Skip to content

Commit c297aa1

Browse files
committed
Update grpc-tools and use package loader to read service descriptor
1 parent fca9892 commit c297aa1

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

package.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
"@types/node": "^12.12.42",
3131
"@types/request": "2.48.0",
3232
"chai": "^3.5.0",
33-
"grpc-tools": "^1.8.1",
33+
"grpc-tools": "^1.9.0",
3434
"kafka-node": "2.6.1",
3535
"mocha": "^7.1.2",
3636
"npm-registry-package-info": "1.0.5",

src/dispatchers/grpc_agent.ts

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,20 +15,21 @@
1515
*/
1616

1717
import * as grpc from '@grpc/grpc-js';
18-
import { SpanAgent } from '../proto_idl_codegen/agent/spanAgent_grpc_pb';
18+
import * as spanAgentPackageDefinition from '../proto_idl_codegen/agent/spanAgent_grpc_pb';
1919
import {Dispatcher} from './dispatcher';
2020
import Span from '../span';
2121
import { Logger, NullLogger } from '../logger';
2222
import Utils from '../utils';
23+
import {ServiceClientConstructor} from '@grpc/grpc-js/build/src/make-client';
2324

2425
export default class RemoteDispatcher implements Dispatcher {
2526
_client: any;
2627
_logger: any;
2728

2829
constructor(agentHost: string = 'haystack-agent', agentPort: number = 35000, logger: Logger = new NullLogger()) {
2930
logger.info(`Initializing the remote grpc agent dispatcher, connecting at ${agentHost}:${agentPort}`);
30-
const spanAgentClient = grpc.makeGenericClientConstructor(SpanAgent, 'SpanAgentService');
31-
this._client = new spanAgentClient(`${agentHost}:${agentPort}`, grpc.credentials.createInsecure());
31+
const protoDescriptor = grpc.loadPackageDefinition(spanAgentPackageDefinition) as {[index: string]: ServiceClientConstructor};
32+
this._client = new protoDescriptor.SpanAgent(`${agentHost}:${agentPort}`, grpc.credentials.createInsecure());
3233
this._logger = logger;
3334
}
3435

0 commit comments

Comments
 (0)