[core] Add basic grpc health check to core and Apricot

AliceO2Group · Jul 19, 2023 · eecef92 · eecef92
1 parent 8554551
commit eecef92
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/apricot/remote/server.go b/apricot/remote/server.go
@@ -37,6 +37,8 @@ import (
 	"github.com/spf13/viper"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/health"
+	"google.golang.org/grpc/health/grpc_health_v1"
 	"google.golang.org/grpc/reflection"
 	"google.golang.org/grpc/status"
 )
@@ -55,6 +57,7 @@ type RpcServer struct {
 
 func NewServer(service configuration.Service) *grpc.Server {
 	s := grpc.NewServer()
+	grpc_health_v1.RegisterHealthServer(s, health.NewServer())
 	apricotpb.RegisterApricotServer(s, &RpcServer{
 		service: service,
 	})

diff --git a/core/server.go b/core/server.go
@@ -42,6 +42,8 @@ import (
 	"github.com/jinzhu/copier"
 	"github.com/spf13/viper"
 	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/health"
+	"google.golang.org/grpc/health/grpc_health_v1"
 	"google.golang.org/grpc/reflection"
 	"google.golang.org/grpc/status"
 
@@ -60,6 +62,7 @@ const MAX_ERROR_LENGTH = 6000 // gRPC seems to impose this limit on the status m
 
 func NewServer(state *globalState) *grpc.Server {
 	s := grpc.NewServer()
+	grpc_health_v1.RegisterHealthServer(s, health.NewServer())
 	pb.RegisterControlServer(s, &RpcServer{
 		state:   state,
 		streams: newSafeStreamsMap(),