admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: [2.5] mockery too unavailable after upgrade golang version (#41522)

Browse Source 
- issue: ##41291
- pr: #41481

Signed-off-by: SimFG <bang.fu@zilliz.com>
This commit is contained in:
SimFG 2025-04-25 14:40:40 +08:00 committed by GitHub
parent 18eb627533
commit 6e18ededab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
116 changed files with 619 additions and 617 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Makefile
View File

@ -54,7 +54,7 @@ GOLANGCI_LINT_VERSION := 1.64.7
GOLANGCI_LINT_OUTPUT := $(shell $(INSTALL_PATH)/golangci-lint --version 2>/dev/null)
INSTALL_GOLANGCI_LINT := $(findstring $(GOLANGCI_LINT_VERSION), $(GOLANGCI_LINT_OUTPUT))
# mockery
MOCKERY_VERSION := 2.46.0
MOCKERY_VERSION := 2.53.3
MOCKERY_OUTPUT := $(shell $(INSTALL_PATH)/mockery --version 2>/dev/null)
INSTALL_MOCKERY := $(findstring $(MOCKERY_VERSION),$(MOCKERY_OUTPUT))
# gci

2
internal/datacoord/allocator/mock_allocator.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package allocator

2
internal/datacoord/broker/mock_broker.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package broker

8
internal/datacoord/mock_channel_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package datacoord
@ -68,7 +68,7 @@ func (_c *MockChannelManager_AddNode_Call) RunAndReturn(run func(int64) error) *
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockChannelManager) Close() {
_m.Called()
}
@ -96,7 +96,7 @@ func (_c *MockChannelManager_Close_Call) Return() *MockChannelManager_Close_Call
}
func (_c *MockChannelManager_Close_Call) RunAndReturn(run func()) *MockChannelManager_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -309,7 +309,7 @@ func (_c *MockChannelManager_GetChannelNamesByCollectionID_Call) RunAndReturn(ru
return _c
}
// GetChannelWatchInfos provides a mock function with given fields:
// GetChannelWatchInfos provides a mock function with no fields
func (_m *MockChannelManager) GetChannelWatchInfos() map[int64]map[string]*datapb.ChannelWatchInfo {
ret := _m.Called()

18
internal/datacoord/mock_channel_store.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package datacoord
@ -46,11 +46,11 @@ func (_c *MockRWChannelStore_AddNode_Call) Return() *MockRWChannelStore_AddNode_
}
func (_c *MockRWChannelStore_AddNode_Call) RunAndReturn(run func(int64)) *MockRWChannelStore_AddNode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// GetBufferChannelInfo provides a mock function with given fields:
// GetBufferChannelInfo provides a mock function with no fields
func (_m *MockRWChannelStore) GetBufferChannelInfo() *NodeChannelInfo {
ret := _m.Called()
@ -256,7 +256,7 @@ func (_c *MockRWChannelStore_GetNodeChannelsByCollectionID_Call) RunAndReturn(ru
return _c
}
// GetNodes provides a mock function with given fields:
// GetNodes provides a mock function with no fields
func (_m *MockRWChannelStore) GetNodes() []int64 {
ret := _m.Called()
@ -303,7 +303,7 @@ func (_c *MockRWChannelStore_GetNodes_Call) RunAndReturn(run func() []int64) *Mo
return _c
}
// GetNodesChannels provides a mock function with given fields:
// GetNodesChannels provides a mock function with no fields
func (_m *MockRWChannelStore) GetNodesChannels() []*NodeChannelInfo {
ret := _m.Called()
@ -396,7 +396,7 @@ func (_c *MockRWChannelStore_HasChannel_Call) RunAndReturn(run func(string) bool
return _c
}
// Reload provides a mock function with given fields:
// Reload provides a mock function with no fields
func (_m *MockRWChannelStore) Reload() error {
ret := _m.Called()
@ -470,7 +470,7 @@ func (_c *MockRWChannelStore_RemoveNode_Call) Return() *MockRWChannelStore_Remov
}
func (_c *MockRWChannelStore_RemoveNode_Call) RunAndReturn(run func(int64)) *MockRWChannelStore_RemoveNode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -516,7 +516,7 @@ func (_c *MockRWChannelStore_SetLegacyChannelByNode_Call) Return() *MockRWChanne
}
func (_c *MockRWChannelStore_SetLegacyChannelByNode_Call) RunAndReturn(run func(...int64)) *MockRWChannelStore_SetLegacyChannelByNode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -598,7 +598,7 @@ func (_c *MockRWChannelStore_UpdateState_Call) Return() *MockRWChannelStore_Upda
}
func (_c *MockRWChannelStore_UpdateState_Call) RunAndReturn(run func(bool, int64, RWChannel, int64)) *MockRWChannelStore_UpdateState_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

10
internal/datacoord/mock_cluster.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package datacoord
@ -24,7 +24,7 @@ func (_m *MockCluster) EXPECT() *MockCluster_Expecter {
return &MockCluster_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockCluster) Close() {
_m.Called()
}
@ -52,7 +52,7 @@ func (_c *MockCluster_Close_Call) Return() *MockCluster_Close_Call {
}
func (_c *MockCluster_Close_Call) RunAndReturn(run func()) *MockCluster_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -201,7 +201,7 @@ func (_c *MockCluster_FlushChannels_Call) RunAndReturn(run func(context.Context,
return _c
}
// GetSessions provides a mock function with given fields:
// GetSessions provides a mock function with no fields
func (_m *MockCluster) GetSessions() []*session.Session {
ret := _m.Called()
@ -460,7 +460,7 @@ func (_c *MockCluster_QueryPreImport_Call) RunAndReturn(run func(int64, *datapb.
return _c
}
// QuerySlots provides a mock function with given fields:
// QuerySlots provides a mock function with no fields
func (_m *MockCluster) QuerySlots() map[int64]int64 {
ret := _m.Called()

14
internal/datacoord/mock_compaction_meta.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package datacoord
@ -299,7 +299,7 @@ func (_c *MockCompactionMeta_DropCompactionTask_Call) RunAndReturn(run func(cont
return _c
}
// GetAnalyzeMeta provides a mock function with given fields:
// GetAnalyzeMeta provides a mock function with no fields
func (_m *MockCompactionMeta) GetAnalyzeMeta() *analyzeMeta {
ret := _m.Called()
@ -346,7 +346,7 @@ func (_c *MockCompactionMeta_GetAnalyzeMeta_Call) RunAndReturn(run func() *analy
return _c
}
// GetCompactionTaskMeta provides a mock function with given fields:
// GetCompactionTaskMeta provides a mock function with no fields
func (_m *MockCompactionMeta) GetCompactionTaskMeta() *compactionTaskMeta {
ret := _m.Called()
@ -539,7 +539,7 @@ func (_c *MockCompactionMeta_GetHealthySegment_Call) RunAndReturn(run func(conte
return _c
}
// GetIndexMeta provides a mock function with given fields:
// GetIndexMeta provides a mock function with no fields
func (_m *MockCompactionMeta) GetIndexMeta() *indexMeta {
ret := _m.Called()
@ -586,7 +586,7 @@ func (_c *MockCompactionMeta_GetIndexMeta_Call) RunAndReturn(run func() *indexMe
return _c
}
// GetPartitionStatsMeta provides a mock function with given fields:
// GetPartitionStatsMeta provides a mock function with no fields
func (_m *MockCompactionMeta) GetPartitionStatsMeta() *partitionStatsMeta {
ret := _m.Called()
@ -870,7 +870,7 @@ func (_c *MockCompactionMeta_SetSegmentStating_Call) Return() *MockCompactionMet
}
func (_c *MockCompactionMeta_SetSegmentStating_Call) RunAndReturn(run func(int64, bool)) *MockCompactionMeta_SetSegmentStating_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -905,7 +905,7 @@ func (_c *MockCompactionMeta_SetSegmentsCompacting_Call) Return() *MockCompactio
}
func (_c *MockCompactionMeta_SetSegmentsCompacting_Call) RunAndReturn(run func(context.Context, []int64, bool)) *MockCompactionMeta_SetSegmentsCompacting_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

16
internal/datacoord/mock_compaction_plan_context.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package datacoord
@ -210,7 +210,7 @@ func (_c *MockCompactionPlanContext_getCompactionTasksNumBySignalID_Call) RunAnd
return _c
}
// isFull provides a mock function with given fields:
// isFull provides a mock function with no fields
func (_m *MockCompactionPlanContext) isFull() bool {
ret := _m.Called()
@ -284,7 +284,7 @@ func (_c *MockCompactionPlanContext_removeTasksByChannel_Call) Return() *MockCom
}
func (_c *MockCompactionPlanContext_removeTasksByChannel_Call) RunAndReturn(run func(string)) *MockCompactionPlanContext_removeTasksByChannel_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -317,11 +317,11 @@ func (_c *MockCompactionPlanContext_setTaskScheduler_Call) Return() *MockCompact
}
func (_c *MockCompactionPlanContext_setTaskScheduler_Call) RunAndReturn(run func(*taskScheduler)) *MockCompactionPlanContext_setTaskScheduler_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// start provides a mock function with given fields:
// start provides a mock function with no fields
func (_m *MockCompactionPlanContext) start() {
_m.Called()
}
@ -349,11 +349,11 @@ func (_c *MockCompactionPlanContext_start_Call) Return() *MockCompactionPlanCont
}
func (_c *MockCompactionPlanContext_start_Call) RunAndReturn(run func()) *MockCompactionPlanContext_start_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// stop provides a mock function with given fields:
// stop provides a mock function with no fields
func (_m *MockCompactionPlanContext) stop() {
_m.Called()
}
@ -381,7 +381,7 @@ func (_c *MockCompactionPlanContext_stop_Call) Return() *MockCompactionPlanConte
}
func (_c *MockCompactionPlanContext_stop_Call) RunAndReturn(run func()) *MockCompactionPlanContext_stop_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

2
internal/datacoord/mock_handler.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package datacoord

20
internal/datacoord/mock_index_engine_version_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package datacoord
@ -49,11 +49,11 @@ func (_c *MockVersionManager_AddNode_Call) Return() *MockVersionManager_AddNode_
}
func (_c *MockVersionManager_AddNode_Call) RunAndReturn(run func(*sessionutil.Session)) *MockVersionManager_AddNode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// GetCurrentIndexEngineVersion provides a mock function with given fields:
// GetCurrentIndexEngineVersion provides a mock function with no fields
func (_m *MockVersionManager) GetCurrentIndexEngineVersion() int32 {
ret := _m.Called()
@ -98,7 +98,7 @@ func (_c *MockVersionManager_GetCurrentIndexEngineVersion_Call) RunAndReturn(run
return _c
}
// GetCurrentScalarIndexEngineVersion provides a mock function with given fields:
// GetCurrentScalarIndexEngineVersion provides a mock function with no fields
func (_m *MockVersionManager) GetCurrentScalarIndexEngineVersion() int32 {
ret := _m.Called()
@ -143,7 +143,7 @@ func (_c *MockVersionManager_GetCurrentScalarIndexEngineVersion_Call) RunAndRetu
return _c
}
// GetIndexNonEncoding provides a mock function with given fields:
// GetIndexNonEncoding provides a mock function with no fields
func (_m *MockVersionManager) GetIndexNonEncoding() bool {
ret := _m.Called()
@ -188,7 +188,7 @@ func (_c *MockVersionManager_GetIndexNonEncoding_Call) RunAndReturn(run func() b
return _c
}
// GetMinimalIndexEngineVersion provides a mock function with given fields:
// GetMinimalIndexEngineVersion provides a mock function with no fields
func (_m *MockVersionManager) GetMinimalIndexEngineVersion() int32 {
ret := _m.Called()
@ -233,7 +233,7 @@ func (_c *MockVersionManager_GetMinimalIndexEngineVersion_Call) RunAndReturn(run
return _c
}
// GetMinimalScalarIndexEngineVersion provides a mock function with given fields:
// GetMinimalScalarIndexEngineVersion provides a mock function with no fields
func (_m *MockVersionManager) GetMinimalScalarIndexEngineVersion() int32 {
ret := _m.Called()
@ -307,7 +307,7 @@ func (_c *MockVersionManager_RemoveNode_Call) Return() *MockVersionManager_Remov
}
func (_c *MockVersionManager_RemoveNode_Call) RunAndReturn(run func(*sessionutil.Session)) *MockVersionManager_RemoveNode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -340,7 +340,7 @@ func (_c *MockVersionManager_Startup_Call) Return() *MockVersionManager_Startup_
}
func (_c *MockVersionManager_Startup_Call) RunAndReturn(run func(map[string]*sessionutil.Session)) *MockVersionManager_Startup_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -373,7 +373,7 @@ func (_c *MockVersionManager_Update_Call) Return() *MockVersionManager_Update_Ca
}
func (_c *MockVersionManager_Update_Call) RunAndReturn(run func(*sessionutil.Session)) *MockVersionManager_Update_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

10
internal/datacoord/mock_segment_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package datacoord
@ -176,7 +176,7 @@ func (_c *MockManager_CleanZeroSealedSegmentsOfChannel_Call) Return() *MockManag
}
func (_c *MockManager_CleanZeroSealedSegmentsOfChannel_Call) RunAndReturn(run func(context.Context, string, uint64)) *MockManager_CleanZeroSealedSegmentsOfChannel_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -211,7 +211,7 @@ func (_c *MockManager_DropSegment_Call) Return() *MockManager_DropSegment_Call {
}
func (_c *MockManager_DropSegment_Call) RunAndReturn(run func(context.Context, string, int64)) *MockManager_DropSegment_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -245,7 +245,7 @@ func (_c *MockManager_DropSegmentsOfChannel_Call) Return() *MockManager_DropSegm
}
func (_c *MockManager_DropSegmentsOfChannel_Call) RunAndReturn(run func(context.Context, string)) *MockManager_DropSegmentsOfChannel_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -280,7 +280,7 @@ func (_c *MockManager_ExpireAllocations_Call) Return() *MockManager_ExpireAlloca
}
func (_c *MockManager_ExpireAllocations_Call) RunAndReturn(run func(context.Context, string, uint64)) *MockManager_ExpireAllocations_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

10
internal/datacoord/mock_stats_job_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package datacoord
@ -116,7 +116,7 @@ func (_c *MockStatsJobManager_GetStatsTask_Call) RunAndReturn(run func(int64, in
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *MockStatsJobManager) Start() {
_m.Called()
}
@ -144,11 +144,11 @@ func (_c *MockStatsJobManager_Start_Call) Return() *MockStatsJobManager_Start_Ca
}
func (_c *MockStatsJobManager_Start_Call) RunAndReturn(run func()) *MockStatsJobManager_Start_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockStatsJobManager) Stop() {
_m.Called()
}
@ -176,7 +176,7 @@ func (_c *MockStatsJobManager_Stop_Call) Return() *MockStatsJobManager_Stop_Call
}
func (_c *MockStatsJobManager_Stop_Call) RunAndReturn(run func()) *MockStatsJobManager_Stop_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

2
internal/datacoord/mock_sub_cluster.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package datacoord

