2019-01-28 23:27:45 +01:00
|
|
|
// Licensed to the Apache Software Foundation(ASF) under one
|
|
|
|
|
// or more contributor license agreements.See the NOTICE file
|
|
|
|
|
// distributed with this work for additional information
|
|
|
|
|
// regarding copyright ownership.The ASF licenses this file
|
|
|
|
|
// to you 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;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Net.Security;
|
|
|
|
|
using System.Security.Cryptography.X509Certificates;
|
|
|
|
|
using System.Threading;
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
using Microsoft.AspNetCore.Builder;
|
|
|
|
|
using Microsoft.AspNetCore.Hosting;
|
|
|
|
|
using Microsoft.Extensions.Configuration;
|
|
|
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
|
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
|
using Thrift;
|
|
|
|
|
using Thrift.Protocol;
|
|
|
|
|
using Thrift.Server;
|
|
|
|
|
using Thrift.Transport;
|
|
|
|
|
using Thrift.Transport.Server;
|
|
|
|
|
using tutorial;
|
|
|
|
|
using shared;
|
|
|
|
|
using Thrift.Processor;
|
|
|
|
|
|
|
|
|
|
namespace Server
|
|
|
|
|
{
|
|
|
|
|
public class Program
|
|
|
|
|
{
|
|
|
|
|
private static readonly ILogger Logger = new LoggerFactory().AddConsole(LogLevel.Trace).AddDebug(LogLevel.Trace).CreateLogger(nameof(Server));
|
|
|
|
|
|
|
|
|
|
public static void Main(string[] args)
|
|
|
|
|
{
|
|
|
|
|
args = args ?? new string[0];
|
|
|
|
|
|
|
|
|
|
if (args.Any(x => x.StartsWith("-help", StringComparison.OrdinalIgnoreCase)))
|
|
|
|
|
{
|
|
|
|
|
DisplayHelp();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
using (var source = new CancellationTokenSource())
|
|
|
|
|
{
|
|
|
|
|
RunAsync(args, source.Token).GetAwaiter().GetResult();
|
|
|
|
|
|
|
|
|
|
Logger.LogInformation("Press any key to stop...");
|
|
|
|
|
|
|
|
|
|
Console.ReadLine();
|
|
|
|
|
source.Cancel();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Logger.LogInformation("Server stopped");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static void DisplayHelp()
|
|
|
|
|
{
|
|
|
|
|
Logger.LogInformation(@"
|
|
|
|
|
Usage:
|
|
|
|
|
Server.exe -help
|
|
|
|
|
will diplay help information
|
|
|
|
|
|
|
|
|
|
Server.exe -tr:<transport> -pr:<protocol>
|
|
|
|
|
will run server with specified arguments (tcp transport and binary protocol by default)
|
|
|
|
|
|
|
|
|
|
Options:
|
|
|
|
|
-tr (transport):
|
|
|
|
|
tcp - (default) tcp transport will be used (host - ""localhost"", port - 9090)
|
|
|
|
|
tcpbuffered - tcp buffered transport will be used (host - ""localhost"", port - 9090)
|
|
|
|
|
namedpipe - namedpipe transport will be used (pipe address - "".test"")
|
|
|
|
|
http - http transport will be used (http address - ""localhost:9090"")
|
|
|
|
|
tcptls - tcp transport with tls will be used (host - ""localhost"", port - 9090)
|
|
|
|
|
framed - tcp framed transport will be used (host - ""localhost"", port - 9090)
|
|
|
|
|
|
|
|
|
|
-pr (protocol):
|
|
|
|
|
binary - (default) binary protocol will be used
|
|
|
|
|
compact - compact protocol will be used
|
|
|
|
|
json - json protocol will be used
|
|
|
|
|
multiplexed - multiplexed protocol will be used
|
|
|
|
|
|
|
|
|
|
Sample:
|
|
|
|
|
Server.exe -tr:tcp
|
|
|
|
|
");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static async Task RunAsync(string[] args, CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
var selectedTransport = GetTransport(args);
|
|
|
|
|
var selectedProtocol = GetProtocol(args);
|
|
|
|
|
|
|
|
|
|
if (selectedTransport == Transport.Http)
|
|
|
|
|
{
|
|
|
|
|
new HttpServerSample().Run(cancellationToken);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
await RunSelectedConfigurationAsync(selectedTransport, selectedProtocol, cancellationToken);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static Protocol GetProtocol(string[] args)
|
|
|
|
|
{
|
|
|
|
|
var transport = args.FirstOrDefault(x => x.StartsWith("-pr"))?.Split(':')?[1];
|
|
|
|
|
|
|
|
|
|
Enum.TryParse(transport, true, out Protocol selectedProtocol);
|
|
|
|
|
|
|
|
|
|
return selectedProtocol;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static Transport GetTransport(string[] args)
|
|
|
|
|
{
|
|
|
|
|
var transport = args.FirstOrDefault(x => x.StartsWith("-tr"))?.Split(':')?[1];
|
|
|
|
|
|
|
|
|
|
Enum.TryParse(transport, true, out Transport selectedTransport);
|
|
|
|
|
|
|
|
|
|
return selectedTransport;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static async Task RunSelectedConfigurationAsync(Transport transport, Protocol protocol, CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
var fabric = new LoggerFactory().AddConsole(LogLevel.Trace).AddDebug(LogLevel.Trace);
|
|
|
|
|
var handler = new CalculatorAsyncHandler();
|
|
|
|
|
ITAsyncProcessor processor = null;
|
|
|
|
|
|
|
|
|
|
TServerTransport serverTransport = null;
|
|
|
|
|
|
|
|
|
|
switch (transport)
|
|
|
|
|
{
|
|
|
|
|
case Transport.Tcp:
|
|
|
|
|
serverTransport = new TServerSocketTransport(9090);
|
|
|
|
|
break;
|
|
|
|
|
case Transport.TcpBuffered:
|
2019-03-14 21:42:16 +01:00
|
|
|
serverTransport = new TServerSocketTransport(port: 9090, clientTimeout: 10000, buffering: Buffering.BufferedTransport);
|
2019-01-28 23:27:45 +01:00
|
|
|
break;
|
|
|
|
|
case Transport.NamedPipe:
|
|
|
|
|
serverTransport = new TNamedPipeServerTransport(".test");
|
|
|
|
|
break;
|
|
|
|
|
case Transport.TcpTls:
|
2019-03-14 21:42:16 +01:00
|
|
|
serverTransport = new TTlsServerSocketTransport(9090, Buffering.None, GetCertificate(), ClientCertValidator, LocalCertificateSelectionCallback);
|
2019-01-28 23:27:45 +01:00
|
|
|
break;
|
|
|
|
|
case Transport.Framed:
|
|
|
|
|
serverTransport = new TServerFramedTransport(9090);
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ITProtocolFactory inputProtocolFactory;
|
|
|
|
|
ITProtocolFactory outputProtocolFactory;
|
|
|
|
|
|
|
|
|
|
switch (protocol)
|
|
|
|
|
{
|
|
|
|
|
case Protocol.Binary:
|
|
|
|
|
{
|
|
|
|
|
inputProtocolFactory = new TBinaryProtocol.Factory();
|
|
|
|
|
outputProtocolFactory = new TBinaryProtocol.Factory();
|
|
|
|
|
processor = new Calculator.AsyncProcessor(handler);
|
|
|
|
|
}
|
|
|
|
|
break;
|
|
|
|
|
case Protocol.Compact:
|
|
|
|
|
{
|
|
|
|
|
inputProtocolFactory = new TCompactProtocol.Factory();
|
|
|
|
|
outputProtocolFactory = new TCompactProtocol.Factory();
|
|
|
|
|
processor = new Calculator.AsyncProcessor(handler);
|
|
|
|
|
}
|
|
|
|
|
break;
|
|
|
|
|
case Protocol.Json:
|
|
|
|
|
{
|
|
|
|
|
inputProtocolFactory = new TJsonProtocol.Factory();
|
|
|
|
|
outputProtocolFactory = new TJsonProtocol.Factory();
|
|
|
|
|
processor = new Calculator.AsyncProcessor(handler);
|
|
|
|
|
}
|
|
|
|
|
break;
|
|
|
|
|
case Protocol.Multiplexed:
|
|
|
|
|
{
|
|
|
|
|
inputProtocolFactory = new TBinaryProtocol.Factory();
|
|
|
|
|
outputProtocolFactory = new TBinaryProtocol.Factory();
|
|
|
|
|
|
|
|
|
|
var calcHandler = new CalculatorAsyncHandler();
|
|
|
|
|
var calcProcessor = new Calculator.AsyncProcessor(calcHandler);
|
|
|
|
|
|
|
|
|
|
var sharedServiceHandler = new SharedServiceAsyncHandler();
|
|
|
|
|
var sharedServiceProcessor = new SharedService.AsyncProcessor(sharedServiceHandler);
|
|
|
|
|
|
|
|
|
|
var multiplexedProcessor = new TMultiplexedProcessor();
|
|
|
|
|
multiplexedProcessor.RegisterProcessor(nameof(Calculator), calcProcessor);
|
|
|
|
|
multiplexedProcessor.RegisterProcessor(nameof(SharedService), sharedServiceProcessor);
|
|
|
|
|
|
|
|
|
|
processor = multiplexedProcessor;
|
|
|
|
|
}
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
throw new ArgumentOutOfRangeException(nameof(protocol), protocol, null);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
Logger.LogInformation(
|
|
|
|
|
$"Selected TAsyncServer with {serverTransport} transport, {processor} processor and {inputProtocolFactory} protocol factories");
|
|
|
|
|
|
|
|
|
|
var server = new TSimpleAsyncServer(processor, serverTransport, inputProtocolFactory, outputProtocolFactory, fabric);
|
|
|
|
|
|
|
|
|
|
Logger.LogInformation("Starting the server...");
|
|
|
|
|
await server.ServeAsync(cancellationToken);
|
|
|
|
|
}
|
|
|
|
|
catch (Exception x)
|
|
|
|
|
{
|
|
|
|
|
Logger.LogInformation(x.ToString());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static X509Certificate2 GetCertificate()
|
|
|
|
|
{
|
|
|
|
|
// due to files location in net core better to take certs from top folder
|
|
|
|
|
var certFile = GetCertPath(Directory.GetParent(Directory.GetCurrentDirectory()));
|
|
|
|
|
return new X509Certificate2(certFile, "ThriftTest");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static string GetCertPath(DirectoryInfo di, int maxCount = 6)
|
|
|
|
|
{
|
|
|
|
|
var topDir = di;
|
|
|
|
|
var certFile =
|
|
|
|
|
topDir.EnumerateFiles("ThriftTest.pfx", SearchOption.AllDirectories)
|
|
|
|
|
.FirstOrDefault();
|
|
|
|
|
if (certFile == null)
|
|
|
|
|
{
|
|
|
|
|
if (maxCount == 0)
|
|
|
|
|
throw new FileNotFoundException("Cannot find file in directories");
|
|
|
|
|
return GetCertPath(di.Parent, maxCount - 1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return certFile.FullName;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static X509Certificate LocalCertificateSelectionCallback(object sender,
|
|
|
|
|
string targetHost, X509CertificateCollection localCertificates,
|
|
|
|
|
X509Certificate remoteCertificate, string[] acceptableIssuers)
|
|
|
|
|
{
|
|
|
|
|
return GetCertificate();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static bool ClientCertValidator(object sender, X509Certificate certificate,
|
|
|
|
|
X509Chain chain, SslPolicyErrors sslPolicyErrors)
|
|
|
|
|
{
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private enum Transport
|
|
|
|
|
{
|
|
|
|
|
Tcp,
|
|
|
|
|
TcpBuffered,
|
|
|
|
|
NamedPipe,
|
|
|
|
|
Http,
|
|
|
|
|
TcpTls,
|
|
|
|
|
Framed
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private enum Protocol
|
|
|
|
|
{
|
|
|
|
|
Binary,
|
|
|
|
|
Compact,
|
|
|
|
|
Json,
|
|
|
|
|
Multiplexed
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class HttpServerSample
|
|
|
|
|
{
|
|
|
|
|
public void Run(CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
var config = new ConfigurationBuilder()
|
|
|
|
|
.AddEnvironmentVariables(prefix: "ASPNETCORE_")
|
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
|
|
var host = new WebHostBuilder()
|
|
|
|
|
.UseConfiguration(config)
|
|
|
|
|
.UseKestrel()
|
|
|
|
|
.UseUrls("http://localhost:9090")
|
|
|
|
|
.UseContentRoot(Directory.GetCurrentDirectory())
|
|
|
|
|
.UseStartup<Startup>()
|
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
|
|
host.RunAsync(cancellationToken).GetAwaiter().GetResult();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class Startup
|
|
|
|
|
{
|
|
|
|
|
public Startup(IHostingEnvironment env)
|
|
|
|
|
{
|
|
|
|
|
var builder = new ConfigurationBuilder()
|
|
|
|
|
.SetBasePath(env.ContentRootPath)
|
|
|
|
|
.AddEnvironmentVariables();
|
|
|
|
|
|
|
|
|
|
Configuration = builder.Build();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public IConfigurationRoot Configuration { get; }
|
|
|
|
|
|
|
|
|
|
// This method gets called by the runtime. Use this method to add services to the container.
|
|
|
|
|
public void ConfigureServices(IServiceCollection services)
|
|
|
|
|
{
|
|
|
|
|
services.AddTransient<Calculator.IAsync, CalculatorAsyncHandler>();
|
|
|
|
|
services.AddTransient<ITAsyncProcessor, Calculator.AsyncProcessor>();
|
|
|
|
|
services.AddTransient<THttpServerTransport, THttpServerTransport>();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// This method gets called by the runtime. Use this method to configure the HTTP request pipeline.
|
|
|
|
|
public void Configure(IApplicationBuilder app, IHostingEnvironment env,
|
|
|
|
|
ILoggerFactory loggerFactory)
|
|
|
|
|
{
|
|
|
|
|
app.UseMiddleware<THttpServerTransport>();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class CalculatorAsyncHandler : Calculator.IAsync
|
|
|
|
|
{
|
|
|
|
|
private readonly Dictionary<int, SharedStruct> _log = new Dictionary<int, SharedStruct>();
|
|
|
|
|
|
|
|
|
|
public CalculatorAsyncHandler()
|
|
|
|
|
{
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public async Task<SharedStruct> getStructAsync(int key,
|
|
|
|
|
CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
Logger.LogInformation("GetStructAsync({0})", key);
|
|
|
|
|
return await Task.FromResult(_log[key]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public async Task pingAsync(CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
Logger.LogInformation("PingAsync()");
|
|
|
|
|
await Task.CompletedTask;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public async Task<int> addAsync(int num1, int num2, CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
Logger.LogInformation($"AddAsync({num1},{num2})");
|
|
|
|
|
return await Task.FromResult(num1 + num2);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public async Task<int> calculateAsync(int logid, Work w, CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
Logger.LogInformation($"CalculateAsync({logid}, [{w.Op},{w.Num1},{w.Num2}])");
|
|
|
|
|
|
|
|
|
|
var val = 0;
|
|
|
|
|
switch (w.Op)
|
|
|
|
|
{
|
|
|
|
|
case Operation.ADD:
|
|
|
|
|
val = w.Num1 + w.Num2;
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
case Operation.SUBTRACT:
|
|
|
|
|
val = w.Num1 - w.Num2;
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
case Operation.MULTIPLY:
|
|
|
|
|
val = w.Num1 * w.Num2;
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
case Operation.DIVIDE:
|
|
|
|
|
if (w.Num2 == 0)
|
|
|
|
|
{
|
|
|
|
|
var io = new InvalidOperation
|
|
|
|
|
{
|
|
|
|
|
WhatOp = (int) w.Op,
|
|
|
|
|
Why = "Cannot divide by 0"
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
throw io;
|
|
|
|
|
}
|
|
|
|
|
val = w.Num1 / w.Num2;
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
default:
|
|
|
|
|
{
|
|
|
|
|
var io = new InvalidOperation
|
|
|
|
|
{
|
|
|
|
|
WhatOp = (int) w.Op,
|
|
|
|
|
Why = "Unknown operation"
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
throw io;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var entry = new SharedStruct
|
|
|
|
|
{
|
|
|
|
|
Key = logid,
|
|
|
|
|
Value = val.ToString()
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
_log[logid] = entry;
|
|
|
|
|
|
|
|
|
|
return await Task.FromResult(val);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public async Task zipAsync(CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
Logger.LogInformation("ZipAsync() with delay 100mc");
|
|
|
|
|
await Task.Delay(100, CancellationToken.None);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class SharedServiceAsyncHandler : SharedService.IAsync
|
|
|
|
|
{
|
|
|
|
|
public async Task<SharedStruct> getStructAsync(int key, CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
Logger.LogInformation("GetStructAsync({0})", key);
|
|
|
|
|
return await Task.FromResult(new SharedStruct()
|
|
|
|
|
{
|
|
|
|
|
Key = key,
|
|
|
|
|
Value = "GetStructAsync"
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
