protected override void ExecuteServerCore(
ConnectionHost host,
CompilerHost compilerHost,
CancellationToken cancellationToken,
EventBus eventBus,
TimeSpan?keepAlive = null)
{
base.ExecuteServerCore(
_host ?? host,
_compilerHost ?? compilerHost,
_cancellationToken,
_eventBus ?? eventBus,
_keepAlive ?? keepAlive);
}
internal static ServerData CreateServer(
string pipeName = null,
CompilerHost compilerHost = null,
ConnectionHost connectionHost = null,
Action <object, EventArgs> onListening = null)
{
pipeName = pipeName ?? Guid.NewGuid().ToString();
compilerHost = compilerHost ?? CompilerHost.Create();
connectionHost = connectionHost ?? ConnectionHost.Create(pipeName);
var serverStatsSource = new TaskCompletionSource <ServerStats>();
var serverListenSource = new TaskCompletionSource <bool>();
var cts = new CancellationTokenSource();
var mutexName = MutexName.GetServerMutexName(pipeName);
var thread = new Thread(_ =>
{
var eventBus = new TestableEventBus();
eventBus.Listening += (sender, e) => { serverListenSource.TrySetResult(true); };
if (onListening != null)
{
eventBus.Listening += (sender, e) => onListening(sender, e);
}
try
{
RunServer(
pipeName,
connectionHost,
compilerHost,
cts.Token,
eventBus,
Timeout.InfiniteTimeSpan);
}
finally
{
var serverStats = new ServerStats(connections: eventBus.ConnectionCount, completedConnections: eventBus.CompletedCount);
serverStatsSource.SetResult(serverStats);
}
});
thread.Start();
// The contract of this function is that it will return once the server has started. Spin here until
// we can verify the server has started or simply failed to start.
while (ServerConnection.WasServerMutexOpen(mutexName) != true && thread.IsAlive)
{
Thread.Yield();
}
return(new ServerData(cts, pipeName, serverStatsSource.Task, serverListenSource.Task));
}
public TestableServerCommand(
ConnectionHost host,
CompilerHost compilerHost,
CancellationToken ct,
EventBus eventBus,
TimeSpan?keepAlive)
: base(new Application(ct, Mock.Of <ExtensionAssemblyLoader>(), Mock.Of <ExtensionDependencyChecker>(), (path, properties) => Mock.Of <PortableExecutableReference>()))
{
_host = host;
_compilerHost = compilerHost;
_cancellationToken = ct;
_eventBus = eventBus;
_keepAlive = keepAlive;
}
internal static int RunServer(
string pipeName,
ConnectionHost host,
CompilerHost compilerHost,
CancellationToken cancellationToken = default,
EventBus eventBus = null,
TimeSpan?keepAlive = null)
{
var command = new TestableServerCommand(host, compilerHost, cancellationToken, eventBus, keepAlive);
var args = new List <string>
{
"-p",
pipeName
};
var result = command.Execute(args.ToArray());
return(result);
}