✨ Add health check

pkg/internal/grpc/server.go

@@ -2,28 +2,33 @@ package grpc
 
 import (
 	"git.solsynth.dev/hydrogen/paperclip/pkg/proto"
-	"net"
-
 	"github.com/spf13/viper"
 	"google.golang.org/grpc"
+	health "google.golang.org/grpc/health/grpc_health_v1"
 	"google.golang.org/grpc/reflection"
+	"net"
 )
 
 type Server struct {
 	proto.UnimplementedAttachmentsServer
 }
 
-func StartGrpc() error {
-	listen, err := net.Listen("tcp", viper.GetString("grpc_bind"))
+var S *grpc.Server
+
+func NewGRPC() {
+	S = grpc.NewServer()
+
+	proto.RegisterAttachmentsServer(S, &Server{})
+	health.RegisterHealthServer(S, &Server{})
+
+	reflection.Register(S)
+}
+
+func ListenGRPC() error {
+	listener, err := net.Listen("tcp", viper.GetString("grpc_bind"))
 	if err != nil {
 		return err
 	}
 
-	server := grpc.NewServer()
-
-	proto.RegisterAttachmentsServer(server, &Server{})
-
-	reflection.Register(server)
-
-	return server.Serve(listen)
+	return S.Serve(listener)
 }