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
|
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.fs4;
import java.io.BufferedOutputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.ByteBuffer;
import java.util.HashMap;
import java.util.Map;
import java.util.logging.Logger;
import com.yahoo.fs4.mplex.FS4Channel;
import com.yahoo.log.LogLevel;
import com.yahoo.search.Query;
/**
* Responsible for dumping query & query result packets
*
* @author tonytv
*/
public class PacketDumper implements PacketListener {
/** High level representation of packet types (e.g. query, result, ...) */
public static enum PacketType {
query(QueryPacket.class),
result(QueryResultPacket.class);
Class<? extends BasicPacket> implementationType;
PacketType(Class<? extends BasicPacket> implementationType) {
this.implementationType = implementationType;
}
}
private static Logger log = Logger.getLogger(PacketDumper.class.getSimpleName());
private volatile boolean disabled = true;
private final File logDirectory;
private final Map<Class<? extends BasicPacket>, DataOutputStream> dumpFiles =
new HashMap<>();
private final String fileNamePattern;
private void handlePacket(FS4Channel channel, BasicPacket packet, ByteBuffer serializedForm, String direction) {
//minimize overhead when disabled:
if (disabled)
return;
try {
DataOutputStream stream = getOutputStream(packet);
if (stream != null) {
synchronized (stream) {
stream.writeChars(packet.getTimestamp() + " " + direction + " packet on channel " + channel.getChannelId());
String indent = " ";
Query query = channel.getQuery();
if (query != null)
stream.writeChars('\n' + indent + "Query: '" + query.getModel().getQueryString());
hexDump(indent, stream, serializedForm);
stream.writeChar('\n');
stream.flush();
}
}
} catch (IOException e) {
log.log(LogLevel.WARNING, "Could not log packet.", e);
}
}
private void hexDump(String indent, DataOutputStream stream, ByteBuffer serializedForm) throws IOException {
HexByteIterator hexByteIterator = new HexByteIterator(serializedForm);
long count = 0;
final int maxNumCharacters = 80;
while (hexByteIterator.hasNext()) {
if (count++ % maxNumCharacters == 0)
stream.writeChar('\n');
stream.writeChars(hexByteIterator.next());
}
}
private synchronized DataOutputStream getOutputStream(BasicPacket packet) {
return dumpFiles.get(packet.getClass());
}
public void packetSent(FS4Channel channel, BasicPacket packet, ByteBuffer serializedForm) {
handlePacket(channel, packet, serializedForm, "Sent");
}
public void packetReceived(FS4Channel channel, BasicPacket packet, ByteBuffer serializedForm) {
handlePacket(channel, packet, serializedForm, "Received");
}
public synchronized void dumpPackets(PacketType packetType, boolean on) throws IOException {
OutputStream stream = dumpFiles.get(packetType.implementationType);
if (!on && stream != null)
closeFile(stream, packetType);
else if (on && stream == null)
openFile(packetType);
}
private void openFile(PacketType packetType) throws FileNotFoundException {
if (!logDirectory.exists() ||
logDirectory.mkdirs()) {
throw new RuntimeException("PacketDumper: Could not create log directory " + logDirectory);
}
String fileName = fileNamePattern.replace("%s", packetType.toString());
boolean append = true;
DataOutputStream outputStream =
new DataOutputStream(
new BufferedOutputStream(
new FileOutputStream(new File(logDirectory, fileName), append)));
dumpFiles.put(packetType.implementationType, outputStream);
disabled = dumpFiles.isEmpty();
}
private void closeFile(OutputStream stream, PacketType packetType) throws IOException {
try {
synchronized (stream) {
stream.close();
}
} finally {
dumpFiles.remove(packetType.implementationType);
disabled = dumpFiles.isEmpty();
}
}
public PacketDumper(File logDirectory, String fileNamePattern) {
this.logDirectory = logDirectory;
this.fileNamePattern = fileNamePattern;
}
}
|
