public virtual IMessageTree Decode(ChannelBuffer buf)
{
DefaultMessageTree tree = new DefaultMessageTree();
Decode(buf, tree);
return(tree);
}
protected internal void DecodeHeader(ChannelBuffer buf, IMessageTree tree)
{
BufferHelper helper = _mBufferHelper;
string id = helper.Read(buf, TAB);
string domain = helper.Read(buf, TAB);
string hostName = helper.Read(buf, TAB);
string ipAddress = helper.Read(buf, TAB);
string threadGroupName = helper.Read(buf, TAB);
string threadId = helper.Read(buf, TAB);
string threadName = helper.Read(buf, TAB);
string messageId = helper.Read(buf, TAB);
string parentMessageId = helper.Read(buf, TAB);
string rootMessageId = helper.Read(buf, TAB);
string sessionToken = helper.Read(buf, LF);
if (ID.Equals(id))
{
tree.Domain = domain;
tree.HostName = hostName;
tree.IpAddress = ipAddress;
tree.ThreadGroupName = threadGroupName;
tree.ThreadId = threadId;
tree.ThreadName = threadName;
tree.MessageId = messageId;
tree.ParentMessageId = parentMessageId;
tree.RootMessageId = rootMessageId;
tree.SessionToken = sessionToken;
}
else
{
throw new Exception("Unrecognized id(" + id + ") for plain text message codec!");
}
}
public virtual IMessageTree Decode(ChannelBuffer buf)
{
DefaultMessageTree tree = new DefaultMessageTree();
Decode(buf, tree);
return tree;
}
protected internal int EncodeHeader(IMessageTree tree, ChannelBuffer buf)
{
BufferHelper helper = _mBufferHelper;
int count = 0;
count += helper.Write(buf, ID);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.Domain);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.HostName);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.IpAddress);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.ThreadGroupName);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.ThreadId);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.ThreadName);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.MessageId);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.ParentMessageId);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.RootMessageId);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.SessionToken);
count += helper.Write(buf, LF);
return(count);
}
protected internal void DecodeHeader(ChannelBuffer buf, IMessageTree tree)
{
BufferHelper helper = _mBufferHelper;
string id = helper.Read(buf, TAB);
string domain = helper.Read(buf, TAB);
string hostName = helper.Read(buf, TAB);
string ipAddress = helper.Read(buf, TAB);
string threadGroupName = helper.Read(buf, TAB);
string threadId = helper.Read(buf, TAB);
string threadName = helper.Read(buf, TAB);
string messageId = helper.Read(buf, TAB);
string parentMessageId = helper.Read(buf, TAB);
string rootMessageId = helper.Read(buf, TAB);
string sessionToken = helper.Read(buf, LF);
if (ID.Equals(id))
{
tree.Domain = domain;
tree.HostName = hostName;
tree.IpAddress = ipAddress;
tree.ThreadGroupName = threadGroupName;
tree.ThreadId = threadId;
tree.ThreadName = threadName;
tree.MessageId = messageId;
tree.ParentMessageId = parentMessageId;
tree.RootMessageId = rootMessageId;
tree.SessionToken = sessionToken;
}
else
{
throw new Exception("Unrecognized id(" + id + ") for plain text message codec!");
}
}
public virtual void Decode(ChannelBuffer buf, IMessageTree tree)
{
DecodeHeader(buf, tree);
if (buf.ReadableBytes() > 0)
{
DecodeMessage(buf, tree);
}
}
public virtual void Decode(ChannelBuffer buf, IMessageTree tree)
{
DecodeHeader(buf, tree);
if (buf.ReadableBytes() > 0)
{
DecodeMessage(buf, tree);
}
}
public int Write(ChannelBuffer buf, string str)
{
if (str == null)
{
str = "null";
}
byte[] data = _mEncoding.GetBytes(str);
buf.WriteBytes(data);
return(data.Length);
}
public int WriteRaw(ChannelBuffer buf, string str)
{
if (str == null)
{
str = "null";
}
byte[] data = _mEncoding.GetBytes(str);
int len = data.Length;
int count = len;
int offset = 0;
for (int i = 0; i < len; i++)
{
byte b = data[i];
if (b == '\t' || b == '\r' || b == '\n' || b == '\\')
{
buf.WriteBytes(data, offset, i - offset);
buf.WriteByte('\\');
if (b == '\t')
{
buf.WriteByte('t');
}
else if (b == '\r')
{
buf.WriteByte('r');
}
else if (b == '\n')
{
buf.WriteByte('n');
}
else
{
buf.WriteByte(b);
}
count++;
offset = i + 1;
}
}
if (len > offset)
{
buf.WriteBytes(data, offset, len - offset);
}
return(count);
}
public virtual void Encode(IMessageTree tree, ChannelBuffer buf)
{
int count = 0;
buf.WriteInt(0); // place-holder
count += EncodeHeader(tree, buf);
if (tree.Message != null)
{
count += EncodeMessage(tree.Message, buf);
}
buf.SetInt(0, count);
}
public virtual void Encode(IMessageTree tree, ChannelBuffer buf)
{
int count = 0;
buf.WriteInt(0); // place-holder
count += EncodeHeader(tree, buf);
if (tree.Message != null)
{
count += EncodeMessage(tree.Message, buf);
}
buf.SetInt(0, count);
}
protected internal int EncodeLine(IMessage message, ChannelBuffer buf, char type, Policy policy)
{
BufferHelper helper = _mBufferHelper;
int count = 0;
count += helper.Write(buf, (byte)type);
if (type == 'T' && message is ITransaction)
{
long duration = ((ITransaction)message).DurationInMillis;
count += helper.Write(buf, _mDateHelper.Format(message.Timestamp + duration));
}
else
{
count += helper.Write(buf, _mDateHelper.Format(message.Timestamp));
}
count += helper.Write(buf, TAB);
count += helper.Write(buf, message.Type);
count += helper.Write(buf, TAB);
count += helper.Write(buf, message.Name);
count += helper.Write(buf, TAB);
if (policy != Policy.WITHOUT_STATUS)
{
count += helper.Write(buf, message.Status);
count += helper.Write(buf, TAB);
Object data = message.Data;
if (policy == Policy.WITH_DURATION && message is ITransaction)
{
long duration0 = ((ITransaction)message).DurationInMicros;
count += helper.Write(buf, duration0.ToString(CultureInfo.InvariantCulture));
//以微秒为单位
count += helper.Write(buf, "us");
count += helper.Write(buf, TAB);
}
count += helper.WriteRaw(buf, data.ToString());
count += helper.Write(buf, TAB);
}
count += helper.Write(buf, LF);
return(count);
}
public string Read(ChannelBuffer buf, byte separator)
{
int count = buf.BytesBefore(separator);
if (count < 0)
{
return(null);
}
byte[] data = new byte[count];
buf.ReadBytes(data);
buf.ReadByte(); // get rid of separator
return(Encoding.UTF8.GetString(data));
}
public string ReadRaw(ChannelBuffer buf, byte separator)
{
int count = buf.BytesBefore(separator);
if (count < 0)
{
return(null);
}
byte[] data = new byte[count];
buf.ReadBytes(data);
buf.ReadByte(); // get rid of separator
int length = data.Length;
for (int i = 0; i < length; i++)
{
if (data[i] == '\\')
{
if (i + 1 < length)
{
byte b = data[i + 1];
if (b == 't')
{
data[i] = (byte)'\t';
}
else if (b == 'r')
{
data[i] = (byte)'\r';
}
else if (b == 'n')
{
data[i] = (byte)'\n';
}
else
{
data[i] = b;
}
Array.Copy(data, i + 2, data, i + 1, length - i - 2);
length--;
}
}
}
return(Encoding.UTF8.GetString(data, 0, length));
}
public int EncodeMessage(IMessage message, ChannelBuffer buf)
{
if (message is ITransaction)
{
var transaction = message as ITransaction;
IList <IMessage> children = transaction.Children;
if (children.Count == 0)
{
return(EncodeLine(transaction, buf, 'A', Policy.WITH_DURATION));
}
int count = 0;
int len = children.Count;
count += EncodeLine(transaction, buf, 't', Policy.WITHOUT_STATUS);
for (int i = 0; i < len; i++)
{
IMessage child = children[i];
count += EncodeMessage(child, buf);
}
count += EncodeLine(transaction, buf, 'T', Policy.WITH_DURATION);
return(count);
}
if (message is IEvent)
{
return(EncodeLine(message, buf, 'E', Policy.DEFAULT));
}
if (message is ITrace)
{
return(EncodeLine(message, buf, 'L', Policy.DEFAULT));
}
if (message is IHeartbeat)
{
return(EncodeLine(message, buf, 'H', Policy.DEFAULT));
}
if (message is IMetric)
{
return(EncodeLine(message, buf, 'M', Policy.DEFAULT));
}
throw new Exception(string.Format("Unsupported message type: {0}.", message.Type));
}
protected internal void DecodeMessage(ChannelBuffer buf, IMessageTree tree)
{
Stack <ITransaction> stack = new Stack <ITransaction>();
IMessage parent = DecodeLine(buf, null, stack, tree);
tree.Message = parent;
while (buf.ReadableBytes() > 0)
{
IMessage message = DecodeLine(buf, (ITransaction)parent, stack, tree);
if (message is ITransaction)
{
parent = message;
}
else
{
break;
}
}
}
public int WriteRaw(ChannelBuffer buf, string str)
{
if (str == null)
{
str = "null";
}
byte[] data = _mEncoding.GetBytes(str);
int len = data.Length;
int count = len;
int offset = 0;
for (int i = 0; i < len; i++)
{
byte b = data[i];
if (b == '\t' || b == '\r' || b == '\n' || b == '\\')
{
buf.WriteBytes(data, offset, i - offset);
buf.WriteByte('\\');
if (b == '\t')
{
buf.WriteByte('t');
}
else if (b == '\r')
{
buf.WriteByte('r');
}
else if (b == '\n')
{
buf.WriteByte('n');
}
else
{
buf.WriteByte(b);
}
count++;
offset = i + 1;
}
}
if (len > offset)
{
buf.WriteBytes(data, offset, len - offset);
}
return count;
}
public int Write(ChannelBuffer buf, string str)
{
if (str == null)
{
str = "null";
}
byte[] data = _mEncoding.GetBytes(str);
buf.WriteBytes(data);
return data.Length;
}
public int Write(ChannelBuffer buf, byte b)
{
buf.WriteByte(b);
return 1;
}
public string ReadRaw(ChannelBuffer buf, byte separator)
{
int count = buf.BytesBefore(separator);
if (count < 0)
{
return null;
}
byte[] data = new byte[count];
buf.ReadBytes(data);
buf.ReadByte(); // get rid of separator
int length = data.Length;
for (int i = 0; i < length; i++)
{
if (data[i] == '\\')
{
if (i + 1 < length)
{
byte b = data[i + 1];
if (b == 't')
{
data[i] = (byte)'\t';
}
else if (b == 'r')
{
data[i] = (byte)'\r';
}
else if (b == 'n')
{
data[i] = (byte)'\n';
}
else
{
data[i] = b;
}
Array.Copy(data, i + 2, data, i + 1, length - i - 2);
length--;
}
}
}
return Encoding.UTF8.GetString(data, 0, length);
}
public string Read(ChannelBuffer buf, byte separator)
{
int count = buf.BytesBefore(separator);
if (count < 0)
{
return null;
}
byte[] data = new byte[count];
buf.ReadBytes(data);
buf.ReadByte(); // get rid of separator
return Encoding.UTF8.GetString(data);
}
protected internal int EncodeLine(IMessage message, ChannelBuffer buf, char type, Policy policy)
{
BufferHelper helper = _mBufferHelper;
int count = 0;
count += helper.Write(buf, (byte)type);
if (type == 'T' && message is ITransaction)
{
long duration = ((ITransaction)message).DurationInMillis;
count += helper.Write(buf, _mDateHelper.Format(message.Timestamp + duration));
}
else
{
count += helper.Write(buf, _mDateHelper.Format(message.Timestamp));
}
count += helper.Write(buf, TAB);
count += helper.Write(buf, message.Type);
count += helper.Write(buf, TAB);
count += helper.Write(buf, message.Name);
count += helper.Write(buf, TAB);
if (policy != Policy.WITHOUT_STATUS)
{
count += helper.Write(buf, message.Status);
count += helper.Write(buf, TAB);
Object data = message.Data;
if (policy == Policy.WITH_DURATION && message is ITransaction)
{
long duration0 = ((ITransaction)message).DurationInMicros;
count += helper.Write(buf, duration0.ToString(CultureInfo.InvariantCulture));
//以微秒为单位
count += helper.Write(buf, "us");
count += helper.Write(buf, TAB);
}
count += helper.WriteRaw(buf, data.ToString());
count += helper.Write(buf, TAB);
}
count += helper.Write(buf, LF);
return count;
}
protected internal int EncodeHeader(IMessageTree tree, ChannelBuffer buf)
{
BufferHelper helper = _mBufferHelper;
int count = 0;
count += helper.Write(buf, ID);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.Domain);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.HostName);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.IpAddress);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.ThreadGroupName);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.ThreadId);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.ThreadName);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.MessageId);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.ParentMessageId);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.RootMessageId);
count += helper.Write(buf, TAB);
count += helper.Write(buf, tree.SessionToken);
count += helper.Write(buf, LF);
return count;
}
protected internal void DecodeMessage(ChannelBuffer buf, IMessageTree tree)
{
Stack<ITransaction> stack = new Stack<ITransaction>();
IMessage parent = DecodeLine(buf, null, stack, tree);
tree.Message = parent;
while (buf.ReadableBytes() > 0)
{
IMessage message = DecodeLine(buf, (ITransaction)parent, stack, tree);
if (message is ITransaction)
{
parent = message;
}
else
{
break;
}
}
}
protected internal IMessage DecodeLine(ChannelBuffer buf, ITransaction parent,
Stack <ITransaction> stack, IMessageTree tree)
{
BufferHelper helper = _mBufferHelper;
char identifier = (char)buf.ReadByte();
string timestamp = helper.Read(buf, TAB);
string type = helper.Read(buf, TAB);
string name = helper.Read(buf, TAB);
switch (identifier)
{
case 't':
IMessage transaction = new DefaultTransaction(type, name, null);
helper.Read(buf, LF); // get rid of line feed
transaction.Timestamp = _mDateHelper.Parse(timestamp);
if (parent != null)
{
parent.AddChild(transaction);
}
stack.Push(parent);
return(transaction);
case 'A':
DefaultTransaction tran = new DefaultTransaction(type, name, null);
string status = helper.Read(buf, TAB);
string duration = helper.Read(buf, TAB);
string data = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
tran.Timestamp = _mDateHelper.Parse(timestamp);
tran.Status = status;
tran.AddData(data);
long d = long.Parse(duration.Substring(0, duration.Length - 2), NumberStyles.Integer);
tran.DurationInMicros = d;
if (parent != null)
{
parent.AddChild(tran);
return(parent);
}
return(tran);
case 'T':
string transactionStatus = helper.Read(buf, TAB);
string transactionDuration = helper.Read(buf, TAB);
string transactionData = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
parent.Status = transactionStatus;
parent.AddData(transactionData);
long transactionD = long.Parse(transactionDuration.Substring(0, transactionDuration.Length - 2), NumberStyles.Integer);
parent.DurationInMicros = transactionD;
return(stack.Pop());
case 'E':
DefaultEvent evt = new DefaultEvent(type, name);
string eventStatus = helper.Read(buf, TAB);
string eventData = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
evt.Timestamp = _mDateHelper.Parse(timestamp);
evt.Status = eventStatus;
evt.AddData(eventData);
if (parent != null)
{
parent.AddChild(evt);
return(parent);
}
return(evt);
case 'M':
DefaultMetric metric = new DefaultMetric(type, name);
string metricStatus = helper.Read(buf, TAB);
string metricData = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF);
metric.Timestamp = _mDateHelper.Parse(timestamp);
metric.Status = metricStatus;
metric.AddData(metricData);
if (parent != null)
{
parent.AddChild(metric);
return(parent);
}
return(metric);
case 'L':
DefaultTrace trace = new DefaultTrace(type, name);
string traceStatus = helper.Read(buf, TAB);
string traceData = helper.Read(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
trace.Timestamp = _mDateHelper.Parse(timestamp);
trace.Status = traceStatus;
trace.AddData(traceData);
if (parent != null)
{
parent.AddChild(trace);
return(parent);
}
return(trace);
case 'H':
DefaultHeartbeat heartbeat = new DefaultHeartbeat(type, name);
string heartbeatStatus = helper.Read(buf, TAB);
string heartbeatData = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
heartbeat.Timestamp = _mDateHelper.Parse(timestamp);
heartbeat.Status = heartbeatStatus;
heartbeat.AddData(heartbeatData);
if (parent != null)
{
parent.AddChild(heartbeat);
return(parent);
}
return(heartbeat);
}
Logger.Error("Unknown identifier(" + identifier + ") of message: " + buf);
//throw new Exception("Unknown identifier int name"); //java版的抛出异常
// unknown message, ignore it
return(parent);
}
public int EncodeMessage(IMessage message, ChannelBuffer buf)
{
if (message is ITransaction)
{
var transaction = message as ITransaction;
IList<IMessage> children = transaction.Children;
if (children.Count == 0)
{
return EncodeLine(transaction, buf, 'A', Policy.WITH_DURATION);
}
int count = 0;
int len = children.Count;
count += EncodeLine(transaction, buf, 't', Policy.WITHOUT_STATUS);
for (int i = 0; i < len; i++)
{
IMessage child = children[i];
count += EncodeMessage(child, buf);
}
count += EncodeLine(transaction, buf, 'T', Policy.WITH_DURATION);
return count;
}
if (message is IEvent)
{
return EncodeLine(message, buf, 'E', Policy.DEFAULT);
}
if (message is ITrace)
{
return EncodeLine(message, buf, 'L', Policy.DEFAULT);
}
if (message is IHeartbeat)
{
return EncodeLine(message, buf, 'H', Policy.DEFAULT);
}
if (message is IMetric)
{
return EncodeLine(message, buf, 'M', Policy.DEFAULT);
}
throw new Exception(string.Format("Unsupported message type: {0}.", message.Type));
}
protected internal IMessage DecodeLine(ChannelBuffer buf, ITransaction parent,
Stack<ITransaction> stack, IMessageTree tree)
{
BufferHelper helper = _mBufferHelper;
char identifier = (char)buf.ReadByte();
string timestamp = helper.Read(buf, TAB);
string type = helper.Read(buf, TAB);
string name = helper.Read(buf, TAB);
switch (identifier)
{
case 't':
IMessage transaction = new DefaultTransaction(type, name, null);
helper.Read(buf, LF); // get rid of line feed
transaction.Timestamp = _mDateHelper.Parse(timestamp);
if (parent != null)
{
parent.AddChild(transaction);
}
stack.Push(parent);
return transaction;
case 'A':
DefaultTransaction tran = new DefaultTransaction(type, name, null);
string status = helper.Read(buf, TAB);
string duration = helper.Read(buf, TAB);
string data = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
tran.Timestamp = _mDateHelper.Parse(timestamp);
tran.Status = status;
tran.AddData(data);
long d = long.Parse(duration.Substring(0, duration.Length - 2), NumberStyles.Integer);
tran.DurationInMicros = d;
if (parent != null)
{
parent.AddChild(tran);
return parent;
}
return tran;
case 'T':
string transactionStatus = helper.Read(buf, TAB);
string transactionDuration = helper.Read(buf, TAB);
string transactionData = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
parent.Status = transactionStatus;
parent.AddData(transactionData);
long transactionD = long.Parse(transactionDuration.Substring(0, transactionDuration.Length - 2), NumberStyles.Integer);
parent.DurationInMicros = transactionD;
return stack.Pop();
case 'E':
DefaultEvent evt = new DefaultEvent(type, name);
string eventStatus = helper.Read(buf, TAB);
string eventData = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
evt.Timestamp = _mDateHelper.Parse(timestamp);
evt.Status = eventStatus;
evt.AddData(eventData);
if (parent != null)
{
parent.AddChild(evt);
return parent;
}
return evt;
case 'M':
DefaultMetric metric = new DefaultMetric(type, name);
string metricStatus = helper.Read(buf, TAB);
string metricData = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF);
metric.Timestamp = _mDateHelper.Parse(timestamp);
metric.Status = metricStatus;
metric.AddData(metricData);
if (parent != null)
{
parent.AddChild(metric);
return parent;
}
return metric;
case 'L':
DefaultTrace trace = new DefaultTrace(type, name);
string traceStatus = helper.Read(buf, TAB);
string traceData = helper.Read(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
trace.Timestamp = _mDateHelper.Parse(timestamp);
trace.Status = traceStatus;
trace.AddData(traceData);
if (parent != null)
{
parent.AddChild(trace);
return parent;
}
return trace;
case 'H':
DefaultHeartbeat heartbeat = new DefaultHeartbeat(type, name);
string heartbeatStatus = helper.Read(buf, TAB);
string heartbeatData = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
heartbeat.Timestamp = _mDateHelper.Parse(timestamp);
heartbeat.Status = heartbeatStatus;
heartbeat.AddData(heartbeatData);
if (parent != null)
{
parent.AddChild(heartbeat);
return parent;
}
return heartbeat;
}
Logger.Error("Unknown identifier(" + identifier + ") of message: " + buf);
//throw new Exception("Unknown identifier int name"); //java版的抛出异常
// unknown message, ignore it
return parent;
}
public int Write(ChannelBuffer buf, byte b)
{
buf.WriteByte(b);
return(1);
}
public override string ToString()
{
PlainTextMessageCodec codec = new PlainTextMessageCodec();
ChannelBuffer buf = new ChannelBuffer(8192);
codec.EncodeMessage(this, buf);
buf.Reset();
return buf.ToString();
}
private void SendInternal(IMessageTree tree, TcpClient activeChannel)
{
if (activeChannel != null && activeChannel.Connected)
{
var buf = new ChannelBuffer(8192);
_codec.Encode(tree, buf);
var data = buf.ToArray();
activeChannel.Client.Send(data);
if (_statistics != null)
{
_statistics.OnBytes(data.Length);
}
}
else
{
Logger.Warn("SendInternal中，Socket关闭");
}
}
