Make OutputBuffers Thrift-compatible (#24701)

Author: shangm2

presto-main/src/main/java/com/facebook/presto/execution/buffer/OutputBuffers.java

@@ -14,6 +14,7 @@
 package com.facebook.presto.execution.buffer;
 
 import com.facebook.drift.annotations.ThriftConstructor;
+import com.facebook.drift.annotations.ThriftEnum;
 import com.facebook.drift.annotations.ThriftField;
 import com.facebook.drift.annotations.ThriftStruct;
 import com.facebook.presto.spi.plan.PartitioningHandle;
@@ -40,6 +41,7 @@
 import static java.lang.Integer.parseInt;
 import static java.util.Objects.requireNonNull;
 
+@ThriftStruct
 public final class OutputBuffers
 {
     public static final int BROADCAST_PARTITION_ID = 0;
@@ -76,6 +78,7 @@ public static OutputBuffers createSpoolingOutputBuffers()
         return SPOOLING_OUTPUT_BUFFERS;
     }
 
+    @ThriftEnum
     public enum BufferType
     {
         PARTITIONED,
@@ -91,6 +94,7 @@ public enum BufferType
     private final Map<OutputBufferId, Integer> buffers;
 
     // Visible only for Jackson... Use the "with" methods instead
+    @ThriftConstructor
     @JsonCreator
     public OutputBuffers(
             @JsonProperty("type") BufferType type,
@@ -104,24 +108,28 @@ public OutputBuffers(
         this.noMoreBufferIds = noMoreBufferIds;
     }
 
+    @ThriftField(1)
     @JsonProperty
     public BufferType getType()
     {
         return type;
     }
 
+    @ThriftField(2)
     @JsonProperty
     public long getVersion()
     {
         return version;
     }
 
+    @ThriftField(3)
     @JsonProperty
     public boolean isNoMoreBufferIds()
     {
         return noMoreBufferIds;
     }
 
+    @ThriftField(4)
     @JsonProperty
     public Map<OutputBufferId, Integer> getBuffers()
     {