grpc
/
grpc-node
mirror of https://github.com/grpc/grpc-node.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Ensure ordering between received messages and status

This commit is contained in:
Michael Lumish 2022-10-14 11:18:26 -07:00
parent 59a2cbceeb
commit 63d9f6a6d6
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
packages/grpc-js/src/load-balancing-call.ts
View File

@ -44,6 +44,8 @@ export class LoadBalancingCall implements Call {
private writeFilterPending = false;
private pendingMessage: {context: MessageContext, message: Buffer} | null = null;
private pendingHalfClose = false;
private readFilterPending = false;
private pendingChildStatus: StatusObject | null = null;
private ended = false;
private serviceUrl: string;
private filterStack: FilterStack;
@ -153,14 +155,23 @@ export class LoadBalancingCall implements Call {
this.listener!.onReceiveMetadata(this.filterStack.receiveMetadata(metadata));
},
onReceiveMessage: message => {
this.readFilterPending = true;
this.filterStack.receiveMessage(message).then(filteredMesssage => {
this.readFilterPending = false;
this.listener!.onReceiveMessage(filteredMesssage);
if (this.pendingChildStatus) {
this.outputStatus(this.pendingChildStatus, 'PROCESSED');
}
}, (status: StatusObject) => {
this.cancelWithStatus(status.code, status.details);
});
},
onReceiveStatus: status => {
this.outputStatus(status, 'PROCESSED');
if (this.readFilterPending) {
this.pendingChildStatus = status;
} else {
this.outputStatus(status, 'PROCESSED');
}
}
});
} catch (error) {