public static IServiceHostClientBuilder UseDotNettyForTransfer(this IServiceHostClientBuilder serviceHostBuilder)
{
serviceHostBuilder.AddInitializer(container =>
{
var factory = container.Resolve <ITransportClientFactory>();
var logger = container.Resolve <ILogger>();
var serializer = container.Resolve <ISerializer>();
var bootstrap = new Bootstrap();
logger.Info($"[config]use dotnetty for transfer");
bootstrap
.Group(new MultithreadEventLoopGroup())
.Channel <TcpSocketChannel>()
.Option(ChannelOption.TcpNodelay, true)
.Handler(new ActionChannelInitializer <IChannel>(channel =>
{
var pipeline = channel.Pipeline;
pipeline.AddLast(new LengthFieldPrepender(4));
pipeline.AddLast(new LengthFieldBasedFrameDecoder(int.MaxValue, 0, 4, 0, 4));
pipeline.AddLast(new ReadClientMessageChannelHandlerAdapter(serializer, logger));
pipeline.AddLast(new ClientHandlerChannelHandlerAdapter(factory, logger));
}));
AttributeKey <IClientSender> clientSenderKey = AttributeKey <IClientSender> .ValueOf(typeof(DefaultTransportClientFactory), nameof(IClientSender));
AttributeKey <IClientListener> clientListenerKey = AttributeKey <IClientListener> .ValueOf(typeof(DefaultTransportClientFactory), nameof(IClientListener));
//AttributeKey<EndPoint> endPointKey = AttributeKey<EndPoint>.ValueOf(typeof(DefaultTransportClientFactory), nameof(EndPoint));
AttributeKey <string> endPointKey = AttributeKey <string> .ValueOf(typeof(DefaultTransportClientFactory), "addresscode");
factory.ClientCreatorDelegate += (JimuAddress address, ref ITransportClient client) =>
{
//if (client == null && address.GetType().IsAssignableFrom(typeof(DotNettyAddress)))
if (client == null && address.ServerFlag == "DotNetty")
{
var ep = address.CreateEndPoint();
var channel = bootstrap.ConnectAsync(ep).Result;
var listener = new DotNettyClientListener();
channel.GetAttribute(clientListenerKey).Set(listener);
var sender = new DotNettyClientSender(channel, serializer);
channel.GetAttribute(clientSenderKey).Set(sender);
channel.GetAttribute(endPointKey).Set($"{address.ServerFlag}-{address.Code}");
client = new DefaultTransportClient(listener, sender, logger);
}
};
});
return(serviceHostBuilder);
}
public static IServiceHostClientBuilder UseHttpForTransfer(this IServiceHostClientBuilder serviceHostBuilder)
{
serviceHostBuilder.AddInitializer(container =>
{
var factory = container.Resolve <ITransportClientFactory>();
var logger = container.Resolve <ILogger>();
logger.Info($"[config]use http for transfer");
factory.ClientCreatorDelegate += (JimuAddress address, ref ITransportClient client) =>
{
//if (client == null && address.GetType() == typeof(HttpAddress))
if (client == null && address.ServerFlag == "Http")
{
var listener = new HttpClientListener();
var sender = new HttpClientSender(address, listener);
client = new DefaultTransportClient(listener, sender, logger);
}
};
});
return(serviceHostBuilder);
}
