Reformat FlightWorkManager

dremio · Aug 12, 2021 · 7c02250 · 7c02250
1 parent 2831b77
commit 7c02250
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 123 deletions.
diff --git a/...ices/arrow-flight/src/main/java/com/dremio/service/flight/DremioFlightLegacyProducer.java b/...ices/arrow-flight/src/main/java/com/dremio/service/flight/DremioFlightLegacyProducer.java
diff --git a/services/arrow-flight/src/main/java/com/dremio/service/flight/impl/FlightWorkManager.java b/services/arrow-flight/src/main/java/com/dremio/service/flight/impl/FlightWorkManager.java
@@ -13,6 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+
 package com.dremio.service.flight.impl;
 
 import java.nio.charset.StandardCharsets;
@@ -67,7 +68,8 @@ public FlightWorkManager(Provider<UserWorker> workerProvider,
    * @return A FlightPreparedStatement which consumes the result of the job.
    */
   public FlightPreparedStatement createPreparedStatement(FlightDescriptor flightDescriptor,
-                                                         Supplier<Boolean> isRequestCancelled, UserSession userSession) {
+                                                         Supplier<Boolean> isRequestCancelled,
+                                                         UserSession userSession) {
     final String query = getQuery(flightDescriptor);
 
     final UserProtos.CreatePreparedStatementArrowReq createPreparedStatementReq =
@@ -88,7 +90,8 @@ public FlightPreparedStatement createPreparedStatement(FlightDescriptor flightDe
     return new FlightPreparedStatement(flightDescriptor, query, createPreparedStatementResponseHandler);
   }
 
-  public void runPreparedStatement(TicketContent.PreparedStatementTicket ticket, FlightProducer.ServerStreamListener listener,
+  public void runPreparedStatement(TicketContent.PreparedStatementTicket ticket,
+                                   FlightProducer.ServerStreamListener listener,
                                    BufferAllocator allocator, UserSession userSession) {
     UserProtos.PreparedStatementHandle handle = ticket.getHandle();
     runPreparedStatement(handle, listener, allocator, userSession);
@@ -112,20 +115,20 @@ public void runPreparedStatement(UserProtos.PreparedStatementHandle handle,
     final UserResponseHandler responseHandler = runQueryResponseHandlerFactory.getHandler(runExternalId, userSession,
       workerProvider, optionManagerProvider, listener, allocator);
 
-    workerProvider.get().submitWork(runExternalId, userSession, responseHandler, userRequest, TerminationListenerRegistry.NOOP);
+    workerProvider.get()
+      .submitWork(runExternalId, userSession, responseHandler, userRequest, TerminationListenerRegistry.NOOP);
   }
 
   public void getCatalogs(FlightProducer.ServerStreamListener listener, BufferAllocator allocator,
                           UserSession userSession) {
     final UserBitShared.ExternalId runExternalId = ExternalIdHelper.generateExternalId();
     final UserRequest userRequest =
-      new UserRequest(UserProtos.RpcType.GET_CATALOGS,
-        UserProtos.GetCatalogsReq.newBuilder()
-          .build()); // TODO
+      new UserRequest(UserProtos.RpcType.GET_CATALOGS, UserProtos.GetCatalogsReq.newBuilder().build());
 
     final UserResponseHandler responseHandler = new GetCatalogsResponseHandler(allocator, listener);
 
-    workerProvider.get().submitWork(runExternalId, userSession, responseHandler, userRequest, TerminationListenerRegistry.NOOP);
+    workerProvider.get()
+      .submitWork(runExternalId, userSession, responseHandler, userRequest, TerminationListenerRegistry.NOOP);
   }
 
   @VisibleForTesting