1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
/*
* JBoss, Home of Professional Open Source
* Copyright 2005, JBoss Inc., and individual contributors as indicated
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.test.remoting.performance.synchronous;
import org.jboss.remoting.callback.Callback;
import org.jboss.remoting.callback.HandleCallbackException;
import org.jboss.remoting.callback.InvokerCallbackHandler;
import EDU.oswego.cs.dl.util.concurrent.SynchronizedInt;
/**
* @author <a href="mailto:tom.elrod@jboss.com">Tom Elrod</a>
*/
public class CallTracker
{
private String clientSessionId;
private int totalCount = 0;
private SynchronizedInt duplicateCount = new SynchronizedInt(0);
private SynchronizedInt receivedCount = new SynchronizedInt(0);
private boolean[] counterArray = null;
private InvokerCallbackHandler callbackHandler;
public CallTracker(String sessionId, InvokerCallbackHandler callbackHandler)
{
this.clientSessionId = sessionId;
this.callbackHandler = callbackHandler;
}
public synchronized void createTotalCount(int totalCount)
{
counterArray = new boolean[totalCount + 1];
this.totalCount = totalCount;
receivedCount.set(0);
duplicateCount.set(0);
}
public void verifyClientInvokeCount(int clientInvokeCallCount)
{
boolean duplicate = addToReceivedCount(clientInvokeCallCount);
if(duplicate)
{
duplicateCount.increment();
}
else
{
receivedCount.increment();
}
int currentDuplicateCount = duplicateCount.get();
int currentReceivedCount = receivedCount.get();
if((currentReceivedCount % 100) == 0)
{
System.out.println(clientSessionId + " -- Received count: " + currentReceivedCount);
System.out.println(clientSessionId + " -- Duplicate count: " + currentDuplicateCount);
System.out.println(clientSessionId + " -- Total count: " + totalCount);
}
if((currentReceivedCount + currentDuplicateCount) == totalCount)
{
System.out.println("\n\n*****************************\n" +
" Test Finished\n" +
"*****************************\n" +
" " + clientSessionId + " -- Received Count = " + currentReceivedCount + "\n" +
" " + clientSessionId + " -- Duplicate Count = " + currentDuplicateCount + "\n" +
"*****************************\n\n");
// now call back on client to indicate finished server processing
if(callbackHandler != null)
{
Callback callback = new Callback(new Integer[]{new Integer(currentReceivedCount),
new Integer(currentDuplicateCount)});
try
{
callbackHandler.handleCallback(callback);
}
catch(HandleCallbackException e)
{
e.printStackTrace();
}
}
}
}
private synchronized boolean addToReceivedCount(int localClientInvokeCount)
{
boolean isDuplicate = false;
if(counterArray == null)
{
System.out.println("Error! Have not received invoke for method 'totalCallCount', so can not process count.");
throw new RuntimeException("Error! Have not received invoke for method 'totalCallCount', so can not process count.");
}
else
{
try
{
isDuplicate = counterArray[localClientInvokeCount];
if(!isDuplicate)
{
counterArray[localClientInvokeCount] = true;
}
}
catch(ArrayIndexOutOfBoundsException e)
{
System.err.println("Got ArrayIndexOutOfBoundsException");
System.err.println("Counter array size = " + counterArray.length);
System.err.println("Received count = " + localClientInvokeCount);
e.printStackTrace();
}
}
return isDuplicate;
}
}
|
