Java tutorial
/* * * Copyright 2014 McEvoy Software Ltd. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package io.milton.grizzly; import com.ning.http.client.AsyncHttpClient; import io.milton.http.HttpManager; import io.milton.mail.MailServer; import java.io.IOException; import java.security.Security; import java.util.concurrent.ExecutorService; import org.apache.commons.lang.StringUtils; import org.glassfish.grizzly.http.server.HttpHandler; import org.glassfish.grizzly.http.server.HttpServer; import org.glassfish.grizzly.http.server.Request; import org.glassfish.grizzly.http.server.Response; import org.glassfish.grizzly.ssl.SSLEngineConfigurator; import org.glassfish.grizzly.threadpool.GrizzlyExecutorService; import org.glassfish.grizzly.threadpool.ThreadPoolConfig; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.BeansException; import org.springframework.context.ConfigurableApplicationContext; import org.springframework.context.support.ClassPathXmlApplicationContext; import org.springframework.context.support.StaticApplicationContext; /** * Use this as a runnable class to start a Grizzly HTTP Server * * On startup it will look for a spring config file in /applicationContext.xml , * and then try to locate a bean named milton.http.manager which must be either * a milton HttpManager or HttpManagerBuilder * * You might find its easiest to copy this class into your app and then add your * own init * * * * @author brad */ public class GrizzlyLoadBalancer { private static final Logger log = LoggerFactory.getLogger(GrizzlyLoadBalancer.class); public static void main(String[] args) throws IOException, InterruptedException { int port = 8080; if (args.length > 0) { port = Integer.parseInt(args[0]); } Integer sslPort = null; if (args.length > 1) { sslPort = Integer.parseInt(args[1]); } GrizzlyLoadBalancer k = new GrizzlyLoadBalancer(); k.start(); System.out.println("Press any key to stop the server..."); System.in.read(); } private HttpServer httpServer; private StaticApplicationContext parent; private HttpManager httpManager; private MailServer mailServer; private MiltonSNIService kademiSNIService; private boolean running; private ExecutorService complexAppExecutorService; public GrizzlyLoadBalancer() { } public void start() throws IOException { start(8080); } public void start(int httpPort) throws IOException { start(httpPort, null); } public boolean start(int httpPort, Integer sslPort) throws IOException { Security.addProvider(new org.bouncycastle.jce.provider.BouncyCastleProvider()); ConfigurableApplicationContext ctx = initSpringApplicationContext(); if (ctx == null) { log.warn("Failed to initialise spring"); return false; } complexAppExecutorService = GrizzlyExecutorService .createInstance(ThreadPoolConfig.defaultConfig().copy().setCorePoolSize(5).setMaxPoolSize(5)); String host = getPropertyOrDefault("host", null); int port = getPropertyOrDefaultInt("port", 8080); int secureHttpPort = getPropertyOrDefaultInt(MiltonSNIService.SYS_SECURE_PORT, MiltonSNIService.SECURE_PORT); if (host == null) { httpServer = HttpServer.createSimpleServer(null, port); } else { httpServer = HttpServer.createSimpleServer(null, host, port); } { // Start the Kademi SNI SSL service MiltonSNICertificateStore store = null; // TODO: allow injection MiltonSNICertificateManager sniCerManager = new MiltonSNICertificateManager(store); SSLEngineConfigurator sniConfig = sniCerManager.createEngineConfigurator(); this.kademiSNIService = new MiltonSNIService(secureHttpPort, sniConfig); this.kademiSNIService.startOn(httpServer); } httpServer.getServerConfiguration().addHttpHandler(new HttpHandler() { @Override public void service(final Request request, final Response response) throws Exception { log.trace("service"); response.suspend(); complexAppExecutorService.execute(new Runnable() { @Override public void run() { try { forwardToCluster(request, response); } finally { response.resume(); } } }); } }, "/"); httpServer.start(); running = true; return true; } @SuppressWarnings("resource") protected ConfigurableApplicationContext initSpringApplicationContext() { log.info("No root spring context"); parent = new StaticApplicationContext(); ConfigurableApplicationContext ctx = null; String[] contextFiles = new String[] { "applicationContext.xml" }; parent.refresh(); try { ctx = new ClassPathXmlApplicationContext(contextFiles, parent); } catch (BeansException e) { log.error("Unable to create a child context for Milton", e); } return ctx; } private void forwardToCluster(Request request, Response response) { // AsyncHttpClient c = new AsyncHttpClient(); // c.prepareGet(null) } private int getPropertyOrDefaultInt(String propSuffix, int defaultVal) { String name = "kademi." + propSuffix; String s = System.getProperty(name); if (StringUtils.isNotBlank(s)) { log.info("Using System property: " + name + " = " + s); return Integer.parseInt(s); } log.info("Using default value " + defaultVal + " for property " + name); return defaultVal; } public static String getPropertyOrDefault(String propSuffix, String defaultVal) { String name = "kademi." + propSuffix; String s = System.getProperty(name); if (StringUtils.isNotBlank(s)) { log.info("Using System property: " + name + " = " + s); return s; } log.info("Using default value " + defaultVal + " for property " + name); return defaultVal; } public class ExtAsyncHttpClient extends AsyncHttpClient { protected BoundRequestBuilder requestBuilder(String method, String url) { //return new BoundRequestBuilder( method, config.isDisableUrlEncodingForBoundedRequests()).setUrl(url).setSignatureCalculator(signatureCalculator); return null; } } }