fix: grpchealth and grpcreflect take multiple name arguments

Signed-off-by: Chris Gianelloni <[email protected]>
blinklabs-io · Sep 5, 2024 · a1ed84d · a1ed84d
1 parent 013556a
commit a1ed84d
Showing 1 changed file with 18 additions and 57 deletions.
diff --git a/internal/utxorpc/api.go b/internal/utxorpc/api.go
@@ -56,73 +56,34 @@ func Start(cfg *config.Config) error {
 	mux.Handle(watchPath, watchHandler)
 	mux.Handle(
 		grpchealth.NewHandler(
-			grpchealth.NewStaticChecker(queryconnect.QueryServiceName),
-			compress1KB,
-		),
-	)
-	mux.Handle(
-		grpchealth.NewHandler(
-			grpchealth.NewStaticChecker(submitconnect.SubmitServiceName),
-			compress1KB,
-		),
-	)
-	mux.Handle(
-		grpchealth.NewHandler(
-			grpchealth.NewStaticChecker(syncconnect.SyncServiceName),
-			compress1KB,
-		),
-	)
-	mux.Handle(
-		grpchealth.NewHandler(
-			grpchealth.NewStaticChecker(watchconnect.WatchServiceName),
-			compress1KB,
-		),
-	)
-	mux.Handle(
-		grpcreflect.NewHandlerV1(
-			grpcreflect.NewStaticReflector(queryconnect.QueryServiceName),
-			compress1KB,
-		),
-	)
-	mux.Handle(
-		grpcreflect.NewHandlerV1(
-			grpcreflect.NewStaticReflector(submitconnect.SubmitServiceName),
-			compress1KB,
-		),
-	)
-	mux.Handle(
-		grpcreflect.NewHandlerV1(
-			grpcreflect.NewStaticReflector(syncconnect.SyncServiceName),
+			grpchealth.NewStaticChecker(
+				queryconnect.QueryServiceName,
+				submitconnect.SubmitServiceName,
+				syncconnect.SyncServiceName,
+				watchconnect.WatchServiceName,
+			),
 			compress1KB,
 		),
 	)
 	mux.Handle(
 		grpcreflect.NewHandlerV1(
-			grpcreflect.NewStaticReflector(watchconnect.WatchServiceName),
-			compress1KB,
-		),
-	)
-	mux.Handle(
-		grpcreflect.NewHandlerV1Alpha(
-			grpcreflect.NewStaticReflector(queryconnect.QueryServiceName),
-			compress1KB,
-		),
-	)
-	mux.Handle(
-		grpcreflect.NewHandlerV1Alpha(
-			grpcreflect.NewStaticReflector(submitconnect.SubmitServiceName),
-			compress1KB,
-		),
-	)
-	mux.Handle(
-		grpcreflect.NewHandlerV1Alpha(
-			grpcreflect.NewStaticReflector(syncconnect.SyncServiceName),
+			grpcreflect.NewStaticReflector(
+				queryconnect.QueryServiceName,
+				submitconnect.SubmitServiceName,
+				syncconnect.SyncServiceName,
+				watchconnect.WatchServiceName,
+			),
 			compress1KB,
 		),
 	)
 	mux.Handle(
 		grpcreflect.NewHandlerV1Alpha(
-			grpcreflect.NewStaticReflector(watchconnect.WatchServiceName),
+			grpcreflect.NewStaticReflector(
+				queryconnect.QueryServiceName,
+				submitconnect.SubmitServiceName,
+				syncconnect.SyncServiceName,
+				watchconnect.WatchServiceName,
+			),
 			compress1KB,
 		),
 	)