More RPC-related logs (#136)

* More RPC-related logs

* Notifier should be set before we use it in the scheduler
This commit is contained in:
Nikolay Edigaryev
2023-09-27 20:16:00 +04:00
committed by GitHub
parent 64987f6d59
commit 40f58e4aee
5 changed files with 20 additions and 8 deletions
+6
View File
@@ -21,6 +21,8 @@ import (
)
func (worker *Worker) watchRPC(ctx context.Context) error {
worker.logger.Infof("connecting to %s over gRPC", worker.client.GRPCTarget())
conn, err := grpc.Dial(worker.client.GRPCTarget(),
grpc.WithTransportCredentials(worker.client.GRPCTransportCredentials()),
grpc.WithKeepaliveParams(keepalive.ClientParameters{
@@ -31,6 +33,8 @@ func (worker *Worker) watchRPC(ctx context.Context) error {
return err
}
worker.logger.Infof("gRPC connection established, starting gRPC stream with the controller")
client := rpc.NewControllerClient(conn)
ctxWithMetadata := metadata.NewOutgoingContext(ctx, worker.grpcMetadata())
@@ -40,6 +44,8 @@ func (worker *Worker) watchRPC(ctx context.Context) error {
return err
}
worker.logger.Infof("running gRPC stream with the controller")
for {
watchFromController, err := stream.Recv()
if err != nil {