private void OnStart(HttpListenerRequest req, HttpListenerResponse resp)
{
if (req.HttpMethod != "POST")
{
resp.Write405();
return;
}
if (!req.HasEntityBody)
{
resp.Write400();
return;
}
CnCStartRequest args;
using (var ist = new StreamReader(req.InputStream))
{
var body = ist.ReadToEnd();
args = body.FromJson<CnCStartRequest>();
}
if (args.job == Guid.Empty)
{
args.job = Guid.NewGuid();
}
if (args.job == Guid.Empty || jobs.ContainsKey(args.job))
{
resp.Write400();
return;
}
if (args.type == "http")
{
if (args.http == null)
{
resp.Write400();
return;
}
args.http.Tag = args.job;
var http = new HttpLoader(args.http);
jobs[args.job] = http;
lock (jobQueue)
{
jobQueue.Enqueue(http);
}
workerSignal.Set();
//speculatively wait a few seconds, in case the job is done real quick like
Thread.Sleep(2000);
resp.WriteJson(new
{
Location = "/status/" + args.job,
Data = new Result(http)
}, http.Status == Status.Complete ? 200 : 202);
}
else
{
resp.Write400();
return;
}
}
private void OnStatusSpecific(HttpListenerRequest req, HttpListenerResponse resp)
{
if (req.HttpMethod != "GET")
{
resp.Write405();
}
var g = req.Url.AbsolutePath.Substring(8);
Guid job;
if (!Guid.TryParse(g, out job) || !jobs.ContainsKey(job))
{
resp.Write404();
return;
}
resp.WriteJson(new Result(jobs[job]));
}
private void StatusGeneral(HttpListenerRequest req, HttpListenerResponse resp)
{
IList<Guid> queued, comp;
lock (jobQueue)
{
queued = jobQueue.Select(j => j.Tag).ToList();
}
lock (completed)
{
comp = completed.Select(c => c.Tag).ToList();
}
var cur = current;
resp.WriteJson(new
{
enqueued = queued,
current = cur != null ? (Guid?)cur.Tag : (Guid?)null,
completed = comp
});
}
private void OnGetSlave(HttpListenerRequest req, HttpListenerResponse res)
{
res.WriteJson(slaves);
}
