From 03fc64eef2019ab83cec57bad9274824c8e33615 Mon Sep 17 00:00:00 2001
From: wusheng <wu.sheng@foxmail.com>
Date: Mon, 28 Nov 2016 16:15:25 +0800
Subject: [PATCH] =?UTF-8?q?move=20=E2=80=98onBatchFinished=E2=80=99=20noti?=
 =?UTF-8?q?ce=20point.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../eye/skywalking/network/grpc/client/SpanStorageClient.java | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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 1bd135d119..2779083f09 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<RequestSpan> requestSpanStreamObserver = spanStorageStub.storageRequestSpan(new StreamObserver<SendResult>() {
             @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<AckSpan> ackSpanStreamObserver = spanStorageStub.storageACKSpan(new StreamObserver<SendResult>() {
             @Override
             public void onNext(SendResult sendResult) {
-                listener.onBatchFinished();
             }
 
             @Override
@@ -61,6 +60,7 @@ public class SpanStorageClient {
 
             @Override
             public void onCompleted() {
+                listener.onBatchFinished();
             }
         });
 
-- 
GitLab