feat: add features field for manager (#385)

Signed-off-by: Gaius <gaius.qi@gmail.com>

Cargo.lock

@@ -190,7 +190,7 @@ dependencies = [
 
 [[package]]
 name = "dragonfly-api"
-version = "2.0.154"
+version = "2.0.155"
 dependencies = [
  "prost",
  "prost-types",

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "dragonfly-api"
-version = "2.0.154"
+version = "2.0.155"
 authors = ["Gaius <gaius.qi@gmail.com>"]
 edition = "2021"
 license = "Apache-2.0"

pkg/apis/manager/v1/mocks/manager_mock.go

@@ -43,26 +43,6 @@ func (m *MockManagerClient) EXPECT() *MockManagerClientMockRecorder {
 	return m.recorder
 }
 
-// CreateModel mocks base method.
-func (m *MockManagerClient) CreateModel(ctx context.Context, in *manager.CreateModelRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) {
-	m.ctrl.T.Helper()
-	varargs := []any{ctx, in}
-	for _, a := range opts {
-		varargs = append(varargs, a)
-	}
-	ret := m.ctrl.Call(m, "CreateModel", varargs...)
-	ret0, _ := ret[0].(*emptypb.Empty)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// CreateModel indicates an expected call of CreateModel.
-func (mr *MockManagerClientMockRecorder) CreateModel(ctx, in any, opts ...any) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	varargs := append([]any{ctx, in}, opts...)
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateModel", reflect.TypeOf((*MockManagerClient)(nil).CreateModel), varargs...)
-}
-
 // GetObjectStorage mocks base method.
 func (m *MockManagerClient) GetObjectStorage(ctx context.Context, in *manager.GetObjectStorageRequest, opts ...grpc.CallOption) (*manager.ObjectStorage, error) {
 	m.ctrl.T.Helper()
@@ -423,21 +403,6 @@ func (m *MockManagerServer) EXPECT() *MockManagerServerMockRecorder {
 	return m.recorder
 }
 
-// CreateModel mocks base method.
-func (m *MockManagerServer) CreateModel(arg0 context.Context, arg1 *manager.CreateModelRequest) (*emptypb.Empty, error) {
-	m.ctrl.T.Helper()
-	ret := m.ctrl.Call(m, "CreateModel", arg0, arg1)
-	ret0, _ := ret[0].(*emptypb.Empty)
-	ret1, _ := ret[1].(error)
-	return ret0, ret1
-}
-
-// CreateModel indicates an expected call of CreateModel.
-func (mr *MockManagerServerMockRecorder) CreateModel(arg0, arg1 any) *gomock.Call {
-	mr.mock.ctrl.T.Helper()
-	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateModel", reflect.TypeOf((*MockManagerServer)(nil).CreateModel), arg0, arg1)
-}
-
 // GetObjectStorage mocks base method.
 func (m *MockManagerServer) GetObjectStorage(arg0 context.Context, arg1 *manager.GetObjectStorageRequest) (*manager.ObjectStorage, error) {
 	m.ctrl.T.Helper()

pkg/apis/manager/v2/manager.proto

@@ -212,6 +212,8 @@ message UpdateSchedulerRequest {
   string ip = 6 [(validate.rules).string = {ip: true}];
   // Scheduler port.
   int32 port = 7 [(validate.rules).int32 = {gte: 1024, lt: 65535}];
+  // Scheduler features.
+  repeated string features = 8 [(validate.rules).string = {min_len: 1, ignore_empty: true}];
 }
 
 // ListSchedulersRequest represents request of ListSchedulers.

proto/manager.proto

@@ -209,6 +209,8 @@ message UpdateSchedulerRequest {
   string ip = 6;
   // Scheduler port.
   int32 port = 7;
+  // Scheduler features.
+  repeated string features = 8;
 }
 
 // ListSchedulersRequest represents request of ListSchedulers.

src/manager.v2.rs

@@ -265,6 +265,9 @@ pub struct UpdateSchedulerRequest {
     /// Scheduler port.
     #[prost(int32, tag = "7")]
     pub port: i32,
+    /// Scheduler features.
+    #[prost(string, repeated, tag = "8")]
+    pub features: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
 }
 /// ListSchedulersRequest represents request of ListSchedulers.
 #[derive(serde::Serialize, serde::Deserialize)]