mirror of
https://github.com/valitydev/thrift.git
synced 2024-11-07 02:45:22 +00:00
b587a12a11
Client: .NET Core Patch: Volodymyr Gotra <vgotra@gmail.com> PR #1088, with significant improvements by Jens Geyer <jensg@apache.org> PR #1149 This closes #1088 This closes #1149
557 lines
20 KiB
C#
557 lines
20 KiB
C#
// 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.Security.Authentication;
|
|
using System.Security.Cryptography.X509Certificates;
|
|
using System.Text;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.Extensions.Logging;
|
|
using ThriftAsync.Test;
|
|
using Thrift;
|
|
using Thrift.Collections;
|
|
using Thrift.Protocols;
|
|
using Thrift.Server;
|
|
using Thrift.Transports;
|
|
using Thrift.Transports.Server;
|
|
|
|
namespace Test
|
|
{
|
|
internal class ServerParam
|
|
{
|
|
internal bool useBufferedSockets = false;
|
|
internal bool useFramed = false;
|
|
internal bool useEncryption = false;
|
|
internal bool compact = false;
|
|
internal bool json = false;
|
|
internal int port = 9090;
|
|
internal string pipe = null;
|
|
|
|
internal void Parse(List<string> args)
|
|
{
|
|
for (var i = 0; i < args.Count; i++)
|
|
{
|
|
if (args[i].StartsWith("--pipe="))
|
|
{
|
|
pipe = args[i].Substring(args[i].IndexOf("=") + 1);
|
|
}
|
|
else if (args[i].StartsWith("--port="))
|
|
{
|
|
port = int.Parse(args[i].Substring(args[i].IndexOf("=") + 1));
|
|
}
|
|
else if (args[i] == "-b" || args[i] == "--buffered" || args[i] == "--transport=buffered")
|
|
{
|
|
useBufferedSockets = true;
|
|
}
|
|
else if (args[i] == "-f" || args[i] == "--framed" || args[i] == "--transport=framed")
|
|
{
|
|
useFramed = true;
|
|
}
|
|
else if (args[i] == "--compact" || args[i] == "--protocol=compact")
|
|
{
|
|
compact = true;
|
|
}
|
|
else if (args[i] == "--json" || args[i] == "--protocol=json")
|
|
{
|
|
json = true;
|
|
}
|
|
else if (args[i] == "--threaded" || args[i] == "--server-type=threaded")
|
|
{
|
|
throw new NotImplementedException(args[i]);
|
|
}
|
|
else if (args[i] == "--threadpool" || args[i] == "--server-type=threadpool")
|
|
{
|
|
throw new NotImplementedException(args[i]);
|
|
}
|
|
else if (args[i] == "--prototype" || args[i] == "--processor=prototype")
|
|
{
|
|
throw new NotImplementedException(args[i]);
|
|
}
|
|
else if (args[i] == "--ssl")
|
|
{
|
|
useEncryption = true;
|
|
}
|
|
else
|
|
{
|
|
throw new ArgumentException(args[i]);
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
public class TestServer
|
|
{
|
|
public static int _clientID = -1;
|
|
public delegate void TestLogDelegate(string msg, params object[] values);
|
|
|
|
public class MyServerEventHandler : TServerEventHandler
|
|
{
|
|
public int callCount = 0;
|
|
|
|
public Task PreServeAsync(CancellationToken cancellationToken)
|
|
{
|
|
callCount++;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task<object> CreateContextAsync(TProtocol input, TProtocol output, CancellationToken cancellationToken)
|
|
{
|
|
callCount++;
|
|
return Task.FromResult<object>(null);
|
|
}
|
|
|
|
public Task DeleteContextAsync(object serverContext, TProtocol input, TProtocol output, CancellationToken cancellationToken)
|
|
{
|
|
callCount++;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task ProcessContextAsync(object serverContext, TClientTransport transport, CancellationToken cancellationToken)
|
|
{
|
|
callCount++;
|
|
return Task.CompletedTask;
|
|
}
|
|
};
|
|
|
|
public class TestHandlerAsync : ThriftAsync.Test.ThriftTest.IAsync
|
|
{
|
|
public TBaseServer server { get; set; }
|
|
private int handlerID;
|
|
private StringBuilder sb = new StringBuilder();
|
|
private TestLogDelegate logger;
|
|
|
|
public TestHandlerAsync()
|
|
{
|
|
handlerID = Interlocked.Increment(ref _clientID);
|
|
logger += testConsoleLogger;
|
|
logger.Invoke("New TestHandler instance created");
|
|
}
|
|
|
|
public void testConsoleLogger(string msg, params object[] values)
|
|
{
|
|
sb.Clear();
|
|
sb.AppendFormat("handler{0:D3}:", handlerID);
|
|
sb.AppendFormat(msg, values);
|
|
sb.AppendLine();
|
|
Console.Write(sb.ToString());
|
|
}
|
|
|
|
public Task testVoidAsync(CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testVoid()");
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task<string> testStringAsync(string thing, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testString({0})", thing);
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<bool> testBoolAsync(bool thing, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testBool({0})", thing);
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<sbyte> testByteAsync(sbyte thing, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testByte({0})", thing);
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<int> testI32Async(int thing, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testI32({0})", thing);
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<long> testI64Async(long thing, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testI64({0})", thing);
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<double> testDoubleAsync(double thing, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testDouble({0})", thing);
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<byte[]> testBinaryAsync(byte[] thing, CancellationToken cancellationToken)
|
|
{
|
|
var hex = BitConverter.ToString(thing).Replace("-", string.Empty);
|
|
logger.Invoke("testBinary({0:X})", hex);
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<Xtruct> testStructAsync(Xtruct thing, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testStruct({{\"{0}\", {1}, {2}, {3}}})", thing.String_thing, thing.Byte_thing, thing.I32_thing, thing.I64_thing);
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<Xtruct2> testNestAsync(Xtruct2 nest, CancellationToken cancellationToken)
|
|
{
|
|
var thing = nest.Struct_thing;
|
|
logger.Invoke("testNest({{{0}, {{\"{1}\", {2}, {3}, {4}, {5}}}}})",
|
|
nest.Byte_thing,
|
|
thing.String_thing,
|
|
thing.Byte_thing,
|
|
thing.I32_thing,
|
|
thing.I64_thing,
|
|
nest.I32_thing);
|
|
return Task.FromResult(nest);
|
|
}
|
|
|
|
public Task<Dictionary<int, int>> testMapAsync(Dictionary<int, int> thing, CancellationToken cancellationToken)
|
|
{
|
|
sb.Clear();
|
|
sb.Append("testMap({{");
|
|
var first = true;
|
|
foreach (var key in thing.Keys)
|
|
{
|
|
if (first)
|
|
{
|
|
first = false;
|
|
}
|
|
else
|
|
{
|
|
sb.Append(", ");
|
|
}
|
|
sb.AppendFormat("{0} => {1}", key, thing[key]);
|
|
}
|
|
sb.Append("}})");
|
|
logger.Invoke(sb.ToString());
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<Dictionary<string, string>> testStringMapAsync(Dictionary<string, string> thing, CancellationToken cancellationToken)
|
|
{
|
|
sb.Clear();
|
|
sb.Append("testStringMap({{");
|
|
var first = true;
|
|
foreach (var key in thing.Keys)
|
|
{
|
|
if (first)
|
|
{
|
|
first = false;
|
|
}
|
|
else
|
|
{
|
|
sb.Append(", ");
|
|
}
|
|
sb.AppendFormat("{0} => {1}", key, thing[key]);
|
|
}
|
|
sb.Append("}})");
|
|
logger.Invoke(sb.ToString());
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<THashSet<int>> testSetAsync(THashSet<int> thing, CancellationToken cancellationToken)
|
|
{
|
|
sb.Clear();
|
|
sb.Append("testSet({{");
|
|
var first = true;
|
|
foreach (int elem in thing)
|
|
{
|
|
if (first)
|
|
{
|
|
first = false;
|
|
}
|
|
else
|
|
{
|
|
sb.Append(", ");
|
|
}
|
|
sb.AppendFormat("{0}", elem);
|
|
}
|
|
sb.Append("}})");
|
|
logger.Invoke(sb.ToString());
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<List<int>> testListAsync(List<int> thing, CancellationToken cancellationToken)
|
|
{
|
|
sb.Clear();
|
|
sb.Append("testList({{");
|
|
var first = true;
|
|
foreach (var elem in thing)
|
|
{
|
|
if (first)
|
|
{
|
|
first = false;
|
|
}
|
|
else
|
|
{
|
|
sb.Append(", ");
|
|
}
|
|
sb.AppendFormat("{0}", elem);
|
|
}
|
|
sb.Append("}})");
|
|
logger.Invoke(sb.ToString());
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<Numberz> testEnumAsync(Numberz thing, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testEnum({0})", thing);
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<long> testTypedefAsync(long thing, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testTypedef({0})", thing);
|
|
return Task.FromResult(thing);
|
|
}
|
|
|
|
public Task<Dictionary<int, Dictionary<int, int>>> testMapMapAsync(int hello, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testMapMap({0})", hello);
|
|
var mapmap = new Dictionary<int, Dictionary<int, int>>();
|
|
|
|
var pos = new Dictionary<int, int>();
|
|
var neg = new Dictionary<int, int>();
|
|
for (var i = 1; i < 5; i++)
|
|
{
|
|
pos[i] = i;
|
|
neg[-i] = -i;
|
|
}
|
|
|
|
mapmap[4] = pos;
|
|
mapmap[-4] = neg;
|
|
|
|
return Task.FromResult(mapmap);
|
|
}
|
|
|
|
public Task<Dictionary<long, Dictionary<Numberz, Insanity>>> testInsanityAsync(Insanity argument, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testInsanity()");
|
|
|
|
/** from ThriftTest.thrift:
|
|
* So you think you've got this all worked, out eh?
|
|
*
|
|
* Creates a the returned map with these values and prints it out:
|
|
* { 1 => { 2 => argument,
|
|
* 3 => argument,
|
|
* },
|
|
* 2 => { 6 => <empty Insanity struct>, },
|
|
* }
|
|
* @return map<UserId, map<Numberz,Insanity>> - a map with the above values
|
|
*/
|
|
|
|
var first_map = new Dictionary<Numberz, Insanity>();
|
|
var second_map = new Dictionary<Numberz, Insanity>(); ;
|
|
|
|
first_map[Numberz.TWO] = argument;
|
|
first_map[Numberz.THREE] = argument;
|
|
|
|
second_map[Numberz.SIX] = new Insanity();
|
|
|
|
var insane = new Dictionary<long, Dictionary<Numberz, Insanity>>
|
|
{
|
|
[1] = first_map,
|
|
[2] = second_map
|
|
};
|
|
|
|
return Task.FromResult(insane);
|
|
}
|
|
|
|
public Task<Xtruct> testMultiAsync(sbyte arg0, int arg1, long arg2, Dictionary<short, string> arg3, Numberz arg4, long arg5,
|
|
CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testMulti()");
|
|
|
|
var hello = new Xtruct(); ;
|
|
hello.String_thing = "Hello2";
|
|
hello.Byte_thing = arg0;
|
|
hello.I32_thing = arg1;
|
|
hello.I64_thing = arg2;
|
|
return Task.FromResult(hello);
|
|
}
|
|
|
|
public Task testExceptionAsync(string arg, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testException({0})", arg);
|
|
if (arg == "Xception")
|
|
{
|
|
var x = new Xception
|
|
{
|
|
ErrorCode = 1001,
|
|
Message = arg
|
|
};
|
|
throw x;
|
|
}
|
|
if (arg == "TException")
|
|
{
|
|
throw new TException();
|
|
}
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task<Xtruct> testMultiExceptionAsync(string arg0, string arg1, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testMultiException({0}, {1})", arg0, arg1);
|
|
if (arg0 == "Xception")
|
|
{
|
|
var x = new Xception
|
|
{
|
|
ErrorCode = 1001,
|
|
Message = "This is an Xception"
|
|
};
|
|
throw x;
|
|
}
|
|
|
|
if (arg0 == "Xception2")
|
|
{
|
|
var x = new Xception2
|
|
{
|
|
ErrorCode = 2002,
|
|
Struct_thing = new Xtruct { String_thing = "This is an Xception2" }
|
|
};
|
|
throw x;
|
|
}
|
|
|
|
var result = new Xtruct { String_thing = arg1 };
|
|
return Task.FromResult(result);
|
|
}
|
|
|
|
public Task testOnewayAsync(int secondsToSleep, CancellationToken cancellationToken)
|
|
{
|
|
logger.Invoke("testOneway({0}), sleeping...", secondsToSleep);
|
|
Thread.Sleep(secondsToSleep * 1000);
|
|
logger.Invoke("testOneway finished");
|
|
|
|
return Task.CompletedTask;
|
|
}
|
|
}
|
|
|
|
|
|
private enum ProcessorFactoryType
|
|
{
|
|
TSingletonProcessorFactory,
|
|
TPrototypeProcessorFactory,
|
|
}
|
|
|
|
internal static void PrintOptionsHelp()
|
|
{
|
|
Console.WriteLine("Server options:");
|
|
Console.WriteLine(" --pipe=<pipe name>");
|
|
Console.WriteLine(" --port=<port number>");
|
|
Console.WriteLine(" --transport=<transport name> one of buffered,framed (defaults to none)");
|
|
Console.WriteLine(" --protocol=<protocol name> one of compact,json (defaults to binary)");
|
|
Console.WriteLine(" --server-type=<type> one of threaded,threadpool (defaults to simple)");
|
|
Console.WriteLine(" --processor=<prototype>");
|
|
Console.WriteLine(" --ssl");
|
|
Console.WriteLine();
|
|
}
|
|
|
|
public static int Execute(List<string> args)
|
|
{
|
|
var logger = new LoggerFactory().CreateLogger("Test");
|
|
|
|
try
|
|
{
|
|
var param = new ServerParam();
|
|
|
|
try
|
|
{
|
|
param.Parse(args);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Console.WriteLine("*** FAILED ***");
|
|
Console.WriteLine("Error while parsing arguments");
|
|
Console.WriteLine(ex.Message + " ST: " + ex.StackTrace);
|
|
return 1;
|
|
}
|
|
|
|
|
|
// Transport
|
|
TServerTransport trans;
|
|
if (param.pipe != null)
|
|
{
|
|
trans = new TNamedPipeServerTransport(param.pipe);
|
|
}
|
|
else
|
|
{
|
|
if (param.useEncryption)
|
|
{
|
|
var certPath = "../keys/server.p12";
|
|
trans = new TTlsServerSocketTransport(param.port, param.useBufferedSockets, new X509Certificate2(certPath, "thrift"), null, null, SslProtocols.Tls12);
|
|
}
|
|
else
|
|
{
|
|
trans = new TServerSocketTransport(param.port, 0, param.useBufferedSockets);
|
|
}
|
|
}
|
|
|
|
ITProtocolFactory proto;
|
|
if (param.compact)
|
|
proto = new TCompactProtocol.Factory();
|
|
else if (param.json)
|
|
proto = new TJsonProtocol.Factory();
|
|
else
|
|
proto = new TBinaryProtocol.Factory();
|
|
|
|
ITProcessorFactory processorFactory;
|
|
|
|
// Processor
|
|
var testHandler = new TestHandlerAsync();
|
|
var testProcessor = new ThriftAsync.Test.ThriftTest.AsyncProcessor(testHandler);
|
|
processorFactory = new SingletonTProcessorFactory(testProcessor);
|
|
|
|
|
|
TTransportFactory transFactory;
|
|
if (param.useFramed)
|
|
throw new NotImplementedException("framed"); // transFactory = new TFramedTransport.Factory();
|
|
else
|
|
transFactory = new TTransportFactory();
|
|
|
|
TBaseServer serverEngine = new AsyncBaseServer(processorFactory, trans, transFactory, transFactory, proto, proto, logger);
|
|
|
|
//Server event handler
|
|
var serverEvents = new MyServerEventHandler();
|
|
serverEngine.SetEventHandler(serverEvents);
|
|
|
|
// Run it
|
|
var where = (! string.IsNullOrEmpty(param.pipe)) ? "on pipe " + param.pipe : "on port " + param.port;
|
|
Console.WriteLine("Starting the AsyncBaseServer " + where +
|
|
" with processor TPrototypeProcessorFactory prototype factory " +
|
|
(param.useBufferedSockets ? " with buffered socket" : "") +
|
|
(param.useFramed ? " with framed transport" : "") +
|
|
(param.useEncryption ? " with encryption" : "") +
|
|
(param.compact ? " with compact protocol" : "") +
|
|
(param.json ? " with json protocol" : "") +
|
|
"...");
|
|
serverEngine.ServeAsync(CancellationToken.None).GetAwaiter().GetResult();
|
|
Console.ReadLine();
|
|
}
|
|
catch (Exception x)
|
|
{
|
|
Console.Error.Write(x);
|
|
return 1;
|
|
}
|
|
Console.WriteLine("done.");
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
}
|