X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=src%2Fdolda%2Fjsvc%2Fscgi%2FDirServer.java;h=ef38a930672aeb49fb941b94f0b23f5332b42720;hb=c39ae6d24864244eab704dea7307f769f2f2e9da;hp=7920977a91c67102e3b11275749d0df5c757ac5f;hpb=22779185ec3cd3ab6fafdbcbe675161ceae9ce7d;p=jsvc.git

diff --git a/src/dolda/jsvc/scgi/DirServer.java b/src/dolda/jsvc/scgi/DirServer.java
index 7920977..ef38a93 100644
--- a/src/dolda/jsvc/scgi/DirServer.java
+++ b/src/dolda/jsvc/scgi/DirServer.java
@@ -7,11 +7,14 @@ import java.util.logging.*;
 import dolda.jsvc.*;
 import dolda.jsvc.util.*;
 import dolda.jsvc.j2ee.PosixArgs;
+import java.lang.management.ManagementFactory;
+import javax.management.*;
 
 public class DirServer extends Server {
     private final Map<File, DSContext> contexts = new HashMap<File, DSContext>();
-    private final Environment env;
+    public final Environment env;
     private final Logger logger = Logger.getLogger("dolda.jsvc.scgi.dirserver");
+    private Thread sdhook = null, main = null;
     
     public DirServer(ServerSocket sk, Environment env) {
 	super(sk);
@@ -24,7 +27,12 @@ public class DirServer extends Server {
 	    String act = "loaded %s as %s";
 	    if(ctx != null) {
 		if(ctx.mtime < file.lastModified()) {
-		    ctx.tg.destroy();
+		    ctx.tg.shutdown();
+		    try {
+			ManagementFactory.getPlatformMBeanServer().unregisterMBean(ctx.mbean.name);
+		    } catch(InstanceNotFoundException e) {
+		    } catch(MBeanRegistrationException e) {
+		    }
 		    contexts.remove(file);
 		    ctx = null;
 		    act = "reloaded %s as %s";
@@ -32,6 +40,12 @@ public class DirServer extends Server {
 	    }
 	    if(ctx == null) {
 		ctx = new DSContext(file, env);
+		try {
+		    ManagementFactory.getPlatformMBeanServer().registerMBean(ctx.mbean, ctx.mbean.name);
+		} catch(InstanceAlreadyExistsException e) {
+		} catch(MBeanRegistrationException e) {
+		} catch(NotCompliantMBeanException e) {
+		}
 		contexts.put(file, ctx);
 		logger.config(String.format(act, file, ctx.name()));
 	    }
@@ -51,19 +65,56 @@ public class DirServer extends Server {
 	RequestThread w = ctx.tg.respond(req);
 	w.start();
     }
+    
+    private class ShutdownHandler extends Thread {
+	public void run() {
+	    sdhook = null;
+	    DirServer.this.stop();
+	    try {
+		main.join();
+	    } catch(InterruptedException e) {}
+	}
+    }
+
+    protected void shutdown() {
+	try {
+	    if(sdhook != null)
+		Runtime.getRuntime().removeShutdownHook(sdhook);
+	} catch(Exception e) {}
+	synchronized(contexts) {
+	    for(Iterator<Map.Entry<File, DSContext>> i = contexts.entrySet().iterator(); i.hasNext();) {
+		Map.Entry<File, DSContext> e = i.next();
+		DSContext ctx = e.getValue();
+		i.remove();
+		ctx.tg.shutdown();
+	    }
+	}
+	super.shutdown();
+	try {
+	    ManagementFactory.getPlatformMBeanServer().unregisterMBean(dolda.jsvc.scgi.jmx.Server.name);
+	} catch(InstanceNotFoundException e) {
+	} catch(MBeanRegistrationException e) {
+	}
+    }
 
     private static void usage(PrintStream out) {
-	out.println("usage: dolda.jsvc.scgi.DirServer [-h] [-e CHARSET] [-d DATADIR] PORT");
+	out.println("usage: dolda.jsvc.scgi.DirServer [-h] [-B BINDADDR] [-e CHARSET] [-d DATADIR] PORT");
     }
     
     public static void main(String[] args) {
-	PosixArgs opt = PosixArgs.getopt(args, "h");
+	PosixArgs opt = PosixArgs.getopt(args, "he:d:B:");
 	if(opt == null) {
 	    usage(System.err);
 	    System.exit(1);
 	}
 	String charset = null;
 	File datroot = null;
+	InetAddress bindaddr;
+	try {
+	    bindaddr = InetAddress.getLocalHost();
+	} catch(UnknownHostException e) {
+	    throw(new Error("WTF?!", e));
+	}
 	for(char c : opt.parsed()) {
 	    switch(c) {
 	    case 'e':
@@ -76,6 +127,14 @@ public class DirServer extends Server {
 		    System.exit(1);
 		}
 		break;
+	    case 'B':
+		try {
+		    bindaddr = InetAddress.getByName(opt.arg);
+		} catch(UnknownHostException e) {
+		    System.err.println(opt.arg  + ": no such host");
+		    System.exit(1);
+		}
+		break;
 	    case 'h':
 		usage(System.out);
 		return;
@@ -89,17 +148,27 @@ public class DirServer extends Server {
 	env.initvm();
 	int port = Integer.parseInt(opt.rest[0]);
 	ServerSocket sk;
+	SocketAddress saddr = new InetSocketAddress(bindaddr, port);
 	try {
-	    sk = new ServerSocket(port);
+	    sk = new ServerSocket();
+	    sk.bind(saddr);
 	} catch(IOException e) {
-	    System.err.println("could not bind to port " + port + ": " + e.getMessage());
+	    System.err.println("could not bind to " + saddr + ": " + e.getMessage());
 	    System.exit(1);
 	    return; /* Because javac is stupid. :-/ */
 	}
 	DirServer s = new DirServer(sk, env);
+	try {
+	    ManagementFactory.getPlatformMBeanServer().registerMBean(new dolda.jsvc.scgi.jmx.Server(s), dolda.jsvc.scgi.jmx.Server.name);
+	} catch(InstanceAlreadyExistsException e) {
+	} catch(MBeanRegistrationException e) {
+	} catch(NotCompliantMBeanException e) {
+	}
 	if(charset != null)
 	    s.headcs = charset;
 	
-	new Thread(s, "SCGI server thread").start();
+	Runtime.getRuntime().addShutdownHook(s.sdhook = s.new ShutdownHandler());
+	s.main = new Thread(s, "SCGI server thread");
+	s.main.start();
     }
 }