com.sun.grizzly.http.jk.server.JkCoyoteHandler.java Source code

Java tutorial

Introduction

Here is the source code for com.sun.grizzly.http.jk.server.JkCoyoteHandler.java

Source

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright (c) 2009-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.
 *
 *
 * This file incorporates work covered by the following copyright and
 * permission notice:
 *
 * Copyright 2004 The Apache Software Foundation
 *
 * 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 com.sun.grizzly.http.jk.server;

import java.io.IOException;
import java.util.Iterator;

import javax.management.MBeanServer;
import javax.management.ObjectName;

import com.sun.grizzly.tcp.Adapter;
import com.sun.grizzly.tcp.ProtocolHandler;
import com.sun.grizzly.tcp.Request;
import com.sun.grizzly.tcp.Response;
import com.sun.grizzly.tcp.RequestInfo;
import com.sun.grizzly.tcp.Constants;
import com.sun.grizzly.util.LoggerUtils;
import java.util.logging.Level;
import com.sun.grizzly.http.jk.core.JkHandler;
import com.sun.grizzly.http.jk.core.Msg;
import com.sun.grizzly.http.jk.core.MsgContext;
import com.sun.grizzly.tcp.http11.GrizzlyListener;
import java.util.concurrent.CountDownLatch;
import org.apache.commons.modeler.Registry;

/** Plugs Jk into Coyote. Must be named "type=JkHandler,name=container"
 *
 * jmx:notification-handler name="org.apache.jk.SEND_PACKET
 * jmx:notification-handler name="com.sun.grizzly.tcp.ACTION_COMMIT
 */
public class JkCoyoteHandler extends JkHandler implements ProtocolHandler, GrizzlyListener {

    // ----------------------------------------------------------- DoPrivileged
    private boolean paused = false;
    int epNote;
    Adapter adapter;
    protected JkMain jkMain = null;

    /** Set a property. Name is a "component.property". JMX should
     * be used instead.
     */
    @Override
    public void setProperty(String name, String value) {
        getJkMain().setProperty(name, value);
        properties.put(name, value);
    }

    @Override
    public String getProperty(String name) {
        return properties.getProperty(name);
    }

    public Iterator getAttributeNames() {
        return properties.keySet().iterator();
    }

    /** Pass config info
     */
    public void setAttribute(String name, Object value) {
        if (LoggerUtils.getLogger().isLoggable(Level.FINEST)) {
            LoggerUtils.getLogger().log(Level.FINEST, "setAttribute " + name + " " + value);
        }
        if (value instanceof String) {
            this.setProperty(name, (String) value);
        }
    }

    /**
     * Retrieve config info.
     * Primarily for use with the admin webapp.
     */
    public Object getAttribute(String name) {
        return getJkMain().getProperty(name);
    }

    /** The adapter, used to call the connector 
     */
    public void setAdapter(Adapter adapter) {
        this.adapter = adapter;
    }

    public Adapter getAdapter() {
        return adapter;
    }

    public JkMain getJkMain() {
        if (jkMain == null) {
            jkMain = new JkMain();
            jkMain.setWorkerEnv(wEnv);

        }
        return jkMain;
    }

    boolean started = false;

    /** Start the protocol
     */
    @Override
    public void init() {
        if (started) {
            return;
        }

        started = true;

        if (wEnv == null) {
            // we are probably not registered - not very good.
            wEnv = getJkMain().getWorkerEnv();
            wEnv.addHandler("container", this);
        }

        try {
            // jkMain.setJkHome() XXX;

            getJkMain().init();

        } catch (Exception ex) {
            LoggerUtils.getLogger().log(Level.SEVERE, "Error during init", ex);
        }
    }

    public void start() {
        try {
            if (oname != null && getJkMain().getDomain() == null) {
                try {
                    ObjectName jkmainOname = new ObjectName(oname.getDomain() + ":type=JkMain");
                    Registry.getRegistry(null, null).registerComponent(getJkMain(), jkmainOname, "JkMain");
                } catch (Exception e) {
                    LoggerUtils.getLogger().log(Level.SEVERE, "Error registering jkmain " + e);
                }
            }
            getJkMain().start();
        } catch (Exception ex) {
            LoggerUtils.getLogger().log(Level.SEVERE, "Error during startup", ex);
        }
    }

    @Override
    public void pause() throws Exception {
        if (!paused) {
            paused = true;
            getJkMain().pause();
        }
    }

    @Override
    public void resume() throws Exception {
        if (paused) {
            paused = false;
            getJkMain().resume();
        }
    }

    @Override
    public void destroy() {
        if (!started) {
            return;
        }

        started = false;
        getJkMain().stop();
    }

    // -------------------- Jk handler implementation --------------------
    // Jk Handler mehod
    @Override
    public int invoke(Msg msg, MsgContext ep) throws IOException {
        if (ep.isLogTimeEnabled()) {
            ep.setLong(MsgContext.TIMER_PRE_REQUEST, System.currentTimeMillis());
        }

        Request req = ep.getRequest();
        Response res = req.getResponse();

        if (LoggerUtils.getLogger().isLoggable(Level.FINEST)) {
            LoggerUtils.getLogger().log(Level.FINEST,
                    "Invoke " + req + " " + res + " " + req.requestURI().toString());
        }

        res.setNote(epNote, ep);
        ep.setStatus(MsgContext.JK_STATUS_HEAD);
        RequestInfo rp = req.getRequestProcessor();
        rp.setStage(Constants.STAGE_SERVICE);
        try {
            adapter.service(req, res);
        } catch (Exception ex) {
            LoggerUtils.getLogger().log(Level.SEVERE, "Error servicing request " + req, ex);
        } finally {
            try {
                adapter.afterService(req, res);
            } catch (Exception ex) {
                LoggerUtils.getLogger().log(Level.SEVERE, "Error servicing request " + req, ex);
            }
        }
        if (ep.getStatus() != MsgContext.JK_STATUS_CLOSED) {
            res.finish();
        }

        req.recycle();
        req.updateCounters();
        res.recycle();
        ep.recycle();
        if (ep.getStatus() == MsgContext.JK_STATUS_ERROR) {
            return ERROR;
        }
        ep.setStatus(MsgContext.JK_STATUS_NEW);
        rp.setStage(Constants.STAGE_KEEPALIVE);
        return OK;
    }

    @Override
    public ObjectName preRegister(MBeanServer server, ObjectName oname) throws Exception {
        // override - we must be registered as "container"
        this.name = "container";
        return super.preRegister(server, oname);
    }

    public String protocol() {
        return "AJP";
    }

    public void listen() throws IOException, InstantiationException {
        init();
        start();
    }
}