interactivespaces.service.web.server.internal.netty.NettyWebServerService.java Source code

Introduction

Here is the source code for interactivespaces.service.web.server.internal.netty.NettyWebServerService.java

Source

/*
 * Copyright (C) 2012 Google 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.
 */

package interactivespaces.service.web.server.internal.netty;

import interactivespaces.service.web.server.WebServer;
import interactivespaces.service.web.server.WebServerService;
import interactivespaces.service.web.server.internal.BaseWebServerService;
import interactivespaces.system.InteractiveSpacesEnvironment;
import interactivespaces.util.io.FileSupport;
import interactivespaces.util.io.FileSupportImpl;
import interactivespaces.util.web.MapExtensionMimeResolver;
import interactivespaces.util.web.MimeResolver;

import org.apache.commons.logging.Log;

import java.io.InputStream;
import java.util.concurrent.ScheduledExecutorService;

/**
 * A {@link WebServerService} which gives NETTY web servers.
 *
 * @author Keith M. Hughes
 */
public class NettyWebServerService extends BaseWebServerService {

    /**
     * Where the default MIME types are found.
     */
    public static final String BUNDLE_LOCATION_WEB_SERVER_MIME_TYPES = "interactivespaces/util/web/mime.types";

    /**
     * The line splitter regex for the internal MIME file.
     */
    public static final String MIME_TYPE_LINE_SPLITTER_REGEX = "\\r?\\n";

    /**
     * The regex for splitting lines of the internal MIME file into their component pieces.
     */
    public static final String MIME_FILE_COMPONENT_SPLITTING_REGEX = "\\s+";

    /**
     * The comment character for the internal MIME file.
     */
    public static final String MIME_FILE_COMMENT_CHARACTER = "#";

    /**
     * The default HTTP MIME resolver to use.
     */
    private MimeResolver defaultHttpMimeResolver;

    /**
     * The file support to use.
     */
    private FileSupport fileSupport = FileSupportImpl.INSTANCE;

    @Override
    public String getName() {
        return WebServerService.SERVICE_NAME;
    }

    @Override
    public void startup() {
        // Because of tests.
        InteractiveSpacesEnvironment spaceEnvironment = getSpaceEnvironment();
        defaultHttpMimeResolver = newDefaultMimeResolver(
                spaceEnvironment != null ? spaceEnvironment.getLog() : null);
    }

    @Override
    public synchronized WebServer newWebServer(String serverName, int port, Log log) {
        WebServer server = newWebServer(log);

        server.setPort(port);
        server.setServerName(serverName);

        return server;
    }

    @Override
    public synchronized WebServer newWebServer(Log log) {
        ScheduledExecutorService threadPool = getSpaceEnvironment().getExecutorService();

        WebServer server = new NettyWebServer(threadPool, threadPool, log);
        server.setDefaultMimeResolver(getDefaultHttpMimeResolver());

        addServer(server);

        return server;
    }

    @SuppressWarnings("unchecked")
    @Override
    public <T extends MimeResolver> T getDefaultHttpMimeResolver() {
        return (T) defaultHttpMimeResolver;
    }

    @Override
    public void setDefaultHttpMimeResolver(MimeResolver resolver) {
        defaultHttpMimeResolver = resolver;
    }

    /**
     * Create a new MIME resolver which maps standard MIME types.
     *
     * @param log
     *          the logger to use while creating the resolver.
     *
     * @return the resolver
     */
    private MapExtensionMimeResolver newDefaultMimeResolver(Log log) {
        MapExtensionMimeResolver resolver = new MapExtensionMimeResolver();

        try {
            InputStream mimeResource = MapExtensionMimeResolver.class.getClassLoader()
                    .getResourceAsStream(BUNDLE_LOCATION_WEB_SERVER_MIME_TYPES);
            if (mimeResource != null) {
                String mimeFile = fileSupport.inputStreamAsString(mimeResource);
                String[] lines = mimeFile.split(MIME_TYPE_LINE_SPLITTER_REGEX);
                for (String line : lines) {
                    line = line.trim();
                    if (!line.isEmpty() && !line.startsWith(MIME_FILE_COMMENT_CHARACTER)) {
                        String[] parts = line.split(MIME_FILE_COMPONENT_SPLITTING_REGEX);
                        if (parts.length >= 2) {
                            String mimeType = parts[0];
                            for (int i = 1; i < parts.length; i++) {
                                resolver.addMimeType(parts[i], mimeType);
                            }
                        }
                    }
                }
            }
        } catch (Exception e) {
            if (log != null) {
                log.warn("Could not read MIME file. MIME resolver is empty", e);
            }
        }
        return resolver;
    }

}