ca.carleton.gcrc.couch.date.DateServlet Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nunaliit2-couch-date Show documentation
There is a newer version: 2.2.7
package ca.carleton.gcrc.couch.date;

import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.util.List;

import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.json.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import ca.carleton.gcrc.couch.date.impl.DateRobotThread;
import ca.carleton.gcrc.couch.date.impl.DateSourceCouchWithCluster;
import ca.carleton.gcrc.couch.date.impl.NowReference;
import ca.carleton.gcrc.couch.date.impl.TimeInterval;
import ca.carleton.gcrc.json.servlet.JsonServlet;

@SuppressWarnings("serial")
public class DateServlet extends JsonServlet {

	final protected Logger logger = LoggerFactory.getLogger(this.getClass());
	
	private DateServletConfiguration configuration = null;
	private DateServiceActions actions = null;
	private DateRobotThread robot = null;
	
	public DateServlet() {
		
	}
	
	public void init(ServletConfig config) throws ServletException {
		super.init(config);
		
		// Pick up configuration
		Object configurationObj = config.getServletContext().getAttribute(DateServletConfiguration.CONFIGURATION_KEY);
		if( null == configurationObj ) {
			throw new ServletException("Can not find configuration object");
		}
		if( configurationObj instanceof DateServletConfiguration ){
			configuration = (DateServletConfiguration)configurationObj;

			DateSourceCouchWithCluster dateSource;
			try {
				dateSource = new DateSourceCouchWithCluster(configuration.getAtlasDesignDocument());
			} catch (Exception e) {
				throw new ServletException("Unable to create date source",e);
			}
			
			actions = new DateServiceActions(dateSource);
			
			try {
				robot = new DateRobotThread(configuration.getAtlasDesignDocument(), dateSource.getClusterTree());
				robot.start();
			} catch (Exception e) {
				throw new ServletException("Unable to start date robot",e);
			}
			
		} else {
			throw new ServletException("Invalid class for configuration: "+configurationObj.getClass().getName());
		}
	}
	
	public void destroy() {
		if( null != robot ){
			DateRobotThread thread = robot;
			robot = null;
			try{
				thread.shutdown();
				thread.join();
			} catch(Exception e) {
				// just ignore. We're shutting down
			}
		}
	}

	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		try {
			List paths = computeRequestPath(request);
			
			if( paths.size() < 1 ) {
				doGetWelcome(request, response);
				
			} else if( paths.size() == 1
			 && "docIdsFromInterval".equals(paths.get(0)) ) {
				// Parameter 'min'
				String[] mins = request.getParameterValues("min");
				if( null == mins || mins.length != 1 ){
					throw new Exception("Parameter 'min' must be specified exactly once");
				}
				long min = Long.parseLong(mins[0]);
				
				// Parameter 'ongoing'
				String[] ongoings = request.getParameterValues("ongoing");
				boolean ongoing = false;
				if( null == ongoings ) {
					// nothing to do
				} else if( ongoings.length == 1 ) {
					ongoing = Boolean.parseBoolean(ongoings[0]);
				} else if( ongoings.length > 1 ){
					throw new Exception("Parameter 'ongoing' must be specified at most once");
				}
				
				// Parameter 'max'
				long max = min;
				if( !ongoing ){
					String[] maxs = request.getParameterValues("max");
					if( null == maxs || maxs.length != 1 ){
						throw new Exception("If not ongoing request, parameter 'max' must be specified exactly once");
					}
					max = Long.parseLong(maxs[0]);
				}

				TimeInterval interval = null;
				if( ongoing ){
					interval = new TimeInterval(min,NowReference.now());
				} else {
					interval = new TimeInterval(min,max);
				}
				
				JSONObject result = actions.getDocIdsFromInterval(interval);
				sendJsonResponse(response, result);
				
			} else if( paths.size() == 1
			 && "getInfo".equals(paths.get(0)) ) {
				response.setContentType("text/plain");
				response.setCharacterEncoding("utf-8");
				OutputStream os = response.getOutputStream();
				OutputStreamWriter osw = new OutputStreamWriter(os,"utf-8");
				PrintWriter pw = new PrintWriter(osw);
				
				actions.getInfo(pw);
				
				pw.flush();
				
			} else if( paths.size() == 1
			 && "getDot".equals(paths.get(0)) ) {
				response.setContentType("text/plain");
				response.setCharacterEncoding("utf-8");
				OutputStream os = response.getOutputStream();
				OutputStreamWriter osw = new OutputStreamWriter(os,"utf-8");
				PrintWriter pw = new PrintWriter(osw);
				
				actions.getDotInfo(pw);
				
				pw.flush();
				
			} else {
				throw new Exception("Unrecognized request");
			}
		} catch (Exception e) {
			reportError(e, response);
		}
	}

	private void doGetWelcome(HttpServletRequest request, HttpServletResponse resp) throws Exception {
		JSONObject result = new JSONObject();
		result.put("ok", true);
		result.put("service", "date");
		sendJsonResponse(resp, result);
	}
}