From 6878c6aee345aa8c19f6e51aab5ee76992c58b65 Mon Sep 17 00:00:00 2001
From: Woonduk Kang <wd.kang@navercorp.com>
Date: Fri, 2 Aug 2013 02:23:49 +0000
Subject: [PATCH] =?UTF-8?q?[=EA=B0=95=EC=9A=B4=EB=8D=95]=20[LUCYSUS-1744]?=
 =?UTF-8?q?=20commit=20=EC=95=88=EB=90=9C=EA=B2=8C=20=EC=9E=88=EC=96=B4=20?=
 =?UTF-8?q?=EC=B6=94=EA=B0=80.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

git-svn-id: http://svn.bds.nhncorp.com/pe/hippo-tomcat-profiler/trunk@2078 84d0f5b1-2673-498c-a247-62c4ff18d310
---
 .../io/SafeHeaderTBaseSerializer.java         | 114 ++++++++++++++++++
 1 file changed, 114 insertions(+)
 create mode 100644 src/main/java/com/nhn/pinpoint/profiler/io/SafeHeaderTBaseSerializer.java

diff --git a/src/main/java/com/nhn/pinpoint/profiler/io/SafeHeaderTBaseSerializer.java b/src/main/java/com/nhn/pinpoint/profiler/io/SafeHeaderTBaseSerializer.java
new file mode 100644
index 000000000..8c0f7fa02
--- /dev/null
+++ b/src/main/java/com/nhn/pinpoint/profiler/io/SafeHeaderTBaseSerializer.java
@@ -0,0 +1,114 @@
+package com.nhn.pinpoint.profiler.io;
+
+import com.nhn.pinpoint.common.dto.Header;
+import com.nhn.pinpoint.common.util.BytesUtils;
+import org.apache.thrift.TBase;
+import org.apache.thrift.TException;
+import org.apache.thrift.protocol.TCompactProtocol;
+import org.apache.thrift.protocol.TProtocol;
+import org.apache.thrift.protocol.TProtocolFactory;
+import org.apache.thrift.transport.TIOStreamTransport;
+
+import java.io.ByteArrayOutputStream;
+import java.io.UnsupportedEncodingException;
+
+/**
+ * Generic utility for easily serializing objects into a byte array or Java
+ * String.
+ */
+public class SafeHeaderTBaseSerializer {
+
+    /**
+     * This is the byte array that data is actually serialized into
+     */
+    // udp 패킷 사이즈에 최대 맞춤.
+    private final ByteArrayOutputStream baos_ = new ByteArrayOutputStream(1024 * 4);
+
+    /**
+     * This transport wraps that byte array
+     */
+    private final TIOStreamTransport transport_ = new TIOStreamTransport(baos_);
+
+    /**
+     * Internal protocol used for serializing objects.
+     */
+    private TProtocol protocol_;
+
+    /**
+     * Create a new TSerializer that uses the TBinaryProtocol by default.
+     */
+    public SafeHeaderTBaseSerializer() {
+
+        this(new TCompactProtocol.Factory(), new DefaultTBaseLocator());
+    }
+
+    private TBaseLocator locator;
+
+    /**
+     * Create a new TSerializer. It will use the TProtocol specified by the
+     * factory that is passed in.
+     *
+     * @param protocolFactory Factory to create a protocol
+     */
+    public SafeHeaderTBaseSerializer(TProtocolFactory protocolFactory, TBaseLocator locator) {
+        protocol_ = protocolFactory.getProtocol(transport_);
+        this.locator = locator;
+    }
+
+    /**
+     * Serialize the Thrift object into a byte array. The process is simple,
+     * just clear the byte array output, write the object into it, and grab the
+     * raw bytes.
+     *
+     * @param base The object to serialize
+     * @return Serialized object in byte[] format
+     */
+    public byte[] serialize(TBase<?, ?> base) throws TException {
+        final Header header = locator.headerLookup(base);
+        baos_.reset();
+        writeHeader(header);
+        base.write(protocol_);
+        return baos_.toByteArray();
+    }
+
+    public int getInterBufferSize() {
+
+        return baos_.size();
+    }
+
+    private void writeHeader(Header header) throws TException {
+        protocol_.writeByte(header.getSignature());
+        protocol_.writeByte(header.getVersion());
+        // 프로토콜 변경에 관계 없이 고정 사이즈의 데이터로 인코딩 하도록 변경.
+        short type = header.getType();
+        protocol_.writeByte(BytesUtils.writeShort1(type));
+        protocol_.writeByte(BytesUtils.writeShort2(type));
+    }
+
+    /**
+     * Serialize the Thrift object into a Java string, using a specified
+     * character set for encoding.
+     *
+     * @param base    The object to serialize
+     * @param charset Valid JVM charset
+     * @return Serialized object as a String
+     */
+    public String toString(TBase<?, ?> base, String charset) throws TException {
+        try {
+            return new String(serialize(base), charset);
+        } catch (UnsupportedEncodingException uex) {
+            throw new TException("JVM DOES NOT SUPPORT ENCODING: " + charset);
+        }
+    }
+
+    /**
+     * Serialize the Thrift object into a Java string, using the default JVM
+     * charset encoding.
+     *
+     * @param base The object to serialize
+     * @return Serialized object as a String
+     */
+    public String toString(TBase<?, ?> base) throws TException {
+        return new String(serialize(base));
+    }
+}