Merge pull request #10303 from dragonchaser/grpc-checks

Grpc checks
owncloud · Oct 15, 2024 · 4f8d0bb · 4f8d0bb
2 parents fde6b4c + 49cf9b0
commit 4f8d0bb
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 3 deletions.
diff --git a/ocis-pkg/handlers/checkgrpc.go b/ocis-pkg/handlers/checkgrpc.go
@@ -0,0 +1,22 @@
+package handlers
+
+import (
+	"context"
+	"fmt"
+
+	"google.golang.org/grpc/credentials/insecure"
+
+	"google.golang.org/grpc"
+)
+
+// NewGRPCCheck checks the reachability of a grpc server.
+func NewGRPCCheck(address string) func(ctx context.Context) error {
+	return func(_ context.Context) error {
+		conn, err := grpc.NewClient(address, grpc.WithTransportCredentials(insecure.NewCredentials()))
+		if err != nil {
+			return fmt.Errorf("could not connect to grpc server: %v", err)
+		}
+		_ = conn.Close()
+		return nil
+	}
+}
diff --git a/services/search/pkg/server/debug/server.go b/services/search/pkg/server/debug/server.go
@@ -14,7 +14,8 @@ func Server(opts ...Option) (*http.Server, error) {
 
 	checkHandler := handlers.NewCheckHandler(
 		handlers.NewCheckHandlerConfiguration().
-			WithLogger(options.Logger),
+			WithLogger(options.Logger).
+			WithCheck("grpc reachability", handlers.NewGRPCCheck(options.Config.GRPC.Addr)),
 	)
 
 	return debug.NewService(

diff --git a/services/settings/pkg/server/debug/server.go b/services/settings/pkg/server/debug/server.go
@@ -15,7 +15,8 @@ func Server(opts ...Option) (*http.Server, error) {
 	checkHandler := handlers.NewCheckHandler(
 		handlers.NewCheckHandlerConfiguration().
 			WithLogger(options.Logger).
-			WithCheck("web reachability", handlers.NewHTTPCheck(options.Config.HTTP.Addr)),
+			WithCheck("web reachability", handlers.NewHTTPCheck(options.Config.HTTP.Addr)).
+			WithCheck("grpc reachability", handlers.NewGRPCCheck(options.Config.GRPC.Addr)),
 	)
 
 	return debug.NewService(

diff --git a/services/thumbnails/pkg/server/debug/server.go b/services/thumbnails/pkg/server/debug/server.go
@@ -15,7 +15,8 @@ func Server(opts ...Option) (*http.Server, error) {
 	checkHandler := handlers.NewCheckHandler(
 		handlers.NewCheckHandlerConfiguration().
 			WithLogger(options.Logger).
-			WithCheck("web reachability", handlers.NewHTTPCheck(options.Config.HTTP.Addr)),
+			WithCheck("web reachability", handlers.NewHTTPCheck(options.Config.HTTP.Addr)).
+			WithCheck("grpc reachability", handlers.NewGRPCCheck(options.Config.GRPC.Addr)),
 	)
 
 	return debug.NewService(