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!");
}
}
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);
}
public virtual IMessageTree Decode(ChannelBuffer buf)
{
DefaultMessageTree tree = new DefaultMessageTree();
Decode(buf, tree);
return tree;
}
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));
}
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 EncodeMessage(IMessage message, ChannelBuffer buf)
{
if (message is IEvent)
{
return(EncodeLine(message, buf, 'E', Policy.DEFAULT));
}
var transaction = message as ITransaction;
if (transaction != null)
{
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 IHeartbeat)
{
return(EncodeLine(message, buf, 'H', Policy.DEFAULT));
}
throw new Exception("Unsupported message type: " + message.Type + ".");
}
public override String ToString()
{
PlainTextMessageCodec codec = new PlainTextMessageCodec();
ChannelBuffer buf = new ChannelBuffer(8192);
codec.Encode(this, buf);
buf.Reset();
buf.Skip(4); // get rid of length
return buf.ToString();
}
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);
}
public int EncodeMessage(IMessage message, ChannelBuffer buf)
{
if (message is IEvent)
{
return EncodeLine(message, buf, 'E', Policy.DEFAULT);
}
var transaction = message as ITransaction;
if (transaction != null)
{
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 IHeartbeat)
{
return EncodeLine(message, buf, 'H', Policy.DEFAULT);
}
throw new Exception("Unsupported message type: " + message.Type + ".");
}
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;
}
public int Write(ChannelBuffer buf, byte b)
{
buf.WriteByte(b);
return(1);
}
protected internal IMessage DecodeLine(ChannelBuffer buf, ITransaction parent,
Stack<ITransaction> stack, IMessageTree tree)
{
BufferHelper helper = _mBufferHelper;
byte identifier = buf.ReadByte();
String timestamp = helper.Read(buf, TAB);
String type = helper.Read(buf, TAB);
String name = helper.Read(buf, TAB);
if (identifier == 'E')
{
IMessage evt = new DefaultEvent(type, name);
String status = helper.Read(buf, TAB);
String data = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
evt.Timestamp = _mDateHelper.Parse(timestamp);
evt.Status = status;
evt.AddData(data);
if (parent != null)
{
parent.AddChild(evt);
return parent;
}
return evt;
}
if (identifier == 'H')
{
IMessage heartbeat = new DefaultHeartbeat(type, name);
String status0 = helper.Read(buf, TAB);
String data1 = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
heartbeat.Timestamp = _mDateHelper.Parse(timestamp);
heartbeat.Status = status0;
heartbeat.AddData(data1);
if (parent != null)
{
parent.AddChild(heartbeat);
return parent;
}
return heartbeat;
}
if (identifier == '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;
}
if (identifier == 'A')
{
ITransaction transaction2 = new DefaultTransaction(type, name, null);
String status3 = helper.Read(buf, TAB);
String duration = helper.Read(buf, TAB);
String data4 = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
transaction2.Timestamp = _mDateHelper.Parse(timestamp);
transaction2.Status = status3;
transaction2.AddData(data4);
long d = Int64.Parse(duration.Substring(0, duration.Length - 2), NumberStyles.Integer);
transaction2.DurationInMicros = d;
if (parent != null)
{
parent.AddChild(transaction2);
return parent;
}
return transaction2;
}
if (identifier == 'T')
{
String status5 = helper.Read(buf, TAB);
String duration6 = helper.Read(buf, TAB);
String data7 = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
parent.Status = status5;
parent.AddData(data7);
long d8 = Int64.Parse(
duration6.Substring(0, duration6.Length - 2),
NumberStyles.Integer);
parent.DurationInMicros = d8;
return stack.Pop();
}
Logger.Error("Unknown identifier(" + identifier + ") of message: " + buf);
// unknown message, ignore it
return parent;
}
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;
}
}
}
private void SendInternal(IMessageTree tree)
{
if (_mLastChannel != null)
{
try
{
Logger.Warn("SendInternal中，_mLastChannel关闭");
_mLastChannel.Close();
}
catch
{
// ignore it
}
_mLastChannel = null;
}
if (_mActiveChannel != null && _mActiveChannel.Connected)
{
ChannelBuffer buf = new ChannelBuffer(8192);
_mCodec.Encode(tree, buf);
byte[] data = buf.ToArray();
_mActiveChannel.Client.Send(data);
if (_mStatistics != null)
{
_mStatistics.OnBytes(data.Length);
}
}
else
{
Logger.Warn("SendInternal中，Socket关闭");
}
}
protected internal IMessage DecodeLine(ChannelBuffer buf, ITransaction parent,
Stack <ITransaction> stack, IMessageTree tree)
{
BufferHelper helper = _mBufferHelper;
byte identifier = buf.ReadByte();
String timestamp = helper.Read(buf, TAB);
String type = helper.Read(buf, TAB);
String name = helper.Read(buf, TAB);
if (identifier == 'E')
{
IMessage evt = new DefaultEvent(type, name);
String status = helper.Read(buf, TAB);
String data = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
evt.Timestamp = _mDateHelper.Parse(timestamp);
evt.Status = status;
evt.AddData(data);
if (parent != null)
{
parent.AddChild(evt);
return(parent);
}
return(evt);
}
if (identifier == 'H')
{
IMessage heartbeat = new DefaultHeartbeat(type, name);
String status0 = helper.Read(buf, TAB);
String data1 = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
heartbeat.Timestamp = _mDateHelper.Parse(timestamp);
heartbeat.Status = status0;
heartbeat.AddData(data1);
if (parent != null)
{
parent.AddChild(heartbeat);
return(parent);
}
return(heartbeat);
}
if (identifier == '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);
}
if (identifier == 'A')
{
ITransaction transaction2 = new DefaultTransaction(type, name, null);
String status3 = helper.Read(buf, TAB);
String duration = helper.Read(buf, TAB);
String data4 = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
transaction2.Timestamp = _mDateHelper.Parse(timestamp);
transaction2.Status = status3;
transaction2.AddData(data4);
long d = Int64.Parse(duration.Substring(0, duration.Length - 2), NumberStyles.Integer);
transaction2.DurationInMicros = d;
if (parent != null)
{
parent.AddChild(transaction2);
return(parent);
}
return(transaction2);
}
if (identifier == 'T')
{
String status5 = helper.Read(buf, TAB);
String duration6 = helper.Read(buf, TAB);
String data7 = helper.ReadRaw(buf, TAB);
helper.Read(buf, LF); // get rid of line feed
parent.Status = status5;
parent.AddData(data7);
long d8 = Int64.Parse(
duration6.Substring(0, duration6.Length - 2),
NumberStyles.Integer);
parent.DurationInMicros = d8;
return(stack.Pop());
}
Logger.Error("Unknown identifier(" + identifier + ") of message: " + buf);
// unknown message, ignore it
return(parent);
}
