ServerDriver.java

/**
 * Driver for WebServer class
 */

import java.io.*;
import java.util.*;
import java.util.logging.*;


public class ServerDriver {

	private static final Logger logger = Logger.getLogger("WebServer"); // global logger
	private static final int TERM_WAIT_TIME = 10000; // wait for server process to terminate (milli-seconds)
    
	public static void main(String[] args) {
        // parse command line args
        HashMap<String, String> params = parseCommandLine(args);
        
        // set the parameters
        int serverPort = Integer.parseInt( params.getOrDefault("-p", "2025") ); // server port number
        int conTimeout = Integer.parseInt( params.getOrDefault("-t", "0") ); // idle connection timeout in milli-seconds, 0 means infinity
        String rootDir = params.getOrDefault("-r", System.getProperty("user.dir")); // root directory of the web server, default is current directory
        Level logLevel = Level.parse( params.getOrDefault("-v", "all").toUpperCase() ); // log levels: all, info, off

        // standard output
        setLogLevel(logLevel);

        System.out.println("log level is set to " + logLevel);
        System.out.println("starting the server on port " + serverPort);
        System.out.println("root directory set to " + rootDir);
        WebServer server = new WebServer(serverPort, rootDir, conTimeout);
        
        // start the server
        server.start();
        System.out.println("server started, type \"quit\" to stop");
        System.out.println(".....................................");

        // wait for quit command
        waitForQuit(server);

        try {
            // shutdown the server
            System.out.println();
            System.out.println("server is shutting down...");
            server.shutdown();
            server.join(TERM_WAIT_TIME);
            System.out.println("server stopped");
        } catch (InterruptedException e) {
            // Ok, ignore
            System.out.println("server did not shutdown properly");
        }

		System.exit(0);
	}


    // wait until user types "quit" or server terminates
    private static void waitForQuit(Thread server) {
        BufferedReader console = new BufferedReader(new InputStreamReader(System.in));
        try {
            while (server.isAlive()) {
                // to avoid blocking on input
                if (console.ready()) 
                    if (console.readLine().equals("quit")) break;
            }
        } catch (IOException e) {
            // Ok, ignore
        }
    }    

	// parse command line arguments
	private static HashMap<String, String> parseCommandLine(String[] args) {
		HashMap<String, String> params = new HashMap<String, String>();

		int i = 0;
		while ((i + 1) < args.length) {
			params.put(args[i], args[i+1]);
			i += 2;
		}
		
		return params;
	}
	
	
	// set the global log level and format
	private static void setLogLevel(Level level) {
		System.setProperty("java.util.logging.SimpleFormatter.format", "%5$s %n");
		
		ConsoleHandler handler = new ConsoleHandler();
        handler.setLevel(level);
        logger.addHandler(handler);		
		logger.setLevel(level);
        logger.setUseParentHandlers(false);
	}
	
}