Unified Storage: Adds health check service for entity server (#86471)

pkg/services/store/entity/sqlstash/sql_storage_server.go

@@ -130,6 +130,19 @@ func (s *sqlEntityServer) Init() error {
 	return nil
 }
 
+func (s *sqlEntityServer) IsHealthy(ctx context.Context, r *entity.HealthCheckRequest) (*entity.HealthCheckResponse, error) {
+	sess, err := s.db.GetSession()
+	if err != nil {
+		return nil, err
+	}
+	_, err = sess.Query(ctx, "SELECT 1")
+	if err != nil {
+		return nil, err
+	}
+
+	return &entity.HealthCheckResponse{Status: entity.HealthCheckResponse_SERVING}, nil
+}
+
 func (s *sqlEntityServer) Stop() {
 	s.cancel()
 }

pkg/services/store/entity/sqlstash/sql_storage_server_test.go

@@ -18,6 +18,13 @@ func TestMain(m *testing.M) {
 	testsuite.Run(m)
 }
 
+func TestIsHealthy(t *testing.T) {
+	s := setUpTestServer(t)
+
+	_, err := s.IsHealthy(context.Background(), &entity.HealthCheckRequest{})
+	require.NoError(t, err)
+}
+
 func TestCreate(t *testing.T) {
 	s := setUpTestServer(t)