/*
 * Copyright (C) 2012 Facebook, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may
 * not use this file except in compliance with the License. You may obtain
 * a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations
 * under the License.
 */
package io.airlift.drift.codec;

public enum ThriftProtocolType
{
    UNKNOWN((byte) 0),
    BOOL((byte) 2),
    BYTE((byte) 3),
    DOUBLE((byte) 4),
    I16((byte) 6),
    I32((byte) 8),
    I64((byte) 10),
    STRING((byte) 11),
    STRUCT((byte) 12),
    MAP((byte) 13),
    SET((byte) 14),
    LIST((byte) 15),
    ENUM((byte) 8), // same as I32 type
    BINARY((byte) 11); // same as STRING type

    private final byte type;

    ThriftProtocolType(byte type)
    {
        this.type = type;
    }

    public byte getType()
    {
        return type;
    }
}

@@ -628,16 +628,58 @@
 codec;%0A%0A
+import org.apache.thrift.protocol.TType;%0A%0A
 public e
@@ -719,16 +719,18 @@
 OWN(
-(byte) 0
+TType.STOP
 ),%0A 
@@ -741,16 +741,18 @@
 OOL(
-(byte) 2
+TType.BOOL
 ),%0A 
@@ -763,16 +763,18 @@
 YTE(
-(byte) 3
+TType.BYTE
 ),%0A 
@@ -787,16 +787,20 @@
 BLE(
-(byte) 4
+TType.DOUBLE
 ),%0A 
@@ -806,23 +806,24 @@
     I16(
-(byte) 
+TType.I1
 6),%0A    
@@ -826,24 +826,25 @@
     I32(
-(byte) 8
+TType.I32
 ),%0A    I
@@ -850,17 +850,17 @@
 I64(
-(byte) 10
+TType.I64
 ),%0A 
@@ -869,25 +869,28 @@
  STRING(
-(byte) 11
+TType.STRING
 ),%0A    S
@@ -899,17 +899,20 @@
 UCT(
-(byte) 12
+TType.STRUCT
 ),%0A 
@@ -922,17 +922,17 @@
 MAP(
-(byte) 13
+TType.MAP
 ),%0A 
@@ -942,17 +942,17 @@
 SET(
-(byte) 14
+TType.SET
 ),%0A 
@@ -963,17 +963,18 @@
 IST(
-(byte) 15
+TType.LIST
 ),%0A 
@@ -985,84 +985,45 @@
 NUM(
-(byte) 8), // same as I32 type%0A    BINARY((byte) 11); // same as 
+TType.I32),%0A    BINARY(TType.
 STRING
- type
+);
 %0A%0A  

Use TType constants in ThriftProtocolType