| // Code generated by protoc-gen-go-grpc. DO NOT EDIT. |
| |
| package si |
| |
| import ( |
| context "context" |
| grpc "google.golang.org/grpc" |
| codes "google.golang.org/grpc/codes" |
| status "google.golang.org/grpc/status" |
| ) |
| |
| // This is a compile-time assertion to ensure that this generated file |
| // is compatible with the grpc package it is being compiled against. |
| // Requires gRPC-Go v1.32.0 or later. |
| const _ = grpc.SupportPackageIsVersion7 |
| |
| // SchedulerClient is the client API for Scheduler service. |
| // |
| // For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream. |
| type SchedulerClient interface { |
| // Register a RM, if it is a reconnect from previous RM the call will |
| // trigger a cleanup of all in-memory data and resync with RM. |
| RegisterResourceManager(ctx context.Context, in *RegisterResourceManagerRequest, opts ...grpc.CallOption) (*RegisterResourceManagerResponse, error) |
| // Update Scheduler status (this includes node status update, allocation request |
| // updates, etc. And receive updates from scheduler for allocation changes, |
| // any required status changes, etc. |
| Update(ctx context.Context, opts ...grpc.CallOption) (Scheduler_UpdateClient, error) |
| } |
| |
| type schedulerClient struct { |
| cc grpc.ClientConnInterface |
| } |
| |
| func NewSchedulerClient(cc grpc.ClientConnInterface) SchedulerClient { |
| return &schedulerClient{cc} |
| } |
| |
| func (c *schedulerClient) RegisterResourceManager(ctx context.Context, in *RegisterResourceManagerRequest, opts ...grpc.CallOption) (*RegisterResourceManagerResponse, error) { |
| out := new(RegisterResourceManagerResponse) |
| err := c.cc.Invoke(ctx, "/si.v1.Scheduler/RegisterResourceManager", in, out, opts...) |
| if err != nil { |
| return nil, err |
| } |
| return out, nil |
| } |
| |
| func (c *schedulerClient) Update(ctx context.Context, opts ...grpc.CallOption) (Scheduler_UpdateClient, error) { |
| stream, err := c.cc.NewStream(ctx, &Scheduler_ServiceDesc.Streams[0], "/si.v1.Scheduler/Update", opts...) |
| if err != nil { |
| return nil, err |
| } |
| x := &schedulerUpdateClient{stream} |
| return x, nil |
| } |
| |
| type Scheduler_UpdateClient interface { |
| Send(*UpdateRequest) error |
| Recv() (*UpdateResponse, error) |
| grpc.ClientStream |
| } |
| |
| type schedulerUpdateClient struct { |
| grpc.ClientStream |
| } |
| |
| func (x *schedulerUpdateClient) Send(m *UpdateRequest) error { |
| return x.ClientStream.SendMsg(m) |
| } |
| |
| func (x *schedulerUpdateClient) Recv() (*UpdateResponse, error) { |
| m := new(UpdateResponse) |
| if err := x.ClientStream.RecvMsg(m); err != nil { |
| return nil, err |
| } |
| return m, nil |
| } |
| |
| // SchedulerServer is the server API for Scheduler service. |
| // All implementations must embed UnimplementedSchedulerServer |
| // for forward compatibility |
| type SchedulerServer interface { |
| // Register a RM, if it is a reconnect from previous RM the call will |
| // trigger a cleanup of all in-memory data and resync with RM. |
| RegisterResourceManager(context.Context, *RegisterResourceManagerRequest) (*RegisterResourceManagerResponse, error) |
| // Update Scheduler status (this includes node status update, allocation request |
| // updates, etc. And receive updates from scheduler for allocation changes, |
| // any required status changes, etc. |
| Update(Scheduler_UpdateServer) error |
| mustEmbedUnimplementedSchedulerServer() |
| } |
| |
| // UnimplementedSchedulerServer must be embedded to have forward compatible implementations. |
| type UnimplementedSchedulerServer struct { |
| } |
| |
| func (UnimplementedSchedulerServer) RegisterResourceManager(context.Context, *RegisterResourceManagerRequest) (*RegisterResourceManagerResponse, error) { |
| return nil, status.Errorf(codes.Unimplemented, "method RegisterResourceManager not implemented") |
| } |
| func (UnimplementedSchedulerServer) Update(Scheduler_UpdateServer) error { |
| return status.Errorf(codes.Unimplemented, "method Update not implemented") |
| } |
| func (UnimplementedSchedulerServer) mustEmbedUnimplementedSchedulerServer() {} |
| |
| // UnsafeSchedulerServer may be embedded to opt out of forward compatibility for this service. |
| // Use of this interface is not recommended, as added methods to SchedulerServer will |
| // result in compilation errors. |
| type UnsafeSchedulerServer interface { |
| mustEmbedUnimplementedSchedulerServer() |
| } |
| |
| func RegisterSchedulerServer(s grpc.ServiceRegistrar, srv SchedulerServer) { |
| s.RegisterService(&Scheduler_ServiceDesc, srv) |
| } |
| |
| func _Scheduler_RegisterResourceManager_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { |
| in := new(RegisterResourceManagerRequest) |
| if err := dec(in); err != nil { |
| return nil, err |
| } |
| if interceptor == nil { |
| return srv.(SchedulerServer).RegisterResourceManager(ctx, in) |
| } |
| info := &grpc.UnaryServerInfo{ |
| Server: srv, |
| FullMethod: "/si.v1.Scheduler/RegisterResourceManager", |
| } |
| handler := func(ctx context.Context, req interface{}) (interface{}, error) { |
| return srv.(SchedulerServer).RegisterResourceManager(ctx, req.(*RegisterResourceManagerRequest)) |
| } |
| return interceptor(ctx, in, info, handler) |
| } |
| |
| func _Scheduler_Update_Handler(srv interface{}, stream grpc.ServerStream) error { |
| return srv.(SchedulerServer).Update(&schedulerUpdateServer{stream}) |
| } |
| |
| type Scheduler_UpdateServer interface { |
| Send(*UpdateResponse) error |
| Recv() (*UpdateRequest, error) |
| grpc.ServerStream |
| } |
| |
| type schedulerUpdateServer struct { |
| grpc.ServerStream |
| } |
| |
| func (x *schedulerUpdateServer) Send(m *UpdateResponse) error { |
| return x.ServerStream.SendMsg(m) |
| } |
| |
| func (x *schedulerUpdateServer) Recv() (*UpdateRequest, error) { |
| m := new(UpdateRequest) |
| if err := x.ServerStream.RecvMsg(m); err != nil { |
| return nil, err |
| } |
| return m, nil |
| } |
| |
| // Scheduler_ServiceDesc is the grpc.ServiceDesc for Scheduler service. |
| // It's only intended for direct use with grpc.RegisterService, |
| // and not to be introspected or modified (even as a copy) |
| var Scheduler_ServiceDesc = grpc.ServiceDesc{ |
| ServiceName: "si.v1.Scheduler", |
| HandlerType: (*SchedulerServer)(nil), |
| Methods: []grpc.MethodDesc{ |
| { |
| MethodName: "RegisterResourceManager", |
| Handler: _Scheduler_RegisterResourceManager_Handler, |
| }, |
| }, |
| Streams: []grpc.StreamDesc{ |
| { |
| StreamName: "Update", |
| Handler: _Scheduler_Update_Handler, |
| ServerStreams: true, |
| ClientStreams: true, |
| }, |
| }, |
| Metadata: "incubator-yunikorn-scheduler-interface/si.proto", |
| } |