|
22 | 22 | */ |
23 | 23 | @Factory |
24 | 24 | public class GrpcStubFactory { |
25 | | - |
26 | | - @Inject |
27 | | - GrpcChannelManager grpcChannelManager; |
28 | | - |
| 25 | + |
29 | 26 | @Bean |
30 | 27 | @Singleton |
31 | | - public WorkerControllerServiceBlockingStub blockingWorkerServiceStub() { |
32 | | - return WorkerControllerServiceGrpc.newBlockingStub(grpcChannelManager.getDefaultChannel()); |
| 28 | + public WorkerControllerServiceBlockingStub blockingWorkerServiceStub(GrpcChannelManager manager) { |
| 29 | + return WorkerControllerServiceGrpc.newBlockingStub(manager.getDefaultChannel()); |
33 | 30 | } |
34 | 31 |
|
35 | 32 | @Bean |
36 | 33 | @Singleton |
37 | | - public WorkerControllerServiceStub asyncWorkerServiceStub() { |
38 | | - return WorkerControllerServiceGrpc.newStub(grpcChannelManager.getDefaultChannel()); |
| 34 | + public WorkerControllerServiceStub asyncWorkerServiceStub(GrpcChannelManager manager) { |
| 35 | + return WorkerControllerServiceGrpc.newStub(manager.getDefaultChannel()); |
39 | 36 | } |
40 | 37 |
|
41 | 38 | @Bean |
42 | 39 | @Singleton |
43 | | - public LivenessControllerServiceBlockingStub workerServiceStub() { |
44 | | - return LivenessControllerServiceGrpc.newBlockingStub(grpcChannelManager.getDefaultChannel()); |
| 40 | + public LivenessControllerServiceBlockingStub workerServiceStub(GrpcChannelManager manager) { |
| 41 | + return LivenessControllerServiceGrpc.newBlockingStub(manager.getDefaultChannel()); |
45 | 42 | } |
46 | 43 |
|
47 | 44 | @Bean |
48 | 45 | @Singleton |
49 | | - public ConnectControllerServiceBlockingStub connectControllerServiceBlockingStub() { |
50 | | - return ConnectControllerServiceGrpc.newBlockingStub(grpcChannelManager.getDefaultChannel()); |
| 46 | + public ConnectControllerServiceBlockingStub connectControllerServiceBlockingStub(GrpcChannelManager manager) { |
| 47 | + return ConnectControllerServiceGrpc.newBlockingStub(manager.getDefaultChannel()); |
51 | 48 | } |
52 | 49 |
|
53 | 50 | @Bean |
54 | 51 | @Singleton |
55 | | - public WorkerFlowMetaStoreServiceBlockingStub workerFlowMetaStoreServiceBlockingStub() { |
56 | | - return WorkerFlowMetaStoreServiceGrpc.newBlockingStub(grpcChannelManager.getDefaultChannel()); |
| 52 | + public WorkerFlowMetaStoreServiceBlockingStub workerFlowMetaStoreServiceBlockingStub(GrpcChannelManager manager) { |
| 53 | + return WorkerFlowMetaStoreServiceGrpc.newBlockingStub(manager.getDefaultChannel()); |
57 | 54 | } |
58 | 55 |
|
59 | 56 | @Bean |
60 | 57 | @Singleton |
61 | | - public KVMetadataServiceBlockingStub kvMetadataServiceBlockingStub() { |
62 | | - return KVMetadataServiceGrpc.newBlockingStub(grpcChannelManager.getDefaultChannel()); |
| 58 | + public KVMetadataServiceBlockingStub kvMetadataServiceBlockingStub(GrpcChannelManager manager) { |
| 59 | + return KVMetadataServiceGrpc.newBlockingStub(manager.getDefaultChannel()); |
63 | 60 | } |
64 | 61 | } |
0 commit comments