amirho/nest
1
0
Fork 0
mirror of https://github.com/nestjs/nest.git synced 2026-02-21 23:11:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(microservices): keepalive option in grpc clinet

Browse Source 
https://github.com/grpc/grpc/blob/master/doc/keepalive.md
This commit is contained in:
CatsMiaow
2020-02-24 14:27:24 +09:00
parent 581d1aedc6
commit 51a968c779
2 changed files with 34 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
packages/microservices/client/client-grpc.ts
View File

@@ -78,10 +78,34 @@ export class ClientGrpcProxy extends ClientProxy implements ClientGrpc {
GRPC_DEFAULT_MAX_RECEIVE_MESSAGE_LENGTH,
),
};
const keepaliveOptions = {};
if (isObject(this.options.keepalive)) {
const keepaliveKeys = {
keepaliveTimeMs: 'grpc.keepalive_time_ms',
keepaliveTimeoutMs: 'grpc.keepalive_timeout_ms',
keepalivePermitWithoutCalls: 'grpc.keepalive_permit_without_calls',
http2MaxPingsWithoutData: 'grpc.http2.max_pings_without_data',
http2MinTimeBetweenPingsMs: 'grpc.http2.min_time_between_pings_ms',
http2MinPingIntervalWithoutDataMs:
'grpc.http2.min_ping_interval_without_data_ms',
http2MaxPingStrikes: 'grpc.http2.max_ping_strikes',
};
for (const [optionKey, optionValue] of Object.entries(
this.options.keepalive,
)) {
const key = keepaliveKeys[optionKey];
if (key) {
keepaliveOptions[key] = optionValue;
}
}
}
const options: any = isObject(this.options)
? {
...this.options,
...maxMessageLengthOptions,
...keepaliveOptions,
loader: '',
}
: {
@@ -92,6 +116,7 @@ export class ClientGrpcProxy extends ClientProxy implements ClientGrpc {
options.credentials || grpcPackage.credentials.createInsecure();
delete options.credentials;
delete options.keepalive;
const grpcClient = new clientRef[name](this.url, credentials, options);
this.clients.set(name, grpcClient);
return grpcClient;