blob: 3e89652502d8496c30e0ca9cfc3640e1c12ac86c (
plain)
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
|
package org.simpleframework.http.socket;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.nio.channels.SelectableChannel;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;
import org.simpleframework.transport.trace.TraceAnalyzer;
import org.simpleframework.transport.trace.Trace;
public class WebSocketAnalyzer implements TraceAnalyzer {
private final Map<SelectableChannel, Integer> map;
private final AtomicInteger count;
private final boolean debug;
public WebSocketAnalyzer() {
this(true);
}
public WebSocketAnalyzer(boolean debug) {
this.map = new ConcurrentHashMap<SelectableChannel, Integer>();
this.count = new AtomicInteger();
this.debug = debug;
}
public Trace attach(SelectableChannel channel) {
if(map.containsKey(channel)) {
throw new IllegalStateException("Can't attach twice");
}
final int counter = count.getAndIncrement();
map.put(channel, counter);
return new Trace() {
public void trace(Object event) {
if(debug) {
trace(event, "");
}
}
public void trace(Object event, Object value) {
if(debug) {
if(value instanceof Throwable) {
StringWriter writer = new StringWriter();
PrintWriter out = new PrintWriter(writer);
((Exception)value).printStackTrace(out);
out.flush();
value = writer.toString();
}
if(value != null && !String.valueOf(value).isEmpty()) {
System.err.printf("(%s) [%s] %s: %s%n", Thread.currentThread().getName(), counter, event, value);
} else {
System.err.printf("(%s) [%s] %s%n", Thread.currentThread().getName(), counter, event);
}
}
}
};
}
public void stop() {
System.err.println("Stop agent");
}
}
|