|
29 | 29 | import org.junit.Assert;
|
30 | 30 | import org.junit.BeforeClass;
|
31 | 31 | import org.junit.Test;
|
| 32 | +import org.slf4j.Logger; |
| 33 | +import org.slf4j.LoggerFactory; |
32 | 34 |
|
33 | 35 | import java.util.HashMap;
|
34 | 36 | import java.util.Map;
|
35 | 37 | import java.util.concurrent.CountDownLatch;
|
36 | 38 | import java.util.concurrent.TimeUnit;
|
37 | 39 |
|
38 | 40 | public class GrpcClientTest {
|
| 41 | + private static final Logger LOGGER = LoggerFactory.getLogger(GrpcClientTest.class); |
39 | 42 | private static GreeterGrpc.GreeterStub stub;
|
40 | 43 | private static GreeterGrpc.GreeterBlockingStub blockingStub;
|
41 | 44 |
|
@@ -76,22 +79,25 @@ public void serverStream() throws InterruptedException {
|
76 | 79 | "sayGreeterServerStream") {
|
77 | 80 | @Override
|
78 | 81 | public void onNext(GreeterReply data) {
|
79 |
| - long lastTimestamp = map.getOrDefault(key, 0L); |
80 | 82 | long now = System.currentTimeMillis();
|
| 83 | + long lastTimestamp = map.getOrDefault(key, 0L); |
81 | 84 | map.put(key, now);
|
| 85 | + latch.countDown(); |
82 | 86 | if (lastTimestamp == 0) {
|
83 |
| - latch.countDown(); |
| 87 | + LOGGER.info("onNext now:{}", now); |
84 | 88 | } else {
|
85 | 89 | long diff = Math.abs(now - lastTimestamp - 1000);
|
86 |
| - System.out.println(diff); |
87 | 90 | if (diff < 50) {
|
88 |
| - latch.countDown(); |
| 91 | + LOGGER.info("onNext now:{} diff:{}", now, diff); |
| 92 | + } else { |
| 93 | + LOGGER.warn("onNext now:{} diff:{} >= 50!", now, diff); |
89 | 94 | }
|
90 | 95 | }
|
91 | 96 | super.onNext(data);
|
92 | 97 | }
|
93 | 98 | };
|
94 | 99 | stub.greetServerStream(request, observer);
|
| 100 | + LOGGER.info("serverStream begin latch.await now:{}", System.currentTimeMillis()); |
95 | 101 | Assert.assertTrue(latch.await(12, TimeUnit.SECONDS));
|
96 | 102 | }
|
97 | 103 |
|
|
0 commit comments