Make bytes available

Author: shangm2

drift-codec-utils/src/main/java/com/facebook/drift/codec/utils/UuidToLeachSalzBinaryEncodingThriftCodec.java

@@ -15,6 +15,7 @@
  */
 package com.facebook.drift.codec.utils;
 
+import com.facebook.drift.codec.CodecThriftType;
 import com.facebook.drift.codec.ThriftCodec;
 import com.facebook.drift.codec.internal.coercion.FromThrift;
 import com.facebook.drift.codec.internal.coercion.ToThrift;
@@ -33,24 +34,32 @@
 
 /**
  * This codec will encode UUIDs into the format specified in RFC 4122:
- *
+ * <p>
  * https://www.ietf.org/rfc/rfc4122.txt
- *
+ * <p>
  * Likewise, it presumes the input is of this format.
  */
 public class UuidToLeachSalzBinaryEncodingThriftCodec
         implements ThriftCodec<UUID>
 {
+    private static final ThriftType THRIFT_TYPE = new ThriftType(ThriftType.BINARY, UUID.class);
+
     @Inject
     public UuidToLeachSalzBinaryEncodingThriftCodec(ThriftCatalog thriftCatalog)
     {
         thriftCatalog.addDefaultCoercions(getClass());
     }
 
+    @CodecThriftType
+    public static ThriftType getThriftType()
+    {
+        return THRIFT_TYPE;
+    }
+
     @Override
     public ThriftType getType()
     {
-        return new ThriftType(ThriftType.BINARY, UUID.class);
+        return THRIFT_TYPE;
     }
 
     @Override

drift-idl-generator/src/main/java/com/facebook/drift/idl/generator/ThriftIdlGenerator.java

@@ -152,7 +152,7 @@ private void loadCustomCodec(String customCodecClassName)
 
         // Look for a static method annotated with @CodecThriftType
         ThriftType thriftType = null;
-        for (Method method : codecClass.getDeclaredMethods()) {
+        for (Method method : codecClass.getMethods()) {
             if (method.isAnnotationPresent(CodecThriftType.class)) {
                 if (!Modifier.isPublic(method.getModifiers())) {
                     throw new ThriftIdlGeneratorException("Method annotated with @CodecThriftType must be public: " + customCodecClassName + "#" + method.getName());
@@ -163,7 +163,7 @@ private void loadCustomCodec(String customCodecClassName)
                 }
 
                 if (method.getParameterCount() != 0) {
-                    throw new ThriftIdlGeneratorException("Method annotated with @CodecThriftType must have no parameters: " + customCodecClassName + "#" + method.getName());
+                    throw new ThriftIdlGeneratorException("Method annotated with @CodecThriftType must have no parameter: " + customCodecClassName + "#" + method.getName());
                 }
 
                 if (!ThriftType.class.isAssignableFrom(method.getReturnType())) {

drift-protocol/src/main/java/com/facebook/drift/protocol/TByteArrayOutputStream.java

@@ -0,0 +1,48 @@
+/*
+ * Copyright (C) 2017 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 com.facebook.drift.protocol;
+
+import java.io.ByteArrayOutputStream;
+import java.nio.charset.Charset;
+
+public class TByteArrayOutputStream
+        extends ByteArrayOutputStream
+{
+    public TByteArrayOutputStream(int size)
+    {
+        super(size);
+    }
+
+    public TByteArrayOutputStream()
+    {
+        super();
+    }
+
+    protected byte[] get()
+    {
+        return buf;
+    }
+
+    public int len()
+    {
+        return count;
+    }
+
+    public String toString(Charset charset)
+    {
+        return new String(buf, 0, count, charset);
+    }
+}

drift-protocol/src/main/java/com/facebook/drift/protocol/TMemoryBufferWriteOnly.java

@@ -0,0 +1,49 @@
+/*
+ * Copyright (C) 2017 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 com.facebook.drift.protocol;
+
+import com.google.common.io.ByteArrayDataOutput;
+import com.google.common.io.ByteStreams;
+
+public class TMemoryBufferWriteOnly
+        implements TTransport
+{
+    private ByteArrayDataOutput data;
+
+    public TMemoryBufferWriteOnly(int size)
+    {
+        data = ByteStreams.newDataOutput(size);
+    }
+
+    @Override
+    public void read(byte[] buf, int off, int len)
+            throws TTransportException
+    {
+        throw new UnsupportedOperationException("Read operation is not supported");
+    }
+
+    @Override
+    public void write(byte[] buf, int off, int len)
+            throws TTransportException
+    {
+        data.write(buf, off, len);
+    }
+
+    public byte[] getBytes()
+    {
+        return data.toByteArray();
+    }
+}