Use POLL_FLIGHT_INFO

apache · Aug 1, 2023 · 5d71b7d · 5d71b7d
1 parent 067320e
commit 5d71b7d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/java/flight/flight-core/src/main/java/org/apache/arrow/flight/FlightMethod.java b/java/flight/flight-core/src/main/java/org/apache/arrow/flight/FlightMethod.java
@@ -32,6 +32,7 @@ public enum FlightMethod {
   DO_ACTION,
   LIST_ACTIONS,
   DO_EXCHANGE,
+  POLL_FLIGHT_INFO,
   ;
 
   /**
@@ -59,7 +60,7 @@ public static FlightMethod fromProtocol(final String methodName) {
     } else if (FlightServiceGrpc.getDoExchangeMethod().getFullMethodName().equals(methodName)) {
       return DO_EXCHANGE;
     } else if (FlightServiceGrpc.getPollFlightInfoMethod().getFullMethodName().equals(methodName)) {
-      return DO_EXCHANGE;
+      return POLL_FLIGHT_INFO;
     }
     throw new IllegalArgumentException("Not a Flight method name in gRPC: " + methodName);
   }