diff --git a/internal/distributed/proxy/client/client.go b/internal/distributed/proxy/client/client.go index 5f7ac68843d4c..fe58e81aae302 100644 --- a/internal/distributed/proxy/client/client.go +++ b/internal/distributed/proxy/client/client.go @@ -198,3 +198,21 @@ func (c *Client) GetDdChannel(ctx context.Context, req *internalpb.GetDdChannelR return client.GetDdChannel(ctx, req) }) } + +func (c *Client) ImportV2(ctx context.Context, req *internalpb.ImportRequest, opts ...grpc.CallOption) (*internalpb.ImportResponse, error) { + return wrapGrpcCall(ctx, c, func(client proxypb.ProxyClient) (*internalpb.ImportResponse, error) { + return client.ImportV2(ctx, req) + }) +} + +func (c *Client) GetImportProgress(ctx context.Context, req *internalpb.GetImportProgressRequest, opts ...grpc.CallOption) (*internalpb.GetImportProgressResponse, error) { + return wrapGrpcCall(ctx, c, func(client proxypb.ProxyClient) (*internalpb.GetImportProgressResponse, error) { + return client.GetImportProgress(ctx, req) + }) +} + +func (c *Client) ListImports(ctx context.Context, req *internalpb.ListImportsRequest, opts ...grpc.CallOption) (*internalpb.ListImportsResponse, error) { + return wrapGrpcCall(ctx, c, func(client proxypb.ProxyClient) (*internalpb.ListImportsResponse, error) { + return client.ListImports(ctx, req) + }) +} diff --git a/internal/distributed/proxy/client/client_test.go b/internal/distributed/proxy/client/client_test.go index 8ee3899ab5a77..eda883255bacd 100644 --- a/internal/distributed/proxy/client/client_test.go +++ b/internal/distributed/proxy/client/client_test.go @@ -433,3 +433,32 @@ func Test_GetDdChannel(t *testing.T) { _, err = client.GetDdChannel(ctx, &internalpb.GetDdChannelRequest{}) assert.ErrorIs(t, err, context.DeadlineExceeded) } + +func Test_ImportV2(t *testing.T) { + paramtable.Init() + ctx := context.Background() + + client, err := NewClient(ctx, "test", 1) + assert.NoError(t, err) + defer client.Close() + + mockProxy := mocks.NewMockProxyClient(t) + mockGrpcClient := mocks.NewMockGrpcClient[proxypb.ProxyClient](t) + mockGrpcClient.EXPECT().Close().Return(nil) + mockGrpcClient.EXPECT().ReCall(mock.Anything, mock.Anything).RunAndReturn(func(ctx context.Context, f func(proxypb.ProxyClient) (interface{}, error)) (interface{}, error) { + return f(mockProxy) + }) + client.(*Client).grpcClient = mockGrpcClient + + mockProxy.EXPECT().ImportV2(mock.Anything, mock.Anything).Return(&internalpb.ImportResponse{Status: merr.Success()}, nil) + _, err = client.ImportV2(ctx, &internalpb.ImportRequest{}) + assert.Nil(t, err) + + mockProxy.EXPECT().GetImportProgress(mock.Anything, mock.Anything).Return(&internalpb.GetImportProgressResponse{Status: merr.Success()}, nil) + _, err = client.GetImportProgress(ctx, &internalpb.GetImportProgressRequest{}) + assert.Nil(t, err) + + mockProxy.EXPECT().ListImports(mock.Anything, mock.Anything).Return(&internalpb.ListImportsResponse{Status: merr.Success()}, nil) + _, err = client.ListImports(ctx, &internalpb.ListImportsRequest{}) + assert.Nil(t, err) +} diff --git a/internal/distributed/proxy/httpserver/handler_v2.go b/internal/distributed/proxy/httpserver/handler_v2.go index 0b7b7bafcdc34..daa4cccee58eb 100644 --- a/internal/distributed/proxy/httpserver/handler_v2.go +++ b/internal/distributed/proxy/httpserver/handler_v2.go @@ -1564,7 +1564,7 @@ func (h *HandlersV2) getImportJobProcess(ctx context.Context, c *gin.Context, an jobIDGetter := anyReq.(JobIDGetter) req := &internalpb.GetImportProgressRequest{ DbName: dbName, - JobID: strconv.FormatInt(jobIDGetter.GetJobID(), 10), + JobID: jobIDGetter.GetJobID(), } resp, err := wrapperProxy(ctx, c, req, h.checkAuth, false, func(reqCtx context.Context, req any) (interface{}, error) { return h.proxy.GetImportProgress(reqCtx, req.(*internalpb.GetImportProgressRequest)) diff --git a/internal/distributed/proxy/httpserver/handler_v2_test.go b/internal/distributed/proxy/httpserver/handler_v2_test.go index 04dbc078015bb..458744f8bda64 100644 --- a/internal/distributed/proxy/httpserver/handler_v2_test.go +++ b/internal/distributed/proxy/httpserver/handler_v2_test.go @@ -810,18 +810,18 @@ func TestMethodPost(t *testing.T) { mp.EXPECT().ListImports(mock.Anything, mock.Anything).Return(&internalpb.ListImportsResponse{ Status: &StatusSuccess, JobIDs: []string{"1", "2", "3", "4"}, - States: []internalpb.ImportState{ - internalpb.ImportState_Pending, - internalpb.ImportState_InProgress, - internalpb.ImportState_Failed, - internalpb.ImportState_Completed, + States: []internalpb.ImportJobState{ + internalpb.ImportJobState_Pending, + internalpb.ImportJobState_Importing, + internalpb.ImportJobState_Failed, + internalpb.ImportJobState_Completed, }, Reasons: []string{"", "", "mock reason", ""}, Progresses: []int64{0, 30, 0, 100}, }, nil).Once() mp.EXPECT().GetImportProgress(mock.Anything, mock.Anything).Return(&internalpb.GetImportProgressResponse{ Status: &StatusSuccess, - State: internalpb.ImportState_Completed, + State: internalpb.ImportJobState_Completed, Reason: "", Progress: 100, }, nil).Once() @@ -903,7 +903,7 @@ func TestMethodPost(t *testing.T) { `"userName": "` + util.UserRoot + `", "password": "Milvus", "newPassword": "milvus", "roleName": "` + util.RoleAdmin + `",` + `"roleName": "` + util.RoleAdmin + `", "objectType": "Global", "objectName": "*", "privilege": "*",` + `"aliasName": "` + DefaultAliasName + `",` + - `"jobID": 1234567890,` + + `"jobID": "1234567890",` + `"files": [["book.json"]]` + `}`)) req := httptest.NewRequest(http.MethodPost, testcase.path, bodyReader) diff --git a/internal/distributed/proxy/httpserver/request_v2.go b/internal/distributed/proxy/httpserver/request_v2.go index f7833b31cbf3b..c0b19dd283b93 100644 --- a/internal/distributed/proxy/httpserver/request_v2.go +++ b/internal/distributed/proxy/httpserver/request_v2.go @@ -69,7 +69,7 @@ func (req *ImportReq) GetCollectionName() string { } func (req *ImportReq) GetPartitionName() string { - return req.CollectionName + return req.PartitionName } func (req *ImportReq) GetFiles() [][]string { @@ -81,10 +81,10 @@ func (req *ImportReq) GetOptions() map[string]string { } type JobIDReq struct { - JobID int64 `json:"jobID" binding:"required"` + JobID string `json:"jobID" binding:"required"` } -func (req *JobIDReq) GetJobID() int64 { return req.JobID } +func (req *JobIDReq) GetJobID() string { return req.JobID } type QueryReqV2 struct { DbName string `json:"dbName"` @@ -211,7 +211,7 @@ type OptionsGetter interface { GetOptions() map[string]string } type JobIDGetter interface { - GetJobID() int64 + GetJobID() string } type PasswordReq struct { diff --git a/internal/distributed/proxy/service.go b/internal/distributed/proxy/service.go index d95f36a3a1541..b0af0611609c6 100644 --- a/internal/distributed/proxy/service.go +++ b/internal/distributed/proxy/service.go @@ -1202,3 +1202,15 @@ func (s *Server) AllocTimestamp(ctx context.Context, req *milvuspb.AllocTimestam func (s *Server) ReplicateMessage(ctx context.Context, req *milvuspb.ReplicateMessageRequest) (*milvuspb.ReplicateMessageResponse, error) { return s.proxy.ReplicateMessage(ctx, req) } + +func (s *Server) ImportV2(ctx context.Context, req *internalpb.ImportRequest) (*internalpb.ImportResponse, error) { + return s.proxy.ImportV2(ctx, req) +} + +func (s *Server) GetImportProgress(ctx context.Context, req *internalpb.GetImportProgressRequest) (*internalpb.GetImportProgressResponse, error) { + return s.proxy.GetImportProgress(ctx, req) +} + +func (s *Server) ListImports(ctx context.Context, req *internalpb.ListImportsRequest) (*internalpb.ListImportsResponse, error) { + return s.proxy.ListImports(ctx, req) +} diff --git a/internal/mocks/mock_datacoord.go b/internal/mocks/mock_datacoord.go index afdf107ef30ab..06a56ad832924 100644 --- a/internal/mocks/mock_datacoord.go +++ b/internal/mocks/mock_datacoord.go @@ -2113,15 +2113,15 @@ func (_c *MockDataCoord_Init_Call) RunAndReturn(run func() error) *MockDataCoord } // ListImports provides a mock function with given fields: _a0, _a1 -func (_m *MockDataCoord) ListImports(_a0 context.Context, _a1 *internalpb.ListImportsRequest) (*internalpb.ListImportsResponse, error) { +func (_m *MockDataCoord) ListImports(_a0 context.Context, _a1 *internalpb.ListImportsRequestInternal) (*internalpb.ListImportsResponse, error) { ret := _m.Called(_a0, _a1) var r0 *internalpb.ListImportsResponse var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequest) (*internalpb.ListImportsResponse, error)); ok { + if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequestInternal) (*internalpb.ListImportsResponse, error)); ok { return rf(_a0, _a1) } - if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequest) *internalpb.ListImportsResponse); ok { + if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequestInternal) *internalpb.ListImportsResponse); ok { r0 = rf(_a0, _a1) } else { if ret.Get(0) != nil { @@ -2129,7 +2129,7 @@ func (_m *MockDataCoord) ListImports(_a0 context.Context, _a1 *internalpb.ListIm } } - if rf, ok := ret.Get(1).(func(context.Context, *internalpb.ListImportsRequest) error); ok { + if rf, ok := ret.Get(1).(func(context.Context, *internalpb.ListImportsRequestInternal) error); ok { r1 = rf(_a0, _a1) } else { r1 = ret.Error(1) @@ -2145,14 +2145,14 @@ type MockDataCoord_ListImports_Call struct { // ListImports is a helper method to define mock.On call // - _a0 context.Context -// - _a1 *internalpb.ListImportsRequest +// - _a1 *internalpb.ListImportsRequestInternal func (_e *MockDataCoord_Expecter) ListImports(_a0 interface{}, _a1 interface{}) *MockDataCoord_ListImports_Call { return &MockDataCoord_ListImports_Call{Call: _e.mock.On("ListImports", _a0, _a1)} } -func (_c *MockDataCoord_ListImports_Call) Run(run func(_a0 context.Context, _a1 *internalpb.ListImportsRequest)) *MockDataCoord_ListImports_Call { +func (_c *MockDataCoord_ListImports_Call) Run(run func(_a0 context.Context, _a1 *internalpb.ListImportsRequestInternal)) *MockDataCoord_ListImports_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].(*internalpb.ListImportsRequest)) + run(args[0].(context.Context), args[1].(*internalpb.ListImportsRequestInternal)) }) return _c } @@ -2162,7 +2162,7 @@ func (_c *MockDataCoord_ListImports_Call) Return(_a0 *internalpb.ListImportsResp return _c } -func (_c *MockDataCoord_ListImports_Call) RunAndReturn(run func(context.Context, *internalpb.ListImportsRequest) (*internalpb.ListImportsResponse, error)) *MockDataCoord_ListImports_Call { +func (_c *MockDataCoord_ListImports_Call) RunAndReturn(run func(context.Context, *internalpb.ListImportsRequestInternal) (*internalpb.ListImportsResponse, error)) *MockDataCoord_ListImports_Call { _c.Call.Return(run) return _c } diff --git a/internal/mocks/mock_datacoord_client.go b/internal/mocks/mock_datacoord_client.go index 05b930444dbf4..efba80d15297f 100644 --- a/internal/mocks/mock_datacoord_client.go +++ b/internal/mocks/mock_datacoord_client.go @@ -2665,7 +2665,7 @@ func (_c *MockDataCoordClient_ImportV2_Call) RunAndReturn(run func(context.Conte } // ListImports provides a mock function with given fields: ctx, in, opts -func (_m *MockDataCoordClient) ListImports(ctx context.Context, in *internalpb.ListImportsRequest, opts ...grpc.CallOption) (*internalpb.ListImportsResponse, error) { +func (_m *MockDataCoordClient) ListImports(ctx context.Context, in *internalpb.ListImportsRequestInternal, opts ...grpc.CallOption) (*internalpb.ListImportsResponse, error) { _va := make([]interface{}, len(opts)) for _i := range opts { _va[_i] = opts[_i] @@ -2677,10 +2677,10 @@ func (_m *MockDataCoordClient) ListImports(ctx context.Context, in *internalpb.L var r0 *internalpb.ListImportsResponse var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequest, ...grpc.CallOption) (*internalpb.ListImportsResponse, error)); ok { + if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequestInternal, ...grpc.CallOption) (*internalpb.ListImportsResponse, error)); ok { return rf(ctx, in, opts...) } - if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequest, ...grpc.CallOption) *internalpb.ListImportsResponse); ok { + if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequestInternal, ...grpc.CallOption) *internalpb.ListImportsResponse); ok { r0 = rf(ctx, in, opts...) } else { if ret.Get(0) != nil { @@ -2688,7 +2688,7 @@ func (_m *MockDataCoordClient) ListImports(ctx context.Context, in *internalpb.L } } - if rf, ok := ret.Get(1).(func(context.Context, *internalpb.ListImportsRequest, ...grpc.CallOption) error); ok { + if rf, ok := ret.Get(1).(func(context.Context, *internalpb.ListImportsRequestInternal, ...grpc.CallOption) error); ok { r1 = rf(ctx, in, opts...) } else { r1 = ret.Error(1) @@ -2704,14 +2704,14 @@ type MockDataCoordClient_ListImports_Call struct { // ListImports is a helper method to define mock.On call // - ctx context.Context -// - in *internalpb.ListImportsRequest +// - in *internalpb.ListImportsRequestInternal // - opts ...grpc.CallOption func (_e *MockDataCoordClient_Expecter) ListImports(ctx interface{}, in interface{}, opts ...interface{}) *MockDataCoordClient_ListImports_Call { return &MockDataCoordClient_ListImports_Call{Call: _e.mock.On("ListImports", append([]interface{}{ctx, in}, opts...)...)} } -func (_c *MockDataCoordClient_ListImports_Call) Run(run func(ctx context.Context, in *internalpb.ListImportsRequest, opts ...grpc.CallOption)) *MockDataCoordClient_ListImports_Call { +func (_c *MockDataCoordClient_ListImports_Call) Run(run func(ctx context.Context, in *internalpb.ListImportsRequestInternal, opts ...grpc.CallOption)) *MockDataCoordClient_ListImports_Call { _c.Call.Run(func(args mock.Arguments) { variadicArgs := make([]grpc.CallOption, len(args)-2) for i, a := range args[2:] { @@ -2719,7 +2719,7 @@ func (_c *MockDataCoordClient_ListImports_Call) Run(run func(ctx context.Context variadicArgs[i] = a.(grpc.CallOption) } } - run(args[0].(context.Context), args[1].(*internalpb.ListImportsRequest), variadicArgs...) + run(args[0].(context.Context), args[1].(*internalpb.ListImportsRequestInternal), variadicArgs...) }) return _c } @@ -2729,7 +2729,7 @@ func (_c *MockDataCoordClient_ListImports_Call) Return(_a0 *internalpb.ListImpor return _c } -func (_c *MockDataCoordClient_ListImports_Call) RunAndReturn(run func(context.Context, *internalpb.ListImportsRequest, ...grpc.CallOption) (*internalpb.ListImportsResponse, error)) *MockDataCoordClient_ListImports_Call { +func (_c *MockDataCoordClient_ListImports_Call) RunAndReturn(run func(context.Context, *internalpb.ListImportsRequestInternal, ...grpc.CallOption) (*internalpb.ListImportsResponse, error)) *MockDataCoordClient_ListImports_Call { _c.Call.Return(run) return _c } diff --git a/internal/mocks/mock_datanode.go b/internal/mocks/mock_datanode.go index 21e3762e40b01..e2ed0a0c5846f 100644 --- a/internal/mocks/mock_datanode.go +++ b/internal/mocks/mock_datanode.go @@ -64,8 +64,8 @@ type MockDataNode_AddImportSegment_Call struct { } // AddImportSegment is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.AddImportSegmentRequest +// - _a0 context.Context +// - _a1 *datapb.AddImportSegmentRequest func (_e *MockDataNode_Expecter) AddImportSegment(_a0 interface{}, _a1 interface{}) *MockDataNode_AddImportSegment_Call { return &MockDataNode_AddImportSegment_Call{Call: _e.mock.On("AddImportSegment", _a0, _a1)} } @@ -119,8 +119,8 @@ type MockDataNode_CheckChannelOperationProgress_Call struct { } // CheckChannelOperationProgress is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.ChannelWatchInfo +// - _a0 context.Context +// - _a1 *datapb.ChannelWatchInfo func (_e *MockDataNode_Expecter) CheckChannelOperationProgress(_a0 interface{}, _a1 interface{}) *MockDataNode_CheckChannelOperationProgress_Call { return &MockDataNode_CheckChannelOperationProgress_Call{Call: _e.mock.On("CheckChannelOperationProgress", _a0, _a1)} } @@ -174,8 +174,8 @@ type MockDataNode_Compaction_Call struct { } // Compaction is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.CompactionPlan +// - _a0 context.Context +// - _a1 *datapb.CompactionPlan func (_e *MockDataNode_Expecter) Compaction(_a0 interface{}, _a1 interface{}) *MockDataNode_Compaction_Call { return &MockDataNode_Compaction_Call{Call: _e.mock.On("Compaction", _a0, _a1)} } @@ -229,8 +229,8 @@ type MockDataNode_DropImport_Call struct { } // DropImport is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.DropImportRequest +// - _a0 context.Context +// - _a1 *datapb.DropImportRequest func (_e *MockDataNode_Expecter) DropImport(_a0 interface{}, _a1 interface{}) *MockDataNode_DropImport_Call { return &MockDataNode_DropImport_Call{Call: _e.mock.On("DropImport", _a0, _a1)} } @@ -284,8 +284,8 @@ type MockDataNode_FlushChannels_Call struct { } // FlushChannels is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.FlushChannelsRequest +// - _a0 context.Context +// - _a1 *datapb.FlushChannelsRequest func (_e *MockDataNode_Expecter) FlushChannels(_a0 interface{}, _a1 interface{}) *MockDataNode_FlushChannels_Call { return &MockDataNode_FlushChannels_Call{Call: _e.mock.On("FlushChannels", _a0, _a1)} } @@ -339,8 +339,8 @@ type MockDataNode_FlushSegments_Call struct { } // FlushSegments is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.FlushSegmentsRequest +// - _a0 context.Context +// - _a1 *datapb.FlushSegmentsRequest func (_e *MockDataNode_Expecter) FlushSegments(_a0 interface{}, _a1 interface{}) *MockDataNode_FlushSegments_Call { return &MockDataNode_FlushSegments_Call{Call: _e.mock.On("FlushSegments", _a0, _a1)} } @@ -435,8 +435,8 @@ type MockDataNode_GetCompactionState_Call struct { } // GetCompactionState is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.CompactionStateRequest +// - _a0 context.Context +// - _a1 *datapb.CompactionStateRequest func (_e *MockDataNode_Expecter) GetCompactionState(_a0 interface{}, _a1 interface{}) *MockDataNode_GetCompactionState_Call { return &MockDataNode_GetCompactionState_Call{Call: _e.mock.On("GetCompactionState", _a0, _a1)} } @@ -490,8 +490,8 @@ type MockDataNode_GetComponentStates_Call struct { } // GetComponentStates is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetComponentStatesRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetComponentStatesRequest func (_e *MockDataNode_Expecter) GetComponentStates(_a0 interface{}, _a1 interface{}) *MockDataNode_GetComponentStates_Call { return &MockDataNode_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", _a0, _a1)} } @@ -545,8 +545,8 @@ type MockDataNode_GetMetrics_Call struct { } // GetMetrics is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetMetricsRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetMetricsRequest func (_e *MockDataNode_Expecter) GetMetrics(_a0 interface{}, _a1 interface{}) *MockDataNode_GetMetrics_Call { return &MockDataNode_GetMetrics_Call{Call: _e.mock.On("GetMetrics", _a0, _a1)} } @@ -682,8 +682,8 @@ type MockDataNode_GetStatisticsChannel_Call struct { } // GetStatisticsChannel is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.GetStatisticsChannelRequest +// - _a0 context.Context +// - _a1 *internalpb.GetStatisticsChannelRequest func (_e *MockDataNode_Expecter) GetStatisticsChannel(_a0 interface{}, _a1 interface{}) *MockDataNode_GetStatisticsChannel_Call { return &MockDataNode_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", _a0, _a1)} } @@ -737,8 +737,8 @@ type MockDataNode_Import_Call struct { } // Import is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.ImportTaskRequest +// - _a0 context.Context +// - _a1 *datapb.ImportTaskRequest func (_e *MockDataNode_Expecter) Import(_a0 interface{}, _a1 interface{}) *MockDataNode_Import_Call { return &MockDataNode_Import_Call{Call: _e.mock.On("Import", _a0, _a1)} } @@ -792,8 +792,8 @@ type MockDataNode_ImportV2_Call struct { } // ImportV2 is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.ImportRequest +// - _a0 context.Context +// - _a1 *datapb.ImportRequest func (_e *MockDataNode_Expecter) ImportV2(_a0 interface{}, _a1 interface{}) *MockDataNode_ImportV2_Call { return &MockDataNode_ImportV2_Call{Call: _e.mock.On("ImportV2", _a0, _a1)} } @@ -888,8 +888,8 @@ type MockDataNode_NotifyChannelOperation_Call struct { } // NotifyChannelOperation is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.ChannelOperationsRequest +// - _a0 context.Context +// - _a1 *datapb.ChannelOperationsRequest func (_e *MockDataNode_Expecter) NotifyChannelOperation(_a0 interface{}, _a1 interface{}) *MockDataNode_NotifyChannelOperation_Call { return &MockDataNode_NotifyChannelOperation_Call{Call: _e.mock.On("NotifyChannelOperation", _a0, _a1)} } @@ -943,8 +943,8 @@ type MockDataNode_PreImport_Call struct { } // PreImport is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.PreImportRequest +// - _a0 context.Context +// - _a1 *datapb.PreImportRequest func (_e *MockDataNode_Expecter) PreImport(_a0 interface{}, _a1 interface{}) *MockDataNode_PreImport_Call { return &MockDataNode_PreImport_Call{Call: _e.mock.On("PreImport", _a0, _a1)} } @@ -998,8 +998,8 @@ type MockDataNode_QueryImport_Call struct { } // QueryImport is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.QueryImportRequest +// - _a0 context.Context +// - _a1 *datapb.QueryImportRequest func (_e *MockDataNode_Expecter) QueryImport(_a0 interface{}, _a1 interface{}) *MockDataNode_QueryImport_Call { return &MockDataNode_QueryImport_Call{Call: _e.mock.On("QueryImport", _a0, _a1)} } @@ -1053,8 +1053,8 @@ type MockDataNode_QueryPreImport_Call struct { } // QueryPreImport is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.QueryPreImportRequest +// - _a0 context.Context +// - _a1 *datapb.QueryPreImportRequest func (_e *MockDataNode_Expecter) QueryPreImport(_a0 interface{}, _a1 interface{}) *MockDataNode_QueryPreImport_Call { return &MockDataNode_QueryPreImport_Call{Call: _e.mock.On("QueryPreImport", _a0, _a1)} } @@ -1149,8 +1149,8 @@ type MockDataNode_ResendSegmentStats_Call struct { } // ResendSegmentStats is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.ResendSegmentStatsRequest +// - _a0 context.Context +// - _a1 *datapb.ResendSegmentStatsRequest func (_e *MockDataNode_Expecter) ResendSegmentStats(_a0 interface{}, _a1 interface{}) *MockDataNode_ResendSegmentStats_Call { return &MockDataNode_ResendSegmentStats_Call{Call: _e.mock.On("ResendSegmentStats", _a0, _a1)} } @@ -1183,7 +1183,7 @@ type MockDataNode_SetAddress_Call struct { } // SetAddress is a helper method to define mock.On call -// - address string +// - address string func (_e *MockDataNode_Expecter) SetAddress(address interface{}) *MockDataNode_SetAddress_Call { return &MockDataNode_SetAddress_Call{Call: _e.mock.On("SetAddress", address)} } @@ -1225,7 +1225,7 @@ type MockDataNode_SetDataCoordClient_Call struct { } // SetDataCoordClient is a helper method to define mock.On call -// - dataCoord types.DataCoordClient +// - dataCoord types.DataCoordClient func (_e *MockDataNode_Expecter) SetDataCoordClient(dataCoord interface{}) *MockDataNode_SetDataCoordClient_Call { return &MockDataNode_SetDataCoordClient_Call{Call: _e.mock.On("SetDataCoordClient", dataCoord)} } @@ -1258,7 +1258,7 @@ type MockDataNode_SetEtcdClient_Call struct { } // SetEtcdClient is a helper method to define mock.On call -// - etcdClient *clientv3.Client +// - etcdClient *clientv3.Client func (_e *MockDataNode_Expecter) SetEtcdClient(etcdClient interface{}) *MockDataNode_SetEtcdClient_Call { return &MockDataNode_SetEtcdClient_Call{Call: _e.mock.On("SetEtcdClient", etcdClient)} } @@ -1300,7 +1300,7 @@ type MockDataNode_SetRootCoordClient_Call struct { } // SetRootCoordClient is a helper method to define mock.On call -// - rootCoord types.RootCoordClient +// - rootCoord types.RootCoordClient func (_e *MockDataNode_Expecter) SetRootCoordClient(rootCoord interface{}) *MockDataNode_SetRootCoordClient_Call { return &MockDataNode_SetRootCoordClient_Call{Call: _e.mock.On("SetRootCoordClient", rootCoord)} } @@ -1354,8 +1354,8 @@ type MockDataNode_ShowConfigurations_Call struct { } // ShowConfigurations is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.ShowConfigurationsRequest +// - _a0 context.Context +// - _a1 *internalpb.ShowConfigurationsRequest func (_e *MockDataNode_Expecter) ShowConfigurations(_a0 interface{}, _a1 interface{}) *MockDataNode_ShowConfigurations_Call { return &MockDataNode_ShowConfigurations_Call{Call: _e.mock.On("ShowConfigurations", _a0, _a1)} } @@ -1491,8 +1491,8 @@ type MockDataNode_SyncSegments_Call struct { } // SyncSegments is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.SyncSegmentsRequest +// - _a0 context.Context +// - _a1 *datapb.SyncSegmentsRequest func (_e *MockDataNode_Expecter) SyncSegments(_a0 interface{}, _a1 interface{}) *MockDataNode_SyncSegments_Call { return &MockDataNode_SyncSegments_Call{Call: _e.mock.On("SyncSegments", _a0, _a1)} } @@ -1525,7 +1525,7 @@ type MockDataNode_UpdateStateCode_Call struct { } // UpdateStateCode is a helper method to define mock.On call -// - stateCode commonpb.StateCode +// - stateCode commonpb.StateCode func (_e *MockDataNode_Expecter) UpdateStateCode(stateCode interface{}) *MockDataNode_UpdateStateCode_Call { return &MockDataNode_UpdateStateCode_Call{Call: _e.mock.On("UpdateStateCode", stateCode)} } @@ -1579,8 +1579,8 @@ type MockDataNode_WatchDmChannels_Call struct { } // WatchDmChannels is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *datapb.WatchDmChannelsRequest +// - _a0 context.Context +// - _a1 *datapb.WatchDmChannelsRequest func (_e *MockDataNode_Expecter) WatchDmChannels(_a0 interface{}, _a1 interface{}) *MockDataNode_WatchDmChannels_Call { return &MockDataNode_WatchDmChannels_Call{Call: _e.mock.On("WatchDmChannels", _a0, _a1)} } diff --git a/internal/mocks/mock_datanode_client.go b/internal/mocks/mock_datanode_client.go index 8d603b84bbb78..c0db2cb6e2dca 100644 --- a/internal/mocks/mock_datanode_client.go +++ b/internal/mocks/mock_datanode_client.go @@ -70,9 +70,9 @@ type MockDataNodeClient_AddImportSegment_Call struct { } // AddImportSegment is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.AddImportSegmentRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.AddImportSegmentRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) AddImportSegment(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_AddImportSegment_Call { return &MockDataNodeClient_AddImportSegment_Call{Call: _e.mock.On("AddImportSegment", append([]interface{}{ctx, in}, opts...)...)} @@ -140,9 +140,9 @@ type MockDataNodeClient_CheckChannelOperationProgress_Call struct { } // CheckChannelOperationProgress is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.ChannelWatchInfo -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.ChannelWatchInfo +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) CheckChannelOperationProgress(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_CheckChannelOperationProgress_Call { return &MockDataNodeClient_CheckChannelOperationProgress_Call{Call: _e.mock.On("CheckChannelOperationProgress", append([]interface{}{ctx, in}, opts...)...)} @@ -251,9 +251,9 @@ type MockDataNodeClient_Compaction_Call struct { } // Compaction is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.CompactionPlan -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.CompactionPlan +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) Compaction(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_Compaction_Call { return &MockDataNodeClient_Compaction_Call{Call: _e.mock.On("Compaction", append([]interface{}{ctx, in}, opts...)...)} @@ -321,9 +321,9 @@ type MockDataNodeClient_DropImport_Call struct { } // DropImport is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.DropImportRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.DropImportRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) DropImport(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_DropImport_Call { return &MockDataNodeClient_DropImport_Call{Call: _e.mock.On("DropImport", append([]interface{}{ctx, in}, opts...)...)} @@ -391,9 +391,9 @@ type MockDataNodeClient_FlushChannels_Call struct { } // FlushChannels is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.FlushChannelsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.FlushChannelsRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) FlushChannels(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_FlushChannels_Call { return &MockDataNodeClient_FlushChannels_Call{Call: _e.mock.On("FlushChannels", append([]interface{}{ctx, in}, opts...)...)} @@ -461,9 +461,9 @@ type MockDataNodeClient_FlushSegments_Call struct { } // FlushSegments is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.FlushSegmentsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.FlushSegmentsRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) FlushSegments(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_FlushSegments_Call { return &MockDataNodeClient_FlushSegments_Call{Call: _e.mock.On("FlushSegments", append([]interface{}{ctx, in}, opts...)...)} @@ -531,9 +531,9 @@ type MockDataNodeClient_GetCompactionState_Call struct { } // GetCompactionState is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.CompactionStateRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.CompactionStateRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) GetCompactionState(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_GetCompactionState_Call { return &MockDataNodeClient_GetCompactionState_Call{Call: _e.mock.On("GetCompactionState", append([]interface{}{ctx, in}, opts...)...)} @@ -601,9 +601,9 @@ type MockDataNodeClient_GetComponentStates_Call struct { } // GetComponentStates is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetComponentStatesRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetComponentStatesRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) GetComponentStates(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_GetComponentStates_Call { return &MockDataNodeClient_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", append([]interface{}{ctx, in}, opts...)...)} @@ -671,9 +671,9 @@ type MockDataNodeClient_GetMetrics_Call struct { } // GetMetrics is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetMetricsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetMetricsRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) GetMetrics(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_GetMetrics_Call { return &MockDataNodeClient_GetMetrics_Call{Call: _e.mock.On("GetMetrics", append([]interface{}{ctx, in}, opts...)...)} @@ -741,9 +741,9 @@ type MockDataNodeClient_GetStatisticsChannel_Call struct { } // GetStatisticsChannel is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.GetStatisticsChannelRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.GetStatisticsChannelRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) GetStatisticsChannel(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_GetStatisticsChannel_Call { return &MockDataNodeClient_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", append([]interface{}{ctx, in}, opts...)...)} @@ -811,9 +811,9 @@ type MockDataNodeClient_Import_Call struct { } // Import is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.ImportTaskRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.ImportTaskRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) Import(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_Import_Call { return &MockDataNodeClient_Import_Call{Call: _e.mock.On("Import", append([]interface{}{ctx, in}, opts...)...)} @@ -881,9 +881,9 @@ type MockDataNodeClient_ImportV2_Call struct { } // ImportV2 is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.ImportRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.ImportRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) ImportV2(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_ImportV2_Call { return &MockDataNodeClient_ImportV2_Call{Call: _e.mock.On("ImportV2", append([]interface{}{ctx, in}, opts...)...)} @@ -951,9 +951,9 @@ type MockDataNodeClient_NotifyChannelOperation_Call struct { } // NotifyChannelOperation is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.ChannelOperationsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.ChannelOperationsRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) NotifyChannelOperation(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_NotifyChannelOperation_Call { return &MockDataNodeClient_NotifyChannelOperation_Call{Call: _e.mock.On("NotifyChannelOperation", append([]interface{}{ctx, in}, opts...)...)} @@ -1021,9 +1021,9 @@ type MockDataNodeClient_PreImport_Call struct { } // PreImport is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.PreImportRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.PreImportRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) PreImport(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_PreImport_Call { return &MockDataNodeClient_PreImport_Call{Call: _e.mock.On("PreImport", append([]interface{}{ctx, in}, opts...)...)} @@ -1091,9 +1091,9 @@ type MockDataNodeClient_QueryImport_Call struct { } // QueryImport is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.QueryImportRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.QueryImportRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) QueryImport(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_QueryImport_Call { return &MockDataNodeClient_QueryImport_Call{Call: _e.mock.On("QueryImport", append([]interface{}{ctx, in}, opts...)...)} @@ -1161,9 +1161,9 @@ type MockDataNodeClient_QueryPreImport_Call struct { } // QueryPreImport is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.QueryPreImportRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.QueryPreImportRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) QueryPreImport(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_QueryPreImport_Call { return &MockDataNodeClient_QueryPreImport_Call{Call: _e.mock.On("QueryPreImport", append([]interface{}{ctx, in}, opts...)...)} @@ -1231,9 +1231,9 @@ type MockDataNodeClient_ResendSegmentStats_Call struct { } // ResendSegmentStats is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.ResendSegmentStatsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.ResendSegmentStatsRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) ResendSegmentStats(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_ResendSegmentStats_Call { return &MockDataNodeClient_ResendSegmentStats_Call{Call: _e.mock.On("ResendSegmentStats", append([]interface{}{ctx, in}, opts...)...)} @@ -1301,9 +1301,9 @@ type MockDataNodeClient_ShowConfigurations_Call struct { } // ShowConfigurations is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.ShowConfigurationsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.ShowConfigurationsRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) ShowConfigurations(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_ShowConfigurations_Call { return &MockDataNodeClient_ShowConfigurations_Call{Call: _e.mock.On("ShowConfigurations", append([]interface{}{ctx, in}, opts...)...)} @@ -1371,9 +1371,9 @@ type MockDataNodeClient_SyncSegments_Call struct { } // SyncSegments is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.SyncSegmentsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.SyncSegmentsRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) SyncSegments(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_SyncSegments_Call { return &MockDataNodeClient_SyncSegments_Call{Call: _e.mock.On("SyncSegments", append([]interface{}{ctx, in}, opts...)...)} @@ -1441,9 +1441,9 @@ type MockDataNodeClient_WatchDmChannels_Call struct { } // WatchDmChannels is a helper method to define mock.On call -// - ctx context.Context -// - in *datapb.WatchDmChannelsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *datapb.WatchDmChannelsRequest +// - opts ...grpc.CallOption func (_e *MockDataNodeClient_Expecter) WatchDmChannels(ctx interface{}, in interface{}, opts ...interface{}) *MockDataNodeClient_WatchDmChannels_Call { return &MockDataNodeClient_WatchDmChannels_Call{Call: _e.mock.On("WatchDmChannels", append([]interface{}{ctx, in}, opts...)...)} diff --git a/internal/mocks/mock_indexnode.go b/internal/mocks/mock_indexnode.go index bcd158dc7b34e..a86a1a99704bd 100644 --- a/internal/mocks/mock_indexnode.go +++ b/internal/mocks/mock_indexnode.go @@ -62,8 +62,8 @@ type MockIndexNode_CreateJob_Call struct { } // CreateJob is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *indexpb.CreateJobRequest +// - _a0 context.Context +// - _a1 *indexpb.CreateJobRequest func (_e *MockIndexNode_Expecter) CreateJob(_a0 interface{}, _a1 interface{}) *MockIndexNode_CreateJob_Call { return &MockIndexNode_CreateJob_Call{Call: _e.mock.On("CreateJob", _a0, _a1)} } @@ -117,8 +117,8 @@ type MockIndexNode_DropJobs_Call struct { } // DropJobs is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *indexpb.DropJobsRequest +// - _a0 context.Context +// - _a1 *indexpb.DropJobsRequest func (_e *MockIndexNode_Expecter) DropJobs(_a0 interface{}, _a1 interface{}) *MockIndexNode_DropJobs_Call { return &MockIndexNode_DropJobs_Call{Call: _e.mock.On("DropJobs", _a0, _a1)} } @@ -213,8 +213,8 @@ type MockIndexNode_GetComponentStates_Call struct { } // GetComponentStates is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetComponentStatesRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetComponentStatesRequest func (_e *MockIndexNode_Expecter) GetComponentStates(_a0 interface{}, _a1 interface{}) *MockIndexNode_GetComponentStates_Call { return &MockIndexNode_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", _a0, _a1)} } @@ -268,8 +268,8 @@ type MockIndexNode_GetJobStats_Call struct { } // GetJobStats is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *indexpb.GetJobStatsRequest +// - _a0 context.Context +// - _a1 *indexpb.GetJobStatsRequest func (_e *MockIndexNode_Expecter) GetJobStats(_a0 interface{}, _a1 interface{}) *MockIndexNode_GetJobStats_Call { return &MockIndexNode_GetJobStats_Call{Call: _e.mock.On("GetJobStats", _a0, _a1)} } @@ -323,8 +323,8 @@ type MockIndexNode_GetMetrics_Call struct { } // GetMetrics is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetMetricsRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetMetricsRequest func (_e *MockIndexNode_Expecter) GetMetrics(_a0 interface{}, _a1 interface{}) *MockIndexNode_GetMetrics_Call { return &MockIndexNode_GetMetrics_Call{Call: _e.mock.On("GetMetrics", _a0, _a1)} } @@ -378,8 +378,8 @@ type MockIndexNode_GetStatisticsChannel_Call struct { } // GetStatisticsChannel is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.GetStatisticsChannelRequest +// - _a0 context.Context +// - _a1 *internalpb.GetStatisticsChannelRequest func (_e *MockIndexNode_Expecter) GetStatisticsChannel(_a0 interface{}, _a1 interface{}) *MockIndexNode_GetStatisticsChannel_Call { return &MockIndexNode_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", _a0, _a1)} } @@ -474,8 +474,8 @@ type MockIndexNode_QueryJobs_Call struct { } // QueryJobs is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *indexpb.QueryJobsRequest +// - _a0 context.Context +// - _a1 *indexpb.QueryJobsRequest func (_e *MockIndexNode_Expecter) QueryJobs(_a0 interface{}, _a1 interface{}) *MockIndexNode_QueryJobs_Call { return &MockIndexNode_QueryJobs_Call{Call: _e.mock.On("QueryJobs", _a0, _a1)} } @@ -549,7 +549,7 @@ type MockIndexNode_SetAddress_Call struct { } // SetAddress is a helper method to define mock.On call -// - address string +// - address string func (_e *MockIndexNode_Expecter) SetAddress(address interface{}) *MockIndexNode_SetAddress_Call { return &MockIndexNode_SetAddress_Call{Call: _e.mock.On("SetAddress", address)} } @@ -582,7 +582,7 @@ type MockIndexNode_SetEtcdClient_Call struct { } // SetEtcdClient is a helper method to define mock.On call -// - etcdClient *clientv3.Client +// - etcdClient *clientv3.Client func (_e *MockIndexNode_Expecter) SetEtcdClient(etcdClient interface{}) *MockIndexNode_SetEtcdClient_Call { return &MockIndexNode_SetEtcdClient_Call{Call: _e.mock.On("SetEtcdClient", etcdClient)} } @@ -636,8 +636,8 @@ type MockIndexNode_ShowConfigurations_Call struct { } // ShowConfigurations is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.ShowConfigurationsRequest +// - _a0 context.Context +// - _a1 *internalpb.ShowConfigurationsRequest func (_e *MockIndexNode_Expecter) ShowConfigurations(_a0 interface{}, _a1 interface{}) *MockIndexNode_ShowConfigurations_Call { return &MockIndexNode_ShowConfigurations_Call{Call: _e.mock.On("ShowConfigurations", _a0, _a1)} } @@ -752,7 +752,7 @@ type MockIndexNode_UpdateStateCode_Call struct { } // UpdateStateCode is a helper method to define mock.On call -// - stateCode commonpb.StateCode +// - stateCode commonpb.StateCode func (_e *MockIndexNode_Expecter) UpdateStateCode(stateCode interface{}) *MockIndexNode_UpdateStateCode_Call { return &MockIndexNode_UpdateStateCode_Call{Call: _e.mock.On("UpdateStateCode", stateCode)} } diff --git a/internal/mocks/mock_indexnode_client.go b/internal/mocks/mock_indexnode_client.go index 1e30de98ac1bd..f300843780687 100644 --- a/internal/mocks/mock_indexnode_client.go +++ b/internal/mocks/mock_indexnode_client.go @@ -111,9 +111,9 @@ type MockIndexNodeClient_CreateJob_Call struct { } // CreateJob is a helper method to define mock.On call -// - ctx context.Context -// - in *indexpb.CreateJobRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *indexpb.CreateJobRequest +// - opts ...grpc.CallOption func (_e *MockIndexNodeClient_Expecter) CreateJob(ctx interface{}, in interface{}, opts ...interface{}) *MockIndexNodeClient_CreateJob_Call { return &MockIndexNodeClient_CreateJob_Call{Call: _e.mock.On("CreateJob", append([]interface{}{ctx, in}, opts...)...)} @@ -181,9 +181,9 @@ type MockIndexNodeClient_DropJobs_Call struct { } // DropJobs is a helper method to define mock.On call -// - ctx context.Context -// - in *indexpb.DropJobsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *indexpb.DropJobsRequest +// - opts ...grpc.CallOption func (_e *MockIndexNodeClient_Expecter) DropJobs(ctx interface{}, in interface{}, opts ...interface{}) *MockIndexNodeClient_DropJobs_Call { return &MockIndexNodeClient_DropJobs_Call{Call: _e.mock.On("DropJobs", append([]interface{}{ctx, in}, opts...)...)} @@ -251,9 +251,9 @@ type MockIndexNodeClient_GetComponentStates_Call struct { } // GetComponentStates is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetComponentStatesRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetComponentStatesRequest +// - opts ...grpc.CallOption func (_e *MockIndexNodeClient_Expecter) GetComponentStates(ctx interface{}, in interface{}, opts ...interface{}) *MockIndexNodeClient_GetComponentStates_Call { return &MockIndexNodeClient_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", append([]interface{}{ctx, in}, opts...)...)} @@ -321,9 +321,9 @@ type MockIndexNodeClient_GetJobStats_Call struct { } // GetJobStats is a helper method to define mock.On call -// - ctx context.Context -// - in *indexpb.GetJobStatsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *indexpb.GetJobStatsRequest +// - opts ...grpc.CallOption func (_e *MockIndexNodeClient_Expecter) GetJobStats(ctx interface{}, in interface{}, opts ...interface{}) *MockIndexNodeClient_GetJobStats_Call { return &MockIndexNodeClient_GetJobStats_Call{Call: _e.mock.On("GetJobStats", append([]interface{}{ctx, in}, opts...)...)} @@ -391,9 +391,9 @@ type MockIndexNodeClient_GetMetrics_Call struct { } // GetMetrics is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetMetricsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetMetricsRequest +// - opts ...grpc.CallOption func (_e *MockIndexNodeClient_Expecter) GetMetrics(ctx interface{}, in interface{}, opts ...interface{}) *MockIndexNodeClient_GetMetrics_Call { return &MockIndexNodeClient_GetMetrics_Call{Call: _e.mock.On("GetMetrics", append([]interface{}{ctx, in}, opts...)...)} @@ -461,9 +461,9 @@ type MockIndexNodeClient_GetStatisticsChannel_Call struct { } // GetStatisticsChannel is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.GetStatisticsChannelRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.GetStatisticsChannelRequest +// - opts ...grpc.CallOption func (_e *MockIndexNodeClient_Expecter) GetStatisticsChannel(ctx interface{}, in interface{}, opts ...interface{}) *MockIndexNodeClient_GetStatisticsChannel_Call { return &MockIndexNodeClient_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", append([]interface{}{ctx, in}, opts...)...)} @@ -531,9 +531,9 @@ type MockIndexNodeClient_QueryJobs_Call struct { } // QueryJobs is a helper method to define mock.On call -// - ctx context.Context -// - in *indexpb.QueryJobsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *indexpb.QueryJobsRequest +// - opts ...grpc.CallOption func (_e *MockIndexNodeClient_Expecter) QueryJobs(ctx interface{}, in interface{}, opts ...interface{}) *MockIndexNodeClient_QueryJobs_Call { return &MockIndexNodeClient_QueryJobs_Call{Call: _e.mock.On("QueryJobs", append([]interface{}{ctx, in}, opts...)...)} @@ -601,9 +601,9 @@ type MockIndexNodeClient_ShowConfigurations_Call struct { } // ShowConfigurations is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.ShowConfigurationsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.ShowConfigurationsRequest +// - opts ...grpc.CallOption func (_e *MockIndexNodeClient_Expecter) ShowConfigurations(ctx interface{}, in interface{}, opts ...interface{}) *MockIndexNodeClient_ShowConfigurations_Call { return &MockIndexNodeClient_ShowConfigurations_Call{Call: _e.mock.On("ShowConfigurations", append([]interface{}{ctx, in}, opts...)...)} diff --git a/internal/mocks/mock_proxy_client.go b/internal/mocks/mock_proxy_client.go index 0d74d3cd46e30..a1d0dcc48dad6 100644 --- a/internal/mocks/mock_proxy_client.go +++ b/internal/mocks/mock_proxy_client.go @@ -111,9 +111,9 @@ type MockProxyClient_GetComponentStates_Call struct { } // GetComponentStates is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetComponentStatesRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetComponentStatesRequest +// - opts ...grpc.CallOption func (_e *MockProxyClient_Expecter) GetComponentStates(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_GetComponentStates_Call { return &MockProxyClient_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", append([]interface{}{ctx, in}, opts...)...)} @@ -181,9 +181,9 @@ type MockProxyClient_GetDdChannel_Call struct { } // GetDdChannel is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.GetDdChannelRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.GetDdChannelRequest +// - opts ...grpc.CallOption func (_e *MockProxyClient_Expecter) GetDdChannel(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_GetDdChannel_Call { return &MockProxyClient_GetDdChannel_Call{Call: _e.mock.On("GetDdChannel", append([]interface{}{ctx, in}, opts...)...)} @@ -212,6 +212,76 @@ func (_c *MockProxyClient_GetDdChannel_Call) RunAndReturn(run func(context.Conte return _c } +// GetImportProgress provides a mock function with given fields: ctx, in, opts +func (_m *MockProxyClient) GetImportProgress(ctx context.Context, in *internalpb.GetImportProgressRequest, opts ...grpc.CallOption) (*internalpb.GetImportProgressResponse, error) { + _va := make([]interface{}, len(opts)) + for _i := range opts { + _va[_i] = opts[_i] + } + var _ca []interface{} + _ca = append(_ca, ctx, in) + _ca = append(_ca, _va...) + ret := _m.Called(_ca...) + + var r0 *internalpb.GetImportProgressResponse + var r1 error + if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetImportProgressRequest, ...grpc.CallOption) (*internalpb.GetImportProgressResponse, error)); ok { + return rf(ctx, in, opts...) + } + if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetImportProgressRequest, ...grpc.CallOption) *internalpb.GetImportProgressResponse); ok { + r0 = rf(ctx, in, opts...) + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(*internalpb.GetImportProgressResponse) + } + } + + if rf, ok := ret.Get(1).(func(context.Context, *internalpb.GetImportProgressRequest, ...grpc.CallOption) error); ok { + r1 = rf(ctx, in, opts...) + } else { + r1 = ret.Error(1) + } + + return r0, r1 +} + +// MockProxyClient_GetImportProgress_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetImportProgress' +type MockProxyClient_GetImportProgress_Call struct { + *mock.Call +} + +// GetImportProgress is a helper method to define mock.On call +// - ctx context.Context +// - in *internalpb.GetImportProgressRequest +// - opts ...grpc.CallOption +func (_e *MockProxyClient_Expecter) GetImportProgress(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_GetImportProgress_Call { + return &MockProxyClient_GetImportProgress_Call{Call: _e.mock.On("GetImportProgress", + append([]interface{}{ctx, in}, opts...)...)} +} + +func (_c *MockProxyClient_GetImportProgress_Call) Run(run func(ctx context.Context, in *internalpb.GetImportProgressRequest, opts ...grpc.CallOption)) *MockProxyClient_GetImportProgress_Call { + _c.Call.Run(func(args mock.Arguments) { + variadicArgs := make([]grpc.CallOption, len(args)-2) + for i, a := range args[2:] { + if a != nil { + variadicArgs[i] = a.(grpc.CallOption) + } + } + run(args[0].(context.Context), args[1].(*internalpb.GetImportProgressRequest), variadicArgs...) + }) + return _c +} + +func (_c *MockProxyClient_GetImportProgress_Call) Return(_a0 *internalpb.GetImportProgressResponse, _a1 error) *MockProxyClient_GetImportProgress_Call { + _c.Call.Return(_a0, _a1) + return _c +} + +func (_c *MockProxyClient_GetImportProgress_Call) RunAndReturn(run func(context.Context, *internalpb.GetImportProgressRequest, ...grpc.CallOption) (*internalpb.GetImportProgressResponse, error)) *MockProxyClient_GetImportProgress_Call { + _c.Call.Return(run) + return _c +} + // GetProxyMetrics provides a mock function with given fields: ctx, in, opts func (_m *MockProxyClient) GetProxyMetrics(ctx context.Context, in *milvuspb.GetMetricsRequest, opts ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error) { _va := make([]interface{}, len(opts)) @@ -251,9 +321,9 @@ type MockProxyClient_GetProxyMetrics_Call struct { } // GetProxyMetrics is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetMetricsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetMetricsRequest +// - opts ...grpc.CallOption func (_e *MockProxyClient_Expecter) GetProxyMetrics(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_GetProxyMetrics_Call { return &MockProxyClient_GetProxyMetrics_Call{Call: _e.mock.On("GetProxyMetrics", append([]interface{}{ctx, in}, opts...)...)} @@ -321,9 +391,9 @@ type MockProxyClient_GetStatisticsChannel_Call struct { } // GetStatisticsChannel is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.GetStatisticsChannelRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.GetStatisticsChannelRequest +// - opts ...grpc.CallOption func (_e *MockProxyClient_Expecter) GetStatisticsChannel(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_GetStatisticsChannel_Call { return &MockProxyClient_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", append([]interface{}{ctx, in}, opts...)...)} @@ -352,6 +422,76 @@ func (_c *MockProxyClient_GetStatisticsChannel_Call) RunAndReturn(run func(conte return _c } +// ImportV2 provides a mock function with given fields: ctx, in, opts +func (_m *MockProxyClient) ImportV2(ctx context.Context, in *internalpb.ImportRequest, opts ...grpc.CallOption) (*internalpb.ImportResponse, error) { + _va := make([]interface{}, len(opts)) + for _i := range opts { + _va[_i] = opts[_i] + } + var _ca []interface{} + _ca = append(_ca, ctx, in) + _ca = append(_ca, _va...) + ret := _m.Called(_ca...) + + var r0 *internalpb.ImportResponse + var r1 error + if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ImportRequest, ...grpc.CallOption) (*internalpb.ImportResponse, error)); ok { + return rf(ctx, in, opts...) + } + if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ImportRequest, ...grpc.CallOption) *internalpb.ImportResponse); ok { + r0 = rf(ctx, in, opts...) + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(*internalpb.ImportResponse) + } + } + + if rf, ok := ret.Get(1).(func(context.Context, *internalpb.ImportRequest, ...grpc.CallOption) error); ok { + r1 = rf(ctx, in, opts...) + } else { + r1 = ret.Error(1) + } + + return r0, r1 +} + +// MockProxyClient_ImportV2_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ImportV2' +type MockProxyClient_ImportV2_Call struct { + *mock.Call +} + +// ImportV2 is a helper method to define mock.On call +// - ctx context.Context +// - in *internalpb.ImportRequest +// - opts ...grpc.CallOption +func (_e *MockProxyClient_Expecter) ImportV2(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_ImportV2_Call { + return &MockProxyClient_ImportV2_Call{Call: _e.mock.On("ImportV2", + append([]interface{}{ctx, in}, opts...)...)} +} + +func (_c *MockProxyClient_ImportV2_Call) Run(run func(ctx context.Context, in *internalpb.ImportRequest, opts ...grpc.CallOption)) *MockProxyClient_ImportV2_Call { + _c.Call.Run(func(args mock.Arguments) { + variadicArgs := make([]grpc.CallOption, len(args)-2) + for i, a := range args[2:] { + if a != nil { + variadicArgs[i] = a.(grpc.CallOption) + } + } + run(args[0].(context.Context), args[1].(*internalpb.ImportRequest), variadicArgs...) + }) + return _c +} + +func (_c *MockProxyClient_ImportV2_Call) Return(_a0 *internalpb.ImportResponse, _a1 error) *MockProxyClient_ImportV2_Call { + _c.Call.Return(_a0, _a1) + return _c +} + +func (_c *MockProxyClient_ImportV2_Call) RunAndReturn(run func(context.Context, *internalpb.ImportRequest, ...grpc.CallOption) (*internalpb.ImportResponse, error)) *MockProxyClient_ImportV2_Call { + _c.Call.Return(run) + return _c +} + // InvalidateCollectionMetaCache provides a mock function with given fields: ctx, in, opts func (_m *MockProxyClient) InvalidateCollectionMetaCache(ctx context.Context, in *proxypb.InvalidateCollMetaCacheRequest, opts ...grpc.CallOption) (*commonpb.Status, error) { _va := make([]interface{}, len(opts)) @@ -391,9 +531,9 @@ type MockProxyClient_InvalidateCollectionMetaCache_Call struct { } // InvalidateCollectionMetaCache is a helper method to define mock.On call -// - ctx context.Context -// - in *proxypb.InvalidateCollMetaCacheRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *proxypb.InvalidateCollMetaCacheRequest +// - opts ...grpc.CallOption func (_e *MockProxyClient_Expecter) InvalidateCollectionMetaCache(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_InvalidateCollectionMetaCache_Call { return &MockProxyClient_InvalidateCollectionMetaCache_Call{Call: _e.mock.On("InvalidateCollectionMetaCache", append([]interface{}{ctx, in}, opts...)...)} @@ -461,9 +601,9 @@ type MockProxyClient_InvalidateCredentialCache_Call struct { } // InvalidateCredentialCache is a helper method to define mock.On call -// - ctx context.Context -// - in *proxypb.InvalidateCredCacheRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *proxypb.InvalidateCredCacheRequest +// - opts ...grpc.CallOption func (_e *MockProxyClient_Expecter) InvalidateCredentialCache(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_InvalidateCredentialCache_Call { return &MockProxyClient_InvalidateCredentialCache_Call{Call: _e.mock.On("InvalidateCredentialCache", append([]interface{}{ctx, in}, opts...)...)} @@ -531,9 +671,9 @@ type MockProxyClient_ListClientInfos_Call struct { } // ListClientInfos is a helper method to define mock.On call -// - ctx context.Context -// - in *proxypb.ListClientInfosRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *proxypb.ListClientInfosRequest +// - opts ...grpc.CallOption func (_e *MockProxyClient_Expecter) ListClientInfos(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_ListClientInfos_Call { return &MockProxyClient_ListClientInfos_Call{Call: _e.mock.On("ListClientInfos", append([]interface{}{ctx, in}, opts...)...)} @@ -562,6 +702,76 @@ func (_c *MockProxyClient_ListClientInfos_Call) RunAndReturn(run func(context.Co return _c } +// ListImports provides a mock function with given fields: ctx, in, opts +func (_m *MockProxyClient) ListImports(ctx context.Context, in *internalpb.ListImportsRequest, opts ...grpc.CallOption) (*internalpb.ListImportsResponse, error) { + _va := make([]interface{}, len(opts)) + for _i := range opts { + _va[_i] = opts[_i] + } + var _ca []interface{} + _ca = append(_ca, ctx, in) + _ca = append(_ca, _va...) + ret := _m.Called(_ca...) + + var r0 *internalpb.ListImportsResponse + var r1 error + if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequest, ...grpc.CallOption) (*internalpb.ListImportsResponse, error)); ok { + return rf(ctx, in, opts...) + } + if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequest, ...grpc.CallOption) *internalpb.ListImportsResponse); ok { + r0 = rf(ctx, in, opts...) + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(*internalpb.ListImportsResponse) + } + } + + if rf, ok := ret.Get(1).(func(context.Context, *internalpb.ListImportsRequest, ...grpc.CallOption) error); ok { + r1 = rf(ctx, in, opts...) + } else { + r1 = ret.Error(1) + } + + return r0, r1 +} + +// MockProxyClient_ListImports_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListImports' +type MockProxyClient_ListImports_Call struct { + *mock.Call +} + +// ListImports is a helper method to define mock.On call +// - ctx context.Context +// - in *internalpb.ListImportsRequest +// - opts ...grpc.CallOption +func (_e *MockProxyClient_Expecter) ListImports(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_ListImports_Call { + return &MockProxyClient_ListImports_Call{Call: _e.mock.On("ListImports", + append([]interface{}{ctx, in}, opts...)...)} +} + +func (_c *MockProxyClient_ListImports_Call) Run(run func(ctx context.Context, in *internalpb.ListImportsRequest, opts ...grpc.CallOption)) *MockProxyClient_ListImports_Call { + _c.Call.Run(func(args mock.Arguments) { + variadicArgs := make([]grpc.CallOption, len(args)-2) + for i, a := range args[2:] { + if a != nil { + variadicArgs[i] = a.(grpc.CallOption) + } + } + run(args[0].(context.Context), args[1].(*internalpb.ListImportsRequest), variadicArgs...) + }) + return _c +} + +func (_c *MockProxyClient_ListImports_Call) Return(_a0 *internalpb.ListImportsResponse, _a1 error) *MockProxyClient_ListImports_Call { + _c.Call.Return(_a0, _a1) + return _c +} + +func (_c *MockProxyClient_ListImports_Call) RunAndReturn(run func(context.Context, *internalpb.ListImportsRequest, ...grpc.CallOption) (*internalpb.ListImportsResponse, error)) *MockProxyClient_ListImports_Call { + _c.Call.Return(run) + return _c +} + // RefreshPolicyInfoCache provides a mock function with given fields: ctx, in, opts func (_m *MockProxyClient) RefreshPolicyInfoCache(ctx context.Context, in *proxypb.RefreshPolicyInfoCacheRequest, opts ...grpc.CallOption) (*commonpb.Status, error) { _va := make([]interface{}, len(opts)) @@ -601,9 +811,9 @@ type MockProxyClient_RefreshPolicyInfoCache_Call struct { } // RefreshPolicyInfoCache is a helper method to define mock.On call -// - ctx context.Context -// - in *proxypb.RefreshPolicyInfoCacheRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *proxypb.RefreshPolicyInfoCacheRequest +// - opts ...grpc.CallOption func (_e *MockProxyClient_Expecter) RefreshPolicyInfoCache(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_RefreshPolicyInfoCache_Call { return &MockProxyClient_RefreshPolicyInfoCache_Call{Call: _e.mock.On("RefreshPolicyInfoCache", append([]interface{}{ctx, in}, opts...)...)} @@ -671,9 +881,9 @@ type MockProxyClient_SetRates_Call struct { } // SetRates is a helper method to define mock.On call -// - ctx context.Context -// - in *proxypb.SetRatesRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *proxypb.SetRatesRequest +// - opts ...grpc.CallOption func (_e *MockProxyClient_Expecter) SetRates(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_SetRates_Call { return &MockProxyClient_SetRates_Call{Call: _e.mock.On("SetRates", append([]interface{}{ctx, in}, opts...)...)} @@ -741,9 +951,9 @@ type MockProxyClient_UpdateCredentialCache_Call struct { } // UpdateCredentialCache is a helper method to define mock.On call -// - ctx context.Context -// - in *proxypb.UpdateCredCacheRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *proxypb.UpdateCredCacheRequest +// - opts ...grpc.CallOption func (_e *MockProxyClient_Expecter) UpdateCredentialCache(ctx interface{}, in interface{}, opts ...interface{}) *MockProxyClient_UpdateCredentialCache_Call { return &MockProxyClient_UpdateCredentialCache_Call{Call: _e.mock.On("UpdateCredentialCache", append([]interface{}{ctx, in}, opts...)...)} diff --git a/internal/mocks/mock_querycoord.go b/internal/mocks/mock_querycoord.go index 2fd90e209595a..c4ad51d91f510 100644 --- a/internal/mocks/mock_querycoord.go +++ b/internal/mocks/mock_querycoord.go @@ -66,8 +66,8 @@ type MockQueryCoord_ActivateChecker_Call struct { } // ActivateChecker is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.ActivateCheckerRequest +// - _a0 context.Context +// - _a1 *querypb.ActivateCheckerRequest func (_e *MockQueryCoord_Expecter) ActivateChecker(_a0 interface{}, _a1 interface{}) *MockQueryCoord_ActivateChecker_Call { return &MockQueryCoord_ActivateChecker_Call{Call: _e.mock.On("ActivateChecker", _a0, _a1)} } @@ -121,8 +121,8 @@ type MockQueryCoord_CheckHealth_Call struct { } // CheckHealth is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.CheckHealthRequest +// - _a0 context.Context +// - _a1 *milvuspb.CheckHealthRequest func (_e *MockQueryCoord_Expecter) CheckHealth(_a0 interface{}, _a1 interface{}) *MockQueryCoord_CheckHealth_Call { return &MockQueryCoord_CheckHealth_Call{Call: _e.mock.On("CheckHealth", _a0, _a1)} } @@ -176,8 +176,8 @@ type MockQueryCoord_CreateResourceGroup_Call struct { } // CreateResourceGroup is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.CreateResourceGroupRequest +// - _a0 context.Context +// - _a1 *milvuspb.CreateResourceGroupRequest func (_e *MockQueryCoord_Expecter) CreateResourceGroup(_a0 interface{}, _a1 interface{}) *MockQueryCoord_CreateResourceGroup_Call { return &MockQueryCoord_CreateResourceGroup_Call{Call: _e.mock.On("CreateResourceGroup", _a0, _a1)} } @@ -231,8 +231,8 @@ type MockQueryCoord_DeactivateChecker_Call struct { } // DeactivateChecker is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.DeactivateCheckerRequest +// - _a0 context.Context +// - _a1 *querypb.DeactivateCheckerRequest func (_e *MockQueryCoord_Expecter) DeactivateChecker(_a0 interface{}, _a1 interface{}) *MockQueryCoord_DeactivateChecker_Call { return &MockQueryCoord_DeactivateChecker_Call{Call: _e.mock.On("DeactivateChecker", _a0, _a1)} } @@ -286,8 +286,8 @@ type MockQueryCoord_DescribeResourceGroup_Call struct { } // DescribeResourceGroup is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.DescribeResourceGroupRequest +// - _a0 context.Context +// - _a1 *querypb.DescribeResourceGroupRequest func (_e *MockQueryCoord_Expecter) DescribeResourceGroup(_a0 interface{}, _a1 interface{}) *MockQueryCoord_DescribeResourceGroup_Call { return &MockQueryCoord_DescribeResourceGroup_Call{Call: _e.mock.On("DescribeResourceGroup", _a0, _a1)} } @@ -341,8 +341,8 @@ type MockQueryCoord_DropResourceGroup_Call struct { } // DropResourceGroup is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DropResourceGroupRequest +// - _a0 context.Context +// - _a1 *milvuspb.DropResourceGroupRequest func (_e *MockQueryCoord_Expecter) DropResourceGroup(_a0 interface{}, _a1 interface{}) *MockQueryCoord_DropResourceGroup_Call { return &MockQueryCoord_DropResourceGroup_Call{Call: _e.mock.On("DropResourceGroup", _a0, _a1)} } @@ -396,8 +396,8 @@ type MockQueryCoord_GetComponentStates_Call struct { } // GetComponentStates is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetComponentStatesRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetComponentStatesRequest func (_e *MockQueryCoord_Expecter) GetComponentStates(_a0 interface{}, _a1 interface{}) *MockQueryCoord_GetComponentStates_Call { return &MockQueryCoord_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", _a0, _a1)} } @@ -451,8 +451,8 @@ type MockQueryCoord_GetMetrics_Call struct { } // GetMetrics is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetMetricsRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetMetricsRequest func (_e *MockQueryCoord_Expecter) GetMetrics(_a0 interface{}, _a1 interface{}) *MockQueryCoord_GetMetrics_Call { return &MockQueryCoord_GetMetrics_Call{Call: _e.mock.On("GetMetrics", _a0, _a1)} } @@ -506,8 +506,8 @@ type MockQueryCoord_GetPartitionStates_Call struct { } // GetPartitionStates is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.GetPartitionStatesRequest +// - _a0 context.Context +// - _a1 *querypb.GetPartitionStatesRequest func (_e *MockQueryCoord_Expecter) GetPartitionStates(_a0 interface{}, _a1 interface{}) *MockQueryCoord_GetPartitionStates_Call { return &MockQueryCoord_GetPartitionStates_Call{Call: _e.mock.On("GetPartitionStates", _a0, _a1)} } @@ -561,8 +561,8 @@ type MockQueryCoord_GetReplicas_Call struct { } // GetReplicas is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetReplicasRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetReplicasRequest func (_e *MockQueryCoord_Expecter) GetReplicas(_a0 interface{}, _a1 interface{}) *MockQueryCoord_GetReplicas_Call { return &MockQueryCoord_GetReplicas_Call{Call: _e.mock.On("GetReplicas", _a0, _a1)} } @@ -616,8 +616,8 @@ type MockQueryCoord_GetSegmentInfo_Call struct { } // GetSegmentInfo is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.GetSegmentInfoRequest +// - _a0 context.Context +// - _a1 *querypb.GetSegmentInfoRequest func (_e *MockQueryCoord_Expecter) GetSegmentInfo(_a0 interface{}, _a1 interface{}) *MockQueryCoord_GetSegmentInfo_Call { return &MockQueryCoord_GetSegmentInfo_Call{Call: _e.mock.On("GetSegmentInfo", _a0, _a1)} } @@ -671,8 +671,8 @@ type MockQueryCoord_GetShardLeaders_Call struct { } // GetShardLeaders is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.GetShardLeadersRequest +// - _a0 context.Context +// - _a1 *querypb.GetShardLeadersRequest func (_e *MockQueryCoord_Expecter) GetShardLeaders(_a0 interface{}, _a1 interface{}) *MockQueryCoord_GetShardLeaders_Call { return &MockQueryCoord_GetShardLeaders_Call{Call: _e.mock.On("GetShardLeaders", _a0, _a1)} } @@ -726,8 +726,8 @@ type MockQueryCoord_GetStatisticsChannel_Call struct { } // GetStatisticsChannel is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.GetStatisticsChannelRequest +// - _a0 context.Context +// - _a1 *internalpb.GetStatisticsChannelRequest func (_e *MockQueryCoord_Expecter) GetStatisticsChannel(_a0 interface{}, _a1 interface{}) *MockQueryCoord_GetStatisticsChannel_Call { return &MockQueryCoord_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", _a0, _a1)} } @@ -781,8 +781,8 @@ type MockQueryCoord_GetTimeTickChannel_Call struct { } // GetTimeTickChannel is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.GetTimeTickChannelRequest +// - _a0 context.Context +// - _a1 *internalpb.GetTimeTickChannelRequest func (_e *MockQueryCoord_Expecter) GetTimeTickChannel(_a0 interface{}, _a1 interface{}) *MockQueryCoord_GetTimeTickChannel_Call { return &MockQueryCoord_GetTimeTickChannel_Call{Call: _e.mock.On("GetTimeTickChannel", _a0, _a1)} } @@ -877,8 +877,8 @@ type MockQueryCoord_ListCheckers_Call struct { } // ListCheckers is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.ListCheckersRequest +// - _a0 context.Context +// - _a1 *querypb.ListCheckersRequest func (_e *MockQueryCoord_Expecter) ListCheckers(_a0 interface{}, _a1 interface{}) *MockQueryCoord_ListCheckers_Call { return &MockQueryCoord_ListCheckers_Call{Call: _e.mock.On("ListCheckers", _a0, _a1)} } @@ -932,8 +932,8 @@ type MockQueryCoord_ListResourceGroups_Call struct { } // ListResourceGroups is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ListResourceGroupsRequest +// - _a0 context.Context +// - _a1 *milvuspb.ListResourceGroupsRequest func (_e *MockQueryCoord_Expecter) ListResourceGroups(_a0 interface{}, _a1 interface{}) *MockQueryCoord_ListResourceGroups_Call { return &MockQueryCoord_ListResourceGroups_Call{Call: _e.mock.On("ListResourceGroups", _a0, _a1)} } @@ -987,8 +987,8 @@ type MockQueryCoord_LoadBalance_Call struct { } // LoadBalance is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.LoadBalanceRequest +// - _a0 context.Context +// - _a1 *querypb.LoadBalanceRequest func (_e *MockQueryCoord_Expecter) LoadBalance(_a0 interface{}, _a1 interface{}) *MockQueryCoord_LoadBalance_Call { return &MockQueryCoord_LoadBalance_Call{Call: _e.mock.On("LoadBalance", _a0, _a1)} } @@ -1042,8 +1042,8 @@ type MockQueryCoord_LoadCollection_Call struct { } // LoadCollection is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.LoadCollectionRequest +// - _a0 context.Context +// - _a1 *querypb.LoadCollectionRequest func (_e *MockQueryCoord_Expecter) LoadCollection(_a0 interface{}, _a1 interface{}) *MockQueryCoord_LoadCollection_Call { return &MockQueryCoord_LoadCollection_Call{Call: _e.mock.On("LoadCollection", _a0, _a1)} } @@ -1097,8 +1097,8 @@ type MockQueryCoord_LoadPartitions_Call struct { } // LoadPartitions is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.LoadPartitionsRequest +// - _a0 context.Context +// - _a1 *querypb.LoadPartitionsRequest func (_e *MockQueryCoord_Expecter) LoadPartitions(_a0 interface{}, _a1 interface{}) *MockQueryCoord_LoadPartitions_Call { return &MockQueryCoord_LoadPartitions_Call{Call: _e.mock.On("LoadPartitions", _a0, _a1)} } @@ -1193,8 +1193,8 @@ type MockQueryCoord_ReleaseCollection_Call struct { } // ReleaseCollection is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.ReleaseCollectionRequest +// - _a0 context.Context +// - _a1 *querypb.ReleaseCollectionRequest func (_e *MockQueryCoord_Expecter) ReleaseCollection(_a0 interface{}, _a1 interface{}) *MockQueryCoord_ReleaseCollection_Call { return &MockQueryCoord_ReleaseCollection_Call{Call: _e.mock.On("ReleaseCollection", _a0, _a1)} } @@ -1248,8 +1248,8 @@ type MockQueryCoord_ReleasePartitions_Call struct { } // ReleasePartitions is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.ReleasePartitionsRequest +// - _a0 context.Context +// - _a1 *querypb.ReleasePartitionsRequest func (_e *MockQueryCoord_Expecter) ReleasePartitions(_a0 interface{}, _a1 interface{}) *MockQueryCoord_ReleasePartitions_Call { return &MockQueryCoord_ReleasePartitions_Call{Call: _e.mock.On("ReleasePartitions", _a0, _a1)} } @@ -1282,7 +1282,7 @@ type MockQueryCoord_SetAddress_Call struct { } // SetAddress is a helper method to define mock.On call -// - address string +// - address string func (_e *MockQueryCoord_Expecter) SetAddress(address interface{}) *MockQueryCoord_SetAddress_Call { return &MockQueryCoord_SetAddress_Call{Call: _e.mock.On("SetAddress", address)} } @@ -1324,7 +1324,7 @@ type MockQueryCoord_SetDataCoordClient_Call struct { } // SetDataCoordClient is a helper method to define mock.On call -// - dataCoord types.DataCoordClient +// - dataCoord types.DataCoordClient func (_e *MockQueryCoord_Expecter) SetDataCoordClient(dataCoord interface{}) *MockQueryCoord_SetDataCoordClient_Call { return &MockQueryCoord_SetDataCoordClient_Call{Call: _e.mock.On("SetDataCoordClient", dataCoord)} } @@ -1357,7 +1357,7 @@ type MockQueryCoord_SetEtcdClient_Call struct { } // SetEtcdClient is a helper method to define mock.On call -// - etcdClient *clientv3.Client +// - etcdClient *clientv3.Client func (_e *MockQueryCoord_Expecter) SetEtcdClient(etcdClient interface{}) *MockQueryCoord_SetEtcdClient_Call { return &MockQueryCoord_SetEtcdClient_Call{Call: _e.mock.On("SetEtcdClient", etcdClient)} } @@ -1390,7 +1390,7 @@ type MockQueryCoord_SetQueryNodeCreator_Call struct { } // SetQueryNodeCreator is a helper method to define mock.On call -// - _a0 func(context.Context , string , int64)(types.QueryNodeClient , error) +// - _a0 func(context.Context , string , int64)(types.QueryNodeClient , error) func (_e *MockQueryCoord_Expecter) SetQueryNodeCreator(_a0 interface{}) *MockQueryCoord_SetQueryNodeCreator_Call { return &MockQueryCoord_SetQueryNodeCreator_Call{Call: _e.mock.On("SetQueryNodeCreator", _a0)} } @@ -1432,7 +1432,7 @@ type MockQueryCoord_SetRootCoordClient_Call struct { } // SetRootCoordClient is a helper method to define mock.On call -// - rootCoord types.RootCoordClient +// - rootCoord types.RootCoordClient func (_e *MockQueryCoord_Expecter) SetRootCoordClient(rootCoord interface{}) *MockQueryCoord_SetRootCoordClient_Call { return &MockQueryCoord_SetRootCoordClient_Call{Call: _e.mock.On("SetRootCoordClient", rootCoord)} } @@ -1465,7 +1465,7 @@ type MockQueryCoord_SetTiKVClient_Call struct { } // SetTiKVClient is a helper method to define mock.On call -// - client *txnkv.Client +// - client *txnkv.Client func (_e *MockQueryCoord_Expecter) SetTiKVClient(client interface{}) *MockQueryCoord_SetTiKVClient_Call { return &MockQueryCoord_SetTiKVClient_Call{Call: _e.mock.On("SetTiKVClient", client)} } @@ -1519,8 +1519,8 @@ type MockQueryCoord_ShowCollections_Call struct { } // ShowCollections is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.ShowCollectionsRequest +// - _a0 context.Context +// - _a1 *querypb.ShowCollectionsRequest func (_e *MockQueryCoord_Expecter) ShowCollections(_a0 interface{}, _a1 interface{}) *MockQueryCoord_ShowCollections_Call { return &MockQueryCoord_ShowCollections_Call{Call: _e.mock.On("ShowCollections", _a0, _a1)} } @@ -1574,8 +1574,8 @@ type MockQueryCoord_ShowConfigurations_Call struct { } // ShowConfigurations is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.ShowConfigurationsRequest +// - _a0 context.Context +// - _a1 *internalpb.ShowConfigurationsRequest func (_e *MockQueryCoord_Expecter) ShowConfigurations(_a0 interface{}, _a1 interface{}) *MockQueryCoord_ShowConfigurations_Call { return &MockQueryCoord_ShowConfigurations_Call{Call: _e.mock.On("ShowConfigurations", _a0, _a1)} } @@ -1629,8 +1629,8 @@ type MockQueryCoord_ShowPartitions_Call struct { } // ShowPartitions is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.ShowPartitionsRequest +// - _a0 context.Context +// - _a1 *querypb.ShowPartitionsRequest func (_e *MockQueryCoord_Expecter) ShowPartitions(_a0 interface{}, _a1 interface{}) *MockQueryCoord_ShowPartitions_Call { return &MockQueryCoord_ShowPartitions_Call{Call: _e.mock.On("ShowPartitions", _a0, _a1)} } @@ -1766,8 +1766,8 @@ type MockQueryCoord_SyncNewCreatedPartition_Call struct { } // SyncNewCreatedPartition is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.SyncNewCreatedPartitionRequest +// - _a0 context.Context +// - _a1 *querypb.SyncNewCreatedPartitionRequest func (_e *MockQueryCoord_Expecter) SyncNewCreatedPartition(_a0 interface{}, _a1 interface{}) *MockQueryCoord_SyncNewCreatedPartition_Call { return &MockQueryCoord_SyncNewCreatedPartition_Call{Call: _e.mock.On("SyncNewCreatedPartition", _a0, _a1)} } @@ -1821,8 +1821,8 @@ type MockQueryCoord_TransferNode_Call struct { } // TransferNode is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.TransferNodeRequest +// - _a0 context.Context +// - _a1 *milvuspb.TransferNodeRequest func (_e *MockQueryCoord_Expecter) TransferNode(_a0 interface{}, _a1 interface{}) *MockQueryCoord_TransferNode_Call { return &MockQueryCoord_TransferNode_Call{Call: _e.mock.On("TransferNode", _a0, _a1)} } @@ -1876,8 +1876,8 @@ type MockQueryCoord_TransferReplica_Call struct { } // TransferReplica is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.TransferReplicaRequest +// - _a0 context.Context +// - _a1 *querypb.TransferReplicaRequest func (_e *MockQueryCoord_Expecter) TransferReplica(_a0 interface{}, _a1 interface{}) *MockQueryCoord_TransferReplica_Call { return &MockQueryCoord_TransferReplica_Call{Call: _e.mock.On("TransferReplica", _a0, _a1)} } @@ -1910,7 +1910,7 @@ type MockQueryCoord_UpdateStateCode_Call struct { } // UpdateStateCode is a helper method to define mock.On call -// - stateCode commonpb.StateCode +// - stateCode commonpb.StateCode func (_e *MockQueryCoord_Expecter) UpdateStateCode(stateCode interface{}) *MockQueryCoord_UpdateStateCode_Call { return &MockQueryCoord_UpdateStateCode_Call{Call: _e.mock.On("UpdateStateCode", stateCode)} } diff --git a/internal/mocks/mock_querycoord_client.go b/internal/mocks/mock_querycoord_client.go index 947bff1387e0c..fc2f385483e00 100644 --- a/internal/mocks/mock_querycoord_client.go +++ b/internal/mocks/mock_querycoord_client.go @@ -70,9 +70,9 @@ type MockQueryCoordClient_ActivateChecker_Call struct { } // ActivateChecker is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.ActivateCheckerRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.ActivateCheckerRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) ActivateChecker(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_ActivateChecker_Call { return &MockQueryCoordClient_ActivateChecker_Call{Call: _e.mock.On("ActivateChecker", append([]interface{}{ctx, in}, opts...)...)} @@ -140,9 +140,9 @@ type MockQueryCoordClient_CheckHealth_Call struct { } // CheckHealth is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.CheckHealthRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.CheckHealthRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) CheckHealth(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_CheckHealth_Call { return &MockQueryCoordClient_CheckHealth_Call{Call: _e.mock.On("CheckHealth", append([]interface{}{ctx, in}, opts...)...)} @@ -251,9 +251,9 @@ type MockQueryCoordClient_CreateResourceGroup_Call struct { } // CreateResourceGroup is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.CreateResourceGroupRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.CreateResourceGroupRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) CreateResourceGroup(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_CreateResourceGroup_Call { return &MockQueryCoordClient_CreateResourceGroup_Call{Call: _e.mock.On("CreateResourceGroup", append([]interface{}{ctx, in}, opts...)...)} @@ -321,9 +321,9 @@ type MockQueryCoordClient_DeactivateChecker_Call struct { } // DeactivateChecker is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.DeactivateCheckerRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.DeactivateCheckerRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) DeactivateChecker(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_DeactivateChecker_Call { return &MockQueryCoordClient_DeactivateChecker_Call{Call: _e.mock.On("DeactivateChecker", append([]interface{}{ctx, in}, opts...)...)} @@ -391,9 +391,9 @@ type MockQueryCoordClient_DescribeResourceGroup_Call struct { } // DescribeResourceGroup is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.DescribeResourceGroupRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.DescribeResourceGroupRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) DescribeResourceGroup(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_DescribeResourceGroup_Call { return &MockQueryCoordClient_DescribeResourceGroup_Call{Call: _e.mock.On("DescribeResourceGroup", append([]interface{}{ctx, in}, opts...)...)} @@ -461,9 +461,9 @@ type MockQueryCoordClient_DropResourceGroup_Call struct { } // DropResourceGroup is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.DropResourceGroupRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.DropResourceGroupRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) DropResourceGroup(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_DropResourceGroup_Call { return &MockQueryCoordClient_DropResourceGroup_Call{Call: _e.mock.On("DropResourceGroup", append([]interface{}{ctx, in}, opts...)...)} @@ -531,9 +531,9 @@ type MockQueryCoordClient_GetComponentStates_Call struct { } // GetComponentStates is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetComponentStatesRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetComponentStatesRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) GetComponentStates(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_GetComponentStates_Call { return &MockQueryCoordClient_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", append([]interface{}{ctx, in}, opts...)...)} @@ -601,9 +601,9 @@ type MockQueryCoordClient_GetMetrics_Call struct { } // GetMetrics is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetMetricsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetMetricsRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) GetMetrics(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_GetMetrics_Call { return &MockQueryCoordClient_GetMetrics_Call{Call: _e.mock.On("GetMetrics", append([]interface{}{ctx, in}, opts...)...)} @@ -671,9 +671,9 @@ type MockQueryCoordClient_GetPartitionStates_Call struct { } // GetPartitionStates is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.GetPartitionStatesRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.GetPartitionStatesRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) GetPartitionStates(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_GetPartitionStates_Call { return &MockQueryCoordClient_GetPartitionStates_Call{Call: _e.mock.On("GetPartitionStates", append([]interface{}{ctx, in}, opts...)...)} @@ -741,9 +741,9 @@ type MockQueryCoordClient_GetReplicas_Call struct { } // GetReplicas is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetReplicasRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetReplicasRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) GetReplicas(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_GetReplicas_Call { return &MockQueryCoordClient_GetReplicas_Call{Call: _e.mock.On("GetReplicas", append([]interface{}{ctx, in}, opts...)...)} @@ -811,9 +811,9 @@ type MockQueryCoordClient_GetSegmentInfo_Call struct { } // GetSegmentInfo is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.GetSegmentInfoRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.GetSegmentInfoRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) GetSegmentInfo(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_GetSegmentInfo_Call { return &MockQueryCoordClient_GetSegmentInfo_Call{Call: _e.mock.On("GetSegmentInfo", append([]interface{}{ctx, in}, opts...)...)} @@ -881,9 +881,9 @@ type MockQueryCoordClient_GetShardLeaders_Call struct { } // GetShardLeaders is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.GetShardLeadersRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.GetShardLeadersRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) GetShardLeaders(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_GetShardLeaders_Call { return &MockQueryCoordClient_GetShardLeaders_Call{Call: _e.mock.On("GetShardLeaders", append([]interface{}{ctx, in}, opts...)...)} @@ -951,9 +951,9 @@ type MockQueryCoordClient_GetStatisticsChannel_Call struct { } // GetStatisticsChannel is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.GetStatisticsChannelRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.GetStatisticsChannelRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) GetStatisticsChannel(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_GetStatisticsChannel_Call { return &MockQueryCoordClient_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", append([]interface{}{ctx, in}, opts...)...)} @@ -1021,9 +1021,9 @@ type MockQueryCoordClient_GetTimeTickChannel_Call struct { } // GetTimeTickChannel is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.GetTimeTickChannelRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.GetTimeTickChannelRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) GetTimeTickChannel(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_GetTimeTickChannel_Call { return &MockQueryCoordClient_GetTimeTickChannel_Call{Call: _e.mock.On("GetTimeTickChannel", append([]interface{}{ctx, in}, opts...)...)} @@ -1091,9 +1091,9 @@ type MockQueryCoordClient_ListCheckers_Call struct { } // ListCheckers is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.ListCheckersRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.ListCheckersRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) ListCheckers(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_ListCheckers_Call { return &MockQueryCoordClient_ListCheckers_Call{Call: _e.mock.On("ListCheckers", append([]interface{}{ctx, in}, opts...)...)} @@ -1161,9 +1161,9 @@ type MockQueryCoordClient_ListResourceGroups_Call struct { } // ListResourceGroups is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.ListResourceGroupsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.ListResourceGroupsRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) ListResourceGroups(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_ListResourceGroups_Call { return &MockQueryCoordClient_ListResourceGroups_Call{Call: _e.mock.On("ListResourceGroups", append([]interface{}{ctx, in}, opts...)...)} @@ -1231,9 +1231,9 @@ type MockQueryCoordClient_LoadBalance_Call struct { } // LoadBalance is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.LoadBalanceRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.LoadBalanceRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) LoadBalance(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_LoadBalance_Call { return &MockQueryCoordClient_LoadBalance_Call{Call: _e.mock.On("LoadBalance", append([]interface{}{ctx, in}, opts...)...)} @@ -1301,9 +1301,9 @@ type MockQueryCoordClient_LoadCollection_Call struct { } // LoadCollection is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.LoadCollectionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.LoadCollectionRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) LoadCollection(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_LoadCollection_Call { return &MockQueryCoordClient_LoadCollection_Call{Call: _e.mock.On("LoadCollection", append([]interface{}{ctx, in}, opts...)...)} @@ -1371,9 +1371,9 @@ type MockQueryCoordClient_LoadPartitions_Call struct { } // LoadPartitions is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.LoadPartitionsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.LoadPartitionsRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) LoadPartitions(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_LoadPartitions_Call { return &MockQueryCoordClient_LoadPartitions_Call{Call: _e.mock.On("LoadPartitions", append([]interface{}{ctx, in}, opts...)...)} @@ -1441,9 +1441,9 @@ type MockQueryCoordClient_ReleaseCollection_Call struct { } // ReleaseCollection is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.ReleaseCollectionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.ReleaseCollectionRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) ReleaseCollection(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_ReleaseCollection_Call { return &MockQueryCoordClient_ReleaseCollection_Call{Call: _e.mock.On("ReleaseCollection", append([]interface{}{ctx, in}, opts...)...)} @@ -1511,9 +1511,9 @@ type MockQueryCoordClient_ReleasePartitions_Call struct { } // ReleasePartitions is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.ReleasePartitionsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.ReleasePartitionsRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) ReleasePartitions(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_ReleasePartitions_Call { return &MockQueryCoordClient_ReleasePartitions_Call{Call: _e.mock.On("ReleasePartitions", append([]interface{}{ctx, in}, opts...)...)} @@ -1581,9 +1581,9 @@ type MockQueryCoordClient_ShowCollections_Call struct { } // ShowCollections is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.ShowCollectionsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.ShowCollectionsRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) ShowCollections(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_ShowCollections_Call { return &MockQueryCoordClient_ShowCollections_Call{Call: _e.mock.On("ShowCollections", append([]interface{}{ctx, in}, opts...)...)} @@ -1651,9 +1651,9 @@ type MockQueryCoordClient_ShowConfigurations_Call struct { } // ShowConfigurations is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.ShowConfigurationsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.ShowConfigurationsRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) ShowConfigurations(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_ShowConfigurations_Call { return &MockQueryCoordClient_ShowConfigurations_Call{Call: _e.mock.On("ShowConfigurations", append([]interface{}{ctx, in}, opts...)...)} @@ -1721,9 +1721,9 @@ type MockQueryCoordClient_ShowPartitions_Call struct { } // ShowPartitions is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.ShowPartitionsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.ShowPartitionsRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) ShowPartitions(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_ShowPartitions_Call { return &MockQueryCoordClient_ShowPartitions_Call{Call: _e.mock.On("ShowPartitions", append([]interface{}{ctx, in}, opts...)...)} @@ -1791,9 +1791,9 @@ type MockQueryCoordClient_SyncNewCreatedPartition_Call struct { } // SyncNewCreatedPartition is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.SyncNewCreatedPartitionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.SyncNewCreatedPartitionRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) SyncNewCreatedPartition(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_SyncNewCreatedPartition_Call { return &MockQueryCoordClient_SyncNewCreatedPartition_Call{Call: _e.mock.On("SyncNewCreatedPartition", append([]interface{}{ctx, in}, opts...)...)} @@ -1861,9 +1861,9 @@ type MockQueryCoordClient_TransferNode_Call struct { } // TransferNode is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.TransferNodeRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.TransferNodeRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) TransferNode(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_TransferNode_Call { return &MockQueryCoordClient_TransferNode_Call{Call: _e.mock.On("TransferNode", append([]interface{}{ctx, in}, opts...)...)} @@ -1931,9 +1931,9 @@ type MockQueryCoordClient_TransferReplica_Call struct { } // TransferReplica is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.TransferReplicaRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.TransferReplicaRequest +// - opts ...grpc.CallOption func (_e *MockQueryCoordClient_Expecter) TransferReplica(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryCoordClient_TransferReplica_Call { return &MockQueryCoordClient_TransferReplica_Call{Call: _e.mock.On("TransferReplica", append([]interface{}{ctx, in}, opts...)...)} diff --git a/internal/mocks/mock_querynode.go b/internal/mocks/mock_querynode.go index e6288f04b1d27..5a59275f8756d 100644 --- a/internal/mocks/mock_querynode.go +++ b/internal/mocks/mock_querynode.go @@ -62,8 +62,8 @@ type MockQueryNode_Delete_Call struct { } // Delete is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.DeleteRequest +// - _a0 context.Context +// - _a1 *querypb.DeleteRequest func (_e *MockQueryNode_Expecter) Delete(_a0 interface{}, _a1 interface{}) *MockQueryNode_Delete_Call { return &MockQueryNode_Delete_Call{Call: _e.mock.On("Delete", _a0, _a1)} } @@ -158,8 +158,8 @@ type MockQueryNode_GetComponentStates_Call struct { } // GetComponentStates is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetComponentStatesRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetComponentStatesRequest func (_e *MockQueryNode_Expecter) GetComponentStates(_a0 interface{}, _a1 interface{}) *MockQueryNode_GetComponentStates_Call { return &MockQueryNode_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", _a0, _a1)} } @@ -213,8 +213,8 @@ type MockQueryNode_GetDataDistribution_Call struct { } // GetDataDistribution is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.GetDataDistributionRequest +// - _a0 context.Context +// - _a1 *querypb.GetDataDistributionRequest func (_e *MockQueryNode_Expecter) GetDataDistribution(_a0 interface{}, _a1 interface{}) *MockQueryNode_GetDataDistribution_Call { return &MockQueryNode_GetDataDistribution_Call{Call: _e.mock.On("GetDataDistribution", _a0, _a1)} } @@ -268,8 +268,8 @@ type MockQueryNode_GetMetrics_Call struct { } // GetMetrics is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetMetricsRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetMetricsRequest func (_e *MockQueryNode_Expecter) GetMetrics(_a0 interface{}, _a1 interface{}) *MockQueryNode_GetMetrics_Call { return &MockQueryNode_GetMetrics_Call{Call: _e.mock.On("GetMetrics", _a0, _a1)} } @@ -364,8 +364,8 @@ type MockQueryNode_GetSegmentInfo_Call struct { } // GetSegmentInfo is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.GetSegmentInfoRequest +// - _a0 context.Context +// - _a1 *querypb.GetSegmentInfoRequest func (_e *MockQueryNode_Expecter) GetSegmentInfo(_a0 interface{}, _a1 interface{}) *MockQueryNode_GetSegmentInfo_Call { return &MockQueryNode_GetSegmentInfo_Call{Call: _e.mock.On("GetSegmentInfo", _a0, _a1)} } @@ -419,8 +419,8 @@ type MockQueryNode_GetStatistics_Call struct { } // GetStatistics is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.GetStatisticsRequest +// - _a0 context.Context +// - _a1 *querypb.GetStatisticsRequest func (_e *MockQueryNode_Expecter) GetStatistics(_a0 interface{}, _a1 interface{}) *MockQueryNode_GetStatistics_Call { return &MockQueryNode_GetStatistics_Call{Call: _e.mock.On("GetStatistics", _a0, _a1)} } @@ -474,8 +474,8 @@ type MockQueryNode_GetStatisticsChannel_Call struct { } // GetStatisticsChannel is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.GetStatisticsChannelRequest +// - _a0 context.Context +// - _a1 *internalpb.GetStatisticsChannelRequest func (_e *MockQueryNode_Expecter) GetStatisticsChannel(_a0 interface{}, _a1 interface{}) *MockQueryNode_GetStatisticsChannel_Call { return &MockQueryNode_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", _a0, _a1)} } @@ -529,8 +529,8 @@ type MockQueryNode_GetTimeTickChannel_Call struct { } // GetTimeTickChannel is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.GetTimeTickChannelRequest +// - _a0 context.Context +// - _a1 *internalpb.GetTimeTickChannelRequest func (_e *MockQueryNode_Expecter) GetTimeTickChannel(_a0 interface{}, _a1 interface{}) *MockQueryNode_GetTimeTickChannel_Call { return &MockQueryNode_GetTimeTickChannel_Call{Call: _e.mock.On("GetTimeTickChannel", _a0, _a1)} } @@ -584,8 +584,8 @@ type MockQueryNode_HybridSearch_Call struct { } // HybridSearch is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.HybridSearchRequest +// - _a0 context.Context +// - _a1 *querypb.HybridSearchRequest func (_e *MockQueryNode_Expecter) HybridSearch(_a0 interface{}, _a1 interface{}) *MockQueryNode_HybridSearch_Call { return &MockQueryNode_HybridSearch_Call{Call: _e.mock.On("HybridSearch", _a0, _a1)} } @@ -680,8 +680,8 @@ type MockQueryNode_LoadPartitions_Call struct { } // LoadPartitions is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.LoadPartitionsRequest +// - _a0 context.Context +// - _a1 *querypb.LoadPartitionsRequest func (_e *MockQueryNode_Expecter) LoadPartitions(_a0 interface{}, _a1 interface{}) *MockQueryNode_LoadPartitions_Call { return &MockQueryNode_LoadPartitions_Call{Call: _e.mock.On("LoadPartitions", _a0, _a1)} } @@ -735,8 +735,8 @@ type MockQueryNode_LoadSegments_Call struct { } // LoadSegments is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.LoadSegmentsRequest +// - _a0 context.Context +// - _a1 *querypb.LoadSegmentsRequest func (_e *MockQueryNode_Expecter) LoadSegments(_a0 interface{}, _a1 interface{}) *MockQueryNode_LoadSegments_Call { return &MockQueryNode_LoadSegments_Call{Call: _e.mock.On("LoadSegments", _a0, _a1)} } @@ -790,8 +790,8 @@ type MockQueryNode_Query_Call struct { } // Query is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.QueryRequest +// - _a0 context.Context +// - _a1 *querypb.QueryRequest func (_e *MockQueryNode_Expecter) Query(_a0 interface{}, _a1 interface{}) *MockQueryNode_Query_Call { return &MockQueryNode_Query_Call{Call: _e.mock.On("Query", _a0, _a1)} } @@ -845,8 +845,8 @@ type MockQueryNode_QuerySegments_Call struct { } // QuerySegments is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.QueryRequest +// - _a0 context.Context +// - _a1 *querypb.QueryRequest func (_e *MockQueryNode_Expecter) QuerySegments(_a0 interface{}, _a1 interface{}) *MockQueryNode_QuerySegments_Call { return &MockQueryNode_QuerySegments_Call{Call: _e.mock.On("QuerySegments", _a0, _a1)} } @@ -888,8 +888,8 @@ type MockQueryNode_QueryStream_Call struct { } // QueryStream is a helper method to define mock.On call -// - _a0 *querypb.QueryRequest -// - _a1 querypb.QueryNode_QueryStreamServer +// - _a0 *querypb.QueryRequest +// - _a1 querypb.QueryNode_QueryStreamServer func (_e *MockQueryNode_Expecter) QueryStream(_a0 interface{}, _a1 interface{}) *MockQueryNode_QueryStream_Call { return &MockQueryNode_QueryStream_Call{Call: _e.mock.On("QueryStream", _a0, _a1)} } @@ -931,8 +931,8 @@ type MockQueryNode_QueryStreamSegments_Call struct { } // QueryStreamSegments is a helper method to define mock.On call -// - _a0 *querypb.QueryRequest -// - _a1 querypb.QueryNode_QueryStreamSegmentsServer +// - _a0 *querypb.QueryRequest +// - _a1 querypb.QueryNode_QueryStreamSegmentsServer func (_e *MockQueryNode_Expecter) QueryStreamSegments(_a0 interface{}, _a1 interface{}) *MockQueryNode_QueryStreamSegments_Call { return &MockQueryNode_QueryStreamSegments_Call{Call: _e.mock.On("QueryStreamSegments", _a0, _a1)} } @@ -1027,8 +1027,8 @@ type MockQueryNode_ReleaseCollection_Call struct { } // ReleaseCollection is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.ReleaseCollectionRequest +// - _a0 context.Context +// - _a1 *querypb.ReleaseCollectionRequest func (_e *MockQueryNode_Expecter) ReleaseCollection(_a0 interface{}, _a1 interface{}) *MockQueryNode_ReleaseCollection_Call { return &MockQueryNode_ReleaseCollection_Call{Call: _e.mock.On("ReleaseCollection", _a0, _a1)} } @@ -1082,8 +1082,8 @@ type MockQueryNode_ReleasePartitions_Call struct { } // ReleasePartitions is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.ReleasePartitionsRequest +// - _a0 context.Context +// - _a1 *querypb.ReleasePartitionsRequest func (_e *MockQueryNode_Expecter) ReleasePartitions(_a0 interface{}, _a1 interface{}) *MockQueryNode_ReleasePartitions_Call { return &MockQueryNode_ReleasePartitions_Call{Call: _e.mock.On("ReleasePartitions", _a0, _a1)} } @@ -1137,8 +1137,8 @@ type MockQueryNode_ReleaseSegments_Call struct { } // ReleaseSegments is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.ReleaseSegmentsRequest +// - _a0 context.Context +// - _a1 *querypb.ReleaseSegmentsRequest func (_e *MockQueryNode_Expecter) ReleaseSegments(_a0 interface{}, _a1 interface{}) *MockQueryNode_ReleaseSegments_Call { return &MockQueryNode_ReleaseSegments_Call{Call: _e.mock.On("ReleaseSegments", _a0, _a1)} } @@ -1192,8 +1192,8 @@ type MockQueryNode_Search_Call struct { } // Search is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.SearchRequest +// - _a0 context.Context +// - _a1 *querypb.SearchRequest func (_e *MockQueryNode_Expecter) Search(_a0 interface{}, _a1 interface{}) *MockQueryNode_Search_Call { return &MockQueryNode_Search_Call{Call: _e.mock.On("Search", _a0, _a1)} } @@ -1247,8 +1247,8 @@ type MockQueryNode_SearchSegments_Call struct { } // SearchSegments is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.SearchRequest +// - _a0 context.Context +// - _a1 *querypb.SearchRequest func (_e *MockQueryNode_Expecter) SearchSegments(_a0 interface{}, _a1 interface{}) *MockQueryNode_SearchSegments_Call { return &MockQueryNode_SearchSegments_Call{Call: _e.mock.On("SearchSegments", _a0, _a1)} } @@ -1281,7 +1281,7 @@ type MockQueryNode_SetAddress_Call struct { } // SetAddress is a helper method to define mock.On call -// - address string +// - address string func (_e *MockQueryNode_Expecter) SetAddress(address interface{}) *MockQueryNode_SetAddress_Call { return &MockQueryNode_SetAddress_Call{Call: _e.mock.On("SetAddress", address)} } @@ -1314,7 +1314,7 @@ type MockQueryNode_SetEtcdClient_Call struct { } // SetEtcdClient is a helper method to define mock.On call -// - etcdClient *clientv3.Client +// - etcdClient *clientv3.Client func (_e *MockQueryNode_Expecter) SetEtcdClient(etcdClient interface{}) *MockQueryNode_SetEtcdClient_Call { return &MockQueryNode_SetEtcdClient_Call{Call: _e.mock.On("SetEtcdClient", etcdClient)} } @@ -1368,8 +1368,8 @@ type MockQueryNode_ShowConfigurations_Call struct { } // ShowConfigurations is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.ShowConfigurationsRequest +// - _a0 context.Context +// - _a1 *internalpb.ShowConfigurationsRequest func (_e *MockQueryNode_Expecter) ShowConfigurations(_a0 interface{}, _a1 interface{}) *MockQueryNode_ShowConfigurations_Call { return &MockQueryNode_ShowConfigurations_Call{Call: _e.mock.On("ShowConfigurations", _a0, _a1)} } @@ -1505,8 +1505,8 @@ type MockQueryNode_SyncDistribution_Call struct { } // SyncDistribution is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.SyncDistributionRequest +// - _a0 context.Context +// - _a1 *querypb.SyncDistributionRequest func (_e *MockQueryNode_Expecter) SyncDistribution(_a0 interface{}, _a1 interface{}) *MockQueryNode_SyncDistribution_Call { return &MockQueryNode_SyncDistribution_Call{Call: _e.mock.On("SyncDistribution", _a0, _a1)} } @@ -1560,8 +1560,8 @@ type MockQueryNode_SyncReplicaSegments_Call struct { } // SyncReplicaSegments is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.SyncReplicaSegmentsRequest +// - _a0 context.Context +// - _a1 *querypb.SyncReplicaSegmentsRequest func (_e *MockQueryNode_Expecter) SyncReplicaSegments(_a0 interface{}, _a1 interface{}) *MockQueryNode_SyncReplicaSegments_Call { return &MockQueryNode_SyncReplicaSegments_Call{Call: _e.mock.On("SyncReplicaSegments", _a0, _a1)} } @@ -1615,8 +1615,8 @@ type MockQueryNode_UnsubDmChannel_Call struct { } // UnsubDmChannel is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.UnsubDmChannelRequest +// - _a0 context.Context +// - _a1 *querypb.UnsubDmChannelRequest func (_e *MockQueryNode_Expecter) UnsubDmChannel(_a0 interface{}, _a1 interface{}) *MockQueryNode_UnsubDmChannel_Call { return &MockQueryNode_UnsubDmChannel_Call{Call: _e.mock.On("UnsubDmChannel", _a0, _a1)} } @@ -1649,7 +1649,7 @@ type MockQueryNode_UpdateStateCode_Call struct { } // UpdateStateCode is a helper method to define mock.On call -// - stateCode commonpb.StateCode +// - stateCode commonpb.StateCode func (_e *MockQueryNode_Expecter) UpdateStateCode(stateCode interface{}) *MockQueryNode_UpdateStateCode_Call { return &MockQueryNode_UpdateStateCode_Call{Call: _e.mock.On("UpdateStateCode", stateCode)} } @@ -1703,8 +1703,8 @@ type MockQueryNode_WatchDmChannels_Call struct { } // WatchDmChannels is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *querypb.WatchDmChannelsRequest +// - _a0 context.Context +// - _a1 *querypb.WatchDmChannelsRequest func (_e *MockQueryNode_Expecter) WatchDmChannels(_a0 interface{}, _a1 interface{}) *MockQueryNode_WatchDmChannels_Call { return &MockQueryNode_WatchDmChannels_Call{Call: _e.mock.On("WatchDmChannels", _a0, _a1)} } diff --git a/internal/mocks/mock_querynode_client.go b/internal/mocks/mock_querynode_client.go index 51f4d9ba3f4b6..1f68bd7769cf0 100644 --- a/internal/mocks/mock_querynode_client.go +++ b/internal/mocks/mock_querynode_client.go @@ -111,9 +111,9 @@ type MockQueryNodeClient_Delete_Call struct { } // Delete is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.DeleteRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.DeleteRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) Delete(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_Delete_Call { return &MockQueryNodeClient_Delete_Call{Call: _e.mock.On("Delete", append([]interface{}{ctx, in}, opts...)...)} @@ -181,9 +181,9 @@ type MockQueryNodeClient_GetComponentStates_Call struct { } // GetComponentStates is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetComponentStatesRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetComponentStatesRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) GetComponentStates(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_GetComponentStates_Call { return &MockQueryNodeClient_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", append([]interface{}{ctx, in}, opts...)...)} @@ -251,9 +251,9 @@ type MockQueryNodeClient_GetDataDistribution_Call struct { } // GetDataDistribution is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.GetDataDistributionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.GetDataDistributionRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) GetDataDistribution(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_GetDataDistribution_Call { return &MockQueryNodeClient_GetDataDistribution_Call{Call: _e.mock.On("GetDataDistribution", append([]interface{}{ctx, in}, opts...)...)} @@ -321,9 +321,9 @@ type MockQueryNodeClient_GetMetrics_Call struct { } // GetMetrics is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetMetricsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetMetricsRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) GetMetrics(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_GetMetrics_Call { return &MockQueryNodeClient_GetMetrics_Call{Call: _e.mock.On("GetMetrics", append([]interface{}{ctx, in}, opts...)...)} @@ -391,9 +391,9 @@ type MockQueryNodeClient_GetSegmentInfo_Call struct { } // GetSegmentInfo is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.GetSegmentInfoRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.GetSegmentInfoRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) GetSegmentInfo(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_GetSegmentInfo_Call { return &MockQueryNodeClient_GetSegmentInfo_Call{Call: _e.mock.On("GetSegmentInfo", append([]interface{}{ctx, in}, opts...)...)} @@ -461,9 +461,9 @@ type MockQueryNodeClient_GetStatistics_Call struct { } // GetStatistics is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.GetStatisticsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.GetStatisticsRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) GetStatistics(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_GetStatistics_Call { return &MockQueryNodeClient_GetStatistics_Call{Call: _e.mock.On("GetStatistics", append([]interface{}{ctx, in}, opts...)...)} @@ -531,9 +531,9 @@ type MockQueryNodeClient_GetStatisticsChannel_Call struct { } // GetStatisticsChannel is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.GetStatisticsChannelRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.GetStatisticsChannelRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) GetStatisticsChannel(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_GetStatisticsChannel_Call { return &MockQueryNodeClient_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", append([]interface{}{ctx, in}, opts...)...)} @@ -601,9 +601,9 @@ type MockQueryNodeClient_GetTimeTickChannel_Call struct { } // GetTimeTickChannel is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.GetTimeTickChannelRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.GetTimeTickChannelRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) GetTimeTickChannel(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_GetTimeTickChannel_Call { return &MockQueryNodeClient_GetTimeTickChannel_Call{Call: _e.mock.On("GetTimeTickChannel", append([]interface{}{ctx, in}, opts...)...)} @@ -671,9 +671,9 @@ type MockQueryNodeClient_HybridSearch_Call struct { } // HybridSearch is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.HybridSearchRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.HybridSearchRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) HybridSearch(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_HybridSearch_Call { return &MockQueryNodeClient_HybridSearch_Call{Call: _e.mock.On("HybridSearch", append([]interface{}{ctx, in}, opts...)...)} @@ -741,9 +741,9 @@ type MockQueryNodeClient_LoadPartitions_Call struct { } // LoadPartitions is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.LoadPartitionsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.LoadPartitionsRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) LoadPartitions(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_LoadPartitions_Call { return &MockQueryNodeClient_LoadPartitions_Call{Call: _e.mock.On("LoadPartitions", append([]interface{}{ctx, in}, opts...)...)} @@ -811,9 +811,9 @@ type MockQueryNodeClient_LoadSegments_Call struct { } // LoadSegments is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.LoadSegmentsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.LoadSegmentsRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) LoadSegments(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_LoadSegments_Call { return &MockQueryNodeClient_LoadSegments_Call{Call: _e.mock.On("LoadSegments", append([]interface{}{ctx, in}, opts...)...)} @@ -881,9 +881,9 @@ type MockQueryNodeClient_Query_Call struct { } // Query is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.QueryRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.QueryRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) Query(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_Query_Call { return &MockQueryNodeClient_Query_Call{Call: _e.mock.On("Query", append([]interface{}{ctx, in}, opts...)...)} @@ -951,9 +951,9 @@ type MockQueryNodeClient_QuerySegments_Call struct { } // QuerySegments is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.QueryRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.QueryRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) QuerySegments(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_QuerySegments_Call { return &MockQueryNodeClient_QuerySegments_Call{Call: _e.mock.On("QuerySegments", append([]interface{}{ctx, in}, opts...)...)} @@ -1021,9 +1021,9 @@ type MockQueryNodeClient_QueryStream_Call struct { } // QueryStream is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.QueryRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.QueryRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) QueryStream(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_QueryStream_Call { return &MockQueryNodeClient_QueryStream_Call{Call: _e.mock.On("QueryStream", append([]interface{}{ctx, in}, opts...)...)} @@ -1091,9 +1091,9 @@ type MockQueryNodeClient_QueryStreamSegments_Call struct { } // QueryStreamSegments is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.QueryRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.QueryRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) QueryStreamSegments(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_QueryStreamSegments_Call { return &MockQueryNodeClient_QueryStreamSegments_Call{Call: _e.mock.On("QueryStreamSegments", append([]interface{}{ctx, in}, opts...)...)} @@ -1161,9 +1161,9 @@ type MockQueryNodeClient_ReleaseCollection_Call struct { } // ReleaseCollection is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.ReleaseCollectionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.ReleaseCollectionRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) ReleaseCollection(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_ReleaseCollection_Call { return &MockQueryNodeClient_ReleaseCollection_Call{Call: _e.mock.On("ReleaseCollection", append([]interface{}{ctx, in}, opts...)...)} @@ -1231,9 +1231,9 @@ type MockQueryNodeClient_ReleasePartitions_Call struct { } // ReleasePartitions is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.ReleasePartitionsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.ReleasePartitionsRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) ReleasePartitions(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_ReleasePartitions_Call { return &MockQueryNodeClient_ReleasePartitions_Call{Call: _e.mock.On("ReleasePartitions", append([]interface{}{ctx, in}, opts...)...)} @@ -1301,9 +1301,9 @@ type MockQueryNodeClient_ReleaseSegments_Call struct { } // ReleaseSegments is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.ReleaseSegmentsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.ReleaseSegmentsRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) ReleaseSegments(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_ReleaseSegments_Call { return &MockQueryNodeClient_ReleaseSegments_Call{Call: _e.mock.On("ReleaseSegments", append([]interface{}{ctx, in}, opts...)...)} @@ -1371,9 +1371,9 @@ type MockQueryNodeClient_Search_Call struct { } // Search is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.SearchRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.SearchRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) Search(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_Search_Call { return &MockQueryNodeClient_Search_Call{Call: _e.mock.On("Search", append([]interface{}{ctx, in}, opts...)...)} @@ -1441,9 +1441,9 @@ type MockQueryNodeClient_SearchSegments_Call struct { } // SearchSegments is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.SearchRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.SearchRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) SearchSegments(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_SearchSegments_Call { return &MockQueryNodeClient_SearchSegments_Call{Call: _e.mock.On("SearchSegments", append([]interface{}{ctx, in}, opts...)...)} @@ -1511,9 +1511,9 @@ type MockQueryNodeClient_ShowConfigurations_Call struct { } // ShowConfigurations is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.ShowConfigurationsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.ShowConfigurationsRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) ShowConfigurations(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_ShowConfigurations_Call { return &MockQueryNodeClient_ShowConfigurations_Call{Call: _e.mock.On("ShowConfigurations", append([]interface{}{ctx, in}, opts...)...)} @@ -1581,9 +1581,9 @@ type MockQueryNodeClient_SyncDistribution_Call struct { } // SyncDistribution is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.SyncDistributionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.SyncDistributionRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) SyncDistribution(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_SyncDistribution_Call { return &MockQueryNodeClient_SyncDistribution_Call{Call: _e.mock.On("SyncDistribution", append([]interface{}{ctx, in}, opts...)...)} @@ -1651,9 +1651,9 @@ type MockQueryNodeClient_SyncReplicaSegments_Call struct { } // SyncReplicaSegments is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.SyncReplicaSegmentsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.SyncReplicaSegmentsRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) SyncReplicaSegments(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_SyncReplicaSegments_Call { return &MockQueryNodeClient_SyncReplicaSegments_Call{Call: _e.mock.On("SyncReplicaSegments", append([]interface{}{ctx, in}, opts...)...)} @@ -1721,9 +1721,9 @@ type MockQueryNodeClient_UnsubDmChannel_Call struct { } // UnsubDmChannel is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.UnsubDmChannelRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.UnsubDmChannelRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) UnsubDmChannel(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_UnsubDmChannel_Call { return &MockQueryNodeClient_UnsubDmChannel_Call{Call: _e.mock.On("UnsubDmChannel", append([]interface{}{ctx, in}, opts...)...)} @@ -1791,9 +1791,9 @@ type MockQueryNodeClient_WatchDmChannels_Call struct { } // WatchDmChannels is a helper method to define mock.On call -// - ctx context.Context -// - in *querypb.WatchDmChannelsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *querypb.WatchDmChannelsRequest +// - opts ...grpc.CallOption func (_e *MockQueryNodeClient_Expecter) WatchDmChannels(ctx interface{}, in interface{}, opts ...interface{}) *MockQueryNodeClient_WatchDmChannels_Call { return &MockQueryNodeClient_WatchDmChannels_Call{Call: _e.mock.On("WatchDmChannels", append([]interface{}{ctx, in}, opts...)...)} diff --git a/internal/mocks/mock_rootcoord_client.go b/internal/mocks/mock_rootcoord_client.go index 30ccb664e3155..0b98cf0e9984a 100644 --- a/internal/mocks/mock_rootcoord_client.go +++ b/internal/mocks/mock_rootcoord_client.go @@ -72,9 +72,9 @@ type MockRootCoordClient_AllocID_Call struct { } // AllocID is a helper method to define mock.On call -// - ctx context.Context -// - in *rootcoordpb.AllocIDRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *rootcoordpb.AllocIDRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) AllocID(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_AllocID_Call { return &MockRootCoordClient_AllocID_Call{Call: _e.mock.On("AllocID", append([]interface{}{ctx, in}, opts...)...)} @@ -142,9 +142,9 @@ type MockRootCoordClient_AllocTimestamp_Call struct { } // AllocTimestamp is a helper method to define mock.On call -// - ctx context.Context -// - in *rootcoordpb.AllocTimestampRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *rootcoordpb.AllocTimestampRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) AllocTimestamp(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_AllocTimestamp_Call { return &MockRootCoordClient_AllocTimestamp_Call{Call: _e.mock.On("AllocTimestamp", append([]interface{}{ctx, in}, opts...)...)} @@ -212,9 +212,9 @@ type MockRootCoordClient_AlterAlias_Call struct { } // AlterAlias is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.AlterAliasRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.AlterAliasRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) AlterAlias(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_AlterAlias_Call { return &MockRootCoordClient_AlterAlias_Call{Call: _e.mock.On("AlterAlias", append([]interface{}{ctx, in}, opts...)...)} @@ -282,9 +282,9 @@ type MockRootCoordClient_AlterCollection_Call struct { } // AlterCollection is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.AlterCollectionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.AlterCollectionRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) AlterCollection(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_AlterCollection_Call { return &MockRootCoordClient_AlterCollection_Call{Call: _e.mock.On("AlterCollection", append([]interface{}{ctx, in}, opts...)...)} @@ -352,9 +352,9 @@ type MockRootCoordClient_CheckHealth_Call struct { } // CheckHealth is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.CheckHealthRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.CheckHealthRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) CheckHealth(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_CheckHealth_Call { return &MockRootCoordClient_CheckHealth_Call{Call: _e.mock.On("CheckHealth", append([]interface{}{ctx, in}, opts...)...)} @@ -463,9 +463,9 @@ type MockRootCoordClient_CreateAlias_Call struct { } // CreateAlias is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.CreateAliasRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.CreateAliasRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) CreateAlias(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_CreateAlias_Call { return &MockRootCoordClient_CreateAlias_Call{Call: _e.mock.On("CreateAlias", append([]interface{}{ctx, in}, opts...)...)} @@ -533,9 +533,9 @@ type MockRootCoordClient_CreateCollection_Call struct { } // CreateCollection is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.CreateCollectionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.CreateCollectionRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) CreateCollection(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_CreateCollection_Call { return &MockRootCoordClient_CreateCollection_Call{Call: _e.mock.On("CreateCollection", append([]interface{}{ctx, in}, opts...)...)} @@ -603,9 +603,9 @@ type MockRootCoordClient_CreateCredential_Call struct { } // CreateCredential is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.CredentialInfo -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.CredentialInfo +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) CreateCredential(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_CreateCredential_Call { return &MockRootCoordClient_CreateCredential_Call{Call: _e.mock.On("CreateCredential", append([]interface{}{ctx, in}, opts...)...)} @@ -673,9 +673,9 @@ type MockRootCoordClient_CreateDatabase_Call struct { } // CreateDatabase is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.CreateDatabaseRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.CreateDatabaseRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) CreateDatabase(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_CreateDatabase_Call { return &MockRootCoordClient_CreateDatabase_Call{Call: _e.mock.On("CreateDatabase", append([]interface{}{ctx, in}, opts...)...)} @@ -743,9 +743,9 @@ type MockRootCoordClient_CreatePartition_Call struct { } // CreatePartition is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.CreatePartitionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.CreatePartitionRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) CreatePartition(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_CreatePartition_Call { return &MockRootCoordClient_CreatePartition_Call{Call: _e.mock.On("CreatePartition", append([]interface{}{ctx, in}, opts...)...)} @@ -813,9 +813,9 @@ type MockRootCoordClient_CreateRole_Call struct { } // CreateRole is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.CreateRoleRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.CreateRoleRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) CreateRole(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_CreateRole_Call { return &MockRootCoordClient_CreateRole_Call{Call: _e.mock.On("CreateRole", append([]interface{}{ctx, in}, opts...)...)} @@ -883,9 +883,9 @@ type MockRootCoordClient_DeleteCredential_Call struct { } // DeleteCredential is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.DeleteCredentialRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.DeleteCredentialRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) DeleteCredential(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_DeleteCredential_Call { return &MockRootCoordClient_DeleteCredential_Call{Call: _e.mock.On("DeleteCredential", append([]interface{}{ctx, in}, opts...)...)} @@ -953,9 +953,9 @@ type MockRootCoordClient_DescribeAlias_Call struct { } // DescribeAlias is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.DescribeAliasRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.DescribeAliasRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) DescribeAlias(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_DescribeAlias_Call { return &MockRootCoordClient_DescribeAlias_Call{Call: _e.mock.On("DescribeAlias", append([]interface{}{ctx, in}, opts...)...)} @@ -1023,9 +1023,9 @@ type MockRootCoordClient_DescribeCollection_Call struct { } // DescribeCollection is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.DescribeCollectionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.DescribeCollectionRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) DescribeCollection(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_DescribeCollection_Call { return &MockRootCoordClient_DescribeCollection_Call{Call: _e.mock.On("DescribeCollection", append([]interface{}{ctx, in}, opts...)...)} @@ -1093,9 +1093,9 @@ type MockRootCoordClient_DescribeCollectionInternal_Call struct { } // DescribeCollectionInternal is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.DescribeCollectionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.DescribeCollectionRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) DescribeCollectionInternal(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_DescribeCollectionInternal_Call { return &MockRootCoordClient_DescribeCollectionInternal_Call{Call: _e.mock.On("DescribeCollectionInternal", append([]interface{}{ctx, in}, opts...)...)} @@ -1163,9 +1163,9 @@ type MockRootCoordClient_DropAlias_Call struct { } // DropAlias is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.DropAliasRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.DropAliasRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) DropAlias(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_DropAlias_Call { return &MockRootCoordClient_DropAlias_Call{Call: _e.mock.On("DropAlias", append([]interface{}{ctx, in}, opts...)...)} @@ -1233,9 +1233,9 @@ type MockRootCoordClient_DropCollection_Call struct { } // DropCollection is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.DropCollectionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.DropCollectionRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) DropCollection(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_DropCollection_Call { return &MockRootCoordClient_DropCollection_Call{Call: _e.mock.On("DropCollection", append([]interface{}{ctx, in}, opts...)...)} @@ -1303,9 +1303,9 @@ type MockRootCoordClient_DropDatabase_Call struct { } // DropDatabase is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.DropDatabaseRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.DropDatabaseRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) DropDatabase(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_DropDatabase_Call { return &MockRootCoordClient_DropDatabase_Call{Call: _e.mock.On("DropDatabase", append([]interface{}{ctx, in}, opts...)...)} @@ -1373,9 +1373,9 @@ type MockRootCoordClient_DropPartition_Call struct { } // DropPartition is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.DropPartitionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.DropPartitionRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) DropPartition(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_DropPartition_Call { return &MockRootCoordClient_DropPartition_Call{Call: _e.mock.On("DropPartition", append([]interface{}{ctx, in}, opts...)...)} @@ -1443,9 +1443,9 @@ type MockRootCoordClient_DropRole_Call struct { } // DropRole is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.DropRoleRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.DropRoleRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) DropRole(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_DropRole_Call { return &MockRootCoordClient_DropRole_Call{Call: _e.mock.On("DropRole", append([]interface{}{ctx, in}, opts...)...)} @@ -1513,9 +1513,9 @@ type MockRootCoordClient_GetComponentStates_Call struct { } // GetComponentStates is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetComponentStatesRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetComponentStatesRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) GetComponentStates(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_GetComponentStates_Call { return &MockRootCoordClient_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", append([]interface{}{ctx, in}, opts...)...)} @@ -1583,9 +1583,9 @@ type MockRootCoordClient_GetCredential_Call struct { } // GetCredential is a helper method to define mock.On call -// - ctx context.Context -// - in *rootcoordpb.GetCredentialRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *rootcoordpb.GetCredentialRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) GetCredential(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_GetCredential_Call { return &MockRootCoordClient_GetCredential_Call{Call: _e.mock.On("GetCredential", append([]interface{}{ctx, in}, opts...)...)} @@ -1653,9 +1653,9 @@ type MockRootCoordClient_GetImportState_Call struct { } // GetImportState is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetImportStateRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetImportStateRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) GetImportState(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_GetImportState_Call { return &MockRootCoordClient_GetImportState_Call{Call: _e.mock.On("GetImportState", append([]interface{}{ctx, in}, opts...)...)} @@ -1723,9 +1723,9 @@ type MockRootCoordClient_GetMetrics_Call struct { } // GetMetrics is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.GetMetricsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.GetMetricsRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) GetMetrics(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_GetMetrics_Call { return &MockRootCoordClient_GetMetrics_Call{Call: _e.mock.On("GetMetrics", append([]interface{}{ctx, in}, opts...)...)} @@ -1793,9 +1793,9 @@ type MockRootCoordClient_GetStatisticsChannel_Call struct { } // GetStatisticsChannel is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.GetStatisticsChannelRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.GetStatisticsChannelRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) GetStatisticsChannel(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_GetStatisticsChannel_Call { return &MockRootCoordClient_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", append([]interface{}{ctx, in}, opts...)...)} @@ -1863,9 +1863,9 @@ type MockRootCoordClient_GetTimeTickChannel_Call struct { } // GetTimeTickChannel is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.GetTimeTickChannelRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.GetTimeTickChannelRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) GetTimeTickChannel(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_GetTimeTickChannel_Call { return &MockRootCoordClient_GetTimeTickChannel_Call{Call: _e.mock.On("GetTimeTickChannel", append([]interface{}{ctx, in}, opts...)...)} @@ -1933,9 +1933,9 @@ type MockRootCoordClient_HasCollection_Call struct { } // HasCollection is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.HasCollectionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.HasCollectionRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) HasCollection(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_HasCollection_Call { return &MockRootCoordClient_HasCollection_Call{Call: _e.mock.On("HasCollection", append([]interface{}{ctx, in}, opts...)...)} @@ -2003,9 +2003,9 @@ type MockRootCoordClient_HasPartition_Call struct { } // HasPartition is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.HasPartitionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.HasPartitionRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) HasPartition(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_HasPartition_Call { return &MockRootCoordClient_HasPartition_Call{Call: _e.mock.On("HasPartition", append([]interface{}{ctx, in}, opts...)...)} @@ -2073,9 +2073,9 @@ type MockRootCoordClient_Import_Call struct { } // Import is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.ImportRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.ImportRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) Import(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_Import_Call { return &MockRootCoordClient_Import_Call{Call: _e.mock.On("Import", append([]interface{}{ctx, in}, opts...)...)} @@ -2143,9 +2143,9 @@ type MockRootCoordClient_InvalidateCollectionMetaCache_Call struct { } // InvalidateCollectionMetaCache is a helper method to define mock.On call -// - ctx context.Context -// - in *proxypb.InvalidateCollMetaCacheRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *proxypb.InvalidateCollMetaCacheRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) InvalidateCollectionMetaCache(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_InvalidateCollectionMetaCache_Call { return &MockRootCoordClient_InvalidateCollectionMetaCache_Call{Call: _e.mock.On("InvalidateCollectionMetaCache", append([]interface{}{ctx, in}, opts...)...)} @@ -2213,9 +2213,9 @@ type MockRootCoordClient_ListAliases_Call struct { } // ListAliases is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.ListAliasesRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.ListAliasesRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) ListAliases(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_ListAliases_Call { return &MockRootCoordClient_ListAliases_Call{Call: _e.mock.On("ListAliases", append([]interface{}{ctx, in}, opts...)...)} @@ -2283,9 +2283,9 @@ type MockRootCoordClient_ListCredUsers_Call struct { } // ListCredUsers is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.ListCredUsersRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.ListCredUsersRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) ListCredUsers(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_ListCredUsers_Call { return &MockRootCoordClient_ListCredUsers_Call{Call: _e.mock.On("ListCredUsers", append([]interface{}{ctx, in}, opts...)...)} @@ -2353,9 +2353,9 @@ type MockRootCoordClient_ListDatabases_Call struct { } // ListDatabases is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.ListDatabasesRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.ListDatabasesRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) ListDatabases(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_ListDatabases_Call { return &MockRootCoordClient_ListDatabases_Call{Call: _e.mock.On("ListDatabases", append([]interface{}{ctx, in}, opts...)...)} @@ -2423,9 +2423,9 @@ type MockRootCoordClient_ListImportTasks_Call struct { } // ListImportTasks is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.ListImportTasksRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.ListImportTasksRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) ListImportTasks(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_ListImportTasks_Call { return &MockRootCoordClient_ListImportTasks_Call{Call: _e.mock.On("ListImportTasks", append([]interface{}{ctx, in}, opts...)...)} @@ -2493,9 +2493,9 @@ type MockRootCoordClient_ListPolicy_Call struct { } // ListPolicy is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.ListPolicyRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.ListPolicyRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) ListPolicy(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_ListPolicy_Call { return &MockRootCoordClient_ListPolicy_Call{Call: _e.mock.On("ListPolicy", append([]interface{}{ctx, in}, opts...)...)} @@ -2563,9 +2563,9 @@ type MockRootCoordClient_OperatePrivilege_Call struct { } // OperatePrivilege is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.OperatePrivilegeRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.OperatePrivilegeRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) OperatePrivilege(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_OperatePrivilege_Call { return &MockRootCoordClient_OperatePrivilege_Call{Call: _e.mock.On("OperatePrivilege", append([]interface{}{ctx, in}, opts...)...)} @@ -2633,9 +2633,9 @@ type MockRootCoordClient_OperateUserRole_Call struct { } // OperateUserRole is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.OperateUserRoleRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.OperateUserRoleRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) OperateUserRole(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_OperateUserRole_Call { return &MockRootCoordClient_OperateUserRole_Call{Call: _e.mock.On("OperateUserRole", append([]interface{}{ctx, in}, opts...)...)} @@ -2703,9 +2703,9 @@ type MockRootCoordClient_RenameCollection_Call struct { } // RenameCollection is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.RenameCollectionRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.RenameCollectionRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) RenameCollection(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_RenameCollection_Call { return &MockRootCoordClient_RenameCollection_Call{Call: _e.mock.On("RenameCollection", append([]interface{}{ctx, in}, opts...)...)} @@ -2773,9 +2773,9 @@ type MockRootCoordClient_ReportImport_Call struct { } // ReportImport is a helper method to define mock.On call -// - ctx context.Context -// - in *rootcoordpb.ImportResult -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *rootcoordpb.ImportResult +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) ReportImport(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_ReportImport_Call { return &MockRootCoordClient_ReportImport_Call{Call: _e.mock.On("ReportImport", append([]interface{}{ctx, in}, opts...)...)} @@ -2843,9 +2843,9 @@ type MockRootCoordClient_SelectGrant_Call struct { } // SelectGrant is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.SelectGrantRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.SelectGrantRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) SelectGrant(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_SelectGrant_Call { return &MockRootCoordClient_SelectGrant_Call{Call: _e.mock.On("SelectGrant", append([]interface{}{ctx, in}, opts...)...)} @@ -2913,9 +2913,9 @@ type MockRootCoordClient_SelectRole_Call struct { } // SelectRole is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.SelectRoleRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.SelectRoleRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) SelectRole(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_SelectRole_Call { return &MockRootCoordClient_SelectRole_Call{Call: _e.mock.On("SelectRole", append([]interface{}{ctx, in}, opts...)...)} @@ -2983,9 +2983,9 @@ type MockRootCoordClient_SelectUser_Call struct { } // SelectUser is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.SelectUserRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.SelectUserRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) SelectUser(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_SelectUser_Call { return &MockRootCoordClient_SelectUser_Call{Call: _e.mock.On("SelectUser", append([]interface{}{ctx, in}, opts...)...)} @@ -3053,9 +3053,9 @@ type MockRootCoordClient_ShowCollections_Call struct { } // ShowCollections is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.ShowCollectionsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.ShowCollectionsRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) ShowCollections(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_ShowCollections_Call { return &MockRootCoordClient_ShowCollections_Call{Call: _e.mock.On("ShowCollections", append([]interface{}{ctx, in}, opts...)...)} @@ -3123,9 +3123,9 @@ type MockRootCoordClient_ShowConfigurations_Call struct { } // ShowConfigurations is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.ShowConfigurationsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.ShowConfigurationsRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) ShowConfigurations(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_ShowConfigurations_Call { return &MockRootCoordClient_ShowConfigurations_Call{Call: _e.mock.On("ShowConfigurations", append([]interface{}{ctx, in}, opts...)...)} @@ -3193,9 +3193,9 @@ type MockRootCoordClient_ShowPartitions_Call struct { } // ShowPartitions is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.ShowPartitionsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.ShowPartitionsRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) ShowPartitions(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_ShowPartitions_Call { return &MockRootCoordClient_ShowPartitions_Call{Call: _e.mock.On("ShowPartitions", append([]interface{}{ctx, in}, opts...)...)} @@ -3263,9 +3263,9 @@ type MockRootCoordClient_ShowPartitionsInternal_Call struct { } // ShowPartitionsInternal is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.ShowPartitionsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.ShowPartitionsRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) ShowPartitionsInternal(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_ShowPartitionsInternal_Call { return &MockRootCoordClient_ShowPartitionsInternal_Call{Call: _e.mock.On("ShowPartitionsInternal", append([]interface{}{ctx, in}, opts...)...)} @@ -3333,9 +3333,9 @@ type MockRootCoordClient_ShowSegments_Call struct { } // ShowSegments is a helper method to define mock.On call -// - ctx context.Context -// - in *milvuspb.ShowSegmentsRequest -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *milvuspb.ShowSegmentsRequest +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) ShowSegments(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_ShowSegments_Call { return &MockRootCoordClient_ShowSegments_Call{Call: _e.mock.On("ShowSegments", append([]interface{}{ctx, in}, opts...)...)} @@ -3403,9 +3403,9 @@ type MockRootCoordClient_UpdateChannelTimeTick_Call struct { } // UpdateChannelTimeTick is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.ChannelTimeTickMsg -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.ChannelTimeTickMsg +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) UpdateChannelTimeTick(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_UpdateChannelTimeTick_Call { return &MockRootCoordClient_UpdateChannelTimeTick_Call{Call: _e.mock.On("UpdateChannelTimeTick", append([]interface{}{ctx, in}, opts...)...)} @@ -3473,9 +3473,9 @@ type MockRootCoordClient_UpdateCredential_Call struct { } // UpdateCredential is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.CredentialInfo -// - opts ...grpc.CallOption +// - ctx context.Context +// - in *internalpb.CredentialInfo +// - opts ...grpc.CallOption func (_e *MockRootCoordClient_Expecter) UpdateCredential(ctx interface{}, in interface{}, opts ...interface{}) *MockRootCoordClient_UpdateCredential_Call { return &MockRootCoordClient_UpdateCredential_Call{Call: _e.mock.On("UpdateCredential", append([]interface{}{ctx, in}, opts...)...)}