mirror of https://github.com/dragonflyoss/api.git
feat: manager add model and model version api (#6)
Signed-off-by: Gaius <gaius.qi@gmail.com>
This commit is contained in:
Gaius
GitHub
parent
02137e70b5
commit
916c66390c
4
go.sum
4
go.sum
|
|
@ -22,7 +22,6 @@ github.com/envoyproxy/protoc-gen-validate v0.6.7 h1:qcZcULcd/abmQg6dwigimCNEyi4g
|
|||
github.com/envoyproxy/protoc-gen-validate v0.6.7/go.mod h1:dyJXwwfPK2VSqiB9Klm1J6romD608Ba7Hij42vrOBCo=
|
||||
github.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04=
|
||||
github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=
|
||||
github.com/golang/mock v1.1.1 h1:G5FRp8JnTd7RQH5kemVNlMeyXQAztQ3mOWV95KxsXH8=
|
||||
github.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=
|
||||
github.com/golang/mock v1.6.0 h1:ErTB+efbowRARo13NNdxyJji2egdxLGQhRaY+DUumQc=
|
||||
github.com/golang/mock v1.6.0/go.mod h1:p6yTPP+5HYm5mzsMV8JkE6ZKdX+/wYM6Hr+LicevLPs=
|
||||
|
|
@ -89,7 +88,6 @@ golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLL
|
|||
golang.org/x/net v0.0.0-20200822124328-c89045814202/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=
|
||||
golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
|
||||
golang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM=
|
||||
golang.org/x/net v0.0.0-20210813160813-60bc85c4be6d h1:LO7XpTYMwTqxjLcGWPijK3vRXg1aWdlNOVOHRq45d7c=
|
||||
golang.org/x/net v0.0.0-20210813160813-60bc85c4be6d/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
|
||||
golang.org/x/net v0.0.0-20220728211354-c7608f3a8462 h1:UreQrH7DbFXSi9ZFox6FNT3WBooWmdANpU+IfkT1T4I=
|
||||
golang.org/x/net v0.0.0-20220728211354-c7608f3a8462/go.mod h1:YDH+HFinaLZZlnHAfSS6ZXJJ9M9t4Dl22yv3iI2vPwk=
|
||||
|
|
@ -110,7 +108,6 @@ golang.org/x/sys v0.0.0-20210119212857-b64e53b001e4/go.mod h1:h1NjWce9XRLGQEsW7w
|
|||
golang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20210510120138-977fb7262007/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20210816183151-1e6c022a8912 h1:uCLL3g5wH2xjxVREVuAbP9JM5PPKjRbXKRa6IBjkzmU=
|
||||
golang.org/x/sys v0.0.0-20210816183151-1e6c022a8912/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20220728004956-3c1f35247d10 h1:WIoqL4EROvwiPdUtaip4VcDdpZ4kha7wBWZrbVKCIZg=
|
||||
golang.org/x/sys v0.0.0-20220728004956-3c1f35247d10/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
|
|
@ -139,7 +136,6 @@ google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7
|
|||
google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=
|
||||
google.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc=
|
||||
google.golang.org/genproto v0.0.0-20200513103714-09dca8ec2884/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c=
|
||||
google.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013 h1:+kGHl1aib/qcwaRi1CbqBZ1rk19r85MNUf8HaBghugY=
|
||||
google.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo=
|
||||
google.golang.org/genproto v0.0.0-20220728213248-dd149ef739b9 h1:d3fKQZK+1rWQMg3xLKQbPMirUCo29I/NRdI2WarSzTg=
|
||||
google.golang.org/genproto v0.0.0-20220728213248-dd149ef739b9/go.mod h1:iHe1svFLAZg9VWz891+QbRMwUv9O/1Ww+/mngYeThbc=
|
||||
|
|
|
|||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
|
|
@ -20,6 +20,7 @@ package manager;
|
|||
|
||||
import "google/protobuf/empty.proto";
|
||||
import "validate/validate.proto";
|
||||
import "google/protobuf/timestamp.proto";
|
||||
|
||||
option go_package = "d7y.io/api/pkg/apis/manager/v1";
|
||||
|
||||
|
|
@ -281,6 +282,180 @@ message ListBucketsResponse {
|
|||
repeated Bucket buckets = 1;
|
||||
}
|
||||
|
||||
// Model represents information of model.
|
||||
message Model {
|
||||
// Model id.
|
||||
string id = 1 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model name.
|
||||
string name = 2 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model version id.
|
||||
string version_id = 3 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Scheduler id.
|
||||
string scheduler_id = 4 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Scheduler hostname.
|
||||
string host_name = 5 [(validate.rules).string.hostname = true];
|
||||
// Scheduler ip.
|
||||
string ip = 6 [(validate.rules).string.ip = true];
|
||||
// Model create time.
|
||||
google.protobuf.Timestamp create_at = 7 [(validate.rules).timestamp.required = true];
|
||||
// Model update time.
|
||||
google.protobuf.Timestamp update_at = 8 [(validate.rules).timestamp.required = true];
|
||||
}
|
||||
|
||||
// ListModelsRequest represents request of ListModels.
|
||||
message ListModelsRequest {
|
||||
// Scheduler id.
|
||||
string scheduler_id = 1 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
}
|
||||
|
||||
// ListModelsResponse represents response of ListModels.
|
||||
message ListModelsResponse {
|
||||
// Model informations.
|
||||
repeated Model models = 1;
|
||||
}
|
||||
|
||||
// GetModelRequest represents request of GetModel.
|
||||
message GetModelRequest {
|
||||
// Scheduler id.
|
||||
string scheduler_id = 1 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model id.
|
||||
string model_id = 2 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
}
|
||||
|
||||
// CreateModelRequest represents request of CreateModel.
|
||||
message CreateModelRequest {
|
||||
// Model id.
|
||||
string model_id = 1 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model name.
|
||||
string name = 2 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model version id.
|
||||
string version_id = 3 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Scheduler id.
|
||||
string scheduler_id = 4 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Scheduler hostname.
|
||||
string host_name = 5 [(validate.rules).string.hostname = true];
|
||||
// Scheduler ip.
|
||||
string ip = 6 [(validate.rules).string.ip = true];
|
||||
}
|
||||
|
||||
// UpdateModelRequest represents request of UpdateModel.
|
||||
message UpdateModelRequest {
|
||||
// Model id.
|
||||
string model_id = 1 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model name.
|
||||
string name = 2 [(validate.rules).string = {min_len: 1, max_len: 1024, ignore_empty: true}];
|
||||
// Model version id.
|
||||
string version_id = 3 [(validate.rules).string = {min_len: 1, max_len: 1024, ignore_empty: true}];
|
||||
// Scheduler id.
|
||||
string scheduler_id = 4 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Scheduler hostname.
|
||||
string host_name = 5 [(validate.rules).string = {hostname: true, ignore_empty: true}];
|
||||
// Scheduler ip.
|
||||
string ip = 6 [(validate.rules).string = {ip: true, ignore_empty: true}];
|
||||
}
|
||||
|
||||
// DeleteModelRequest represents request of DeleteModel.
|
||||
message DeleteModelRequest {
|
||||
// Scheduler id.
|
||||
string scheduler_id = 1 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model id.
|
||||
string model_id = 2 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
}
|
||||
|
||||
// ModelVersion represents information of model version.
|
||||
message ModelVersion {
|
||||
// Model version id.
|
||||
string id = 1 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model version name.
|
||||
bytes date = 2 [(validate.rules).bytes.min_len = 1];
|
||||
// Model version mae.
|
||||
double mae = 3 [(validate.rules).double.gte = 0];
|
||||
// Model version mse.
|
||||
double mse = 4 [(validate.rules).double.gte = 0];
|
||||
// Model version rmse.
|
||||
double rmse = 5 [(validate.rules).double.gte = 0];
|
||||
// Model version r^2.
|
||||
double r2 = 6 [(validate.rules).double.gte = 0];
|
||||
// Model create time.
|
||||
google.protobuf.Timestamp create_at = 7 [(validate.rules).timestamp.required = true];
|
||||
// Model update time.
|
||||
google.protobuf.Timestamp update_at = 8 [(validate.rules).timestamp.required = true];
|
||||
}
|
||||
|
||||
// ListModelVersionsRequest represents request of ListModelVersions.
|
||||
message ListModelVersionsRequest {
|
||||
// Scheduler id.
|
||||
string scheduler_id = 1 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model id.
|
||||
string model_id = 2 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
}
|
||||
|
||||
// ListModelVersionsResponse represents response of ListModelVersions.
|
||||
message ListModelVersionsResponse {
|
||||
// Model version informations.
|
||||
repeated ModelVersion model_versions = 1;
|
||||
}
|
||||
|
||||
// GetModelVersionRequest represents request of GetModelVersion.
|
||||
message GetModelVersionRequest {
|
||||
// Scheduler id.
|
||||
string scheduler_id = 1 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model id.
|
||||
string model_id = 2 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model version id.
|
||||
string model_version_id = 3 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
}
|
||||
|
||||
// CreateModelVersionRequest represents request of CreateModelVersion.
|
||||
message CreateModelVersionRequest {
|
||||
// Model version id.
|
||||
string id = 1 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Scheduler id.
|
||||
string scheduler_id = 2 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model id.
|
||||
string model_id = 3 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model version name.
|
||||
bytes date = 4 [(validate.rules).bytes.min_len = 1];
|
||||
// Model version mae.
|
||||
double mae = 5 [(validate.rules).double.gte = 0];
|
||||
// Model version mse.
|
||||
double mse = 6 [(validate.rules).double.gte = 0];
|
||||
// Model version rmse.
|
||||
double rmse = 7 [(validate.rules).double.gte = 0];
|
||||
// Model version r^2.
|
||||
double r2 = 8 [(validate.rules).double.gte = 0];
|
||||
}
|
||||
|
||||
// UpdateModelVersionRequest represents request of UpdateModelVersion.
|
||||
message UpdateModelVersionRequest {
|
||||
// Model version id.
|
||||
string id = 1 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Scheduler id.
|
||||
string scheduler_id = 2 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model id.
|
||||
string model_id = 3 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model version name.
|
||||
bytes date = 4 [(validate.rules).bytes = {min_len: 1, ignore_empty: true}];
|
||||
// Model version mae.
|
||||
double mae = 5 [(validate.rules).double = {gte: 0, ignore_empty: true}];
|
||||
// Model version mse.
|
||||
double mse = 6 [(validate.rules).double = {gte: 0, ignore_empty: true}];
|
||||
// Model version rmse.
|
||||
double rmse = 7 [(validate.rules).double = {gte: 0, ignore_empty: true}];
|
||||
// Model version r^2.
|
||||
double r2 = 8 [(validate.rules).double = {gte: 0, ignore_empty: true}];
|
||||
}
|
||||
|
||||
// DeleteModelVersionRequest represents request of DeleteModelVersion.
|
||||
message DeleteModelVersionRequest {
|
||||
// Scheduler id.
|
||||
string scheduler_id = 1 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model id.
|
||||
string model_id = 2 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
// Model version id.
|
||||
string model_version_id = 3 [(validate.rules).string = {min_len: 1, max_len: 1024}];
|
||||
}
|
||||
|
||||
// KeepAliveRequest represents request of KeepAlive.
|
||||
message KeepAliveRequest {
|
||||
// Request source type.
|
||||
|
|
@ -316,6 +491,36 @@ service Manager {
|
|||
// List buckets configuration.
|
||||
rpc ListBuckets(ListBucketsRequest)returns(ListBucketsResponse);
|
||||
|
||||
// List models information.
|
||||
rpc ListModels(ListModelsRequest)returns(ListModelsResponse);
|
||||
|
||||
// Get model information.
|
||||
rpc GetModel(GetModelRequest)returns(Model);
|
||||
|
||||
// Create model information.
|
||||
rpc CreateModel(CreateModelRequest)returns(Model);
|
||||
|
||||
// Update model information.
|
||||
rpc UpdateModel(UpdateModelRequest)returns(Model);
|
||||
|
||||
// Delete model information.
|
||||
rpc DeleteModel(DeleteModelRequest)returns(google.protobuf.Empty);
|
||||
|
||||
// List model versions information.
|
||||
rpc ListModelVersions(ListModelVersionsRequest)returns(ListModelVersionsResponse);
|
||||
|
||||
// Get model version information.
|
||||
rpc GetModelVersion(GetModelVersionRequest)returns(ModelVersion);
|
||||
|
||||
// Create model version information.
|
||||
rpc CreateModelVersion(CreateModelVersionRequest)returns(ModelVersion);
|
||||
|
||||
// Update model version information.
|
||||
rpc UpdateModelVersion(UpdateModelVersionRequest)returns(ModelVersion);
|
||||
|
||||
// Delete model version information.
|
||||
rpc DeleteModelVersion(DeleteModelVersionRequest)returns(google.protobuf.Empty);
|
||||
|
||||
// KeepAlive with manager.
|
||||
rpc KeepAlive(stream KeepAliveRequest)returns(google.protobuf.Empty);
|
||||
}
|
||||
|
|
|
|||
|
|
@ -38,6 +38,126 @@ func (m *MockManagerClient) EXPECT() *MockManagerClientMockRecorder {
|
|||
return m.recorder
|
||||
}
|
||||
|
||||
// CreateModel mocks base method.
|
||||
func (m *MockManagerClient) CreateModel(ctx context.Context, in *v1.CreateModelRequest, opts ...grpc.CallOption) (*v1.Model, error) {
|
||||
m.ctrl.T.Helper()
|
||||
varargs := []interface{}{ctx, in}
|
||||
for _, a := range opts {
|
||||
varargs = append(varargs, a)
|
||||
}
|
||||
ret := m.ctrl.Call(m, "CreateModel", varargs...)
|
||||
ret0, _ := ret[0].(*v1.Model)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// CreateModel indicates an expected call of CreateModel.
|
||||
func (mr *MockManagerClientMockRecorder) CreateModel(ctx, in interface{}, opts ...interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
varargs := append([]interface{}{ctx, in}, opts...)
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateModel", reflect.TypeOf((*MockManagerClient)(nil).CreateModel), varargs...)
|
||||
}
|
||||
|
||||
// CreateModelVersion mocks base method.
|
||||
func (m *MockManagerClient) CreateModelVersion(ctx context.Context, in *v1.CreateModelVersionRequest, opts ...grpc.CallOption) (*v1.ModelVersion, error) {
|
||||
m.ctrl.T.Helper()
|
||||
varargs := []interface{}{ctx, in}
|
||||
for _, a := range opts {
|
||||
varargs = append(varargs, a)
|
||||
}
|
||||
ret := m.ctrl.Call(m, "CreateModelVersion", varargs...)
|
||||
ret0, _ := ret[0].(*v1.ModelVersion)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// CreateModelVersion indicates an expected call of CreateModelVersion.
|
||||
func (mr *MockManagerClientMockRecorder) CreateModelVersion(ctx, in interface{}, opts ...interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
varargs := append([]interface{}{ctx, in}, opts...)
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateModelVersion", reflect.TypeOf((*MockManagerClient)(nil).CreateModelVersion), varargs...)
|
||||
}
|
||||
|
||||
// DeleteModel mocks base method.
|
||||
func (m *MockManagerClient) DeleteModel(ctx context.Context, in *v1.DeleteModelRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) {
|
||||
m.ctrl.T.Helper()
|
||||
varargs := []interface{}{ctx, in}
|
||||
for _, a := range opts {
|
||||
varargs = append(varargs, a)
|
||||
}
|
||||
ret := m.ctrl.Call(m, "DeleteModel", varargs...)
|
||||
ret0, _ := ret[0].(*emptypb.Empty)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// DeleteModel indicates an expected call of DeleteModel.
|
||||
func (mr *MockManagerClientMockRecorder) DeleteModel(ctx, in interface{}, opts ...interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
varargs := append([]interface{}{ctx, in}, opts...)
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteModel", reflect.TypeOf((*MockManagerClient)(nil).DeleteModel), varargs...)
|
||||
}
|
||||
|
||||
// DeleteModelVersion mocks base method.
|
||||
func (m *MockManagerClient) DeleteModelVersion(ctx context.Context, in *v1.DeleteModelVersionRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) {
|
||||
m.ctrl.T.Helper()
|
||||
varargs := []interface{}{ctx, in}
|
||||
for _, a := range opts {
|
||||
varargs = append(varargs, a)
|
||||
}
|
||||
ret := m.ctrl.Call(m, "DeleteModelVersion", varargs...)
|
||||
ret0, _ := ret[0].(*emptypb.Empty)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// DeleteModelVersion indicates an expected call of DeleteModelVersion.
|
||||
func (mr *MockManagerClientMockRecorder) DeleteModelVersion(ctx, in interface{}, opts ...interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
varargs := append([]interface{}{ctx, in}, opts...)
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteModelVersion", reflect.TypeOf((*MockManagerClient)(nil).DeleteModelVersion), varargs...)
|
||||
}
|
||||
|
||||
// GetModel mocks base method.
|
||||
func (m *MockManagerClient) GetModel(ctx context.Context, in *v1.GetModelRequest, opts ...grpc.CallOption) (*v1.Model, error) {
|
||||
m.ctrl.T.Helper()
|
||||
varargs := []interface{}{ctx, in}
|
||||
for _, a := range opts {
|
||||
varargs = append(varargs, a)
|
||||
}
|
||||
ret := m.ctrl.Call(m, "GetModel", varargs...)
|
||||
ret0, _ := ret[0].(*v1.Model)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// GetModel indicates an expected call of GetModel.
|
||||
func (mr *MockManagerClientMockRecorder) GetModel(ctx, in interface{}, opts ...interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
varargs := append([]interface{}{ctx, in}, opts...)
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetModel", reflect.TypeOf((*MockManagerClient)(nil).GetModel), varargs...)
|
||||
}
|
||||
|
||||
// GetModelVersion mocks base method.
|
||||
func (m *MockManagerClient) GetModelVersion(ctx context.Context, in *v1.GetModelVersionRequest, opts ...grpc.CallOption) (*v1.ModelVersion, error) {
|
||||
m.ctrl.T.Helper()
|
||||
varargs := []interface{}{ctx, in}
|
||||
for _, a := range opts {
|
||||
varargs = append(varargs, a)
|
||||
}
|
||||
ret := m.ctrl.Call(m, "GetModelVersion", varargs...)
|
||||
ret0, _ := ret[0].(*v1.ModelVersion)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// GetModelVersion indicates an expected call of GetModelVersion.
|
||||
func (mr *MockManagerClientMockRecorder) GetModelVersion(ctx, in interface{}, opts ...interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
varargs := append([]interface{}{ctx, in}, opts...)
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetModelVersion", reflect.TypeOf((*MockManagerClient)(nil).GetModelVersion), varargs...)
|
||||
}
|
||||
|
||||
// GetObjectStorage mocks base method.
|
||||
func (m *MockManagerClient) GetObjectStorage(ctx context.Context, in *v1.GetObjectStorageRequest, opts ...grpc.CallOption) (*v1.ObjectStorage, error) {
|
||||
m.ctrl.T.Helper()
|
||||
|
|
@ -138,6 +258,46 @@ func (mr *MockManagerClientMockRecorder) ListBuckets(ctx, in interface{}, opts .
|
|||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListBuckets", reflect.TypeOf((*MockManagerClient)(nil).ListBuckets), varargs...)
|
||||
}
|
||||
|
||||
// ListModelVersions mocks base method.
|
||||
func (m *MockManagerClient) ListModelVersions(ctx context.Context, in *v1.ListModelVersionsRequest, opts ...grpc.CallOption) (*v1.ListModelVersionsResponse, error) {
|
||||
m.ctrl.T.Helper()
|
||||
varargs := []interface{}{ctx, in}
|
||||
for _, a := range opts {
|
||||
varargs = append(varargs, a)
|
||||
}
|
||||
ret := m.ctrl.Call(m, "ListModelVersions", varargs...)
|
||||
ret0, _ := ret[0].(*v1.ListModelVersionsResponse)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// ListModelVersions indicates an expected call of ListModelVersions.
|
||||
func (mr *MockManagerClientMockRecorder) ListModelVersions(ctx, in interface{}, opts ...interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
varargs := append([]interface{}{ctx, in}, opts...)
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListModelVersions", reflect.TypeOf((*MockManagerClient)(nil).ListModelVersions), varargs...)
|
||||
}
|
||||
|
||||
// ListModels mocks base method.
|
||||
func (m *MockManagerClient) ListModels(ctx context.Context, in *v1.ListModelsRequest, opts ...grpc.CallOption) (*v1.ListModelsResponse, error) {
|
||||
m.ctrl.T.Helper()
|
||||
varargs := []interface{}{ctx, in}
|
||||
for _, a := range opts {
|
||||
varargs = append(varargs, a)
|
||||
}
|
||||
ret := m.ctrl.Call(m, "ListModels", varargs...)
|
||||
ret0, _ := ret[0].(*v1.ListModelsResponse)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// ListModels indicates an expected call of ListModels.
|
||||
func (mr *MockManagerClientMockRecorder) ListModels(ctx, in interface{}, opts ...interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
varargs := append([]interface{}{ctx, in}, opts...)
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListModels", reflect.TypeOf((*MockManagerClient)(nil).ListModels), varargs...)
|
||||
}
|
||||
|
||||
// ListSchedulers mocks base method.
|
||||
func (m *MockManagerClient) ListSchedulers(ctx context.Context, in *v1.ListSchedulersRequest, opts ...grpc.CallOption) (*v1.ListSchedulersResponse, error) {
|
||||
m.ctrl.T.Helper()
|
||||
|
|
@ -158,6 +318,46 @@ func (mr *MockManagerClientMockRecorder) ListSchedulers(ctx, in interface{}, opt
|
|||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListSchedulers", reflect.TypeOf((*MockManagerClient)(nil).ListSchedulers), varargs...)
|
||||
}
|
||||
|
||||
// UpdateModel mocks base method.
|
||||
func (m *MockManagerClient) UpdateModel(ctx context.Context, in *v1.UpdateModelRequest, opts ...grpc.CallOption) (*v1.Model, error) {
|
||||
m.ctrl.T.Helper()
|
||||
varargs := []interface{}{ctx, in}
|
||||
for _, a := range opts {
|
||||
varargs = append(varargs, a)
|
||||
}
|
||||
ret := m.ctrl.Call(m, "UpdateModel", varargs...)
|
||||
ret0, _ := ret[0].(*v1.Model)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// UpdateModel indicates an expected call of UpdateModel.
|
||||
func (mr *MockManagerClientMockRecorder) UpdateModel(ctx, in interface{}, opts ...interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
varargs := append([]interface{}{ctx, in}, opts...)
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateModel", reflect.TypeOf((*MockManagerClient)(nil).UpdateModel), varargs...)
|
||||
}
|
||||
|
||||
// UpdateModelVersion mocks base method.
|
||||
func (m *MockManagerClient) UpdateModelVersion(ctx context.Context, in *v1.UpdateModelVersionRequest, opts ...grpc.CallOption) (*v1.ModelVersion, error) {
|
||||
m.ctrl.T.Helper()
|
||||
varargs := []interface{}{ctx, in}
|
||||
for _, a := range opts {
|
||||
varargs = append(varargs, a)
|
||||
}
|
||||
ret := m.ctrl.Call(m, "UpdateModelVersion", varargs...)
|
||||
ret0, _ := ret[0].(*v1.ModelVersion)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// UpdateModelVersion indicates an expected call of UpdateModelVersion.
|
||||
func (mr *MockManagerClientMockRecorder) UpdateModelVersion(ctx, in interface{}, opts ...interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
varargs := append([]interface{}{ctx, in}, opts...)
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateModelVersion", reflect.TypeOf((*MockManagerClient)(nil).UpdateModelVersion), varargs...)
|
||||
}
|
||||
|
||||
// UpdateScheduler mocks base method.
|
||||
func (m *MockManagerClient) UpdateScheduler(ctx context.Context, in *v1.UpdateSchedulerRequest, opts ...grpc.CallOption) (*v1.Scheduler, error) {
|
||||
m.ctrl.T.Helper()
|
||||
|
|
@ -358,6 +558,96 @@ func (m *MockManagerServer) EXPECT() *MockManagerServerMockRecorder {
|
|||
return m.recorder
|
||||
}
|
||||
|
||||
// CreateModel mocks base method.
|
||||
func (m *MockManagerServer) CreateModel(arg0 context.Context, arg1 *v1.CreateModelRequest) (*v1.Model, error) {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "CreateModel", arg0, arg1)
|
||||
ret0, _ := ret[0].(*v1.Model)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// CreateModel indicates an expected call of CreateModel.
|
||||
func (mr *MockManagerServerMockRecorder) CreateModel(arg0, arg1 interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateModel", reflect.TypeOf((*MockManagerServer)(nil).CreateModel), arg0, arg1)
|
||||
}
|
||||
|
||||
// CreateModelVersion mocks base method.
|
||||
func (m *MockManagerServer) CreateModelVersion(arg0 context.Context, arg1 *v1.CreateModelVersionRequest) (*v1.ModelVersion, error) {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "CreateModelVersion", arg0, arg1)
|
||||
ret0, _ := ret[0].(*v1.ModelVersion)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// CreateModelVersion indicates an expected call of CreateModelVersion.
|
||||
func (mr *MockManagerServerMockRecorder) CreateModelVersion(arg0, arg1 interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateModelVersion", reflect.TypeOf((*MockManagerServer)(nil).CreateModelVersion), arg0, arg1)
|
||||
}
|
||||
|
||||
// DeleteModel mocks base method.
|
||||
func (m *MockManagerServer) DeleteModel(arg0 context.Context, arg1 *v1.DeleteModelRequest) (*emptypb.Empty, error) {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "DeleteModel", arg0, arg1)
|
||||
ret0, _ := ret[0].(*emptypb.Empty)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// DeleteModel indicates an expected call of DeleteModel.
|
||||
func (mr *MockManagerServerMockRecorder) DeleteModel(arg0, arg1 interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteModel", reflect.TypeOf((*MockManagerServer)(nil).DeleteModel), arg0, arg1)
|
||||
}
|
||||
|
||||
// DeleteModelVersion mocks base method.
|
||||
func (m *MockManagerServer) DeleteModelVersion(arg0 context.Context, arg1 *v1.DeleteModelVersionRequest) (*emptypb.Empty, error) {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "DeleteModelVersion", arg0, arg1)
|
||||
ret0, _ := ret[0].(*emptypb.Empty)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// DeleteModelVersion indicates an expected call of DeleteModelVersion.
|
||||
func (mr *MockManagerServerMockRecorder) DeleteModelVersion(arg0, arg1 interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteModelVersion", reflect.TypeOf((*MockManagerServer)(nil).DeleteModelVersion), arg0, arg1)
|
||||
}
|
||||
|
||||
// GetModel mocks base method.
|
||||
func (m *MockManagerServer) GetModel(arg0 context.Context, arg1 *v1.GetModelRequest) (*v1.Model, error) {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "GetModel", arg0, arg1)
|
||||
ret0, _ := ret[0].(*v1.Model)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// GetModel indicates an expected call of GetModel.
|
||||
func (mr *MockManagerServerMockRecorder) GetModel(arg0, arg1 interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetModel", reflect.TypeOf((*MockManagerServer)(nil).GetModel), arg0, arg1)
|
||||
}
|
||||
|
||||
// GetModelVersion mocks base method.
|
||||
func (m *MockManagerServer) GetModelVersion(arg0 context.Context, arg1 *v1.GetModelVersionRequest) (*v1.ModelVersion, error) {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "GetModelVersion", arg0, arg1)
|
||||
ret0, _ := ret[0].(*v1.ModelVersion)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// GetModelVersion indicates an expected call of GetModelVersion.
|
||||
func (mr *MockManagerServerMockRecorder) GetModelVersion(arg0, arg1 interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetModelVersion", reflect.TypeOf((*MockManagerServer)(nil).GetModelVersion), arg0, arg1)
|
||||
}
|
||||
|
||||
// GetObjectStorage mocks base method.
|
||||
func (m *MockManagerServer) GetObjectStorage(arg0 context.Context, arg1 *v1.GetObjectStorageRequest) (*v1.ObjectStorage, error) {
|
||||
m.ctrl.T.Helper()
|
||||
|
|
@ -432,6 +722,36 @@ func (mr *MockManagerServerMockRecorder) ListBuckets(arg0, arg1 interface{}) *go
|
|||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListBuckets", reflect.TypeOf((*MockManagerServer)(nil).ListBuckets), arg0, arg1)
|
||||
}
|
||||
|
||||
// ListModelVersions mocks base method.
|
||||
func (m *MockManagerServer) ListModelVersions(arg0 context.Context, arg1 *v1.ListModelVersionsRequest) (*v1.ListModelVersionsResponse, error) {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "ListModelVersions", arg0, arg1)
|
||||
ret0, _ := ret[0].(*v1.ListModelVersionsResponse)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// ListModelVersions indicates an expected call of ListModelVersions.
|
||||
func (mr *MockManagerServerMockRecorder) ListModelVersions(arg0, arg1 interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListModelVersions", reflect.TypeOf((*MockManagerServer)(nil).ListModelVersions), arg0, arg1)
|
||||
}
|
||||
|
||||
// ListModels mocks base method.
|
||||
func (m *MockManagerServer) ListModels(arg0 context.Context, arg1 *v1.ListModelsRequest) (*v1.ListModelsResponse, error) {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "ListModels", arg0, arg1)
|
||||
ret0, _ := ret[0].(*v1.ListModelsResponse)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// ListModels indicates an expected call of ListModels.
|
||||
func (mr *MockManagerServerMockRecorder) ListModels(arg0, arg1 interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListModels", reflect.TypeOf((*MockManagerServer)(nil).ListModels), arg0, arg1)
|
||||
}
|
||||
|
||||
// ListSchedulers mocks base method.
|
||||
func (m *MockManagerServer) ListSchedulers(arg0 context.Context, arg1 *v1.ListSchedulersRequest) (*v1.ListSchedulersResponse, error) {
|
||||
m.ctrl.T.Helper()
|
||||
|
|
@ -447,6 +767,36 @@ func (mr *MockManagerServerMockRecorder) ListSchedulers(arg0, arg1 interface{})
|
|||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListSchedulers", reflect.TypeOf((*MockManagerServer)(nil).ListSchedulers), arg0, arg1)
|
||||
}
|
||||
|
||||
// UpdateModel mocks base method.
|
||||
func (m *MockManagerServer) UpdateModel(arg0 context.Context, arg1 *v1.UpdateModelRequest) (*v1.Model, error) {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "UpdateModel", arg0, arg1)
|
||||
ret0, _ := ret[0].(*v1.Model)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// UpdateModel indicates an expected call of UpdateModel.
|
||||
func (mr *MockManagerServerMockRecorder) UpdateModel(arg0, arg1 interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateModel", reflect.TypeOf((*MockManagerServer)(nil).UpdateModel), arg0, arg1)
|
||||
}
|
||||
|
||||
// UpdateModelVersion mocks base method.
|
||||
func (m *MockManagerServer) UpdateModelVersion(arg0 context.Context, arg1 *v1.UpdateModelVersionRequest) (*v1.ModelVersion, error) {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "UpdateModelVersion", arg0, arg1)
|
||||
ret0, _ := ret[0].(*v1.ModelVersion)
|
||||
ret1, _ := ret[1].(error)
|
||||
return ret0, ret1
|
||||
}
|
||||
|
||||
// UpdateModelVersion indicates an expected call of UpdateModelVersion.
|
||||
func (mr *MockManagerServerMockRecorder) UpdateModelVersion(arg0, arg1 interface{}) *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateModelVersion", reflect.TypeOf((*MockManagerServer)(nil).UpdateModelVersion), arg0, arg1)
|
||||
}
|
||||
|
||||
// UpdateScheduler mocks base method.
|
||||
func (m *MockManagerServer) UpdateScheduler(arg0 context.Context, arg1 *v1.UpdateSchedulerRequest) (*v1.Scheduler, error) {
|
||||
m.ctrl.T.Helper()
|
||||
|
|
|
|||
Loading…
Reference in New Issue