com.kixeye.chassis.transport.http.HttpTransportConfiguration.java Source code

Java tutorial

Introduction

Here is the source code for com.kixeye.chassis.transport.http.HttpTransportConfiguration.java

Source

package com.kixeye.chassis.transport.http;

/*
 * #%L
 * Chassis Transport Core
 * %%
 * Copyright (C) 2014 KIXEYE, Inc
 * %%
 * 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.
 * #L%
 */

import java.net.InetSocketAddress;
import java.util.Collection;
import java.util.Collections;
import java.util.Map;

import com.codahale.metrics.health.HealthCheckRegistry;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.common.base.Function;
import com.google.common.collect.Collections2;
import com.kixeye.chassis.transport.serde.JacksonMessageSerDe;
import com.kixeye.chassis.transport.shared.HealthServlet;
import org.apache.commons.lang.StringUtils;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.eclipse.jetty.servlet.ServletHolder;
import org.eclipse.jetty.servlets.GzipFilter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Condition;
import org.springframework.context.annotation.ConditionContext;
import org.springframework.context.annotation.Conditional;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.support.PropertySourcesPlaceholderConfigurer;
import org.springframework.core.annotation.Order;
import org.springframework.core.type.AnnotatedTypeMetadata;
import org.springframework.web.context.ConfigurableWebApplicationContext;
import org.springframework.web.context.support.AnnotationConfigWebApplicationContext;
import org.springframework.web.servlet.DispatcherServlet;

import com.codahale.metrics.MetricRegistry;
import com.codahale.metrics.jetty9.InstrumentedQueuedThreadPool;
import com.kixeye.chassis.transport.SpringMvcConfiguration;
import com.kixeye.chassis.transport.http.HttpTransportConfiguration.HttpEnabledCondition;
import com.kixeye.chassis.transport.shared.JettyConnectorRegistry;
import com.kixeye.chassis.transport.swagger.SwaggerRegistry;
import com.kixeye.chassis.transport.util.SpringContextWrapper;
import javax.annotation.Nullable;

/**
 * Configures the Http transport.
 * 
 * @author ebahtijaragic
 */
@Configuration
@Conditional(HttpEnabledCondition.class)
@ComponentScan(basePackageClasses = HttpTransportConfiguration.class)
public class HttpTransportConfiguration {
    public static final String HTTP_TRANSPORT_CHILD_CONTEXT_BEAN_NAME = "transportWebMvcContext";

    @Autowired(required = false)
    private MetricRegistry metricRegistry;

    @Value("${http.metrics.threadpool.enabled}")
    private boolean monitorThreadpool;

    @Value("${http.metrics.handler.enabled}")
    private boolean monitorHandler;

    @Autowired(required = false)
    private HealthCheckRegistry healthCheckRegistry;

