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

pkg/services/store/entity/entity_grpc.pb.go

@@ -19,13 +19,14 @@ import (
 const _ = grpc.SupportPackageIsVersion7
 
 const (
-	EntityStore_Read_FullMethodName    = "/entity.EntityStore/Read"
-	EntityStore_Create_FullMethodName  = "/entity.EntityStore/Create"
-	EntityStore_Update_FullMethodName  = "/entity.EntityStore/Update"
-	EntityStore_Delete_FullMethodName  = "/entity.EntityStore/Delete"
-	EntityStore_History_FullMethodName = "/entity.EntityStore/History"
-	EntityStore_List_FullMethodName    = "/entity.EntityStore/List"
-	EntityStore_Watch_FullMethodName   = "/entity.EntityStore/Watch"
+	EntityStore_Read_FullMethodName      = "/entity.EntityStore/Read"
+	EntityStore_Create_FullMethodName    = "/entity.EntityStore/Create"
+	EntityStore_Update_FullMethodName    = "/entity.EntityStore/Update"
+	EntityStore_Delete_FullMethodName    = "/entity.EntityStore/Delete"
+	EntityStore_History_FullMethodName   = "/entity.EntityStore/History"
+	EntityStore_List_FullMethodName      = "/entity.EntityStore/List"
+	EntityStore_Watch_FullMethodName     = "/entity.EntityStore/Watch"
+	EntityStore_IsHealthy_FullMethodName = "/entity.EntityStore/IsHealthy"
 )
 
 // EntityStoreClient is the client API for EntityStore service.
@@ -39,6 +40,7 @@ type EntityStoreClient interface {
 	History(ctx context.Context, in *EntityHistoryRequest, opts ...grpc.CallOption) (*EntityHistoryResponse, error)
 	List(ctx context.Context, in *EntityListRequest, opts ...grpc.CallOption) (*EntityListResponse, error)
 	Watch(ctx context.Context, opts ...grpc.CallOption) (EntityStore_WatchClient, error)
+	IsHealthy(ctx context.Context, in *HealthCheckRequest, opts ...grpc.CallOption) (*HealthCheckResponse, error)
 }
 
 type entityStoreClient struct {
@@ -134,6 +136,15 @@ func (x *entityStoreWatchClient) Recv() (*EntityWatchResponse, error) {
 	return m, nil
 }
 
+func (c *entityStoreClient) IsHealthy(ctx context.Context, in *HealthCheckRequest, opts ...grpc.CallOption) (*HealthCheckResponse, error) {
+	out := new(HealthCheckResponse)
+	err := c.cc.Invoke(ctx, EntityStore_IsHealthy_FullMethodName, in, out, opts...)
+	if err != nil {
+		return nil, err
+	}
+	return out, nil
+}
+
 // EntityStoreServer is the server API for EntityStore service.
 // All implementations should embed UnimplementedEntityStoreServer
 // for forward compatibility
@@ -145,6 +156,7 @@ type EntityStoreServer interface {
 	History(context.Context, *EntityHistoryRequest) (*EntityHistoryResponse, error)
 	List(context.Context, *EntityListRequest) (*EntityListResponse, error)
 	Watch(EntityStore_WatchServer) error
+	IsHealthy(context.Context, *HealthCheckRequest) (*HealthCheckResponse, error)
 }
 
 // UnimplementedEntityStoreServer should be embedded to have forward compatible implementations.
@@ -172,6 +184,9 @@ func (UnimplementedEntityStoreServer) List(context.Context, *EntityListRequest)
 func (UnimplementedEntityStoreServer) Watch(EntityStore_WatchServer) error {
 	return status.Errorf(codes.Unimplemented, "method Watch not implemented")
 }
+func (UnimplementedEntityStoreServer) IsHealthy(context.Context, *HealthCheckRequest) (*HealthCheckResponse, error) {
+	return nil, status.Errorf(codes.Unimplemented, "method IsHealthy not implemented")
+}
 
 // UnsafeEntityStoreServer may be embedded to opt out of forward compatibility for this service.
 // Use of this interface is not recommended, as added methods to EntityStoreServer will
@@ -318,6 +333,24 @@ func (x *entityStoreWatchServer) Recv() (*EntityWatchRequest, error) {
 	return m, nil
 }
 
+func _EntityStore_IsHealthy_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
+	in := new(HealthCheckRequest)
+	if err := dec(in); err != nil {
+		return nil, err
+	}
+	if interceptor == nil {
+		return srv.(EntityStoreServer).IsHealthy(ctx, in)
+	}
+	info := &grpc.UnaryServerInfo{
+		Server:     srv,
+		FullMethod: EntityStore_IsHealthy_FullMethodName,
+	}
+	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
+		return srv.(EntityStoreServer).IsHealthy(ctx, req.(*HealthCheckRequest))
+	}
+	return interceptor(ctx, in, info, handler)
+}
+
 // EntityStore_ServiceDesc is the grpc.ServiceDesc for EntityStore service.
 // It's only intended for direct use with grpc.RegisterService,
 // and not to be introspected or modified (even as a copy)
@@ -349,6 +382,10 @@ var EntityStore_ServiceDesc = grpc.ServiceDesc{
 			MethodName: "List",
 			Handler:    _EntityStore_List_Handler,
 		},
+		{
+			MethodName: "IsHealthy",
+			Handler:    _EntityStore_IsHealthy_Handler,
+		},
 	},
 	Streams: []grpc.StreamDesc{
 		{