diff --git a/skywalking-network/src/main/java/com/a/eye/skywalking/network/grpc/client/SpanStorageClient.java b/skywalking-network/src/main/java/com/a/eye/skywalking/network/grpc/client/SpanStorageClient.java index 1bd135d11..2779083f0 100644 --- a/skywalking-network/src/main/java/com/a/eye/skywalking/network/grpc/client/SpanStorageClient.java +++ b/skywalking-network/src/main/java/com/a/eye/skywalking/network/grpc/client/SpanStorageClient.java @@ -27,7 +27,6 @@ public class SpanStorageClient { StreamObserver requestSpanStreamObserver = spanStorageStub.storageRequestSpan(new StreamObserver() { @Override public void onNext(SendResult sendResult) { - listener.onBatchFinished(); } @Override @@ -37,6 +36,7 @@ public class SpanStorageClient { @Override public void onCompleted() { + listener.onBatchFinished(); } }); @@ -51,7 +51,6 @@ public class SpanStorageClient { StreamObserver ackSpanStreamObserver = spanStorageStub.storageACKSpan(new StreamObserver() { @Override public void onNext(SendResult sendResult) { - listener.onBatchFinished(); } @Override @@ -61,6 +60,7 @@ public class SpanStorageClient { @Override public void onCompleted() { + listener.onBatchFinished(); } });