From 5c0bb405494764b56567692c63b4d903d7031007 Mon Sep 17 00:00:00 2001 From: Bingyi Sun Date: Mon, 11 Mar 2024 17:31:03 +0800 Subject: [PATCH] fix: merge index params when creating index (#31127) issue: https://github.com/milvus-io/milvus/issues/31102 --------- Signed-off-by: sunby --- internal/datacoord/index_service.go | 13 + .../metastore/mocks/mock_datacoord_catalog.go | 106 ++--- internal/mocks/mock_proxy.go | 394 +++++++++--------- internal/mocks/mock_proxy_client.go | 78 ++-- tests/integration/import/util_test.go | 2 +- 5 files changed, 303 insertions(+), 290 deletions(-) diff --git a/internal/datacoord/index_service.go b/internal/datacoord/index_service.go index c16124eb2c00e..513682d493275 100644 --- a/internal/datacoord/index_service.go +++ b/internal/datacoord/index_service.go @@ -205,6 +205,19 @@ func (s *Server) CreateIndex(ctx context.Context, req *indexpb.CreateIndexReques return merr.Status(err), nil } + // merge with previous params because create index would not pass mmap params + indexes := s.meta.indexMeta.GetFieldIndexes(req.GetCollectionID(), req.GetFieldID(), req.GetIndexName()) + if len(indexes) == 1 { + req.UserIndexParams, err = UpdateParams(indexes[0], indexes[0].UserIndexParams, req.GetUserIndexParams()) + if err != nil { + return merr.Status(err), nil + } + req.IndexParams, err = UpdateParams(indexes[0], indexes[0].IndexParams, req.GetIndexParams()) + if err != nil { + return merr.Status(err), nil + } + } + if indexID == 0 { indexID, err = s.allocator.allocID(ctx) if err != nil { diff --git a/internal/metastore/mocks/mock_datacoord_catalog.go b/internal/metastore/mocks/mock_datacoord_catalog.go index b786fa6c633ac..0fec3d050b83f 100644 --- a/internal/metastore/mocks/mock_datacoord_catalog.go +++ b/internal/metastore/mocks/mock_datacoord_catalog.go @@ -48,8 +48,8 @@ type DataCoordCatalog_AddSegment_Call struct { } // AddSegment is a helper method to define mock.On call -// - ctx context.Context -// - segment *datapb.SegmentInfo +// - ctx context.Context +// - segment *datapb.SegmentInfo func (_e *DataCoordCatalog_Expecter) AddSegment(ctx interface{}, segment interface{}) *DataCoordCatalog_AddSegment_Call { return &DataCoordCatalog_AddSegment_Call{Call: _e.mock.On("AddSegment", ctx, segment)} } @@ -91,8 +91,8 @@ type DataCoordCatalog_AlterIndexes_Call struct { } // AlterIndexes is a helper method to define mock.On call -// - ctx context.Context -// - newIndexes []*model.Index +// - ctx context.Context +// - newIndexes []*model.Index func (_e *DataCoordCatalog_Expecter) AlterIndexes(ctx interface{}, newIndexes interface{}) *DataCoordCatalog_AlterIndexes_Call { return &DataCoordCatalog_AlterIndexes_Call{Call: _e.mock.On("AlterIndexes", ctx, newIndexes)} } @@ -134,8 +134,8 @@ type DataCoordCatalog_AlterSegmentIndexes_Call struct { } // AlterSegmentIndexes is a helper method to define mock.On call -// - ctx context.Context -// - newSegIdxes []*model.SegmentIndex +// - ctx context.Context +// - newSegIdxes []*model.SegmentIndex func (_e *DataCoordCatalog_Expecter) AlterSegmentIndexes(ctx interface{}, newSegIdxes interface{}) *DataCoordCatalog_AlterSegmentIndexes_Call { return &DataCoordCatalog_AlterSegmentIndexes_Call{Call: _e.mock.On("AlterSegmentIndexes", ctx, newSegIdxes)} } @@ -184,9 +184,9 @@ type DataCoordCatalog_AlterSegments_Call struct { } // AlterSegments is a helper method to define mock.On call -// - ctx context.Context -// - newSegments []*datapb.SegmentInfo -// - binlogs ...metastore.BinlogsIncrement +// - ctx context.Context +// - newSegments []*datapb.SegmentInfo +// - binlogs ...metastore.BinlogsIncrement func (_e *DataCoordCatalog_Expecter) AlterSegments(ctx interface{}, newSegments interface{}, binlogs ...interface{}) *DataCoordCatalog_AlterSegments_Call { return &DataCoordCatalog_AlterSegments_Call{Call: _e.mock.On("AlterSegments", append([]interface{}{ctx, newSegments}, binlogs...)...)} @@ -235,8 +235,8 @@ type DataCoordCatalog_ChannelExists_Call struct { } // ChannelExists is a helper method to define mock.On call -// - ctx context.Context -// - channel string +// - ctx context.Context +// - channel string func (_e *DataCoordCatalog_Expecter) ChannelExists(ctx interface{}, channel interface{}) *DataCoordCatalog_ChannelExists_Call { return &DataCoordCatalog_ChannelExists_Call{Call: _e.mock.On("ChannelExists", ctx, channel)} } @@ -278,8 +278,8 @@ type DataCoordCatalog_CreateIndex_Call struct { } // CreateIndex is a helper method to define mock.On call -// - ctx context.Context -// - index *model.Index +// - ctx context.Context +// - index *model.Index func (_e *DataCoordCatalog_Expecter) CreateIndex(ctx interface{}, index interface{}) *DataCoordCatalog_CreateIndex_Call { return &DataCoordCatalog_CreateIndex_Call{Call: _e.mock.On("CreateIndex", ctx, index)} } @@ -321,8 +321,8 @@ type DataCoordCatalog_CreateSegmentIndex_Call struct { } // CreateSegmentIndex is a helper method to define mock.On call -// - ctx context.Context -// - segIdx *model.SegmentIndex +// - ctx context.Context +// - segIdx *model.SegmentIndex func (_e *DataCoordCatalog_Expecter) CreateSegmentIndex(ctx interface{}, segIdx interface{}) *DataCoordCatalog_CreateSegmentIndex_Call { return &DataCoordCatalog_CreateSegmentIndex_Call{Call: _e.mock.On("CreateSegmentIndex", ctx, segIdx)} } @@ -364,8 +364,8 @@ type DataCoordCatalog_DropChannel_Call struct { } // DropChannel is a helper method to define mock.On call -// - ctx context.Context -// - channel string +// - ctx context.Context +// - channel string func (_e *DataCoordCatalog_Expecter) DropChannel(ctx interface{}, channel interface{}) *DataCoordCatalog_DropChannel_Call { return &DataCoordCatalog_DropChannel_Call{Call: _e.mock.On("DropChannel", ctx, channel)} } @@ -407,8 +407,8 @@ type DataCoordCatalog_DropChannelCheckpoint_Call struct { } // DropChannelCheckpoint is a helper method to define mock.On call -// - ctx context.Context -// - vChannel string +// - ctx context.Context +// - vChannel string func (_e *DataCoordCatalog_Expecter) DropChannelCheckpoint(ctx interface{}, vChannel interface{}) *DataCoordCatalog_DropChannelCheckpoint_Call { return &DataCoordCatalog_DropChannelCheckpoint_Call{Call: _e.mock.On("DropChannelCheckpoint", ctx, vChannel)} } @@ -450,7 +450,7 @@ type DataCoordCatalog_DropImportJob_Call struct { } // DropImportJob is a helper method to define mock.On call -// - jobID int64 +// - jobID int64 func (_e *DataCoordCatalog_Expecter) DropImportJob(jobID interface{}) *DataCoordCatalog_DropImportJob_Call { return &DataCoordCatalog_DropImportJob_Call{Call: _e.mock.On("DropImportJob", jobID)} } @@ -492,7 +492,7 @@ type DataCoordCatalog_DropImportTask_Call struct { } // DropImportTask is a helper method to define mock.On call -// - taskID int64 +// - taskID int64 func (_e *DataCoordCatalog_Expecter) DropImportTask(taskID interface{}) *DataCoordCatalog_DropImportTask_Call { return &DataCoordCatalog_DropImportTask_Call{Call: _e.mock.On("DropImportTask", taskID)} } @@ -534,9 +534,9 @@ type DataCoordCatalog_DropIndex_Call struct { } // DropIndex is a helper method to define mock.On call -// - ctx context.Context -// - collID int64 -// - dropIdxID int64 +// - ctx context.Context +// - collID int64 +// - dropIdxID int64 func (_e *DataCoordCatalog_Expecter) DropIndex(ctx interface{}, collID interface{}, dropIdxID interface{}) *DataCoordCatalog_DropIndex_Call { return &DataCoordCatalog_DropIndex_Call{Call: _e.mock.On("DropIndex", ctx, collID, dropIdxID)} } @@ -578,7 +578,7 @@ type DataCoordCatalog_DropPreImportTask_Call struct { } // DropPreImportTask is a helper method to define mock.On call -// - taskID int64 +// - taskID int64 func (_e *DataCoordCatalog_Expecter) DropPreImportTask(taskID interface{}) *DataCoordCatalog_DropPreImportTask_Call { return &DataCoordCatalog_DropPreImportTask_Call{Call: _e.mock.On("DropPreImportTask", taskID)} } @@ -620,8 +620,8 @@ type DataCoordCatalog_DropSegment_Call struct { } // DropSegment is a helper method to define mock.On call -// - ctx context.Context -// - segment *datapb.SegmentInfo +// - ctx context.Context +// - segment *datapb.SegmentInfo func (_e *DataCoordCatalog_Expecter) DropSegment(ctx interface{}, segment interface{}) *DataCoordCatalog_DropSegment_Call { return &DataCoordCatalog_DropSegment_Call{Call: _e.mock.On("DropSegment", ctx, segment)} } @@ -663,11 +663,11 @@ type DataCoordCatalog_DropSegmentIndex_Call struct { } // DropSegmentIndex is a helper method to define mock.On call -// - ctx context.Context -// - collID int64 -// - partID int64 -// - segID int64 -// - buildID int64 +// - ctx context.Context +// - collID int64 +// - partID int64 +// - segID int64 +// - buildID int64 func (_e *DataCoordCatalog_Expecter) DropSegmentIndex(ctx interface{}, collID interface{}, partID interface{}, segID interface{}, buildID interface{}) *DataCoordCatalog_DropSegmentIndex_Call { return &DataCoordCatalog_DropSegmentIndex_Call{Call: _e.mock.On("DropSegmentIndex", ctx, collID, partID, segID, buildID)} } @@ -709,9 +709,9 @@ type DataCoordCatalog_GcConfirm_Call struct { } // GcConfirm is a helper method to define mock.On call -// - ctx context.Context -// - collectionID int64 -// - partitionID int64 +// - ctx context.Context +// - collectionID int64 +// - partitionID int64 func (_e *DataCoordCatalog_Expecter) GcConfirm(ctx interface{}, collectionID interface{}, partitionID interface{}) *DataCoordCatalog_GcConfirm_Call { return &DataCoordCatalog_GcConfirm_Call{Call: _e.mock.On("GcConfirm", ctx, collectionID, partitionID)} } @@ -765,7 +765,7 @@ type DataCoordCatalog_ListChannelCheckpoint_Call struct { } // ListChannelCheckpoint is a helper method to define mock.On call -// - ctx context.Context +// - ctx context.Context func (_e *DataCoordCatalog_Expecter) ListChannelCheckpoint(ctx interface{}) *DataCoordCatalog_ListChannelCheckpoint_Call { return &DataCoordCatalog_ListChannelCheckpoint_Call{Call: _e.mock.On("ListChannelCheckpoint", ctx)} } @@ -925,7 +925,7 @@ type DataCoordCatalog_ListIndexes_Call struct { } // ListIndexes is a helper method to define mock.On call -// - ctx context.Context +// - ctx context.Context func (_e *DataCoordCatalog_Expecter) ListIndexes(ctx interface{}) *DataCoordCatalog_ListIndexes_Call { return &DataCoordCatalog_ListIndexes_Call{Call: _e.mock.On("ListIndexes", ctx)} } @@ -1032,7 +1032,7 @@ type DataCoordCatalog_ListSegmentIndexes_Call struct { } // ListSegmentIndexes is a helper method to define mock.On call -// - ctx context.Context +// - ctx context.Context func (_e *DataCoordCatalog_Expecter) ListSegmentIndexes(ctx interface{}) *DataCoordCatalog_ListSegmentIndexes_Call { return &DataCoordCatalog_ListSegmentIndexes_Call{Call: _e.mock.On("ListSegmentIndexes", ctx)} } @@ -1086,7 +1086,7 @@ type DataCoordCatalog_ListSegments_Call struct { } // ListSegments is a helper method to define mock.On call -// - ctx context.Context +// - ctx context.Context func (_e *DataCoordCatalog_Expecter) ListSegments(ctx interface{}) *DataCoordCatalog_ListSegments_Call { return &DataCoordCatalog_ListSegments_Call{Call: _e.mock.On("ListSegments", ctx)} } @@ -1128,8 +1128,8 @@ type DataCoordCatalog_MarkChannelAdded_Call struct { } // MarkChannelAdded is a helper method to define mock.On call -// - ctx context.Context -// - channel string +// - ctx context.Context +// - channel string func (_e *DataCoordCatalog_Expecter) MarkChannelAdded(ctx interface{}, channel interface{}) *DataCoordCatalog_MarkChannelAdded_Call { return &DataCoordCatalog_MarkChannelAdded_Call{Call: _e.mock.On("MarkChannelAdded", ctx, channel)} } @@ -1171,8 +1171,8 @@ type DataCoordCatalog_MarkChannelDeleted_Call struct { } // MarkChannelDeleted is a helper method to define mock.On call -// - ctx context.Context -// - channel string +// - ctx context.Context +// - channel string func (_e *DataCoordCatalog_Expecter) MarkChannelDeleted(ctx interface{}, channel interface{}) *DataCoordCatalog_MarkChannelDeleted_Call { return &DataCoordCatalog_MarkChannelDeleted_Call{Call: _e.mock.On("MarkChannelDeleted", ctx, channel)} } @@ -1214,9 +1214,9 @@ type DataCoordCatalog_SaveChannelCheckpoint_Call struct { } // SaveChannelCheckpoint is a helper method to define mock.On call -// - ctx context.Context -// - vChannel string -// - pos *msgpb.MsgPosition +// - ctx context.Context +// - vChannel string +// - pos *msgpb.MsgPosition func (_e *DataCoordCatalog_Expecter) SaveChannelCheckpoint(ctx interface{}, vChannel interface{}, pos interface{}) *DataCoordCatalog_SaveChannelCheckpoint_Call { return &DataCoordCatalog_SaveChannelCheckpoint_Call{Call: _e.mock.On("SaveChannelCheckpoint", ctx, vChannel, pos)} } @@ -1301,8 +1301,8 @@ type DataCoordCatalog_SaveDroppedSegmentsInBatch_Call struct { } // SaveDroppedSegmentsInBatch is a helper method to define mock.On call -// - ctx context.Context -// - segments []*datapb.SegmentInfo +// - ctx context.Context +// - segments []*datapb.SegmentInfo func (_e *DataCoordCatalog_Expecter) SaveDroppedSegmentsInBatch(ctx interface{}, segments interface{}) *DataCoordCatalog_SaveDroppedSegmentsInBatch_Call { return &DataCoordCatalog_SaveDroppedSegmentsInBatch_Call{Call: _e.mock.On("SaveDroppedSegmentsInBatch", ctx, segments)} } @@ -1344,7 +1344,7 @@ type DataCoordCatalog_SaveImportJob_Call struct { } // SaveImportJob is a helper method to define mock.On call -// - job *datapb.ImportJob +// - job *datapb.ImportJob func (_e *DataCoordCatalog_Expecter) SaveImportJob(job interface{}) *DataCoordCatalog_SaveImportJob_Call { return &DataCoordCatalog_SaveImportJob_Call{Call: _e.mock.On("SaveImportJob", job)} } @@ -1386,7 +1386,7 @@ type DataCoordCatalog_SaveImportTask_Call struct { } // SaveImportTask is a helper method to define mock.On call -// - task *datapb.ImportTaskV2 +// - task *datapb.ImportTaskV2 func (_e *DataCoordCatalog_Expecter) SaveImportTask(task interface{}) *DataCoordCatalog_SaveImportTask_Call { return &DataCoordCatalog_SaveImportTask_Call{Call: _e.mock.On("SaveImportTask", task)} } @@ -1428,7 +1428,7 @@ type DataCoordCatalog_SavePreImportTask_Call struct { } // SavePreImportTask is a helper method to define mock.On call -// - task *datapb.PreImportTask +// - task *datapb.PreImportTask func (_e *DataCoordCatalog_Expecter) SavePreImportTask(task interface{}) *DataCoordCatalog_SavePreImportTask_Call { return &DataCoordCatalog_SavePreImportTask_Call{Call: _e.mock.On("SavePreImportTask", task)} } @@ -1470,8 +1470,8 @@ type DataCoordCatalog_ShouldDropChannel_Call struct { } // ShouldDropChannel is a helper method to define mock.On call -// - ctx context.Context -// - channel string +// - ctx context.Context +// - channel string func (_e *DataCoordCatalog_Expecter) ShouldDropChannel(ctx interface{}, channel interface{}) *DataCoordCatalog_ShouldDropChannel_Call { return &DataCoordCatalog_ShouldDropChannel_Call{Call: _e.mock.On("ShouldDropChannel", ctx, channel)} } diff --git a/internal/mocks/mock_proxy.go b/internal/mocks/mock_proxy.go index c380b9a5070fa..bb8b37661543b 100644 --- a/internal/mocks/mock_proxy.go +++ b/internal/mocks/mock_proxy.go @@ -66,8 +66,8 @@ type MockProxy_AllocTimestamp_Call struct { } // AllocTimestamp is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.AllocTimestampRequest +// - _a0 context.Context +// - _a1 *milvuspb.AllocTimestampRequest func (_e *MockProxy_Expecter) AllocTimestamp(_a0 interface{}, _a1 interface{}) *MockProxy_AllocTimestamp_Call { return &MockProxy_AllocTimestamp_Call{Call: _e.mock.On("AllocTimestamp", _a0, _a1)} } @@ -121,8 +121,8 @@ type MockProxy_AlterAlias_Call struct { } // AlterAlias is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.AlterAliasRequest +// - _a0 context.Context +// - _a1 *milvuspb.AlterAliasRequest func (_e *MockProxy_Expecter) AlterAlias(_a0 interface{}, _a1 interface{}) *MockProxy_AlterAlias_Call { return &MockProxy_AlterAlias_Call{Call: _e.mock.On("AlterAlias", _a0, _a1)} } @@ -176,8 +176,8 @@ type MockProxy_AlterCollection_Call struct { } // AlterCollection is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.AlterCollectionRequest +// - _a0 context.Context +// - _a1 *milvuspb.AlterCollectionRequest func (_e *MockProxy_Expecter) AlterCollection(_a0 interface{}, _a1 interface{}) *MockProxy_AlterCollection_Call { return &MockProxy_AlterCollection_Call{Call: _e.mock.On("AlterCollection", _a0, _a1)} } @@ -231,8 +231,8 @@ type MockProxy_AlterIndex_Call struct { } // AlterIndex is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.AlterIndexRequest +// - _a0 context.Context +// - _a1 *milvuspb.AlterIndexRequest func (_e *MockProxy_Expecter) AlterIndex(_a0 interface{}, _a1 interface{}) *MockProxy_AlterIndex_Call { return &MockProxy_AlterIndex_Call{Call: _e.mock.On("AlterIndex", _a0, _a1)} } @@ -286,8 +286,8 @@ type MockProxy_CalcDistance_Call struct { } // CalcDistance is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.CalcDistanceRequest +// - _a0 context.Context +// - _a1 *milvuspb.CalcDistanceRequest func (_e *MockProxy_Expecter) CalcDistance(_a0 interface{}, _a1 interface{}) *MockProxy_CalcDistance_Call { return &MockProxy_CalcDistance_Call{Call: _e.mock.On("CalcDistance", _a0, _a1)} } @@ -341,8 +341,8 @@ type MockProxy_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 *MockProxy_Expecter) CheckHealth(_a0 interface{}, _a1 interface{}) *MockProxy_CheckHealth_Call { return &MockProxy_CheckHealth_Call{Call: _e.mock.On("CheckHealth", _a0, _a1)} } @@ -396,8 +396,8 @@ type MockProxy_Connect_Call struct { } // Connect is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ConnectRequest +// - _a0 context.Context +// - _a1 *milvuspb.ConnectRequest func (_e *MockProxy_Expecter) Connect(_a0 interface{}, _a1 interface{}) *MockProxy_Connect_Call { return &MockProxy_Connect_Call{Call: _e.mock.On("Connect", _a0, _a1)} } @@ -451,8 +451,8 @@ type MockProxy_CreateAlias_Call struct { } // CreateAlias is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.CreateAliasRequest +// - _a0 context.Context +// - _a1 *milvuspb.CreateAliasRequest func (_e *MockProxy_Expecter) CreateAlias(_a0 interface{}, _a1 interface{}) *MockProxy_CreateAlias_Call { return &MockProxy_CreateAlias_Call{Call: _e.mock.On("CreateAlias", _a0, _a1)} } @@ -506,8 +506,8 @@ type MockProxy_CreateCollection_Call struct { } // CreateCollection is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.CreateCollectionRequest +// - _a0 context.Context +// - _a1 *milvuspb.CreateCollectionRequest func (_e *MockProxy_Expecter) CreateCollection(_a0 interface{}, _a1 interface{}) *MockProxy_CreateCollection_Call { return &MockProxy_CreateCollection_Call{Call: _e.mock.On("CreateCollection", _a0, _a1)} } @@ -561,8 +561,8 @@ type MockProxy_CreateCredential_Call struct { } // CreateCredential is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.CreateCredentialRequest +// - _a0 context.Context +// - _a1 *milvuspb.CreateCredentialRequest func (_e *MockProxy_Expecter) CreateCredential(_a0 interface{}, _a1 interface{}) *MockProxy_CreateCredential_Call { return &MockProxy_CreateCredential_Call{Call: _e.mock.On("CreateCredential", _a0, _a1)} } @@ -616,8 +616,8 @@ type MockProxy_CreateDatabase_Call struct { } // CreateDatabase is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.CreateDatabaseRequest +// - _a0 context.Context +// - _a1 *milvuspb.CreateDatabaseRequest func (_e *MockProxy_Expecter) CreateDatabase(_a0 interface{}, _a1 interface{}) *MockProxy_CreateDatabase_Call { return &MockProxy_CreateDatabase_Call{Call: _e.mock.On("CreateDatabase", _a0, _a1)} } @@ -671,8 +671,8 @@ type MockProxy_CreateIndex_Call struct { } // CreateIndex is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.CreateIndexRequest +// - _a0 context.Context +// - _a1 *milvuspb.CreateIndexRequest func (_e *MockProxy_Expecter) CreateIndex(_a0 interface{}, _a1 interface{}) *MockProxy_CreateIndex_Call { return &MockProxy_CreateIndex_Call{Call: _e.mock.On("CreateIndex", _a0, _a1)} } @@ -726,8 +726,8 @@ type MockProxy_CreatePartition_Call struct { } // CreatePartition is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.CreatePartitionRequest +// - _a0 context.Context +// - _a1 *milvuspb.CreatePartitionRequest func (_e *MockProxy_Expecter) CreatePartition(_a0 interface{}, _a1 interface{}) *MockProxy_CreatePartition_Call { return &MockProxy_CreatePartition_Call{Call: _e.mock.On("CreatePartition", _a0, _a1)} } @@ -781,8 +781,8 @@ type MockProxy_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 *MockProxy_Expecter) CreateResourceGroup(_a0 interface{}, _a1 interface{}) *MockProxy_CreateResourceGroup_Call { return &MockProxy_CreateResourceGroup_Call{Call: _e.mock.On("CreateResourceGroup", _a0, _a1)} } @@ -836,8 +836,8 @@ type MockProxy_CreateRole_Call struct { } // CreateRole is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.CreateRoleRequest +// - _a0 context.Context +// - _a1 *milvuspb.CreateRoleRequest func (_e *MockProxy_Expecter) CreateRole(_a0 interface{}, _a1 interface{}) *MockProxy_CreateRole_Call { return &MockProxy_CreateRole_Call{Call: _e.mock.On("CreateRole", _a0, _a1)} } @@ -891,8 +891,8 @@ type MockProxy_Delete_Call struct { } // Delete is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DeleteRequest +// - _a0 context.Context +// - _a1 *milvuspb.DeleteRequest func (_e *MockProxy_Expecter) Delete(_a0 interface{}, _a1 interface{}) *MockProxy_Delete_Call { return &MockProxy_Delete_Call{Call: _e.mock.On("Delete", _a0, _a1)} } @@ -946,8 +946,8 @@ type MockProxy_DeleteCredential_Call struct { } // DeleteCredential is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DeleteCredentialRequest +// - _a0 context.Context +// - _a1 *milvuspb.DeleteCredentialRequest func (_e *MockProxy_Expecter) DeleteCredential(_a0 interface{}, _a1 interface{}) *MockProxy_DeleteCredential_Call { return &MockProxy_DeleteCredential_Call{Call: _e.mock.On("DeleteCredential", _a0, _a1)} } @@ -1001,8 +1001,8 @@ type MockProxy_DescribeAlias_Call struct { } // DescribeAlias is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DescribeAliasRequest +// - _a0 context.Context +// - _a1 *milvuspb.DescribeAliasRequest func (_e *MockProxy_Expecter) DescribeAlias(_a0 interface{}, _a1 interface{}) *MockProxy_DescribeAlias_Call { return &MockProxy_DescribeAlias_Call{Call: _e.mock.On("DescribeAlias", _a0, _a1)} } @@ -1056,8 +1056,8 @@ type MockProxy_DescribeCollection_Call struct { } // DescribeCollection is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DescribeCollectionRequest +// - _a0 context.Context +// - _a1 *milvuspb.DescribeCollectionRequest func (_e *MockProxy_Expecter) DescribeCollection(_a0 interface{}, _a1 interface{}) *MockProxy_DescribeCollection_Call { return &MockProxy_DescribeCollection_Call{Call: _e.mock.On("DescribeCollection", _a0, _a1)} } @@ -1111,8 +1111,8 @@ type MockProxy_DescribeIndex_Call struct { } // DescribeIndex is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DescribeIndexRequest +// - _a0 context.Context +// - _a1 *milvuspb.DescribeIndexRequest func (_e *MockProxy_Expecter) DescribeIndex(_a0 interface{}, _a1 interface{}) *MockProxy_DescribeIndex_Call { return &MockProxy_DescribeIndex_Call{Call: _e.mock.On("DescribeIndex", _a0, _a1)} } @@ -1166,8 +1166,8 @@ type MockProxy_DescribeResourceGroup_Call struct { } // DescribeResourceGroup is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DescribeResourceGroupRequest +// - _a0 context.Context +// - _a1 *milvuspb.DescribeResourceGroupRequest func (_e *MockProxy_Expecter) DescribeResourceGroup(_a0 interface{}, _a1 interface{}) *MockProxy_DescribeResourceGroup_Call { return &MockProxy_DescribeResourceGroup_Call{Call: _e.mock.On("DescribeResourceGroup", _a0, _a1)} } @@ -1221,8 +1221,8 @@ type MockProxy_DescribeSegmentIndexData_Call struct { } // DescribeSegmentIndexData is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *federpb.DescribeSegmentIndexDataRequest +// - _a0 context.Context +// - _a1 *federpb.DescribeSegmentIndexDataRequest func (_e *MockProxy_Expecter) DescribeSegmentIndexData(_a0 interface{}, _a1 interface{}) *MockProxy_DescribeSegmentIndexData_Call { return &MockProxy_DescribeSegmentIndexData_Call{Call: _e.mock.On("DescribeSegmentIndexData", _a0, _a1)} } @@ -1276,8 +1276,8 @@ type MockProxy_DropAlias_Call struct { } // DropAlias is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DropAliasRequest +// - _a0 context.Context +// - _a1 *milvuspb.DropAliasRequest func (_e *MockProxy_Expecter) DropAlias(_a0 interface{}, _a1 interface{}) *MockProxy_DropAlias_Call { return &MockProxy_DropAlias_Call{Call: _e.mock.On("DropAlias", _a0, _a1)} } @@ -1331,8 +1331,8 @@ type MockProxy_DropCollection_Call struct { } // DropCollection is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DropCollectionRequest +// - _a0 context.Context +// - _a1 *milvuspb.DropCollectionRequest func (_e *MockProxy_Expecter) DropCollection(_a0 interface{}, _a1 interface{}) *MockProxy_DropCollection_Call { return &MockProxy_DropCollection_Call{Call: _e.mock.On("DropCollection", _a0, _a1)} } @@ -1386,8 +1386,8 @@ type MockProxy_DropDatabase_Call struct { } // DropDatabase is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DropDatabaseRequest +// - _a0 context.Context +// - _a1 *milvuspb.DropDatabaseRequest func (_e *MockProxy_Expecter) DropDatabase(_a0 interface{}, _a1 interface{}) *MockProxy_DropDatabase_Call { return &MockProxy_DropDatabase_Call{Call: _e.mock.On("DropDatabase", _a0, _a1)} } @@ -1441,8 +1441,8 @@ type MockProxy_DropIndex_Call struct { } // DropIndex is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DropIndexRequest +// - _a0 context.Context +// - _a1 *milvuspb.DropIndexRequest func (_e *MockProxy_Expecter) DropIndex(_a0 interface{}, _a1 interface{}) *MockProxy_DropIndex_Call { return &MockProxy_DropIndex_Call{Call: _e.mock.On("DropIndex", _a0, _a1)} } @@ -1496,8 +1496,8 @@ type MockProxy_DropPartition_Call struct { } // DropPartition is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DropPartitionRequest +// - _a0 context.Context +// - _a1 *milvuspb.DropPartitionRequest func (_e *MockProxy_Expecter) DropPartition(_a0 interface{}, _a1 interface{}) *MockProxy_DropPartition_Call { return &MockProxy_DropPartition_Call{Call: _e.mock.On("DropPartition", _a0, _a1)} } @@ -1551,8 +1551,8 @@ type MockProxy_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 *MockProxy_Expecter) DropResourceGroup(_a0 interface{}, _a1 interface{}) *MockProxy_DropResourceGroup_Call { return &MockProxy_DropResourceGroup_Call{Call: _e.mock.On("DropResourceGroup", _a0, _a1)} } @@ -1606,8 +1606,8 @@ type MockProxy_DropRole_Call struct { } // DropRole is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DropRoleRequest +// - _a0 context.Context +// - _a1 *milvuspb.DropRoleRequest func (_e *MockProxy_Expecter) DropRole(_a0 interface{}, _a1 interface{}) *MockProxy_DropRole_Call { return &MockProxy_DropRole_Call{Call: _e.mock.On("DropRole", _a0, _a1)} } @@ -1661,8 +1661,8 @@ type MockProxy_Dummy_Call struct { } // Dummy is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.DummyRequest +// - _a0 context.Context +// - _a1 *milvuspb.DummyRequest func (_e *MockProxy_Expecter) Dummy(_a0 interface{}, _a1 interface{}) *MockProxy_Dummy_Call { return &MockProxy_Dummy_Call{Call: _e.mock.On("Dummy", _a0, _a1)} } @@ -1716,8 +1716,8 @@ type MockProxy_Flush_Call struct { } // Flush is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.FlushRequest +// - _a0 context.Context +// - _a1 *milvuspb.FlushRequest func (_e *MockProxy_Expecter) Flush(_a0 interface{}, _a1 interface{}) *MockProxy_Flush_Call { return &MockProxy_Flush_Call{Call: _e.mock.On("Flush", _a0, _a1)} } @@ -1771,8 +1771,8 @@ type MockProxy_FlushAll_Call struct { } // FlushAll is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.FlushAllRequest +// - _a0 context.Context +// - _a1 *milvuspb.FlushAllRequest func (_e *MockProxy_Expecter) FlushAll(_a0 interface{}, _a1 interface{}) *MockProxy_FlushAll_Call { return &MockProxy_FlushAll_Call{Call: _e.mock.On("FlushAll", _a0, _a1)} } @@ -1867,8 +1867,8 @@ type MockProxy_GetCollectionStatistics_Call struct { } // GetCollectionStatistics is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetCollectionStatisticsRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetCollectionStatisticsRequest func (_e *MockProxy_Expecter) GetCollectionStatistics(_a0 interface{}, _a1 interface{}) *MockProxy_GetCollectionStatistics_Call { return &MockProxy_GetCollectionStatistics_Call{Call: _e.mock.On("GetCollectionStatistics", _a0, _a1)} } @@ -1922,8 +1922,8 @@ type MockProxy_GetCompactionState_Call struct { } // GetCompactionState is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetCompactionStateRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetCompactionStateRequest func (_e *MockProxy_Expecter) GetCompactionState(_a0 interface{}, _a1 interface{}) *MockProxy_GetCompactionState_Call { return &MockProxy_GetCompactionState_Call{Call: _e.mock.On("GetCompactionState", _a0, _a1)} } @@ -1977,8 +1977,8 @@ type MockProxy_GetCompactionStateWithPlans_Call struct { } // GetCompactionStateWithPlans is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetCompactionPlansRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetCompactionPlansRequest func (_e *MockProxy_Expecter) GetCompactionStateWithPlans(_a0 interface{}, _a1 interface{}) *MockProxy_GetCompactionStateWithPlans_Call { return &MockProxy_GetCompactionStateWithPlans_Call{Call: _e.mock.On("GetCompactionStateWithPlans", _a0, _a1)} } @@ -2032,8 +2032,8 @@ type MockProxy_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 *MockProxy_Expecter) GetComponentStates(_a0 interface{}, _a1 interface{}) *MockProxy_GetComponentStates_Call { return &MockProxy_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", _a0, _a1)} } @@ -2087,8 +2087,8 @@ type MockProxy_GetDdChannel_Call struct { } // GetDdChannel is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.GetDdChannelRequest +// - _a0 context.Context +// - _a1 *internalpb.GetDdChannelRequest func (_e *MockProxy_Expecter) GetDdChannel(_a0 interface{}, _a1 interface{}) *MockProxy_GetDdChannel_Call { return &MockProxy_GetDdChannel_Call{Call: _e.mock.On("GetDdChannel", _a0, _a1)} } @@ -2142,8 +2142,8 @@ type MockProxy_GetFlushAllState_Call struct { } // GetFlushAllState is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetFlushAllStateRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetFlushAllStateRequest func (_e *MockProxy_Expecter) GetFlushAllState(_a0 interface{}, _a1 interface{}) *MockProxy_GetFlushAllState_Call { return &MockProxy_GetFlushAllState_Call{Call: _e.mock.On("GetFlushAllState", _a0, _a1)} } @@ -2197,8 +2197,8 @@ type MockProxy_GetFlushState_Call struct { } // GetFlushState is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetFlushStateRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetFlushStateRequest func (_e *MockProxy_Expecter) GetFlushState(_a0 interface{}, _a1 interface{}) *MockProxy_GetFlushState_Call { return &MockProxy_GetFlushState_Call{Call: _e.mock.On("GetFlushState", _a0, _a1)} } @@ -2252,8 +2252,8 @@ type MockProxy_GetImportProgress_Call struct { } // GetImportProgress is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.GetImportProgressRequest +// - _a0 context.Context +// - _a1 *internalpb.GetImportProgressRequest func (_e *MockProxy_Expecter) GetImportProgress(_a0 interface{}, _a1 interface{}) *MockProxy_GetImportProgress_Call { return &MockProxy_GetImportProgress_Call{Call: _e.mock.On("GetImportProgress", _a0, _a1)} } @@ -2307,8 +2307,8 @@ type MockProxy_GetImportState_Call struct { } // GetImportState is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetImportStateRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetImportStateRequest func (_e *MockProxy_Expecter) GetImportState(_a0 interface{}, _a1 interface{}) *MockProxy_GetImportState_Call { return &MockProxy_GetImportState_Call{Call: _e.mock.On("GetImportState", _a0, _a1)} } @@ -2362,8 +2362,8 @@ type MockProxy_GetIndexBuildProgress_Call struct { } // GetIndexBuildProgress is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetIndexBuildProgressRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetIndexBuildProgressRequest func (_e *MockProxy_Expecter) GetIndexBuildProgress(_a0 interface{}, _a1 interface{}) *MockProxy_GetIndexBuildProgress_Call { return &MockProxy_GetIndexBuildProgress_Call{Call: _e.mock.On("GetIndexBuildProgress", _a0, _a1)} } @@ -2417,8 +2417,8 @@ type MockProxy_GetIndexState_Call struct { } // GetIndexState is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetIndexStateRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetIndexStateRequest func (_e *MockProxy_Expecter) GetIndexState(_a0 interface{}, _a1 interface{}) *MockProxy_GetIndexState_Call { return &MockProxy_GetIndexState_Call{Call: _e.mock.On("GetIndexState", _a0, _a1)} } @@ -2472,8 +2472,8 @@ type MockProxy_GetIndexStatistics_Call struct { } // GetIndexStatistics is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetIndexStatisticsRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetIndexStatisticsRequest func (_e *MockProxy_Expecter) GetIndexStatistics(_a0 interface{}, _a1 interface{}) *MockProxy_GetIndexStatistics_Call { return &MockProxy_GetIndexStatistics_Call{Call: _e.mock.On("GetIndexStatistics", _a0, _a1)} } @@ -2527,8 +2527,8 @@ type MockProxy_GetLoadState_Call struct { } // GetLoadState is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetLoadStateRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetLoadStateRequest func (_e *MockProxy_Expecter) GetLoadState(_a0 interface{}, _a1 interface{}) *MockProxy_GetLoadState_Call { return &MockProxy_GetLoadState_Call{Call: _e.mock.On("GetLoadState", _a0, _a1)} } @@ -2582,8 +2582,8 @@ type MockProxy_GetLoadingProgress_Call struct { } // GetLoadingProgress is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetLoadingProgressRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetLoadingProgressRequest func (_e *MockProxy_Expecter) GetLoadingProgress(_a0 interface{}, _a1 interface{}) *MockProxy_GetLoadingProgress_Call { return &MockProxy_GetLoadingProgress_Call{Call: _e.mock.On("GetLoadingProgress", _a0, _a1)} } @@ -2637,8 +2637,8 @@ type MockProxy_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 *MockProxy_Expecter) GetMetrics(_a0 interface{}, _a1 interface{}) *MockProxy_GetMetrics_Call { return &MockProxy_GetMetrics_Call{Call: _e.mock.On("GetMetrics", _a0, _a1)} } @@ -2692,8 +2692,8 @@ type MockProxy_GetPartitionStatistics_Call struct { } // GetPartitionStatistics is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetPartitionStatisticsRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetPartitionStatisticsRequest func (_e *MockProxy_Expecter) GetPartitionStatistics(_a0 interface{}, _a1 interface{}) *MockProxy_GetPartitionStatistics_Call { return &MockProxy_GetPartitionStatistics_Call{Call: _e.mock.On("GetPartitionStatistics", _a0, _a1)} } @@ -2747,8 +2747,8 @@ type MockProxy_GetPersistentSegmentInfo_Call struct { } // GetPersistentSegmentInfo is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetPersistentSegmentInfoRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetPersistentSegmentInfoRequest func (_e *MockProxy_Expecter) GetPersistentSegmentInfo(_a0 interface{}, _a1 interface{}) *MockProxy_GetPersistentSegmentInfo_Call { return &MockProxy_GetPersistentSegmentInfo_Call{Call: _e.mock.On("GetPersistentSegmentInfo", _a0, _a1)} } @@ -2802,8 +2802,8 @@ type MockProxy_GetProxyMetrics_Call struct { } // GetProxyMetrics is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetMetricsRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetMetricsRequest func (_e *MockProxy_Expecter) GetProxyMetrics(_a0 interface{}, _a1 interface{}) *MockProxy_GetProxyMetrics_Call { return &MockProxy_GetProxyMetrics_Call{Call: _e.mock.On("GetProxyMetrics", _a0, _a1)} } @@ -2857,8 +2857,8 @@ type MockProxy_GetQuerySegmentInfo_Call struct { } // GetQuerySegmentInfo is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetQuerySegmentInfoRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetQuerySegmentInfoRequest func (_e *MockProxy_Expecter) GetQuerySegmentInfo(_a0 interface{}, _a1 interface{}) *MockProxy_GetQuerySegmentInfo_Call { return &MockProxy_GetQuerySegmentInfo_Call{Call: _e.mock.On("GetQuerySegmentInfo", _a0, _a1)} } @@ -2965,8 +2965,8 @@ type MockProxy_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 *MockProxy_Expecter) GetReplicas(_a0 interface{}, _a1 interface{}) *MockProxy_GetReplicas_Call { return &MockProxy_GetReplicas_Call{Call: _e.mock.On("GetReplicas", _a0, _a1)} } @@ -3020,8 +3020,8 @@ type MockProxy_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 *MockProxy_Expecter) GetStatisticsChannel(_a0 interface{}, _a1 interface{}) *MockProxy_GetStatisticsChannel_Call { return &MockProxy_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", _a0, _a1)} } @@ -3075,8 +3075,8 @@ type MockProxy_GetVersion_Call struct { } // GetVersion is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.GetVersionRequest +// - _a0 context.Context +// - _a1 *milvuspb.GetVersionRequest func (_e *MockProxy_Expecter) GetVersion(_a0 interface{}, _a1 interface{}) *MockProxy_GetVersion_Call { return &MockProxy_GetVersion_Call{Call: _e.mock.On("GetVersion", _a0, _a1)} } @@ -3130,8 +3130,8 @@ type MockProxy_HasCollection_Call struct { } // HasCollection is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.HasCollectionRequest +// - _a0 context.Context +// - _a1 *milvuspb.HasCollectionRequest func (_e *MockProxy_Expecter) HasCollection(_a0 interface{}, _a1 interface{}) *MockProxy_HasCollection_Call { return &MockProxy_HasCollection_Call{Call: _e.mock.On("HasCollection", _a0, _a1)} } @@ -3185,8 +3185,8 @@ type MockProxy_HasPartition_Call struct { } // HasPartition is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.HasPartitionRequest +// - _a0 context.Context +// - _a1 *milvuspb.HasPartitionRequest func (_e *MockProxy_Expecter) HasPartition(_a0 interface{}, _a1 interface{}) *MockProxy_HasPartition_Call { return &MockProxy_HasPartition_Call{Call: _e.mock.On("HasPartition", _a0, _a1)} } @@ -3240,8 +3240,8 @@ type MockProxy_HybridSearch_Call struct { } // HybridSearch is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.HybridSearchRequest +// - _a0 context.Context +// - _a1 *milvuspb.HybridSearchRequest func (_e *MockProxy_Expecter) HybridSearch(_a0 interface{}, _a1 interface{}) *MockProxy_HybridSearch_Call { return &MockProxy_HybridSearch_Call{Call: _e.mock.On("HybridSearch", _a0, _a1)} } @@ -3295,8 +3295,8 @@ type MockProxy_Import_Call struct { } // Import is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ImportRequest +// - _a0 context.Context +// - _a1 *milvuspb.ImportRequest func (_e *MockProxy_Expecter) Import(_a0 interface{}, _a1 interface{}) *MockProxy_Import_Call { return &MockProxy_Import_Call{Call: _e.mock.On("Import", _a0, _a1)} } @@ -3350,8 +3350,8 @@ type MockProxy_ImportV2_Call struct { } // ImportV2 is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.ImportRequest +// - _a0 context.Context +// - _a1 *internalpb.ImportRequest func (_e *MockProxy_Expecter) ImportV2(_a0 interface{}, _a1 interface{}) *MockProxy_ImportV2_Call { return &MockProxy_ImportV2_Call{Call: _e.mock.On("ImportV2", _a0, _a1)} } @@ -3446,8 +3446,8 @@ type MockProxy_Insert_Call struct { } // Insert is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.InsertRequest +// - _a0 context.Context +// - _a1 *milvuspb.InsertRequest func (_e *MockProxy_Expecter) Insert(_a0 interface{}, _a1 interface{}) *MockProxy_Insert_Call { return &MockProxy_Insert_Call{Call: _e.mock.On("Insert", _a0, _a1)} } @@ -3501,8 +3501,8 @@ type MockProxy_InvalidateCollectionMetaCache_Call struct { } // InvalidateCollectionMetaCache is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *proxypb.InvalidateCollMetaCacheRequest +// - _a0 context.Context +// - _a1 *proxypb.InvalidateCollMetaCacheRequest func (_e *MockProxy_Expecter) InvalidateCollectionMetaCache(_a0 interface{}, _a1 interface{}) *MockProxy_InvalidateCollectionMetaCache_Call { return &MockProxy_InvalidateCollectionMetaCache_Call{Call: _e.mock.On("InvalidateCollectionMetaCache", _a0, _a1)} } @@ -3556,8 +3556,8 @@ type MockProxy_InvalidateCredentialCache_Call struct { } // InvalidateCredentialCache is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *proxypb.InvalidateCredCacheRequest +// - _a0 context.Context +// - _a1 *proxypb.InvalidateCredCacheRequest func (_e *MockProxy_Expecter) InvalidateCredentialCache(_a0 interface{}, _a1 interface{}) *MockProxy_InvalidateCredentialCache_Call { return &MockProxy_InvalidateCredentialCache_Call{Call: _e.mock.On("InvalidateCredentialCache", _a0, _a1)} } @@ -3611,8 +3611,8 @@ type MockProxy_ListAliases_Call struct { } // ListAliases is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ListAliasesRequest +// - _a0 context.Context +// - _a1 *milvuspb.ListAliasesRequest func (_e *MockProxy_Expecter) ListAliases(_a0 interface{}, _a1 interface{}) *MockProxy_ListAliases_Call { return &MockProxy_ListAliases_Call{Call: _e.mock.On("ListAliases", _a0, _a1)} } @@ -3666,8 +3666,8 @@ type MockProxy_ListClientInfos_Call struct { } // ListClientInfos is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *proxypb.ListClientInfosRequest +// - _a0 context.Context +// - _a1 *proxypb.ListClientInfosRequest func (_e *MockProxy_Expecter) ListClientInfos(_a0 interface{}, _a1 interface{}) *MockProxy_ListClientInfos_Call { return &MockProxy_ListClientInfos_Call{Call: _e.mock.On("ListClientInfos", _a0, _a1)} } @@ -3721,8 +3721,8 @@ type MockProxy_ListCredUsers_Call struct { } // ListCredUsers is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ListCredUsersRequest +// - _a0 context.Context +// - _a1 *milvuspb.ListCredUsersRequest func (_e *MockProxy_Expecter) ListCredUsers(_a0 interface{}, _a1 interface{}) *MockProxy_ListCredUsers_Call { return &MockProxy_ListCredUsers_Call{Call: _e.mock.On("ListCredUsers", _a0, _a1)} } @@ -3776,8 +3776,8 @@ type MockProxy_ListDatabases_Call struct { } // ListDatabases is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ListDatabasesRequest +// - _a0 context.Context +// - _a1 *milvuspb.ListDatabasesRequest func (_e *MockProxy_Expecter) ListDatabases(_a0 interface{}, _a1 interface{}) *MockProxy_ListDatabases_Call { return &MockProxy_ListDatabases_Call{Call: _e.mock.On("ListDatabases", _a0, _a1)} } @@ -3831,8 +3831,8 @@ type MockProxy_ListImportTasks_Call struct { } // ListImportTasks is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ListImportTasksRequest +// - _a0 context.Context +// - _a1 *milvuspb.ListImportTasksRequest func (_e *MockProxy_Expecter) ListImportTasks(_a0 interface{}, _a1 interface{}) *MockProxy_ListImportTasks_Call { return &MockProxy_ListImportTasks_Call{Call: _e.mock.On("ListImportTasks", _a0, _a1)} } @@ -3886,8 +3886,8 @@ type MockProxy_ListImports_Call struct { } // ListImports is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *internalpb.ListImportsRequest +// - _a0 context.Context +// - _a1 *internalpb.ListImportsRequest func (_e *MockProxy_Expecter) ListImports(_a0 interface{}, _a1 interface{}) *MockProxy_ListImports_Call { return &MockProxy_ListImports_Call{Call: _e.mock.On("ListImports", _a0, _a1)} } @@ -3941,8 +3941,8 @@ type MockProxy_ListIndexedSegment_Call struct { } // ListIndexedSegment is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *federpb.ListIndexedSegmentRequest +// - _a0 context.Context +// - _a1 *federpb.ListIndexedSegmentRequest func (_e *MockProxy_Expecter) ListIndexedSegment(_a0 interface{}, _a1 interface{}) *MockProxy_ListIndexedSegment_Call { return &MockProxy_ListIndexedSegment_Call{Call: _e.mock.On("ListIndexedSegment", _a0, _a1)} } @@ -3996,8 +3996,8 @@ type MockProxy_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 *MockProxy_Expecter) ListResourceGroups(_a0 interface{}, _a1 interface{}) *MockProxy_ListResourceGroups_Call { return &MockProxy_ListResourceGroups_Call{Call: _e.mock.On("ListResourceGroups", _a0, _a1)} } @@ -4051,8 +4051,8 @@ type MockProxy_LoadBalance_Call struct { } // LoadBalance is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.LoadBalanceRequest +// - _a0 context.Context +// - _a1 *milvuspb.LoadBalanceRequest func (_e *MockProxy_Expecter) LoadBalance(_a0 interface{}, _a1 interface{}) *MockProxy_LoadBalance_Call { return &MockProxy_LoadBalance_Call{Call: _e.mock.On("LoadBalance", _a0, _a1)} } @@ -4106,8 +4106,8 @@ type MockProxy_LoadCollection_Call struct { } // LoadCollection is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.LoadCollectionRequest +// - _a0 context.Context +// - _a1 *milvuspb.LoadCollectionRequest func (_e *MockProxy_Expecter) LoadCollection(_a0 interface{}, _a1 interface{}) *MockProxy_LoadCollection_Call { return &MockProxy_LoadCollection_Call{Call: _e.mock.On("LoadCollection", _a0, _a1)} } @@ -4161,8 +4161,8 @@ type MockProxy_LoadPartitions_Call struct { } // LoadPartitions is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.LoadPartitionsRequest +// - _a0 context.Context +// - _a1 *milvuspb.LoadPartitionsRequest func (_e *MockProxy_Expecter) LoadPartitions(_a0 interface{}, _a1 interface{}) *MockProxy_LoadPartitions_Call { return &MockProxy_LoadPartitions_Call{Call: _e.mock.On("LoadPartitions", _a0, _a1)} } @@ -4216,8 +4216,8 @@ type MockProxy_ManualCompaction_Call struct { } // ManualCompaction is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ManualCompactionRequest +// - _a0 context.Context +// - _a1 *milvuspb.ManualCompactionRequest func (_e *MockProxy_Expecter) ManualCompaction(_a0 interface{}, _a1 interface{}) *MockProxy_ManualCompaction_Call { return &MockProxy_ManualCompaction_Call{Call: _e.mock.On("ManualCompaction", _a0, _a1)} } @@ -4271,8 +4271,8 @@ type MockProxy_OperatePrivilege_Call struct { } // OperatePrivilege is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.OperatePrivilegeRequest +// - _a0 context.Context +// - _a1 *milvuspb.OperatePrivilegeRequest func (_e *MockProxy_Expecter) OperatePrivilege(_a0 interface{}, _a1 interface{}) *MockProxy_OperatePrivilege_Call { return &MockProxy_OperatePrivilege_Call{Call: _e.mock.On("OperatePrivilege", _a0, _a1)} } @@ -4326,8 +4326,8 @@ type MockProxy_OperateUserRole_Call struct { } // OperateUserRole is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.OperateUserRoleRequest +// - _a0 context.Context +// - _a1 *milvuspb.OperateUserRoleRequest func (_e *MockProxy_Expecter) OperateUserRole(_a0 interface{}, _a1 interface{}) *MockProxy_OperateUserRole_Call { return &MockProxy_OperateUserRole_Call{Call: _e.mock.On("OperateUserRole", _a0, _a1)} } @@ -4381,8 +4381,8 @@ type MockProxy_Query_Call struct { } // Query is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.QueryRequest +// - _a0 context.Context +// - _a1 *milvuspb.QueryRequest func (_e *MockProxy_Expecter) Query(_a0 interface{}, _a1 interface{}) *MockProxy_Query_Call { return &MockProxy_Query_Call{Call: _e.mock.On("Query", _a0, _a1)} } @@ -4436,8 +4436,8 @@ type MockProxy_RefreshPolicyInfoCache_Call struct { } // RefreshPolicyInfoCache is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *proxypb.RefreshPolicyInfoCacheRequest +// - _a0 context.Context +// - _a1 *proxypb.RefreshPolicyInfoCacheRequest func (_e *MockProxy_Expecter) RefreshPolicyInfoCache(_a0 interface{}, _a1 interface{}) *MockProxy_RefreshPolicyInfoCache_Call { return &MockProxy_RefreshPolicyInfoCache_Call{Call: _e.mock.On("RefreshPolicyInfoCache", _a0, _a1)} } @@ -4532,8 +4532,8 @@ type MockProxy_RegisterLink_Call struct { } // RegisterLink is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.RegisterLinkRequest +// - _a0 context.Context +// - _a1 *milvuspb.RegisterLinkRequest func (_e *MockProxy_Expecter) RegisterLink(_a0 interface{}, _a1 interface{}) *MockProxy_RegisterLink_Call { return &MockProxy_RegisterLink_Call{Call: _e.mock.On("RegisterLink", _a0, _a1)} } @@ -4587,8 +4587,8 @@ type MockProxy_ReleaseCollection_Call struct { } // ReleaseCollection is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ReleaseCollectionRequest +// - _a0 context.Context +// - _a1 *milvuspb.ReleaseCollectionRequest func (_e *MockProxy_Expecter) ReleaseCollection(_a0 interface{}, _a1 interface{}) *MockProxy_ReleaseCollection_Call { return &MockProxy_ReleaseCollection_Call{Call: _e.mock.On("ReleaseCollection", _a0, _a1)} } @@ -4642,8 +4642,8 @@ type MockProxy_ReleasePartitions_Call struct { } // ReleasePartitions is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ReleasePartitionsRequest +// - _a0 context.Context +// - _a1 *milvuspb.ReleasePartitionsRequest func (_e *MockProxy_Expecter) ReleasePartitions(_a0 interface{}, _a1 interface{}) *MockProxy_ReleasePartitions_Call { return &MockProxy_ReleasePartitions_Call{Call: _e.mock.On("ReleasePartitions", _a0, _a1)} } @@ -4697,8 +4697,8 @@ type MockProxy_RenameCollection_Call struct { } // RenameCollection is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.RenameCollectionRequest +// - _a0 context.Context +// - _a1 *milvuspb.RenameCollectionRequest func (_e *MockProxy_Expecter) RenameCollection(_a0 interface{}, _a1 interface{}) *MockProxy_RenameCollection_Call { return &MockProxy_RenameCollection_Call{Call: _e.mock.On("RenameCollection", _a0, _a1)} } @@ -4752,8 +4752,8 @@ type MockProxy_ReplicateMessage_Call struct { } // ReplicateMessage is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ReplicateMessageRequest +// - _a0 context.Context +// - _a1 *milvuspb.ReplicateMessageRequest func (_e *MockProxy_Expecter) ReplicateMessage(_a0 interface{}, _a1 interface{}) *MockProxy_ReplicateMessage_Call { return &MockProxy_ReplicateMessage_Call{Call: _e.mock.On("ReplicateMessage", _a0, _a1)} } @@ -4807,8 +4807,8 @@ type MockProxy_Search_Call struct { } // Search is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.SearchRequest +// - _a0 context.Context +// - _a1 *milvuspb.SearchRequest func (_e *MockProxy_Expecter) Search(_a0 interface{}, _a1 interface{}) *MockProxy_Search_Call { return &MockProxy_Search_Call{Call: _e.mock.On("Search", _a0, _a1)} } @@ -4862,8 +4862,8 @@ type MockProxy_SelectGrant_Call struct { } // SelectGrant is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.SelectGrantRequest +// - _a0 context.Context +// - _a1 *milvuspb.SelectGrantRequest func (_e *MockProxy_Expecter) SelectGrant(_a0 interface{}, _a1 interface{}) *MockProxy_SelectGrant_Call { return &MockProxy_SelectGrant_Call{Call: _e.mock.On("SelectGrant", _a0, _a1)} } @@ -4917,8 +4917,8 @@ type MockProxy_SelectRole_Call struct { } // SelectRole is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.SelectRoleRequest +// - _a0 context.Context +// - _a1 *milvuspb.SelectRoleRequest func (_e *MockProxy_Expecter) SelectRole(_a0 interface{}, _a1 interface{}) *MockProxy_SelectRole_Call { return &MockProxy_SelectRole_Call{Call: _e.mock.On("SelectRole", _a0, _a1)} } @@ -4972,8 +4972,8 @@ type MockProxy_SelectUser_Call struct { } // SelectUser is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.SelectUserRequest +// - _a0 context.Context +// - _a1 *milvuspb.SelectUserRequest func (_e *MockProxy_Expecter) SelectUser(_a0 interface{}, _a1 interface{}) *MockProxy_SelectUser_Call { return &MockProxy_SelectUser_Call{Call: _e.mock.On("SelectUser", _a0, _a1)} } @@ -5006,7 +5006,7 @@ type MockProxy_SetAddress_Call struct { } // SetAddress is a helper method to define mock.On call -// - address string +// - address string func (_e *MockProxy_Expecter) SetAddress(address interface{}) *MockProxy_SetAddress_Call { return &MockProxy_SetAddress_Call{Call: _e.mock.On("SetAddress", address)} } @@ -5039,7 +5039,7 @@ type MockProxy_SetDataCoordClient_Call struct { } // SetDataCoordClient is a helper method to define mock.On call -// - dataCoord types.DataCoordClient +// - dataCoord types.DataCoordClient func (_e *MockProxy_Expecter) SetDataCoordClient(dataCoord interface{}) *MockProxy_SetDataCoordClient_Call { return &MockProxy_SetDataCoordClient_Call{Call: _e.mock.On("SetDataCoordClient", dataCoord)} } @@ -5072,7 +5072,7 @@ type MockProxy_SetEtcdClient_Call struct { } // SetEtcdClient is a helper method to define mock.On call -// - etcdClient *clientv3.Client +// - etcdClient *clientv3.Client func (_e *MockProxy_Expecter) SetEtcdClient(etcdClient interface{}) *MockProxy_SetEtcdClient_Call { return &MockProxy_SetEtcdClient_Call{Call: _e.mock.On("SetEtcdClient", etcdClient)} } @@ -5105,7 +5105,7 @@ type MockProxy_SetQueryCoordClient_Call struct { } // SetQueryCoordClient is a helper method to define mock.On call -// - queryCoord types.QueryCoordClient +// - queryCoord types.QueryCoordClient func (_e *MockProxy_Expecter) SetQueryCoordClient(queryCoord interface{}) *MockProxy_SetQueryCoordClient_Call { return &MockProxy_SetQueryCoordClient_Call{Call: _e.mock.On("SetQueryCoordClient", queryCoord)} } @@ -5138,7 +5138,7 @@ type MockProxy_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 *MockProxy_Expecter) SetQueryNodeCreator(_a0 interface{}) *MockProxy_SetQueryNodeCreator_Call { return &MockProxy_SetQueryNodeCreator_Call{Call: _e.mock.On("SetQueryNodeCreator", _a0)} } @@ -5192,8 +5192,8 @@ type MockProxy_SetRates_Call struct { } // SetRates is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *proxypb.SetRatesRequest +// - _a0 context.Context +// - _a1 *proxypb.SetRatesRequest func (_e *MockProxy_Expecter) SetRates(_a0 interface{}, _a1 interface{}) *MockProxy_SetRates_Call { return &MockProxy_SetRates_Call{Call: _e.mock.On("SetRates", _a0, _a1)} } @@ -5226,7 +5226,7 @@ type MockProxy_SetRootCoordClient_Call struct { } // SetRootCoordClient is a helper method to define mock.On call -// - rootCoord types.RootCoordClient +// - rootCoord types.RootCoordClient func (_e *MockProxy_Expecter) SetRootCoordClient(rootCoord interface{}) *MockProxy_SetRootCoordClient_Call { return &MockProxy_SetRootCoordClient_Call{Call: _e.mock.On("SetRootCoordClient", rootCoord)} } @@ -5280,8 +5280,8 @@ type MockProxy_ShowCollections_Call struct { } // ShowCollections is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ShowCollectionsRequest +// - _a0 context.Context +// - _a1 *milvuspb.ShowCollectionsRequest func (_e *MockProxy_Expecter) ShowCollections(_a0 interface{}, _a1 interface{}) *MockProxy_ShowCollections_Call { return &MockProxy_ShowCollections_Call{Call: _e.mock.On("ShowCollections", _a0, _a1)} } @@ -5335,8 +5335,8 @@ type MockProxy_ShowPartitions_Call struct { } // ShowPartitions is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.ShowPartitionsRequest +// - _a0 context.Context +// - _a1 *milvuspb.ShowPartitionsRequest func (_e *MockProxy_Expecter) ShowPartitions(_a0 interface{}, _a1 interface{}) *MockProxy_ShowPartitions_Call { return &MockProxy_ShowPartitions_Call{Call: _e.mock.On("ShowPartitions", _a0, _a1)} } @@ -5472,8 +5472,8 @@ type MockProxy_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 *MockProxy_Expecter) TransferNode(_a0 interface{}, _a1 interface{}) *MockProxy_TransferNode_Call { return &MockProxy_TransferNode_Call{Call: _e.mock.On("TransferNode", _a0, _a1)} } @@ -5527,8 +5527,8 @@ type MockProxy_TransferReplica_Call struct { } // TransferReplica is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.TransferReplicaRequest +// - _a0 context.Context +// - _a1 *milvuspb.TransferReplicaRequest func (_e *MockProxy_Expecter) TransferReplica(_a0 interface{}, _a1 interface{}) *MockProxy_TransferReplica_Call { return &MockProxy_TransferReplica_Call{Call: _e.mock.On("TransferReplica", _a0, _a1)} } @@ -5582,8 +5582,8 @@ type MockProxy_UpdateCredential_Call struct { } // UpdateCredential is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.UpdateCredentialRequest +// - _a0 context.Context +// - _a1 *milvuspb.UpdateCredentialRequest func (_e *MockProxy_Expecter) UpdateCredential(_a0 interface{}, _a1 interface{}) *MockProxy_UpdateCredential_Call { return &MockProxy_UpdateCredential_Call{Call: _e.mock.On("UpdateCredential", _a0, _a1)} } @@ -5637,8 +5637,8 @@ type MockProxy_UpdateCredentialCache_Call struct { } // UpdateCredentialCache is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *proxypb.UpdateCredCacheRequest +// - _a0 context.Context +// - _a1 *proxypb.UpdateCredCacheRequest func (_e *MockProxy_Expecter) UpdateCredentialCache(_a0 interface{}, _a1 interface{}) *MockProxy_UpdateCredentialCache_Call { return &MockProxy_UpdateCredentialCache_Call{Call: _e.mock.On("UpdateCredentialCache", _a0, _a1)} } @@ -5726,7 +5726,7 @@ type MockProxy_UpdateStateCode_Call struct { } // UpdateStateCode is a helper method to define mock.On call -// - stateCode commonpb.StateCode +// - stateCode commonpb.StateCode func (_e *MockProxy_Expecter) UpdateStateCode(stateCode interface{}) *MockProxy_UpdateStateCode_Call { return &MockProxy_UpdateStateCode_Call{Call: _e.mock.On("UpdateStateCode", stateCode)} } @@ -5780,8 +5780,8 @@ type MockProxy_Upsert_Call struct { } // Upsert is a helper method to define mock.On call -// - _a0 context.Context -// - _a1 *milvuspb.UpsertRequest +// - _a0 context.Context +// - _a1 *milvuspb.UpsertRequest func (_e *MockProxy_Expecter) Upsert(_a0 interface{}, _a1 interface{}) *MockProxy_Upsert_Call { return &MockProxy_Upsert_Call{Call: _e.mock.On("Upsert", _a0, _a1)} } diff --git a/internal/mocks/mock_proxy_client.go b/internal/mocks/mock_proxy_client.go index a1d0dcc48dad6..ae700007a5824 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...)...)} @@ -251,9 +251,9 @@ type MockProxyClient_GetImportProgress_Call struct { } // GetImportProgress is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.GetImportProgressRequest -// - opts ...grpc.CallOption +// - 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...)...)} @@ -321,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...)...)} @@ -391,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...)...)} @@ -461,9 +461,9 @@ type MockProxyClient_ImportV2_Call struct { } // ImportV2 is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.ImportRequest -// - opts ...grpc.CallOption +// - 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...)...)} @@ -531,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...)...)} @@ -601,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...)...)} @@ -671,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...)...)} @@ -741,9 +741,9 @@ type MockProxyClient_ListImports_Call struct { } // ListImports is a helper method to define mock.On call -// - ctx context.Context -// - in *internalpb.ListImportsRequest -// - opts ...grpc.CallOption +// - 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...)...)} @@ -811,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...)...)} @@ -881,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...)...)} @@ -951,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/tests/integration/import/util_test.go b/tests/integration/import/util_test.go index e964a083d503f..d05dc19897254 100644 --- a/tests/integration/import/util_test.go +++ b/tests/integration/import/util_test.go @@ -652,7 +652,7 @@ func GenerateJSONFile(t *testing.T, filePath string, schema *schemapb.Collection jsonBytes, err := json.Marshal(rows) assert.NoError(t, err) - err = os.WriteFile(filePath, jsonBytes, 0644) // nolint + err = os.WriteFile(filePath, jsonBytes, 0o644) // nolint assert.NoError(t, err) }