JobEntryBase.java

/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package com.datascience.service;

import java.lang.reflect.Type;
import java.util.Collection;
import java.util.Properties;
import java.util.logging.Logger;

import javax.servlet.ServletContext;
import javax.ws.rs.*;
import javax.ws.rs.core.*;

import com.datascience.datastoring.jobs.JobsManager;
import com.datascience.datastoring.jobs.JobsLocksManager;
import com.datascience.core.base.LObject;
import com.datascience.core.base.Project;
import com.datascience.core.commands.*;
import com.datascience.core.commands.Utils.ShallowAssign;
import com.datascience.executor.ICommandStatusesContainer;
import com.datascience.scheduler.SchedulerCommands;
import com.datascience.serialization.ISerializer;
import com.datascience.datastoring.jobs.JobCommand;
import com.datascience.executor.ProjectCommandExecutor;

/**
 *
 * @author konrad
 */
public abstract class JobEntryBase<T extends Project> {
	
	ServletContext context;
	Request request;
	UriInfo uriInfo;
	String jid;

	ResponseBuilder responser;
	ProjectCommandExecutor executor;
	ISerializer serializer;
	ICommandStatusesContainer statusesContainer;
	JobsManager jobsManager;
	JobsLocksManager jobsLocksManager;

	Type objectsType;
	Type assignsType;

	public JobEntryBase(ServletContext context, Request request, UriInfo uriInfo, String jid) throws Exception{
		this.context = context;
		this.request = request;
		this.uriInfo = uriInfo;
		this.jid = jid;
		postConstruct();
	}

	protected abstract JobCommand getPredictionZipCommand(String path);

	public void postConstruct() throws Exception{
		jobsManager = (JobsManager) context.getAttribute(Constants.JOBS_MANAGER);
		responser = (ResponseBuilder) context.getAttribute(Constants.RESPONSER);
		executor = (ProjectCommandExecutor) context.getAttribute(Constants.COMMAND_EXECUTOR);
		statusesContainer = (ICommandStatusesContainer) context.getAttribute(Constants.COMMAND_STATUSES_CONTAINER);
		serializer = responser.getSerializer();
		jobsLocksManager = (JobsLocksManager) context.getAttribute(Constants.JOBS_LOCKS_MANAGER);

		Logger.getAnonymousLogger().info(uriInfo.getPath());
	}

	protected Response buildResponseOnCommand(JobCommand command){
		command.setJobId(jid);
		command.setJobsManager(jobsManager);
		RequestExecutorCommand rec = new RequestExecutorCommand(
				statusesContainer.initNewStatus(), command, jobsLocksManager.getLock(jid), statusesContainer);
		executor.add(rec);
		return responser.makeRedirectResponse(String.format("responses/%s/%s/%s", rec.commandId, request.getMethod(), uriInfo.getPath()));
	}

	@Path("prediction/zip")
	@GET
	public Response getPredictionsZip(){
		return buildResponseOnCommand(getPredictionZipCommand(
				((Properties)context.getAttribute(Constants.PROPERTIES)).getProperty(Constants.DOWNLOADS_PATH)));
	}

	@Path("")
	@GET
	public Response getJobInfo(){
		return buildResponseOnCommand(new ProjectCommands.GetProjectInfo());
	}

	@Path("objects")
	@GET
	public Response getObjects(){
		return buildResponseOnCommand(new ObjectCommands.GetObjects());
	}

	@Path("objects/{oid:[a-zA-Z_0-9/:.-]+}/info")
	@GET
	public Response getObject(@PathParam("oid") String objectId){
		return buildResponseOnCommand(new ObjectCommands.GetObject(objectId));
	}

	@Path("objects/{oid:[a-zA-Z_0-9/:.-]+}/assigns")
	@GET
	public Response getObjectAssigns(@PathParam("oid") String objectId){
		return buildResponseOnCommand(new ObjectCommands.GetObjectAssigns(objectId));
	}

	@Consumes(MediaType.APPLICATION_JSON)
	@Path("objects")
	@POST
	public Response addObjects(String json){
		Collection<LObject> objects = serializer.parse(json, objectsType);
		return buildResponseOnCommand(new ObjectCommands.AddObjects(objects));
	}

	@Path("goldObjects")
	@GET
	public Response getGoldObjects(){
		return buildResponseOnCommand(new GoldObjectsCommands.GetGoldObjects());
	}

	@Path("goldObjects/{oid:[a-zA-Z_0-9/:.-]+}")
	@GET
	public Response getGoldObject(@PathParam("oid") String objectId){
		return buildResponseOnCommand(new GoldObjectsCommands.GetGoldObject(objectId));
	}

	@Consumes(MediaType.APPLICATION_JSON)
	@Path("goldObjects")
	@POST
	public Response addGoldObjects(String json){
		Collection<LObject> goldObjects = serializer.parse(json, objectsType);
		return buildResponseOnCommand(new ObjectCommands.AddObjects(goldObjects));
	}

	@Path("evaluationObjects")
	@GET
	public Response getEvaluationObjects(){
		return buildResponseOnCommand(new EvaluationObjectsCommands.GetEvaluationObjects());
	}

	@Path("evaluationObjects/{oid:[a-zA-Z_0-9/:.-]+}")
	@GET
	public Response getEvaluationObject(@PathParam("oid") String objectId){
		return buildResponseOnCommand(new EvaluationObjectsCommands.GetEvaluationObject(objectId));
	}

	@Consumes(MediaType.APPLICATION_JSON)
	@Path("evaluationObjects")
	@POST
	public Response addEvaluationObjects(String json){
		Collection<LObject> objects = serializer.parse(json, objectsType);
		return buildResponseOnCommand(new ObjectCommands.AddObjects(objects));
	}

	@Path("workers")
	@GET
	public Response getWorkers(){
		return buildResponseOnCommand(new WorkerCommands.GetWorkers());
	}

	@Path("workers/{wid:[a-zA-Z_0-9/:.-]+}/info")
	@GET
	public Response getWorker(@PathParam("wid") String worker){
		return buildResponseOnCommand(new WorkerCommands.GetWorker(worker));
	}

	@Path("workers/{wid:[a-zA-Z_0-9/:.-]+}/assigns")
	@GET
	public Response getWorkerAssigns(@PathParam("wid") String worker){
		return buildResponseOnCommand(new AssignsCommands.GetWorkerAssigns<T>(worker));
	}

	@Path("assigns")
	@GET
	public Response getAssigns(){
		return buildResponseOnCommand(new AssignsCommands.GetAssigns<T>());
	}

	@Consumes(MediaType.APPLICATION_JSON)
	@Path("assigns")
	@POST
	public Response addAssigns(String json){
		Collection<ShallowAssign> assigns = serializer.parse(json, assignsType);
		return buildResponseOnCommand(new AssignsCommands.AddAssigns(assigns));
	}

	@Path("compute/")
	@POST
	public Response compute(){
		return buildResponseOnCommand(new ProjectCommands.Compute());
	}

	@Path("nextObject/")
	@GET
	public Response schedulerNextObject(){
		return buildResponseOnCommand(new SchedulerCommands.GetNextObject<T>());
	}

	@Path("nextObject/{wid:[a-zA-Z_0-9/:.-]+}")
	@GET
	public Response schedulerNextObject(@PathParam("wid") String worker){
		return buildResponseOnCommand(new SchedulerCommands.GetNextObjectForWorker(worker));
	}
}