jvm-jersey-env (#1677)

A new Jersey framework based JVM environment which is lightweight and simpler.
2020-09-08 21:24:25 +05:30
parent 975286c2ac
commit 54b384efec
28 changed files with 1049 additions and 25 deletions
@@ -0,0 +1,35 @@
+package io.fission;
+
+import org.codehaus.jackson.annotate.JsonIgnoreProperties;
+
+@JsonIgnoreProperties(ignoreUnknown = true)
+public class FunctionLoadRequest {
+
+	private String filepath;
+	private String functionName;
+	private String url;
+
+	String getFilepath() {
+		return filepath;
+	}
+
+	void setFilepath(String filepath) {
+		this.filepath = filepath;
+	}
+
+	String getUrl() {
+		return url;
+	}
+
+	void setUrl(String url) {
+		this.url = url;
+	}
+
+	public String getFunctionName() {
+		return functionName;
+	}
+
+	public void setFunctionName(String functionName) {
+		this.functionName = functionName;
+	}
+}
@@ -0,0 +1,147 @@
+package io.fission;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.PrintWriter;
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.util.Enumeration;
+import java.util.jar.JarEntry;
+import java.util.jar.JarFile;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+import javax.ws.rs.GET;
+import javax.ws.rs.POST;
+import javax.ws.rs.PUT;
+import javax.ws.rs.DELETE;
+import javax.ws.rs.Path;
+import javax.ws.rs.core.Response;
+import javax.ws.rs.core.Context;
+import javax.ws.rs.container.ContainerRequestContext;
+
+import io.fission.Function;
+
+@Path("/")
+public class JerseyServer {
+
+	private static Function<ContainerRequestContext,Response> fn;
+
+	private static final int CLASS_LENGTH = 6;
+
+	private static Logger logger = Logger.getGlobal();
+
+	@GET
+	public Response home(@Context ContainerRequestContext request) {
+		return callUserFunction(request);
+	}
+
+	@POST
+	public Response homePost(@Context ContainerRequestContext request) {
+		return callUserFunction(request);
+	}
+
+	@PUT
+	public Response homePut(@Context ContainerRequestContext request) {
+		return callUserFunction(request);
+	}
+
+	@DELETE
+	public Response homeDelete(@Context ContainerRequestContext request) {
+		return callUserFunction(request);
+	}
+
+	@Path("v2/specialize")
+	@POST
+	public Response specialize(FunctionLoadRequest req) {
+
+		long startTime = System.nanoTime();
+		File file = new File(req.getFilepath());
+
+		if (!file.exists()) {
+			return Response.status(Response.Status.BAD_REQUEST).entity("/userfunc/usernot found").build();
+		}
+
+		String entryPoint = req.getFunctionName();
+		logger.log(Level.INFO, "Entrypoint class:" + entryPoint);
+		if (entryPoint == null) {
+			return Response.status(Response.Status.BAD_REQUEST).entity("Entrypoint class is missing in the JAR or the name is incorrect")
+					.build();
+		}
+
+		JarFile jarFile = null;
+		ClassLoader cl = null;
+		try {
+
+			jarFile = new JarFile(file);
+			Enumeration<JarEntry> e = jarFile.entries();
+			URL[] urls = { new URL("jar:file:" + file + "!/") };
+
+			// TODO Check if the classloading can be improved for ex. use something like:
+			// Thread.currentThread().setContextClassLoader(cl);
+			if (this.getClass().getClassLoader() == null) {
+				cl = URLClassLoader.newInstance(urls);
+			} else {
+				cl = URLClassLoader.newInstance(urls, this.getClass().getClassLoader());
+			}
+
+			if (cl == null) {
+
+				return Response.status(Response.Status.BAD_REQUEST).entity("Failed to initialize the classloader")
+						.build();
+			}
+
+			// Load all dependent classes from libraries etc.
+			while (e.hasMoreElements()) {
+				JarEntry je = e.nextElement();
+				if (je.isDirectory() || !je.getName().endsWith(".class")) {
+					continue;
+				}
+				String className = je.getName().substring(0, je.getName().length() - CLASS_LENGTH);
+				className = className.replace('/', '.');
+				cl.loadClass(className);
+			}
+
+			// Instantiate the function class
+			fn = (Function) cl.loadClass(entryPoint).newInstance();
+		} catch (MalformedURLException e) {
+			e.printStackTrace();
+			return Response.status(Response.Status.BAD_REQUEST).entity("Entrypoint class is missing in the function")
+					.build();
+		} catch (ClassNotFoundException e) {
+			e.printStackTrace();
+			return Response.status(Response.Status.BAD_REQUEST).entity("Error loading Function or dependent class")
+					.build();
+		} catch (InstantiationException e) {
+			e.printStackTrace();
+			return Response.status(Response.Status.BAD_REQUEST)
+					.entity("Error creating a new instance of function class").build();
+		} catch (IllegalAccessException e) {
+			e.printStackTrace();
+			return Response.status(Response.Status.BAD_REQUEST)
+					.entity("Error creating a new instance of function class").build();
+		} catch (IOException e) {
+			e.printStackTrace();
+			return Response.status(Response.Status.BAD_REQUEST).entity("Error reading the JAR file").build();
+		} finally {
+			try {
+				jarFile.close();
+			} catch (IOException e) {
+				e.printStackTrace();
+				return Response.status(Response.Status.BAD_REQUEST)
+						.entity("Error closing the file while loading the class").build();
+			}
+		}
+		long elapsedTime = System.nanoTime() - startTime;
+		logger.log(Level.INFO, "Specialize call done in: " + elapsedTime / 1000000 + " ms");
+		return Response.status(Response.Status.OK).entity("Done").build();
+	}
+
+	private Response callUserFunction(ContainerRequestContext httpRequest) {
+		if (fn == null) {
+			return Response.status(Response.Status.BAD_REQUEST).entity("Container not specialized").build();
+		} else {
+			return fn.call(httpRequest, null);
+		}
+	}
+}
@@ -0,0 +1,50 @@
+package io.fission;
+
+import org.eclipse.jetty.server.Server;
+import org.eclipse.jetty.servlet.ServletContextHandler;
+import org.eclipse.jetty.servlet.ServletHolder;
+import org.glassfish.jersey.jackson.JacksonFeature;
+import org.glassfish.jersey.server.ResourceConfig;
+import org.glassfish.jersey.servlet.ServletContainer;
+
+public class Main {
+
+	private static final int DEFAULT_PORT = 8888;
+	
+	private int serverPort;
+	
+	public Main(int serverPort) throws Exception {
+		this.serverPort = serverPort;
+		Server server = configureServer();	        
+        server.start();
+        server.join();
+	}	
+
+	private Server configureServer() {
+		ResourceConfig resourceConfig = new ResourceConfig();		
+		resourceConfig.packages(JerseyServer.class.getPackage().getName());
+		resourceConfig.register(JacksonFeature.class);
+		ServletContainer servletContainer = new ServletContainer(resourceConfig);
+		ServletHolder sh = new ServletHolder(servletContainer);
+		Server server = new Server(serverPort);		
+		ServletContextHandler context = new ServletContextHandler(ServletContextHandler.SESSIONS);
+        context.setContextPath("/");
+        context.addServlet(sh, "/*");
+		server.setHandler(context);
+		return server;
+	}
+	
+	public static void main(String[] args) throws Exception {
+		
+		int serverPort = DEFAULT_PORT;
+		
+		if(args.length >= 1) {
+			try {
+				serverPort = Integer.parseInt(args[0]);
+			} catch (NumberFormatException e) {
+				e.printStackTrace();
+			}
+		}
+		new Main(serverPort);	
+	}
+}