Add Drift serialization compatibility for NodeStatus

prestodb · Apr 19, 2021 · b1306fc · b1306fc
1 parent 3ab2e8e
commit b1306fc
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/presto-main/src/main/java/com/facebook/presto/metadata/InternalNode.java b/presto-main/src/main/java/com/facebook/presto/metadata/InternalNode.java
@@ -14,6 +14,8 @@
 package com.facebook.presto.metadata;
 
 import com.facebook.drift.annotations.ThriftConstructor;
+import com.facebook.drift.annotations.ThriftEnum;
+import com.facebook.drift.annotations.ThriftEnumValue;
 import com.facebook.drift.annotations.ThriftField;
 import com.facebook.drift.annotations.ThriftStruct;
 import com.facebook.presto.client.NodeVersion;
@@ -36,10 +38,25 @@
 public class InternalNode
         implements Node
 {
+    @ThriftEnum
     public enum NodeStatus
     {
-        ALIVE,
-        DEAD
+        ALIVE(1),
+        DEAD(2),
+        /**/;
+
+        private final int statusCode;
+
+        NodeStatus(int statusCode)
+        {
+            this.statusCode = statusCode;
+        }
+
+        @ThriftEnumValue
+        public int getStatusCode()
+        {
+            return statusCode;
+        }
     }
 
     private final String nodeIdentifier;