afonso/core-extra
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

corefx properly cancel server stream from client for session events

Browse source
This commit is contained in:
Blake J. Harnden 2019-05-31 16:53:16 -07:00
parent b075181796
commit 43e18d820f
1 changed files with 34 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
corefx/src/main/java/com/core/client/grpc/CoreGrpcClient.java
View file

@ -1129,6 +1129,8 @@ public class CoreGrpcClient implements ICoreClient {
Iterator<CoreProto.Event> events = blockingStub.events(request); Iterator<CoreProto.Event> events = blockingStub.events(request);
executorService.submit(() -> { executorService.submit(() -> {
Context.CancellableContext context = Context.current().withCancellation();
context.run(() -> {
try { try {
while (handlingEvents) { while (handlingEvents) {
CoreProto.Event event = events.next(); CoreProto.Event event = events.next();
@ -1158,8 +1160,12 @@ public class CoreGrpcClient implements ICoreClient {
} }
} catch (StatusRuntimeException ex) { } catch (StatusRuntimeException ex) {
logger.error("error handling session events", ex); logger.error("error handling session events", ex);
} finally {
context.cancel(null);
context.close();
} }
}); });
});
} catch (StatusRuntimeException ex) { } catch (StatusRuntimeException ex) {
throw new IOException("setup event handlers error", ex); throw new IOException("setup event handlers error", ex);
} }