com.sun.grizzly.config.GrizzlyConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of grizzly-config Show documentation
There is a newer version: 1.9.65
/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright (c) 2008-2010 Oracle and/or its affiliates. All rights reserved.
 *
 * The contents of this file are subject to the terms of either the GNU
 * General Public License Version 2 only ("GPL") or the Common Development
 * and Distribution License("CDDL") (collectively, the "License").  You
 * may not use this file except in compliance with the License.  You can
 * obtain a copy of the License at
 * https://glassfish.dev.java.net/public/CDDL+GPL_1_1.html
 * or packager/legal/LICENSE.txt.  See the License for the specific
 * language governing permissions and limitations under the License.
 *
 * When distributing the software, include this License Header Notice in each
 * file and include the License file at packager/legal/LICENSE.txt.
 *
 * GPL Classpath Exception:
 * Oracle designates this particular file as subject to the "Classpath"
 * exception as provided by Oracle in the GPL Version 2 section of the License
 * file that accompanied this code.
 *
 * Modifications:
 * If applicable, add the following below the License Header, with the fields
 * enclosed by brackets [] replaced by your own identifying information:
 * "Portions Copyright [year] [name of copyright owner]"
 *
 * Contributor(s):
 * If you wish your version of this file to be governed by only the CDDL or
 * only the GPL Version 2, indicate your decision by adding "[Contributor]
 * elects to include this software in this distribution under the [CDDL or GPL
 * Version 2] license."  If you don't indicate a single choice of license, a
 * recipient has the option to distribute your version of this file under
 * either the CDDL, the GPL Version 2 or to extend the choice of license to
 * its licensees as provided above.  However, if you add GPL Version 2 code
 * and therefore, elected the GPL Version 2 license, then the option applies
 * only if the new code is made subject to such option by the copyright
 * holder.
 */

package com.sun.grizzly.config;

import com.sun.grizzly.Controller;
import com.sun.grizzly.ControllerStateListenerAdapter;
import com.sun.grizzly.config.dom.NetworkConfig;
import com.sun.grizzly.config.dom.NetworkListener;
import com.sun.grizzly.util.FutureImpl;
import com.sun.grizzly.util.LoggerUtils;
import com.sun.grizzly.util.WorkerThreadImpl;
import org.jvnet.hk2.component.Habitat;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * Created Nov 24, 2008
 *
 * @author Justin Lee
 */
public class GrizzlyConfig {
    private static final Logger logger = LoggerUtils.getLogger();
    private final NetworkConfig config;
    private final Habitat habitat;
    private final List listeners = new ArrayList();

    public GrizzlyConfig(String file) {
        habitat = Utils.getHabitat(file);
        config = habitat.getComponent(NetworkConfig.class);
    }

    public NetworkConfig getConfig() {
        return config;
    }

    public Habitat getHabitat() {
        return habitat;
    }

    public List getListeners() {
        return listeners;
    }

    public void setupNetwork() {
        validateConfig(config);
        synchronized (listeners) {
            final List networkListeners = config.getNetworkListeners().getNetworkListener();

            final FutureImpl future = new FutureImpl();
            final AtomicInteger counter = new AtomicInteger(networkListeners.size());

            for (final NetworkListener listener : networkListeners) {
                final Controller controller = new Controller() {
                    @Override
                    public void logVersion() {
                        //no-op
                    }
                };
                controller.addStateListener(new ControllerStateListenerAdapter() {
                    @Override
                    public void onReady() {
                        if (counter.decrementAndGet() == 0) {
                            future.setResult(Boolean.TRUE);
                        }
                    }
                    
                    @Override
                    public void onException(Throwable e) {
                        future.setException(new IllegalStateException(
                                "Failed to start listener on port=" + listener.getPort(), e));
                    }
                });
                
                final GrizzlyServiceListener grizzlyListener = new GrizzlyServiceListener(controller);
                grizzlyListener.configure(listener, habitat);
                listeners.add(grizzlyListener);
                final Thread thread = new WorkerThreadImpl(new ListenerRunnable(grizzlyListener));
                thread.setDaemon(true);
                thread.start();
            }

            try {
                future.get(10, TimeUnit.SECONDS); // wait for the system to finish setting up the listener
            } catch (Exception e) {
                logger.warning(e.getMessage());
                throw new RuntimeException(e);
            }
        }
    }

    public void shutdownNetwork() {
        synchronized (listeners) {
            for (GrizzlyServiceListener listener : listeners) {
                try {
                    listener.stop();
                } catch (Exception e) {
                }
            }
            listeners.clear();
        }
    }

    private static void validateConfig(NetworkConfig config) {
        for (final NetworkListener listener : config.getNetworkListeners().getNetworkListener()) {
            listener.findHttpProtocol();
        }
    }

    public void shutdown() {
        final List copy;
        synchronized(listeners) {
            copy = new ArrayList(listeners);
        }

        for (GrizzlyServiceListener listener : copy) {
            listener.stop();
        }
    }

    public static boolean toBoolean(String value) {
        final String v = null != value ? value.trim() : value;
        return "true".equals(v) || "yes".equals(v) || "on".equals(v) || "1".equals(v);
    }

    private static class ListenerRunnable implements Runnable {
        private final GrizzlyServiceListener grizzlyListener;

        public ListenerRunnable(GrizzlyServiceListener grizzlyListener) {
            this.grizzlyListener = grizzlyListener;
        }

        public void run() {
            try {
                grizzlyListener.start();
            } catch (IOException e) {
                logger.log(Level.SEVERE, e.getMessage(), e);
                throw new RuntimeException(e.getMessage());
            } catch (InstantiationException e) {
                logger.log(Level.SEVERE, e.getMessage(), e);
                throw new RuntimeException(e.getMessage());
            }
        }
    }
}