Quadcap Embeddable Server

com/quadcap/net/server/Worker.java

Go to the documentation of this file.

package com.quadcap.net.server;

/* Copyright 1999 - 2003 Quadcap Software.  All rights reserved.
 *
 * This software is distributed under the Quadcap Free Software License.
 * This software may be used or modified for any purpose, personal or
 * commercial.  Open Source redistributions are permitted.  Commercial
 * redistribution of larger works derived from, or works which bundle
 * this software requires a "Commercial Redistribution License"; see
 * http://www.quadcap.com/purchase.
 *
 * Redistributions qualify as "Open Source" under  one of the following terms:
 *   
 *    Redistributions are made at no charge beyond the reasonable cost of
 *    materials and delivery.
 *
 *    Redistributions are accompanied by a copy of the Source Code or by an
 *    irrevocable offer to provide a copy of the Source Code for up to three
 *    years at the cost of materials and delivery.  Such redistributions
 *    must allow further use, modification, and redistribution of the Source
 *    Code under substantially the same terms as this license.
 *
 * Redistributions of source code must retain the copyright notices as they
 * appear in each source code file, these license terms, and the
 * disclaimer/limitation of liability set forth as paragraph 6 below.
 *
 * Redistributions in binary form must reproduce this Copyright Notice,
 * these license terms, and the disclaimer/limitation of liability set
 * forth as paragraph 6 below, in the documentation and/or other materials
 * provided with the distribution.
 *
 * The Software is provided on an "AS IS" basis.  No warranty is
 * provided that the Software is free of defects, or fit for a
 * particular purpose.  
 *
 * Limitation of Liability. Quadcap Software shall not be liable
 * for any damages suffered by the Licensee or any third party resulting
 * from use of the Software.
 */

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import java.util.Hashtable;

import java.net.InetAddress;
import java.net.Socket;

import com.quadcap.util.ConfigString;
import com.quadcap.util.Debug;

/**
 * A worker is a runnable with a context, in/out streams, and (when active)
 * a socket.
 *
 * @author Stan Bailes
 */
public abstract class Worker implements Runnable {
    Server                      server;
    Object                      lock            = new Object();
    protected Socket            socket          = null;
    int                         sport           = -1;
    boolean                     terminate       = false;

    static int                  wcnt            = 0;
    int                         cnt             = wcnt++;

    protected Object             context;
    //#ifdef DEBUG
    protected FileOutputStream  log             = null;
    //#endif
    protected WorkerInputStream  win            = null;
    protected WorkerOutputStream wout           = null;

    public String toString() { return "Worker " + cnt; }
    
    public void init(Server server, Object context) {
        this.server = server;
        this.context = context;
    }

    public void init(Server server, Object context, String name)
        throws IOException
    {
        init(server, context);
        //#ifdef DEBUG
        this.log = null;
        int trc = Integer.parseInt(
            ConfigString.find("com.quadcap.net.server.trace", "0").toString());
        if (trc > 0) {
            try { new File("logs").mkdir(); } catch (Throwable t) {}
            log = new FileOutputStream("logs/" + name + "-" + cnt + ".log");
        }
        win = new WorkerInputStream(log);
        wout = new WorkerOutputStream(log);
        //#else
//-        win = new WorkerInputStream(null);
//-        wout = new WorkerOutputStream(null);
        //#endif
    }

    public int getId() { return cnt; }

    void handle(Socket socket, int sport) {
        this.socket = socket;
        this.sport = sport;
        synchronized (lock) {
            lock.notify();
        }
    }

    public void run() {
        //Jni j = new Jni("work");
        final WorkerInputStream lwin = win;
        final WorkerOutputStream lwout = wout;
        while (!terminate) {
            //j.reset();
            boolean interrupted = false;
            synchronized (lock) {
                //j.dump("sync");
                try {
                    if (socket == null) lock.wait();
                    //j.dump("got socket");
                } catch (InterruptedException e) {
                }
            }
            if (terminate) return;
            try {
                //j.dump("in try");
                lwin.reset(socket.getInputStream());
                //j.dump("win.reset");
                lwout.reset(socket.getOutputStream());
                //j.dump("wout.reset");
                doSession();
                //j.dump("doSession");
                
            } catch (Throwable t) {
                Debug.print(t);
            } finally {
                try { wout.close(); } catch (Throwable t) {}
                try { socket.close(); } catch (Throwable t) {}
                socket = null;
                server.returnIdleWorker(this);
            }
        }
    }

    public final WorkerInputStream getInputStream() {
        return win;
    }

    public final WorkerOutputStream getOutputStream() {
        return wout;
    }

    public final Socket getSocket() {
        return socket;
    }

    public final String getHostName() {
        return socket.getLocalAddress().getHostName();
    }

    public final int getPort() {
        return sport;
    }

    /**
     * Return the IP address of the agent that sent the request.
     */
    public String getRemoteAddr() {
        InetAddress ia = socket.getInetAddress();
        return ia.getHostAddress();
    }

    public String getRemoteHost() {
        InetAddress ia = socket.getInetAddress();
        return ia.getHostName();
    }
    
    public abstract void doSession() throws Exception;

    public void stop() {
        terminate = true;
        try {
            synchronized (lock) {
                lock.notifyAll();
            }
        } catch (Throwable t) {}
    }
}