10
internal/datacoord/mock_trigger.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package datacoord
@ -78,7 +78,7 @@ func (_c *MockTrigger_TriggerCompaction_Call) RunAndReturn(run func(context.Cont
return _c
}
// start provides a mock function with given fields:
// start provides a mock function with no fields
func (_m *MockTrigger) start() {
_m.Called()
}
@ -106,11 +106,11 @@ func (_c *MockTrigger_start_Call) Return() *MockTrigger_start_Call {
}
func (_c *MockTrigger_start_Call) RunAndReturn(run func()) *MockTrigger_start_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// stop provides a mock function with given fields:
// stop provides a mock function with no fields
func (_m *MockTrigger) stop() {
_m.Called()
}
@ -138,7 +138,7 @@ func (_c *MockTrigger_stop_Call) Return() *MockTrigger_stop_Call {
}
func (_c *MockTrigger_stop_Call) RunAndReturn(run func()) *MockTrigger_stop_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

12
internal/datacoord/mock_trigger_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package datacoord
@ -108,11 +108,11 @@ func (_c *MockTriggerManager_OnCollectionUpdate_Call) Return() *MockTriggerManag
}
func (_c *MockTriggerManager_OnCollectionUpdate_Call) RunAndReturn(run func(int64)) *MockTriggerManager_OnCollectionUpdate_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *MockTriggerManager) Start() {
_m.Called()
}
@ -140,11 +140,11 @@ func (_c *MockTriggerManager_Start_Call) Return() *MockTriggerManager_Start_Call
}
func (_c *MockTriggerManager_Start_Call) RunAndReturn(run func()) *MockTriggerManager_Start_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockTriggerManager) Stop() {
_m.Called()
}
@ -172,7 +172,7 @@ func (_c *MockTriggerManager_Stop_Call) Return() *MockTriggerManager_Stop_Call {
}
func (_c *MockTriggerManager_Stop_Call) RunAndReturn(run func()) *MockTriggerManager_Stop_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

