✨ Add health check

pkg/internal/grpc/attachments.go

@@ -7,7 +7,7 @@ import (
 	"git.solsynth.dev/hydrogen/paperclip/pkg/proto"
 	"google.golang.org/protobuf/types/known/emptypb"
 
-	"git.solsynth.dev/hydrogen/paperclip/pkg/models"
+	"git.solsynth.dev/hydrogen/paperclip/pkg/internal/models"
 	jsoniter "github.com/json-iterator/go"
 )
 

pkg/internal/grpc/client.go

@@ -1,22 +0,0 @@
-package grpc
-
-import (
-	idpb "git.solsynth.dev/hydrogen/passport/pkg/grpc/proto"
-	"google.golang.org/grpc/credentials/insecure"
-
-	"github.com/spf13/viper"
-	"google.golang.org/grpc"
-)
-
-var Auth idpb.AuthClient
-
-func ConnectPassport() error {
-	addr := viper.GetString("passport.grpc_endpoint")
-	if conn, err := grpc.Dial(addr, grpc.WithTransportCredentials(insecure.NewCredentials())); err != nil {
-		return err
-	} else {
-		Auth = idpb.NewAuthClient(conn)
-	}
-
-	return nil
-}

pkg/internal/grpc/health.go

@@ -0,0 +1,26 @@
+package grpc
+
+import (
+	"context"
+	health "google.golang.org/grpc/health/grpc_health_v1"
+	"time"
+)
+
+func (v *Server) Check(ctx context.Context, request *health.HealthCheckRequest) (*health.HealthCheckResponse, error) {
+	return &health.HealthCheckResponse{
+		Status: health.HealthCheckResponse_SERVING,
+	}, nil
+}
+
+func (v *Server) Watch(request *health.HealthCheckRequest, server health.Health_WatchServer) error {
+	for {
+		if server.Send(&health.HealthCheckResponse{
+			Status: health.HealthCheckResponse_SERVING,
+		}) != nil {
+			break
+		}
+		time.Sleep(1000 * time.Millisecond)
+	}
+
+	return nil
+}

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)
 }