com.tvd12.ezyhttp.server.jetty.JettyApplicationBootstrap Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ezyhttp-server-jetty Show documentation
There is a newer version: 1.3.7
package com.tvd12.ezyhttp.server.jetty;

import java.util.ArrayList;
import java.util.EnumSet;
import java.util.List;

import javax.servlet.DispatcherType;
import javax.servlet.MultipartConfigElement;

import org.eclipse.jetty.server.Connector;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.ServerConnector;
import org.eclipse.jetty.servlet.FilterHolder;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.eclipse.jetty.servlets.CrossOriginFilter;
import org.eclipse.jetty.util.thread.QueuedThreadPool;

import com.tvd12.ezyfox.annotation.EzyProperty;
import com.tvd12.ezyfox.util.EzyLoggable;
import com.tvd12.ezyhttp.core.util.FileSizes;
import com.tvd12.ezyhttp.server.core.ApplicationEntry;
import com.tvd12.ezyhttp.server.core.annotation.ApplicationBootstrap;

import lombok.AccessLevel;
import lombok.Setter;

@Setter
@ApplicationBootstrap
public class JettyApplicationBootstrap extends EzyLoggable implements ApplicationEntry {
	
	@EzyProperty("server.port")
	protected int port = 8080;
	
	@EzyProperty("server.host")
	protected String host = "0.0.0.0";
	
	@EzyProperty("server.max_threads")
	protected int maxThreads = 256;
	
	@EzyProperty("server.min_threads")
	protected int minThreads = 16;
	
	@EzyProperty("server.idle_timeout")
	protected int idleTimeout = 150 * 1000;
	
	@EzyProperty("server.multipart.location")
	protected String multipartLocation = "tmp";
	
	@EzyProperty("server.multipart.file_size_threshold")
	protected String multipartFileSizeThreshold = "5MB";
	
	@EzyProperty("server.multipart.max_file_size")
	protected String multipartMaxFileSize = "5MB";
	
	@EzyProperty("server.multipart.max_request_size")
	protected String multipartMaxRequestSize = "5MB";
	
	@EzyProperty("cors.enable")
	protected boolean corsEnable;
	
	@EzyProperty("cors.allowed_origins")
	protected String allowedOrigins = "*";
	
	@EzyProperty("cors.allowed_headers")
    protected String allowedHeaders = "*";
	
	@EzyProperty("management.enable")
	protected boolean managementEnable;
	
	@EzyProperty("management.host")
	protected String managementHost = "0.0.0.0";
	
	@EzyProperty("management.port")
	protected int managementPort = 18080;

	@Setter(AccessLevel.NONE)
	protected Server server;
	 
	@Override   
    public void start() throws Exception {
    	QueuedThreadPool threadPool = new QueuedThreadPool(maxThreads, minThreads, idleTimeout);
        server = new Server(threadPool);
        ServerConnector connector = new ServerConnector(server);
        connector.setHost(host);
        connector.setPort(port);
        List connectors = new ArrayList<>();
        connectors.add(connector);
        if(managementEnable) {
        	ServerConnector managementConnector = new ServerConnector(server);
        	managementConnector.setHost(managementHost);
        	managementConnector.setPort(managementPort);
        	connectors.add(managementConnector);
        }
        server.setConnectors(connectors.toArray(new Connector[connectors.size()]));
        ServletContextHandler servletHandler = newServletHandler();
        server.setHandler(servletHandler);
        server.start();
        logger.info("http server started on: {}:{}", host, port);
        if(managementEnable)
        	logger.info("management started on: {}:{}", managementHost, managementPort);
    }
	
    protected ServletContextHandler newServletHandler() {
    	ServletContextHandler servletHandler = new ServletContextHandler();
        servletHandler.addServlet(JettyBlockingServlet.class, "/*")
        	.getRegistration()
        	.setMultipartConfig(new MultipartConfigElement(
        		multipartLocation,
        		FileSizes.toByteSize(multipartMaxFileSize),
        		(int)FileSizes.toByteSize(multipartMaxRequestSize),
        		(int)FileSizes.toByteSize(multipartFileSizeThreshold)
        	));;
        logger.info("cors.enable = {}", corsEnable);
        if(corsEnable) {
	        FilterHolder crossOriginFilter = newCrossOriginFilter();
	        addFilter(servletHandler, crossOriginFilter);
        }
        return servletHandler;
    }
    
    protected void addFilter(ServletContextHandler servletHandler, FilterHolder filter) {
    	servletHandler.addFilter(filter, "/*", EnumSet.of(DispatcherType.REQUEST));
    }
    
    protected FilterHolder newCrossOriginFilter() {
    	FilterHolder filter = new FilterHolder();
    	filter.setName("cross-origin");
    	filter.setInitParameter(CrossOriginFilter.ALLOWED_ORIGINS_PARAM, allowedOrigins);
    	filter.setInitParameter(CrossOriginFilter.ALLOWED_METHODS_PARAM, "GET,HEAD,POST,PUT,DELETE,CONNECT,TRACE,PATCH,OPTIONS");
    	filter.setInitParameter(CrossOriginFilter.ALLOWED_HEADERS_PARAM, allowedHeaders);
    	filter.setInitParameter(CrossOriginFilter.ACCESS_CONTROL_ALLOW_ORIGIN_HEADER, allowedOrigins);
    	filter.setInitParameter(CrossOriginFilter.ACCESS_CONTROL_ALLOW_METHODS_HEADER, "GET,HEAD,POST,PUT,DELETE,CONNECT,TRACE,PATCH,OPTIONS");
    	filter.setInitParameter(CrossOriginFilter.ACCESS_CONTROL_ALLOW_HEADERS_HEADER, allowedHeaders);
    	CrossOriginFilter corsFilter = new CrossOriginFilter();
    	filter.setFilter(corsFilter);
    	return filter;
    }
}