← Back to branch summary

~ubuntu-branches/ubuntu/raring/libjboss-remoting-java/raring

« back to all changes in this revision

Viewing changes to src/tests/org/jboss/test/remoting/soak/ServerLauncher.java

  • Committer: Package Import Robot
  • Author(s): Torsten Werner
  • Date: 2011-09-09 14:01:03 UTC
  • mfrom: (1.1.6 upstream)
  • Revision ID: package-import@ubuntu.com-20110909140103-hqokx61534tas9rg
Tags: 2.5.3.SP1-1
http://bugs.debian.org/628303
* Newer but not newest upstream release. Do not build samples.
* Change debian/watch to upstream's svn repo.
* Add patch to fix compile error caused by tomcat update.
  (Closes: #628303)
* Switch to source format 3.0.
* Switch to debhelper level 7.
* Remove useless Depends.
* Update Standards-Version: 3.9.2.
* Update README.source.

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/tests/org/jboss/test/remoting/soak/ServerLauncher.java
 
1
/*
 
2
* JBoss, Home of Professional Open Source
 
3
* Copyright 2005, JBoss Inc., and individual contributors as indicated
 
4
* by the @authors tag. See the copyright.txt in the distribution for a
 
5
* full listing of individual contributors.
 
6
*
 
7
* This is free software; you can redistribute it and/or modify it
 
8
* under the terms of the GNU Lesser General Public License as
 
9
* published by the Free Software Foundation; either version 2.1 of
 
10
* the License, or (at your option) any later version.
 
11
*
 
12
* This software is distributed in the hope that it will be useful,
 
13
* but WITHOUT ANY WARRANTY; without even the implied warranty of
 
14
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 
15
* Lesser General Public License for more details.
 
16
*
 
17
* You should have received a copy of the GNU Lesser General Public
 
18
* License along with this software; if not, write to the Free
 
19
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 
20
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 
21
*/
 
22
package org.jboss.test.remoting.soak;
 
23
 
 
24
import java.net.InetAddress;
 
25
import java.util.HashMap;
 
26
import java.util.HashSet;
 
27
import java.util.Iterator;
 
28
import java.util.Map;
 
29
import java.util.Set;
 
30
 
 
31
import javax.management.MBeanServer;
 
32
 
 
33
import org.apache.log4j.ConsoleAppender;
 
34
import org.apache.log4j.Level;
 
35
import org.apache.log4j.Logger;
 
36
import org.apache.log4j.PatternLayout;
 
37
import org.jboss.logging.XLevel;
 
38
import org.jboss.remoting.InvocationRequest;
 
39
import org.jboss.remoting.InvokerLocator;
 
40
import org.jboss.remoting.ServerInvocationHandler;
 
41
import org.jboss.remoting.ServerInvoker;
 
42
import org.jboss.remoting.callback.Callback;
 
43
import org.jboss.remoting.callback.InvokerCallbackHandler;
 
44
import org.jboss.remoting.callback.ServerInvokerCallbackHandler;
 
45
import org.jboss.remoting.transport.Connector;
 
46
 
 
47
 
 
48
/**
 
49
 * 
 
50
 * @author <a href="ron.sigal@jboss.com">Ron Sigal</a>
 
51
 * @version $Revision: 1.1 $
 
52
 * <p>
 
53
 * Copyright Mar 13, 2008
 
54
 * </p>
 
55
 */
 
56
public class ServerLauncher extends SoakConstants
 
57
{
 
58
   private static Logger log = Logger.getLogger(ServerLauncher.class);
 
59
   private static Map locators = new HashMap();
 
60
   private static Connector[] connectors = new Connector[4];
 
61
   
 
62
   public static Map getLocators()
 
63
   {
 
64
      return locators;
 
65
   }
 
66
   
 
67
   public static void main(String[] args)
 
68
   {
 
69
      try
 
70
      {
 
71
         Logger.getLogger("org.jboss.remoting").setLevel(XLevel.INFO);
 
72
         Logger.getLogger("org.jboss.test.remoting").setLevel(Level.INFO);
 
73
         String pattern = "[%d{ABSOLUTE}] [%t] %5p (%F:%L) - %m%n";
 
74
         PatternLayout layout = new PatternLayout(pattern);
 
75
         ConsoleAppender consoleAppender = new ConsoleAppender(layout);
 
76
         Logger.getRootLogger().addAppender(consoleAppender); 
 
77
         
 
78
         String host = InetAddress.getLocalHost().getHostAddress();
 
79
         
 
80
         connectors[0] = setupServer(host, 6666, "bisocket");
 
81
         locators.put("bisocket", connectors[0].getLocator().getLocatorURI());
 
82
         
 
83
         connectors[1] = setupServer(host, 6667, "http");
 
84
         locators.put("http", connectors[1].getLocator().getLocatorURI());
 
85
         
 
86
         connectors[2] = setupServer(host, 6668, "rmi");
 
87
         locators.put("rmi", connectors[2].getLocator().getLocatorURI());
 
88
         
 
89
         connectors[3] = setupServer(host, 6669, "socket");
 
90
         locators.put("socket", connectors[3].getLocator().getLocatorURI());
 
91
         
 
92
         log.info("SERVERS CREATED: " + locators);
 
93
         
 
94
         System.in.read();
 
95
         System.in.read();
 
96
         System.in.read();
 
97
         
 
98
         log.info("SHUTTING DOWN SERVERS");
 
99
         for (int i = 0; i < connectors.length; i++)
 
100
         {
 
101
            connectors[i].stop();
 
102
         }
 
103
         log.info("SERVERS SHUT DOWN");
 
104
         
 
105
      }
 
106
      catch (Exception e)
 
107
      {
 
108
         log.error("Error", e);
 
109
      }
 
110
   }
 
111
   
 
112
   
 
113
   protected static Connector setupServer(String host, int port, String transport) throws Exception
 
114
   {
 
115
      String locatorURI = transport + "://" + host + ":" + port + "/?timeout=0"; 
 
116
      InvokerLocator serverLocator = new InvokerLocator(locatorURI);
 
117
      log.info("Starting remoting server with locator uri of: " + locatorURI);
 
118
      HashMap config = new HashMap();
 
119
      config.put(InvokerLocator.FORCE_REMOTE, "true");
 
120
      Connector connector = new Connector(serverLocator, config);
 
121
      connector.create();
 
122
      TestInvocationHandler invocationHandler = new TestInvocationHandler();
 
123
      connector.addInvocationHandler("test", invocationHandler);
 
124
      connector.start();
 
125
      return connector;
 
126
   }
 
127
   
 
128
   
 
129
   static class TestInvocationHandler implements ServerInvocationHandler
 
130
   {
 
131
      Map listeners = new HashMap();
 
132
      Object lock = new Object();
 
133
      
 
134
      public void addListener(InvokerCallbackHandler callbackHandler)
 
135
      {
 
136
         log.debug("entering addListener()");
 
137
         synchronized (lock)
 
138
         {
 
139
            String id = ((ServerInvokerCallbackHandler)callbackHandler).getClientSessionId();
 
140
            listeners.put(id, callbackHandler);
 
141
         }
 
142
         log.debug("added InvokerCallbackHandler: " + listeners);
 
143
      }
 
144
      
 
145
      public Object invoke(final InvocationRequest invocation) throws Throwable
 
146
      {
 
147
         String command = (String) invocation.getParameter();
 
148
         log.debug("command: " + command);
 
149
         
 
150
         if (COPY.equals(command))
 
151
         {
 
152
            Object o = invocation.getRequestPayload().get(PAYLOAD);
 
153
            return o;
 
154
         }
 
155
         else if (SPIN.equals(command))
 
156
         {
 
157
            String s = (String) invocation.getRequestPayload().get(SPIN_TIME);
 
158
            int spinTime = Integer.parseInt(s);
 
159
            SpinThread t = new SpinThread();
 
160
            t.start();
 
161
            Thread.sleep(spinTime);
 
162
            t.setStop();
 
163
            return "done";
 
164
         }
 
165
         else if (CALLBACK.equals(command))
 
166
         {
 
167
            String id = invocation.getSessionId();
 
168
            Map requestPayload = invocation.getRequestPayload();
 
169
            String s = (String) requestPayload.get(NUMBER_OF_CALLBACKS);
 
170
            int callbacks = Integer.parseInt(s);
 
171
            InvokerCallbackHandler callbackHandler = null;
 
172
 
 
173
            synchronized (lock)
 
174
            {
 
175
               callbackHandler = (InvokerCallbackHandler) listeners.get(id);
 
176
            }
 
177
 
 
178
            if (callbackHandler == null)
 
179
            {
 
180
               log.debug("sessionId: " + id);
 
181
               log.debug("listeners: " + listeners);
 
182
            }
 
183
            Callback callback = new Callback("callback");
 
184
            for (int i = 0; i < callbacks; i++)
 
185
            {
 
186
               callbackHandler.handleCallback(callback);
 
187
            }
 
188
         }
 
189
 
 
190
         return command;
 
191
      }
 
192
      
 
193
      public void removeListener(InvokerCallbackHandler callbackHandler)
 
194
      {
 
195
         synchronized (lock)
 
196
         {
 
197
            String id = ((ServerInvokerCallbackHandler) callbackHandler).getClientSessionId();
 
198
            listeners.remove(id);
 
199
         }
 
200
      }
 
201
      
 
202
      public void setMBeanServer(MBeanServer server) {}
 
203
      public void setInvoker(ServerInvoker invoker) {}
 
204
   }
 
205
   
 
206
   static class SpinThread extends Thread
 
207
   {
 
208
      boolean stop;
 
209
      static int counter;
 
210
      static Object lock = new Object();
 
211
      
 
212
      public SpinThread()
 
213
      {
 
214
         synchronized (lock)
 
215
         {
 
216
            setName("spinThread:" + counter++);
 
217
         }
 
218
      }
 
219
      public void setStop()
 
220
      {
 
221
         stop = true;
 
222
         log.debug(this + " stop = " + stop);
 
223
      }
 
224
      
 
225
      public void run()
 
226
      {
 
227
         int n = 0;
 
228
         while (!stop)
 
229
         {
 
230
            n++;
 
231
            if ((n + 1) % 10000 == 0)
 
232
               log.debug(this + "stop = " + stop);
 
233
         }
 
234
         log.debug("SpinThread done");
 
235
      }
 
236
   }
 
237
}
 
238