18
internal/datacoord/session/mock_data_node_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package session
@ -53,7 +53,7 @@ func (_c *MockDataNodeManager_AddSession_Call) Return() *MockDataNodeManager_Add
}
func (_c *MockDataNodeManager_AddSession_Call) RunAndReturn(run func(*NodeInfo)) *MockDataNodeManager_AddSession_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -163,7 +163,7 @@ func (_c *MockDataNodeManager_CheckHealth_Call) RunAndReturn(run func(context.Co
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockDataNodeManager) Close() {
_m.Called()
}
@ -191,7 +191,7 @@ func (_c *MockDataNodeManager_Close_Call) Return() *MockDataNodeManager_Close_Ca
}
func (_c *MockDataNodeManager_Close_Call) RunAndReturn(run func()) *MockDataNodeManager_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -272,7 +272,7 @@ func (_c *MockDataNodeManager_DeleteSession_Call) Return() *MockDataNodeManager_
}
func (_c *MockDataNodeManager_DeleteSession_Call) RunAndReturn(run func(*NodeInfo)) *MockDataNodeManager_DeleteSession_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -401,7 +401,7 @@ func (_c *MockDataNodeManager_Flush_Call) Return() *MockDataNodeManager_Flush_Ca
}
func (_c *MockDataNodeManager_Flush_Call) RunAndReturn(run func(context.Context, int64, *datapb.FlushSegmentsRequest)) *MockDataNodeManager_Flush_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -512,7 +512,7 @@ func (_c *MockDataNodeManager_GetCompactionPlanResult_Call) RunAndReturn(run fun
return _c
}
// GetCompactionPlansResults provides a mock function with given fields:
// GetCompactionPlansResults provides a mock function with no fields
func (_m *MockDataNodeManager) GetCompactionPlansResults() (map[int64]*typeutil.Pair[int64, *datapb.CompactionPlanResult], error) {
ret := _m.Called()
@ -627,7 +627,7 @@ func (_c *MockDataNodeManager_GetSession_Call) RunAndReturn(run func(int64) (*Se
return _c
}
// GetSessionIDs provides a mock function with given fields:
// GetSessionIDs provides a mock function with no fields
func (_m *MockDataNodeManager) GetSessionIDs() []int64 {
ret := _m.Called()
@ -674,7 +674,7 @@ func (_c *MockDataNodeManager_GetSessionIDs_Call) RunAndReturn(run func() []int6
return _c
}
// GetSessions provides a mock function with given fields:
// GetSessions provides a mock function with no fields
func (_m *MockDataNodeManager) GetSessions() []*Session {
ret := _m.Called()

12
internal/datacoord/session/mock_worker_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package session
@ -67,7 +67,7 @@ func (_c *MockWorkerManager_AddNode_Call) RunAndReturn(run func(int64, string) e
return _c
}
// GetAllClients provides a mock function with given fields:
// GetAllClients provides a mock function with no fields
func (_m *MockWorkerManager) GetAllClients() map[int64]types.IndexNodeClient {
ret := _m.Called()
@ -172,7 +172,7 @@ func (_c *MockWorkerManager_GetClientByID_Call) RunAndReturn(run func(int64) (ty
return _c
}
// PickClient provides a mock function with given fields:
// PickClient provides a mock function with no fields
func (_m *MockWorkerManager) PickClient() (int64, types.IndexNodeClient) {
ret := _m.Called()
@ -229,7 +229,7 @@ func (_c *MockWorkerManager_PickClient_Call) RunAndReturn(run func() (int64, typ
return _c
}
// QuerySlots provides a mock function with given fields:
// QuerySlots provides a mock function with no fields
func (_m *MockWorkerManager) QuerySlots() map[int64]*WorkerSlots {
ret := _m.Called()
@ -305,7 +305,7 @@ func (_c *MockWorkerManager_RemoveNode_Call) Return() *MockWorkerManager_RemoveN
}
func (_c *MockWorkerManager_RemoveNode_Call) RunAndReturn(run func(int64)) *MockWorkerManager_RemoveNode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -338,7 +338,7 @@ func (_c *MockWorkerManager_StoppingNode_Call) Return() *MockWorkerManager_Stopp
}
func (_c *MockWorkerManager_StoppingNode_Call) RunAndReturn(run func(int64)) *MockWorkerManager_StoppingNode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

6
internal/kv/mocks/meta_kv.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -23,7 +23,7 @@ func (_m *MetaKv) EXPECT() *MetaKv_Expecter {
return &MetaKv_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MetaKv) Close() {
_m.Called()
}
@ -51,7 +51,7 @@ func (_c *MetaKv_Close_Call) Return() *MetaKv_Close_Call {
}
func (_c *MetaKv_Close_Call) RunAndReturn(run func()) *MetaKv_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

2
internal/kv/mocks/snapshot_kv.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks

6
internal/kv/mocks/txn_kv.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -23,7 +23,7 @@ func (_m *TxnKV) EXPECT() *TxnKV_Expecter {
return &TxnKV_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *TxnKV) Close() {
_m.Called()
}
@ -51,7 +51,7 @@ func (_c *TxnKV_Close_Call) Return() *TxnKV_Close_Call {
}
func (_c *TxnKV_Close_Call) RunAndReturn(run func()) *TxnKV_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

6
internal/kv/mocks/watch_kv.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -25,7 +25,7 @@ func (_m *WatchKV) EXPECT() *WatchKV_Expecter {
return &WatchKV_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *WatchKV) Close() {
_m.Called()
}
@ -53,7 +53,7 @@ func (_c *WatchKV_Close_Call) Return() *WatchKV_Close_Call {
}
func (_c *WatchKV_Close_Call) RunAndReturn(run func()) *WatchKV_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

4
internal/mocks/distributed/mock_streaming/mock_WALAccesser.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_streaming
@ -149,7 +149,7 @@ func (_c *MockWALAccesser_AppendMessagesWithOption_Call) RunAndReturn(run func(c
return _c
}
// Broadcast provides a mock function with given fields:
// Broadcast provides a mock function with no fields
func (_m *MockWALAccesser) Broadcast() streaming.Broadcast {
ret := _m.Called()

12
internal/mocks/google.golang.org/grpc/mock_balancer/mock_SubConn.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_balancer
@ -22,7 +22,7 @@ func (_m *MockSubConn) EXPECT() *MockSubConn_Expecter {
return &MockSubConn_Expecter{mock: &_m.Mock}
}
// Connect provides a mock function with given fields:
// Connect provides a mock function with no fields
func (_m *MockSubConn) Connect() {
_m.Called()
}
@ -50,7 +50,7 @@ func (_c *MockSubConn_Connect_Call) Return() *MockSubConn_Connect_Call {
}
func (_c *MockSubConn_Connect_Call) RunAndReturn(run func()) *MockSubConn_Connect_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -114,7 +114,7 @@ func (_c *MockSubConn_GetOrBuildProducer_Call) RunAndReturn(run func(balancer.Pr
return _c
}
// Shutdown provides a mock function with given fields:
// Shutdown provides a mock function with no fields
func (_m *MockSubConn) Shutdown() {
_m.Called()
}
@ -142,7 +142,7 @@ func (_c *MockSubConn_Shutdown_Call) Return() *MockSubConn_Shutdown_Call {
}
func (_c *MockSubConn_Shutdown_Call) RunAndReturn(run func()) *MockSubConn_Shutdown_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -175,7 +175,7 @@ func (_c *MockSubConn_UpdateAddresses_Call) Return() *MockSubConn_UpdateAddresse
}
func (_c *MockSubConn_UpdateAddresses_Call) RunAndReturn(run func([]resolver.Address)) *MockSubConn_UpdateAddresses_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

6
internal/mocks/google.golang.org/grpc/mock_resolver/mock_ClientConn.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_resolver
@ -51,7 +51,7 @@ func (_c *MockClientConn_NewAddress_Call) Return() *MockClientConn_NewAddress_Ca
}
func (_c *MockClientConn_NewAddress_Call) RunAndReturn(run func([]resolver.Address)) *MockClientConn_NewAddress_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -132,7 +132,7 @@ func (_c *MockClientConn_ReportError_Call) Return() *MockClientConn_ReportError_
}
func (_c *MockClientConn_ReportError_Call) RunAndReturn(run func(error)) *MockClientConn_ReportError_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

10
internal/mocks/google.golang.org/mock_grpc/mock_ClientStream.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_grpc
@ -23,7 +23,7 @@ func (_m *MockClientStream) EXPECT() *MockClientStream_Expecter {
return &MockClientStream_Expecter{mock: &_m.Mock}
}
// CloseSend provides a mock function with given fields:
// CloseSend provides a mock function with no fields
func (_m *MockClientStream) CloseSend() error {
ret := _m.Called()
@ -68,7 +68,7 @@ func (_c *MockClientStream_CloseSend_Call) RunAndReturn(run func() error) *MockC
return _c
}
// Context provides a mock function with given fields:
// Context provides a mock function with no fields
func (_m *MockClientStream) Context() context.Context {
ret := _m.Called()
@ -115,7 +115,7 @@ func (_c *MockClientStream_Context_Call) RunAndReturn(run func() context.Context
return _c
}
// Header provides a mock function with given fields:
// Header provides a mock function with no fields
func (_m *MockClientStream) Header() (metadata.MD, error) {
ret := _m.Called()
@ -264,7 +264,7 @@ func (_c *MockClientStream_SendMsg_Call) RunAndReturn(run func(interface{}) erro
return _c
}
// Trailer provides a mock function with given fields:
// Trailer provides a mock function with no fields
func (_m *MockClientStream) Trailer() metadata.MD {
ret := _m.Called()

22
internal/mocks/mock_datacoord.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -2278,7 +2278,7 @@ func (_c *MockDataCoord_ImportV2_Call) RunAndReturn(run func(context.Context, *i
return _c
}
// Init provides a mock function with given fields:
// Init provides a mock function with no fields
func (_m *MockDataCoord) Init() error {
ret := _m.Called()
@ -2559,7 +2559,7 @@ func (_c *MockDataCoord_MarkSegmentsDropped_Call) RunAndReturn(run func(context.
return _c
}
// Register provides a mock function with given fields:
// Register provides a mock function with no fields
func (_m *MockDataCoord) Register() error {
ret := _m.Called()
@ -2751,7 +2751,7 @@ func (_c *MockDataCoord_SetAddress_Call) Return() *MockDataCoord_SetAddress_Call
}
func (_c *MockDataCoord_SetAddress_Call) RunAndReturn(run func(string)) *MockDataCoord_SetAddress_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -2784,7 +2784,7 @@ func (_c *MockDataCoord_SetDataNodeCreator_Call) Return() *MockDataCoord_SetData
}
func (_c *MockDataCoord_SetDataNodeCreator_Call) RunAndReturn(run func(func(context.Context, string, int64) (types.DataNodeClient, error))) *MockDataCoord_SetDataNodeCreator_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -2817,7 +2817,7 @@ func (_c *MockDataCoord_SetEtcdClient_Call) Return() *MockDataCoord_SetEtcdClien
}
func (_c *MockDataCoord_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockDataCoord_SetEtcdClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -2850,7 +2850,7 @@ func (_c *MockDataCoord_SetIndexNodeCreator_Call) Return() *MockDataCoord_SetInd
}
func (_c *MockDataCoord_SetIndexNodeCreator_Call) RunAndReturn(run func(func(context.Context, string, int64) (types.IndexNodeClient, error))) *MockDataCoord_SetIndexNodeCreator_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -2883,7 +2883,7 @@ func (_c *MockDataCoord_SetRootCoordClient_Call) Return() *MockDataCoord_SetRoot
}
func (_c *MockDataCoord_SetRootCoordClient_Call) RunAndReturn(run func(types.RootCoordClient)) *MockDataCoord_SetRootCoordClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -2975,7 +2975,7 @@ func (_c *MockDataCoord_SetTiKVClient_Call) Return() *MockDataCoord_SetTiKVClien
}
func (_c *MockDataCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client)) *MockDataCoord_SetTiKVClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -3038,7 +3038,7 @@ func (_c *MockDataCoord_ShowConfigurations_Call) RunAndReturn(run func(context.C
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *MockDataCoord) Start() error {
ret := _m.Called()
@ -3083,7 +3083,7 @@ func (_c *MockDataCoord_Start_Call) RunAndReturn(run func() error) *MockDataCoor
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockDataCoord) Stop() error {
ret := _m.Called()

4
internal/mocks/mock_datacoord_client.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -403,7 +403,7 @@ func (_c *MockDataCoordClient_CheckHealth_Call) RunAndReturn(run func(context.Co
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockDataCoordClient) Close() error {
ret := _m.Called()

22
internal/mocks/mock_datanode.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -386,7 +386,7 @@ func (_c *MockDataNode_FlushSegments_Call) RunAndReturn(run func(context.Context
return _c
}
// GetAddress provides a mock function with given fields:
// GetAddress provides a mock function with no fields
func (_m *MockDataNode) GetAddress() string {
ret := _m.Called()
@ -608,7 +608,7 @@ func (_c *MockDataNode_GetMetrics_Call) RunAndReturn(run func(context.Context, *
return _c
}
// GetNodeID provides a mock function with given fields:
// GetNodeID provides a mock function with no fields
func (_m *MockDataNode) GetNodeID() int64 {
ret := _m.Called()
@ -653,7 +653,7 @@ func (_c *MockDataNode_GetNodeID_Call) RunAndReturn(run func() int64) *MockDataN
return _c
}
// GetStateCode provides a mock function with given fields:
// GetStateCode provides a mock function with no fields
func (_m *MockDataNode) GetStateCode() commonpb.StateCode {
ret := _m.Called()
@ -816,7 +816,7 @@ func (_c *MockDataNode_ImportV2_Call) RunAndReturn(run func(context.Context, *da
return _c
}
// Init provides a mock function with given fields:
// Init provides a mock function with no fields
func (_m *MockDataNode) Init() error {
ret := _m.Called()
@ -1156,7 +1156,7 @@ func (_c *MockDataNode_QuerySlot_Call) RunAndReturn(run func(context.Context, *d
return _c
}
// Register provides a mock function with given fields:
// Register provides a mock function with no fields
func (_m *MockDataNode) Register() error {
ret := _m.Called()
@ -1289,7 +1289,7 @@ func (_c *MockDataNode_SetAddress_Call) Return() *MockDataNode_SetAddress_Call {
}
func (_c *MockDataNode_SetAddress_Call) RunAndReturn(run func(string)) *MockDataNode_SetAddress_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1368,7 +1368,7 @@ func (_c *MockDataNode_SetEtcdClient_Call) Return() *MockDataNode_SetEtcdClient_
}
func (_c *MockDataNode_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockDataNode_SetEtcdClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1477,7 +1477,7 @@ func (_c *MockDataNode_ShowConfigurations_Call) RunAndReturn(run func(context.Co
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *MockDataNode) Start() error {
ret := _m.Called()
@ -1522,7 +1522,7 @@ func (_c *MockDataNode_Start_Call) RunAndReturn(run func() error) *MockDataNode_
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockDataNode) Stop() error {
ret := _m.Called()
@ -1655,7 +1655,7 @@ func (_c *MockDataNode_UpdateStateCode_Call) Return() *MockDataNode_UpdateStateC
}
func (_c *MockDataNode_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *MockDataNode_UpdateStateCode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

4
internal/mocks/mock_datanode_client.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -105,7 +105,7 @@ func (_c *MockDataNodeClient_CheckChannelOperationProgress_Call) RunAndReturn(ru
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockDataNodeClient) Close() error {
ret := _m.Called()

18
internal/mocks/mock_indexnode.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -266,7 +266,7 @@ func (_c *MockIndexNode_DropJobsV2_Call) RunAndReturn(run func(context.Context,
return _c
}
// GetAddress provides a mock function with given fields:
// GetAddress provides a mock function with no fields
func (_m *MockIndexNode) GetAddress() string {
ret := _m.Called()
@ -547,7 +547,7 @@ func (_c *MockIndexNode_GetStatisticsChannel_Call) RunAndReturn(run func(context
return _c
}
// Init provides a mock function with given fields:
// Init provides a mock function with no fields
func (_m *MockIndexNode) Init() error {
ret := _m.Called()
@ -710,7 +710,7 @@ func (_c *MockIndexNode_QueryJobsV2_Call) RunAndReturn(run func(context.Context,
return _c
}
// Register provides a mock function with given fields:
// Register provides a mock function with no fields
func (_m *MockIndexNode) Register() error {
ret := _m.Called()
@ -784,7 +784,7 @@ func (_c *MockIndexNode_SetAddress_Call) Return() *MockIndexNode_SetAddress_Call
}
func (_c *MockIndexNode_SetAddress_Call) RunAndReturn(run func(string)) *MockIndexNode_SetAddress_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -817,7 +817,7 @@ func (_c *MockIndexNode_SetEtcdClient_Call) Return() *MockIndexNode_SetEtcdClien
}
func (_c *MockIndexNode_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockIndexNode_SetEtcdClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -880,7 +880,7 @@ func (_c *MockIndexNode_ShowConfigurations_Call) RunAndReturn(run func(context.C
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *MockIndexNode) Start() error {
ret := _m.Called()
@ -925,7 +925,7 @@ func (_c *MockIndexNode_Start_Call) RunAndReturn(run func() error) *MockIndexNod
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockIndexNode) Stop() error {
ret := _m.Called()
@ -999,7 +999,7 @@ func (_c *MockIndexNode_UpdateStateCode_Call) Return() *MockIndexNode_UpdateStat
}
func (_c *MockIndexNode_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *MockIndexNode_UpdateStateCode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

4
internal/mocks/mock_indexnode_client.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -31,7 +31,7 @@ func (_m *MockIndexNodeClient) EXPECT() *MockIndexNodeClient_Expecter {
return &MockIndexNodeClient_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockIndexNodeClient) Close() error {
ret := _m.Called()

2
internal/mocks/mock_metastore/mock_StreamingCoordCataLog.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_metastore

2
internal/mocks/mock_metastore/mock_StreamingNodeCataLog.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_metastore

28
internal/mocks/mock_proxy.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -2276,7 +2276,7 @@ func (_c *MockProxy_FlushAll_Call) RunAndReturn(run func(context.Context, *milvu
return _c
}
// GetAddress provides a mock function with given fields:
// GetAddress provides a mock function with no fields
func (_m *MockProxy) GetAddress() string {
ret := _m.Called()
@ -3442,7 +3442,7 @@ func (_c *MockProxy_GetQuerySegmentInfo_Call) RunAndReturn(run func(context.Cont
return _c
}
// GetRateLimiter provides a mock function with given fields:
// GetRateLimiter provides a mock function with no fields
func (_m *MockProxy) GetRateLimiter() (types.Limiter, error) {
ret := _m.Called()
@ -4030,7 +4030,7 @@ func (_c *MockProxy_ImportV2_Call) RunAndReturn(run func(context.Context, *inter
return _c
}
// Init provides a mock function with given fields:
// Init provides a mock function with no fields
func (_m *MockProxy) Init() error {
ret := _m.Called()
@ -5432,7 +5432,7 @@ func (_c *MockProxy_RefreshPolicyInfoCache_Call) RunAndReturn(run func(context.C
return _c
}
// Register provides a mock function with given fields:
// Register provides a mock function with no fields
func (_m *MockProxy) Register() error {
ret := _m.Called()
@ -6155,7 +6155,7 @@ func (_c *MockProxy_SetAddress_Call) Return() *MockProxy_SetAddress_Call {
}
func (_c *MockProxy_SetAddress_Call) RunAndReturn(run func(string)) *MockProxy_SetAddress_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -6188,7 +6188,7 @@ func (_c *MockProxy_SetDataCoordClient_Call) Return() *MockProxy_SetDataCoordCli
}
func (_c *MockProxy_SetDataCoordClient_Call) RunAndReturn(run func(types.DataCoordClient)) *MockProxy_SetDataCoordClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -6221,7 +6221,7 @@ func (_c *MockProxy_SetEtcdClient_Call) Return() *MockProxy_SetEtcdClient_Call {
}
func (_c *MockProxy_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockProxy_SetEtcdClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -6254,7 +6254,7 @@ func (_c *MockProxy_SetQueryCoordClient_Call) Return() *MockProxy_SetQueryCoordC
}
func (_c *MockProxy_SetQueryCoordClient_Call) RunAndReturn(run func(types.QueryCoordClient)) *MockProxy_SetQueryCoordClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -6287,7 +6287,7 @@ func (_c *MockProxy_SetQueryNodeCreator_Call) Return() *MockProxy_SetQueryNodeCr
}
func (_c *MockProxy_SetQueryNodeCreator_Call) RunAndReturn(run func(func(context.Context, string, int64) (types.QueryNodeClient, error))) *MockProxy_SetQueryNodeCreator_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -6379,7 +6379,7 @@ func (_c *MockProxy_SetRootCoordClient_Call) Return() *MockProxy_SetRootCoordCli
}
func (_c *MockProxy_SetRootCoordClient_Call) RunAndReturn(run func(types.RootCoordClient)) *MockProxy_SetRootCoordClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -6501,7 +6501,7 @@ func (_c *MockProxy_ShowPartitions_Call) RunAndReturn(run func(context.Context,
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *MockProxy) Start() error {
ret := _m.Called()
@ -6546,7 +6546,7 @@ func (_c *MockProxy_Start_Call) RunAndReturn(run func() error) *MockProxy_Start_
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockProxy) Stop() error {
ret := _m.Called()
@ -6915,7 +6915,7 @@ func (_c *MockProxy_UpdateStateCode_Call) Return() *MockProxy_UpdateStateCode_Ca
}
func (_c *MockProxy_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *MockProxy_UpdateStateCode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

4
internal/mocks/mock_proxy_client.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -31,7 +31,7 @@ func (_m *MockProxyClient) EXPECT() *MockProxyClient_Expecter {
return &MockProxyClient_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockProxyClient) Close() error {
ret := _m.Called()

20
internal/mocks/mock_querycoord.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -1037,7 +1037,7 @@ func (_c *MockQueryCoord_GetTimeTickChannel_Call) RunAndReturn(run func(context.
return _c
}
// Init provides a mock function with given fields:
// Init provides a mock function with no fields
func (_m *MockQueryCoord) Init() error {
ret := _m.Called()
@ -1436,7 +1436,7 @@ func (_c *MockQueryCoord_LoadPartitions_Call) RunAndReturn(run func(context.Cont
return _c
}
// Register provides a mock function with given fields:
// Register provides a mock function with no fields
func (_m *MockQueryCoord) Register() error {
ret := _m.Called()
@ -1746,7 +1746,7 @@ func (_c *MockQueryCoord_SetAddress_Call) Return() *MockQueryCoord_SetAddress_Ca
}
func (_c *MockQueryCoord_SetAddress_Call) RunAndReturn(run func(string)) *MockQueryCoord_SetAddress_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1825,7 +1825,7 @@ func (_c *MockQueryCoord_SetEtcdClient_Call) Return() *MockQueryCoord_SetEtcdCli
}
func (_c *MockQueryCoord_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockQueryCoord_SetEtcdClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1858,7 +1858,7 @@ func (_c *MockQueryCoord_SetQueryNodeCreator_Call) Return() *MockQueryCoord_SetQ
}
func (_c *MockQueryCoord_SetQueryNodeCreator_Call) RunAndReturn(run func(func(context.Context, string, int64) (types.QueryNodeClient, error))) *MockQueryCoord_SetQueryNodeCreator_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1937,7 +1937,7 @@ func (_c *MockQueryCoord_SetTiKVClient_Call) Return() *MockQueryCoord_SetTiKVCli
}
func (_c *MockQueryCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client)) *MockQueryCoord_SetTiKVClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -2118,7 +2118,7 @@ func (_c *MockQueryCoord_ShowPartitions_Call) RunAndReturn(run func(context.Cont
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *MockQueryCoord) Start() error {
ret := _m.Called()
@ -2163,7 +2163,7 @@ func (_c *MockQueryCoord_Start_Call) RunAndReturn(run func() error) *MockQueryCo
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockQueryCoord) Stop() error {
ret := _m.Called()
@ -2768,7 +2768,7 @@ func (_c *MockQueryCoord_UpdateStateCode_Call) Return() *MockQueryCoord_UpdateSt
}
func (_c *MockQueryCoord_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *MockQueryCoord_UpdateStateCode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

4
internal/mocks/mock_querycoord_client.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -327,7 +327,7 @@ func (_c *MockQueryCoordClient_CheckQueryNodeDistribution_Call) RunAndReturn(run
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockQueryCoordClient) Close() error {
ret := _m.Called()

20
internal/mocks/mock_querynode.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -148,7 +148,7 @@ func (_c *MockQueryNode_DeleteBatch_Call) RunAndReturn(run func(context.Context,
return _c
}
// GetAddress provides a mock function with given fields:
// GetAddress provides a mock function with no fields
func (_m *MockQueryNode) GetAddress() string {
ret := _m.Called()
@ -370,7 +370,7 @@ func (_c *MockQueryNode_GetMetrics_Call) RunAndReturn(run func(context.Context,
return _c
}
// GetNodeID provides a mock function with given fields:
// GetNodeID provides a mock function with no fields
func (_m *MockQueryNode) GetNodeID() int64 {
ret := _m.Called()
@ -651,7 +651,7 @@ func (_c *MockQueryNode_GetTimeTickChannel_Call) RunAndReturn(run func(context.C
return _c
}
// Init provides a mock function with given fields:
// Init provides a mock function with no fields
func (_m *MockQueryNode) Init() error {
ret := _m.Called()
@ -1026,7 +1026,7 @@ func (_c *MockQueryNode_QueryStreamSegments_Call) RunAndReturn(run func(*querypb
return _c
}
// Register provides a mock function with given fields:
// Register provides a mock function with no fields
func (_m *MockQueryNode) Register() error {
ret := _m.Called()
@ -1395,7 +1395,7 @@ func (_c *MockQueryNode_SetAddress_Call) Return() *MockQueryNode_SetAddress_Call
}
func (_c *MockQueryNode_SetAddress_Call) RunAndReturn(run func(string)) *MockQueryNode_SetAddress_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1428,7 +1428,7 @@ func (_c *MockQueryNode_SetEtcdClient_Call) Return() *MockQueryNode_SetEtcdClien
}
func (_c *MockQueryNode_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockQueryNode_SetEtcdClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1491,7 +1491,7 @@ func (_c *MockQueryNode_ShowConfigurations_Call) RunAndReturn(run func(context.C
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *MockQueryNode) Start() error {
ret := _m.Called()
@ -1536,7 +1536,7 @@ func (_c *MockQueryNode_Start_Call) RunAndReturn(run func() error) *MockQueryNod
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockQueryNode) Stop() error {
ret := _m.Called()
@ -1787,7 +1787,7 @@ func (_c *MockQueryNode_UpdateStateCode_Call) Return() *MockQueryNode_UpdateStat
}
func (_c *MockQueryNode_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *MockQueryNode_UpdateStateCode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

4
internal/mocks/mock_querynode_client.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -31,7 +31,7 @@ func (_m *MockQueryNodeClient) EXPECT() *MockQueryNodeClient_Expecter {
return &MockQueryNodeClient_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockQueryNodeClient) Close() error {
ret := _m.Called()

26
internal/mocks/mock_rootcoord.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -1926,7 +1926,7 @@ func (_c *RootCoord_GetTimeTickChannel_Call) RunAndReturn(run func(context.Conte
return _c
}
// GracefulStop provides a mock function with given fields:
// GracefulStop provides a mock function with no fields
func (_m *RootCoord) GracefulStop() {
_m.Called()
}
@ -1954,7 +1954,7 @@ func (_c *RootCoord_GracefulStop_Call) Return() *RootCoord_GracefulStop_Call {
}
func (_c *RootCoord_GracefulStop_Call) RunAndReturn(run func()) *RootCoord_GracefulStop_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -2076,7 +2076,7 @@ func (_c *RootCoord_HasPartition_Call) RunAndReturn(run func(context.Context, *m
return _c
}
// Init provides a mock function with given fields:
// Init provides a mock function with no fields
func (_m *RootCoord) Init() error {
ret := _m.Called()
@ -2652,7 +2652,7 @@ func (_c *RootCoord_OperateUserRole_Call) RunAndReturn(run func(context.Context,
return _c
}
// Register provides a mock function with given fields:
// Register provides a mock function with no fields
func (_m *RootCoord) Register() error {
ret := _m.Called()
@ -2726,7 +2726,7 @@ func (_c *RootCoord_RegisterStreamingCoordGRPCService_Call) Return() *RootCoord_
}
func (_c *RootCoord_RegisterStreamingCoordGRPCService_Call) RunAndReturn(run func(*grpc.Server)) *RootCoord_RegisterStreamingCoordGRPCService_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -3054,7 +3054,7 @@ func (_c *RootCoord_SetAddress_Call) Return() *RootCoord_SetAddress_Call {
}
func (_c *RootCoord_SetAddress_Call) RunAndReturn(run func(string)) *RootCoord_SetAddress_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -3133,7 +3133,7 @@ func (_c *RootCoord_SetEtcdClient_Call) Return() *RootCoord_SetEtcdClient_Call {
}
func (_c *RootCoord_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *RootCoord_SetEtcdClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -3166,7 +3166,7 @@ func (_c *RootCoord_SetProxyCreator_Call) Return() *RootCoord_SetProxyCreator_Ca
}
func (_c *RootCoord_SetProxyCreator_Call) RunAndReturn(run func(func(context.Context, string, int64) (types.ProxyClient, error))) *RootCoord_SetProxyCreator_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -3245,7 +3245,7 @@ func (_c *RootCoord_SetTiKVClient_Call) Return() *RootCoord_SetTiKVClient_Call {
}
func (_c *RootCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client)) *RootCoord_SetTiKVClient_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -3603,7 +3603,7 @@ func (_c *RootCoord_ShowSegments_Call) RunAndReturn(run func(context.Context, *m
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *RootCoord) Start() error {
ret := _m.Called()
@ -3648,7 +3648,7 @@ func (_c *RootCoord_Start_Call) RunAndReturn(run func() error) *RootCoord_Start_
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *RootCoord) Stop() error {
ret := _m.Called()
@ -3840,7 +3840,7 @@ func (_c *RootCoord_UpdateStateCode_Call) Return() *RootCoord_UpdateStateCode_Ca
}
func (_c *RootCoord_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *RootCoord_UpdateStateCode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

4
internal/mocks/mock_rootcoord_client.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -625,7 +625,7 @@ func (_c *MockRootCoordClient_CheckHealth_Call) RunAndReturn(run func(context.Co
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockRootCoordClient) Close() error {
ret := _m.Called()

6
internal/mocks/streamingcoord/mock_client/mock_BroadcastService.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_client
@ -130,7 +130,7 @@ func (_c *MockBroadcastService_Broadcast_Call) RunAndReturn(run func(context.Con
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockBroadcastService) Close() {
_m.Called()
}
@ -158,7 +158,7 @@ func (_c *MockBroadcastService_Close_Call) Return() *MockBroadcastService_Close_
}
func (_c *MockBroadcastService_Close_Call) RunAndReturn(run func()) *MockBroadcastService_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

10
internal/mocks/streamingcoord/mock_client/mock_Client.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_client
@ -20,7 +20,7 @@ func (_m *MockClient) EXPECT() *MockClient_Expecter {
return &MockClient_Expecter{mock: &_m.Mock}
}
// Assignment provides a mock function with given fields:
// Assignment provides a mock function with no fields
func (_m *MockClient) Assignment() client.AssignmentService {
ret := _m.Called()
@ -67,7 +67,7 @@ func (_c *MockClient_Assignment_Call) RunAndReturn(run func() client.AssignmentS
return _c
}
// Broadcast provides a mock function with given fields:
// Broadcast provides a mock function with no fields
func (_m *MockClient) Broadcast() client.BroadcastService {
ret := _m.Called()
@ -114,7 +114,7 @@ func (_c *MockClient_Broadcast_Call) RunAndReturn(run func() client.BroadcastSer
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockClient) Close() {
_m.Called()
}
@ -142,7 +142,7 @@ func (_c *MockClient_Close_Call) Return() *MockClient_Close_Call {
}
func (_c *MockClient_Close_Call) RunAndReturn(run func()) *MockClient_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

6
internal/mocks/streamingcoord/server/mock_balancer/mock_Balancer.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_balancer
@ -24,7 +24,7 @@ func (_m *MockBalancer) EXPECT() *MockBalancer_Expecter {
return &MockBalancer_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockBalancer) Close() {
_m.Called()
}
@ -52,7 +52,7 @@ func (_c *MockBalancer_Close_Call) Return() *MockBalancer_Close_Call {
}
func (_c *MockBalancer_Close_Call) RunAndReturn(run func()) *MockBalancer_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

16
internal/mocks/streamingcoord/server/mock_broadcaster/mock_AppendOperator.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_broadcaster
@ -8,7 +8,7 @@ import (
message "github.com/milvus-io/milvus/pkg/v2/streaming/util/message"
mock "github.com/stretchr/testify/mock"
streaming "github.com/milvus-io/milvus/internal/distributed/streaming"
types "github.com/milvus-io/milvus/pkg/v2/streaming/util/types"
)
// MockAppendOperator is an autogenerated mock type for the AppendOperator type
@ -25,7 +25,7 @@ func (_m *MockAppendOperator) EXPECT() *MockAppendOperator_Expecter {
}
// AppendMessages provides a mock function with given fields: ctx, msgs
func (_m *MockAppendOperator) AppendMessages(ctx context.Context, msgs ...message.MutableMessage) streaming.AppendResponses {
func (_m *MockAppendOperator) AppendMessages(ctx context.Context, msgs ...message.MutableMessage) types.AppendResponses {
_va := make([]interface{}, len(msgs))
for _i := range msgs {
_va[_i] = msgs[_i]
@ -39,11 +39,11 @@ func (_m *MockAppendOperator) AppendMessages(ctx context.Context, msgs ...messag
panic("no return value specified for AppendMessages")
}
var r0 streaming.AppendResponses
if rf, ok := ret.Get(0).(func(context.Context, ...message.MutableMessage) streaming.AppendResponses); ok {
var r0 types.AppendResponses
if rf, ok := ret.Get(0).(func(context.Context, ...message.MutableMessage) types.AppendResponses); ok {
r0 = rf(ctx, msgs...)
} else {
r0 = ret.Get(0).(streaming.AppendResponses)
r0 = ret.Get(0).(types.AppendResponses)
}
return r0
@ -75,12 +75,12 @@ func (_c *MockAppendOperator_AppendMessages_Call) Run(run func(ctx context.Conte
return _c
}
func (_c *MockAppendOperator_AppendMessages_Call) Return(_a0 streaming.AppendResponses) *MockAppendOperator_AppendMessages_Call {
func (_c *MockAppendOperator_AppendMessages_Call) Return(_a0 types.AppendResponses) *MockAppendOperator_AppendMessages_Call {
_c.Call.Return(_a0)
return _c
}
func (_c *MockAppendOperator_AppendMessages_Call) RunAndReturn(run func(context.Context, ...message.MutableMessage) streaming.AppendResponses) *MockAppendOperator_AppendMessages_Call {
func (_c *MockAppendOperator_AppendMessages_Call) RunAndReturn(run func(context.Context, ...message.MutableMessage) types.AppendResponses) *MockAppendOperator_AppendMessages_Call {
_c.Call.Return(run)
return _c
}

6
internal/mocks/streamingnode/client/handler/mock_assignment/mock_Watcher.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_assignment
@ -22,7 +22,7 @@ func (_m *MockWatcher) EXPECT() *MockWatcher_Expecter {
return &MockWatcher_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockWatcher) Close() {
_m.Called()
}
@ -50,7 +50,7 @@ func (_c *MockWatcher_Close_Call) Return() *MockWatcher_Close_Call {
}
func (_c *MockWatcher_Close_Call) RunAndReturn(run func()) *MockWatcher_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

10
internal/mocks/streamingnode/client/handler/mock_consumer/mock_Consumer.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_consumer
@ -17,7 +17,7 @@ func (_m *MockConsumer) EXPECT() *MockConsumer_Expecter {
return &MockConsumer_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockConsumer) Close() {
_m.Called()
}
@ -45,11 +45,11 @@ func (_c *MockConsumer_Close_Call) Return() *MockConsumer_Close_Call {
}
func (_c *MockConsumer_Close_Call) RunAndReturn(run func()) *MockConsumer_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Done provides a mock function with given fields:
// Done provides a mock function with no fields
func (_m *MockConsumer) Done() <-chan struct{} {
ret := _m.Called()
@ -96,7 +96,7 @@ func (_c *MockConsumer_Done_Call) RunAndReturn(run func() <-chan struct{}) *Mock
return _c
}
// Error provides a mock function with given fields:
// Error provides a mock function with no fields
func (_m *MockConsumer) Error() error {
ret := _m.Called()

12
internal/mocks/streamingnode/client/handler/mock_producer/mock_Producer.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_producer
@ -24,7 +24,7 @@ func (_m *MockProducer) EXPECT() *MockProducer_Expecter {
return &MockProducer_Expecter{mock: &_m.Mock}
}
// Assignment provides a mock function with given fields:
// Assignment provides a mock function with no fields
func (_m *MockProducer) Assignment() types.PChannelInfoAssigned {
ret := _m.Called()
@ -69,7 +69,7 @@ func (_c *MockProducer_Assignment_Call) RunAndReturn(run func() types.PChannelIn
return _c
}
// Available provides a mock function with given fields:
// Available provides a mock function with no fields
func (_m *MockProducer) Available() <-chan struct{} {
ret := _m.Called()
@ -116,7 +116,7 @@ func (_c *MockProducer_Available_Call) RunAndReturn(run func() <-chan struct{})
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockProducer) Close() {
_m.Called()
}
@ -144,11 +144,11 @@ func (_c *MockProducer_Close_Call) Return() *MockProducer_Close_Call {
}
func (_c *MockProducer_Close_Call) RunAndReturn(run func()) *MockProducer_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// IsAvailable provides a mock function with given fields:
// IsAvailable provides a mock function with no fields
func (_m *MockProducer) IsAvailable() bool {
ret := _m.Called()

6
internal/mocks/streamingnode/client/mock_handler/mock_HandlerClient.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_handler
@ -27,7 +27,7 @@ func (_m *MockHandlerClient) EXPECT() *MockHandlerClient_Expecter {
return &MockHandlerClient_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockHandlerClient) Close() {
_m.Called()
}
@ -55,7 +55,7 @@ func (_c *MockHandlerClient_Close_Call) Return() *MockHandlerClient_Close_Call {
}
func (_c *MockHandlerClient_Close_Call) RunAndReturn(run func()) *MockHandlerClient_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

6
internal/mocks/streamingnode/client/mock_manager/mock_ManagerClient.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_manager
@ -70,7 +70,7 @@ func (_c *MockManagerClient_Assign_Call) RunAndReturn(run func(context.Context,
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockManagerClient) Close() {
_m.Called()
}
@ -98,7 +98,7 @@ func (_c *MockManagerClient_Close_Call) Return() *MockManagerClient_Close_Call {
}
func (_c *MockManagerClient_Close_Call) RunAndReturn(run func()) *MockManagerClient_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

16
internal/mocks/streamingnode/server/mock_flusher/mock_Flusher.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_flusher
@ -97,11 +97,11 @@ func (_c *MockFlusher_RegisterVChannel_Call) Return() *MockFlusher_RegisterVChan
}
func (_c *MockFlusher_RegisterVChannel_Call) RunAndReturn(run func(string, wal.WAL)) *MockFlusher_RegisterVChannel_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *MockFlusher) Start() {
_m.Called()
}
@ -129,11 +129,11 @@ func (_c *MockFlusher_Start_Call) Return() *MockFlusher_Start_Call {
}
func (_c *MockFlusher_Start_Call) RunAndReturn(run func()) *MockFlusher_Start_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockFlusher) Stop() {
_m.Called()
}
@ -161,7 +161,7 @@ func (_c *MockFlusher_Stop_Call) Return() *MockFlusher_Stop_Call {
}
func (_c *MockFlusher_Stop_Call) RunAndReturn(run func()) *MockFlusher_Stop_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -194,7 +194,7 @@ func (_c *MockFlusher_UnregisterPChannel_Call) Return() *MockFlusher_UnregisterP
}
func (_c *MockFlusher_UnregisterPChannel_Call) RunAndReturn(run func(string)) *MockFlusher_UnregisterPChannel_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -227,7 +227,7 @@ func (_c *MockFlusher_UnregisterVChannel_Call) Return() *MockFlusher_UnregisterV
}
func (_c *MockFlusher_UnregisterVChannel_Call) RunAndReturn(run func(string)) *MockFlusher_UnregisterVChannel_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

6
internal/mocks/streamingnode/server/mock_wal/mock_Opener.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_wal
@ -22,7 +22,7 @@ func (_m *MockOpener) EXPECT() *MockOpener_Expecter {
return &MockOpener_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockOpener) Close() {
_m.Called()
}
@ -50,7 +50,7 @@ func (_c *MockOpener_Close_Call) Return() *MockOpener_Close_Call {
}
func (_c *MockOpener_Close_Call) RunAndReturn(run func()) *MockOpener_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

6
internal/mocks/streamingnode/server/mock_wal/mock_OpenerBuilder.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_wal
@ -20,7 +20,7 @@ func (_m *MockOpenerBuilder) EXPECT() *MockOpenerBuilder_Expecter {
return &MockOpenerBuilder_Expecter{mock: &_m.Mock}
}
// Build provides a mock function with given fields:
// Build provides a mock function with no fields
func (_m *MockOpenerBuilder) Build() (wal.Opener, error) {
ret := _m.Called()
@ -77,7 +77,7 @@ func (_c *MockOpenerBuilder_Build_Call) RunAndReturn(run func() (wal.Opener, err
return _c
}
// Name provides a mock function with given fields:
// Name provides a mock function with no fields
func (_m *MockOpenerBuilder) Name() string {
ret := _m.Called()

12
internal/mocks/streamingnode/server/mock_wal/mock_Scanner.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_wal
@ -22,7 +22,7 @@ func (_m *MockScanner) EXPECT() *MockScanner_Expecter {
return &MockScanner_Expecter{mock: &_m.Mock}
}
// Chan provides a mock function with given fields:
// Chan provides a mock function with no fields
func (_m *MockScanner) Chan() <-chan message.ImmutableMessage {
ret := _m.Called()
@ -69,7 +69,7 @@ func (_c *MockScanner_Chan_Call) RunAndReturn(run func() <-chan message.Immutabl
return _c
}
// Channel provides a mock function with given fields:
// Channel provides a mock function with no fields
func (_m *MockScanner) Channel() types.PChannelInfo {
ret := _m.Called()
@ -114,7 +114,7 @@ func (_c *MockScanner_Channel_Call) RunAndReturn(run func() types.PChannelInfo)
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockScanner) Close() error {
ret := _m.Called()
@ -159,7 +159,7 @@ func (_c *MockScanner_Close_Call) RunAndReturn(run func() error) *MockScanner_Cl
return _c
}
// Done provides a mock function with given fields:
// Done provides a mock function with no fields
func (_m *MockScanner) Done() <-chan struct{} {
ret := _m.Called()
@ -206,7 +206,7 @@ func (_c *MockScanner_Done_Call) RunAndReturn(run func() <-chan struct{}) *MockS
return _c
}
// Error provides a mock function with given fields:
// Error provides a mock function with no fields
func (_m *MockScanner) Error() error {
ret := _m.Called()

16
internal/mocks/streamingnode/server/mock_wal/mock_WAL.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_wal
@ -116,11 +116,11 @@ func (_c *MockWAL_AppendAsync_Call) Return() *MockWAL_AppendAsync_Call {
}
func (_c *MockWAL_AppendAsync_Call) RunAndReturn(run func(context.Context, message.MutableMessage, func(*types.AppendResult, error))) *MockWAL_AppendAsync_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Available provides a mock function with given fields:
// Available provides a mock function with no fields
func (_m *MockWAL) Available() <-chan struct{} {
ret := _m.Called()
@ -167,7 +167,7 @@ func (_c *MockWAL_Available_Call) RunAndReturn(run func() <-chan struct{}) *Mock
return _c
}
// Channel provides a mock function with given fields:
// Channel provides a mock function with no fields
func (_m *MockWAL) Channel() types.PChannelInfo {
ret := _m.Called()
@ -212,7 +212,7 @@ func (_c *MockWAL_Channel_Call) RunAndReturn(run func() types.PChannelInfo) *Moc
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockWAL) Close() {
_m.Called()
}
@ -240,11 +240,11 @@ func (_c *MockWAL_Close_Call) Return() *MockWAL_Close_Call {
}
func (_c *MockWAL_Close_Call) RunAndReturn(run func()) *MockWAL_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// IsAvailable provides a mock function with given fields:
// IsAvailable provides a mock function with no fields
func (_m *MockWAL) IsAvailable() bool {
ret := _m.Called()
@ -348,7 +348,7 @@ func (_c *MockWAL_Read_Call) RunAndReturn(run func(context.Context, wal.ReadOpti
return _c
}
// WALName provides a mock function with given fields:
// WALName provides a mock function with no fields
func (_m *MockWAL) WALName() string {
ret := _m.Called()

8
internal/mocks/streamingnode/server/mock_walmanager/mock_Manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_walmanager
@ -24,7 +24,7 @@ func (_m *MockManager) EXPECT() *MockManager_Expecter {
return &MockManager_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockManager) Close() {
_m.Called()
}
@ -52,11 +52,11 @@ func (_c *MockManager_Close_Call) Return() *MockManager_Close_Call {
}
func (_c *MockManager_Close_Call) RunAndReturn(run func()) *MockManager_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// GetAllAvailableChannels provides a mock function with given fields:
// GetAllAvailableChannels provides a mock function with no fields
func (_m *MockManager) GetAllAvailableChannels() ([]types.PChannelInfo, error) {
ret := _m.Called()

12
internal/mocks/streamingnode/server/wal/interceptors/segment/mock_inspector/mock_SealOperator.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_inspector
@ -25,7 +25,7 @@ func (_m *MockSealOperator) EXPECT() *MockSealOperator_Expecter {
return &MockSealOperator_Expecter{mock: &_m.Mock}
}
// Channel provides a mock function with given fields:
// Channel provides a mock function with no fields
func (_m *MockSealOperator) Channel() types.PChannelInfo {
ret := _m.Called()
@ -70,7 +70,7 @@ func (_c *MockSealOperator_Channel_Call) RunAndReturn(run func() types.PChannelI
return _c
}
// IsNoWaitSeal provides a mock function with given fields:
// IsNoWaitSeal provides a mock function with no fields
func (_m *MockSealOperator) IsNoWaitSeal() bool {
ret := _m.Called()
@ -159,7 +159,7 @@ func (_c *MockSealOperator_MustSealSegments_Call) Return() *MockSealOperator_Mus
}
func (_c *MockSealOperator_MustSealSegments_Call) RunAndReturn(run func(context.Context, ...stats.SegmentBelongs)) *MockSealOperator_MustSealSegments_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -207,7 +207,7 @@ func (_c *MockSealOperator_TryToSealSegments_Call) Return() *MockSealOperator_Tr
}
func (_c *MockSealOperator_TryToSealSegments_Call) RunAndReturn(run func(context.Context, ...stats.SegmentBelongs)) *MockSealOperator_TryToSealSegments_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -240,7 +240,7 @@ func (_c *MockSealOperator_TryToSealWaitedSegment_Call) Return() *MockSealOperat
}
func (_c *MockSealOperator_TryToSealWaitedSegment_Call) RunAndReturn(run func(context.Context)) *MockSealOperator_TryToSealWaitedSegment_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

8
internal/mocks/streamingnode/server/wal/interceptors/timetick/mock_inspector/mock_TimeTickSyncOperator.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_inspector
@ -24,7 +24,7 @@ func (_m *MockTimeTickSyncOperator) EXPECT() *MockTimeTickSyncOperator_Expecter
return &MockTimeTickSyncOperator_Expecter{mock: &_m.Mock}
}
// Channel provides a mock function with given fields:
// Channel provides a mock function with no fields
func (_m *MockTimeTickSyncOperator) Channel() types.PChannelInfo {
ret := _m.Called()
@ -98,11 +98,11 @@ func (_c *MockTimeTickSyncOperator_Sync_Call) Return() *MockTimeTickSyncOperator
}
func (_c *MockTimeTickSyncOperator_Sync_Call) RunAndReturn(run func(context.Context)) *MockTimeTickSyncOperator_Sync_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// TimeTickNotifier provides a mock function with given fields:
// TimeTickNotifier provides a mock function with no fields
func (_m *MockTimeTickSyncOperator) TimeTickNotifier() *inspector.TimeTickNotifier {
ret := _m.Called()

6
internal/mocks/streamingnode/server/wal/mock_interceptors/mock_Interceptor.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_interceptors
@ -23,7 +23,7 @@ func (_m *MockInterceptor) EXPECT() *MockInterceptor_Expecter {
return &MockInterceptor_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockInterceptor) Close() {
_m.Called()
}
@ -51,7 +51,7 @@ func (_c *MockInterceptor_Close_Call) Return() *MockInterceptor_Close_Call {
}
func (_c *MockInterceptor_Close_Call) RunAndReturn(run func()) *MockInterceptor_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

2
internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorBuilder.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_interceptors

8
internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorWithReady.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_interceptors
@ -23,7 +23,7 @@ func (_m *MockInterceptorWithReady) EXPECT() *MockInterceptorWithReady_Expecter
return &MockInterceptorWithReady_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockInterceptorWithReady) Close() {
_m.Called()
}
@ -51,7 +51,7 @@ func (_c *MockInterceptorWithReady_Close_Call) Return() *MockInterceptorWithRead
}
func (_c *MockInterceptorWithReady_Close_Call) RunAndReturn(run func()) *MockInterceptorWithReady_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -115,7 +115,7 @@ func (_c *MockInterceptorWithReady_DoAppend_Call) RunAndReturn(run func(context.
return _c
}
// Ready provides a mock function with given fields:
// Ready provides a mock function with no fields
func (_m *MockInterceptorWithReady) Ready() <-chan struct{} {
ret := _m.Called()

14
internal/mocks/util/mock_segcore/mock_CSegment.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_segcore
@ -186,7 +186,7 @@ func (_c *MockCSegment_HasRawData_Call) RunAndReturn(run func(int64) bool) *Mock
return _c
}
// ID provides a mock function with given fields:
// ID provides a mock function with no fields
func (_m *MockCSegment) ID() int64 {
ret := _m.Called()
@ -349,7 +349,7 @@ func (_c *MockCSegment_LoadFieldData_Call) RunAndReturn(run func(context.Context
return _c
}
// MemSize provides a mock function with given fields:
// MemSize provides a mock function with no fields
func (_m *MockCSegment) MemSize() int64 {
ret := _m.Called()
@ -394,7 +394,7 @@ func (_c *MockCSegment_MemSize_Call) RunAndReturn(run func() int64) *MockCSegmen
return _c
}
// RawPointer provides a mock function with given fields:
// RawPointer provides a mock function with no fields
func (_m *MockCSegment) RawPointer() segcore.CSegmentInterface {
ret := _m.Called()
@ -439,7 +439,7 @@ func (_c *MockCSegment_RawPointer_Call) RunAndReturn(run func() segcore.CSegment
return _c
}
// Release provides a mock function with given fields:
// Release provides a mock function with no fields
func (_m *MockCSegment) Release() {
_m.Called()
}
@ -467,7 +467,7 @@ func (_c *MockCSegment_Release_Call) Return() *MockCSegment_Release_Call {
}
func (_c *MockCSegment_Release_Call) RunAndReturn(run func()) *MockCSegment_Release_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -589,7 +589,7 @@ func (_c *MockCSegment_RetrieveByOffsets_Call) RunAndReturn(run func(context.Con
return _c
}
// RowNum provides a mock function with given fields:
// RowNum provides a mock function with no fields
func (_m *MockCSegment) RowNum() int64 {
ret := _m.Called()

2
internal/mocks/util/searchutil/mock_optimizers/mock_QueryHook.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_optimizers

4
internal/mocks/util/streamingutil/service/mock_discoverer/mock_Discoverer.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_discoverer
@ -69,7 +69,7 @@ func (_c *MockDiscoverer_Discover_Call) RunAndReturn(run func(context.Context, f
return _c
}
// NewVersionedState provides a mock function with given fields:
// NewVersionedState provides a mock function with no fields
func (_m *MockDiscoverer) NewVersionedState() discoverer.VersionedState {
ret := _m.Called()

10
internal/mocks/util/streamingutil/service/mock_lazygrpc/mock_Service.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_lazygrpc
@ -23,7 +23,7 @@ func (_m *MockService[T]) EXPECT() *MockService_Expecter[T] {
return &MockService_Expecter[T]{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockService[T]) Close() {
_m.Called()
}
@ -51,7 +51,7 @@ func (_c *MockService_Close_Call[T]) Return() *MockService_Close_Call[T] {
}
func (_c *MockService_Close_Call[T]) RunAndReturn(run func()) *MockService_Close_Call[T] {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -129,7 +129,9 @@ func (_m *MockService[T]) GetService(ctx context.Context) (T, error) {
if rf, ok := ret.Get(0).(func(context.Context) T); ok {
r0 = rf(ctx)
} else {
r0 = ret.Get(0).(T)
if ret.Get(0) != nil {
r0 = ret.Get(0).(T)
}
}
if rf, ok := ret.Get(1).(func(context.Context) error); ok {

10
internal/mocks/util/streamingutil/service/mock_resolver/mock_Builder.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_resolver
@ -82,7 +82,7 @@ func (_c *MockBuilder_Build_Call) RunAndReturn(run func(resolver.Target, resolve
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockBuilder) Close() {
_m.Called()
}
@ -110,11 +110,11 @@ func (_c *MockBuilder_Close_Call) Return() *MockBuilder_Close_Call {
}
func (_c *MockBuilder_Close_Call) RunAndReturn(run func()) *MockBuilder_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Resolver provides a mock function with given fields:
// Resolver provides a mock function with no fields
func (_m *MockBuilder) Resolver() serviceresolver.Resolver {
ret := _m.Called()
@ -161,7 +161,7 @@ func (_c *MockBuilder_Resolver_Call) RunAndReturn(run func() serviceresolver.Res
return _c
}
// Scheme provides a mock function with given fields:
// Scheme provides a mock function with no fields
func (_m *MockBuilder) Scheme() string {
ret := _m.Called()

4
internal/mocks/util/streamingutil/service/mock_resolver/mock_Resolver.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_resolver
@ -22,7 +22,7 @@ func (_m *MockResolver) EXPECT() *MockResolver_Expecter {
return &MockResolver_Expecter{mock: &_m.Mock}
}
// GetLatestState provides a mock function with given fields:
// GetLatestState provides a mock function with no fields
func (_m *MockResolver) GetLatestState() discoverer.VersionedState {
ret := _m.Called()

18
internal/proxy/mock_cache.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package proxy
@ -110,7 +110,7 @@ func (_c *MockCache_DeprecateShardCache_Call) Return() *MockCache_DeprecateShard
}
func (_c *MockCache_DeprecateShardCache_Call) RunAndReturn(run func(string, string)) *MockCache_DeprecateShardCache_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -944,7 +944,7 @@ func (_c *MockCache_InitPolicyInfo_Call) Return() *MockCache_InitPolicyInfo_Call
}
func (_c *MockCache_InitPolicyInfo_Call) RunAndReturn(run func([]string, []string)) *MockCache_InitPolicyInfo_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -977,11 +977,11 @@ func (_c *MockCache_InvalidateShardLeaderCache_Call) Return() *MockCache_Invalid
}
func (_c *MockCache_InvalidateShardLeaderCache_Call) RunAndReturn(run func([]int64)) *MockCache_InvalidateShardLeaderCache_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// ListShardLocation provides a mock function with given fields:
// ListShardLocation provides a mock function with no fields
func (_m *MockCache) ListShardLocation() map[int64]nodeInfo {
ret := _m.Called()
@ -1105,7 +1105,7 @@ func (_c *MockCache_RemoveCollection_Call) Return() *MockCache_RemoveCollection_
}
func (_c *MockCache_RemoveCollection_Call) RunAndReturn(run func(context.Context, string, string)) *MockCache_RemoveCollection_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1189,7 +1189,7 @@ func (_c *MockCache_RemoveCredential_Call) Return() *MockCache_RemoveCredential_
}
func (_c *MockCache_RemoveCredential_Call) RunAndReturn(run func(string)) *MockCache_RemoveCredential_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1223,7 +1223,7 @@ func (_c *MockCache_RemoveDatabase_Call) Return() *MockCache_RemoveDatabase_Call
}
func (_c *MockCache_RemoveDatabase_Call) RunAndReturn(run func(context.Context, string)) *MockCache_RemoveDatabase_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1256,7 +1256,7 @@ func (_c *MockCache_UpdateCredential_Call) Return() *MockCache_UpdateCredential_
}
func (_c *MockCache_UpdateCredential_Call) RunAndReturn(run func(*internalpb.CredentialInfo)) *MockCache_UpdateCredential_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

8
internal/proxy/mock_channels_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package proxy
@ -197,7 +197,7 @@ func (_c *MockChannelsMgr_getVChannels_Call) RunAndReturn(run func(int64) ([]str
return _c
}
// removeAllDMLStream provides a mock function with given fields:
// removeAllDMLStream provides a mock function with no fields
func (_m *MockChannelsMgr) removeAllDMLStream() {
_m.Called()
}
@ -225,7 +225,7 @@ func (_c *MockChannelsMgr_removeAllDMLStream_Call) Return() *MockChannelsMgr_rem
}
func (_c *MockChannelsMgr_removeAllDMLStream_Call) RunAndReturn(run func()) *MockChannelsMgr_removeAllDMLStream_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -258,7 +258,7 @@ func (_c *MockChannelsMgr_removeDMLStream_Call) Return() *MockChannelsMgr_remove
}
func (_c *MockChannelsMgr_removeDMLStream_Call) RunAndReturn(run func(int64)) *MockChannelsMgr_removeDMLStream_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

14
internal/proxy/mock_lb_balancer.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package proxy
@ -52,11 +52,11 @@ func (_c *MockLBBalancer_CancelWorkload_Call) Return() *MockLBBalancer_CancelWor
}
func (_c *MockLBBalancer_CancelWorkload_Call) RunAndReturn(run func(int64, int64)) *MockLBBalancer_CancelWorkload_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockLBBalancer) Close() {
_m.Called()
}
@ -84,7 +84,7 @@ func (_c *MockLBBalancer_Close_Call) Return() *MockLBBalancer_Close_Call {
}
func (_c *MockLBBalancer_Close_Call) RunAndReturn(run func()) *MockLBBalancer_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -117,7 +117,7 @@ func (_c *MockLBBalancer_RegisterNodeInfo_Call) Return() *MockLBBalancer_Registe
}
func (_c *MockLBBalancer_RegisterNodeInfo_Call) RunAndReturn(run func([]nodeInfo)) *MockLBBalancer_RegisterNodeInfo_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -208,7 +208,7 @@ func (_c *MockLBBalancer_Start_Call) Return() *MockLBBalancer_Start_Call {
}
func (_c *MockLBBalancer_Start_Call) RunAndReturn(run func(context.Context)) *MockLBBalancer_Start_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -242,7 +242,7 @@ func (_c *MockLBBalancer_UpdateCostMetrics_Call) Return() *MockLBBalancer_Update
}
func (_c *MockLBBalancer_UpdateCostMetrics_Call) RunAndReturn(run func(int64, *internalpb.CostAggregation)) *MockLBBalancer_UpdateCostMetrics_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

10
internal/proxy/mock_lb_policy.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package proxy
@ -22,7 +22,7 @@ func (_m *MockLBPolicy) EXPECT() *MockLBPolicy_Expecter {
return &MockLBPolicy_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockLBPolicy) Close() {
_m.Called()
}
@ -50,7 +50,7 @@ func (_c *MockLBPolicy_Close_Call) Return() *MockLBPolicy_Close_Call {
}
func (_c *MockLBPolicy_Close_Call) RunAndReturn(run func()) *MockLBPolicy_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -177,7 +177,7 @@ func (_c *MockLBPolicy_Start_Call) Return() *MockLBPolicy_Start_Call {
}
func (_c *MockLBPolicy_Start_Call) RunAndReturn(run func(context.Context)) *MockLBPolicy_Start_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -211,7 +211,7 @@ func (_c *MockLBPolicy_UpdateCostMetrics_Call) Return() *MockLBPolicy_UpdateCost
}
func (_c *MockLBPolicy_UpdateCostMetrics_Call) RunAndReturn(run func(int64, *internalpb.CostAggregation)) *MockLBPolicy_UpdateCostMetrics_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

8
internal/proxy/mock_shardclient_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package proxy
@ -22,7 +22,7 @@ func (_m *MockShardClientManager) EXPECT() *MockShardClientManager_Expecter {
return &MockShardClientManager_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockShardClientManager) Close() {
_m.Called()
}
@ -50,7 +50,7 @@ func (_c *MockShardClientManager_Close_Call) Return() *MockShardClientManager_Cl
}
func (_c *MockShardClientManager_Close_Call) RunAndReturn(run func()) *MockShardClientManager_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -142,7 +142,7 @@ func (_c *MockShardClientManager_SetClientCreatorFunc_Call) Return() *MockShardC
}
func (_c *MockShardClientManager_SetClientCreatorFunc_Call) RunAndReturn(run func(queryNodeCreatorFunc)) *MockShardClientManager_SetClientCreatorFunc_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

2
internal/proxy/mock_tso_test.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package proxy

2
internal/querycoordv2/balance/mock_balancer.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package balance

12
internal/querycoordv2/dist/mock_controller.go vendored
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package dist
@ -50,7 +50,7 @@ func (_c *MockController_Remove_Call) Return() *MockController_Remove_Call {
}
func (_c *MockController_Remove_Call) RunAndReturn(run func(int64)) *MockController_Remove_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -84,11 +84,11 @@ func (_c *MockController_StartDistInstance_Call) Return() *MockController_StartD
}
func (_c *MockController_StartDistInstance_Call) RunAndReturn(run func(context.Context, int64)) *MockController_StartDistInstance_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockController) Stop() {
_m.Called()
}
@ -116,7 +116,7 @@ func (_c *MockController_Stop_Call) Return() *MockController_Stop_Call {
}
func (_c *MockController_Stop_Call) RunAndReturn(run func()) *MockController_Stop_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -149,7 +149,7 @@ func (_c *MockController_SyncAll_Call) Return() *MockController_SyncAll_Call {
}
func (_c *MockController_SyncAll_Call) RunAndReturn(run func(context.Context)) *MockController_SyncAll_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

2
internal/querycoordv2/meta/mock_broker.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package meta

8
internal/querycoordv2/meta/mock_target_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package meta
@ -953,7 +953,7 @@ func (_c *MockTargetManager_RemoveCollection_Call) Return() *MockTargetManager_R
}
func (_c *MockTargetManager_RemoveCollection_Call) RunAndReturn(run func(context.Context, int64)) *MockTargetManager_RemoveCollection_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1002,7 +1002,7 @@ func (_c *MockTargetManager_RemovePartition_Call) Return() *MockTargetManager_Re
}
func (_c *MockTargetManager_RemovePartition_Call) RunAndReturn(run func(context.Context, int64, ...int64)) *MockTargetManager_RemovePartition_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1036,7 +1036,7 @@ func (_c *MockTargetManager_SaveCurrentTarget_Call) Return() *MockTargetManager_
}
func (_c *MockTargetManager_SaveCurrentTarget_Call) RunAndReturn(run func(context.Context, metastore.QueryCoordCatalog)) *MockTargetManager_SaveCurrentTarget_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

10
internal/querycoordv2/session/mock_cluster.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package session
@ -446,7 +446,7 @@ func (_c *MockCluster_ReleaseSegments_Call) RunAndReturn(run func(context.Contex
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *MockCluster) Start() {
_m.Called()
}
@ -474,11 +474,11 @@ func (_c *MockCluster_Start_Call) Return() *MockCluster_Start_Call {
}
func (_c *MockCluster_Start_Call) RunAndReturn(run func()) *MockCluster_Start_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockCluster) Stop() {
_m.Called()
}
@ -506,7 +506,7 @@ func (_c *MockCluster_Stop_Call) Return() *MockCluster_Stop_Call {
}
func (_c *MockCluster_Stop_Call) RunAndReturn(run func()) *MockCluster_Stop_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

36
internal/querycoordv2/task/mock_scheduler.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package task
@ -17,9 +17,9 @@ func (_m *MockScheduler) EXPECT() *MockScheduler_Expecter {
return &MockScheduler_Expecter{mock: &_m.Mock}
}
// Add provides a mock function with given fields: task
func (_m *MockScheduler) Add(task Task) error {
ret := _m.Called(task)
// Add provides a mock function with given fields: _a0
func (_m *MockScheduler) Add(_a0 Task) error {
ret := _m.Called(_a0)
if len(ret) == 0 {
panic("no return value specified for Add")
@ -27,7 +27,7 @@ func (_m *MockScheduler) Add(task Task) error {
var r0 error
if rf, ok := ret.Get(0).(func(Task) error); ok {
r0 = rf(task)
r0 = rf(_a0)
} else {
r0 = ret.Error(0)
}
@ -41,12 +41,12 @@ type MockScheduler_Add_Call struct {
}
// Add is a helper method to define mock.On call
// - task Task
func (_e *MockScheduler_Expecter) Add(task interface{}) *MockScheduler_Add_Call {
return &MockScheduler_Add_Call{Call: _e.mock.On("Add", task)}
// - _a0 Task
func (_e *MockScheduler_Expecter) Add(_a0 interface{}) *MockScheduler_Add_Call {
return &MockScheduler_Add_Call{Call: _e.mock.On("Add", _a0)}
}
func (_c *MockScheduler_Add_Call) Run(run func(task Task)) *MockScheduler_Add_Call {
func (_c *MockScheduler_Add_Call) Run(run func(_a0 Task)) *MockScheduler_Add_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(Task))
})
@ -92,7 +92,7 @@ func (_c *MockScheduler_AddExecutor_Call) Return() *MockScheduler_AddExecutor_Ca
}
func (_c *MockScheduler_AddExecutor_Call) RunAndReturn(run func(int64)) *MockScheduler_AddExecutor_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -125,7 +125,7 @@ func (_c *MockScheduler_Dispatch_Call) Return() *MockScheduler_Dispatch_Call {
}
func (_c *MockScheduler_Dispatch_Call) RunAndReturn(run func(int64)) *MockScheduler_Dispatch_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -389,7 +389,7 @@ func (_c *MockScheduler_GetSegmentTaskNum_Call) RunAndReturn(run func(...TaskFil
return _c
}
// GetTasksJSON provides a mock function with given fields:
// GetTasksJSON provides a mock function with no fields
func (_m *MockScheduler) GetTasksJSON() string {
ret := _m.Called()
@ -463,7 +463,7 @@ func (_c *MockScheduler_RemoveByNode_Call) Return() *MockScheduler_RemoveByNode_
}
func (_c *MockScheduler_RemoveByNode_Call) RunAndReturn(run func(int64)) *MockScheduler_RemoveByNode_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -496,11 +496,11 @@ func (_c *MockScheduler_RemoveExecutor_Call) Return() *MockScheduler_RemoveExecu
}
func (_c *MockScheduler_RemoveExecutor_Call) RunAndReturn(run func(int64)) *MockScheduler_RemoveExecutor_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *MockScheduler) Start() {
_m.Called()
}
@ -528,11 +528,11 @@ func (_c *MockScheduler_Start_Call) Return() *MockScheduler_Start_Call {
}
func (_c *MockScheduler_Start_Call) RunAndReturn(run func()) *MockScheduler_Start_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockScheduler) Stop() {
_m.Called()
}
@ -560,7 +560,7 @@ func (_c *MockScheduler_Stop_Call) Return() *MockScheduler_Stop_Call {
}
func (_c *MockScheduler_Stop_Call) RunAndReturn(run func()) *MockScheduler_Stop_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

2
internal/querynodev2/cluster/mock_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package cluster

8
internal/querynodev2/cluster/mock_worker.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package cluster
@ -191,7 +191,7 @@ func (_c *MockWorker_GetStatistics_Call) RunAndReturn(run func(context.Context,
return _c
}
// IsHealthy provides a mock function with given fields:
// IsHealthy provides a mock function with no fields
func (_m *MockWorker) IsHealthy() bool {
ret := _m.Called()
@ -496,7 +496,7 @@ func (_c *MockWorker_SearchSegments_Call) RunAndReturn(run func(context.Context,
return _c
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *MockWorker) Stop() {
_m.Called()
}
@ -524,7 +524,7 @@ func (_c *MockWorker_Stop_Call) Return() *MockWorker_Stop_Call {
}
func (_c *MockWorker_Stop_Call) RunAndReturn(run func()) *MockWorker_Stop_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

38
internal/querynodev2/delegator/mock_delegator.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package delegator
@ -57,11 +57,11 @@ func (_c *MockShardDelegator_AddExcludedSegments_Call) Return() *MockShardDelega
}
func (_c *MockShardDelegator_AddExcludedSegments_Call) RunAndReturn(run func(map[int64]uint64)) *MockShardDelegator_AddExcludedSegments_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockShardDelegator) Close() {
_m.Called()
}
@ -89,11 +89,11 @@ func (_c *MockShardDelegator_Close_Call) Return() *MockShardDelegator_Close_Call
}
func (_c *MockShardDelegator_Close_Call) RunAndReturn(run func()) *MockShardDelegator_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Collection provides a mock function with given fields:
// Collection provides a mock function with no fields
func (_m *MockShardDelegator) Collection() int64 {
ret := _m.Called()
@ -138,7 +138,7 @@ func (_c *MockShardDelegator_Collection_Call) RunAndReturn(run func() int64) *Mo
return _c
}
// GetDeleteBufferSize provides a mock function with given fields:
// GetDeleteBufferSize provides a mock function with no fields
func (_m *MockShardDelegator) GetDeleteBufferSize() (int64, int64) {
ret := _m.Called()
@ -360,7 +360,7 @@ func (_c *MockShardDelegator_GetStatistics_Call) RunAndReturn(run func(context.C
return _c
}
// GetTSafe provides a mock function with given fields:
// GetTSafe provides a mock function with no fields
func (_m *MockShardDelegator) GetTSafe() uint64 {
ret := _m.Called()
@ -405,7 +405,7 @@ func (_c *MockShardDelegator_GetTSafe_Call) RunAndReturn(run func() uint64) *Moc
return _c
}
// GetTargetVersion provides a mock function with given fields:
// GetTargetVersion provides a mock function with no fields
func (_m *MockShardDelegator) GetTargetVersion() int64 {
ret := _m.Called()
@ -575,7 +575,7 @@ func (_c *MockShardDelegator_ProcessDelete_Call) Return() *MockShardDelegator_Pr
}
func (_c *MockShardDelegator_ProcessDelete_Call) RunAndReturn(run func([]*DeleteData, uint64)) *MockShardDelegator_ProcessDelete_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -608,7 +608,7 @@ func (_c *MockShardDelegator_ProcessInsert_Call) Return() *MockShardDelegator_Pr
}
func (_c *MockShardDelegator_ProcessInsert_Call) RunAndReturn(run func(map[int64]*InsertData)) *MockShardDelegator_ProcessInsert_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -826,7 +826,7 @@ func (_c *MockShardDelegator_Search_Call) RunAndReturn(run func(context.Context,
return _c
}
// Serviceable provides a mock function with given fields:
// Serviceable provides a mock function with no fields
func (_m *MockShardDelegator) Serviceable() bool {
ret := _m.Called()
@ -871,7 +871,7 @@ func (_c *MockShardDelegator_Serviceable_Call) RunAndReturn(run func() bool) *Mo
return _c
}
// Start provides a mock function with given fields:
// Start provides a mock function with no fields
func (_m *MockShardDelegator) Start() {
_m.Called()
}
@ -899,7 +899,7 @@ func (_c *MockShardDelegator_Start_Call) Return() *MockShardDelegator_Start_Call
}
func (_c *MockShardDelegator_Start_Call) RunAndReturn(run func()) *MockShardDelegator_Start_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -947,7 +947,7 @@ func (_c *MockShardDelegator_SyncDistribution_Call) Return() *MockShardDelegator
}
func (_c *MockShardDelegator_SyncDistribution_Call) RunAndReturn(run func(context.Context, ...SegmentEntry)) *MockShardDelegator_SyncDistribution_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -981,7 +981,7 @@ func (_c *MockShardDelegator_SyncPartitionStats_Call) Return() *MockShardDelegat
}
func (_c *MockShardDelegator_SyncPartitionStats_Call) RunAndReturn(run func(context.Context, map[int64]int64)) *MockShardDelegator_SyncPartitionStats_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1019,7 +1019,7 @@ func (_c *MockShardDelegator_SyncTargetVersion_Call) Return() *MockShardDelegato
}
func (_c *MockShardDelegator_SyncTargetVersion_Call) RunAndReturn(run func(int64, []int64, []int64, []int64, []int64, *msgpb.MsgPosition)) *MockShardDelegator_SyncTargetVersion_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1052,7 +1052,7 @@ func (_c *MockShardDelegator_TryCleanExcludedSegments_Call) Return() *MockShardD
}
func (_c *MockShardDelegator_TryCleanExcludedSegments_Call) RunAndReturn(run func(uint64)) *MockShardDelegator_TryCleanExcludedSegments_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1085,7 +1085,7 @@ func (_c *MockShardDelegator_UpdateTSafe_Call) Return() *MockShardDelegator_Upda
}
func (_c *MockShardDelegator_UpdateTSafe_Call) RunAndReturn(run func(uint64)) *MockShardDelegator_UpdateTSafe_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1136,7 +1136,7 @@ func (_c *MockShardDelegator_VerifyExcludedSegments_Call) RunAndReturn(run func(
return _c
}
// Version provides a mock function with given fields:
// Version provides a mock function with no fields
func (_m *MockShardDelegator) Version() int64 {
ret := _m.Called()

6
internal/querynodev2/segments/mock_collection_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package segments
@ -71,7 +71,7 @@ func (_c *MockCollectionManager_Get_Call) RunAndReturn(run func(int64) *Collecti
return _c
}
// List provides a mock function with given fields:
// List provides a mock function with no fields
func (_m *MockCollectionManager) List() []int64 {
ret := _m.Called()
@ -118,7 +118,7 @@ func (_c *MockCollectionManager_List_Call) RunAndReturn(run func() []int64) *Moc
return _c
}
// ListWithName provides a mock function with given fields:
// ListWithName provides a mock function with no fields
func (_m *MockCollectionManager) ListWithName() map[int64]string {
ret := _m.Called()

26
internal/querynodev2/segments/mock_loader.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package segments
@ -33,11 +33,11 @@ func (_m *MockLoader) EXPECT() *MockLoader_Expecter {
return &MockLoader_Expecter{mock: &_m.Mock}
}
// Load provides a mock function with given fields: ctx, collectionID, segmentType, version, segments
func (_m *MockLoader) Load(ctx context.Context, collectionID int64, segmentType commonpb.SegmentState, version int64, segments ...*querypb.SegmentLoadInfo) ([]Segment, error) {
_va := make([]interface{}, len(segments))
for _i := range segments {
_va[_i] = segments[_i]
// Load provides a mock function with given fields: ctx, collectionID, segmentType, version, _a4
func (_m *MockLoader) Load(ctx context.Context, collectionID int64, segmentType commonpb.SegmentState, version int64, _a4 ...*querypb.SegmentLoadInfo) ([]Segment, error) {
_va := make([]interface{}, len(_a4))
for _i := range _a4 {
_va[_i] = _a4[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx, collectionID, segmentType, version)
@ -51,10 +51,10 @@ func (_m *MockLoader) Load(ctx context.Context, collectionID int64, segmentType
var r0 []Segment
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, int64, commonpb.SegmentState, int64, ...*querypb.SegmentLoadInfo) ([]Segment, error)); ok {
return rf(ctx, collectionID, segmentType, version, segments...)
return rf(ctx, collectionID, segmentType, version, _a4...)
}
if rf, ok := ret.Get(0).(func(context.Context, int64, commonpb.SegmentState, int64, ...*querypb.SegmentLoadInfo) []Segment); ok {
r0 = rf(ctx, collectionID, segmentType, version, segments...)
r0 = rf(ctx, collectionID, segmentType, version, _a4...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]Segment)
@ -62,7 +62,7 @@ func (_m *MockLoader) Load(ctx context.Context, collectionID int64, segmentType
}
if rf, ok := ret.Get(1).(func(context.Context, int64, commonpb.SegmentState, int64, ...*querypb.SegmentLoadInfo) error); ok {
r1 = rf(ctx, collectionID, segmentType, version, segments...)
r1 = rf(ctx, collectionID, segmentType, version, _a4...)
} else {
r1 = ret.Error(1)
}
@ -80,13 +80,13 @@ type MockLoader_Load_Call struct {
// - collectionID int64
// - segmentType commonpb.SegmentState
// - version int64
// - segments ...*querypb.SegmentLoadInfo
func (_e *MockLoader_Expecter) Load(ctx interface{}, collectionID interface{}, segmentType interface{}, version interface{}, segments ...interface{}) *MockLoader_Load_Call {
// - _a4 ...*querypb.SegmentLoadInfo
func (_e *MockLoader_Expecter) Load(ctx interface{}, collectionID interface{}, segmentType interface{}, version interface{}, _a4 ...interface{}) *MockLoader_Load_Call {
return &MockLoader_Load_Call{Call: _e.mock.On("Load",
append([]interface{}{ctx, collectionID, segmentType, version}, segments...)...)}
append([]interface{}{ctx, collectionID, segmentType, version}, _a4...)...)}
}
func (_c *MockLoader_Load_Call) Run(run func(ctx context.Context, collectionID int64, segmentType commonpb.SegmentState, version int64, segments ...*querypb.SegmentLoadInfo)) *MockLoader_Load_Call {
func (_c *MockLoader_Load_Call) Run(run func(ctx context.Context, collectionID int64, segmentType commonpb.SegmentState, version int64, _a4 ...*querypb.SegmentLoadInfo)) *MockLoader_Load_Call {
_c.Call.Run(func(args mock.Arguments) {
variadicArgs := make([]*querypb.SegmentLoadInfo, len(args)-4)
for i, a := range args[4:] {

62
internal/querynodev2/segments/mock_segment.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package segments
@ -132,7 +132,7 @@ func (_c *MockSegment_CASVersion_Call) RunAndReturn(run func(int64, int64) bool)
return _c
}
// Collection provides a mock function with given fields:
// Collection provides a mock function with no fields
func (_m *MockSegment) Collection() int64 {
ret := _m.Called()
@ -177,7 +177,7 @@ func (_c *MockSegment_Collection_Call) RunAndReturn(run func() int64) *MockSegme
return _c
}
// DatabaseName provides a mock function with given fields:
// DatabaseName provides a mock function with no fields
func (_m *MockSegment) DatabaseName() string {
ret := _m.Called()
@ -316,7 +316,7 @@ func (_c *MockSegment_ExistIndex_Call) RunAndReturn(run func(int64) bool) *MockS
return _c
}
// GetBM25Stats provides a mock function with given fields:
// GetBM25Stats provides a mock function with no fields
func (_m *MockSegment) GetBM25Stats() map[int64]*storage.BM25Stats {
ret := _m.Called()
@ -363,7 +363,7 @@ func (_c *MockSegment_GetBM25Stats_Call) RunAndReturn(run func() map[int64]*stor
return _c
}
// GetFieldJSONIndexStats provides a mock function with given fields:
// GetFieldJSONIndexStats provides a mock function with no fields
func (_m *MockSegment) GetFieldJSONIndexStats() []int64 {
ret := _m.Called()
@ -552,7 +552,7 @@ func (_c *MockSegment_HasRawData_Call) RunAndReturn(run func(int64) bool) *MockS
return _c
}
// ID provides a mock function with given fields:
// ID provides a mock function with no fields
func (_m *MockSegment) ID() int64 {
ret := _m.Called()
@ -597,7 +597,7 @@ func (_c *MockSegment_ID_Call) RunAndReturn(run func() int64) *MockSegment_ID_Ca
return _c
}
// Indexes provides a mock function with given fields:
// Indexes provides a mock function with no fields
func (_m *MockSegment) Indexes() []*IndexedFieldInfo {
ret := _m.Called()
@ -693,7 +693,7 @@ func (_c *MockSegment_Insert_Call) RunAndReturn(run func(context.Context, []int6
return _c
}
// InsertCount provides a mock function with given fields:
// InsertCount provides a mock function with no fields
func (_m *MockSegment) InsertCount() int64 {
ret := _m.Called()
@ -738,7 +738,7 @@ func (_c *MockSegment_InsertCount_Call) RunAndReturn(run func() int64) *MockSegm
return _c
}
// IsLazyLoad provides a mock function with given fields:
// IsLazyLoad provides a mock function with no fields
func (_m *MockSegment) IsLazyLoad() bool {
ret := _m.Called()
@ -783,7 +783,7 @@ func (_c *MockSegment_IsLazyLoad_Call) RunAndReturn(run func() bool) *MockSegmen
return _c
}
// IsSorted provides a mock function with given fields:
// IsSorted provides a mock function with no fields
func (_m *MockSegment) IsSorted() bool {
ret := _m.Called()
@ -828,7 +828,7 @@ func (_c *MockSegment_IsSorted_Call) RunAndReturn(run func() bool) *MockSegment_
return _c
}
// LastDeltaTimestamp provides a mock function with given fields:
// LastDeltaTimestamp provides a mock function with no fields
func (_m *MockSegment) LastDeltaTimestamp() uint64 {
ret := _m.Called()
@ -873,7 +873,7 @@ func (_c *MockSegment_LastDeltaTimestamp_Call) RunAndReturn(run func() uint64) *
return _c
}
// Level provides a mock function with given fields:
// Level provides a mock function with no fields
func (_m *MockSegment) Level() datapb.SegmentLevel {
ret := _m.Called()
@ -965,7 +965,7 @@ func (_c *MockSegment_LoadDeltaData_Call) RunAndReturn(run func(context.Context,
return _c
}
// LoadInfo provides a mock function with given fields:
// LoadInfo provides a mock function with no fields
func (_m *MockSegment) LoadInfo() *querypb.SegmentLoadInfo {
ret := _m.Called()
@ -1058,7 +1058,7 @@ func (_c *MockSegment_MayPkExist_Call) RunAndReturn(run func(*storage.LocationsC
return _c
}
// MemSize provides a mock function with given fields:
// MemSize provides a mock function with no fields
func (_m *MockSegment) MemSize() int64 {
ret := _m.Called()
@ -1103,7 +1103,7 @@ func (_c *MockSegment_MemSize_Call) RunAndReturn(run func() int64) *MockSegment_
return _c
}
// NeedUpdatedVersion provides a mock function with given fields:
// NeedUpdatedVersion provides a mock function with no fields
func (_m *MockSegment) NeedUpdatedVersion() int64 {
ret := _m.Called()
@ -1148,7 +1148,7 @@ func (_c *MockSegment_NeedUpdatedVersion_Call) RunAndReturn(run func() int64) *M
return _c
}
// Partition provides a mock function with given fields:
// Partition provides a mock function with no fields
func (_m *MockSegment) Partition() int64 {
ret := _m.Called()
@ -1193,7 +1193,7 @@ func (_c *MockSegment_Partition_Call) RunAndReturn(run func() int64) *MockSegmen
return _c
}
// PinIfNotReleased provides a mock function with given fields:
// PinIfNotReleased provides a mock function with no fields
func (_m *MockSegment) PinIfNotReleased() error {
ret := _m.Called()
@ -1282,11 +1282,11 @@ func (_c *MockSegment_Release_Call) Return() *MockSegment_Release_Call {
}
func (_c *MockSegment_Release_Call) RunAndReturn(run func(context.Context, ...releaseOption)) *MockSegment_Release_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// RemoveUnusedFieldFiles provides a mock function with given fields:
// RemoveUnusedFieldFiles provides a mock function with no fields
func (_m *MockSegment) RemoveUnusedFieldFiles() error {
ret := _m.Called()
@ -1360,11 +1360,11 @@ func (_c *MockSegment_ResetIndexesLazyLoad_Call) Return() *MockSegment_ResetInde
}
func (_c *MockSegment_ResetIndexesLazyLoad_Call) RunAndReturn(run func(bool)) *MockSegment_ResetIndexesLazyLoad_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// ResourceGroup provides a mock function with given fields:
// ResourceGroup provides a mock function with no fields
func (_m *MockSegment) ResourceGroup() string {
ret := _m.Called()
@ -1409,7 +1409,7 @@ func (_c *MockSegment_ResourceGroup_Call) RunAndReturn(run func() string) *MockS
return _c
}
// ResourceUsageEstimate provides a mock function with given fields:
// ResourceUsageEstimate provides a mock function with no fields
func (_m *MockSegment) ResourceUsageEstimate() ResourceUsage {
ret := _m.Called()
@ -1572,7 +1572,7 @@ func (_c *MockSegment_RetrieveByOffsets_Call) RunAndReturn(run func(context.Cont
return _c
}
// RowNum provides a mock function with given fields:
// RowNum provides a mock function with no fields
func (_m *MockSegment) RowNum() int64 {
ret := _m.Called()
@ -1676,7 +1676,7 @@ func (_c *MockSegment_Search_Call) RunAndReturn(run func(context.Context, *segco
return _c
}
// Shard provides a mock function with given fields:
// Shard provides a mock function with no fields
func (_m *MockSegment) Shard() metautil.Channel {
ret := _m.Called()
@ -1721,7 +1721,7 @@ func (_c *MockSegment_Shard_Call) RunAndReturn(run func() metautil.Channel) *Moc
return _c
}
// StartPosition provides a mock function with given fields:
// StartPosition provides a mock function with no fields
func (_m *MockSegment) StartPosition() *msgpb.MsgPosition {
ret := _m.Called()
@ -1768,7 +1768,7 @@ func (_c *MockSegment_StartPosition_Call) RunAndReturn(run func() *msgpb.MsgPosi
return _c
}
// Type provides a mock function with given fields:
// Type provides a mock function with no fields
func (_m *MockSegment) Type() commonpb.SegmentState {
ret := _m.Called()
@ -1813,7 +1813,7 @@ func (_c *MockSegment_Type_Call) RunAndReturn(run func() commonpb.SegmentState)
return _c
}
// Unpin provides a mock function with given fields:
// Unpin provides a mock function with no fields
func (_m *MockSegment) Unpin() {
_m.Called()
}
@ -1841,7 +1841,7 @@ func (_c *MockSegment_Unpin_Call) Return() *MockSegment_Unpin_Call {
}
func (_c *MockSegment_Unpin_Call) RunAndReturn(run func()) *MockSegment_Unpin_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1874,7 +1874,7 @@ func (_c *MockSegment_UpdateBM25Stats_Call) Return() *MockSegment_UpdateBM25Stat
}
func (_c *MockSegment_UpdateBM25Stats_Call) RunAndReturn(run func(map[int64]*storage.BM25Stats)) *MockSegment_UpdateBM25Stats_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -1907,11 +1907,11 @@ func (_c *MockSegment_UpdateBloomFilter_Call) Return() *MockSegment_UpdateBloomF
}
func (_c *MockSegment_UpdateBloomFilter_Call) RunAndReturn(run func([]storage.PrimaryKey)) *MockSegment_UpdateBloomFilter_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Version provides a mock function with given fields:
// Version provides a mock function with no fields
func (_m *MockSegment) Version() int64 {
ret := _m.Called()

62
internal/querynodev2/segments/mock_segment_manager.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package segments
@ -54,11 +54,11 @@ func (_c *MockSegmentManager_Clear_Call) Return() *MockSegmentManager_Clear_Call
}
func (_c *MockSegmentManager_Clear_Call) RunAndReturn(run func(context.Context)) *MockSegmentManager_Clear_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
// Empty provides a mock function with given fields:
// Empty provides a mock function with no fields
func (_m *MockSegmentManager) Empty() bool {
ret := _m.Called()
@ -198,14 +198,14 @@ func (_c *MockSegmentManager_Get_Call) RunAndReturn(run func(int64) Segment) *Mo
return _c
}
// GetAndPin provides a mock function with given fields: segments, filters
func (_m *MockSegmentManager) GetAndPin(segments []int64, filters ...SegmentFilter) ([]Segment, error) {
// GetAndPin provides a mock function with given fields: _a0, filters
func (_m *MockSegmentManager) GetAndPin(_a0 []int64, filters ...SegmentFilter) ([]Segment, error) {
_va := make([]interface{}, len(filters))
for _i := range filters {
_va[_i] = filters[_i]
}
var _ca []interface{}
_ca = append(_ca, segments)
_ca = append(_ca, _a0)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
@ -216,10 +216,10 @@ func (_m *MockSegmentManager) GetAndPin(segments []int64, filters ...SegmentFilt
var r0 []Segment
var r1 error
if rf, ok := ret.Get(0).(func([]int64, ...SegmentFilter) ([]Segment, error)); ok {
return rf(segments, filters...)
return rf(_a0, filters...)
}
if rf, ok := ret.Get(0).(func([]int64, ...SegmentFilter) []Segment); ok {
r0 = rf(segments, filters...)
r0 = rf(_a0, filters...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]Segment)
@ -227,7 +227,7 @@ func (_m *MockSegmentManager) GetAndPin(segments []int64, filters ...SegmentFilt
}
if rf, ok := ret.Get(1).(func([]int64, ...SegmentFilter) error); ok {
r1 = rf(segments, filters...)
r1 = rf(_a0, filters...)
} else {
r1 = ret.Error(1)
}
@ -241,14 +241,14 @@ type MockSegmentManager_GetAndPin_Call struct {
}
// GetAndPin is a helper method to define mock.On call
// - segments []int64
// - _a0 []int64
// - filters ...SegmentFilter
func (_e *MockSegmentManager_Expecter) GetAndPin(segments interface{}, filters ...interface{}) *MockSegmentManager_GetAndPin_Call {
func (_e *MockSegmentManager_Expecter) GetAndPin(_a0 interface{}, filters ...interface{}) *MockSegmentManager_GetAndPin_Call {
return &MockSegmentManager_GetAndPin_Call{Call: _e.mock.On("GetAndPin",
append([]interface{}{segments}, filters...)...)}
append([]interface{}{_a0}, filters...)...)}
}
func (_c *MockSegmentManager_GetAndPin_Call) Run(run func(segments []int64, filters ...SegmentFilter)) *MockSegmentManager_GetAndPin_Call {
func (_c *MockSegmentManager_GetAndPin_Call) Run(run func(_a0 []int64, filters ...SegmentFilter)) *MockSegmentManager_GetAndPin_Call {
_c.Call.Run(func(args mock.Arguments) {
variadicArgs := make([]SegmentFilter, len(args)-1)
for i, a := range args[1:] {
@ -548,11 +548,11 @@ func (_c *MockSegmentManager_GetWithType_Call) RunAndReturn(run func(int64, comm
return _c
}
// Put provides a mock function with given fields: ctx, segmentType, segments
func (_m *MockSegmentManager) Put(ctx context.Context, segmentType commonpb.SegmentState, segments ...Segment) {
_va := make([]interface{}, len(segments))
for _i := range segments {
_va[_i] = segments[_i]
// Put provides a mock function with given fields: ctx, segmentType, _a2
func (_m *MockSegmentManager) Put(ctx context.Context, segmentType commonpb.SegmentState, _a2 ...Segment) {
_va := make([]interface{}, len(_a2))
for _i := range _a2 {
_va[_i] = _a2[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx, segmentType)
@ -568,13 +568,13 @@ type MockSegmentManager_Put_Call struct {
// Put is a helper method to define mock.On call
// - ctx context.Context
// - segmentType commonpb.SegmentState
// - segments ...Segment
func (_e *MockSegmentManager_Expecter) Put(ctx interface{}, segmentType interface{}, segments ...interface{}) *MockSegmentManager_Put_Call {
// - _a2 ...Segment
func (_e *MockSegmentManager_Expecter) Put(ctx interface{}, segmentType interface{}, _a2 ...interface{}) *MockSegmentManager_Put_Call {
return &MockSegmentManager_Put_Call{Call: _e.mock.On("Put",
append([]interface{}{ctx, segmentType}, segments...)...)}
append([]interface{}{ctx, segmentType}, _a2...)...)}
}
func (_c *MockSegmentManager_Put_Call) Run(run func(ctx context.Context, segmentType commonpb.SegmentState, segments ...Segment)) *MockSegmentManager_Put_Call {
func (_c *MockSegmentManager_Put_Call) Run(run func(ctx context.Context, segmentType commonpb.SegmentState, _a2 ...Segment)) *MockSegmentManager_Put_Call {
_c.Call.Run(func(args mock.Arguments) {
variadicArgs := make([]Segment, len(args)-2)
for i, a := range args[2:] {
@ -593,7 +593,7 @@ func (_c *MockSegmentManager_Put_Call) Return() *MockSegmentManager_Put_Call {
}
func (_c *MockSegmentManager_Put_Call) RunAndReturn(run func(context.Context, commonpb.SegmentState, ...Segment)) *MockSegmentManager_Put_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -726,9 +726,9 @@ func (_c *MockSegmentManager_RemoveBy_Call) RunAndReturn(run func(context.Contex
return _c
}
// Unpin provides a mock function with given fields: segments
func (_m *MockSegmentManager) Unpin(segments []Segment) {
_m.Called(segments)
// Unpin provides a mock function with given fields: _a0
func (_m *MockSegmentManager) Unpin(_a0 []Segment) {
_m.Called(_a0)
}
// MockSegmentManager_Unpin_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Unpin'
@ -737,12 +737,12 @@ type MockSegmentManager_Unpin_Call struct {
}
// Unpin is a helper method to define mock.On call
// - segments []Segment
func (_e *MockSegmentManager_Expecter) Unpin(segments interface{}) *MockSegmentManager_Unpin_Call {
return &MockSegmentManager_Unpin_Call{Call: _e.mock.On("Unpin", segments)}
// - _a0 []Segment
func (_e *MockSegmentManager_Expecter) Unpin(_a0 interface{}) *MockSegmentManager_Unpin_Call {
return &MockSegmentManager_Unpin_Call{Call: _e.mock.On("Unpin", _a0)}
}
func (_c *MockSegmentManager_Unpin_Call) Run(run func(segments []Segment)) *MockSegmentManager_Unpin_Call {
func (_c *MockSegmentManager_Unpin_Call) Run(run func(_a0 []Segment)) *MockSegmentManager_Unpin_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].([]Segment))
})
@ -755,7 +755,7 @@ func (_c *MockSegmentManager_Unpin_Call) Return() *MockSegmentManager_Unpin_Call
}
func (_c *MockSegmentManager_Unpin_Call) RunAndReturn(run func([]Segment)) *MockSegmentManager_Unpin_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

10
internal/rootcoord/mocks/garbage_collector.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mockrootcoord
@ -168,7 +168,7 @@ func (_c *GarbageCollector_ReDropCollection_Call) Return() *GarbageCollector_ReD
}
func (_c *GarbageCollector_ReDropCollection_Call) RunAndReturn(run func(*model.Collection, uint64)) *GarbageCollector_ReDropCollection_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -205,7 +205,7 @@ func (_c *GarbageCollector_ReDropPartition_Call) Return() *GarbageCollector_ReDr
}
func (_c *GarbageCollector_ReDropPartition_Call) RunAndReturn(run func(int64, []string, []string, *model.Partition, uint64)) *GarbageCollector_ReDropPartition_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -238,7 +238,7 @@ func (_c *GarbageCollector_RemoveCreatingCollection_Call) Return() *GarbageColle
}
func (_c *GarbageCollector_RemoveCreatingCollection_Call) RunAndReturn(run func(*model.Collection)) *GarbageCollector_RemoveCreatingCollection_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -273,7 +273,7 @@ func (_c *GarbageCollector_RemoveCreatingPartition_Call) Return() *GarbageCollec
}
func (_c *GarbageCollector_RemoveCreatingPartition_Call) RunAndReturn(run func(int64, *model.Partition, uint64)) *GarbageCollector_RemoveCreatingPartition_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

2
internal/rootcoord/mocks/meta_table.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mockrootcoord

8
pkg/eventlog/mock_logger.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package eventlog
@ -17,7 +17,7 @@ func (_m *MockLogger) EXPECT() *MockLogger_Expecter {
return &MockLogger_Expecter{mock: &_m.Mock}
}
// Flush provides a mock function with given fields:
// Flush provides a mock function with no fields
func (_m *MockLogger) Flush() error {
ret := _m.Called()
@ -91,7 +91,7 @@ func (_c *MockLogger_Record_Call) Return() *MockLogger_Record_Call {
}
func (_c *MockLogger_Record_Call) RunAndReturn(run func(Evt)) *MockLogger_Record_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}
@ -125,7 +125,7 @@ func (_c *MockLogger_RecordFunc_Call) Return() *MockLogger_RecordFunc_Call {
}
func (_c *MockLogger_RecordFunc_Call) RunAndReturn(run func(Level, func() Evt)) *MockLogger_RecordFunc_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

10
pkg/kv/predicates/mock_predicate.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package predicates
@ -63,7 +63,7 @@ func (_c *MockPredicate_IsTrue_Call) RunAndReturn(run func(interface{}) bool) *M
return _c
}
// Key provides a mock function with given fields:
// Key provides a mock function with no fields
func (_m *MockPredicate) Key() string {
ret := _m.Called()
@ -108,7 +108,7 @@ func (_c *MockPredicate_Key_Call) RunAndReturn(run func() string) *MockPredicate
return _c
}
// Target provides a mock function with given fields:
// Target provides a mock function with no fields
func (_m *MockPredicate) Target() PredicateTarget {
ret := _m.Called()
@ -153,7 +153,7 @@ func (_c *MockPredicate_Target_Call) RunAndReturn(run func() PredicateTarget) *M
return _c
}
// TargetValue provides a mock function with given fields:
// TargetValue provides a mock function with no fields
func (_m *MockPredicate) TargetValue() interface{} {
ret := _m.Called()
@ -200,7 +200,7 @@ func (_c *MockPredicate_TargetValue_Call) RunAndReturn(run func() interface{}) *
return _c
}
// Type provides a mock function with given fields:
// Type provides a mock function with no fields
func (_m *MockPredicate) Type() PredicateType {
ret := _m.Called()

6
pkg/mocks/mock_kv/mock_MetaKv.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_kv
@ -23,7 +23,7 @@ func (_m *MockMetaKv) EXPECT() *MockMetaKv_Expecter {
return &MockMetaKv_Expecter{mock: &_m.Mock}
}
// Close provides a mock function with given fields:
// Close provides a mock function with no fields
func (_m *MockMetaKv) Close() {
_m.Called()
}
@ -51,7 +51,7 @@ func (_c *MockMetaKv_Close_Call) Return() *MockMetaKv_Close_Call {
}
func (_c *MockMetaKv_Close_Call) RunAndReturn(run func()) *MockMetaKv_Close_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

2
pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_streamingpb

12
pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_streamingpb
@ -24,7 +24,7 @@ func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) EXPECT()
return &MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Expecter{mock: &_m.Mock}
}
// CloseSend provides a mock function with given fields:
// CloseSend provides a mock function with no fields
func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) CloseSend() error {
ret := _m.Called()
@ -69,7 +69,7 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_CloseSend
return _c
}
// Context provides a mock function with given fields:
// Context provides a mock function with no fields
func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Context() context.Context {
ret := _m.Called()
@ -116,7 +116,7 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Context_C
return _c
}
// Header provides a mock function with given fields:
// Header provides a mock function with no fields
func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Header() (metadata.MD, error) {
ret := _m.Called()
@ -173,7 +173,7 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Header_Ca
return _c
}
// Recv provides a mock function with given fields:
// Recv provides a mock function with no fields
func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Recv() (*streamingpb.AssignmentDiscoverResponse, error) {
ret := _m.Called()
@ -368,7 +368,7 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_SendMsg_C
return _c
}
// Trailer provides a mock function with given fields:
// Trailer provides a mock function with no fields
func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Trailer() metadata.MD {
ret := _m.Called()

8
pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_streamingpb
@ -24,7 +24,7 @@ func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) EXPECT()
return &MockStreamingCoordAssignmentService_AssignmentDiscoverServer_Expecter{mock: &_m.Mock}
}
// Context provides a mock function with given fields:
// Context provides a mock function with no fields
func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) Context() context.Context {
ret := _m.Called()
@ -71,7 +71,7 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_Context_C
return _c
}
// Recv provides a mock function with given fields:
// Recv provides a mock function with no fields
func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) Recv() (*streamingpb.AssignmentDiscoverRequest, error) {
ret := _m.Called()
@ -387,7 +387,7 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_SetTraile
}
func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_SetTrailer_Call) RunAndReturn(run func(metadata.MD)) *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_SetTrailer_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

2
pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordBroadcastServiceClient.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_streamingpb

2
pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_streamingpb

12
pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_streamingpb
@ -24,7 +24,7 @@ func (_m *MockStreamingNodeHandlerService_ConsumeClient) EXPECT() *MockStreaming
return &MockStreamingNodeHandlerService_ConsumeClient_Expecter{mock: &_m.Mock}
}
// CloseSend provides a mock function with given fields:
// CloseSend provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ConsumeClient) CloseSend() error {
ret := _m.Called()
@ -69,7 +69,7 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_CloseSend_Call) RunAndRe
return _c
}
// Context provides a mock function with given fields:
// Context provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ConsumeClient) Context() context.Context {
ret := _m.Called()
@ -116,7 +116,7 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_Context_Call) RunAndRetu
return _c
}
// Header provides a mock function with given fields:
// Header provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ConsumeClient) Header() (metadata.MD, error) {
ret := _m.Called()
@ -173,7 +173,7 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_Header_Call) RunAndRetur
return _c
}
// Recv provides a mock function with given fields:
// Recv provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ConsumeClient) Recv() (*streamingpb.ConsumeResponse, error) {
ret := _m.Called()
@ -368,7 +368,7 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_SendMsg_Call) RunAndRetu
return _c
}
// Trailer provides a mock function with given fields:
// Trailer provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ConsumeClient) Trailer() metadata.MD {
ret := _m.Called()

8
pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_streamingpb
@ -24,7 +24,7 @@ func (_m *MockStreamingNodeHandlerService_ConsumeServer) EXPECT() *MockStreaming
return &MockStreamingNodeHandlerService_ConsumeServer_Expecter{mock: &_m.Mock}
}
// Context provides a mock function with given fields:
// Context provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ConsumeServer) Context() context.Context {
ret := _m.Called()
@ -71,7 +71,7 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_Context_Call) RunAndRetu
return _c
}
// Recv provides a mock function with given fields:
// Recv provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ConsumeServer) Recv() (*streamingpb.ConsumeRequest, error) {
ret := _m.Called()
@ -387,7 +387,7 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_SetTrailer_Call) Return(
}
func (_c *MockStreamingNodeHandlerService_ConsumeServer_SetTrailer_Call) RunAndReturn(run func(metadata.MD)) *MockStreamingNodeHandlerService_ConsumeServer_SetTrailer_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

12
pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_streamingpb
@ -24,7 +24,7 @@ func (_m *MockStreamingNodeHandlerService_ProduceClient) EXPECT() *MockStreaming
return &MockStreamingNodeHandlerService_ProduceClient_Expecter{mock: &_m.Mock}
}
// CloseSend provides a mock function with given fields:
// CloseSend provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ProduceClient) CloseSend() error {
ret := _m.Called()
@ -69,7 +69,7 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_CloseSend_Call) RunAndRe
return _c
}
// Context provides a mock function with given fields:
// Context provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ProduceClient) Context() context.Context {
ret := _m.Called()
@ -116,7 +116,7 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_Context_Call) RunAndRetu
return _c
}
// Header provides a mock function with given fields:
// Header provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ProduceClient) Header() (metadata.MD, error) {
ret := _m.Called()
@ -173,7 +173,7 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_Header_Call) RunAndRetur
return _c
}
// Recv provides a mock function with given fields:
// Recv provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ProduceClient) Recv() (*streamingpb.ProduceResponse, error) {
ret := _m.Called()
@ -368,7 +368,7 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_SendMsg_Call) RunAndRetu
return _c
}
// Trailer provides a mock function with given fields:
// Trailer provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ProduceClient) Trailer() metadata.MD {
ret := _m.Called()

8
pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.46.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mock_streamingpb
@ -24,7 +24,7 @@ func (_m *MockStreamingNodeHandlerService_ProduceServer) EXPECT() *MockStreaming
return &MockStreamingNodeHandlerService_ProduceServer_Expecter{mock: &_m.Mock}
}
// Context provides a mock function with given fields:
// Context provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ProduceServer) Context() context.Context {
ret := _m.Called()
@ -71,7 +71,7 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_Context_Call) RunAndRetu
return _c
}
// Recv provides a mock function with given fields:
// Recv provides a mock function with no fields
func (_m *MockStreamingNodeHandlerService_ProduceServer) Recv() (*streamingpb.ProduceRequest, error) {
ret := _m.Called()
@ -387,7 +387,7 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_SetTrailer_Call) Return(
}
func (_c *MockStreamingNodeHandlerService_ProduceServer_SetTrailer_Call) RunAndReturn(run func(metadata.MD)) *MockStreamingNodeHandlerService_ProduceServer_SetTrailer_Call {
_c.Call.Return(run)
_c.Run(run)
return _c
}

Some files were not shown because too many files have changed in this diff Show More