feat: add health check server (#589)

ionorg · Oct 26, 2021 · 09707c7 · 09707c7
1 parent e42ee58
commit 09707c7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/cmd/signal/allrpc/server/server.go b/cmd/signal/allrpc/server/server.go
@@ -14,6 +14,8 @@ import (
 	"github.com/pion/ion-sfu/pkg/sfu"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"google.golang.org/grpc"
+	"google.golang.org/grpc/health"
+	"google.golang.org/grpc/health/grpc_health_v1"
 
 	// pprof
 	_ "net/http/pprof"
@@ -51,6 +53,7 @@ func (s *Server) ServeGRPC(gaddr string) error {
 		grpc.StreamInterceptor(grpc_prometheus.StreamServerInterceptor),
 	)
 	pb.RegisterSFUServer(gs, grpcServer.NewServer(s.sfu))
+	grpc_health_v1.RegisterHealthServer(gs, health.NewServer())
 	s.logger.Info("GRPC Listening", "addr", gaddr)
 
 	if err := gs.Serve(l); err != nil {

diff --git a/cmd/signal/grpc/main.go b/cmd/signal/grpc/main.go
@@ -19,6 +19,8 @@ import (
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"github.com/spf13/viper"
 	"google.golang.org/grpc"
+	"google.golang.org/grpc/health"
+	"google.golang.org/grpc/health/grpc_health_v1"
 )
 
 type grpcConfig struct {
@@ -188,6 +190,7 @@ func main() {
 	dc.Use(datachannel.SubscriberAPI)
 
 	pb.RegisterSFUServer(s, server.NewServer(nsfu))
+	grpc_health_v1.RegisterHealthServer(s, health.NewServer())
 	grpc_prometheus.Register(s)
 
 	go startMetrics(metricsAddr)