mirror of https://github.com/grpc/grpc-node.git
chore: move new functions towards the end of the class
This commit is contained in:
AVVS
parent
e0b900dd69
commit
a4a676d378
|
|
@ -1191,6 +1191,187 @@ export class Server {
|
||||||
channelzSessionInfo?.streamTracker.addCallFailed();
|
channelzSessionInfo?.streamTracker.addCallFailed();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private _channelzHandler(http2Server: http2.Http2Server) {
|
||||||
|
return (
|
||||||
|
stream: http2.ServerHttp2Stream,
|
||||||
|
headers: http2.IncomingHttpHeaders
|
||||||
|
) => {
|
||||||
|
// for handling idle timeout
|
||||||
|
this.onStreamOpened(stream);
|
||||||
|
|
||||||
|
const channelzSessionInfo = this.sessions.get(
|
||||||
|
stream.session as http2.ServerHttp2Session
|
||||||
|
);
|
||||||
|
|
||||||
|
this.callTracker.addCallStarted();
|
||||||
|
channelzSessionInfo?.streamTracker.addCallStarted();
|
||||||
|
|
||||||
|
if (!this._verifyContentType(stream, headers)) {
|
||||||
|
this.callTracker.addCallFailed();
|
||||||
|
channelzSessionInfo?.streamTracker.addCallFailed();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const path = headers[HTTP2_HEADER_PATH] as string;
|
||||||
|
|
||||||
|
const handler = this._retrieveHandler(path);
|
||||||
|
if (!handler) {
|
||||||
|
this._respondWithError(
|
||||||
|
getUnimplementedStatusResponse(path),
|
||||||
|
stream,
|
||||||
|
channelzSessionInfo
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const callEventTracker: CallEventTracker = {
|
||||||
|
addMessageSent: () => {
|
||||||
|
if (channelzSessionInfo) {
|
||||||
|
channelzSessionInfo.messagesSent += 1;
|
||||||
|
channelzSessionInfo.lastMessageSentTimestamp = new Date();
|
||||||
|
}
|
||||||
|
},
|
||||||
|
addMessageReceived: () => {
|
||||||
|
if (channelzSessionInfo) {
|
||||||
|
channelzSessionInfo.messagesReceived += 1;
|
||||||
|
channelzSessionInfo.lastMessageReceivedTimestamp = new Date();
|
||||||
|
}
|
||||||
|
},
|
||||||
|
onCallEnd: status => {
|
||||||
|
if (status.code === Status.OK) {
|
||||||
|
this.callTracker.addCallSucceeded();
|
||||||
|
} else {
|
||||||
|
this.callTracker.addCallFailed();
|
||||||
|
}
|
||||||
|
},
|
||||||
|
onStreamEnd: success => {
|
||||||
|
if (channelzSessionInfo) {
|
||||||
|
if (success) {
|
||||||
|
channelzSessionInfo.streamTracker.addCallSucceeded();
|
||||||
|
} else {
|
||||||
|
channelzSessionInfo.streamTracker.addCallFailed();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
const call = getServerInterceptingCall(
|
||||||
|
this.interceptors,
|
||||||
|
stream,
|
||||||
|
headers,
|
||||||
|
callEventTracker,
|
||||||
|
handler,
|
||||||
|
this.options
|
||||||
|
);
|
||||||
|
|
||||||
|
if (!this._runHandlerForCall(call, handler)) {
|
||||||
|
this.callTracker.addCallFailed();
|
||||||
|
channelzSessionInfo?.streamTracker.addCallFailed();
|
||||||
|
|
||||||
|
call.sendStatus({
|
||||||
|
code: Status.INTERNAL,
|
||||||
|
details: `Unknown handler type: ${handler.type}`,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
private _streamHandler(http2Server: http2.Http2Server) {
|
||||||
|
return (
|
||||||
|
stream: http2.ServerHttp2Stream,
|
||||||
|
headers: http2.IncomingHttpHeaders
|
||||||
|
) => {
|
||||||
|
// for handling idle timeout
|
||||||
|
this.onStreamOpened(stream);
|
||||||
|
|
||||||
|
if (this._verifyContentType(stream, headers) !== true) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const path = headers[HTTP2_HEADER_PATH] as string;
|
||||||
|
|
||||||
|
const handler = this._retrieveHandler(path);
|
||||||
|
if (!handler) {
|
||||||
|
this._respondWithError(
|
||||||
|
getUnimplementedStatusResponse(path),
|
||||||
|
stream,
|
||||||
|
null
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const call = getServerInterceptingCall(
|
||||||
|
this.interceptors,
|
||||||
|
stream,
|
||||||
|
headers,
|
||||||
|
null,
|
||||||
|
handler,
|
||||||
|
this.options
|
||||||
|
);
|
||||||
|
|
||||||
|
if (!this._runHandlerForCall(call, handler)) {
|
||||||
|
call.sendStatus({
|
||||||
|
code: Status.INTERNAL,
|
||||||
|
details: `Unknown handler type: ${handler.type}`,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
private _runHandlerForCall(
|
||||||
|
call: ServerInterceptingCallInterface,
|
||||||
|
handler:
|
||||||
|
| UntypedUnaryHandler
|
||||||
|
| UntypedClientStreamingHandler
|
||||||
|
| UntypedServerStreamingHandler
|
||||||
|
| UntypedBidiStreamingHandler
|
||||||
|
): boolean {
|
||||||
|
const { type } = handler;
|
||||||
|
if (type === 'unary') {
|
||||||
|
handleUnary(call, handler);
|
||||||
|
} else if (type === 'clientStream') {
|
||||||
|
handleClientStreaming(call, handler);
|
||||||
|
} else if (type === 'serverStream') {
|
||||||
|
handleServerStreaming(call, handler);
|
||||||
|
} else if (type === 'bidi') {
|
||||||
|
handleBidiStreaming(call, handler);
|
||||||
|
} else {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
private _setupHandlers(
|
||||||
|
http2Server: http2.Http2Server | http2.Http2SecureServer
|
||||||
|
): void {
|
||||||
|
if (http2Server === null) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const serverAddress = http2Server.address();
|
||||||
|
let serverAddressString = 'null';
|
||||||
|
if (serverAddress) {
|
||||||
|
if (typeof serverAddress === 'string') {
|
||||||
|
serverAddressString = serverAddress;
|
||||||
|
} else {
|
||||||
|
serverAddressString = serverAddress.address + ':' + serverAddress.port;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
this.serverAddressString = serverAddressString;
|
||||||
|
|
||||||
|
const handler = this.channelzEnabled
|
||||||
|
? this._channelzHandler(http2Server)
|
||||||
|
: this._streamHandler(http2Server);
|
||||||
|
|
||||||
|
const sessionHandler = this.channelzEnabled
|
||||||
|
? this._channelzSessionHandler(http2Server)
|
||||||
|
: this._sessionHandler(http2Server);
|
||||||
|
|
||||||
|
http2Server.on('stream', handler);
|
||||||
|
http2Server.on('session', sessionHandler);
|
||||||
|
}
|
||||||
|
|
||||||
private _sessionHandler(
|
private _sessionHandler(
|
||||||
http2Server: http2.Http2Server | http2.Http2SecureServer
|
http2Server: http2.Http2Server | http2.Http2SecureServer
|
||||||
) {
|
) {
|
||||||
|
|
@ -1432,187 +1613,6 @@ export class Server {
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
private _channelzHandler(http2Server: http2.Http2Server) {
|
|
||||||
return (
|
|
||||||
stream: http2.ServerHttp2Stream,
|
|
||||||
headers: http2.IncomingHttpHeaders
|
|
||||||
) => {
|
|
||||||
// for handling idle timeout
|
|
||||||
this.onStreamOpened(stream);
|
|
||||||
|
|
||||||
const channelzSessionInfo = this.sessions.get(
|
|
||||||
stream.session as http2.ServerHttp2Session
|
|
||||||
);
|
|
||||||
|
|
||||||
this.callTracker.addCallStarted();
|
|
||||||
channelzSessionInfo?.streamTracker.addCallStarted();
|
|
||||||
|
|
||||||
if (!this._verifyContentType(stream, headers)) {
|
|
||||||
this.callTracker.addCallFailed();
|
|
||||||
channelzSessionInfo?.streamTracker.addCallFailed();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const path = headers[HTTP2_HEADER_PATH] as string;
|
|
||||||
|
|
||||||
const handler = this._retrieveHandler(path);
|
|
||||||
if (!handler) {
|
|
||||||
this._respondWithError(
|
|
||||||
getUnimplementedStatusResponse(path),
|
|
||||||
stream,
|
|
||||||
channelzSessionInfo
|
|
||||||
);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const callEventTracker: CallEventTracker = {
|
|
||||||
addMessageSent: () => {
|
|
||||||
if (channelzSessionInfo) {
|
|
||||||
channelzSessionInfo.messagesSent += 1;
|
|
||||||
channelzSessionInfo.lastMessageSentTimestamp = new Date();
|
|
||||||
}
|
|
||||||
},
|
|
||||||
addMessageReceived: () => {
|
|
||||||
if (channelzSessionInfo) {
|
|
||||||
channelzSessionInfo.messagesReceived += 1;
|
|
||||||
channelzSessionInfo.lastMessageReceivedTimestamp = new Date();
|
|
||||||
}
|
|
||||||
},
|
|
||||||
onCallEnd: status => {
|
|
||||||
if (status.code === Status.OK) {
|
|
||||||
this.callTracker.addCallSucceeded();
|
|
||||||
} else {
|
|
||||||
this.callTracker.addCallFailed();
|
|
||||||
}
|
|
||||||
},
|
|
||||||
onStreamEnd: success => {
|
|
||||||
if (channelzSessionInfo) {
|
|
||||||
if (success) {
|
|
||||||
channelzSessionInfo.streamTracker.addCallSucceeded();
|
|
||||||
} else {
|
|
||||||
channelzSessionInfo.streamTracker.addCallFailed();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
const call = getServerInterceptingCall(
|
|
||||||
this.interceptors,
|
|
||||||
stream,
|
|
||||||
headers,
|
|
||||||
callEventTracker,
|
|
||||||
handler,
|
|
||||||
this.options
|
|
||||||
);
|
|
||||||
|
|
||||||
if (!this._runHandlerForCall(call, handler)) {
|
|
||||||
this.callTracker.addCallFailed();
|
|
||||||
channelzSessionInfo?.streamTracker.addCallFailed();
|
|
||||||
|
|
||||||
call.sendStatus({
|
|
||||||
code: Status.INTERNAL,
|
|
||||||
details: `Unknown handler type: ${handler.type}`,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
private _streamHandler(http2Server: http2.Http2Server) {
|
|
||||||
return (
|
|
||||||
stream: http2.ServerHttp2Stream,
|
|
||||||
headers: http2.IncomingHttpHeaders
|
|
||||||
) => {
|
|
||||||
// for handling idle timeout
|
|
||||||
this.onStreamOpened(stream);
|
|
||||||
|
|
||||||
if (this._verifyContentType(stream, headers) !== true) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const path = headers[HTTP2_HEADER_PATH] as string;
|
|
||||||
|
|
||||||
const handler = this._retrieveHandler(path);
|
|
||||||
if (!handler) {
|
|
||||||
this._respondWithError(
|
|
||||||
getUnimplementedStatusResponse(path),
|
|
||||||
stream,
|
|
||||||
null
|
|
||||||
);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const call = getServerInterceptingCall(
|
|
||||||
this.interceptors,
|
|
||||||
stream,
|
|
||||||
headers,
|
|
||||||
null,
|
|
||||||
handler,
|
|
||||||
this.options
|
|
||||||
);
|
|
||||||
|
|
||||||
if (!this._runHandlerForCall(call, handler)) {
|
|
||||||
call.sendStatus({
|
|
||||||
code: Status.INTERNAL,
|
|
||||||
details: `Unknown handler type: ${handler.type}`,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
private _runHandlerForCall(
|
|
||||||
call: ServerInterceptingCallInterface,
|
|
||||||
handler:
|
|
||||||
| UntypedUnaryHandler
|
|
||||||
| UntypedClientStreamingHandler
|
|
||||||
| UntypedServerStreamingHandler
|
|
||||||
| UntypedBidiStreamingHandler
|
|
||||||
): boolean {
|
|
||||||
const { type } = handler;
|
|
||||||
if (type === 'unary') {
|
|
||||||
handleUnary(call, handler);
|
|
||||||
} else if (type === 'clientStream') {
|
|
||||||
handleClientStreaming(call, handler);
|
|
||||||
} else if (type === 'serverStream') {
|
|
||||||
handleServerStreaming(call, handler);
|
|
||||||
} else if (type === 'bidi') {
|
|
||||||
handleBidiStreaming(call, handler);
|
|
||||||
} else {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
private _setupHandlers(
|
|
||||||
http2Server: http2.Http2Server | http2.Http2SecureServer
|
|
||||||
): void {
|
|
||||||
if (http2Server === null) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const serverAddress = http2Server.address();
|
|
||||||
let serverAddressString = 'null';
|
|
||||||
if (serverAddress) {
|
|
||||||
if (typeof serverAddress === 'string') {
|
|
||||||
serverAddressString = serverAddress;
|
|
||||||
} else {
|
|
||||||
serverAddressString = serverAddress.address + ':' + serverAddress.port;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
this.serverAddressString = serverAddressString;
|
|
||||||
|
|
||||||
const handler = this.channelzEnabled
|
|
||||||
? this._channelzHandler(http2Server)
|
|
||||||
: this._streamHandler(http2Server);
|
|
||||||
|
|
||||||
const sessionHandler = this.channelzEnabled
|
|
||||||
? this._channelzSessionHandler(http2Server)
|
|
||||||
: this._sessionHandler(http2Server);
|
|
||||||
|
|
||||||
http2Server.on('stream', handler);
|
|
||||||
http2Server.on('session', sessionHandler);
|
|
||||||
}
|
|
||||||
|
|
||||||
private enableIdleTimeout(session: http2.ServerHttp2Session) {
|
private enableIdleTimeout(session: http2.ServerHttp2Session) {
|
||||||
const idleTimeoutObj = {
|
const idleTimeoutObj = {
|
||||||
activeStreams: 0,
|
activeStreams: 0,
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue