tempo-query: register gRPC health server (#4178) (#4186)

Signed-off-by: Benedikt Bongartz <[email protected]>
Co-authored-by: Joe Elliott <[email protected]>
(cherry picked from commit 4726342)

Co-authored-by: Ben B <[email protected]>

CHANGELOG.md

@@ -1,5 +1,6 @@
 ## main / unreleased
 
+* [ENHANCEMENT] Register gRPC health server to tempo-query [#4178](https://github.com/grafana/tempo/pull/4178) (@frzifus)
 * [ENHANCEMENT] Support Tempo on IBM s390x [#4172](https://github.com/grafana/tempo/pull/4175) (@pavolloffay)
 * [ENHANCEMENT] Changed log level from INFO to DEBUG for the TempoDB Find operation using traceId to reduce excessive/unwanted logs in log search. [#4179](https://github.com/grafana/tempo/pull/4179) (@Aki0x137)
 * [ENHANCEMENT] tempo-query: separate tls settings for server and client [#4177](https://github.com/grafana/tempo/pull/4177) (@frzifus)

cmd/tempo-query/main.go

@@ -15,6 +15,8 @@ import (
 	"go.uber.org/zap/zapcore"
 	google_grpc "google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
+	"google.golang.org/grpc/health"
+	"google.golang.org/grpc/health/grpc_health_v1"
 
 	"github.com/grafana/tempo/cmd/tempo-query/tempo"
 )
@@ -72,6 +74,10 @@ func main() {
 	storage_v1.RegisterDependenciesReaderPluginServer(srv, backend)
 	storage_v1.RegisterSpanWriterPluginServer(srv, backend)
 
+	healthServer := health.NewServer()
+	grpc_health_v1.RegisterHealthServer(srv, healthServer)
+	healthServer.SetServingStatus("", grpc_health_v1.HealthCheckResponse_SERVING)
+
 	lis, err := net.Listen("tcp", cfg.Address)
 	if err != nil {
 		logger.Error("failed to listen", zap.Error(err))