我正在制作输入流速率计。它基本上是一个公开请求流调用的服务,并计算它可以处理的每秒消息数。
由于客户端在发送消息时完全异步,我使用ClientCallStreamObserver在流准备就绪时开始发送消息,以避免内存溢出。
客户端代码如下所示:
public static void main(String[] args) throws Exception {
ManagedChannel channel = ManagedChannelBuilder.forAddress("server", 4242).usePlaintext(true).build();
ServerGrpc.ServerStub asyncStub = ServerGrpc.newStub(channel);
StreamObserver<MarketDataOuterClass.Trade> inputStream = asyncStub.reportNewTradeStream(new StreamObserver<Empty>() {
@Override
public void onNext(Empty empty) {
}
@Override
public void onError(Throwable throwable) {
logger.info("on error response stream");
}
@Override
public void onCompleted() {
logger.info("on completed response stream");
}
});
final ClientCallStreamObserver<MarketDataOuterClass.Trade> clientCallObserver = (ClientCallStreamObserver<MarketDataOuterClass.Trade>) inputStream;
while (!clientCallObserver.isReady()) {
Thread.sleep(2000);
logger.info("stream not ready yet");
}
counter.setLastTic(System.nanoTime());
while (true) {
counter.inc();
if (counter.getCounter() % 15000 == 0 ) {
long now = System.nanoTime();
double rate = (double) NANOSEC_TO_SEC * counter.getCounter() / (now - counter.getLastTic());
logger.info("rate: " + rate + " msgs per sec");
counter.clear();
counter.setLastTic(now);
}
inputStream.onNext(createRandomTrade());
}
}
我的观察循环isReady永远不会结束。
OBS:我正在使用kubernetes集群来为我的测试服务,服务器正在接收调用并返回StreamObserver实现。
答案 0 :(得分:1)
isReady
最终应该返回true
。但是代码没有正确地观察流量控制。
每次致电onNext()
发送请求isReady()
后,都可以开始返回false
。您的while (true)
循环应该在每次迭代开始时进行isReady()
检查。
最好是在呼叫准备好发送时通知serverCallObserver.setOnReadyHandler(yourRunnable)
,而不是轮询。请注意,您仍应检查isReady()
中的yourRunnable
,因为可能存在虚假/过期通知。