|
1
22 package org.jboss;
23
24 import java.lang.reflect.InvocationHandler  ;
25 import java.lang.reflect.Method ;
26 import java.lang.reflect.Proxy ;
27 import java.lang.reflect.UndeclaredThrowableException ;
28 import java.util.Hashtable ;
29 import java.util.ArrayList ;
30 import java.io.BufferedReader ;
31 import java.io.InputStreamReader ;
32 import javax.management.MBeanServerConnection ;
33 import javax.management.ObjectName ;
34 import javax.naming.Context ;
35 import javax.naming.InitialContext ;
36
37
38 import gnu.getopt.Getopt;
39 import gnu.getopt.LongOpt;
40
41 import org.jboss.system.server.Server;
42 import org.jboss.system.server.ServerImplMBean;
43 import org.jboss.security.SecurityAssociation;
44 import org.jboss.security.SimplePrincipal;
45 import org.jnp.interfaces.NamingContext;
46
47
55 public class Shutdown
56 {
57
61 public static final String PROGRAM_NAME = System.getProperty("program.name", "shutdown");
62
63 protected static void displayUsage()
64 {
65 System.out.println("A JMX client to shutdown (exit or halt) a remote JBoss server.");
66 System.out.println();
67 System.out.println("usage: " + PROGRAM_NAME + " [options] <operation>");
68 System.out.println();
69 System.out.println("options:");
70 System.out.println(" -h, --help Show this help message (default)");
71 System.out.println(" -D<name>[=<value>] Set a system property");
72 System.out.println(" -- Stop processing options");
73 System.out.println(" -s, --server=<url> Specify the JNDI URL of the remote server");
74 System.out.println(" -n, --serverName=<url> Specify the JMX name of the ServerImpl");
75 System.out.println(" -a, --adapter=<name> Specify JNDI name of the MBeanServerConnection to use");
76 System.out.println(" -u, --user=<name> Specify the username for authentication");
77 System.out.println(" -p, --password=<name> Specify the password for authentication");
78 System.out.println();
79 System.out.println("operations:");
80 System.out.println(" -S, --shutdown Shutdown the server");
81 System.out.println(" -e, --exit=<code> Force the VM to exit with a status code");
82 System.out.println(" -H, --halt=<code> Force the VM to halt with a status code");
83 System.out.println();
84 }
85
86 public static void main(final String [] args) throws Exception
87 {
88 if (args.length == 0)
89 {
90 displayUsage();
91 System.exit(0);
92 }
93
94 String sopts = "-:hD:s:n:a:u:p:Se:H:";
95 LongOpt[] lopts =
96 {
97 new LongOpt("help", LongOpt.NO_ARGUMENT, null, 'h'),
98 new LongOpt("server", LongOpt.REQUIRED_ARGUMENT, null, 's'),
99 new LongOpt("adapter", LongOpt.REQUIRED_ARGUMENT, null, 'a'),
100 new LongOpt("serverName", LongOpt.REQUIRED_ARGUMENT, null, 'n'),
101 new LongOpt("shutdown", LongOpt.NO_ARGUMENT, null, 'S'),
102 new LongOpt("exit", LongOpt.REQUIRED_ARGUMENT, null, 'e'),
103 new LongOpt("halt", LongOpt.REQUIRED_ARGUMENT, null, 'H'),
104 new LongOpt("user", LongOpt.REQUIRED_ARGUMENT, null, 'u'),
105 new LongOpt("password", LongOpt.REQUIRED_ARGUMENT, null, 'p'),
106 };
107
108 Getopt getopt = new Getopt(PROGRAM_NAME, args, sopts, lopts);
109 int code;
110 String arg;
111
112 String serverURL = null;
113 String adapterName = "jmx/rmi/RMIAdaptor";
114 String username = null;
115 String password = null;
116 ObjectName serverJMXName = ServerImplMBean.OBJECT_NAME;
117 boolean exit = false;
118 boolean halt = false;
119 int exitcode = -1;
120
121 while ((code = getopt.getopt()) != -1)
122 {
123 switch (code)
124 {
125 case ':':
126 case '?':
127 System.exit(1);
129 break;
130
131 case 1:
132 System.err.println(PROGRAM_NAME + ": unused non-option argument: " +
135 getopt.getOptarg());
136 break;
137 case 'h':
138 displayUsage();
139 System.exit(0);
140 break;
141 case 'D':
142 {
143 arg = getopt.getOptarg();
145 String name, value;
146 int i = arg.indexOf("=");
147 if (i == -1)
148 {
149 name = arg;
150 value = "true";
151 }
152 else
153 {
154 name = arg.substring(0, i);
155 value = arg.substring(i + 1, arg.length());
156 }
157 System.setProperty(name, value);
158 break;
159 }
160 case 's':
161 serverURL = getopt.getOptarg();
162 break;
163 case 'n':
164 serverJMXName = new ObjectName (getopt.getOptarg());
165 break;
166 case 'S':
167 break;
169 case 'a':
170 adapterName = getopt.getOptarg();
171 break;
172 case 'u':
173 username = getopt.getOptarg();
174 SecurityAssociation.setPrincipal(new SimplePrincipal(username));
175 break;
176 case 'p':
177 password = getopt.getOptarg();
178 SecurityAssociation.setCredential(password);
179 break;
180 case 'e':
181 exitcode = Integer.parseInt(getopt.getOptarg());
182 exit = true;
183 break;
184 case 'H':
185 exitcode = Integer.parseInt(getopt.getOptarg());
186 halt = true;
187 break;
188 }
189 }
190
191 InitialContext ctx;
192
193 if( username != null && password == null )
195 {
196 System.out.print("Enter password for "+username+": ");
197 BufferedReader br = new BufferedReader (new InputStreamReader (System.in));
198 password = br.readLine();
199 SecurityAssociation.setCredential(password);
200 }
201
202 if (serverURL == null)
203 {
204 ctx = new InitialContext ();
205 }
206 else
207 {
208 Hashtable env = new Hashtable ();
209 env.put(Context.PROVIDER_URL, serverURL);
210 env.put(NamingContext.JNP_DISABLE_DISCOVERY, "true");
211 ctx = new InitialContext (env);
212 }
213
214 Object obj = ctx.lookup(adapterName);
215 if (!(obj instanceof MBeanServerConnection ))
216 {
217 throw new RuntimeException ("Object not of type: MBeanServerConnection, but: " +
218 (obj == null ? "not found" : obj.getClass().getName()));
219 }
220
221 MBeanServerConnection adaptor = (MBeanServerConnection ) obj;
222 ServerProxyHandler handler = new ServerProxyHandler(adaptor, serverJMXName);
223 Class [] ifaces = {Server.class};
224 ClassLoader tcl = Thread.currentThread().getContextClassLoader();
225 Server server = (Server) Proxy.newProxyInstance(tcl, ifaces, handler);
226
227 if (exit)
228 {
229 server.exit(exitcode);
230 }
231 else if (halt)
232 {
233 server.halt(exitcode);
234 }
235 else
236 {
237 server.shutdown();
238 }
239 System.out.println("Shutdown message has been posted to the server.");
240 System.out.println("Server shutdown may take a while - check logfiles for completion");
241 }
242
243 private static class ServerProxyHandler implements InvocationHandler
244 {
245 ObjectName serverName;
246 MBeanServerConnection server;
247 ServerProxyHandler(MBeanServerConnection server, ObjectName serverName)
248 {
249 this.server = server;
250 this.serverName = serverName;
251 }
252
253 public Object invoke(Object proxy, Method method, Object [] args)
254 throws Throwable
255 {
256 String methodName = method.getName();
257 Class [] sigTypes = method.getParameterTypes();
258 ArrayList sigStrings = new ArrayList ();
259 for(int s = 0; s < sigTypes.length; s ++)
260 sigStrings.add(sigTypes[s].getName());
261 String [] sig = new String [sigTypes.length];
262 sigStrings.toArray(sig);
263 Object value = null;
264 try
265 {
266 value = server.invoke(serverName, methodName, args, sig);
267 }
268 catch(UndeclaredThrowableException e)
269 {
270 System.out.println("getUndeclaredThrowable: "+e.getUndeclaredThrowable());
271 throw e.getUndeclaredThrowable();
272 }
273 return value;
274 }
275 }
276 }
277 |
Popular Tags |
Java API By Example, From Geeks To Geeks.