From 0a8c589b9be0e24e48d136831c7dbaf25ec9f744 Mon Sep 17 00:00:00 2001 From: trevleon Date: Sat, 1 Jun 2024 22:02:24 -0700 Subject: [PATCH] printing --- go/cmd/globalscheduler/main.go | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/go/cmd/globalscheduler/main.go b/go/cmd/globalscheduler/main.go index afb4a50..e2e7437 100644 --- a/go/cmd/globalscheduler/main.go +++ b/go/cmd/globalscheduler/main.go @@ -67,7 +67,7 @@ type ObjClient interface { func (s *server) Heartbeat(ctx context.Context, req *pb.HeartbeatRequest ) (*pb.StatusResponse, error) { - // log.Printf("heartbeat from %v", req.NodeId) + log.Printf("heartbeat from %v", req.NodeId) mu.Lock() s.status[req.NodeId] = HeartbeatEntry{timeReceived: time.Now(), numRunningTasks: req.RunningTasks, numQueuedTasks: req.QueuedTasks, avgRunningTime: req.AvgRunningTime, avgBandwidth: req.AvgBandwidth} mu.Unlock() @@ -87,7 +87,14 @@ func (s *server) LiveNodesHeartbeat(ctx context.Context) (error) { func(s *server) SendLiveNodes(ctx context.Context) (error) { liveNodes := make(map[uint64]bool) for uid, heartbeat := range s.status { + liveNodes[uid] = time.Since(heartbeat.timeReceived) < LIVE_NODE_TIMEOUT + if _, val := liveNodes[uid]; val { + log.Printf("%v sent as live", req.NodeId) + } + else { + log.Printf("%v sent as dead", req.NodeId) + } } s.gcsClient.RegisterLiveNodes(ctx, &pb.LiveNodesRequest{LiveNodes: liveNodes}) return nil