001: /*
002: * Licensed to the Apache Software Foundation (ASF) under one or more
003: * contributor license agreements. See the NOTICE file distributed with
004: * this work for additional information regarding copyright ownership.
005: * The ASF licenses this file to You under the Apache License, Version 2.0
006: * (the "License"); you may not use this file except in compliance with
007: * the License. You may obtain a copy of the License at
008: *
009: * http://www.apache.org/licenses/LICENSE-2.0
010: *
011: * Unless required by applicable law or agreed to in writing, software
012: * distributed under the License is distributed on an "AS IS" BASIS,
013: * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014: * See the License for the specific language governing permissions and
015: * limitations under the License.
016: *
017: */
018:
019: package org.apache.jmeter.engine;
020:
021: import java.net.InetAddress;
022: import java.rmi.RemoteException;
023: import java.rmi.registry.LocateRegistry;
024: import java.rmi.registry.Registry;
025: import java.util.Date;
026: import java.util.Properties;
027:
028: import org.apache.jmeter.util.JMeterUtils;
029: import org.apache.jorphan.collections.HashTree;
030: import org.apache.jorphan.logging.LoggingManager;
031: import org.apache.log.Logger;
032:
033: /**
034: * This is the JMeter server main code.
035: */
036: public class RemoteJMeterEngineImpl extends
037: java.rmi.server.UnicastRemoteObject implements
038: RemoteJMeterEngine {
039: private static final Logger log = LoggingManager
040: .getLoggerForClass();
041:
042: private JMeterEngine backingEngine;
043:
044: private String hostName;
045:
046: public static final int DEFAULT_RMI_PORT = JMeterUtils
047: .getPropDefault("server.rmi.port", 1099); // $NON-NLS-1$
048:
049: // Should we create our own copy of the RMI registry?
050: private static final boolean createServer = JMeterUtils
051: .getPropDefault("server.rmi.create", true); // $NON-NLS-1$
052:
053: public RemoteJMeterEngineImpl() throws RemoteException {
054: init(DEFAULT_RMI_PORT);
055: }
056:
057: public RemoteJMeterEngineImpl(int port) throws RemoteException {
058: init(port == 0 ? DEFAULT_RMI_PORT : port);
059: }
060:
061: private void init(int port) throws RemoteException {
062: log.info("Starting backing engine on " + port);
063: log.debug("This = " + this );
064: if (createServer) {
065: log.info("Creating RMI registry (server.rmi.create=true)");
066: try {
067: LocateRegistry.createRegistry(port);
068: } catch (RemoteException e) {
069: String msg = "Problem creating registry: " + e;
070: log.warn(msg);
071: System.err.println(msg);
072: System.err.println("Continuing...");
073: }
074: }
075: try {
076: Registry reg = LocateRegistry.getRegistry(port);
077: hostName = InetAddress.getLocalHost().getHostName();
078: log.info("Creating JMeter engine on host " + hostName);
079: backingEngine = new StandardJMeterEngine(hostName);// see setHost()
080: reg.rebind("JMeterEngine", this ); // $NON-NLS-1$
081: log.info("Bound to registry on port " + port);
082: } catch (Exception ex) {
083: log
084: .error("rmiregistry needs to be running to start JMeter in server "
085: + "mode\n\t" + ex.toString());
086: // Throw an Exception to ensure caller knows ...
087: throw new RemoteException(
088: "Cannot start. See server log file.");
089: }
090: }
091:
092: // TODO: is this really needed? The hostname is passed in when the engine is created
093: public void setHost(String host) {
094: hostName = host;
095: log.info("received host: " + host);
096: backingEngine.setHost(host);
097: }
098:
099: /**
100: * Adds a feature to the ThreadGroup attribute of the RemoteJMeterEngineImpl
101: * object.
102: *
103: * @param testTree
104: * the feature to be added to the ThreadGroup attribute
105: */
106: public void configure(HashTree testTree) throws RemoteException {
107: log.info("received test tree");
108: backingEngine.configure(testTree);
109: }
110:
111: public void runTest() throws RemoteException, JMeterEngineException {
112: log.info("running test");
113: log.debug("This = " + this );
114: long now = System.currentTimeMillis();
115: System.out.println("Starting the test on host " + hostName
116: + " @ " + new Date(now) + " (" + now + ")");
117: backingEngine.runTest();
118: }
119:
120: public void reset() throws RemoteException {
121: log.info("Reset");
122: backingEngine.reset();
123: }
124:
125: public void stopTest() throws RemoteException {
126: log.info("Stopping test");
127: backingEngine.stopTest();// TODO: askThreadsToStop() instead?
128: }
129:
130: public void exit() throws RemoteException {
131: log.info("Exitting");
132: backingEngine.exit();
133: }
134:
135: public void setProperties(Properties p) throws RemoteException {
136: backingEngine.setProperties(p);
137: }
138: }
|