ADH-5241

- minor refactoring
arenadata · Nov 28, 2024 · f5b0722 · f5b0722
1 parent 9cabe0a
commit f5b0722
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/...ain/java/io/trino/plugin/adb/connector/protocol/gpfdist/load/process/GpfdistPageSink.java b/...ain/java/io/trino/plugin/adb/connector/protocol/gpfdist/load/process/GpfdistPageSink.java
@@ -56,13 +56,12 @@ public GpfdistPageSink(ContextManager<WriteContext> writeContextManager,
     @Override
     public CompletableFuture<?> appendPage(Page page)
     {
-        pageProcessingFuture = CompletableFuture.supplyAsync(() -> {
+        pageProcessingFuture = CompletableFuture.runAsync(() -> {
             waitForProcessors();
             if (writeContext.getAdbQueryException().get() == null) {
                 PageProcessor pageProcessor = writeContext.getPageProcessors().poll();
                 writeContext.getPageProcessors().offer(pageProcessor);
                 pageProcessor.process(page);
-                return null;
             }
             else {
                 throw writeContext.getAdbQueryException().get();
@@ -110,10 +109,7 @@ public CompletableFuture<Collection<Slice>> finish()
 
     private CompletableFuture<Void> closeCtx()
     {
-        return CompletableFuture.supplyAsync(() -> {
-            writeContext.close();
-            return null;
-        });
+        return CompletableFuture.runAsync(writeContext::close);
     }
 
     private Collection<Slice> createFinishResult()