    @Bean(initMethod = "start", destroyMethod = "stop")
    @Order(0)
    public Server httpServer(@Value("${http.enabled:false}") boolean httpEnabled,
            @Value("${http.hostname:}") String httpHostname, @Value("${http.port:-1}") int httpPort,

            @Value("${https.enabled:false}") boolean httpsEnabled,
            @Value("${https.hostname:}") String httpsHostname, @Value("${https.port:-1}") int httpsPort,
            @Value("${https.selfSigned:false}") boolean selfSigned,
            @Value("${https.mutualSsl:false}") boolean mutualSsl,

            @Value("${https.keyStorePath:}") String keyStorePath,
            @Value("${https.keyStoreData:}") String keyStoreData,
            @Value("${https.keyStorePassword:}") String keyStorePassword,
            @Value("${https.keyManagerPassword:}") String keyManagerPassword,

            @Value("${https.trustStorePath:}") String trustStorePath,
            @Value("${https.trustStoreData:}") String trustStoreData,
            @Value("${https.trustStorePassword:}") String trustStorePassword,

            @Value("${https.excludedCipherSuites:}") String[] excludedCipherSuites,

            ConfigurableWebApplicationContext webApplicationContext) throws Exception {

        // set up servlets
        ServletContextHandler context = servletContextHandler();

        // create a new child application context
        AnnotationConfigWebApplicationContext childApplicationContext = (AnnotationConfigWebApplicationContext) transportWebMvcContext(
                webApplicationContext, context).getContext();

        // register swagger
        childApplicationContext.getBean(SwaggerRegistry.class).registerSwagger(context,
                getObjectMappers(webApplicationContext));

        // configure the spring mvc dispatcher
        DispatcherServlet dispatcher = new DispatcherServlet(childApplicationContext);

        // enable gzip
        context.addFilter(GzipFilter.class, "/*", null);

        // map application servlets
        context.addServlet(new ServletHolder(dispatcher), "/");
        if (healthCheckRegistry != null) {
            context.addServlet(new ServletHolder(new HealthServlet(healthCheckRegistry)), "/healthcheck");
        }

        // create the server
        Server server;
        if (metricRegistry == null || !monitorThreadpool) {
            server = new Server();

            server.setHandler(context);
        } else {
            server = new Server(new InstrumentedQueuedThreadPool(metricRegistry));

            InstrumentedHandler instrumented = new InstrumentedHandler(metricRegistry);
            instrumented.setHandler(context);

            server.setHandler(instrumented);
        }

        // set up connectors
        if (httpEnabled) {
            InetSocketAddress address = StringUtils.isBlank(httpHostname) ? new InetSocketAddress(httpPort)
                    : new InetSocketAddress(httpHostname, httpPort);

            JettyConnectorRegistry.registerHttpConnector(server, address);
        }

        if (httpsEnabled) {
            InetSocketAddress address = StringUtils.isBlank(httpsHostname) ? new InetSocketAddress(httpsPort)
                    : new InetSocketAddress(httpsHostname, httpsPort);

            JettyConnectorRegistry.registerHttpsConnector(server, address, selfSigned, mutualSsl, keyStorePath,
                    keyStoreData, keyStorePassword, keyManagerPassword, trustStorePath, trustStoreData,
                    trustStorePassword, excludedCipherSuites);
        }

        return server;
    }

    private Collection<ObjectMapper> getObjectMappers(ConfigurableWebApplicationContext webApplicationContext) {
        Map<String, JacksonMessageSerDe> serdes = webApplicationContext.getBeansOfType(JacksonMessageSerDe.class);
        if (serdes.isEmpty()) {
            return Collections.emptyList();
        }
        return Collections2.transform(serdes.values(), new Function<JacksonMessageSerDe, ObjectMapper>() {
            @Nullable
            @Override
            public ObjectMapper apply(@Nullable JacksonMessageSerDe serDe) {
                return serDe.getObjectMapper();
            }
        });
    }

    @Bean
    public ServletContextHandler servletContextHandler() {
        ServletContextHandler context = new ServletContextHandler(
                ServletContextHandler.NO_SESSIONS | ServletContextHandler.NO_SECURITY);
        context.setErrorHandler(null);
        context.setWelcomeFiles(new String[] { "/" });

        return context;
    }

    @Bean(name = HTTP_TRANSPORT_CHILD_CONTEXT_BEAN_NAME)
    public SpringContextWrapper transportWebMvcContext(ConfigurableWebApplicationContext parentContext,
            ServletContextHandler servletContextHandler) {
        AnnotationConfigWebApplicationContext transportWebMvcContext = new AnnotationConfigWebApplicationContext();
        transportWebMvcContext.setDisplayName("httpTransport-webMvcContext");
        transportWebMvcContext.setServletContext(servletContextHandler.getServletContext());
        transportWebMvcContext.setParent(parentContext);
        transportWebMvcContext.setEnvironment(parentContext.getEnvironment());
        transportWebMvcContext.register(SpringMvcConfiguration.class);
        transportWebMvcContext.register(PropertySourcesPlaceholderConfigurer.class);
        transportWebMvcContext.refresh();

        return new SpringContextWrapper(transportWebMvcContext);
    }

    /**
    * A condition to check whether HTTP is enabled.
    * 
    * @author ebahtijaragic
    */
    public static class HttpEnabledCondition implements Condition {
        public boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
            return StringUtils.equalsIgnoreCase(context.getEnvironment().resolvePlaceholders("${http.enabled}"),
                    "true")
                    || StringUtils.equalsIgnoreCase(
                            context.getEnvironment().resolvePlaceholders("${https.enabled}"), "true");
        }
    }
}