prompto.server.DataServlet Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Server Show documentation
The newest version!
package prompto.server;

import java.io.IOException;
import java.util.List;
import java.util.Map;

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

import prompto.error.PromptoError;
import prompto.expression.FetchManyExpression;
import prompto.expression.IFetchExpression;
import prompto.literal.IntegerLiteral;
import prompto.parser.ECleverParser;
import prompto.store.IStore;
import prompto.store.IStored;
import prompto.utils.Logger;

@SuppressWarnings("serial")
public class DataServlet extends CleverServlet {

	static Logger logger = new Logger();
	
	static Map stores;
	
	public static void setStores(Map stores) {
		DataServlet.stores = stores;
	}
	
	public static Map getStores() {
		return DataServlet.stores;
	}
	
	@Override
	public void init(ServletConfig config) throws ServletException {
		super.init(config);
	}

	@Override
	protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
		Thread.currentThread().setName(this.getClass().getSimpleName());
		String path = req.getPathInfo();
		if(path!=null) switch(path) {
		case "/fetch":
			doFetch(req, resp);
			break;
		default:
			resp.sendError(HttpServletResponse.SC_NOT_FOUND);
		}
	}
	protected void doFetch(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
		try(var stream = resp.getOutputStream()) {
			try {
				String store = req.getParameter("store");
				if(store==null || store.trim().isEmpty()) {
					writeJsonResponseError("No store specified!", stream);
					return;
				}
				IStore dataStore = stores.get(store);
				if(dataStore==null) {
					writeJsonResponseError("Invalid store: " + store, stream);
					return;
				}
				String query = req.getParameter("query");
				if(query==null || query.trim().isEmpty()) {
					writeJsonResponseError("Empty query!", stream);
					return;
				}
				String first = req.getParameter("first");
				String last = req.getParameter("last");
				String format = req.getParameter("format");
				if(format==null)
					format = "list";
				ECleverParser parser = new ECleverParser(query);
				IFetchExpression fetch = parser.parse_fetch_expression();
				if(fetch==null) {
					writeJsonResponseError("Invalid query: " + query, stream);
					return;
				}
				adjustQueryRange(fetch, first, last);
				logger.info(()->"Running query: " + fetch.toString());
				if("list".equals(format.toLowerCase())) {
					resp.setContentType("application/json");
					Object fetched = fetch.fetchRaw(dataStore);
					JsonRecordsWriter writer = new JsonRecordsWriter(stream, dataStore::getAttributeInfo, dataStore, false);
					writer.writeRecords(fetched);
				} else
					writeJsonResponseError("Invalid query!", stream);
			} catch(PromptoError e) {
				writeJsonResponseError("Invalid query!", stream);
			} catch(Throwable t) {
				t.printStackTrace(System.err);
				resp.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
			}
		}
	}
	
	private static void adjustQueryRange(IFetchExpression fetch, String first, String last) {
		if(fetch instanceof FetchManyExpression) {
			FetchManyExpression many = (FetchManyExpression)fetch;
			if(many.getFirst()==null && first!=null)
				many.setFirst(new IntegerLiteral(first));
			if(many.getLast()==null && last!=null)
				many.setLast(new IntegerLiteral(last));
		}
		
	}
	

	@SuppressWarnings("unchecked")
	static String readCategory(IStored value) {
		List categories = (List)((IStored)value).getData("category");
		if(categories==null || categories.size()<1)
			return "";
		else
			return categories.get(categories.size()-1);
	}

	




}