From 7aa81408930357e8388691b82aeb0e869611acf9 Mon Sep 17 00:00:00 2001
From: Siddharth More <Siddhi More>
Date: Fri, 29 Mar 2024 15:35:13 -0700
Subject: [PATCH] Changes

---
 node/grpc/server.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/node/grpc/server.go b/node/grpc/server.go
index a21461b11c..04c7fd6f0f 100644
--- a/node/grpc/server.go
+++ b/node/grpc/server.go
@@ -11,6 +11,7 @@ import (
 	"github.com/Layr-Labs/eigenda/api"
 	pb "github.com/Layr-Labs/eigenda/api/grpc/node"
 	"github.com/Layr-Labs/eigenda/common"
+	"github.com/Layr-Labs/eigenda/common/healthcheck"
 	"github.com/Layr-Labs/eigenda/core"
 	"github.com/Layr-Labs/eigenda/encoding"
 	"github.com/Layr-Labs/eigenda/node"
@@ -76,7 +77,6 @@ func (s *Server) Start() {
 			s.logger.Error("retrieval server failed; restarting.", "err", err)
 		}
 	}()
-
 }
 
 func (s *Server) serveDispersal() error {
@@ -95,6 +95,7 @@ func (s *Server) serveDispersal() error {
 	reflection.Register(gs)
 
 	pb.RegisterDispersalServer(gs, s)
+	healthcheck.RegisterHealthServer("node.Dispersal", gs)
 
 	s.logger.Info("port", s.config.InternalDispersalPort, "address", listener.Addr().String(), "GRPC Listening")
 	if err := gs.Serve(listener); err != nil {
@@ -119,6 +120,7 @@ func (s *Server) serveRetrieval() error {
 	reflection.Register(gs)
 
 	pb.RegisterRetrievalServer(gs, s)
+	healthcheck.RegisterHealthServer("node.Retrieval", gs)
 
 	s.logger.Info("port", s.config.InternalRetrievalPort, "address", listener.Addr().String(), "GRPC Listening")
 	if err := gs.Serve(listener); err != nil {