public ProcessTrackerTests()
{
HostProcess = Substitute.For<IProcess>();
JobObject = Substitute.For<JobObject>();
ProcessHelper = Substitute.For<ProcessHelper>();
Transport = Substitute.For<IMessageTransport>();
ProcessTracker = new ProcessTracker(Transport, JobObject, HostProcess, ProcessHelper);
}
static void Main(string[] args)
{
//Debugger.Launch();
if (args.Length == 0)
ExitWithError("Must specify container-id as the first argument.", -1);
containerId = args[0];
hostJobObject = new JobObject(null);
hostProcess = ProcessHelper.WrapProcess(Process.GetCurrentProcess());
var input = Console.In;
var output = Console.Out;
using (var transport = MessageTransport.Create(input, output))
{
processTracker = new ProcessTracker(transport, hostJobObject, hostProcess, new ProcessHelper());
var createProcessHandler = new CreateProcessHandler(new ProcessRunner(), processTracker);
var pingHandler = new PingHandler();
var findProcessByIdHandler = new FindProcessByIdHandler(processTracker);
var stopProcessHandler = new StopProcessHandler(processTracker);
var stopAllProcessesHandler = new StopAllProcessesHandler(processTracker);
var waitForProcessExitHandler = new WaitForProcessExitHandler(processTracker);
var dispatcher = new MessageDispatcher();
dispatcher.RegisterMethod<CreateProcessRequest>(
CreateProcessRequest.MethodName,
async (request) =>
{
var result = await createProcessHandler.ExecuteAsync(request.@params);
return new CreateProcessResponse(request.id, result);
});
dispatcher.RegisterMethod<PingRequest>(
PingRequest.MethodName,
async (request) =>
{
await pingHandler.ExecuteAsync();
return new PingResponse(request.id);
});
dispatcher.RegisterMethod<FindProcessByIdRequest>(
FindProcessByIdRequest.MethodName,
async (request) =>
{
var result = await findProcessByIdHandler.ExecuteAsync(request.@params);
return new FindProcessByIdResponse(request.id, result);
});
dispatcher.RegisterMethod<StopProcessRequest>(
StopProcessRequest.MethodName,
async (request) =>
{
await stopProcessHandler.ExecuteAsync(request.@params);
return new StopProcessResponse(request.id);
});
dispatcher.RegisterMethod<StopAllProcessesRequest>(
StopAllProcessesRequest.MethodName,
async (request) =>
{
await stopAllProcessesHandler.ExecuteAsync(request.@params);
return new StopAllProcessesResponse(request.id);
});
dispatcher.RegisterMethod<WaitForProcessExitRequest>(
WaitForProcessExitRequest.MethodName,
async (request) =>
{
var result = await waitForProcessExitHandler.ExecuteAsync(request.@params);
return new WaitForProcessExitResponse(request.id, result);
});
transport.SubscribeRequest(
async (request) =>
{
var response = await dispatcher.DispatchAsync(request);
await transport.PublishResponseAsync(response);
});
transport.Start();
ReportOk();
exitEvent.WaitOne();
}
}
static void Main(string[] args)
{
//Debugger.Launch();
if (args.Length == 0)
{
ExitWithError("Must specify container-id as the first argument.", -1);
}
containerId = args[0];
hostJobObject = new JobObject(null);
hostProcess = ProcessHelper.WrapProcess(Process.GetCurrentProcess());
var input = Console.In;
var output = Console.Out;
using (var transport = MessageTransport.Create(input, output))
{
processTracker = new ProcessTracker(transport, hostJobObject, hostProcess, new ProcessHelper());
var createProcessHandler = new CreateProcessHandler(new ProcessRunner(), processTracker);
var pingHandler = new PingHandler();
var findProcessByIdHandler = new FindProcessByIdHandler(processTracker);
var stopProcessHandler = new StopProcessHandler(processTracker);
var stopAllProcessesHandler = new StopAllProcessesHandler(processTracker);
var waitForProcessExitHandler = new WaitForProcessExitHandler(processTracker);
var dispatcher = new MessageDispatcher();
dispatcher.RegisterMethod <CreateProcessRequest>(
CreateProcessRequest.MethodName,
async(request) =>
{
var result = await createProcessHandler.ExecuteAsync(request.@params);
return(new CreateProcessResponse(request.id, result));
});
dispatcher.RegisterMethod <PingRequest>(
PingRequest.MethodName,
async(request) =>
{
await pingHandler.ExecuteAsync();
return(new PingResponse(request.id));
});
dispatcher.RegisterMethod <FindProcessByIdRequest>(
FindProcessByIdRequest.MethodName,
async(request) =>
{
var result = await findProcessByIdHandler.ExecuteAsync(request.@params);
return(new FindProcessByIdResponse(request.id, result));
});
dispatcher.RegisterMethod <StopProcessRequest>(
StopProcessRequest.MethodName,
async(request) =>
{
await stopProcessHandler.ExecuteAsync(request.@params);
return(new StopProcessResponse(request.id));
});
dispatcher.RegisterMethod <StopAllProcessesRequest>(
StopAllProcessesRequest.MethodName,
async(request) =>
{
await stopAllProcessesHandler.ExecuteAsync(request.@params);
return(new StopAllProcessesResponse(request.id));
});
dispatcher.RegisterMethod <WaitForProcessExitRequest>(
WaitForProcessExitRequest.MethodName,
async(request) =>
{
var result = await waitForProcessExitHandler.ExecuteAsync(request.@params);
return(new WaitForProcessExitResponse(request.id, result));
});
transport.SubscribeRequest(
async(request) =>
{
var response = await dispatcher.DispatchAsync(request);
await transport.PublishResponseAsync(response);
});
transport.Start();
ReportOk();
exitEvent.WaitOne();
}
}