mirror of
https://github.com/microsoft/autogen.git
synced 2025-11-14 17:13:29 +00:00
edbd20167b
Restoring the grpc + Orleans server into the project ## Why are these changes needed? This is the distributed agent runtime for .NET that can manage routing messages amongst a fleet of grpc agent runtimes. ## Related issue number ## Checks - [ ] I've included any doc changes needed for https://microsoft.github.io/autogen/. See https://microsoft.github.io/autogen/docs/Contribute#documentation to build and test documentation locally. - [x] I've added tests (if relevant) corresponding to the changes introduced in this PR. - [x] I've made sure all auto checks have passed. --------- Co-authored-by: Jack Gerrits <jack@jackgerrits.com> Co-authored-by: Jacob Alber <jaalber@microsoft.com>
87 lines
2.3 KiB
C#
87 lines
2.3 KiB
C#
#pragma warning disable IDE0073
|
|
// Copyright 2019 The gRPC Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
using System.Threading.Channels;
|
|
using Grpc.Core;
|
|
|
|
namespace Microsoft.AutoGen.RuntimeGateway.Grpc.Tests.Helpers.Grpc;
|
|
|
|
public class TestServerStreamWriter<T> : IDisposable, IServerStreamWriter<T> where T : class
|
|
{
|
|
private readonly ServerCallContext _serverCallContext;
|
|
private readonly Channel<T> _channel;
|
|
|
|
public WriteOptions? WriteOptions { get; set; }
|
|
|
|
public TestServerStreamWriter(ServerCallContext serverCallContext)
|
|
{
|
|
_channel = Channel.CreateUnbounded<T>();
|
|
|
|
_serverCallContext = serverCallContext;
|
|
}
|
|
|
|
public void Complete()
|
|
{
|
|
_channel.Writer.Complete();
|
|
}
|
|
|
|
public IAsyncEnumerable<T> ReadAllAsync()
|
|
{
|
|
return _channel.Reader.ReadAllAsync();
|
|
}
|
|
|
|
public async Task<T?> ReadNextAsync()
|
|
{
|
|
if (await _channel.Reader.WaitToReadAsync())
|
|
{
|
|
_channel.Reader.TryRead(out var message);
|
|
return message;
|
|
}
|
|
else
|
|
{
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public Task WriteAsync(T message, CancellationToken cancellationToken)
|
|
{
|
|
if (cancellationToken.IsCancellationRequested)
|
|
{
|
|
return Task.FromCanceled(cancellationToken);
|
|
}
|
|
if (_serverCallContext.CancellationToken.IsCancellationRequested)
|
|
{
|
|
return Task.FromCanceled(_serverCallContext.CancellationToken);
|
|
}
|
|
|
|
if (!_channel.Writer.TryWrite(message))
|
|
{
|
|
throw new InvalidOperationException("Unable to write message.");
|
|
}
|
|
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task WriteAsync(T message)
|
|
{
|
|
return WriteAsync(message, CancellationToken.None);
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
Complete();
|
|
}
|
|
}
|