在使用Arrow Flight传输大型数据集时，如何确定“ listener.completed”的时间安排？

如何解决在使用Arrow Flight传输大型数据集时，如何确定“ listener.completed”的时间安排？

我正在尝试使用Arrow Flight传输大型数据集，但由于“ listener.completed”的时间而感到困惑。

当数据集很大时，如果FlightClient立即在最后一个“ putNext”之后执行“ listener.completed”，FlightProducer将获得如下所示的异常：

CallStatus{code=CANCELLED,cause=null,description='cancelled before receiving half close',Metadata='org.apache.arrow.flight.ErrorFlightMetadata@57b6c7f4'}

我搜索了异常原因，这是因为“ listener.completed”函数在服务器接收到数据集之前关闭了流。这是搜索结果：https://github.com/grpc/grpc-java/issues/6560。

但是函数“ putNext”不会返回有关传输过程的任何信号。使用Arrow Flight传输大型数据集时如何确定“ listener.completed”的时间？

详细代码如下所示

 FlightClient.ClientStreamListener listener = client.startPut(FlightDescriptor.path("map-shuffle",dataDesc.toString()),root,new FlightClient.PutListener() {
       @Override
       public void getResult() {
           System.out.println("fail");
       }

       @Override
       public void onNext(PutResult putResult) {
           System.out.println("fail");

       }

       @Override
       public void onError(Throwable throwable) {
           System.out.println("fail");

       }

       @Override
       public void onCompleted() {
           System.out.println("success");

       }
   });
   listener.putNext();
   try {
       Thread.sleep(5000);
   } catch (InterruptedException e) {
       e.printstacktrace();
   }
   listener.completed();