diff --git a/manager/api/grpc/client.go b/manager/api/grpc/client.go index 4dffe2c8..9e1c8306 100644 --- a/manager/api/grpc/client.go +++ b/manager/api/grpc/client.go @@ -73,7 +73,13 @@ func (client ManagerClient) Process(ctx context.Context, cancel context.CancelFu cancel() return errors.Wrap(errTerminationFromServer, errors.New(mes.TerminateReq.Message)) case *pkgmanager.ServerStreamMessage_StopComputation: + msg := &pkgmanager.ClientStreamMessage_StopComputationRes{StopComputationRes: &pkgmanager.StopComputationResponse{ + ComputationId: mes.StopComputation.ComputationId, + }} if err := client.svc.Stop(ctx, mes.StopComputation.ComputationId); err != nil { + msg.StopComputationRes.Message = err.Error() + } + if err := client.stream.Send(&pkgmanager.ClientStreamMessage{Message: msg}); err != nil { return err } case *pkgmanager.ServerStreamMessage_BackendInfoReq: diff --git a/manager/manager.proto b/manager/manager.proto index b3654b71..65aa316f 100644 --- a/manager/manager.proto +++ b/manager/manager.proto @@ -21,6 +21,11 @@ message StopComputation { string computation_id = 1; } +message StopComputationResponse { + string computation_id = 1; + string message = 2; +} + message RunResponse{ string agent_port = 1; string computation_id = 2; @@ -53,6 +58,7 @@ message ClientStreamMessage { AgentEvent agent_event = 2; RunResponse run_res = 3; BackendInfo backendInfo = 4; + StopComputationResponse stopComputationRes = 5; } } diff --git a/manager/service.go b/manager/service.go index 8dc6f9ee..bce53b65 100644 --- a/manager/service.go +++ b/manager/service.go @@ -167,12 +167,15 @@ func (ms *managerService) Run(ctx context.Context, c *manager.ComputationRunReq) func (ms *managerService) Stop(ctx context.Context, computationID string) error { cvm, ok := ms.vms[computationID] if !ok { + defer ms.publishEvent("stop-computation", computationID, "failed", json.RawMessage{}) return ErrNotFound } if err := cvm.Stop(); err != nil { + defer ms.publishEvent("stop-computation", computationID, "failed", json.RawMessage{}) return err } delete(ms.vms, computationID) + defer ms.publishEvent("stop-computation", computationID, "complete", json.RawMessage{}) return nil } diff --git a/manager/service_test.go b/manager/service_test.go index 87ce002d..e01ac1f7 100644 --- a/manager/service_test.go +++ b/manager/service_test.go @@ -115,7 +115,6 @@ func TestRun(t *testing.T) { vmf.AssertExpectations(t) - // Clear the events channel for len(eventsChan) > 0 { <-eventsChan } @@ -124,6 +123,10 @@ func TestRun(t *testing.T) { } func TestStop(t *testing.T) { + vmf := new(mocks.Provider) + vmMock := new(mocks.VM) + vmf.On("Execute", mock.Anything, mock.Anything, mock.Anything).Return(vmMock) + tests := []struct { name string computationID string @@ -156,8 +159,12 @@ func TestStop(t *testing.T) { for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { + logger := slog.Default() + eventsChan := make(chan *manager.ClientStreamMessage, 10) ms := &managerService{ - vms: make(map[string]vm.VM), + logger: logger, + vms: make(map[string]vm.VM), + eventsChan: eventsChan, } vmMock := new(mocks.VM) @@ -180,6 +187,10 @@ func TestStop(t *testing.T) { assert.NoError(t, err) assert.Len(t, ms.vms, 0) } + + for len(eventsChan) > 0 { + <-eventsChan + } }) } } diff --git a/pkg/manager/manager.pb.go b/pkg/manager/manager.pb.go index f3b8bafe..80ccb028 100644 --- a/pkg/manager/manager.pb.go +++ b/pkg/manager/manager.pb.go @@ -118,6 +118,61 @@ func (x *StopComputation) GetComputationId() string { return "" } +type StopComputationResponse struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + ComputationId string `protobuf:"bytes,1,opt,name=computation_id,json=computationId,proto3" json:"computation_id,omitempty"` + Message string `protobuf:"bytes,2,opt,name=message,proto3" json:"message,omitempty"` +} + +func (x *StopComputationResponse) Reset() { + *x = StopComputationResponse{} + if protoimpl.UnsafeEnabled { + mi := &file_manager_manager_proto_msgTypes[2] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) + } +} + +func (x *StopComputationResponse) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*StopComputationResponse) ProtoMessage() {} + +func (x *StopComputationResponse) ProtoReflect() protoreflect.Message { + mi := &file_manager_manager_proto_msgTypes[2] + if protoimpl.UnsafeEnabled && x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use StopComputationResponse.ProtoReflect.Descriptor instead. +func (*StopComputationResponse) Descriptor() ([]byte, []int) { + return file_manager_manager_proto_rawDescGZIP(), []int{2} +} + +func (x *StopComputationResponse) GetComputationId() string { + if x != nil { + return x.ComputationId + } + return "" +} + +func (x *StopComputationResponse) GetMessage() string { + if x != nil { + return x.Message + } + return "" +} + type RunResponse struct { state protoimpl.MessageState sizeCache protoimpl.SizeCache @@ -130,7 +185,7 @@ type RunResponse struct { func (x *RunResponse) Reset() { *x = RunResponse{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[2] + mi := &file_manager_manager_proto_msgTypes[3] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -143,7 +198,7 @@ func (x *RunResponse) String() string { func (*RunResponse) ProtoMessage() {} func (x *RunResponse) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[2] + mi := &file_manager_manager_proto_msgTypes[3] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -156,7 +211,7 @@ func (x *RunResponse) ProtoReflect() protoreflect.Message { // Deprecated: Use RunResponse.ProtoReflect.Descriptor instead. func (*RunResponse) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{2} + return file_manager_manager_proto_rawDescGZIP(), []int{3} } func (x *RunResponse) GetAgentPort() string { @@ -185,7 +240,7 @@ type BackendInfo struct { func (x *BackendInfo) Reset() { *x = BackendInfo{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[3] + mi := &file_manager_manager_proto_msgTypes[4] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -198,7 +253,7 @@ func (x *BackendInfo) String() string { func (*BackendInfo) ProtoMessage() {} func (x *BackendInfo) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[3] + mi := &file_manager_manager_proto_msgTypes[4] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -211,7 +266,7 @@ func (x *BackendInfo) ProtoReflect() protoreflect.Message { // Deprecated: Use BackendInfo.ProtoReflect.Descriptor instead. func (*BackendInfo) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{3} + return file_manager_manager_proto_rawDescGZIP(), []int{4} } func (x *BackendInfo) GetInfo() []byte { @@ -244,7 +299,7 @@ type AgentEvent struct { func (x *AgentEvent) Reset() { *x = AgentEvent{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[4] + mi := &file_manager_manager_proto_msgTypes[5] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -257,7 +312,7 @@ func (x *AgentEvent) String() string { func (*AgentEvent) ProtoMessage() {} func (x *AgentEvent) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[4] + mi := &file_manager_manager_proto_msgTypes[5] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -270,7 +325,7 @@ func (x *AgentEvent) ProtoReflect() protoreflect.Message { // Deprecated: Use AgentEvent.ProtoReflect.Descriptor instead. func (*AgentEvent) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{4} + return file_manager_manager_proto_rawDescGZIP(), []int{5} } func (x *AgentEvent) GetEventType() string { @@ -329,7 +384,7 @@ type AgentLog struct { func (x *AgentLog) Reset() { *x = AgentLog{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[5] + mi := &file_manager_manager_proto_msgTypes[6] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -342,7 +397,7 @@ func (x *AgentLog) String() string { func (*AgentLog) ProtoMessage() {} func (x *AgentLog) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[5] + mi := &file_manager_manager_proto_msgTypes[6] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -355,7 +410,7 @@ func (x *AgentLog) ProtoReflect() protoreflect.Message { // Deprecated: Use AgentLog.ProtoReflect.Descriptor instead. func (*AgentLog) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{5} + return file_manager_manager_proto_rawDescGZIP(), []int{6} } func (x *AgentLog) GetMessage() string { @@ -397,13 +452,14 @@ type ClientStreamMessage struct { // *ClientStreamMessage_AgentEvent // *ClientStreamMessage_RunRes // *ClientStreamMessage_BackendInfo + // *ClientStreamMessage_StopComputationRes Message isClientStreamMessage_Message `protobuf_oneof:"message"` } func (x *ClientStreamMessage) Reset() { *x = ClientStreamMessage{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[6] + mi := &file_manager_manager_proto_msgTypes[7] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -416,7 +472,7 @@ func (x *ClientStreamMessage) String() string { func (*ClientStreamMessage) ProtoMessage() {} func (x *ClientStreamMessage) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[6] + mi := &file_manager_manager_proto_msgTypes[7] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -429,7 +485,7 @@ func (x *ClientStreamMessage) ProtoReflect() protoreflect.Message { // Deprecated: Use ClientStreamMessage.ProtoReflect.Descriptor instead. func (*ClientStreamMessage) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{6} + return file_manager_manager_proto_rawDescGZIP(), []int{7} } func (m *ClientStreamMessage) GetMessage() isClientStreamMessage_Message { @@ -467,6 +523,13 @@ func (x *ClientStreamMessage) GetBackendInfo() *BackendInfo { return nil } +func (x *ClientStreamMessage) GetStopComputationRes() *StopComputationResponse { + if x, ok := x.GetMessage().(*ClientStreamMessage_StopComputationRes); ok { + return x.StopComputationRes + } + return nil +} + type isClientStreamMessage_Message interface { isClientStreamMessage_Message() } @@ -487,6 +550,10 @@ type ClientStreamMessage_BackendInfo struct { BackendInfo *BackendInfo `protobuf:"bytes,4,opt,name=backendInfo,proto3,oneof"` } +type ClientStreamMessage_StopComputationRes struct { + StopComputationRes *StopComputationResponse `protobuf:"bytes,5,opt,name=stopComputationRes,proto3,oneof"` +} + func (*ClientStreamMessage_AgentLog) isClientStreamMessage_Message() {} func (*ClientStreamMessage_AgentEvent) isClientStreamMessage_Message() {} @@ -495,6 +562,8 @@ func (*ClientStreamMessage_RunRes) isClientStreamMessage_Message() {} func (*ClientStreamMessage_BackendInfo) isClientStreamMessage_Message() {} +func (*ClientStreamMessage_StopComputationRes) isClientStreamMessage_Message() {} + type ServerStreamMessage struct { state protoimpl.MessageState sizeCache protoimpl.SizeCache @@ -513,7 +582,7 @@ type ServerStreamMessage struct { func (x *ServerStreamMessage) Reset() { *x = ServerStreamMessage{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[7] + mi := &file_manager_manager_proto_msgTypes[8] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -526,7 +595,7 @@ func (x *ServerStreamMessage) String() string { func (*ServerStreamMessage) ProtoMessage() {} func (x *ServerStreamMessage) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[7] + mi := &file_manager_manager_proto_msgTypes[8] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -539,7 +608,7 @@ func (x *ServerStreamMessage) ProtoReflect() protoreflect.Message { // Deprecated: Use ServerStreamMessage.ProtoReflect.Descriptor instead. func (*ServerStreamMessage) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{7} + return file_manager_manager_proto_rawDescGZIP(), []int{8} } func (m *ServerStreamMessage) GetMessage() isServerStreamMessage_Message { @@ -629,7 +698,7 @@ type RunReqChunks struct { func (x *RunReqChunks) Reset() { *x = RunReqChunks{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[8] + mi := &file_manager_manager_proto_msgTypes[9] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -642,7 +711,7 @@ func (x *RunReqChunks) String() string { func (*RunReqChunks) ProtoMessage() {} func (x *RunReqChunks) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[8] + mi := &file_manager_manager_proto_msgTypes[9] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -655,7 +724,7 @@ func (x *RunReqChunks) ProtoReflect() protoreflect.Message { // Deprecated: Use RunReqChunks.ProtoReflect.Descriptor instead. func (*RunReqChunks) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{8} + return file_manager_manager_proto_rawDescGZIP(), []int{9} } func (x *RunReqChunks) GetData() []byte { @@ -682,7 +751,7 @@ type ComputationRunReq struct { func (x *ComputationRunReq) Reset() { *x = ComputationRunReq{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[9] + mi := &file_manager_manager_proto_msgTypes[10] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -695,7 +764,7 @@ func (x *ComputationRunReq) String() string { func (*ComputationRunReq) ProtoMessage() {} func (x *ComputationRunReq) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[9] + mi := &file_manager_manager_proto_msgTypes[10] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -708,7 +777,7 @@ func (x *ComputationRunReq) ProtoReflect() protoreflect.Message { // Deprecated: Use ComputationRunReq.ProtoReflect.Descriptor instead. func (*ComputationRunReq) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{9} + return file_manager_manager_proto_rawDescGZIP(), []int{10} } func (x *ComputationRunReq) GetId() string { @@ -771,7 +840,7 @@ type BackendInfoReq struct { func (x *BackendInfoReq) Reset() { *x = BackendInfoReq{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[10] + mi := &file_manager_manager_proto_msgTypes[11] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -784,7 +853,7 @@ func (x *BackendInfoReq) String() string { func (*BackendInfoReq) ProtoMessage() {} func (x *BackendInfoReq) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[10] + mi := &file_manager_manager_proto_msgTypes[11] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -797,7 +866,7 @@ func (x *BackendInfoReq) ProtoReflect() protoreflect.Message { // Deprecated: Use BackendInfoReq.ProtoReflect.Descriptor instead. func (*BackendInfoReq) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{10} + return file_manager_manager_proto_rawDescGZIP(), []int{11} } func (x *BackendInfoReq) GetId() string { @@ -818,7 +887,7 @@ type ResultConsumer struct { func (x *ResultConsumer) Reset() { *x = ResultConsumer{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[11] + mi := &file_manager_manager_proto_msgTypes[12] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -831,7 +900,7 @@ func (x *ResultConsumer) String() string { func (*ResultConsumer) ProtoMessage() {} func (x *ResultConsumer) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[11] + mi := &file_manager_manager_proto_msgTypes[12] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -844,7 +913,7 @@ func (x *ResultConsumer) ProtoReflect() protoreflect.Message { // Deprecated: Use ResultConsumer.ProtoReflect.Descriptor instead. func (*ResultConsumer) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{11} + return file_manager_manager_proto_rawDescGZIP(), []int{12} } func (x *ResultConsumer) GetUserKey() []byte { @@ -867,7 +936,7 @@ type Dataset struct { func (x *Dataset) Reset() { *x = Dataset{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[12] + mi := &file_manager_manager_proto_msgTypes[13] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -880,7 +949,7 @@ func (x *Dataset) String() string { func (*Dataset) ProtoMessage() {} func (x *Dataset) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[12] + mi := &file_manager_manager_proto_msgTypes[13] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -893,7 +962,7 @@ func (x *Dataset) ProtoReflect() protoreflect.Message { // Deprecated: Use Dataset.ProtoReflect.Descriptor instead. func (*Dataset) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{12} + return file_manager_manager_proto_rawDescGZIP(), []int{13} } func (x *Dataset) GetHash() []byte { @@ -929,7 +998,7 @@ type Algorithm struct { func (x *Algorithm) Reset() { *x = Algorithm{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[13] + mi := &file_manager_manager_proto_msgTypes[14] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -942,7 +1011,7 @@ func (x *Algorithm) String() string { func (*Algorithm) ProtoMessage() {} func (x *Algorithm) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[13] + mi := &file_manager_manager_proto_msgTypes[14] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -955,7 +1024,7 @@ func (x *Algorithm) ProtoReflect() protoreflect.Message { // Deprecated: Use Algorithm.ProtoReflect.Descriptor instead. func (*Algorithm) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{13} + return file_manager_manager_proto_rawDescGZIP(), []int{14} } func (x *Algorithm) GetHash() []byte { @@ -990,7 +1059,7 @@ type AgentConfig struct { func (x *AgentConfig) Reset() { *x = AgentConfig{} if protoimpl.UnsafeEnabled { - mi := &file_manager_manager_proto_msgTypes[14] + mi := &file_manager_manager_proto_msgTypes[15] ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) ms.StoreMessageInfo(mi) } @@ -1003,7 +1072,7 @@ func (x *AgentConfig) String() string { func (*AgentConfig) ProtoMessage() {} func (x *AgentConfig) ProtoReflect() protoreflect.Message { - mi := &file_manager_manager_proto_msgTypes[14] + mi := &file_manager_manager_proto_msgTypes[15] if protoimpl.UnsafeEnabled && x != nil { ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) if ms.LoadMessageInfo() == nil { @@ -1016,7 +1085,7 @@ func (x *AgentConfig) ProtoReflect() protoreflect.Message { // Deprecated: Use AgentConfig.ProtoReflect.Descriptor instead. func (*AgentConfig) Descriptor() ([]byte, []int) { - return file_manager_manager_proto_rawDescGZIP(), []int{14} + return file_manager_manager_proto_rawDescGZIP(), []int{15} } func (x *AgentConfig) GetPort() string { @@ -1088,54 +1157,65 @@ var file_manager_manager_proto_rawDesc = []byte{ 0x43, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x12, 0x25, 0x0a, 0x0e, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, - 0x49, 0x64, 0x22, 0x53, 0x0a, 0x0b, 0x52, 0x75, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, - 0x65, 0x12, 0x1d, 0x0a, 0x0a, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x70, 0x6f, 0x72, 0x74, 0x18, - 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x09, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x50, 0x6f, 0x72, 0x74, - 0x12, 0x25, 0x0a, 0x0e, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, - 0x69, 0x64, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, - 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x49, 0x64, 0x22, 0x31, 0x0a, 0x0b, 0x42, 0x61, 0x63, 0x6b, 0x65, - 0x6e, 0x64, 0x49, 0x6e, 0x66, 0x6f, 0x12, 0x12, 0x0a, 0x04, 0x69, 0x6e, 0x66, 0x6f, 0x18, 0x01, - 0x20, 0x01, 0x28, 0x0c, 0x52, 0x04, 0x69, 0x6e, 0x66, 0x6f, 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, - 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x02, 0x69, 0x64, 0x22, 0xde, 0x01, 0x0a, 0x0a, 0x41, - 0x67, 0x65, 0x6e, 0x74, 0x45, 0x76, 0x65, 0x6e, 0x74, 0x12, 0x1d, 0x0a, 0x0a, 0x65, 0x76, 0x65, - 0x6e, 0x74, 0x5f, 0x74, 0x79, 0x70, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x09, 0x65, - 0x76, 0x65, 0x6e, 0x74, 0x54, 0x79, 0x70, 0x65, 0x12, 0x38, 0x0a, 0x09, 0x74, 0x69, 0x6d, 0x65, - 0x73, 0x74, 0x61, 0x6d, 0x70, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1a, 0x2e, 0x67, 0x6f, - 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x54, 0x69, - 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x52, 0x09, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, - 0x6d, 0x70, 0x12, 0x25, 0x0a, 0x0e, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, - 0x6e, 0x5f, 0x69, 0x64, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, - 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x49, 0x64, 0x12, 0x18, 0x0a, 0x07, 0x64, 0x65, 0x74, - 0x61, 0x69, 0x6c, 0x73, 0x18, 0x04, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x07, 0x64, 0x65, 0x74, 0x61, - 0x69, 0x6c, 0x73, 0x12, 0x1e, 0x0a, 0x0a, 0x6f, 0x72, 0x69, 0x67, 0x69, 0x6e, 0x61, 0x74, 0x6f, - 0x72, 0x18, 0x05, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0a, 0x6f, 0x72, 0x69, 0x67, 0x69, 0x6e, 0x61, - 0x74, 0x6f, 0x72, 0x12, 0x16, 0x0a, 0x06, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x18, 0x06, 0x20, - 0x01, 0x28, 0x09, 0x52, 0x06, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x22, 0x9b, 0x01, 0x0a, 0x08, - 0x41, 0x67, 0x65, 0x6e, 0x74, 0x4c, 0x6f, 0x67, 0x12, 0x18, 0x0a, 0x07, 0x6d, 0x65, 0x73, 0x73, - 0x61, 0x67, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, - 0x67, 0x65, 0x12, 0x25, 0x0a, 0x0e, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, - 0x6e, 0x5f, 0x69, 0x64, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, - 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x49, 0x64, 0x12, 0x14, 0x0a, 0x05, 0x6c, 0x65, 0x76, - 0x65, 0x6c, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x05, 0x6c, 0x65, 0x76, 0x65, 0x6c, 0x12, - 0x38, 0x0a, 0x09, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x18, 0x04, 0x20, 0x01, - 0x28, 0x0b, 0x32, 0x1a, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, - 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x54, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x52, 0x09, - 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x22, 0xf5, 0x01, 0x0a, 0x13, 0x43, 0x6c, - 0x69, 0x65, 0x6e, 0x74, 0x53, 0x74, 0x72, 0x65, 0x61, 0x6d, 0x4d, 0x65, 0x73, 0x73, 0x61, 0x67, - 0x65, 0x12, 0x30, 0x0a, 0x09, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x6c, 0x6f, 0x67, 0x18, 0x01, - 0x20, 0x01, 0x28, 0x0b, 0x32, 0x11, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, 0x41, - 0x67, 0x65, 0x6e, 0x74, 0x4c, 0x6f, 0x67, 0x48, 0x00, 0x52, 0x08, 0x61, 0x67, 0x65, 0x6e, 0x74, - 0x4c, 0x6f, 0x67, 0x12, 0x36, 0x0a, 0x0b, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x65, 0x76, 0x65, - 0x6e, 0x74, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x13, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, - 0x65, 0x72, 0x2e, 0x41, 0x67, 0x65, 0x6e, 0x74, 0x45, 0x76, 0x65, 0x6e, 0x74, 0x48, 0x00, 0x52, - 0x0a, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x45, 0x76, 0x65, 0x6e, 0x74, 0x12, 0x2f, 0x0a, 0x07, 0x72, - 0x75, 0x6e, 0x5f, 0x72, 0x65, 0x73, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x14, 0x2e, 0x6d, - 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, 0x52, 0x75, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, - 0x73, 0x65, 0x48, 0x00, 0x52, 0x06, 0x72, 0x75, 0x6e, 0x52, 0x65, 0x73, 0x12, 0x38, 0x0a, 0x0b, - 0x62, 0x61, 0x63, 0x6b, 0x65, 0x6e, 0x64, 0x49, 0x6e, 0x66, 0x6f, 0x18, 0x04, 0x20, 0x01, 0x28, - 0x0b, 0x32, 0x14, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, 0x42, 0x61, 0x63, 0x6b, - 0x65, 0x6e, 0x64, 0x49, 0x6e, 0x66, 0x6f, 0x48, 0x00, 0x52, 0x0b, 0x62, 0x61, 0x63, 0x6b, 0x65, - 0x6e, 0x64, 0x49, 0x6e, 0x66, 0x6f, 0x42, 0x09, 0x0a, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, + 0x49, 0x64, 0x22, 0x5a, 0x0a, 0x17, 0x53, 0x74, 0x6f, 0x70, 0x43, 0x6f, 0x6d, 0x70, 0x75, 0x74, + 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x25, 0x0a, + 0x0e, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, 0x69, 0x64, 0x18, + 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, + 0x6f, 0x6e, 0x49, 0x64, 0x12, 0x18, 0x0a, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x18, + 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x22, 0x53, + 0x0a, 0x0b, 0x52, 0x75, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x1d, 0x0a, + 0x0a, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x70, 0x6f, 0x72, 0x74, 0x18, 0x01, 0x20, 0x01, 0x28, + 0x09, 0x52, 0x09, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x50, 0x6f, 0x72, 0x74, 0x12, 0x25, 0x0a, 0x0e, + 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, 0x69, 0x64, 0x18, 0x02, + 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, + 0x6e, 0x49, 0x64, 0x22, 0x31, 0x0a, 0x0b, 0x42, 0x61, 0x63, 0x6b, 0x65, 0x6e, 0x64, 0x49, 0x6e, + 0x66, 0x6f, 0x12, 0x12, 0x0a, 0x04, 0x69, 0x6e, 0x66, 0x6f, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, + 0x52, 0x04, 0x69, 0x6e, 0x66, 0x6f, 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x02, 0x20, 0x01, + 0x28, 0x09, 0x52, 0x02, 0x69, 0x64, 0x22, 0xde, 0x01, 0x0a, 0x0a, 0x41, 0x67, 0x65, 0x6e, 0x74, + 0x45, 0x76, 0x65, 0x6e, 0x74, 0x12, 0x1d, 0x0a, 0x0a, 0x65, 0x76, 0x65, 0x6e, 0x74, 0x5f, 0x74, + 0x79, 0x70, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x09, 0x65, 0x76, 0x65, 0x6e, 0x74, + 0x54, 0x79, 0x70, 0x65, 0x12, 0x38, 0x0a, 0x09, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, + 0x70, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1a, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, + 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x54, 0x69, 0x6d, 0x65, 0x73, 0x74, + 0x61, 0x6d, 0x70, 0x52, 0x09, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x12, 0x25, + 0x0a, 0x0e, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, 0x69, 0x64, + 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, + 0x69, 0x6f, 0x6e, 0x49, 0x64, 0x12, 0x18, 0x0a, 0x07, 0x64, 0x65, 0x74, 0x61, 0x69, 0x6c, 0x73, + 0x18, 0x04, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x07, 0x64, 0x65, 0x74, 0x61, 0x69, 0x6c, 0x73, 0x12, + 0x1e, 0x0a, 0x0a, 0x6f, 0x72, 0x69, 0x67, 0x69, 0x6e, 0x61, 0x74, 0x6f, 0x72, 0x18, 0x05, 0x20, + 0x01, 0x28, 0x09, 0x52, 0x0a, 0x6f, 0x72, 0x69, 0x67, 0x69, 0x6e, 0x61, 0x74, 0x6f, 0x72, 0x12, + 0x16, 0x0a, 0x06, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x18, 0x06, 0x20, 0x01, 0x28, 0x09, 0x52, + 0x06, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x22, 0x9b, 0x01, 0x0a, 0x08, 0x41, 0x67, 0x65, 0x6e, + 0x74, 0x4c, 0x6f, 0x67, 0x12, 0x18, 0x0a, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x18, + 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x12, 0x25, + 0x0a, 0x0e, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, 0x69, 0x64, + 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, + 0x69, 0x6f, 0x6e, 0x49, 0x64, 0x12, 0x14, 0x0a, 0x05, 0x6c, 0x65, 0x76, 0x65, 0x6c, 0x18, 0x03, + 0x20, 0x01, 0x28, 0x09, 0x52, 0x05, 0x6c, 0x65, 0x76, 0x65, 0x6c, 0x12, 0x38, 0x0a, 0x09, 0x74, + 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x18, 0x04, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1a, + 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, + 0x2e, 0x54, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x52, 0x09, 0x74, 0x69, 0x6d, 0x65, + 0x73, 0x74, 0x61, 0x6d, 0x70, 0x22, 0xc9, 0x02, 0x0a, 0x13, 0x43, 0x6c, 0x69, 0x65, 0x6e, 0x74, + 0x53, 0x74, 0x72, 0x65, 0x61, 0x6d, 0x4d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x12, 0x30, 0x0a, + 0x09, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x6c, 0x6f, 0x67, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, + 0x32, 0x11, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, 0x41, 0x67, 0x65, 0x6e, 0x74, + 0x4c, 0x6f, 0x67, 0x48, 0x00, 0x52, 0x08, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x4c, 0x6f, 0x67, 0x12, + 0x36, 0x0a, 0x0b, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x65, 0x76, 0x65, 0x6e, 0x74, 0x18, 0x02, + 0x20, 0x01, 0x28, 0x0b, 0x32, 0x13, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, 0x41, + 0x67, 0x65, 0x6e, 0x74, 0x45, 0x76, 0x65, 0x6e, 0x74, 0x48, 0x00, 0x52, 0x0a, 0x61, 0x67, 0x65, + 0x6e, 0x74, 0x45, 0x76, 0x65, 0x6e, 0x74, 0x12, 0x2f, 0x0a, 0x07, 0x72, 0x75, 0x6e, 0x5f, 0x72, + 0x65, 0x73, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x14, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, + 0x65, 0x72, 0x2e, 0x52, 0x75, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x48, 0x00, + 0x52, 0x06, 0x72, 0x75, 0x6e, 0x52, 0x65, 0x73, 0x12, 0x38, 0x0a, 0x0b, 0x62, 0x61, 0x63, 0x6b, + 0x65, 0x6e, 0x64, 0x49, 0x6e, 0x66, 0x6f, 0x18, 0x04, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x14, 0x2e, + 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, 0x42, 0x61, 0x63, 0x6b, 0x65, 0x6e, 0x64, 0x49, + 0x6e, 0x66, 0x6f, 0x48, 0x00, 0x52, 0x0b, 0x62, 0x61, 0x63, 0x6b, 0x65, 0x6e, 0x64, 0x49, 0x6e, + 0x66, 0x6f, 0x12, 0x52, 0x0a, 0x12, 0x73, 0x74, 0x6f, 0x70, 0x43, 0x6f, 0x6d, 0x70, 0x75, 0x74, + 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x73, 0x18, 0x05, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x20, + 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, 0x53, 0x74, 0x6f, 0x70, 0x43, 0x6f, 0x6d, + 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, + 0x48, 0x00, 0x52, 0x12, 0x73, 0x74, 0x6f, 0x70, 0x43, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, + 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x73, 0x42, 0x09, 0x0a, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x22, 0xd6, 0x02, 0x0a, 0x13, 0x53, 0x65, 0x72, 0x76, 0x65, 0x72, 0x53, 0x74, 0x72, 0x65, 0x61, 0x6d, 0x4d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x12, 0x3b, 0x0a, 0x0c, 0x72, 0x75, 0x6e, 0x52, 0x65, 0x71, 0x43, 0x68, 0x75, 0x6e, 0x6b, 0x73, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, @@ -1231,48 +1311,50 @@ func file_manager_manager_proto_rawDescGZIP() []byte { return file_manager_manager_proto_rawDescData } -var file_manager_manager_proto_msgTypes = make([]protoimpl.MessageInfo, 15) +var file_manager_manager_proto_msgTypes = make([]protoimpl.MessageInfo, 16) var file_manager_manager_proto_goTypes = []any{ - (*Terminate)(nil), // 0: manager.Terminate - (*StopComputation)(nil), // 1: manager.StopComputation - (*RunResponse)(nil), // 2: manager.RunResponse - (*BackendInfo)(nil), // 3: manager.BackendInfo - (*AgentEvent)(nil), // 4: manager.AgentEvent - (*AgentLog)(nil), // 5: manager.AgentLog - (*ClientStreamMessage)(nil), // 6: manager.ClientStreamMessage - (*ServerStreamMessage)(nil), // 7: manager.ServerStreamMessage - (*RunReqChunks)(nil), // 8: manager.RunReqChunks - (*ComputationRunReq)(nil), // 9: manager.ComputationRunReq - (*BackendInfoReq)(nil), // 10: manager.BackendInfoReq - (*ResultConsumer)(nil), // 11: manager.ResultConsumer - (*Dataset)(nil), // 12: manager.Dataset - (*Algorithm)(nil), // 13: manager.Algorithm - (*AgentConfig)(nil), // 14: manager.AgentConfig - (*timestamppb.Timestamp)(nil), // 15: google.protobuf.Timestamp + (*Terminate)(nil), // 0: manager.Terminate + (*StopComputation)(nil), // 1: manager.StopComputation + (*StopComputationResponse)(nil), // 2: manager.StopComputationResponse + (*RunResponse)(nil), // 3: manager.RunResponse + (*BackendInfo)(nil), // 4: manager.BackendInfo + (*AgentEvent)(nil), // 5: manager.AgentEvent + (*AgentLog)(nil), // 6: manager.AgentLog + (*ClientStreamMessage)(nil), // 7: manager.ClientStreamMessage + (*ServerStreamMessage)(nil), // 8: manager.ServerStreamMessage + (*RunReqChunks)(nil), // 9: manager.RunReqChunks + (*ComputationRunReq)(nil), // 10: manager.ComputationRunReq + (*BackendInfoReq)(nil), // 11: manager.BackendInfoReq + (*ResultConsumer)(nil), // 12: manager.ResultConsumer + (*Dataset)(nil), // 13: manager.Dataset + (*Algorithm)(nil), // 14: manager.Algorithm + (*AgentConfig)(nil), // 15: manager.AgentConfig + (*timestamppb.Timestamp)(nil), // 16: google.protobuf.Timestamp } var file_manager_manager_proto_depIdxs = []int32{ - 15, // 0: manager.AgentEvent.timestamp:type_name -> google.protobuf.Timestamp - 15, // 1: manager.AgentLog.timestamp:type_name -> google.protobuf.Timestamp - 5, // 2: manager.ClientStreamMessage.agent_log:type_name -> manager.AgentLog - 4, // 3: manager.ClientStreamMessage.agent_event:type_name -> manager.AgentEvent - 2, // 4: manager.ClientStreamMessage.run_res:type_name -> manager.RunResponse - 3, // 5: manager.ClientStreamMessage.backendInfo:type_name -> manager.BackendInfo - 8, // 6: manager.ServerStreamMessage.runReqChunks:type_name -> manager.RunReqChunks - 9, // 7: manager.ServerStreamMessage.runReq:type_name -> manager.ComputationRunReq - 0, // 8: manager.ServerStreamMessage.terminateReq:type_name -> manager.Terminate - 1, // 9: manager.ServerStreamMessage.stopComputation:type_name -> manager.StopComputation - 10, // 10: manager.ServerStreamMessage.backendInfoReq:type_name -> manager.BackendInfoReq - 12, // 11: manager.ComputationRunReq.datasets:type_name -> manager.Dataset - 13, // 12: manager.ComputationRunReq.algorithm:type_name -> manager.Algorithm - 11, // 13: manager.ComputationRunReq.result_consumers:type_name -> manager.ResultConsumer - 14, // 14: manager.ComputationRunReq.agent_config:type_name -> manager.AgentConfig - 6, // 15: manager.ManagerService.Process:input_type -> manager.ClientStreamMessage - 7, // 16: manager.ManagerService.Process:output_type -> manager.ServerStreamMessage - 16, // [16:17] is the sub-list for method output_type - 15, // [15:16] is the sub-list for method input_type - 15, // [15:15] is the sub-list for extension type_name - 15, // [15:15] is the sub-list for extension extendee - 0, // [0:15] is the sub-list for field type_name + 16, // 0: manager.AgentEvent.timestamp:type_name -> google.protobuf.Timestamp + 16, // 1: manager.AgentLog.timestamp:type_name -> google.protobuf.Timestamp + 6, // 2: manager.ClientStreamMessage.agent_log:type_name -> manager.AgentLog + 5, // 3: manager.ClientStreamMessage.agent_event:type_name -> manager.AgentEvent + 3, // 4: manager.ClientStreamMessage.run_res:type_name -> manager.RunResponse + 4, // 5: manager.ClientStreamMessage.backendInfo:type_name -> manager.BackendInfo + 2, // 6: manager.ClientStreamMessage.stopComputationRes:type_name -> manager.StopComputationResponse + 9, // 7: manager.ServerStreamMessage.runReqChunks:type_name -> manager.RunReqChunks + 10, // 8: manager.ServerStreamMessage.runReq:type_name -> manager.ComputationRunReq + 0, // 9: manager.ServerStreamMessage.terminateReq:type_name -> manager.Terminate + 1, // 10: manager.ServerStreamMessage.stopComputation:type_name -> manager.StopComputation + 11, // 11: manager.ServerStreamMessage.backendInfoReq:type_name -> manager.BackendInfoReq + 13, // 12: manager.ComputationRunReq.datasets:type_name -> manager.Dataset + 14, // 13: manager.ComputationRunReq.algorithm:type_name -> manager.Algorithm + 12, // 14: manager.ComputationRunReq.result_consumers:type_name -> manager.ResultConsumer + 15, // 15: manager.ComputationRunReq.agent_config:type_name -> manager.AgentConfig + 7, // 16: manager.ManagerService.Process:input_type -> manager.ClientStreamMessage + 8, // 17: manager.ManagerService.Process:output_type -> manager.ServerStreamMessage + 17, // [17:18] is the sub-list for method output_type + 16, // [16:17] is the sub-list for method input_type + 16, // [16:16] is the sub-list for extension type_name + 16, // [16:16] is the sub-list for extension extendee + 0, // [0:16] is the sub-list for field type_name } func init() { file_manager_manager_proto_init() } @@ -1306,7 +1388,7 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[2].Exporter = func(v any, i int) any { - switch v := v.(*RunResponse); i { + switch v := v.(*StopComputationResponse); i { case 0: return &v.state case 1: @@ -1318,7 +1400,7 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[3].Exporter = func(v any, i int) any { - switch v := v.(*BackendInfo); i { + switch v := v.(*RunResponse); i { case 0: return &v.state case 1: @@ -1330,7 +1412,7 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[4].Exporter = func(v any, i int) any { - switch v := v.(*AgentEvent); i { + switch v := v.(*BackendInfo); i { case 0: return &v.state case 1: @@ -1342,7 +1424,7 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[5].Exporter = func(v any, i int) any { - switch v := v.(*AgentLog); i { + switch v := v.(*AgentEvent); i { case 0: return &v.state case 1: @@ -1354,7 +1436,7 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[6].Exporter = func(v any, i int) any { - switch v := v.(*ClientStreamMessage); i { + switch v := v.(*AgentLog); i { case 0: return &v.state case 1: @@ -1366,7 +1448,7 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[7].Exporter = func(v any, i int) any { - switch v := v.(*ServerStreamMessage); i { + switch v := v.(*ClientStreamMessage); i { case 0: return &v.state case 1: @@ -1378,7 +1460,7 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[8].Exporter = func(v any, i int) any { - switch v := v.(*RunReqChunks); i { + switch v := v.(*ServerStreamMessage); i { case 0: return &v.state case 1: @@ -1390,7 +1472,7 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[9].Exporter = func(v any, i int) any { - switch v := v.(*ComputationRunReq); i { + switch v := v.(*RunReqChunks); i { case 0: return &v.state case 1: @@ -1402,7 +1484,7 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[10].Exporter = func(v any, i int) any { - switch v := v.(*BackendInfoReq); i { + switch v := v.(*ComputationRunReq); i { case 0: return &v.state case 1: @@ -1414,7 +1496,7 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[11].Exporter = func(v any, i int) any { - switch v := v.(*ResultConsumer); i { + switch v := v.(*BackendInfoReq); i { case 0: return &v.state case 1: @@ -1426,7 +1508,7 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[12].Exporter = func(v any, i int) any { - switch v := v.(*Dataset); i { + switch v := v.(*ResultConsumer); i { case 0: return &v.state case 1: @@ -1438,7 +1520,7 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[13].Exporter = func(v any, i int) any { - switch v := v.(*Algorithm); i { + switch v := v.(*Dataset); i { case 0: return &v.state case 1: @@ -1450,6 +1532,18 @@ func file_manager_manager_proto_init() { } } file_manager_manager_proto_msgTypes[14].Exporter = func(v any, i int) any { + switch v := v.(*Algorithm); i { + case 0: + return &v.state + case 1: + return &v.sizeCache + case 2: + return &v.unknownFields + default: + return nil + } + } + file_manager_manager_proto_msgTypes[15].Exporter = func(v any, i int) any { switch v := v.(*AgentConfig); i { case 0: return &v.state @@ -1462,13 +1556,14 @@ func file_manager_manager_proto_init() { } } } - file_manager_manager_proto_msgTypes[6].OneofWrappers = []any{ + file_manager_manager_proto_msgTypes[7].OneofWrappers = []any{ (*ClientStreamMessage_AgentLog)(nil), (*ClientStreamMessage_AgentEvent)(nil), (*ClientStreamMessage_RunRes)(nil), (*ClientStreamMessage_BackendInfo)(nil), + (*ClientStreamMessage_StopComputationRes)(nil), } - file_manager_manager_proto_msgTypes[7].OneofWrappers = []any{ + file_manager_manager_proto_msgTypes[8].OneofWrappers = []any{ (*ServerStreamMessage_RunReqChunks)(nil), (*ServerStreamMessage_RunReq)(nil), (*ServerStreamMessage_TerminateReq)(nil), @@ -1481,7 +1576,7 @@ func file_manager_manager_proto_init() { GoPackagePath: reflect.TypeOf(x{}).PkgPath(), RawDescriptor: file_manager_manager_proto_rawDesc, NumEnums: 0, - NumMessages: 15, + NumMessages: 16, NumExtensions: 0, NumServices: 1, },