fix: fix `make generate-mockery` panic with go1.22 (#36830)

https://github.com/milvus-io/milvus/issues/36831
Fix `make generate-mockery` panic.

Signed-off-by: sunby <sunbingyi1992@gmail.com>

Makefile

@@ -49,7 +49,7 @@ GOLANGCI_LINT_VERSION := 1.55.2
 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.32.4
+MOCKERY_VERSION := 2.46.0
 MOCKERY_OUTPUT := $(shell $(INSTALL_PATH)/mockery --version 2>/dev/null)
 INSTALL_MOCKERY := $(findstring $(MOCKERY_VERSION),$(MOCKERY_OUTPUT))
 # gci

internal/datacoord/allocator/mock_allocator.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package allocator
 
@@ -25,6 +25,10 @@ func (_m *MockAllocator) EXPECT() *MockAllocator_Expecter {
 func (_m *MockAllocator) AllocID(_a0 context.Context) (int64, error) {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocID")
+	}
+
 	var r0 int64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (int64, error)); ok {
@@ -77,6 +81,10 @@ func (_c *MockAllocator_AllocID_Call) RunAndReturn(run func(context.Context) (in
 func (_m *MockAllocator) AllocN(n int64) (int64, int64, error) {
 	ret := _m.Called(n)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocN")
+	}
+
 	var r0 int64
 	var r1 int64
 	var r2 error
@@ -136,6 +144,10 @@ func (_c *MockAllocator_AllocN_Call) RunAndReturn(run func(int64) (int64, int64,
 func (_m *MockAllocator) AllocTimestamp(_a0 context.Context) (uint64, error) {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocTimestamp")
+	}
+
 	var r0 uint64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (uint64, error)); ok {

internal/datacoord/broker/mock_coordinator_broker.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package broker
 
@@ -26,6 +26,10 @@ func (_m *MockBroker) EXPECT() *MockBroker_Expecter {
 func (_m *MockBroker) DescribeCollectionInternal(ctx context.Context, collectionID int64) (*milvuspb.DescribeCollectionResponse, error) {
 	ret := _m.Called(ctx, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeCollectionInternal")
+	}
+
 	var r0 *milvuspb.DescribeCollectionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) (*milvuspb.DescribeCollectionResponse, error)); ok {
@@ -81,6 +85,10 @@ func (_c *MockBroker_DescribeCollectionInternal_Call) RunAndReturn(run func(cont
 func (_m *MockBroker) HasCollection(ctx context.Context, collectionID int64) (bool, error) {
 	ret := _m.Called(ctx, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for HasCollection")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) (bool, error)); ok {
@@ -134,6 +142,10 @@ func (_c *MockBroker_HasCollection_Call) RunAndReturn(run func(context.Context,
 func (_m *MockBroker) ListDatabases(ctx context.Context) (*milvuspb.ListDatabasesResponse, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListDatabases")
+	}
+
 	var r0 *milvuspb.ListDatabasesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (*milvuspb.ListDatabasesResponse, error)); ok {
@@ -188,6 +200,10 @@ func (_c *MockBroker_ListDatabases_Call) RunAndReturn(run func(context.Context)
 func (_m *MockBroker) ShowCollections(ctx context.Context, dbName string) (*milvuspb.ShowCollectionsResponse, error) {
 	ret := _m.Called(ctx, dbName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowCollections")
+	}
+
 	var r0 *milvuspb.ShowCollectionsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*milvuspb.ShowCollectionsResponse, error)); ok {
@@ -243,6 +259,10 @@ func (_c *MockBroker_ShowCollections_Call) RunAndReturn(run func(context.Context
 func (_m *MockBroker) ShowPartitionsInternal(ctx context.Context, collectionID int64) ([]int64, error) {
 	ret := _m.Called(ctx, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowPartitionsInternal")
+	}
+
 	var r0 []int64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) ([]int64, error)); ok {

internal/datacoord/mock_channel_store.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package datacoord
 
@@ -54,6 +54,10 @@ func (_c *MockRWChannelStore_AddNode_Call) RunAndReturn(run func(int64)) *MockRW
 func (_m *MockRWChannelStore) GetBufferChannelInfo() *NodeChannelInfo {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetBufferChannelInfo")
+	}
+
 	var r0 *NodeChannelInfo
 	if rf, ok := ret.Get(0).(func() *NodeChannelInfo); ok {
 		r0 = rf()
@@ -97,6 +101,10 @@ func (_c *MockRWChannelStore_GetBufferChannelInfo_Call) RunAndReturn(run func()
 func (_m *MockRWChannelStore) GetNode(nodeID int64) *NodeChannelInfo {
 	ret := _m.Called(nodeID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetNode")
+	}
+
 	var r0 *NodeChannelInfo
 	if rf, ok := ret.Get(0).(func(int64) *NodeChannelInfo); ok {
 		r0 = rf(nodeID)
@@ -148,6 +156,10 @@ func (_m *MockRWChannelStore) GetNodeChannelsBy(nodeSelector NodeSelector, chann
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetNodeChannelsBy")
+	}
+
 	var r0 []*NodeChannelInfo
 	if rf, ok := ret.Get(0).(func(NodeSelector, ...ChannelSelector) []*NodeChannelInfo); ok {
 		r0 = rf(nodeSelector, channelSelectors...)
@@ -200,6 +212,10 @@ func (_c *MockRWChannelStore_GetNodeChannelsBy_Call) RunAndReturn(run func(NodeS
 func (_m *MockRWChannelStore) GetNodeChannelsByCollectionID(collectionID int64) map[int64][]string {
 	ret := _m.Called(collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetNodeChannelsByCollectionID")
+	}
+
 	var r0 map[int64][]string
 	if rf, ok := ret.Get(0).(func(int64) map[int64][]string); ok {
 		r0 = rf(collectionID)
@@ -244,6 +260,10 @@ func (_c *MockRWChannelStore_GetNodeChannelsByCollectionID_Call) RunAndReturn(ru
 func (_m *MockRWChannelStore) GetNodes() []int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetNodes")
+	}
+
 	var r0 []int64
 	if rf, ok := ret.Get(0).(func() []int64); ok {
 		r0 = rf()
@@ -287,6 +307,10 @@ func (_c *MockRWChannelStore_GetNodes_Call) RunAndReturn(run func() []int64) *Mo
 func (_m *MockRWChannelStore) GetNodesChannels() []*NodeChannelInfo {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetNodesChannels")
+	}
+
 	var r0 []*NodeChannelInfo
 	if rf, ok := ret.Get(0).(func() []*NodeChannelInfo); ok {
 		r0 = rf()
@@ -330,6 +354,10 @@ func (_c *MockRWChannelStore_GetNodesChannels_Call) RunAndReturn(run func() []*N
 func (_m *MockRWChannelStore) Reload() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Reload")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -450,6 +478,10 @@ func (_c *MockRWChannelStore_SetLegacyChannelByNode_Call) RunAndReturn(run func(
 func (_m *MockRWChannelStore) Update(op *ChannelOpSet) error {
 	ret := _m.Called(op)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Update")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*ChannelOpSet) error); ok {
 		r0 = rf(op)

internal/datacoord/mock_channelmanager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package datacoord
 
@@ -25,6 +25,10 @@ func (_m *MockChannelManager) EXPECT() *MockChannelManager_Expecter {
 func (_m *MockChannelManager) AddNode(nodeID int64) error {
 	ret := _m.Called(nodeID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AddNode")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64) error); ok {
 		r0 = rf(nodeID)
@@ -99,6 +103,10 @@ func (_c *MockChannelManager_Close_Call) RunAndReturn(run func()) *MockChannelMa
 func (_m *MockChannelManager) DeleteNode(nodeID int64) error {
 	ret := _m.Called(nodeID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteNode")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64) error); ok {
 		r0 = rf(nodeID)
@@ -141,6 +149,10 @@ func (_c *MockChannelManager_DeleteNode_Call) RunAndReturn(run func(int64) error
 func (_m *MockChannelManager) FindWatcher(channel string) (int64, error) {
 	ret := _m.Called(channel)
 
+	if len(ret) == 0 {
+		panic("no return value specified for FindWatcher")
+	}
+
 	var r0 int64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (int64, error)); ok {
@@ -193,6 +205,10 @@ func (_c *MockChannelManager_FindWatcher_Call) RunAndReturn(run func(string) (in
 func (_m *MockChannelManager) GetChannel(nodeID int64, channel string) (RWChannel, bool) {
 	ret := _m.Called(nodeID, channel)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetChannel")
+	}
+
 	var r0 RWChannel
 	var r1 bool
 	if rf, ok := ret.Get(0).(func(int64, string) (RWChannel, bool)); ok {
@@ -248,6 +264,10 @@ func (_c *MockChannelManager_GetChannel_Call) RunAndReturn(run func(int64, strin
 func (_m *MockChannelManager) GetChannelNamesByCollectionID(collectionID int64) []string {
 	ret := _m.Called(collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetChannelNamesByCollectionID")
+	}
+
 	var r0 []string
 	if rf, ok := ret.Get(0).(func(int64) []string); ok {
 		r0 = rf(collectionID)
@@ -292,6 +312,10 @@ func (_c *MockChannelManager_GetChannelNamesByCollectionID_Call) RunAndReturn(ru
 func (_m *MockChannelManager) GetChannelsByCollectionID(collectionID int64) []RWChannel {
 	ret := _m.Called(collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetChannelsByCollectionID")
+	}
+
 	var r0 []RWChannel
 	if rf, ok := ret.Get(0).(func(int64) []RWChannel); ok {
 		r0 = rf(collectionID)
@@ -336,6 +360,10 @@ func (_c *MockChannelManager_GetChannelsByCollectionID_Call) RunAndReturn(run fu
 func (_m *MockChannelManager) GetNodeChannelsByCollectionID(collectionID int64) map[int64][]string {
 	ret := _m.Called(collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetNodeChannelsByCollectionID")
+	}
+
 	var r0 map[int64][]string
 	if rf, ok := ret.Get(0).(func(int64) map[int64][]string); ok {
 		r0 = rf(collectionID)
@@ -380,6 +408,10 @@ func (_c *MockChannelManager_GetNodeChannelsByCollectionID_Call) RunAndReturn(ru
 func (_m *MockChannelManager) Match(nodeID int64, channel string) bool {
 	ret := _m.Called(nodeID, channel)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Match")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(int64, string) bool); ok {
 		r0 = rf(nodeID, channel)
@@ -423,6 +455,10 @@ func (_c *MockChannelManager_Match_Call) RunAndReturn(run func(int64, string) bo
 func (_m *MockChannelManager) Release(nodeID int64, channelName string) error {
 	ret := _m.Called(nodeID, channelName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Release")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64, string) error); ok {
 		r0 = rf(nodeID, channelName)
@@ -466,6 +502,10 @@ func (_c *MockChannelManager_Release_Call) RunAndReturn(run func(int64, string)
 func (_m *MockChannelManager) Startup(ctx context.Context, legacyNodes []int64, allNodes []int64) error {
 	ret := _m.Called(ctx, legacyNodes, allNodes)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Startup")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, []int64, []int64) error); ok {
 		r0 = rf(ctx, legacyNodes, allNodes)
@@ -510,6 +550,10 @@ func (_c *MockChannelManager_Startup_Call) RunAndReturn(run func(context.Context
 func (_m *MockChannelManager) Watch(ctx context.Context, ch RWChannel) error {
 	ret := _m.Called(ctx, ch)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Watch")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, RWChannel) error); ok {
 		r0 = rf(ctx, ch)

internal/datacoord/mock_cluster.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package datacoord
 
@@ -60,6 +60,10 @@ func (_c *MockCluster_Close_Call) RunAndReturn(run func()) *MockCluster_Close_Ca
 func (_m *MockCluster) DropImport(nodeID int64, in *datapb.DropImportRequest) error {
 	ret := _m.Called(nodeID, in)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropImport")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64, *datapb.DropImportRequest) error); ok {
 		r0 = rf(nodeID, in)
@@ -103,6 +107,10 @@ func (_c *MockCluster_DropImport_Call) RunAndReturn(run func(int64, *datapb.Drop
 func (_m *MockCluster) Flush(ctx context.Context, nodeID int64, channel string, segments []*datapb.SegmentInfo) error {
 	ret := _m.Called(ctx, nodeID, channel, segments)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Flush")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, string, []*datapb.SegmentInfo) error); ok {
 		r0 = rf(ctx, nodeID, channel, segments)
@@ -148,6 +156,10 @@ func (_c *MockCluster_Flush_Call) RunAndReturn(run func(context.Context, int64,
 func (_m *MockCluster) FlushChannels(ctx context.Context, nodeID int64, flushTs uint64, channels []string) error {
 	ret := _m.Called(ctx, nodeID, flushTs, channels)
 
+	if len(ret) == 0 {
+		panic("no return value specified for FlushChannels")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, uint64, []string) error); ok {
 		r0 = rf(ctx, nodeID, flushTs, channels)
@@ -193,6 +205,10 @@ func (_c *MockCluster_FlushChannels_Call) RunAndReturn(run func(context.Context,
 func (_m *MockCluster) GetSessions() []*session.Session {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSessions")
+	}
+
 	var r0 []*session.Session
 	if rf, ok := ret.Get(0).(func() []*session.Session); ok {
 		r0 = rf()
@@ -236,6 +252,10 @@ func (_c *MockCluster_GetSessions_Call) RunAndReturn(run func() []*session.Sessi
 func (_m *MockCluster) ImportV2(nodeID int64, in *datapb.ImportRequest) error {
 	ret := _m.Called(nodeID, in)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ImportV2")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64, *datapb.ImportRequest) error); ok {
 		r0 = rf(nodeID, in)
@@ -279,6 +299,10 @@ func (_c *MockCluster_ImportV2_Call) RunAndReturn(run func(int64, *datapb.Import
 func (_m *MockCluster) PreImport(nodeID int64, in *datapb.PreImportRequest) error {
 	ret := _m.Called(nodeID, in)
 
+	if len(ret) == 0 {
+		panic("no return value specified for PreImport")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64, *datapb.PreImportRequest) error); ok {
 		r0 = rf(nodeID, in)
@@ -322,6 +346,10 @@ func (_c *MockCluster_PreImport_Call) RunAndReturn(run func(int64, *datapb.PreIm
 func (_m *MockCluster) QueryImport(nodeID int64, in *datapb.QueryImportRequest) (*datapb.QueryImportResponse, error) {
 	ret := _m.Called(nodeID, in)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryImport")
+	}
+
 	var r0 *datapb.QueryImportResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(int64, *datapb.QueryImportRequest) (*datapb.QueryImportResponse, error)); ok {
@@ -377,6 +405,10 @@ func (_c *MockCluster_QueryImport_Call) RunAndReturn(run func(int64, *datapb.Que
 func (_m *MockCluster) QueryPreImport(nodeID int64, in *datapb.QueryPreImportRequest) (*datapb.QueryPreImportResponse, error) {
 	ret := _m.Called(nodeID, in)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryPreImport")
+	}
+
 	var r0 *datapb.QueryPreImportResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(int64, *datapb.QueryPreImportRequest) (*datapb.QueryPreImportResponse, error)); ok {
@@ -432,6 +464,10 @@ func (_c *MockCluster_QueryPreImport_Call) RunAndReturn(run func(int64, *datapb.
 func (_m *MockCluster) QuerySlots() map[int64]int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for QuerySlots")
+	}
+
 	var r0 map[int64]int64
 	if rf, ok := ret.Get(0).(func() map[int64]int64); ok {
 		r0 = rf()
@@ -475,6 +511,10 @@ func (_c *MockCluster_QuerySlots_Call) RunAndReturn(run func() map[int64]int64)
 func (_m *MockCluster) Register(node *session.NodeInfo) error {
 	ret := _m.Called(node)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Register")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*session.NodeInfo) error); ok {
 		r0 = rf(node)
@@ -517,6 +557,10 @@ func (_c *MockCluster_Register_Call) RunAndReturn(run func(*session.NodeInfo) er
 func (_m *MockCluster) Startup(ctx context.Context, nodes []*session.NodeInfo) error {
 	ret := _m.Called(ctx, nodes)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Startup")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, []*session.NodeInfo) error); ok {
 		r0 = rf(ctx, nodes)
@@ -560,6 +604,10 @@ func (_c *MockCluster_Startup_Call) RunAndReturn(run func(context.Context, []*se
 func (_m *MockCluster) UnRegister(node *session.NodeInfo) error {
 	ret := _m.Called(node)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UnRegister")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*session.NodeInfo) error); ok {
 		r0 = rf(node)
@@ -602,6 +650,10 @@ func (_c *MockCluster_UnRegister_Call) RunAndReturn(run func(*session.NodeInfo)
 func (_m *MockCluster) Watch(ctx context.Context, ch RWChannel) error {
 	ret := _m.Called(ctx, ch)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Watch")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, RWChannel) error); ok {
 		r0 = rf(ctx, ch)

internal/datacoord/mock_compaction_meta.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package datacoord
 
@@ -24,6 +24,10 @@ func (_m *MockCompactionMeta) EXPECT() *MockCompactionMeta_Expecter {
 func (_m *MockCompactionMeta) CheckAndSetSegmentsCompacting(segmentIDs []int64) (bool, bool) {
 	ret := _m.Called(segmentIDs)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckAndSetSegmentsCompacting")
+	}
+
 	var r0 bool
 	var r1 bool
 	if rf, ok := ret.Get(0).(func([]int64) (bool, bool)); ok {
@@ -76,6 +80,10 @@ func (_c *MockCompactionMeta_CheckAndSetSegmentsCompacting_Call) RunAndReturn(ru
 func (_m *MockCompactionMeta) CleanPartitionStatsInfo(info *datapb.PartitionStatsInfo) error {
 	ret := _m.Called(info)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CleanPartitionStatsInfo")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*datapb.PartitionStatsInfo) error); ok {
 		r0 = rf(info)
@@ -118,6 +126,10 @@ func (_c *MockCompactionMeta_CleanPartitionStatsInfo_Call) RunAndReturn(run func
 func (_m *MockCompactionMeta) CompleteCompactionMutation(t *datapb.CompactionTask, result *datapb.CompactionPlanResult) ([]*SegmentInfo, *segMetricMutation, error) {
 	ret := _m.Called(t, result)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CompleteCompactionMutation")
+	}
+
 	var r0 []*SegmentInfo
 	var r1 *segMetricMutation
 	var r2 error
@@ -182,6 +194,10 @@ func (_c *MockCompactionMeta_CompleteCompactionMutation_Call) RunAndReturn(run f
 func (_m *MockCompactionMeta) DropCompactionTask(task *datapb.CompactionTask) error {
 	ret := _m.Called(task)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropCompactionTask")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*datapb.CompactionTask) error); ok {
 		r0 = rf(task)
@@ -224,6 +240,10 @@ func (_c *MockCompactionMeta_DropCompactionTask_Call) RunAndReturn(run func(*dat
 func (_m *MockCompactionMeta) GetAnalyzeMeta() *analyzeMeta {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAnalyzeMeta")
+	}
+
 	var r0 *analyzeMeta
 	if rf, ok := ret.Get(0).(func() *analyzeMeta); ok {
 		r0 = rf()
@@ -267,6 +287,10 @@ func (_c *MockCompactionMeta_GetAnalyzeMeta_Call) RunAndReturn(run func() *analy
 func (_m *MockCompactionMeta) GetCompactionTaskMeta() *compactionTaskMeta {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCompactionTaskMeta")
+	}
+
 	var r0 *compactionTaskMeta
 	if rf, ok := ret.Get(0).(func() *compactionTaskMeta); ok {
 		r0 = rf()
@@ -310,6 +334,10 @@ func (_c *MockCompactionMeta_GetCompactionTaskMeta_Call) RunAndReturn(run func()
 func (_m *MockCompactionMeta) GetCompactionTasks() map[int64][]*datapb.CompactionTask {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCompactionTasks")
+	}
+
 	var r0 map[int64][]*datapb.CompactionTask
 	if rf, ok := ret.Get(0).(func() map[int64][]*datapb.CompactionTask); ok {
 		r0 = rf()
@@ -353,6 +381,10 @@ func (_c *MockCompactionMeta_GetCompactionTasks_Call) RunAndReturn(run func() ma
 func (_m *MockCompactionMeta) GetCompactionTasksByTriggerID(triggerID int64) []*datapb.CompactionTask {
 	ret := _m.Called(triggerID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCompactionTasksByTriggerID")
+	}
+
 	var r0 []*datapb.CompactionTask
 	if rf, ok := ret.Get(0).(func(int64) []*datapb.CompactionTask); ok {
 		r0 = rf(triggerID)
@@ -397,6 +429,10 @@ func (_c *MockCompactionMeta_GetCompactionTasksByTriggerID_Call) RunAndReturn(ru
 func (_m *MockCompactionMeta) GetHealthySegment(segID int64) *SegmentInfo {
 	ret := _m.Called(segID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetHealthySegment")
+	}
+
 	var r0 *SegmentInfo
 	if rf, ok := ret.Get(0).(func(int64) *SegmentInfo); ok {
 		r0 = rf(segID)
@@ -441,6 +477,10 @@ func (_c *MockCompactionMeta_GetHealthySegment_Call) RunAndReturn(run func(int64
 func (_m *MockCompactionMeta) GetIndexMeta() *indexMeta {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetIndexMeta")
+	}
+
 	var r0 *indexMeta
 	if rf, ok := ret.Get(0).(func() *indexMeta); ok {
 		r0 = rf()
@@ -484,6 +524,10 @@ func (_c *MockCompactionMeta_GetIndexMeta_Call) RunAndReturn(run func() *indexMe
 func (_m *MockCompactionMeta) GetPartitionStatsMeta() *partitionStatsMeta {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPartitionStatsMeta")
+	}
+
 	var r0 *partitionStatsMeta
 	if rf, ok := ret.Get(0).(func() *partitionStatsMeta); ok {
 		r0 = rf()
@@ -527,6 +571,10 @@ func (_c *MockCompactionMeta_GetPartitionStatsMeta_Call) RunAndReturn(run func()
 func (_m *MockCompactionMeta) GetSegment(segID int64) *SegmentInfo {
 	ret := _m.Called(segID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegment")
+	}
+
 	var r0 *SegmentInfo
 	if rf, ok := ret.Get(0).(func(int64) *SegmentInfo); ok {
 		r0 = rf(segID)
@@ -571,6 +619,10 @@ func (_c *MockCompactionMeta_GetSegment_Call) RunAndReturn(run func(int64) *Segm
 func (_m *MockCompactionMeta) SaveCompactionTask(task *datapb.CompactionTask) error {
 	ret := _m.Called(task)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SaveCompactionTask")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*datapb.CompactionTask) error); ok {
 		r0 = rf(task)
@@ -619,6 +671,10 @@ func (_m *MockCompactionMeta) SelectSegments(filters ...SegmentFilter) []*Segmen
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SelectSegments")
+	}
+
 	var r0 []*SegmentInfo
 	if rf, ok := ret.Get(0).(func(...SegmentFilter) []*SegmentInfo); ok {
 		r0 = rf(filters...)
@@ -710,6 +766,10 @@ func (_m *MockCompactionMeta) UpdateSegmentsInfo(operators ...UpdateOperator) er
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateSegmentsInfo")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(...UpdateOperator) error); ok {
 		r0 = rf(operators...)

internal/datacoord/mock_compaction_plan_context.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package datacoord
 
@@ -24,6 +24,10 @@ func (_m *MockCompactionPlanContext) EXPECT() *MockCompactionPlanContext_Expecte
 func (_m *MockCompactionPlanContext) enqueueCompaction(task *datapb.CompactionTask) error {
 	ret := _m.Called(task)
 
+	if len(ret) == 0 {
+		panic("no return value specified for enqueueCompaction")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*datapb.CompactionTask) error); ok {
 		r0 = rf(task)
@@ -66,6 +70,10 @@ func (_c *MockCompactionPlanContext_enqueueCompaction_Call) RunAndReturn(run fun
 func (_m *MockCompactionPlanContext) getCompactionInfo(signalID int64) *compactionInfo {
 	ret := _m.Called(signalID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for getCompactionInfo")
+	}
+
 	var r0 *compactionInfo
 	if rf, ok := ret.Get(0).(func(int64) *compactionInfo); ok {
 		r0 = rf(signalID)
@@ -110,6 +118,10 @@ func (_c *MockCompactionPlanContext_getCompactionInfo_Call) RunAndReturn(run fun
 func (_m *MockCompactionPlanContext) getCompactionTasksNumBySignalID(signalID int64) int {
 	ret := _m.Called(signalID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for getCompactionTasksNumBySignalID")
+	}
+
 	var r0 int
 	if rf, ok := ret.Get(0).(func(int64) int); ok {
 		r0 = rf(signalID)
@@ -152,6 +164,10 @@ func (_c *MockCompactionPlanContext_getCompactionTasksNumBySignalID_Call) RunAnd
 func (_m *MockCompactionPlanContext) isFull() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for isFull")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()

internal/datacoord/mock_handler.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package datacoord
 
@@ -26,6 +26,10 @@ func (_m *NMockHandler) EXPECT() *NMockHandler_Expecter {
 func (_m *NMockHandler) CheckShouldDropChannel(ch string) bool {
 	ret := _m.Called(ch)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckShouldDropChannel")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(string) bool); ok {
 		r0 = rf(ch)
@@ -68,6 +72,10 @@ func (_c *NMockHandler_CheckShouldDropChannel_Call) RunAndReturn(run func(string
 func (_m *NMockHandler) FinishDropChannel(ch string, collectionID int64) error {
 	ret := _m.Called(ch, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for FinishDropChannel")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, int64) error); ok {
 		r0 = rf(ch, collectionID)
@@ -111,6 +119,10 @@ func (_c *NMockHandler_FinishDropChannel_Call) RunAndReturn(run func(string, int
 func (_m *NMockHandler) GetCollection(ctx context.Context, collectionID int64) (*collectionInfo, error) {
 	ret := _m.Called(ctx, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollection")
+	}
+
 	var r0 *collectionInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) (*collectionInfo, error)); ok {
@@ -166,6 +178,10 @@ func (_c *NMockHandler_GetCollection_Call) RunAndReturn(run func(context.Context
 func (_m *NMockHandler) GetDataVChanPositions(ch RWChannel, partitionID int64) *datapb.VchannelInfo {
 	ret := _m.Called(ch, partitionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDataVChanPositions")
+	}
+
 	var r0 *datapb.VchannelInfo
 	if rf, ok := ret.Get(0).(func(RWChannel, int64) *datapb.VchannelInfo); ok {
 		r0 = rf(ch, partitionID)
@@ -218,6 +234,10 @@ func (_m *NMockHandler) GetQueryVChanPositions(ch RWChannel, partitionIDs ...int
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetQueryVChanPositions")
+	}
+
 	var r0 *datapb.VchannelInfo
 	if rf, ok := ret.Get(0).(func(RWChannel, ...int64) *datapb.VchannelInfo); ok {
 		r0 = rf(ch, partitionIDs...)

internal/datacoord/mock_index_engine_version_manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package datacoord
 
@@ -57,6 +57,10 @@ func (_c *MockVersionManager_AddNode_Call) RunAndReturn(run func(*sessionutil.Se
 func (_m *MockVersionManager) GetCurrentIndexEngineVersion() int32 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCurrentIndexEngineVersion")
+	}
+
 	var r0 int32
 	if rf, ok := ret.Get(0).(func() int32); ok {
 		r0 = rf()
@@ -98,6 +102,10 @@ func (_c *MockVersionManager_GetCurrentIndexEngineVersion_Call) RunAndReturn(run
 func (_m *MockVersionManager) GetMinimalIndexEngineVersion() int32 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMinimalIndexEngineVersion")
+	}
+
 	var r0 int32
 	if rf, ok := ret.Get(0).(func() int32); ok {
 		r0 = rf()

internal/datacoord/mock_job_manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package datacoord
 
@@ -24,6 +24,10 @@ func (_m *MockStatsJobManager) EXPECT() *MockStatsJobManager_Expecter {
 func (_m *MockStatsJobManager) DropStatsTask(originSegmentID int64, subJobType indexpb.StatsSubJob) error {
 	ret := _m.Called(originSegmentID, subJobType)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropStatsTask")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64, indexpb.StatsSubJob) error); ok {
 		r0 = rf(originSegmentID, subJobType)
@@ -67,6 +71,10 @@ func (_c *MockStatsJobManager_DropStatsTask_Call) RunAndReturn(run func(int64, i
 func (_m *MockStatsJobManager) GetStatsTaskState(originSegmentID int64, subJobType indexpb.StatsSubJob) indexpb.JobState {
 	ret := _m.Called(originSegmentID, subJobType)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatsTaskState")
+	}
+
 	var r0 indexpb.JobState
 	if rf, ok := ret.Get(0).(func(int64, indexpb.StatsSubJob) indexpb.JobState); ok {
 		r0 = rf(originSegmentID, subJobType)
@@ -174,6 +182,10 @@ func (_c *MockStatsJobManager_Stop_Call) RunAndReturn(run func()) *MockStatsJobM
 func (_m *MockStatsJobManager) SubmitStatsTask(originSegmentID int64, targetSegmentID int64, subJobType indexpb.StatsSubJob, canRecycle bool) error {
 	ret := _m.Called(originSegmentID, targetSegmentID, subJobType, canRecycle)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SubmitStatsTask")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64, int64, indexpb.StatsSubJob, bool) error); ok {
 		r0 = rf(originSegmentID, targetSegmentID, subJobType, canRecycle)

internal/datacoord/mock_segment_manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package datacoord
 
@@ -25,6 +25,10 @@ func (_m *MockManager) EXPECT() *MockManager_Expecter {
 func (_m *MockManager) AllocNewGrowingSegment(ctx context.Context, collectionID int64, partitionID int64, segmentID int64, channelName string) (*SegmentInfo, error) {
 	ret := _m.Called(ctx, collectionID, partitionID, segmentID, channelName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocNewGrowingSegment")
+	}
+
 	var r0 *SegmentInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, int64, int64, string) (*SegmentInfo, error)); ok {
@@ -83,6 +87,10 @@ func (_c *MockManager_AllocNewGrowingSegment_Call) RunAndReturn(run func(context
 func (_m *MockManager) AllocSegment(ctx context.Context, collectionID int64, partitionID int64, channelName string, requestRows int64) ([]*Allocation, error) {
 	ret := _m.Called(ctx, collectionID, partitionID, channelName, requestRows)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocSegment")
+	}
+
 	var r0 []*Allocation
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, int64, string, int64) ([]*Allocation, error)); ok {
@@ -209,6 +217,10 @@ func (_c *MockManager_DropSegmentsOfChannel_Call) RunAndReturn(run func(context.
 func (_m *MockManager) ExpireAllocations(channel string, ts uint64) error {
 	ret := _m.Called(channel, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ExpireAllocations")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, uint64) error); ok {
 		r0 = rf(channel, ts)
@@ -252,6 +264,10 @@ func (_c *MockManager_ExpireAllocations_Call) RunAndReturn(run func(string, uint
 func (_m *MockManager) GetFlushableSegments(ctx context.Context, channel string, ts uint64) ([]int64, error) {
 	ret := _m.Called(ctx, channel, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetFlushableSegments")
+	}
+
 	var r0 []int64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, uint64) ([]int64, error)); ok {
@@ -308,6 +324,10 @@ func (_c *MockManager_GetFlushableSegments_Call) RunAndReturn(run func(context.C
 func (_m *MockManager) SealAllSegments(ctx context.Context, collectionID int64, segIDs []int64) ([]int64, error) {
 	ret := _m.Called(ctx, collectionID, segIDs)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SealAllSegments")
+	}
+
 	var r0 []int64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, []int64) ([]int64, error)); ok {

internal/datacoord/mock_subcluster.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package datacoord
 
@@ -26,6 +26,10 @@ func (_m *MockSubCluster) EXPECT() *MockSubCluster_Expecter {
 func (_m *MockSubCluster) CheckChannelOperationProgress(ctx context.Context, nodeID int64, info *datapb.ChannelWatchInfo) (*datapb.ChannelOperationProgressResponse, error) {
 	ret := _m.Called(ctx, nodeID, info)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckChannelOperationProgress")
+	}
+
 	var r0 *datapb.ChannelOperationProgressResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *datapb.ChannelWatchInfo) (*datapb.ChannelOperationProgressResponse, error)); ok {
@@ -82,6 +86,10 @@ func (_c *MockSubCluster_CheckChannelOperationProgress_Call) RunAndReturn(run fu
 func (_m *MockSubCluster) NotifyChannelOperation(ctx context.Context, nodeID int64, req *datapb.ChannelOperationsRequest) error {
 	ret := _m.Called(ctx, nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for NotifyChannelOperation")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *datapb.ChannelOperationsRequest) error); ok {
 		r0 = rf(ctx, nodeID, req)

internal/datacoord/mock_trigger_manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package datacoord
 
@@ -25,6 +25,10 @@ func (_m *MockTriggerManager) EXPECT() *MockTriggerManager_Expecter {
 func (_m *MockTriggerManager) ManualTrigger(ctx context.Context, collectionID int64, clusteringCompaction bool) (int64, error) {
 	ret := _m.Called(ctx, collectionID, clusteringCompaction)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ManualTrigger")
+	}
+
 	var r0 int64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, bool) (int64, error)); ok {

internal/datacoord/session/mock_datanode_manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package session
 
@@ -61,6 +61,10 @@ func (_c *MockDataNodeManager_AddSession_Call) RunAndReturn(run func(*NodeInfo))
 func (_m *MockDataNodeManager) CheckChannelOperationProgress(ctx context.Context, nodeID int64, info *datapb.ChannelWatchInfo) (*datapb.ChannelOperationProgressResponse, error) {
 	ret := _m.Called(ctx, nodeID, info)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckChannelOperationProgress")
+	}
+
 	var r0 *datapb.ChannelOperationProgressResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *datapb.ChannelWatchInfo) (*datapb.ChannelOperationProgressResponse, error)); ok {
@@ -117,6 +121,10 @@ func (_c *MockDataNodeManager_CheckChannelOperationProgress_Call) RunAndReturn(r
 func (_m *MockDataNodeManager) CheckHealth(ctx context.Context) error {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckHealth")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
 		r0 = rf(ctx)
@@ -191,6 +199,10 @@ func (_c *MockDataNodeManager_Close_Call) RunAndReturn(run func()) *MockDataNode
 func (_m *MockDataNodeManager) Compaction(ctx context.Context, nodeID int64, plan *datapb.CompactionPlan) error {
 	ret := _m.Called(ctx, nodeID, plan)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Compaction")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *datapb.CompactionPlan) error); ok {
 		r0 = rf(ctx, nodeID, plan)
@@ -268,6 +280,10 @@ func (_c *MockDataNodeManager_DeleteSession_Call) RunAndReturn(run func(*NodeInf
 func (_m *MockDataNodeManager) DropCompactionPlan(nodeID int64, req *datapb.DropCompactionPlanRequest) error {
 	ret := _m.Called(nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropCompactionPlan")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64, *datapb.DropCompactionPlanRequest) error); ok {
 		r0 = rf(nodeID, req)
@@ -311,6 +327,10 @@ func (_c *MockDataNodeManager_DropCompactionPlan_Call) RunAndReturn(run func(int
 func (_m *MockDataNodeManager) DropImport(nodeID int64, in *datapb.DropImportRequest) error {
 	ret := _m.Called(nodeID, in)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropImport")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64, *datapb.DropImportRequest) error); ok {
 		r0 = rf(nodeID, in)
@@ -389,6 +409,10 @@ func (_c *MockDataNodeManager_Flush_Call) RunAndReturn(run func(context.Context,
 func (_m *MockDataNodeManager) FlushChannels(ctx context.Context, nodeID int64, req *datapb.FlushChannelsRequest) error {
 	ret := _m.Called(ctx, nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for FlushChannels")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *datapb.FlushChannelsRequest) error); ok {
 		r0 = rf(ctx, nodeID, req)
@@ -433,6 +457,10 @@ func (_c *MockDataNodeManager_FlushChannels_Call) RunAndReturn(run func(context.
 func (_m *MockDataNodeManager) GetCompactionPlanResult(nodeID int64, planID int64) (*datapb.CompactionPlanResult, error) {
 	ret := _m.Called(nodeID, planID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCompactionPlanResult")
+	}
+
 	var r0 *datapb.CompactionPlanResult
 	var r1 error
 	if rf, ok := ret.Get(0).(func(int64, int64) (*datapb.CompactionPlanResult, error)); ok {
@@ -488,6 +516,10 @@ func (_c *MockDataNodeManager_GetCompactionPlanResult_Call) RunAndReturn(run fun
 func (_m *MockDataNodeManager) GetCompactionPlansResults() (map[int64]*typeutil.Pair[int64, *datapb.CompactionPlanResult], error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCompactionPlansResults")
+	}
+
 	var r0 map[int64]*typeutil.Pair[int64, *datapb.CompactionPlanResult]
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (map[int64]*typeutil.Pair[int64, *datapb.CompactionPlanResult], error)); ok {
@@ -541,6 +573,10 @@ func (_c *MockDataNodeManager_GetCompactionPlansResults_Call) RunAndReturn(run f
 func (_m *MockDataNodeManager) GetSession(_a0 int64) (*Session, bool) {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSession")
+	}
+
 	var r0 *Session
 	var r1 bool
 	if rf, ok := ret.Get(0).(func(int64) (*Session, bool)); ok {
@@ -595,6 +631,10 @@ func (_c *MockDataNodeManager_GetSession_Call) RunAndReturn(run func(int64) (*Se
 func (_m *MockDataNodeManager) GetSessionIDs() []int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSessionIDs")
+	}
+
 	var r0 []int64
 	if rf, ok := ret.Get(0).(func() []int64); ok {
 		r0 = rf()
@@ -638,6 +678,10 @@ func (_c *MockDataNodeManager_GetSessionIDs_Call) RunAndReturn(run func() []int6
 func (_m *MockDataNodeManager) GetSessions() []*Session {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSessions")
+	}
+
 	var r0 []*Session
 	if rf, ok := ret.Get(0).(func() []*Session); ok {
 		r0 = rf()
@@ -681,6 +725,10 @@ func (_c *MockDataNodeManager_GetSessions_Call) RunAndReturn(run func() []*Sessi
 func (_m *MockDataNodeManager) ImportV2(nodeID int64, in *datapb.ImportRequest) error {
 	ret := _m.Called(nodeID, in)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ImportV2")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64, *datapb.ImportRequest) error); ok {
 		r0 = rf(nodeID, in)
@@ -724,6 +772,10 @@ func (_c *MockDataNodeManager_ImportV2_Call) RunAndReturn(run func(int64, *datap
 func (_m *MockDataNodeManager) NotifyChannelOperation(ctx context.Context, nodeID int64, req *datapb.ChannelOperationsRequest) error {
 	ret := _m.Called(ctx, nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for NotifyChannelOperation")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *datapb.ChannelOperationsRequest) error); ok {
 		r0 = rf(ctx, nodeID, req)
@@ -768,6 +820,10 @@ func (_c *MockDataNodeManager_NotifyChannelOperation_Call) RunAndReturn(run func
 func (_m *MockDataNodeManager) PreImport(nodeID int64, in *datapb.PreImportRequest) error {
 	ret := _m.Called(nodeID, in)
 
+	if len(ret) == 0 {
+		panic("no return value specified for PreImport")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64, *datapb.PreImportRequest) error); ok {
 		r0 = rf(nodeID, in)
@@ -811,6 +867,10 @@ func (_c *MockDataNodeManager_PreImport_Call) RunAndReturn(run func(int64, *data
 func (_m *MockDataNodeManager) QueryImport(nodeID int64, in *datapb.QueryImportRequest) (*datapb.QueryImportResponse, error) {
 	ret := _m.Called(nodeID, in)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryImport")
+	}
+
 	var r0 *datapb.QueryImportResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(int64, *datapb.QueryImportRequest) (*datapb.QueryImportResponse, error)); ok {
@@ -866,6 +926,10 @@ func (_c *MockDataNodeManager_QueryImport_Call) RunAndReturn(run func(int64, *da
 func (_m *MockDataNodeManager) QueryPreImport(nodeID int64, in *datapb.QueryPreImportRequest) (*datapb.QueryPreImportResponse, error) {
 	ret := _m.Called(nodeID, in)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryPreImport")
+	}
+
 	var r0 *datapb.QueryPreImportResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(int64, *datapb.QueryPreImportRequest) (*datapb.QueryPreImportResponse, error)); ok {
@@ -921,6 +985,10 @@ func (_c *MockDataNodeManager_QueryPreImport_Call) RunAndReturn(run func(int64,
 func (_m *MockDataNodeManager) QuerySlot(nodeID int64) (*datapb.QuerySlotResponse, error) {
 	ret := _m.Called(nodeID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QuerySlot")
+	}
+
 	var r0 *datapb.QuerySlotResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(int64) (*datapb.QuerySlotResponse, error)); ok {
@@ -975,6 +1043,10 @@ func (_c *MockDataNodeManager_QuerySlot_Call) RunAndReturn(run func(int64) (*dat
 func (_m *MockDataNodeManager) SyncSegments(nodeID int64, req *datapb.SyncSegmentsRequest) error {
 	ret := _m.Called(nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SyncSegments")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64, *datapb.SyncSegmentsRequest) error); ok {
 		r0 = rf(nodeID, req)

internal/datacoord/session/mock_worker_manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package session
 
@@ -24,6 +24,10 @@ func (_m *MockWorkerManager) EXPECT() *MockWorkerManager_Expecter {
 func (_m *MockWorkerManager) AddNode(nodeID int64, address string) error {
 	ret := _m.Called(nodeID, address)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AddNode")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64, string) error); ok {
 		r0 = rf(nodeID, address)
@@ -67,6 +71,10 @@ func (_c *MockWorkerManager_AddNode_Call) RunAndReturn(run func(int64, string) e
 func (_m *MockWorkerManager) ClientSupportDisk() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for ClientSupportDisk")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()
@@ -108,6 +116,10 @@ func (_c *MockWorkerManager_ClientSupportDisk_Call) RunAndReturn(run func() bool
 func (_m *MockWorkerManager) GetAllClients() map[int64]types.IndexNodeClient {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAllClients")
+	}
+
 	var r0 map[int64]types.IndexNodeClient
 	if rf, ok := ret.Get(0).(func() map[int64]types.IndexNodeClient); ok {
 		r0 = rf()
@@ -151,6 +163,10 @@ func (_c *MockWorkerManager_GetAllClients_Call) RunAndReturn(run func() map[int6
 func (_m *MockWorkerManager) GetClientByID(nodeID int64) (types.IndexNodeClient, bool) {
 	ret := _m.Called(nodeID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetClientByID")
+	}
+
 	var r0 types.IndexNodeClient
 	var r1 bool
 	if rf, ok := ret.Get(0).(func(int64) (types.IndexNodeClient, bool)); ok {
@@ -205,6 +221,10 @@ func (_c *MockWorkerManager_GetClientByID_Call) RunAndReturn(run func(int64) (ty
 func (_m *MockWorkerManager) PickClient() (int64, types.IndexNodeClient) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for PickClient")
+	}
+
 	var r0 int64
 	var r1 types.IndexNodeClient
 	if rf, ok := ret.Get(0).(func() (int64, types.IndexNodeClient)); ok {

internal/kv/mocks/meta_kv.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -56,6 +56,10 @@ func (_c *MetaKv_Close_Call) RunAndReturn(run func()) *MetaKv_Close_Call {
 func (_m *MetaKv) CompareVersionAndSwap(key string, version int64, target string) (bool, error) {
 	ret := _m.Called(key, version, target)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CompareVersionAndSwap")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string, int64, string) (bool, error)); ok {
@@ -110,6 +114,10 @@ func (_c *MetaKv_CompareVersionAndSwap_Call) RunAndReturn(run func(string, int64
 func (_m *MetaKv) GetPath(key string) string {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPath")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func(string) string); ok {
 		r0 = rf(key)
@@ -152,6 +160,10 @@ func (_c *MetaKv_GetPath_Call) RunAndReturn(run func(string) string) *MetaKv_Get
 func (_m *MetaKv) Has(key string) (bool, error) {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Has")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (bool, error)); ok {
@@ -204,6 +216,10 @@ func (_c *MetaKv_Has_Call) RunAndReturn(run func(string) (bool, error)) *MetaKv_
 func (_m *MetaKv) HasPrefix(prefix string) (bool, error) {
 	ret := _m.Called(prefix)
 
+	if len(ret) == 0 {
+		panic("no return value specified for HasPrefix")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (bool, error)); ok {
@@ -256,6 +272,10 @@ func (_c *MetaKv_HasPrefix_Call) RunAndReturn(run func(string) (bool, error)) *M
 func (_m *MetaKv) Load(key string) (string, error) {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Load")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (string, error)); ok {
@@ -308,6 +328,10 @@ func (_c *MetaKv_Load_Call) RunAndReturn(run func(string) (string, error)) *Meta
 func (_m *MetaKv) LoadWithPrefix(key string) ([]string, []string, error) {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadWithPrefix")
+	}
+
 	var r0 []string
 	var r1 []string
 	var r2 error
@@ -371,6 +395,10 @@ func (_c *MetaKv_LoadWithPrefix_Call) RunAndReturn(run func(string) ([]string, [
 func (_m *MetaKv) MultiLoad(keys []string) ([]string, error) {
 	ret := _m.Called(keys)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiLoad")
+	}
+
 	var r0 []string
 	var r1 error
 	if rf, ok := ret.Get(0).(func([]string) ([]string, error)); ok {
@@ -425,6 +453,10 @@ func (_c *MetaKv_MultiLoad_Call) RunAndReturn(run func([]string) ([]string, erro
 func (_m *MetaKv) MultiRemove(keys []string) error {
 	ret := _m.Called(keys)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiRemove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func([]string) error); ok {
 		r0 = rf(keys)
@@ -467,6 +499,10 @@ func (_c *MetaKv_MultiRemove_Call) RunAndReturn(run func([]string) error) *MetaK
 func (_m *MetaKv) MultiSave(kvs map[string]string) error {
 	ret := _m.Called(kvs)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSave")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string) error); ok {
 		r0 = rf(kvs)
@@ -516,6 +552,10 @@ func (_m *MetaKv) MultiSaveAndRemove(saves map[string]string, removals []string,
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSaveAndRemove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok {
 		r0 = rf(saves, removals, preds...)
@@ -574,6 +614,10 @@ func (_m *MetaKv) MultiSaveAndRemoveWithPrefix(saves map[string]string, removals
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSaveAndRemoveWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok {
 		r0 = rf(saves, removals, preds...)
@@ -625,6 +669,10 @@ func (_c *MetaKv_MultiSaveAndRemoveWithPrefix_Call) RunAndReturn(run func(map[st
 func (_m *MetaKv) Remove(key string) error {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Remove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(key)
@@ -667,6 +715,10 @@ func (_c *MetaKv_Remove_Call) RunAndReturn(run func(string) error) *MetaKv_Remov
 func (_m *MetaKv) RemoveWithPrefix(key string) error {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemoveWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(key)
@@ -709,6 +761,10 @@ func (_c *MetaKv_RemoveWithPrefix_Call) RunAndReturn(run func(string) error) *Me
 func (_m *MetaKv) Save(key string, value string) error {
 	ret := _m.Called(key, value)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Save")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, string) error); ok {
 		r0 = rf(key, value)
@@ -752,6 +808,10 @@ func (_c *MetaKv_Save_Call) RunAndReturn(run func(string, string) error) *MetaKv
 func (_m *MetaKv) WalkWithPrefix(prefix string, paginationSize int, fn func([]byte, []byte) error) error {
 	ret := _m.Called(prefix, paginationSize, fn)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WalkWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, int, func([]byte, []byte) error) error); ok {
 		r0 = rf(prefix, paginationSize, fn)

internal/kv/mocks/snapshot_kv.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -21,6 +21,10 @@ func (_m *SnapShotKV) EXPECT() *SnapShotKV_Expecter {
 func (_m *SnapShotKV) Load(key string, ts uint64) (string, error) {
 	ret := _m.Called(key, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Load")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string, uint64) (string, error)); ok {
@@ -74,6 +78,10 @@ func (_c *SnapShotKV_Load_Call) RunAndReturn(run func(string, uint64) (string, e
 func (_m *SnapShotKV) LoadWithPrefix(key string, ts uint64) ([]string, []string, error) {
 	ret := _m.Called(key, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadWithPrefix")
+	}
+
 	var r0 []string
 	var r1 []string
 	var r2 error
@@ -138,6 +146,10 @@ func (_c *SnapShotKV_LoadWithPrefix_Call) RunAndReturn(run func(string, uint64)
 func (_m *SnapShotKV) MultiSave(kvs map[string]string, ts uint64) error {
 	ret := _m.Called(kvs, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSave")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string, uint64) error); ok {
 		r0 = rf(kvs, ts)
@@ -181,6 +193,10 @@ func (_c *SnapShotKV_MultiSave_Call) RunAndReturn(run func(map[string]string, ui
 func (_m *SnapShotKV) MultiSaveAndRemove(saves map[string]string, removals []string, ts uint64) error {
 	ret := _m.Called(saves, removals, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSaveAndRemove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string, []string, uint64) error); ok {
 		r0 = rf(saves, removals, ts)
@@ -225,6 +241,10 @@ func (_c *SnapShotKV_MultiSaveAndRemove_Call) RunAndReturn(run func(map[string]s
 func (_m *SnapShotKV) MultiSaveAndRemoveWithPrefix(saves map[string]string, removals []string, ts uint64) error {
 	ret := _m.Called(saves, removals, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSaveAndRemoveWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string, []string, uint64) error); ok {
 		r0 = rf(saves, removals, ts)
@@ -269,6 +289,10 @@ func (_c *SnapShotKV_MultiSaveAndRemoveWithPrefix_Call) RunAndReturn(run func(ma
 func (_m *SnapShotKV) Save(key string, value string, ts uint64) error {
 	ret := _m.Called(key, value, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Save")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, string, uint64) error); ok {
 		r0 = rf(key, value, ts)

internal/kv/mocks/txn_kv.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -56,6 +56,10 @@ func (_c *TxnKV_Close_Call) RunAndReturn(run func()) *TxnKV_Close_Call {
 func (_m *TxnKV) Has(key string) (bool, error) {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Has")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (bool, error)); ok {
@@ -108,6 +112,10 @@ func (_c *TxnKV_Has_Call) RunAndReturn(run func(string) (bool, error)) *TxnKV_Ha
 func (_m *TxnKV) HasPrefix(prefix string) (bool, error) {
 	ret := _m.Called(prefix)
 
+	if len(ret) == 0 {
+		panic("no return value specified for HasPrefix")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (bool, error)); ok {
@@ -160,6 +168,10 @@ func (_c *TxnKV_HasPrefix_Call) RunAndReturn(run func(string) (bool, error)) *Tx
 func (_m *TxnKV) Load(key string) (string, error) {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Load")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (string, error)); ok {
@@ -212,6 +224,10 @@ func (_c *TxnKV_Load_Call) RunAndReturn(run func(string) (string, error)) *TxnKV
 func (_m *TxnKV) LoadWithPrefix(key string) ([]string, []string, error) {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadWithPrefix")
+	}
+
 	var r0 []string
 	var r1 []string
 	var r2 error
@@ -275,6 +291,10 @@ func (_c *TxnKV_LoadWithPrefix_Call) RunAndReturn(run func(string) ([]string, []
 func (_m *TxnKV) MultiLoad(keys []string) ([]string, error) {
 	ret := _m.Called(keys)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiLoad")
+	}
+
 	var r0 []string
 	var r1 error
 	if rf, ok := ret.Get(0).(func([]string) ([]string, error)); ok {
@@ -329,6 +349,10 @@ func (_c *TxnKV_MultiLoad_Call) RunAndReturn(run func([]string) ([]string, error
 func (_m *TxnKV) MultiRemove(keys []string) error {
 	ret := _m.Called(keys)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiRemove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func([]string) error); ok {
 		r0 = rf(keys)
@@ -371,6 +395,10 @@ func (_c *TxnKV_MultiRemove_Call) RunAndReturn(run func([]string) error) *TxnKV_
 func (_m *TxnKV) MultiSave(kvs map[string]string) error {
 	ret := _m.Called(kvs)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSave")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string) error); ok {
 		r0 = rf(kvs)
@@ -420,6 +448,10 @@ func (_m *TxnKV) MultiSaveAndRemove(saves map[string]string, removals []string,
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSaveAndRemove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok {
 		r0 = rf(saves, removals, preds...)
@@ -478,6 +510,10 @@ func (_m *TxnKV) MultiSaveAndRemoveWithPrefix(saves map[string]string, removals
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSaveAndRemoveWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok {
 		r0 = rf(saves, removals, preds...)
@@ -529,6 +565,10 @@ func (_c *TxnKV_MultiSaveAndRemoveWithPrefix_Call) RunAndReturn(run func(map[str
 func (_m *TxnKV) Remove(key string) error {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Remove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(key)
@@ -571,6 +611,10 @@ func (_c *TxnKV_Remove_Call) RunAndReturn(run func(string) error) *TxnKV_Remove_
 func (_m *TxnKV) RemoveWithPrefix(key string) error {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemoveWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(key)
@@ -613,6 +657,10 @@ func (_c *TxnKV_RemoveWithPrefix_Call) RunAndReturn(run func(string) error) *Txn
 func (_m *TxnKV) Save(key string, value string) error {
 	ret := _m.Called(key, value)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Save")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, string) error); ok {
 		r0 = rf(key, value)

internal/kv/mocks/watch_kv.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -59,6 +59,10 @@ func (_c *WatchKV_Close_Call) RunAndReturn(run func()) *WatchKV_Close_Call {
 func (_m *WatchKV) CompareVersionAndSwap(key string, version int64, target string) (bool, error) {
 	ret := _m.Called(key, version, target)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CompareVersionAndSwap")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string, int64, string) (bool, error)); ok {
@@ -113,6 +117,10 @@ func (_c *WatchKV_CompareVersionAndSwap_Call) RunAndReturn(run func(string, int6
 func (_m *WatchKV) GetPath(key string) string {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPath")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func(string) string); ok {
 		r0 = rf(key)
@@ -155,6 +163,10 @@ func (_c *WatchKV_GetPath_Call) RunAndReturn(run func(string) string) *WatchKV_G
 func (_m *WatchKV) Has(key string) (bool, error) {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Has")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (bool, error)); ok {
@@ -207,6 +219,10 @@ func (_c *WatchKV_Has_Call) RunAndReturn(run func(string) (bool, error)) *WatchK
 func (_m *WatchKV) HasPrefix(prefix string) (bool, error) {
 	ret := _m.Called(prefix)
 
+	if len(ret) == 0 {
+		panic("no return value specified for HasPrefix")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (bool, error)); ok {
@@ -259,6 +275,10 @@ func (_c *WatchKV_HasPrefix_Call) RunAndReturn(run func(string) (bool, error)) *
 func (_m *WatchKV) Load(key string) (string, error) {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Load")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (string, error)); ok {
@@ -311,6 +331,10 @@ func (_c *WatchKV_Load_Call) RunAndReturn(run func(string) (string, error)) *Wat
 func (_m *WatchKV) LoadWithPrefix(key string) ([]string, []string, error) {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadWithPrefix")
+	}
+
 	var r0 []string
 	var r1 []string
 	var r2 error
@@ -374,6 +398,10 @@ func (_c *WatchKV_LoadWithPrefix_Call) RunAndReturn(run func(string) ([]string,
 func (_m *WatchKV) MultiLoad(keys []string) ([]string, error) {
 	ret := _m.Called(keys)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiLoad")
+	}
+
 	var r0 []string
 	var r1 error
 	if rf, ok := ret.Get(0).(func([]string) ([]string, error)); ok {
@@ -428,6 +456,10 @@ func (_c *WatchKV_MultiLoad_Call) RunAndReturn(run func([]string) ([]string, err
 func (_m *WatchKV) MultiRemove(keys []string) error {
 	ret := _m.Called(keys)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiRemove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func([]string) error); ok {
 		r0 = rf(keys)
@@ -470,6 +502,10 @@ func (_c *WatchKV_MultiRemove_Call) RunAndReturn(run func([]string) error) *Watc
 func (_m *WatchKV) MultiSave(kvs map[string]string) error {
 	ret := _m.Called(kvs)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSave")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string) error); ok {
 		r0 = rf(kvs)
@@ -519,6 +555,10 @@ func (_m *WatchKV) MultiSaveAndRemove(saves map[string]string, removals []string
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSaveAndRemove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok {
 		r0 = rf(saves, removals, preds...)
@@ -577,6 +617,10 @@ func (_m *WatchKV) MultiSaveAndRemoveWithPrefix(saves map[string]string, removal
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSaveAndRemoveWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok {
 		r0 = rf(saves, removals, preds...)
@@ -628,6 +672,10 @@ func (_c *WatchKV_MultiSaveAndRemoveWithPrefix_Call) RunAndReturn(run func(map[s
 func (_m *WatchKV) Remove(key string) error {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Remove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(key)
@@ -670,6 +718,10 @@ func (_c *WatchKV_Remove_Call) RunAndReturn(run func(string) error) *WatchKV_Rem
 func (_m *WatchKV) RemoveWithPrefix(key string) error {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemoveWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(key)
@@ -712,6 +764,10 @@ func (_c *WatchKV_RemoveWithPrefix_Call) RunAndReturn(run func(string) error) *W
 func (_m *WatchKV) Save(key string, value string) error {
 	ret := _m.Called(key, value)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Save")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, string) error); ok {
 		r0 = rf(key, value)
@@ -755,6 +811,10 @@ func (_c *WatchKV_Save_Call) RunAndReturn(run func(string, string) error) *Watch
 func (_m *WatchKV) WalkWithPrefix(prefix string, paginationSize int, fn func([]byte, []byte) error) error {
 	ret := _m.Called(prefix, paginationSize, fn)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WalkWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, int, func([]byte, []byte) error) error); ok {
 		r0 = rf(prefix, paginationSize, fn)
@@ -799,6 +859,10 @@ func (_c *WatchKV_WalkWithPrefix_Call) RunAndReturn(run func(string, int, func([
 func (_m *WatchKV) Watch(key string) clientv3.WatchChan {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Watch")
+	}
+
 	var r0 clientv3.WatchChan
 	if rf, ok := ret.Get(0).(func(string) clientv3.WatchChan); ok {
 		r0 = rf(key)
@@ -843,6 +907,10 @@ func (_c *WatchKV_Watch_Call) RunAndReturn(run func(string) clientv3.WatchChan)
 func (_m *WatchKV) WatchWithPrefix(key string) clientv3.WatchChan {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WatchWithPrefix")
+	}
+
 	var r0 clientv3.WatchChan
 	if rf, ok := ret.Get(0).(func(string) clientv3.WatchChan); ok {
 		r0 = rf(key)
@@ -887,6 +955,10 @@ func (_c *WatchKV_WatchWithPrefix_Call) RunAndReturn(run func(string) clientv3.W
 func (_m *WatchKV) WatchWithRevision(key string, revision int64) clientv3.WatchChan {
 	ret := _m.Called(key, revision)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WatchWithRevision")
+	}
+
 	var r0 clientv3.WatchChan
 	if rf, ok := ret.Get(0).(func(string, int64) clientv3.WatchChan); ok {
 		r0 = rf(key, revision)

internal/mocks/distributed/mock_streaming/mock_WALAccesser.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_streaming
 
@@ -37,6 +37,10 @@ func (_m *MockWALAccesser) AppendMessages(ctx context.Context, msgs ...message.M
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		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 {
 		r0 = rf(ctx, msgs...)
@@ -94,6 +98,10 @@ func (_m *MockWALAccesser) AppendMessagesWithOption(ctx context.Context, opts st
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AppendMessagesWithOption")
+	}
+
 	var r0 streaming.AppendResponses
 	if rf, ok := ret.Get(0).(func(context.Context, streaming.AppendOption, ...message.MutableMessage) streaming.AppendResponses); ok {
 		r0 = rf(ctx, opts, msgs...)
@@ -152,6 +160,10 @@ func (_m *MockWALAccesser) RawAppend(ctx context.Context, msgs message.MutableMe
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RawAppend")
+	}
+
 	var r0 *types.AppendResult
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, message.MutableMessage, ...streaming.AppendOption) (*types.AppendResult, error)); ok {
@@ -215,6 +227,10 @@ func (_c *MockWALAccesser_RawAppend_Call) RunAndReturn(run func(context.Context,
 func (_m *MockWALAccesser) Read(ctx context.Context, opts streaming.ReadOption) streaming.Scanner {
 	ret := _m.Called(ctx, opts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Read")
+	}
+
 	var r0 streaming.Scanner
 	if rf, ok := ret.Get(0).(func(context.Context, streaming.ReadOption) streaming.Scanner); ok {
 		r0 = rf(ctx, opts)
@@ -260,6 +276,10 @@ func (_c *MockWALAccesser_Read_Call) RunAndReturn(run func(context.Context, stre
 func (_m *MockWALAccesser) Txn(ctx context.Context, opts streaming.TxnOption) (streaming.Txn, error) {
 	ret := _m.Called(ctx, opts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Txn")
+	}
+
 	var r0 streaming.Txn
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, streaming.TxnOption) (streaming.Txn, error)); ok {

internal/mocks/google.golang.org/grpc/mock_balancer/mock_SubConn.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_balancer
 
@@ -58,6 +58,10 @@ func (_c *MockSubConn_Connect_Call) RunAndReturn(run func()) *MockSubConn_Connec
 func (_m *MockSubConn) GetOrBuildProducer(_a0 balancer.ProducerBuilder) (balancer.Producer, func()) {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetOrBuildProducer")
+	}
+
 	var r0 balancer.Producer
 	var r1 func()
 	if rf, ok := ret.Get(0).(func(balancer.ProducerBuilder) (balancer.Producer, func())); ok {

internal/mocks/google.golang.org/grpc/mock_resolver/mock_ClientConn.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_resolver
 
@@ -59,6 +59,10 @@ func (_c *MockClientConn_NewAddress_Call) RunAndReturn(run func([]resolver.Addre
 func (_m *MockClientConn) ParseServiceConfig(serviceConfigJSON string) *serviceconfig.ParseResult {
 	ret := _m.Called(serviceConfigJSON)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ParseServiceConfig")
+	}
+
 	var r0 *serviceconfig.ParseResult
 	if rf, ok := ret.Get(0).(func(string) *serviceconfig.ParseResult); ok {
 		r0 = rf(serviceConfigJSON)
@@ -136,6 +140,10 @@ func (_c *MockClientConn_ReportError_Call) RunAndReturn(run func(error)) *MockCl
 func (_m *MockClientConn) UpdateState(_a0 resolver.State) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateState")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(resolver.State) error); ok {
 		r0 = rf(_a0)

internal/mocks/google.golang.org/mock_grpc/mock_ClientStream.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_grpc
 
@@ -27,6 +27,10 @@ func (_m *MockClientStream) EXPECT() *MockClientStream_Expecter {
 func (_m *MockClientStream) CloseSend() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for CloseSend")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -68,6 +72,10 @@ func (_c *MockClientStream_CloseSend_Call) RunAndReturn(run func() error) *MockC
 func (_m *MockClientStream) Context() context.Context {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Context")
+	}
+
 	var r0 context.Context
 	if rf, ok := ret.Get(0).(func() context.Context); ok {
 		r0 = rf()
@@ -111,6 +119,10 @@ func (_c *MockClientStream_Context_Call) RunAndReturn(run func() context.Context
 func (_m *MockClientStream) Header() (metadata.MD, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Header")
+	}
+
 	var r0 metadata.MD
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (metadata.MD, error)); ok {
@@ -164,6 +176,10 @@ func (_c *MockClientStream_Header_Call) RunAndReturn(run func() (metadata.MD, er
 func (_m *MockClientStream) RecvMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RecvMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -206,6 +222,10 @@ func (_c *MockClientStream_RecvMsg_Call) RunAndReturn(run func(interface{}) erro
 func (_m *MockClientStream) SendMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SendMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -248,6 +268,10 @@ func (_c *MockClientStream_SendMsg_Call) RunAndReturn(run func(interface{}) erro
 func (_m *MockClientStream) Trailer() metadata.MD {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Trailer")
+	}
+
 	var r0 metadata.MD
 	if rf, ok := ret.Get(0).(func() metadata.MD); ok {
 		r0 = rf()

internal/mocks/mock_datacoord.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -42,6 +42,10 @@ func (_m *MockDataCoord) EXPECT() *MockDataCoord_Expecter {
 func (_m *MockDataCoord) AllocSegment(_a0 context.Context, _a1 *datapb.AllocSegmentRequest) (*datapb.AllocSegmentResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocSegment")
+	}
+
 	var r0 *datapb.AllocSegmentResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.AllocSegmentRequest) (*datapb.AllocSegmentResponse, error)); ok {
@@ -97,6 +101,10 @@ func (_c *MockDataCoord_AllocSegment_Call) RunAndReturn(run func(context.Context
 func (_m *MockDataCoord) AlterIndex(_a0 context.Context, _a1 *indexpb.AlterIndexRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AlterIndex")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.AlterIndexRequest) (*commonpb.Status, error)); ok {
@@ -152,6 +160,10 @@ func (_c *MockDataCoord_AlterIndex_Call) RunAndReturn(run func(context.Context,
 func (_m *MockDataCoord) AssignSegmentID(_a0 context.Context, _a1 *datapb.AssignSegmentIDRequest) (*datapb.AssignSegmentIDResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AssignSegmentID")
+	}
+
 	var r0 *datapb.AssignSegmentIDResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.AssignSegmentIDRequest) (*datapb.AssignSegmentIDResponse, error)); ok {
@@ -207,6 +219,10 @@ func (_c *MockDataCoord_AssignSegmentID_Call) RunAndReturn(run func(context.Cont
 func (_m *MockDataCoord) BroadcastAlteredCollection(_a0 context.Context, _a1 *datapb.AlterCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for BroadcastAlteredCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.AlterCollectionRequest) (*commonpb.Status, error)); ok {
@@ -262,6 +278,10 @@ func (_c *MockDataCoord_BroadcastAlteredCollection_Call) RunAndReturn(run func(c
 func (_m *MockDataCoord) CheckHealth(_a0 context.Context, _a1 *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckHealth")
+	}
+
 	var r0 *milvuspb.CheckHealthResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error)); ok {
@@ -317,6 +337,10 @@ func (_c *MockDataCoord_CheckHealth_Call) RunAndReturn(run func(context.Context,
 func (_m *MockDataCoord) CreateIndex(_a0 context.Context, _a1 *indexpb.CreateIndexRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateIndex")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.CreateIndexRequest) (*commonpb.Status, error)); ok {
@@ -372,6 +396,10 @@ func (_c *MockDataCoord_CreateIndex_Call) RunAndReturn(run func(context.Context,
 func (_m *MockDataCoord) DescribeIndex(_a0 context.Context, _a1 *indexpb.DescribeIndexRequest) (*indexpb.DescribeIndexResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeIndex")
+	}
+
 	var r0 *indexpb.DescribeIndexResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.DescribeIndexRequest) (*indexpb.DescribeIndexResponse, error)); ok {
@@ -427,6 +455,10 @@ func (_c *MockDataCoord_DescribeIndex_Call) RunAndReturn(run func(context.Contex
 func (_m *MockDataCoord) DropIndex(_a0 context.Context, _a1 *indexpb.DropIndexRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropIndex")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.DropIndexRequest) (*commonpb.Status, error)); ok {
@@ -482,6 +514,10 @@ func (_c *MockDataCoord_DropIndex_Call) RunAndReturn(run func(context.Context, *
 func (_m *MockDataCoord) DropVirtualChannel(_a0 context.Context, _a1 *datapb.DropVirtualChannelRequest) (*datapb.DropVirtualChannelResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropVirtualChannel")
+	}
+
 	var r0 *datapb.DropVirtualChannelResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.DropVirtualChannelRequest) (*datapb.DropVirtualChannelResponse, error)); ok {
@@ -537,6 +573,10 @@ func (_c *MockDataCoord_DropVirtualChannel_Call) RunAndReturn(run func(context.C
 func (_m *MockDataCoord) Flush(_a0 context.Context, _a1 *datapb.FlushRequest) (*datapb.FlushResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Flush")
+	}
+
 	var r0 *datapb.FlushResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.FlushRequest) (*datapb.FlushResponse, error)); ok {
@@ -592,6 +632,10 @@ func (_c *MockDataCoord_Flush_Call) RunAndReturn(run func(context.Context, *data
 func (_m *MockDataCoord) GcConfirm(_a0 context.Context, _a1 *datapb.GcConfirmRequest) (*datapb.GcConfirmResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GcConfirm")
+	}
+
 	var r0 *datapb.GcConfirmResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GcConfirmRequest) (*datapb.GcConfirmResponse, error)); ok {
@@ -647,6 +691,10 @@ func (_c *MockDataCoord_GcConfirm_Call) RunAndReturn(run func(context.Context, *
 func (_m *MockDataCoord) GcControl(_a0 context.Context, _a1 *datapb.GcControlRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GcControl")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GcControlRequest) (*commonpb.Status, error)); ok {
@@ -702,6 +750,10 @@ func (_c *MockDataCoord_GcControl_Call) RunAndReturn(run func(context.Context, *
 func (_m *MockDataCoord) GetChannelRecoveryInfo(_a0 context.Context, _a1 *datapb.GetChannelRecoveryInfoRequest) (*datapb.GetChannelRecoveryInfoResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetChannelRecoveryInfo")
+	}
+
 	var r0 *datapb.GetChannelRecoveryInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetChannelRecoveryInfoRequest) (*datapb.GetChannelRecoveryInfoResponse, error)); ok {
@@ -757,6 +809,10 @@ func (_c *MockDataCoord_GetChannelRecoveryInfo_Call) RunAndReturn(run func(conte
 func (_m *MockDataCoord) GetCollectionStatistics(_a0 context.Context, _a1 *datapb.GetCollectionStatisticsRequest) (*datapb.GetCollectionStatisticsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollectionStatistics")
+	}
+
 	var r0 *datapb.GetCollectionStatisticsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetCollectionStatisticsRequest) (*datapb.GetCollectionStatisticsResponse, error)); ok {
@@ -812,6 +868,10 @@ func (_c *MockDataCoord_GetCollectionStatistics_Call) RunAndReturn(run func(cont
 func (_m *MockDataCoord) GetCompactionState(_a0 context.Context, _a1 *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCompactionState")
+	}
+
 	var r0 *milvuspb.GetCompactionStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error)); ok {
@@ -867,6 +927,10 @@ func (_c *MockDataCoord_GetCompactionState_Call) RunAndReturn(run func(context.C
 func (_m *MockDataCoord) GetCompactionStateWithPlans(_a0 context.Context, _a1 *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCompactionStateWithPlans")
+	}
+
 	var r0 *milvuspb.GetCompactionPlansResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error)); ok {
@@ -922,6 +986,10 @@ func (_c *MockDataCoord_GetCompactionStateWithPlans_Call) RunAndReturn(run func(
 func (_m *MockDataCoord) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok {
@@ -977,6 +1045,10 @@ func (_c *MockDataCoord_GetComponentStates_Call) RunAndReturn(run func(context.C
 func (_m *MockDataCoord) GetFlushAllState(_a0 context.Context, _a1 *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetFlushAllState")
+	}
+
 	var r0 *milvuspb.GetFlushAllStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error)); ok {
@@ -1032,6 +1104,10 @@ func (_c *MockDataCoord_GetFlushAllState_Call) RunAndReturn(run func(context.Con
 func (_m *MockDataCoord) GetFlushState(_a0 context.Context, _a1 *datapb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetFlushState")
+	}
+
 	var r0 *milvuspb.GetFlushStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error)); ok {
@@ -1087,6 +1163,10 @@ func (_c *MockDataCoord_GetFlushState_Call) RunAndReturn(run func(context.Contex
 func (_m *MockDataCoord) GetFlushedSegments(_a0 context.Context, _a1 *datapb.GetFlushedSegmentsRequest) (*datapb.GetFlushedSegmentsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetFlushedSegments")
+	}
+
 	var r0 *datapb.GetFlushedSegmentsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetFlushedSegmentsRequest) (*datapb.GetFlushedSegmentsResponse, error)); ok {
@@ -1142,6 +1222,10 @@ func (_c *MockDataCoord_GetFlushedSegments_Call) RunAndReturn(run func(context.C
 func (_m *MockDataCoord) GetImportProgress(_a0 context.Context, _a1 *internalpb.GetImportProgressRequest) (*internalpb.GetImportProgressResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetImportProgress")
+	}
+
 	var r0 *internalpb.GetImportProgressResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetImportProgressRequest) (*internalpb.GetImportProgressResponse, error)); ok {
@@ -1197,6 +1281,10 @@ func (_c *MockDataCoord_GetImportProgress_Call) RunAndReturn(run func(context.Co
 func (_m *MockDataCoord) GetIndexBuildProgress(_a0 context.Context, _a1 *indexpb.GetIndexBuildProgressRequest) (*indexpb.GetIndexBuildProgressResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetIndexBuildProgress")
+	}
+
 	var r0 *indexpb.GetIndexBuildProgressResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexBuildProgressRequest) (*indexpb.GetIndexBuildProgressResponse, error)); ok {
@@ -1252,6 +1340,10 @@ func (_c *MockDataCoord_GetIndexBuildProgress_Call) RunAndReturn(run func(contex
 func (_m *MockDataCoord) GetIndexInfos(_a0 context.Context, _a1 *indexpb.GetIndexInfoRequest) (*indexpb.GetIndexInfoResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetIndexInfos")
+	}
+
 	var r0 *indexpb.GetIndexInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexInfoRequest) (*indexpb.GetIndexInfoResponse, error)); ok {
@@ -1307,6 +1399,10 @@ func (_c *MockDataCoord_GetIndexInfos_Call) RunAndReturn(run func(context.Contex
 func (_m *MockDataCoord) GetIndexState(_a0 context.Context, _a1 *indexpb.GetIndexStateRequest) (*indexpb.GetIndexStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetIndexState")
+	}
+
 	var r0 *indexpb.GetIndexStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexStateRequest) (*indexpb.GetIndexStateResponse, error)); ok {
@@ -1362,6 +1458,10 @@ func (_c *MockDataCoord_GetIndexState_Call) RunAndReturn(run func(context.Contex
 func (_m *MockDataCoord) GetIndexStatistics(_a0 context.Context, _a1 *indexpb.GetIndexStatisticsRequest) (*indexpb.GetIndexStatisticsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetIndexStatistics")
+	}
+
 	var r0 *indexpb.GetIndexStatisticsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexStatisticsRequest) (*indexpb.GetIndexStatisticsResponse, error)); ok {
@@ -1417,6 +1517,10 @@ func (_c *MockDataCoord_GetIndexStatistics_Call) RunAndReturn(run func(context.C
 func (_m *MockDataCoord) GetInsertBinlogPaths(_a0 context.Context, _a1 *datapb.GetInsertBinlogPathsRequest) (*datapb.GetInsertBinlogPathsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetInsertBinlogPaths")
+	}
+
 	var r0 *datapb.GetInsertBinlogPathsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetInsertBinlogPathsRequest) (*datapb.GetInsertBinlogPathsResponse, error)); ok {
@@ -1472,6 +1576,10 @@ func (_c *MockDataCoord_GetInsertBinlogPaths_Call) RunAndReturn(run func(context
 func (_m *MockDataCoord) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -1527,6 +1635,10 @@ func (_c *MockDataCoord_GetMetrics_Call) RunAndReturn(run func(context.Context,
 func (_m *MockDataCoord) GetPartitionStatistics(_a0 context.Context, _a1 *datapb.GetPartitionStatisticsRequest) (*datapb.GetPartitionStatisticsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPartitionStatistics")
+	}
+
 	var r0 *datapb.GetPartitionStatisticsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetPartitionStatisticsRequest) (*datapb.GetPartitionStatisticsResponse, error)); ok {
@@ -1582,6 +1694,10 @@ func (_c *MockDataCoord_GetPartitionStatistics_Call) RunAndReturn(run func(conte
 func (_m *MockDataCoord) GetRecoveryInfo(_a0 context.Context, _a1 *datapb.GetRecoveryInfoRequest) (*datapb.GetRecoveryInfoResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetRecoveryInfo")
+	}
+
 	var r0 *datapb.GetRecoveryInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetRecoveryInfoRequest) (*datapb.GetRecoveryInfoResponse, error)); ok {
@@ -1637,6 +1753,10 @@ func (_c *MockDataCoord_GetRecoveryInfo_Call) RunAndReturn(run func(context.Cont
 func (_m *MockDataCoord) GetRecoveryInfoV2(_a0 context.Context, _a1 *datapb.GetRecoveryInfoRequestV2) (*datapb.GetRecoveryInfoResponseV2, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetRecoveryInfoV2")
+	}
+
 	var r0 *datapb.GetRecoveryInfoResponseV2
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetRecoveryInfoRequestV2) (*datapb.GetRecoveryInfoResponseV2, error)); ok {
@@ -1692,6 +1812,10 @@ func (_c *MockDataCoord_GetRecoveryInfoV2_Call) RunAndReturn(run func(context.Co
 func (_m *MockDataCoord) GetSegmentIndexState(_a0 context.Context, _a1 *indexpb.GetSegmentIndexStateRequest) (*indexpb.GetSegmentIndexStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentIndexState")
+	}
+
 	var r0 *indexpb.GetSegmentIndexStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetSegmentIndexStateRequest) (*indexpb.GetSegmentIndexStateResponse, error)); ok {
@@ -1747,6 +1871,10 @@ func (_c *MockDataCoord_GetSegmentIndexState_Call) RunAndReturn(run func(context
 func (_m *MockDataCoord) GetSegmentInfo(_a0 context.Context, _a1 *datapb.GetSegmentInfoRequest) (*datapb.GetSegmentInfoResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentInfo")
+	}
+
 	var r0 *datapb.GetSegmentInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentInfoRequest) (*datapb.GetSegmentInfoResponse, error)); ok {
@@ -1802,6 +1930,10 @@ func (_c *MockDataCoord_GetSegmentInfo_Call) RunAndReturn(run func(context.Conte
 func (_m *MockDataCoord) GetSegmentInfoChannel(_a0 context.Context, _a1 *datapb.GetSegmentInfoChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentInfoChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentInfoChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -1857,6 +1989,10 @@ func (_c *MockDataCoord_GetSegmentInfoChannel_Call) RunAndReturn(run func(contex
 func (_m *MockDataCoord) GetSegmentStates(_a0 context.Context, _a1 *datapb.GetSegmentStatesRequest) (*datapb.GetSegmentStatesResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentStates")
+	}
+
 	var r0 *datapb.GetSegmentStatesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentStatesRequest) (*datapb.GetSegmentStatesResponse, error)); ok {
@@ -1912,6 +2048,10 @@ func (_c *MockDataCoord_GetSegmentStates_Call) RunAndReturn(run func(context.Con
 func (_m *MockDataCoord) GetSegmentsByStates(_a0 context.Context, _a1 *datapb.GetSegmentsByStatesRequest) (*datapb.GetSegmentsByStatesResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentsByStates")
+	}
+
 	var r0 *datapb.GetSegmentsByStatesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentsByStatesRequest) (*datapb.GetSegmentsByStatesResponse, error)); ok {
@@ -1967,6 +2107,10 @@ func (_c *MockDataCoord_GetSegmentsByStates_Call) RunAndReturn(run func(context.
 func (_m *MockDataCoord) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -2022,6 +2166,10 @@ func (_c *MockDataCoord_GetStatisticsChannel_Call) RunAndReturn(run func(context
 func (_m *MockDataCoord) GetTimeTickChannel(_a0 context.Context, _a1 *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetTimeTickChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -2077,6 +2225,10 @@ func (_c *MockDataCoord_GetTimeTickChannel_Call) RunAndReturn(run func(context.C
 func (_m *MockDataCoord) ImportV2(_a0 context.Context, _a1 *internalpb.ImportRequestInternal) (*internalpb.ImportResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ImportV2")
+	}
+
 	var r0 *internalpb.ImportResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ImportRequestInternal) (*internalpb.ImportResponse, error)); ok {
@@ -2132,6 +2284,10 @@ func (_c *MockDataCoord_ImportV2_Call) RunAndReturn(run func(context.Context, *i
 func (_m *MockDataCoord) Init() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Init")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -2173,6 +2329,10 @@ func (_c *MockDataCoord_Init_Call) RunAndReturn(run func() error) *MockDataCoord
 func (_m *MockDataCoord) ListImports(_a0 context.Context, _a1 *internalpb.ListImportsRequestInternal) (*internalpb.ListImportsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListImports")
+	}
+
 	var r0 *internalpb.ListImportsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequestInternal) (*internalpb.ListImportsResponse, error)); ok {
@@ -2228,6 +2388,10 @@ func (_c *MockDataCoord_ListImports_Call) RunAndReturn(run func(context.Context,
 func (_m *MockDataCoord) ListIndexes(_a0 context.Context, _a1 *indexpb.ListIndexesRequest) (*indexpb.ListIndexesResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListIndexes")
+	}
+
 	var r0 *indexpb.ListIndexesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.ListIndexesRequest) (*indexpb.ListIndexesResponse, error)); ok {
@@ -2283,6 +2447,10 @@ func (_c *MockDataCoord_ListIndexes_Call) RunAndReturn(run func(context.Context,
 func (_m *MockDataCoord) ManualCompaction(_a0 context.Context, _a1 *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ManualCompaction")
+	}
+
 	var r0 *milvuspb.ManualCompactionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error)); ok {
@@ -2338,6 +2506,10 @@ func (_c *MockDataCoord_ManualCompaction_Call) RunAndReturn(run func(context.Con
 func (_m *MockDataCoord) MarkSegmentsDropped(_a0 context.Context, _a1 *datapb.MarkSegmentsDroppedRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MarkSegmentsDropped")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.MarkSegmentsDroppedRequest) (*commonpb.Status, error)); ok {
@@ -2393,6 +2565,10 @@ func (_c *MockDataCoord_MarkSegmentsDropped_Call) RunAndReturn(run func(context.
 func (_m *MockDataCoord) Register() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Register")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -2467,6 +2643,10 @@ func (_c *MockDataCoord_RegisterStreamingCoordGRPCService_Call) RunAndReturn(run
 func (_m *MockDataCoord) ReportDataNodeTtMsgs(_a0 context.Context, _a1 *datapb.ReportDataNodeTtMsgsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReportDataNodeTtMsgs")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.ReportDataNodeTtMsgsRequest) (*commonpb.Status, error)); ok {
@@ -2522,6 +2702,10 @@ func (_c *MockDataCoord_ReportDataNodeTtMsgs_Call) RunAndReturn(run func(context
 func (_m *MockDataCoord) SaveBinlogPaths(_a0 context.Context, _a1 *datapb.SaveBinlogPathsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SaveBinlogPaths")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.SaveBinlogPathsRequest) (*commonpb.Status, error)); ok {
@@ -2742,6 +2926,10 @@ func (_c *MockDataCoord_SetRootCoordClient_Call) RunAndReturn(run func(types.Roo
 func (_m *MockDataCoord) SetSegmentState(_a0 context.Context, _a1 *datapb.SetSegmentStateRequest) (*datapb.SetSegmentStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetSegmentState")
+	}
+
 	var r0 *datapb.SetSegmentStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.SetSegmentStateRequest) (*datapb.SetSegmentStateResponse, error)); ok {
@@ -2830,6 +3018,10 @@ func (_c *MockDataCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client)
 func (_m *MockDataCoord) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok {
@@ -2885,6 +3077,10 @@ func (_c *MockDataCoord_ShowConfigurations_Call) RunAndReturn(run func(context.C
 func (_m *MockDataCoord) Start() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Start")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -2926,6 +3122,10 @@ func (_c *MockDataCoord_Start_Call) RunAndReturn(run func() error) *MockDataCoor
 func (_m *MockDataCoord) Stop() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Stop")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -2967,6 +3167,10 @@ func (_c *MockDataCoord_Stop_Call) RunAndReturn(run func() error) *MockDataCoord
 func (_m *MockDataCoord) UpdateChannelCheckpoint(_a0 context.Context, _a1 *datapb.UpdateChannelCheckpointRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateChannelCheckpoint")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.UpdateChannelCheckpointRequest) (*commonpb.Status, error)); ok {
@@ -3022,6 +3226,10 @@ func (_c *MockDataCoord_UpdateChannelCheckpoint_Call) RunAndReturn(run func(cont
 func (_m *MockDataCoord) UpdateSegmentStatistics(_a0 context.Context, _a1 *datapb.UpdateSegmentStatisticsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateSegmentStatistics")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.UpdateSegmentStatisticsRequest) (*commonpb.Status, error)); ok {
@@ -3077,6 +3285,10 @@ func (_c *MockDataCoord_UpdateSegmentStatistics_Call) RunAndReturn(run func(cont
 func (_m *MockDataCoord) WatchChannels(_a0 context.Context, _a1 *datapb.WatchChannelsRequest) (*datapb.WatchChannelsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WatchChannels")
+	}
+
 	var r0 *datapb.WatchChannelsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.WatchChannelsRequest) (*datapb.WatchChannelsResponse, error)); ok {

internal/mocks/mock_datacoord_client.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -44,6 +44,10 @@ func (_m *MockDataCoordClient) AllocSegment(ctx context.Context, in *datapb.Allo
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocSegment")
+	}
+
 	var r0 *datapb.AllocSegmentResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.AllocSegmentRequest, ...grpc.CallOption) (*datapb.AllocSegmentResponse, error)); ok {
@@ -114,6 +118,10 @@ func (_m *MockDataCoordClient) AlterIndex(ctx context.Context, in *indexpb.Alter
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AlterIndex")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.AlterIndexRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -184,6 +192,10 @@ func (_m *MockDataCoordClient) AssignSegmentID(ctx context.Context, in *datapb.A
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AssignSegmentID")
+	}
+
 	var r0 *datapb.AssignSegmentIDResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.AssignSegmentIDRequest, ...grpc.CallOption) (*datapb.AssignSegmentIDResponse, error)); ok {
@@ -254,6 +266,10 @@ func (_m *MockDataCoordClient) BroadcastAlteredCollection(ctx context.Context, i
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for BroadcastAlteredCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.AlterCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -324,6 +340,10 @@ func (_m *MockDataCoordClient) CheckHealth(ctx context.Context, in *milvuspb.Che
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckHealth")
+	}
+
 	var r0 *milvuspb.CheckHealthResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest, ...grpc.CallOption) (*milvuspb.CheckHealthResponse, error)); ok {
@@ -387,6 +407,10 @@ func (_c *MockDataCoordClient_CheckHealth_Call) RunAndReturn(run func(context.Co
 func (_m *MockDataCoordClient) Close() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Close")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -435,6 +459,10 @@ func (_m *MockDataCoordClient) CreateIndex(ctx context.Context, in *indexpb.Crea
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateIndex")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.CreateIndexRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -505,6 +533,10 @@ func (_m *MockDataCoordClient) DescribeIndex(ctx context.Context, in *indexpb.De
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeIndex")
+	}
+
 	var r0 *indexpb.DescribeIndexResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.DescribeIndexRequest, ...grpc.CallOption) (*indexpb.DescribeIndexResponse, error)); ok {
@@ -575,6 +607,10 @@ func (_m *MockDataCoordClient) DropIndex(ctx context.Context, in *indexpb.DropIn
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropIndex")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.DropIndexRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -645,6 +681,10 @@ func (_m *MockDataCoordClient) DropVirtualChannel(ctx context.Context, in *datap
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropVirtualChannel")
+	}
+
 	var r0 *datapb.DropVirtualChannelResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.DropVirtualChannelRequest, ...grpc.CallOption) (*datapb.DropVirtualChannelResponse, error)); ok {
@@ -715,6 +755,10 @@ func (_m *MockDataCoordClient) Flush(ctx context.Context, in *datapb.FlushReques
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Flush")
+	}
+
 	var r0 *datapb.FlushResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.FlushRequest, ...grpc.CallOption) (*datapb.FlushResponse, error)); ok {
@@ -785,6 +829,10 @@ func (_m *MockDataCoordClient) GcConfirm(ctx context.Context, in *datapb.GcConfi
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GcConfirm")
+	}
+
 	var r0 *datapb.GcConfirmResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GcConfirmRequest, ...grpc.CallOption) (*datapb.GcConfirmResponse, error)); ok {
@@ -855,6 +903,10 @@ func (_m *MockDataCoordClient) GcControl(ctx context.Context, in *datapb.GcContr
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GcControl")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GcControlRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -925,6 +977,10 @@ func (_m *MockDataCoordClient) GetChannelRecoveryInfo(ctx context.Context, in *d
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetChannelRecoveryInfo")
+	}
+
 	var r0 *datapb.GetChannelRecoveryInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetChannelRecoveryInfoRequest, ...grpc.CallOption) (*datapb.GetChannelRecoveryInfoResponse, error)); ok {
@@ -995,6 +1051,10 @@ func (_m *MockDataCoordClient) GetCollectionStatistics(ctx context.Context, in *
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollectionStatistics")
+	}
+
 	var r0 *datapb.GetCollectionStatisticsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetCollectionStatisticsRequest, ...grpc.CallOption) (*datapb.GetCollectionStatisticsResponse, error)); ok {
@@ -1065,6 +1125,10 @@ func (_m *MockDataCoordClient) GetCompactionState(ctx context.Context, in *milvu
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCompactionState")
+	}
+
 	var r0 *milvuspb.GetCompactionStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionStateRequest, ...grpc.CallOption) (*milvuspb.GetCompactionStateResponse, error)); ok {
@@ -1135,6 +1199,10 @@ func (_m *MockDataCoordClient) GetCompactionStateWithPlans(ctx context.Context,
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCompactionStateWithPlans")
+	}
+
 	var r0 *milvuspb.GetCompactionPlansResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionPlansRequest, ...grpc.CallOption) (*milvuspb.GetCompactionPlansResponse, error)); ok {
@@ -1205,6 +1273,10 @@ func (_m *MockDataCoordClient) GetComponentStates(ctx context.Context, in *milvu
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok {
@@ -1275,6 +1347,10 @@ func (_m *MockDataCoordClient) GetFlushAllState(ctx context.Context, in *milvusp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetFlushAllState")
+	}
+
 	var r0 *milvuspb.GetFlushAllStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushAllStateRequest, ...grpc.CallOption) (*milvuspb.GetFlushAllStateResponse, error)); ok {
@@ -1345,6 +1421,10 @@ func (_m *MockDataCoordClient) GetFlushState(ctx context.Context, in *datapb.Get
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetFlushState")
+	}
+
 	var r0 *milvuspb.GetFlushStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetFlushStateRequest, ...grpc.CallOption) (*milvuspb.GetFlushStateResponse, error)); ok {
@@ -1415,6 +1495,10 @@ func (_m *MockDataCoordClient) GetFlushedSegments(ctx context.Context, in *datap
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetFlushedSegments")
+	}
+
 	var r0 *datapb.GetFlushedSegmentsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetFlushedSegmentsRequest, ...grpc.CallOption) (*datapb.GetFlushedSegmentsResponse, error)); ok {
@@ -1485,6 +1569,10 @@ func (_m *MockDataCoordClient) GetImportProgress(ctx context.Context, in *intern
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetImportProgress")
+	}
+
 	var r0 *internalpb.GetImportProgressResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetImportProgressRequest, ...grpc.CallOption) (*internalpb.GetImportProgressResponse, error)); ok {
@@ -1555,6 +1643,10 @@ func (_m *MockDataCoordClient) GetIndexBuildProgress(ctx context.Context, in *in
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetIndexBuildProgress")
+	}
+
 	var r0 *indexpb.GetIndexBuildProgressResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexBuildProgressRequest, ...grpc.CallOption) (*indexpb.GetIndexBuildProgressResponse, error)); ok {
@@ -1625,6 +1717,10 @@ func (_m *MockDataCoordClient) GetIndexInfos(ctx context.Context, in *indexpb.Ge
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetIndexInfos")
+	}
+
 	var r0 *indexpb.GetIndexInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexInfoRequest, ...grpc.CallOption) (*indexpb.GetIndexInfoResponse, error)); ok {
@@ -1695,6 +1791,10 @@ func (_m *MockDataCoordClient) GetIndexState(ctx context.Context, in *indexpb.Ge
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetIndexState")
+	}
+
 	var r0 *indexpb.GetIndexStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexStateRequest, ...grpc.CallOption) (*indexpb.GetIndexStateResponse, error)); ok {
@@ -1765,6 +1865,10 @@ func (_m *MockDataCoordClient) GetIndexStatistics(ctx context.Context, in *index
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetIndexStatistics")
+	}
+
 	var r0 *indexpb.GetIndexStatisticsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexStatisticsRequest, ...grpc.CallOption) (*indexpb.GetIndexStatisticsResponse, error)); ok {
@@ -1835,6 +1939,10 @@ func (_m *MockDataCoordClient) GetInsertBinlogPaths(ctx context.Context, in *dat
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetInsertBinlogPaths")
+	}
+
 	var r0 *datapb.GetInsertBinlogPathsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetInsertBinlogPathsRequest, ...grpc.CallOption) (*datapb.GetInsertBinlogPathsResponse, error)); ok {
@@ -1905,6 +2013,10 @@ func (_m *MockDataCoordClient) GetMetrics(ctx context.Context, in *milvuspb.GetM
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -1975,6 +2087,10 @@ func (_m *MockDataCoordClient) GetPartitionStatistics(ctx context.Context, in *d
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPartitionStatistics")
+	}
+
 	var r0 *datapb.GetPartitionStatisticsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetPartitionStatisticsRequest, ...grpc.CallOption) (*datapb.GetPartitionStatisticsResponse, error)); ok {
@@ -2045,6 +2161,10 @@ func (_m *MockDataCoordClient) GetRecoveryInfo(ctx context.Context, in *datapb.G
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetRecoveryInfo")
+	}
+
 	var r0 *datapb.GetRecoveryInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetRecoveryInfoRequest, ...grpc.CallOption) (*datapb.GetRecoveryInfoResponse, error)); ok {
@@ -2115,6 +2235,10 @@ func (_m *MockDataCoordClient) GetRecoveryInfoV2(ctx context.Context, in *datapb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetRecoveryInfoV2")
+	}
+
 	var r0 *datapb.GetRecoveryInfoResponseV2
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetRecoveryInfoRequestV2, ...grpc.CallOption) (*datapb.GetRecoveryInfoResponseV2, error)); ok {
@@ -2185,6 +2309,10 @@ func (_m *MockDataCoordClient) GetSegmentIndexState(ctx context.Context, in *ind
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentIndexState")
+	}
+
 	var r0 *indexpb.GetSegmentIndexStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetSegmentIndexStateRequest, ...grpc.CallOption) (*indexpb.GetSegmentIndexStateResponse, error)); ok {
@@ -2255,6 +2383,10 @@ func (_m *MockDataCoordClient) GetSegmentInfo(ctx context.Context, in *datapb.Ge
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentInfo")
+	}
+
 	var r0 *datapb.GetSegmentInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentInfoRequest, ...grpc.CallOption) (*datapb.GetSegmentInfoResponse, error)); ok {
@@ -2325,6 +2457,10 @@ func (_m *MockDataCoordClient) GetSegmentInfoChannel(ctx context.Context, in *da
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentInfoChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentInfoChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -2395,6 +2531,10 @@ func (_m *MockDataCoordClient) GetSegmentStates(ctx context.Context, in *datapb.
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentStates")
+	}
+
 	var r0 *datapb.GetSegmentStatesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentStatesRequest, ...grpc.CallOption) (*datapb.GetSegmentStatesResponse, error)); ok {
@@ -2465,6 +2605,10 @@ func (_m *MockDataCoordClient) GetSegmentsByStates(ctx context.Context, in *data
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentsByStates")
+	}
+
 	var r0 *datapb.GetSegmentsByStatesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentsByStatesRequest, ...grpc.CallOption) (*datapb.GetSegmentsByStatesResponse, error)); ok {
@@ -2535,6 +2679,10 @@ func (_m *MockDataCoordClient) GetStatisticsChannel(ctx context.Context, in *int
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -2605,6 +2753,10 @@ func (_m *MockDataCoordClient) GetTimeTickChannel(ctx context.Context, in *inter
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetTimeTickChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -2675,6 +2827,10 @@ func (_m *MockDataCoordClient) ImportV2(ctx context.Context, in *internalpb.Impo
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ImportV2")
+	}
+
 	var r0 *internalpb.ImportResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ImportRequestInternal, ...grpc.CallOption) (*internalpb.ImportResponse, error)); ok {
@@ -2745,6 +2901,10 @@ func (_m *MockDataCoordClient) ListImports(ctx context.Context, in *internalpb.L
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListImports")
+	}
+
 	var r0 *internalpb.ListImportsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequestInternal, ...grpc.CallOption) (*internalpb.ListImportsResponse, error)); ok {
@@ -2815,6 +2975,10 @@ func (_m *MockDataCoordClient) ListIndexes(ctx context.Context, in *indexpb.List
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListIndexes")
+	}
+
 	var r0 *indexpb.ListIndexesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *indexpb.ListIndexesRequest, ...grpc.CallOption) (*indexpb.ListIndexesResponse, error)); ok {
@@ -2885,6 +3049,10 @@ func (_m *MockDataCoordClient) ManualCompaction(ctx context.Context, in *milvusp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ManualCompaction")
+	}
+
 	var r0 *milvuspb.ManualCompactionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ManualCompactionRequest, ...grpc.CallOption) (*milvuspb.ManualCompactionResponse, error)); ok {
@@ -2955,6 +3123,10 @@ func (_m *MockDataCoordClient) MarkSegmentsDropped(ctx context.Context, in *data
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MarkSegmentsDropped")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.MarkSegmentsDroppedRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -3025,6 +3197,10 @@ func (_m *MockDataCoordClient) ReportDataNodeTtMsgs(ctx context.Context, in *dat
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReportDataNodeTtMsgs")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.ReportDataNodeTtMsgsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -3095,6 +3271,10 @@ func (_m *MockDataCoordClient) SaveBinlogPaths(ctx context.Context, in *datapb.S
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SaveBinlogPaths")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.SaveBinlogPathsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -3165,6 +3345,10 @@ func (_m *MockDataCoordClient) SetSegmentState(ctx context.Context, in *datapb.S
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetSegmentState")
+	}
+
 	var r0 *datapb.SetSegmentStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.SetSegmentStateRequest, ...grpc.CallOption) (*datapb.SetSegmentStateResponse, error)); ok {
@@ -3235,6 +3419,10 @@ func (_m *MockDataCoordClient) ShowConfigurations(ctx context.Context, in *inter
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest, ...grpc.CallOption) (*internalpb.ShowConfigurationsResponse, error)); ok {
@@ -3305,6 +3493,10 @@ func (_m *MockDataCoordClient) UpdateChannelCheckpoint(ctx context.Context, in *
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateChannelCheckpoint")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.UpdateChannelCheckpointRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -3375,6 +3567,10 @@ func (_m *MockDataCoordClient) UpdateSegmentStatistics(ctx context.Context, in *
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateSegmentStatistics")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.UpdateSegmentStatisticsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -3445,6 +3641,10 @@ func (_m *MockDataCoordClient) WatchChannels(ctx context.Context, in *datapb.Wat
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WatchChannels")
+	}
+
 	var r0 *datapb.WatchChannelsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.WatchChannelsRequest, ...grpc.CallOption) (*datapb.WatchChannelsResponse, error)); ok {

internal/mocks/mock_datanode.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -36,6 +36,10 @@ func (_m *MockDataNode) EXPECT() *MockDataNode_Expecter {
 func (_m *MockDataNode) CheckChannelOperationProgress(_a0 context.Context, _a1 *datapb.ChannelWatchInfo) (*datapb.ChannelOperationProgressResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckChannelOperationProgress")
+	}
+
 	var r0 *datapb.ChannelOperationProgressResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.ChannelWatchInfo) (*datapb.ChannelOperationProgressResponse, error)); ok {
@@ -91,6 +95,10 @@ func (_c *MockDataNode_CheckChannelOperationProgress_Call) RunAndReturn(run func
 func (_m *MockDataNode) CompactionV2(_a0 context.Context, _a1 *datapb.CompactionPlan) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CompactionV2")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.CompactionPlan) (*commonpb.Status, error)); ok {
@@ -146,6 +154,10 @@ func (_c *MockDataNode_CompactionV2_Call) RunAndReturn(run func(context.Context,
 func (_m *MockDataNode) DropCompactionPlan(_a0 context.Context, _a1 *datapb.DropCompactionPlanRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropCompactionPlan")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.DropCompactionPlanRequest) (*commonpb.Status, error)); ok {
@@ -201,6 +213,10 @@ func (_c *MockDataNode_DropCompactionPlan_Call) RunAndReturn(run func(context.Co
 func (_m *MockDataNode) DropImport(_a0 context.Context, _a1 *datapb.DropImportRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropImport")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.DropImportRequest) (*commonpb.Status, error)); ok {
@@ -256,6 +272,10 @@ func (_c *MockDataNode_DropImport_Call) RunAndReturn(run func(context.Context, *
 func (_m *MockDataNode) FlushChannels(_a0 context.Context, _a1 *datapb.FlushChannelsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for FlushChannels")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.FlushChannelsRequest) (*commonpb.Status, error)); ok {
@@ -311,6 +331,10 @@ func (_c *MockDataNode_FlushChannels_Call) RunAndReturn(run func(context.Context
 func (_m *MockDataNode) FlushSegments(_a0 context.Context, _a1 *datapb.FlushSegmentsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for FlushSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.FlushSegmentsRequest) (*commonpb.Status, error)); ok {
@@ -366,6 +390,10 @@ func (_c *MockDataNode_FlushSegments_Call) RunAndReturn(run func(context.Context
 func (_m *MockDataNode) GetAddress() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAddress")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()
@@ -407,6 +435,10 @@ func (_c *MockDataNode_GetAddress_Call) RunAndReturn(run func() string) *MockDat
 func (_m *MockDataNode) GetCompactionState(_a0 context.Context, _a1 *datapb.CompactionStateRequest) (*datapb.CompactionStateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCompactionState")
+	}
+
 	var r0 *datapb.CompactionStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.CompactionStateRequest) (*datapb.CompactionStateResponse, error)); ok {
@@ -462,6 +494,10 @@ func (_c *MockDataNode_GetCompactionState_Call) RunAndReturn(run func(context.Co
 func (_m *MockDataNode) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok {
@@ -517,6 +553,10 @@ func (_c *MockDataNode_GetComponentStates_Call) RunAndReturn(run func(context.Co
 func (_m *MockDataNode) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -572,6 +612,10 @@ func (_c *MockDataNode_GetMetrics_Call) RunAndReturn(run func(context.Context, *
 func (_m *MockDataNode) GetNodeID() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetNodeID")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()
@@ -613,6 +657,10 @@ func (_c *MockDataNode_GetNodeID_Call) RunAndReturn(run func() int64) *MockDataN
 func (_m *MockDataNode) GetStateCode() commonpb.StateCode {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStateCode")
+	}
+
 	var r0 commonpb.StateCode
 	if rf, ok := ret.Get(0).(func() commonpb.StateCode); ok {
 		r0 = rf()
@@ -654,6 +702,10 @@ func (_c *MockDataNode_GetStateCode_Call) RunAndReturn(run func() commonpb.State
 func (_m *MockDataNode) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -709,6 +761,10 @@ func (_c *MockDataNode_GetStatisticsChannel_Call) RunAndReturn(run func(context.
 func (_m *MockDataNode) ImportV2(_a0 context.Context, _a1 *datapb.ImportRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ImportV2")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.ImportRequest) (*commonpb.Status, error)); ok {
@@ -764,6 +820,10 @@ func (_c *MockDataNode_ImportV2_Call) RunAndReturn(run func(context.Context, *da
 func (_m *MockDataNode) Init() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Init")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -805,6 +865,10 @@ func (_c *MockDataNode_Init_Call) RunAndReturn(run func() error) *MockDataNode_I
 func (_m *MockDataNode) NotifyChannelOperation(_a0 context.Context, _a1 *datapb.ChannelOperationsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for NotifyChannelOperation")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.ChannelOperationsRequest) (*commonpb.Status, error)); ok {
@@ -860,6 +924,10 @@ func (_c *MockDataNode_NotifyChannelOperation_Call) RunAndReturn(run func(contex
 func (_m *MockDataNode) PreImport(_a0 context.Context, _a1 *datapb.PreImportRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for PreImport")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.PreImportRequest) (*commonpb.Status, error)); ok {
@@ -915,6 +983,10 @@ func (_c *MockDataNode_PreImport_Call) RunAndReturn(run func(context.Context, *d
 func (_m *MockDataNode) QueryImport(_a0 context.Context, _a1 *datapb.QueryImportRequest) (*datapb.QueryImportResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryImport")
+	}
+
 	var r0 *datapb.QueryImportResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.QueryImportRequest) (*datapb.QueryImportResponse, error)); ok {
@@ -970,6 +1042,10 @@ func (_c *MockDataNode_QueryImport_Call) RunAndReturn(run func(context.Context,
 func (_m *MockDataNode) QueryPreImport(_a0 context.Context, _a1 *datapb.QueryPreImportRequest) (*datapb.QueryPreImportResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryPreImport")
+	}
+
 	var r0 *datapb.QueryPreImportResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.QueryPreImportRequest) (*datapb.QueryPreImportResponse, error)); ok {
@@ -1025,6 +1101,10 @@ func (_c *MockDataNode_QueryPreImport_Call) RunAndReturn(run func(context.Contex
 func (_m *MockDataNode) QuerySlot(_a0 context.Context, _a1 *datapb.QuerySlotRequest) (*datapb.QuerySlotResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QuerySlot")
+	}
+
 	var r0 *datapb.QuerySlotResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.QuerySlotRequest) (*datapb.QuerySlotResponse, error)); ok {
@@ -1080,6 +1160,10 @@ func (_c *MockDataNode_QuerySlot_Call) RunAndReturn(run func(context.Context, *d
 func (_m *MockDataNode) Register() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Register")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -1121,6 +1205,10 @@ func (_c *MockDataNode_Register_Call) RunAndReturn(run func() error) *MockDataNo
 func (_m *MockDataNode) ResendSegmentStats(_a0 context.Context, _a1 *datapb.ResendSegmentStatsRequest) (*datapb.ResendSegmentStatsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ResendSegmentStats")
+	}
+
 	var r0 *datapb.ResendSegmentStatsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.ResendSegmentStatsRequest) (*datapb.ResendSegmentStatsResponse, error)); ok {
@@ -1209,6 +1297,10 @@ func (_c *MockDataNode_SetAddress_Call) RunAndReturn(run func(string)) *MockData
 func (_m *MockDataNode) SetDataCoordClient(dataCoord types.DataCoordClient) error {
 	ret := _m.Called(dataCoord)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetDataCoordClient")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(types.DataCoordClient) error); ok {
 		r0 = rf(dataCoord)
@@ -1284,6 +1376,10 @@ func (_c *MockDataNode_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Clien
 func (_m *MockDataNode) SetRootCoordClient(rootCoord types.RootCoordClient) error {
 	ret := _m.Called(rootCoord)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetRootCoordClient")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(types.RootCoordClient) error); ok {
 		r0 = rf(rootCoord)
@@ -1326,6 +1422,10 @@ func (_c *MockDataNode_SetRootCoordClient_Call) RunAndReturn(run func(types.Root
 func (_m *MockDataNode) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok {
@@ -1381,6 +1481,10 @@ func (_c *MockDataNode_ShowConfigurations_Call) RunAndReturn(run func(context.Co
 func (_m *MockDataNode) Start() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Start")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -1422,6 +1526,10 @@ func (_c *MockDataNode_Start_Call) RunAndReturn(run func() error) *MockDataNode_
 func (_m *MockDataNode) Stop() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Stop")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -1463,6 +1571,10 @@ func (_c *MockDataNode_Stop_Call) RunAndReturn(run func() error) *MockDataNode_S
 func (_m *MockDataNode) SyncSegments(_a0 context.Context, _a1 *datapb.SyncSegmentsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SyncSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.SyncSegmentsRequest) (*commonpb.Status, error)); ok {
@@ -1551,6 +1663,10 @@ func (_c *MockDataNode_UpdateStateCode_Call) RunAndReturn(run func(commonpb.Stat
 func (_m *MockDataNode) WatchDmChannels(_a0 context.Context, _a1 *datapb.WatchDmChannelsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WatchDmChannels")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.WatchDmChannelsRequest) (*commonpb.Status, error)); ok {

internal/mocks/mock_datanode_client.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -42,6 +42,10 @@ func (_m *MockDataNodeClient) CheckChannelOperationProgress(ctx context.Context,
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckChannelOperationProgress")
+	}
+
 	var r0 *datapb.ChannelOperationProgressResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.ChannelWatchInfo, ...grpc.CallOption) (*datapb.ChannelOperationProgressResponse, error)); ok {
@@ -105,6 +109,10 @@ func (_c *MockDataNodeClient_CheckChannelOperationProgress_Call) RunAndReturn(ru
 func (_m *MockDataNodeClient) Close() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Close")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -153,6 +161,10 @@ func (_m *MockDataNodeClient) CompactionV2(ctx context.Context, in *datapb.Compa
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CompactionV2")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.CompactionPlan, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -223,6 +235,10 @@ func (_m *MockDataNodeClient) DropCompactionPlan(ctx context.Context, in *datapb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropCompactionPlan")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.DropCompactionPlanRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -293,6 +309,10 @@ func (_m *MockDataNodeClient) DropImport(ctx context.Context, in *datapb.DropImp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropImport")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.DropImportRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -363,6 +383,10 @@ func (_m *MockDataNodeClient) FlushChannels(ctx context.Context, in *datapb.Flus
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for FlushChannels")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.FlushChannelsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -433,6 +457,10 @@ func (_m *MockDataNodeClient) FlushSegments(ctx context.Context, in *datapb.Flus
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for FlushSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.FlushSegmentsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -503,6 +531,10 @@ func (_m *MockDataNodeClient) GetCompactionState(ctx context.Context, in *datapb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCompactionState")
+	}
+
 	var r0 *datapb.CompactionStateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.CompactionStateRequest, ...grpc.CallOption) (*datapb.CompactionStateResponse, error)); ok {
@@ -573,6 +605,10 @@ func (_m *MockDataNodeClient) GetComponentStates(ctx context.Context, in *milvus
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok {
@@ -643,6 +679,10 @@ func (_m *MockDataNodeClient) GetMetrics(ctx context.Context, in *milvuspb.GetMe
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -713,6 +753,10 @@ func (_m *MockDataNodeClient) GetStatisticsChannel(ctx context.Context, in *inte
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -783,6 +827,10 @@ func (_m *MockDataNodeClient) ImportV2(ctx context.Context, in *datapb.ImportReq
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ImportV2")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.ImportRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -853,6 +901,10 @@ func (_m *MockDataNodeClient) NotifyChannelOperation(ctx context.Context, in *da
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for NotifyChannelOperation")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.ChannelOperationsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -923,6 +975,10 @@ func (_m *MockDataNodeClient) PreImport(ctx context.Context, in *datapb.PreImpor
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for PreImport")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.PreImportRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -993,6 +1049,10 @@ func (_m *MockDataNodeClient) QueryImport(ctx context.Context, in *datapb.QueryI
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryImport")
+	}
+
 	var r0 *datapb.QueryImportResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.QueryImportRequest, ...grpc.CallOption) (*datapb.QueryImportResponse, error)); ok {
@@ -1063,6 +1123,10 @@ func (_m *MockDataNodeClient) QueryPreImport(ctx context.Context, in *datapb.Que
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryPreImport")
+	}
+
 	var r0 *datapb.QueryPreImportResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.QueryPreImportRequest, ...grpc.CallOption) (*datapb.QueryPreImportResponse, error)); ok {
@@ -1133,6 +1197,10 @@ func (_m *MockDataNodeClient) QuerySlot(ctx context.Context, in *datapb.QuerySlo
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QuerySlot")
+	}
+
 	var r0 *datapb.QuerySlotResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.QuerySlotRequest, ...grpc.CallOption) (*datapb.QuerySlotResponse, error)); ok {
@@ -1203,6 +1271,10 @@ func (_m *MockDataNodeClient) ResendSegmentStats(ctx context.Context, in *datapb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ResendSegmentStats")
+	}
+
 	var r0 *datapb.ResendSegmentStatsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.ResendSegmentStatsRequest, ...grpc.CallOption) (*datapb.ResendSegmentStatsResponse, error)); ok {
@@ -1273,6 +1345,10 @@ func (_m *MockDataNodeClient) ShowConfigurations(ctx context.Context, in *intern
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest, ...grpc.CallOption) (*internalpb.ShowConfigurationsResponse, error)); ok {
@@ -1343,6 +1419,10 @@ func (_m *MockDataNodeClient) SyncSegments(ctx context.Context, in *datapb.SyncS
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SyncSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.SyncSegmentsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1413,6 +1493,10 @@ func (_m *MockDataNodeClient) WatchDmChannels(ctx context.Context, in *datapb.Wa
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WatchDmChannels")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *datapb.WatchDmChannelsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {

internal/mocks/mock_indexnode.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -34,6 +34,10 @@ func (_m *MockIndexNode) EXPECT() *MockIndexNode_Expecter {
 func (_m *MockIndexNode) CreateJob(_a0 context.Context, _a1 *workerpb.CreateJobRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateJob")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.CreateJobRequest) (*commonpb.Status, error)); ok {
@@ -89,6 +93,10 @@ func (_c *MockIndexNode_CreateJob_Call) RunAndReturn(run func(context.Context, *
 func (_m *MockIndexNode) CreateJobV2(_a0 context.Context, _a1 *workerpb.CreateJobV2Request) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateJobV2")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.CreateJobV2Request) (*commonpb.Status, error)); ok {
@@ -144,6 +152,10 @@ func (_c *MockIndexNode_CreateJobV2_Call) RunAndReturn(run func(context.Context,
 func (_m *MockIndexNode) DropJobs(_a0 context.Context, _a1 *workerpb.DropJobsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropJobs")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.DropJobsRequest) (*commonpb.Status, error)); ok {
@@ -199,6 +211,10 @@ func (_c *MockIndexNode_DropJobs_Call) RunAndReturn(run func(context.Context, *w
 func (_m *MockIndexNode) DropJobsV2(_a0 context.Context, _a1 *workerpb.DropJobsV2Request) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropJobsV2")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.DropJobsV2Request) (*commonpb.Status, error)); ok {
@@ -254,6 +270,10 @@ func (_c *MockIndexNode_DropJobsV2_Call) RunAndReturn(run func(context.Context,
 func (_m *MockIndexNode) GetAddress() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAddress")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()
@@ -295,6 +315,10 @@ func (_c *MockIndexNode_GetAddress_Call) RunAndReturn(run func() string) *MockIn
 func (_m *MockIndexNode) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok {
@@ -350,6 +374,10 @@ func (_c *MockIndexNode_GetComponentStates_Call) RunAndReturn(run func(context.C
 func (_m *MockIndexNode) GetJobStats(_a0 context.Context, _a1 *workerpb.GetJobStatsRequest) (*workerpb.GetJobStatsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetJobStats")
+	}
+
 	var r0 *workerpb.GetJobStatsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.GetJobStatsRequest) (*workerpb.GetJobStatsResponse, error)); ok {
@@ -405,6 +433,10 @@ func (_c *MockIndexNode_GetJobStats_Call) RunAndReturn(run func(context.Context,
 func (_m *MockIndexNode) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -460,6 +492,10 @@ func (_c *MockIndexNode_GetMetrics_Call) RunAndReturn(run func(context.Context,
 func (_m *MockIndexNode) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -515,6 +551,10 @@ func (_c *MockIndexNode_GetStatisticsChannel_Call) RunAndReturn(run func(context
 func (_m *MockIndexNode) Init() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Init")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -556,6 +596,10 @@ func (_c *MockIndexNode_Init_Call) RunAndReturn(run func() error) *MockIndexNode
 func (_m *MockIndexNode) QueryJobs(_a0 context.Context, _a1 *workerpb.QueryJobsRequest) (*workerpb.QueryJobsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryJobs")
+	}
+
 	var r0 *workerpb.QueryJobsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.QueryJobsRequest) (*workerpb.QueryJobsResponse, error)); ok {
@@ -611,6 +655,10 @@ func (_c *MockIndexNode_QueryJobs_Call) RunAndReturn(run func(context.Context, *
 func (_m *MockIndexNode) QueryJobsV2(_a0 context.Context, _a1 *workerpb.QueryJobsV2Request) (*workerpb.QueryJobsV2Response, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryJobsV2")
+	}
+
 	var r0 *workerpb.QueryJobsV2Response
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.QueryJobsV2Request) (*workerpb.QueryJobsV2Response, error)); ok {
@@ -666,6 +714,10 @@ func (_c *MockIndexNode_QueryJobsV2_Call) RunAndReturn(run func(context.Context,
 func (_m *MockIndexNode) Register() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Register")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -773,6 +825,10 @@ func (_c *MockIndexNode_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Clie
 func (_m *MockIndexNode) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok {
@@ -828,6 +884,10 @@ func (_c *MockIndexNode_ShowConfigurations_Call) RunAndReturn(run func(context.C
 func (_m *MockIndexNode) Start() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Start")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -869,6 +929,10 @@ func (_c *MockIndexNode_Start_Call) RunAndReturn(run func() error) *MockIndexNod
 func (_m *MockIndexNode) Stop() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Stop")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()

internal/mocks/mock_indexnode_client.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -35,6 +35,10 @@ func (_m *MockIndexNodeClient) EXPECT() *MockIndexNodeClient_Expecter {
 func (_m *MockIndexNodeClient) Close() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Close")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -83,6 +87,10 @@ func (_m *MockIndexNodeClient) CreateJob(ctx context.Context, in *workerpb.Creat
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateJob")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.CreateJobRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -153,6 +161,10 @@ func (_m *MockIndexNodeClient) CreateJobV2(ctx context.Context, in *workerpb.Cre
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateJobV2")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.CreateJobV2Request, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -223,6 +235,10 @@ func (_m *MockIndexNodeClient) DropJobs(ctx context.Context, in *workerpb.DropJo
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropJobs")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.DropJobsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -293,6 +309,10 @@ func (_m *MockIndexNodeClient) DropJobsV2(ctx context.Context, in *workerpb.Drop
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropJobsV2")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.DropJobsV2Request, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -363,6 +383,10 @@ func (_m *MockIndexNodeClient) GetComponentStates(ctx context.Context, in *milvu
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok {
@@ -433,6 +457,10 @@ func (_m *MockIndexNodeClient) GetJobStats(ctx context.Context, in *workerpb.Get
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetJobStats")
+	}
+
 	var r0 *workerpb.GetJobStatsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.GetJobStatsRequest, ...grpc.CallOption) (*workerpb.GetJobStatsResponse, error)); ok {
@@ -503,6 +531,10 @@ func (_m *MockIndexNodeClient) GetMetrics(ctx context.Context, in *milvuspb.GetM
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -573,6 +605,10 @@ func (_m *MockIndexNodeClient) GetStatisticsChannel(ctx context.Context, in *int
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -643,6 +679,10 @@ func (_m *MockIndexNodeClient) QueryJobs(ctx context.Context, in *workerpb.Query
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryJobs")
+	}
+
 	var r0 *workerpb.QueryJobsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.QueryJobsRequest, ...grpc.CallOption) (*workerpb.QueryJobsResponse, error)); ok {
@@ -713,6 +753,10 @@ func (_m *MockIndexNodeClient) QueryJobsV2(ctx context.Context, in *workerpb.Que
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryJobsV2")
+	}
+
 	var r0 *workerpb.QueryJobsV2Response
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *workerpb.QueryJobsV2Request, ...grpc.CallOption) (*workerpb.QueryJobsV2Response, error)); ok {
@@ -783,6 +827,10 @@ func (_m *MockIndexNodeClient) ShowConfigurations(ctx context.Context, in *inter
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest, ...grpc.CallOption) (*internalpb.ShowConfigurationsResponse, error)); ok {

internal/mocks/mock_metastore/mock_StreamingCoordCataLog.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_metastore
 
@@ -27,6 +27,10 @@ func (_m *MockStreamingCoordCataLog) EXPECT() *MockStreamingCoordCataLog_Expecte
 func (_m *MockStreamingCoordCataLog) ListPChannel(ctx context.Context) ([]*streamingpb.PChannelMeta, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListPChannel")
+	}
+
 	var r0 []*streamingpb.PChannelMeta
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) ([]*streamingpb.PChannelMeta, error)); ok {
@@ -81,6 +85,10 @@ func (_c *MockStreamingCoordCataLog_ListPChannel_Call) RunAndReturn(run func(con
 func (_m *MockStreamingCoordCataLog) SavePChannels(ctx context.Context, info []*streamingpb.PChannelMeta) error {
 	ret := _m.Called(ctx, info)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SavePChannels")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, []*streamingpb.PChannelMeta) error); ok {
 		r0 = rf(ctx, info)

internal/mocks/mock_metastore/mock_StreamingNodeCataLog.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_metastore
 
@@ -27,6 +27,10 @@ func (_m *MockStreamingNodeCataLog) EXPECT() *MockStreamingNodeCataLog_Expecter
 func (_m *MockStreamingNodeCataLog) ListSegmentAssignment(ctx context.Context, pChannelName string) ([]*streamingpb.SegmentAssignmentMeta, error) {
 	ret := _m.Called(ctx, pChannelName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListSegmentAssignment")
+	}
+
 	var r0 []*streamingpb.SegmentAssignmentMeta
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) ([]*streamingpb.SegmentAssignmentMeta, error)); ok {
@@ -82,6 +86,10 @@ func (_c *MockStreamingNodeCataLog_ListSegmentAssignment_Call) RunAndReturn(run
 func (_m *MockStreamingNodeCataLog) SaveSegmentAssignments(ctx context.Context, pChannelName string, infos []*streamingpb.SegmentAssignmentMeta) error {
 	ret := _m.Called(ctx, pChannelName, infos)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SaveSegmentAssignments")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, []*streamingpb.SegmentAssignmentMeta) error); ok {
 		r0 = rf(ctx, pChannelName, infos)

internal/mocks/mock_proxy_client.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -35,6 +35,10 @@ func (_m *MockProxyClient) EXPECT() *MockProxyClient_Expecter {
 func (_m *MockProxyClient) Close() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Close")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -83,6 +87,10 @@ func (_m *MockProxyClient) GetComponentStates(ctx context.Context, in *milvuspb.
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok {
@@ -153,6 +161,10 @@ func (_m *MockProxyClient) GetDdChannel(ctx context.Context, in *internalpb.GetD
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDdChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetDdChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -223,6 +235,10 @@ func (_m *MockProxyClient) GetImportProgress(ctx context.Context, in *internalpb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetImportProgress")
+	}
+
 	var r0 *internalpb.GetImportProgressResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetImportProgressRequest, ...grpc.CallOption) (*internalpb.GetImportProgressResponse, error)); ok {
@@ -293,6 +309,10 @@ func (_m *MockProxyClient) GetProxyMetrics(ctx context.Context, in *milvuspb.Get
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetProxyMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -363,6 +383,10 @@ func (_m *MockProxyClient) GetStatisticsChannel(ctx context.Context, in *interna
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -433,6 +457,10 @@ func (_m *MockProxyClient) ImportV2(ctx context.Context, in *internalpb.ImportRe
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ImportV2")
+	}
+
 	var r0 *internalpb.ImportResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ImportRequest, ...grpc.CallOption) (*internalpb.ImportResponse, error)); ok {
@@ -503,6 +531,10 @@ func (_m *MockProxyClient) InvalidateCollectionMetaCache(ctx context.Context, in
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InvalidateCollectionMetaCache")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCollMetaCacheRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -573,6 +605,10 @@ func (_m *MockProxyClient) InvalidateCredentialCache(ctx context.Context, in *pr
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InvalidateCredentialCache")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCredCacheRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -643,6 +679,10 @@ func (_m *MockProxyClient) InvalidateShardLeaderCache(ctx context.Context, in *p
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InvalidateShardLeaderCache")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateShardLeaderCacheRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -713,6 +753,10 @@ func (_m *MockProxyClient) ListClientInfos(ctx context.Context, in *proxypb.List
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListClientInfos")
+	}
+
 	var r0 *proxypb.ListClientInfosResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.ListClientInfosRequest, ...grpc.CallOption) (*proxypb.ListClientInfosResponse, error)); ok {
@@ -783,6 +827,10 @@ func (_m *MockProxyClient) ListImports(ctx context.Context, in *internalpb.ListI
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListImports")
+	}
+
 	var r0 *internalpb.ListImportsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequest, ...grpc.CallOption) (*internalpb.ListImportsResponse, error)); ok {
@@ -853,6 +901,10 @@ func (_m *MockProxyClient) RefreshPolicyInfoCache(ctx context.Context, in *proxy
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RefreshPolicyInfoCache")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.RefreshPolicyInfoCacheRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -923,6 +975,10 @@ func (_m *MockProxyClient) SetRates(ctx context.Context, in *proxypb.SetRatesReq
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetRates")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.SetRatesRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -993,6 +1049,10 @@ func (_m *MockProxyClient) UpdateCredentialCache(ctx context.Context, in *proxyp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateCredentialCache")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.UpdateCredCacheRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {

internal/mocks/mock_querycoord.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -38,6 +38,10 @@ func (_m *MockQueryCoord) EXPECT() *MockQueryCoord_Expecter {
 func (_m *MockQueryCoord) ActivateChecker(_a0 context.Context, _a1 *querypb.ActivateCheckerRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ActivateChecker")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ActivateCheckerRequest) (*commonpb.Status, error)); ok {
@@ -93,6 +97,10 @@ func (_c *MockQueryCoord_ActivateChecker_Call) RunAndReturn(run func(context.Con
 func (_m *MockQueryCoord) CheckHealth(_a0 context.Context, _a1 *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckHealth")
+	}
+
 	var r0 *milvuspb.CheckHealthResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error)); ok {
@@ -148,6 +156,10 @@ func (_c *MockQueryCoord_CheckHealth_Call) RunAndReturn(run func(context.Context
 func (_m *MockQueryCoord) CheckQueryNodeDistribution(_a0 context.Context, _a1 *querypb.CheckQueryNodeDistributionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckQueryNodeDistribution")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.CheckQueryNodeDistributionRequest) (*commonpb.Status, error)); ok {
@@ -203,6 +215,10 @@ func (_c *MockQueryCoord_CheckQueryNodeDistribution_Call) RunAndReturn(run func(
 func (_m *MockQueryCoord) CreateResourceGroup(_a0 context.Context, _a1 *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateResourceGroup")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error)); ok {
@@ -258,6 +274,10 @@ func (_c *MockQueryCoord_CreateResourceGroup_Call) RunAndReturn(run func(context
 func (_m *MockQueryCoord) DeactivateChecker(_a0 context.Context, _a1 *querypb.DeactivateCheckerRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeactivateChecker")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.DeactivateCheckerRequest) (*commonpb.Status, error)); ok {
@@ -313,6 +333,10 @@ func (_c *MockQueryCoord_DeactivateChecker_Call) RunAndReturn(run func(context.C
 func (_m *MockQueryCoord) DescribeResourceGroup(_a0 context.Context, _a1 *querypb.DescribeResourceGroupRequest) (*querypb.DescribeResourceGroupResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeResourceGroup")
+	}
+
 	var r0 *querypb.DescribeResourceGroupResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.DescribeResourceGroupRequest) (*querypb.DescribeResourceGroupResponse, error)); ok {
@@ -368,6 +392,10 @@ func (_c *MockQueryCoord_DescribeResourceGroup_Call) RunAndReturn(run func(conte
 func (_m *MockQueryCoord) DropResourceGroup(_a0 context.Context, _a1 *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropResourceGroup")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error)); ok {
@@ -423,6 +451,10 @@ func (_c *MockQueryCoord_DropResourceGroup_Call) RunAndReturn(run func(context.C
 func (_m *MockQueryCoord) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok {
@@ -478,6 +510,10 @@ func (_c *MockQueryCoord_GetComponentStates_Call) RunAndReturn(run func(context.
 func (_m *MockQueryCoord) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -533,6 +569,10 @@ func (_c *MockQueryCoord_GetMetrics_Call) RunAndReturn(run func(context.Context,
 func (_m *MockQueryCoord) GetPartitionStates(_a0 context.Context, _a1 *querypb.GetPartitionStatesRequest) (*querypb.GetPartitionStatesResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPartitionStates")
+	}
+
 	var r0 *querypb.GetPartitionStatesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetPartitionStatesRequest) (*querypb.GetPartitionStatesResponse, error)); ok {
@@ -588,6 +628,10 @@ func (_c *MockQueryCoord_GetPartitionStates_Call) RunAndReturn(run func(context.
 func (_m *MockQueryCoord) GetQueryNodeDistribution(_a0 context.Context, _a1 *querypb.GetQueryNodeDistributionRequest) (*querypb.GetQueryNodeDistributionResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetQueryNodeDistribution")
+	}
+
 	var r0 *querypb.GetQueryNodeDistributionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetQueryNodeDistributionRequest) (*querypb.GetQueryNodeDistributionResponse, error)); ok {
@@ -643,6 +687,10 @@ func (_c *MockQueryCoord_GetQueryNodeDistribution_Call) RunAndReturn(run func(co
 func (_m *MockQueryCoord) GetReplicas(_a0 context.Context, _a1 *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetReplicas")
+	}
+
 	var r0 *milvuspb.GetReplicasResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error)); ok {
@@ -698,6 +746,10 @@ func (_c *MockQueryCoord_GetReplicas_Call) RunAndReturn(run func(context.Context
 func (_m *MockQueryCoord) GetSegmentInfo(_a0 context.Context, _a1 *querypb.GetSegmentInfoRequest) (*querypb.GetSegmentInfoResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentInfo")
+	}
+
 	var r0 *querypb.GetSegmentInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetSegmentInfoRequest) (*querypb.GetSegmentInfoResponse, error)); ok {
@@ -753,6 +805,10 @@ func (_c *MockQueryCoord_GetSegmentInfo_Call) RunAndReturn(run func(context.Cont
 func (_m *MockQueryCoord) GetShardLeaders(_a0 context.Context, _a1 *querypb.GetShardLeadersRequest) (*querypb.GetShardLeadersResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetShardLeaders")
+	}
+
 	var r0 *querypb.GetShardLeadersResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetShardLeadersRequest) (*querypb.GetShardLeadersResponse, error)); ok {
@@ -808,6 +864,10 @@ func (_c *MockQueryCoord_GetShardLeaders_Call) RunAndReturn(run func(context.Con
 func (_m *MockQueryCoord) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -863,6 +923,10 @@ func (_c *MockQueryCoord_GetStatisticsChannel_Call) RunAndReturn(run func(contex
 func (_m *MockQueryCoord) GetTimeTickChannel(_a0 context.Context, _a1 *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetTimeTickChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -918,6 +982,10 @@ func (_c *MockQueryCoord_GetTimeTickChannel_Call) RunAndReturn(run func(context.
 func (_m *MockQueryCoord) Init() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Init")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -959,6 +1027,10 @@ func (_c *MockQueryCoord_Init_Call) RunAndReturn(run func() error) *MockQueryCoo
 func (_m *MockQueryCoord) ListCheckers(_a0 context.Context, _a1 *querypb.ListCheckersRequest) (*querypb.ListCheckersResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListCheckers")
+	}
+
 	var r0 *querypb.ListCheckersResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ListCheckersRequest) (*querypb.ListCheckersResponse, error)); ok {
@@ -1014,6 +1086,10 @@ func (_c *MockQueryCoord_ListCheckers_Call) RunAndReturn(run func(context.Contex
 func (_m *MockQueryCoord) ListQueryNode(_a0 context.Context, _a1 *querypb.ListQueryNodeRequest) (*querypb.ListQueryNodeResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListQueryNode")
+	}
+
 	var r0 *querypb.ListQueryNodeResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ListQueryNodeRequest) (*querypb.ListQueryNodeResponse, error)); ok {
@@ -1069,6 +1145,10 @@ func (_c *MockQueryCoord_ListQueryNode_Call) RunAndReturn(run func(context.Conte
 func (_m *MockQueryCoord) ListResourceGroups(_a0 context.Context, _a1 *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListResourceGroups")
+	}
+
 	var r0 *milvuspb.ListResourceGroupsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error)); ok {
@@ -1124,6 +1204,10 @@ func (_c *MockQueryCoord_ListResourceGroups_Call) RunAndReturn(run func(context.
 func (_m *MockQueryCoord) LoadBalance(_a0 context.Context, _a1 *querypb.LoadBalanceRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadBalance")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadBalanceRequest) (*commonpb.Status, error)); ok {
@@ -1179,6 +1263,10 @@ func (_c *MockQueryCoord_LoadBalance_Call) RunAndReturn(run func(context.Context
 func (_m *MockQueryCoord) LoadCollection(_a0 context.Context, _a1 *querypb.LoadCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadCollectionRequest) (*commonpb.Status, error)); ok {
@@ -1234,6 +1322,10 @@ func (_c *MockQueryCoord_LoadCollection_Call) RunAndReturn(run func(context.Cont
 func (_m *MockQueryCoord) LoadPartitions(_a0 context.Context, _a1 *querypb.LoadPartitionsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadPartitions")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadPartitionsRequest) (*commonpb.Status, error)); ok {
@@ -1289,6 +1381,10 @@ func (_c *MockQueryCoord_LoadPartitions_Call) RunAndReturn(run func(context.Cont
 func (_m *MockQueryCoord) Register() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Register")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -1330,6 +1426,10 @@ func (_c *MockQueryCoord_Register_Call) RunAndReturn(run func() error) *MockQuer
 func (_m *MockQueryCoord) ReleaseCollection(_a0 context.Context, _a1 *querypb.ReleaseCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleaseCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseCollectionRequest) (*commonpb.Status, error)); ok {
@@ -1385,6 +1485,10 @@ func (_c *MockQueryCoord_ReleaseCollection_Call) RunAndReturn(run func(context.C
 func (_m *MockQueryCoord) ReleasePartitions(_a0 context.Context, _a1 *querypb.ReleasePartitionsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleasePartitions")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleasePartitionsRequest) (*commonpb.Status, error)); ok {
@@ -1440,6 +1544,10 @@ func (_c *MockQueryCoord_ReleasePartitions_Call) RunAndReturn(run func(context.C
 func (_m *MockQueryCoord) ResumeBalance(_a0 context.Context, _a1 *querypb.ResumeBalanceRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ResumeBalance")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ResumeBalanceRequest) (*commonpb.Status, error)); ok {
@@ -1495,6 +1603,10 @@ func (_c *MockQueryCoord_ResumeBalance_Call) RunAndReturn(run func(context.Conte
 func (_m *MockQueryCoord) ResumeNode(_a0 context.Context, _a1 *querypb.ResumeNodeRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ResumeNode")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ResumeNodeRequest) (*commonpb.Status, error)); ok {
@@ -1583,6 +1695,10 @@ func (_c *MockQueryCoord_SetAddress_Call) RunAndReturn(run func(string)) *MockQu
 func (_m *MockQueryCoord) SetDataCoordClient(dataCoord types.DataCoordClient) error {
 	ret := _m.Called(dataCoord)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetDataCoordClient")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(types.DataCoordClient) error); ok {
 		r0 = rf(dataCoord)
@@ -1691,6 +1807,10 @@ func (_c *MockQueryCoord_SetQueryNodeCreator_Call) RunAndReturn(run func(func(co
 func (_m *MockQueryCoord) SetRootCoordClient(rootCoord types.RootCoordClient) error {
 	ret := _m.Called(rootCoord)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetRootCoordClient")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(types.RootCoordClient) error); ok {
 		r0 = rf(rootCoord)
@@ -1766,6 +1886,10 @@ func (_c *MockQueryCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client
 func (_m *MockQueryCoord) ShowCollections(_a0 context.Context, _a1 *querypb.ShowCollectionsRequest) (*querypb.ShowCollectionsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowCollections")
+	}
+
 	var r0 *querypb.ShowCollectionsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ShowCollectionsRequest) (*querypb.ShowCollectionsResponse, error)); ok {
@@ -1821,6 +1945,10 @@ func (_c *MockQueryCoord_ShowCollections_Call) RunAndReturn(run func(context.Con
 func (_m *MockQueryCoord) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok {
@@ -1876,6 +2004,10 @@ func (_c *MockQueryCoord_ShowConfigurations_Call) RunAndReturn(run func(context.
 func (_m *MockQueryCoord) ShowPartitions(_a0 context.Context, _a1 *querypb.ShowPartitionsRequest) (*querypb.ShowPartitionsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowPartitions")
+	}
+
 	var r0 *querypb.ShowPartitionsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ShowPartitionsRequest) (*querypb.ShowPartitionsResponse, error)); ok {
@@ -1931,6 +2063,10 @@ func (_c *MockQueryCoord_ShowPartitions_Call) RunAndReturn(run func(context.Cont
 func (_m *MockQueryCoord) Start() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Start")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -1972,6 +2108,10 @@ func (_c *MockQueryCoord_Start_Call) RunAndReturn(run func() error) *MockQueryCo
 func (_m *MockQueryCoord) Stop() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Stop")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -2013,6 +2153,10 @@ func (_c *MockQueryCoord_Stop_Call) RunAndReturn(run func() error) *MockQueryCoo
 func (_m *MockQueryCoord) SuspendBalance(_a0 context.Context, _a1 *querypb.SuspendBalanceRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SuspendBalance")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SuspendBalanceRequest) (*commonpb.Status, error)); ok {
@@ -2068,6 +2212,10 @@ func (_c *MockQueryCoord_SuspendBalance_Call) RunAndReturn(run func(context.Cont
 func (_m *MockQueryCoord) SuspendNode(_a0 context.Context, _a1 *querypb.SuspendNodeRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SuspendNode")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SuspendNodeRequest) (*commonpb.Status, error)); ok {
@@ -2123,6 +2271,10 @@ func (_c *MockQueryCoord_SuspendNode_Call) RunAndReturn(run func(context.Context
 func (_m *MockQueryCoord) SyncNewCreatedPartition(_a0 context.Context, _a1 *querypb.SyncNewCreatedPartitionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SyncNewCreatedPartition")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncNewCreatedPartitionRequest) (*commonpb.Status, error)); ok {
@@ -2178,6 +2330,10 @@ func (_c *MockQueryCoord_SyncNewCreatedPartition_Call) RunAndReturn(run func(con
 func (_m *MockQueryCoord) TransferChannel(_a0 context.Context, _a1 *querypb.TransferChannelRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for TransferChannel")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.TransferChannelRequest) (*commonpb.Status, error)); ok {
@@ -2233,6 +2389,10 @@ func (_c *MockQueryCoord_TransferChannel_Call) RunAndReturn(run func(context.Con
 func (_m *MockQueryCoord) TransferNode(_a0 context.Context, _a1 *milvuspb.TransferNodeRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for TransferNode")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferNodeRequest) (*commonpb.Status, error)); ok {
@@ -2288,6 +2448,10 @@ func (_c *MockQueryCoord_TransferNode_Call) RunAndReturn(run func(context.Contex
 func (_m *MockQueryCoord) TransferReplica(_a0 context.Context, _a1 *querypb.TransferReplicaRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for TransferReplica")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.TransferReplicaRequest) (*commonpb.Status, error)); ok {
@@ -2343,6 +2507,10 @@ func (_c *MockQueryCoord_TransferReplica_Call) RunAndReturn(run func(context.Con
 func (_m *MockQueryCoord) TransferSegment(_a0 context.Context, _a1 *querypb.TransferSegmentRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for TransferSegment")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.TransferSegmentRequest) (*commonpb.Status, error)); ok {
@@ -2398,6 +2566,10 @@ func (_c *MockQueryCoord_TransferSegment_Call) RunAndReturn(run func(context.Con
 func (_m *MockQueryCoord) UpdateLoadConfig(_a0 context.Context, _a1 *querypb.UpdateLoadConfigRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateLoadConfig")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.UpdateLoadConfigRequest) (*commonpb.Status, error)); ok {
@@ -2453,6 +2625,10 @@ func (_c *MockQueryCoord_UpdateLoadConfig_Call) RunAndReturn(run func(context.Co
 func (_m *MockQueryCoord) UpdateResourceGroups(_a0 context.Context, _a1 *querypb.UpdateResourceGroupsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateResourceGroups")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.UpdateResourceGroupsRequest) (*commonpb.Status, error)); ok {

internal/mocks/mock_querycoord_client.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -42,6 +42,10 @@ func (_m *MockQueryCoordClient) ActivateChecker(ctx context.Context, in *querypb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ActivateChecker")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ActivateCheckerRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -112,6 +116,10 @@ func (_m *MockQueryCoordClient) CheckHealth(ctx context.Context, in *milvuspb.Ch
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckHealth")
+	}
+
 	var r0 *milvuspb.CheckHealthResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest, ...grpc.CallOption) (*milvuspb.CheckHealthResponse, error)); ok {
@@ -182,6 +190,10 @@ func (_m *MockQueryCoordClient) CheckQueryNodeDistribution(ctx context.Context,
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckQueryNodeDistribution")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.CheckQueryNodeDistributionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -245,6 +257,10 @@ func (_c *MockQueryCoordClient_CheckQueryNodeDistribution_Call) RunAndReturn(run
 func (_m *MockQueryCoordClient) Close() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Close")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -293,6 +309,10 @@ func (_m *MockQueryCoordClient) CreateResourceGroup(ctx context.Context, in *mil
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateResourceGroup")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateResourceGroupRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -363,6 +383,10 @@ func (_m *MockQueryCoordClient) DeactivateChecker(ctx context.Context, in *query
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeactivateChecker")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.DeactivateCheckerRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -433,6 +457,10 @@ func (_m *MockQueryCoordClient) DescribeResourceGroup(ctx context.Context, in *q
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeResourceGroup")
+	}
+
 	var r0 *querypb.DescribeResourceGroupResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.DescribeResourceGroupRequest, ...grpc.CallOption) (*querypb.DescribeResourceGroupResponse, error)); ok {
@@ -503,6 +531,10 @@ func (_m *MockQueryCoordClient) DropResourceGroup(ctx context.Context, in *milvu
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropResourceGroup")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropResourceGroupRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -573,6 +605,10 @@ func (_m *MockQueryCoordClient) GetComponentStates(ctx context.Context, in *milv
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok {
@@ -643,6 +679,10 @@ func (_m *MockQueryCoordClient) GetMetrics(ctx context.Context, in *milvuspb.Get
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -713,6 +753,10 @@ func (_m *MockQueryCoordClient) GetPartitionStates(ctx context.Context, in *quer
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPartitionStates")
+	}
+
 	var r0 *querypb.GetPartitionStatesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetPartitionStatesRequest, ...grpc.CallOption) (*querypb.GetPartitionStatesResponse, error)); ok {
@@ -783,6 +827,10 @@ func (_m *MockQueryCoordClient) GetQueryNodeDistribution(ctx context.Context, in
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetQueryNodeDistribution")
+	}
+
 	var r0 *querypb.GetQueryNodeDistributionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetQueryNodeDistributionRequest, ...grpc.CallOption) (*querypb.GetQueryNodeDistributionResponse, error)); ok {
@@ -853,6 +901,10 @@ func (_m *MockQueryCoordClient) GetReplicas(ctx context.Context, in *milvuspb.Ge
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetReplicas")
+	}
+
 	var r0 *milvuspb.GetReplicasResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetReplicasRequest, ...grpc.CallOption) (*milvuspb.GetReplicasResponse, error)); ok {
@@ -923,6 +975,10 @@ func (_m *MockQueryCoordClient) GetSegmentInfo(ctx context.Context, in *querypb.
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentInfo")
+	}
+
 	var r0 *querypb.GetSegmentInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetSegmentInfoRequest, ...grpc.CallOption) (*querypb.GetSegmentInfoResponse, error)); ok {
@@ -993,6 +1049,10 @@ func (_m *MockQueryCoordClient) GetShardLeaders(ctx context.Context, in *querypb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetShardLeaders")
+	}
+
 	var r0 *querypb.GetShardLeadersResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetShardLeadersRequest, ...grpc.CallOption) (*querypb.GetShardLeadersResponse, error)); ok {
@@ -1063,6 +1123,10 @@ func (_m *MockQueryCoordClient) GetStatisticsChannel(ctx context.Context, in *in
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -1133,6 +1197,10 @@ func (_m *MockQueryCoordClient) GetTimeTickChannel(ctx context.Context, in *inte
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetTimeTickChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -1203,6 +1271,10 @@ func (_m *MockQueryCoordClient) ListCheckers(ctx context.Context, in *querypb.Li
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListCheckers")
+	}
+
 	var r0 *querypb.ListCheckersResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ListCheckersRequest, ...grpc.CallOption) (*querypb.ListCheckersResponse, error)); ok {
@@ -1273,6 +1345,10 @@ func (_m *MockQueryCoordClient) ListQueryNode(ctx context.Context, in *querypb.L
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListQueryNode")
+	}
+
 	var r0 *querypb.ListQueryNodeResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ListQueryNodeRequest, ...grpc.CallOption) (*querypb.ListQueryNodeResponse, error)); ok {
@@ -1343,6 +1419,10 @@ func (_m *MockQueryCoordClient) ListResourceGroups(ctx context.Context, in *milv
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListResourceGroups")
+	}
+
 	var r0 *milvuspb.ListResourceGroupsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListResourceGroupsRequest, ...grpc.CallOption) (*milvuspb.ListResourceGroupsResponse, error)); ok {
@@ -1413,6 +1493,10 @@ func (_m *MockQueryCoordClient) LoadBalance(ctx context.Context, in *querypb.Loa
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadBalance")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadBalanceRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1483,6 +1567,10 @@ func (_m *MockQueryCoordClient) LoadCollection(ctx context.Context, in *querypb.
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1553,6 +1641,10 @@ func (_m *MockQueryCoordClient) LoadPartitions(ctx context.Context, in *querypb.
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadPartitions")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadPartitionsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1623,6 +1715,10 @@ func (_m *MockQueryCoordClient) ReleaseCollection(ctx context.Context, in *query
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleaseCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1693,6 +1789,10 @@ func (_m *MockQueryCoordClient) ReleasePartitions(ctx context.Context, in *query
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleasePartitions")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleasePartitionsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1763,6 +1863,10 @@ func (_m *MockQueryCoordClient) ResumeBalance(ctx context.Context, in *querypb.R
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ResumeBalance")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ResumeBalanceRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1833,6 +1937,10 @@ func (_m *MockQueryCoordClient) ResumeNode(ctx context.Context, in *querypb.Resu
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ResumeNode")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ResumeNodeRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1903,6 +2011,10 @@ func (_m *MockQueryCoordClient) ShowCollections(ctx context.Context, in *querypb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowCollections")
+	}
+
 	var r0 *querypb.ShowCollectionsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ShowCollectionsRequest, ...grpc.CallOption) (*querypb.ShowCollectionsResponse, error)); ok {
@@ -1973,6 +2085,10 @@ func (_m *MockQueryCoordClient) ShowConfigurations(ctx context.Context, in *inte
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest, ...grpc.CallOption) (*internalpb.ShowConfigurationsResponse, error)); ok {
@@ -2043,6 +2159,10 @@ func (_m *MockQueryCoordClient) ShowPartitions(ctx context.Context, in *querypb.
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowPartitions")
+	}
+
 	var r0 *querypb.ShowPartitionsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ShowPartitionsRequest, ...grpc.CallOption) (*querypb.ShowPartitionsResponse, error)); ok {
@@ -2113,6 +2233,10 @@ func (_m *MockQueryCoordClient) SuspendBalance(ctx context.Context, in *querypb.
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SuspendBalance")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SuspendBalanceRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2183,6 +2307,10 @@ func (_m *MockQueryCoordClient) SuspendNode(ctx context.Context, in *querypb.Sus
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SuspendNode")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SuspendNodeRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2253,6 +2381,10 @@ func (_m *MockQueryCoordClient) SyncNewCreatedPartition(ctx context.Context, in
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SyncNewCreatedPartition")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncNewCreatedPartitionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2323,6 +2455,10 @@ func (_m *MockQueryCoordClient) TransferChannel(ctx context.Context, in *querypb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for TransferChannel")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.TransferChannelRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2393,6 +2529,10 @@ func (_m *MockQueryCoordClient) TransferNode(ctx context.Context, in *milvuspb.T
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for TransferNode")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferNodeRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2463,6 +2603,10 @@ func (_m *MockQueryCoordClient) TransferReplica(ctx context.Context, in *querypb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for TransferReplica")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.TransferReplicaRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2533,6 +2677,10 @@ func (_m *MockQueryCoordClient) TransferSegment(ctx context.Context, in *querypb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for TransferSegment")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.TransferSegmentRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2603,6 +2751,10 @@ func (_m *MockQueryCoordClient) UpdateLoadConfig(ctx context.Context, in *queryp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateLoadConfig")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.UpdateLoadConfigRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2673,6 +2825,10 @@ func (_m *MockQueryCoordClient) UpdateResourceGroups(ctx context.Context, in *qu
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateResourceGroups")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.UpdateResourceGroupsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {

internal/mocks/mock_querynode.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -34,6 +34,10 @@ func (_m *MockQueryNode) EXPECT() *MockQueryNode_Expecter {
 func (_m *MockQueryNode) Delete(_a0 context.Context, _a1 *querypb.DeleteRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Delete")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.DeleteRequest) (*commonpb.Status, error)); ok {
@@ -89,6 +93,10 @@ func (_c *MockQueryNode_Delete_Call) RunAndReturn(run func(context.Context, *que
 func (_m *MockQueryNode) GetAddress() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAddress")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()
@@ -130,6 +138,10 @@ func (_c *MockQueryNode_GetAddress_Call) RunAndReturn(run func() string) *MockQu
 func (_m *MockQueryNode) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok {
@@ -185,6 +197,10 @@ func (_c *MockQueryNode_GetComponentStates_Call) RunAndReturn(run func(context.C
 func (_m *MockQueryNode) GetDataDistribution(_a0 context.Context, _a1 *querypb.GetDataDistributionRequest) (*querypb.GetDataDistributionResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDataDistribution")
+	}
+
 	var r0 *querypb.GetDataDistributionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetDataDistributionRequest) (*querypb.GetDataDistributionResponse, error)); ok {
@@ -240,6 +256,10 @@ func (_c *MockQueryNode_GetDataDistribution_Call) RunAndReturn(run func(context.
 func (_m *MockQueryNode) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -295,6 +315,10 @@ func (_c *MockQueryNode_GetMetrics_Call) RunAndReturn(run func(context.Context,
 func (_m *MockQueryNode) GetNodeID() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetNodeID")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()
@@ -336,6 +360,10 @@ func (_c *MockQueryNode_GetNodeID_Call) RunAndReturn(run func() int64) *MockQuer
 func (_m *MockQueryNode) GetSegmentInfo(_a0 context.Context, _a1 *querypb.GetSegmentInfoRequest) (*querypb.GetSegmentInfoResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentInfo")
+	}
+
 	var r0 *querypb.GetSegmentInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetSegmentInfoRequest) (*querypb.GetSegmentInfoResponse, error)); ok {
@@ -391,6 +419,10 @@ func (_c *MockQueryNode_GetSegmentInfo_Call) RunAndReturn(run func(context.Conte
 func (_m *MockQueryNode) GetStatistics(_a0 context.Context, _a1 *querypb.GetStatisticsRequest) (*internalpb.GetStatisticsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatistics")
+	}
+
 	var r0 *internalpb.GetStatisticsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetStatisticsRequest) (*internalpb.GetStatisticsResponse, error)); ok {
@@ -446,6 +478,10 @@ func (_c *MockQueryNode_GetStatistics_Call) RunAndReturn(run func(context.Contex
 func (_m *MockQueryNode) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -501,6 +537,10 @@ func (_c *MockQueryNode_GetStatisticsChannel_Call) RunAndReturn(run func(context
 func (_m *MockQueryNode) GetTimeTickChannel(_a0 context.Context, _a1 *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetTimeTickChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -556,6 +596,10 @@ func (_c *MockQueryNode_GetTimeTickChannel_Call) RunAndReturn(run func(context.C
 func (_m *MockQueryNode) Init() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Init")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -597,6 +641,10 @@ func (_c *MockQueryNode_Init_Call) RunAndReturn(run func() error) *MockQueryNode
 func (_m *MockQueryNode) LoadPartitions(_a0 context.Context, _a1 *querypb.LoadPartitionsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadPartitions")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadPartitionsRequest) (*commonpb.Status, error)); ok {
@@ -652,6 +700,10 @@ func (_c *MockQueryNode_LoadPartitions_Call) RunAndReturn(run func(context.Conte
 func (_m *MockQueryNode) LoadSegments(_a0 context.Context, _a1 *querypb.LoadSegmentsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadSegmentsRequest) (*commonpb.Status, error)); ok {
@@ -707,6 +759,10 @@ func (_c *MockQueryNode_LoadSegments_Call) RunAndReturn(run func(context.Context
 func (_m *MockQueryNode) Query(_a0 context.Context, _a1 *querypb.QueryRequest) (*internalpb.RetrieveResults, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Query")
+	}
+
 	var r0 *internalpb.RetrieveResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest) (*internalpb.RetrieveResults, error)); ok {
@@ -762,6 +818,10 @@ func (_c *MockQueryNode_Query_Call) RunAndReturn(run func(context.Context, *quer
 func (_m *MockQueryNode) QuerySegments(_a0 context.Context, _a1 *querypb.QueryRequest) (*internalpb.RetrieveResults, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QuerySegments")
+	}
+
 	var r0 *internalpb.RetrieveResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest) (*internalpb.RetrieveResults, error)); ok {
@@ -817,6 +877,10 @@ func (_c *MockQueryNode_QuerySegments_Call) RunAndReturn(run func(context.Contex
 func (_m *MockQueryNode) QueryStream(_a0 *querypb.QueryRequest, _a1 querypb.QueryNode_QueryStreamServer) error {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryStream")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*querypb.QueryRequest, querypb.QueryNode_QueryStreamServer) error); ok {
 		r0 = rf(_a0, _a1)
@@ -860,6 +924,10 @@ func (_c *MockQueryNode_QueryStream_Call) RunAndReturn(run func(*querypb.QueryRe
 func (_m *MockQueryNode) QueryStreamSegments(_a0 *querypb.QueryRequest, _a1 querypb.QueryNode_QueryStreamSegmentsServer) error {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryStreamSegments")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*querypb.QueryRequest, querypb.QueryNode_QueryStreamSegmentsServer) error); ok {
 		r0 = rf(_a0, _a1)
@@ -903,6 +971,10 @@ func (_c *MockQueryNode_QueryStreamSegments_Call) RunAndReturn(run func(*querypb
 func (_m *MockQueryNode) Register() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Register")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -944,6 +1016,10 @@ func (_c *MockQueryNode_Register_Call) RunAndReturn(run func() error) *MockQuery
 func (_m *MockQueryNode) ReleaseCollection(_a0 context.Context, _a1 *querypb.ReleaseCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleaseCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseCollectionRequest) (*commonpb.Status, error)); ok {
@@ -999,6 +1075,10 @@ func (_c *MockQueryNode_ReleaseCollection_Call) RunAndReturn(run func(context.Co
 func (_m *MockQueryNode) ReleasePartitions(_a0 context.Context, _a1 *querypb.ReleasePartitionsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleasePartitions")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleasePartitionsRequest) (*commonpb.Status, error)); ok {
@@ -1054,6 +1134,10 @@ func (_c *MockQueryNode_ReleasePartitions_Call) RunAndReturn(run func(context.Co
 func (_m *MockQueryNode) ReleaseSegments(_a0 context.Context, _a1 *querypb.ReleaseSegmentsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleaseSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseSegmentsRequest) (*commonpb.Status, error)); ok {
@@ -1109,6 +1193,10 @@ func (_c *MockQueryNode_ReleaseSegments_Call) RunAndReturn(run func(context.Cont
 func (_m *MockQueryNode) Search(_a0 context.Context, _a1 *querypb.SearchRequest) (*internalpb.SearchResults, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Search")
+	}
+
 	var r0 *internalpb.SearchResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest) (*internalpb.SearchResults, error)); ok {
@@ -1164,6 +1252,10 @@ func (_c *MockQueryNode_Search_Call) RunAndReturn(run func(context.Context, *que
 func (_m *MockQueryNode) SearchSegments(_a0 context.Context, _a1 *querypb.SearchRequest) (*internalpb.SearchResults, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SearchSegments")
+	}
+
 	var r0 *internalpb.SearchResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest) (*internalpb.SearchResults, error)); ok {
@@ -1285,6 +1377,10 @@ func (_c *MockQueryNode_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Clie
 func (_m *MockQueryNode) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok {
@@ -1340,6 +1436,10 @@ func (_c *MockQueryNode_ShowConfigurations_Call) RunAndReturn(run func(context.C
 func (_m *MockQueryNode) Start() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Start")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -1381,6 +1481,10 @@ func (_c *MockQueryNode_Start_Call) RunAndReturn(run func() error) *MockQueryNod
 func (_m *MockQueryNode) Stop() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Stop")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -1422,6 +1526,10 @@ func (_c *MockQueryNode_Stop_Call) RunAndReturn(run func() error) *MockQueryNode
 func (_m *MockQueryNode) SyncDistribution(_a0 context.Context, _a1 *querypb.SyncDistributionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SyncDistribution")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncDistributionRequest) (*commonpb.Status, error)); ok {
@@ -1477,6 +1585,10 @@ func (_c *MockQueryNode_SyncDistribution_Call) RunAndReturn(run func(context.Con
 func (_m *MockQueryNode) SyncReplicaSegments(_a0 context.Context, _a1 *querypb.SyncReplicaSegmentsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SyncReplicaSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncReplicaSegmentsRequest) (*commonpb.Status, error)); ok {
@@ -1532,6 +1644,10 @@ func (_c *MockQueryNode_SyncReplicaSegments_Call) RunAndReturn(run func(context.
 func (_m *MockQueryNode) UnsubDmChannel(_a0 context.Context, _a1 *querypb.UnsubDmChannelRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UnsubDmChannel")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.UnsubDmChannelRequest) (*commonpb.Status, error)); ok {
@@ -1620,6 +1736,10 @@ func (_c *MockQueryNode_UpdateStateCode_Call) RunAndReturn(run func(commonpb.Sta
 func (_m *MockQueryNode) WatchDmChannels(_a0 context.Context, _a1 *querypb.WatchDmChannelsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WatchDmChannels")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.WatchDmChannelsRequest) (*commonpb.Status, error)); ok {

internal/mocks/mock_querynode_client.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -35,6 +35,10 @@ func (_m *MockQueryNodeClient) EXPECT() *MockQueryNodeClient_Expecter {
 func (_m *MockQueryNodeClient) Close() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Close")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -83,6 +87,10 @@ func (_m *MockQueryNodeClient) Delete(ctx context.Context, in *querypb.DeleteReq
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Delete")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.DeleteRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -153,6 +161,10 @@ func (_m *MockQueryNodeClient) GetComponentStates(ctx context.Context, in *milvu
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok {
@@ -223,6 +235,10 @@ func (_m *MockQueryNodeClient) GetDataDistribution(ctx context.Context, in *quer
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDataDistribution")
+	}
+
 	var r0 *querypb.GetDataDistributionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetDataDistributionRequest, ...grpc.CallOption) (*querypb.GetDataDistributionResponse, error)); ok {
@@ -293,6 +309,10 @@ func (_m *MockQueryNodeClient) GetMetrics(ctx context.Context, in *milvuspb.GetM
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -363,6 +383,10 @@ func (_m *MockQueryNodeClient) GetSegmentInfo(ctx context.Context, in *querypb.G
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentInfo")
+	}
+
 	var r0 *querypb.GetSegmentInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetSegmentInfoRequest, ...grpc.CallOption) (*querypb.GetSegmentInfoResponse, error)); ok {
@@ -433,6 +457,10 @@ func (_m *MockQueryNodeClient) GetStatistics(ctx context.Context, in *querypb.Ge
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatistics")
+	}
+
 	var r0 *internalpb.GetStatisticsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetStatisticsRequest, ...grpc.CallOption) (*internalpb.GetStatisticsResponse, error)); ok {
@@ -503,6 +531,10 @@ func (_m *MockQueryNodeClient) GetStatisticsChannel(ctx context.Context, in *int
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -573,6 +605,10 @@ func (_m *MockQueryNodeClient) GetTimeTickChannel(ctx context.Context, in *inter
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetTimeTickChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -643,6 +679,10 @@ func (_m *MockQueryNodeClient) LoadPartitions(ctx context.Context, in *querypb.L
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadPartitions")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadPartitionsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -713,6 +753,10 @@ func (_m *MockQueryNodeClient) LoadSegments(ctx context.Context, in *querypb.Loa
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadSegmentsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -783,6 +827,10 @@ func (_m *MockQueryNodeClient) Query(ctx context.Context, in *querypb.QueryReque
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Query")
+	}
+
 	var r0 *internalpb.RetrieveResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest, ...grpc.CallOption) (*internalpb.RetrieveResults, error)); ok {
@@ -853,6 +901,10 @@ func (_m *MockQueryNodeClient) QuerySegments(ctx context.Context, in *querypb.Qu
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QuerySegments")
+	}
+
 	var r0 *internalpb.RetrieveResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest, ...grpc.CallOption) (*internalpb.RetrieveResults, error)); ok {
@@ -923,6 +975,10 @@ func (_m *MockQueryNodeClient) QueryStream(ctx context.Context, in *querypb.Quer
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryStream")
+	}
+
 	var r0 querypb.QueryNode_QueryStreamClient
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest, ...grpc.CallOption) (querypb.QueryNode_QueryStreamClient, error)); ok {
@@ -993,6 +1049,10 @@ func (_m *MockQueryNodeClient) QueryStreamSegments(ctx context.Context, in *quer
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryStreamSegments")
+	}
+
 	var r0 querypb.QueryNode_QueryStreamSegmentsClient
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest, ...grpc.CallOption) (querypb.QueryNode_QueryStreamSegmentsClient, error)); ok {
@@ -1063,6 +1123,10 @@ func (_m *MockQueryNodeClient) ReleaseCollection(ctx context.Context, in *queryp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleaseCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1133,6 +1197,10 @@ func (_m *MockQueryNodeClient) ReleasePartitions(ctx context.Context, in *queryp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleasePartitions")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleasePartitionsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1203,6 +1271,10 @@ func (_m *MockQueryNodeClient) ReleaseSegments(ctx context.Context, in *querypb.
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleaseSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseSegmentsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1273,6 +1345,10 @@ func (_m *MockQueryNodeClient) Search(ctx context.Context, in *querypb.SearchReq
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Search")
+	}
+
 	var r0 *internalpb.SearchResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest, ...grpc.CallOption) (*internalpb.SearchResults, error)); ok {
@@ -1343,6 +1419,10 @@ func (_m *MockQueryNodeClient) SearchSegments(ctx context.Context, in *querypb.S
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SearchSegments")
+	}
+
 	var r0 *internalpb.SearchResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest, ...grpc.CallOption) (*internalpb.SearchResults, error)); ok {
@@ -1413,6 +1493,10 @@ func (_m *MockQueryNodeClient) ShowConfigurations(ctx context.Context, in *inter
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest, ...grpc.CallOption) (*internalpb.ShowConfigurationsResponse, error)); ok {
@@ -1483,6 +1567,10 @@ func (_m *MockQueryNodeClient) SyncDistribution(ctx context.Context, in *querypb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SyncDistribution")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncDistributionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1553,6 +1641,10 @@ func (_m *MockQueryNodeClient) SyncReplicaSegments(ctx context.Context, in *quer
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SyncReplicaSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncReplicaSegmentsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1623,6 +1715,10 @@ func (_m *MockQueryNodeClient) UnsubDmChannel(ctx context.Context, in *querypb.U
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UnsubDmChannel")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.UnsubDmChannelRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1693,6 +1789,10 @@ func (_m *MockQueryNodeClient) WatchDmChannels(ctx context.Context, in *querypb.
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WatchDmChannels")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.WatchDmChannelsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {

internal/mocks/mock_rootcoord.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -40,6 +40,10 @@ func (_m *RootCoord) EXPECT() *RootCoord_Expecter {
 func (_m *RootCoord) AllocID(_a0 context.Context, _a1 *rootcoordpb.AllocIDRequest) (*rootcoordpb.AllocIDResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocID")
+	}
+
 	var r0 *rootcoordpb.AllocIDResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AllocIDRequest) (*rootcoordpb.AllocIDResponse, error)); ok {
@@ -95,6 +99,10 @@ func (_c *RootCoord_AllocID_Call) RunAndReturn(run func(context.Context, *rootco
 func (_m *RootCoord) AllocTimestamp(_a0 context.Context, _a1 *rootcoordpb.AllocTimestampRequest) (*rootcoordpb.AllocTimestampResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocTimestamp")
+	}
+
 	var r0 *rootcoordpb.AllocTimestampResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AllocTimestampRequest) (*rootcoordpb.AllocTimestampResponse, error)); ok {
@@ -150,6 +158,10 @@ func (_c *RootCoord_AllocTimestamp_Call) RunAndReturn(run func(context.Context,
 func (_m *RootCoord) AlterAlias(_a0 context.Context, _a1 *milvuspb.AlterAliasRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AlterAlias")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterAliasRequest) (*commonpb.Status, error)); ok {
@@ -205,6 +217,10 @@ func (_c *RootCoord_AlterAlias_Call) RunAndReturn(run func(context.Context, *mil
 func (_m *RootCoord) AlterCollection(_a0 context.Context, _a1 *milvuspb.AlterCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AlterCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterCollectionRequest) (*commonpb.Status, error)); ok {
@@ -260,6 +276,10 @@ func (_c *RootCoord_AlterCollection_Call) RunAndReturn(run func(context.Context,
 func (_m *RootCoord) AlterDatabase(_a0 context.Context, _a1 *rootcoordpb.AlterDatabaseRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AlterDatabase")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AlterDatabaseRequest) (*commonpb.Status, error)); ok {
@@ -315,6 +335,10 @@ func (_c *RootCoord_AlterDatabase_Call) RunAndReturn(run func(context.Context, *
 func (_m *RootCoord) BackupRBAC(_a0 context.Context, _a1 *milvuspb.BackupRBACMetaRequest) (*milvuspb.BackupRBACMetaResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for BackupRBAC")
+	}
+
 	var r0 *milvuspb.BackupRBACMetaResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.BackupRBACMetaRequest) (*milvuspb.BackupRBACMetaResponse, error)); ok {
@@ -370,6 +394,10 @@ func (_c *RootCoord_BackupRBAC_Call) RunAndReturn(run func(context.Context, *mil
 func (_m *RootCoord) CheckHealth(_a0 context.Context, _a1 *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckHealth")
+	}
+
 	var r0 *milvuspb.CheckHealthResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error)); ok {
@@ -425,6 +453,10 @@ func (_c *RootCoord_CheckHealth_Call) RunAndReturn(run func(context.Context, *mi
 func (_m *RootCoord) CreateAlias(_a0 context.Context, _a1 *milvuspb.CreateAliasRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateAlias")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateAliasRequest) (*commonpb.Status, error)); ok {
@@ -480,6 +512,10 @@ func (_c *RootCoord_CreateAlias_Call) RunAndReturn(run func(context.Context, *mi
 func (_m *RootCoord) CreateCollection(_a0 context.Context, _a1 *milvuspb.CreateCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCollectionRequest) (*commonpb.Status, error)); ok {
@@ -535,6 +571,10 @@ func (_c *RootCoord_CreateCollection_Call) RunAndReturn(run func(context.Context
 func (_m *RootCoord) CreateCredential(_a0 context.Context, _a1 *internalpb.CredentialInfo) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateCredential")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.CredentialInfo) (*commonpb.Status, error)); ok {
@@ -590,6 +630,10 @@ func (_c *RootCoord_CreateCredential_Call) RunAndReturn(run func(context.Context
 func (_m *RootCoord) CreateDatabase(_a0 context.Context, _a1 *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateDatabase")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error)); ok {
@@ -645,6 +689,10 @@ func (_c *RootCoord_CreateDatabase_Call) RunAndReturn(run func(context.Context,
 func (_m *RootCoord) CreatePartition(_a0 context.Context, _a1 *milvuspb.CreatePartitionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreatePartition")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreatePartitionRequest) (*commonpb.Status, error)); ok {
@@ -700,6 +748,10 @@ func (_c *RootCoord_CreatePartition_Call) RunAndReturn(run func(context.Context,
 func (_m *RootCoord) CreateRole(_a0 context.Context, _a1 *milvuspb.CreateRoleRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateRole")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRoleRequest) (*commonpb.Status, error)); ok {
@@ -755,6 +807,10 @@ func (_c *RootCoord_CreateRole_Call) RunAndReturn(run func(context.Context, *mil
 func (_m *RootCoord) DeleteCredential(_a0 context.Context, _a1 *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteCredential")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error)); ok {
@@ -810,6 +866,10 @@ func (_c *RootCoord_DeleteCredential_Call) RunAndReturn(run func(context.Context
 func (_m *RootCoord) DescribeAlias(_a0 context.Context, _a1 *milvuspb.DescribeAliasRequest) (*milvuspb.DescribeAliasResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeAlias")
+	}
+
 	var r0 *milvuspb.DescribeAliasResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeAliasRequest) (*milvuspb.DescribeAliasResponse, error)); ok {
@@ -865,6 +925,10 @@ func (_c *RootCoord_DescribeAlias_Call) RunAndReturn(run func(context.Context, *
 func (_m *RootCoord) DescribeCollection(_a0 context.Context, _a1 *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeCollection")
+	}
+
 	var r0 *milvuspb.DescribeCollectionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error)); ok {
@@ -920,6 +984,10 @@ func (_c *RootCoord_DescribeCollection_Call) RunAndReturn(run func(context.Conte
 func (_m *RootCoord) DescribeCollectionInternal(_a0 context.Context, _a1 *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeCollectionInternal")
+	}
+
 	var r0 *milvuspb.DescribeCollectionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error)); ok {
@@ -975,6 +1043,10 @@ func (_c *RootCoord_DescribeCollectionInternal_Call) RunAndReturn(run func(conte
 func (_m *RootCoord) DescribeDatabase(_a0 context.Context, _a1 *rootcoordpb.DescribeDatabaseRequest) (*rootcoordpb.DescribeDatabaseResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeDatabase")
+	}
+
 	var r0 *rootcoordpb.DescribeDatabaseResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.DescribeDatabaseRequest) (*rootcoordpb.DescribeDatabaseResponse, error)); ok {
@@ -1030,6 +1102,10 @@ func (_c *RootCoord_DescribeDatabase_Call) RunAndReturn(run func(context.Context
 func (_m *RootCoord) DropAlias(_a0 context.Context, _a1 *milvuspb.DropAliasRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropAlias")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropAliasRequest) (*commonpb.Status, error)); ok {
@@ -1085,6 +1161,10 @@ func (_c *RootCoord_DropAlias_Call) RunAndReturn(run func(context.Context, *milv
 func (_m *RootCoord) DropCollection(_a0 context.Context, _a1 *milvuspb.DropCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropCollectionRequest) (*commonpb.Status, error)); ok {
@@ -1140,6 +1220,10 @@ func (_c *RootCoord_DropCollection_Call) RunAndReturn(run func(context.Context,
 func (_m *RootCoord) DropDatabase(_a0 context.Context, _a1 *milvuspb.DropDatabaseRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropDatabase")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropDatabaseRequest) (*commonpb.Status, error)); ok {
@@ -1195,6 +1279,10 @@ func (_c *RootCoord_DropDatabase_Call) RunAndReturn(run func(context.Context, *m
 func (_m *RootCoord) DropPartition(_a0 context.Context, _a1 *milvuspb.DropPartitionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropPartition")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropPartitionRequest) (*commonpb.Status, error)); ok {
@@ -1250,6 +1338,10 @@ func (_c *RootCoord_DropPartition_Call) RunAndReturn(run func(context.Context, *
 func (_m *RootCoord) DropRole(_a0 context.Context, _a1 *milvuspb.DropRoleRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropRole")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRoleRequest) (*commonpb.Status, error)); ok {
@@ -1305,6 +1397,10 @@ func (_c *RootCoord_DropRole_Call) RunAndReturn(run func(context.Context, *milvu
 func (_m *RootCoord) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok {
@@ -1360,6 +1456,10 @@ func (_c *RootCoord_GetComponentStates_Call) RunAndReturn(run func(context.Conte
 func (_m *RootCoord) GetCredential(_a0 context.Context, _a1 *rootcoordpb.GetCredentialRequest) (*rootcoordpb.GetCredentialResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCredential")
+	}
+
 	var r0 *rootcoordpb.GetCredentialResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.GetCredentialRequest) (*rootcoordpb.GetCredentialResponse, error)); ok {
@@ -1415,6 +1515,10 @@ func (_c *RootCoord_GetCredential_Call) RunAndReturn(run func(context.Context, *
 func (_m *RootCoord) GetMetrics(ctx context.Context, req *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) {
 	ret := _m.Called(ctx, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -1470,6 +1574,10 @@ func (_c *RootCoord_GetMetrics_Call) RunAndReturn(run func(context.Context, *mil
 func (_m *RootCoord) GetPChannelInfo(_a0 context.Context, _a1 *rootcoordpb.GetPChannelInfoRequest) (*rootcoordpb.GetPChannelInfoResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPChannelInfo")
+	}
+
 	var r0 *rootcoordpb.GetPChannelInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.GetPChannelInfoRequest) (*rootcoordpb.GetPChannelInfoResponse, error)); ok {
@@ -1525,6 +1633,10 @@ func (_c *RootCoord_GetPChannelInfo_Call) RunAndReturn(run func(context.Context,
 func (_m *RootCoord) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -1580,6 +1692,10 @@ func (_c *RootCoord_GetStatisticsChannel_Call) RunAndReturn(run func(context.Con
 func (_m *RootCoord) GetTimeTickChannel(_a0 context.Context, _a1 *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetTimeTickChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -1635,6 +1751,10 @@ func (_c *RootCoord_GetTimeTickChannel_Call) RunAndReturn(run func(context.Conte
 func (_m *RootCoord) HasCollection(_a0 context.Context, _a1 *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for HasCollection")
+	}
+
 	var r0 *milvuspb.BoolResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error)); ok {
@@ -1690,6 +1810,10 @@ func (_c *RootCoord_HasCollection_Call) RunAndReturn(run func(context.Context, *
 func (_m *RootCoord) HasPartition(_a0 context.Context, _a1 *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for HasPartition")
+	}
+
 	var r0 *milvuspb.BoolResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error)); ok {
@@ -1745,6 +1869,10 @@ func (_c *RootCoord_HasPartition_Call) RunAndReturn(run func(context.Context, *m
 func (_m *RootCoord) Init() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Init")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -1786,6 +1914,10 @@ func (_c *RootCoord_Init_Call) RunAndReturn(run func() error) *RootCoord_Init_Ca
 func (_m *RootCoord) InvalidateCollectionMetaCache(_a0 context.Context, _a1 *proxypb.InvalidateCollMetaCacheRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InvalidateCollectionMetaCache")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCollMetaCacheRequest) (*commonpb.Status, error)); ok {
@@ -1841,6 +1973,10 @@ func (_c *RootCoord_InvalidateCollectionMetaCache_Call) RunAndReturn(run func(co
 func (_m *RootCoord) ListAliases(_a0 context.Context, _a1 *milvuspb.ListAliasesRequest) (*milvuspb.ListAliasesResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListAliases")
+	}
+
 	var r0 *milvuspb.ListAliasesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListAliasesRequest) (*milvuspb.ListAliasesResponse, error)); ok {
@@ -1896,6 +2032,10 @@ func (_c *RootCoord_ListAliases_Call) RunAndReturn(run func(context.Context, *mi
 func (_m *RootCoord) ListCredUsers(_a0 context.Context, _a1 *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListCredUsers")
+	}
+
 	var r0 *milvuspb.ListCredUsersResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error)); ok {
@@ -1951,6 +2091,10 @@ func (_c *RootCoord_ListCredUsers_Call) RunAndReturn(run func(context.Context, *
 func (_m *RootCoord) ListDatabases(_a0 context.Context, _a1 *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListDatabases")
+	}
+
 	var r0 *milvuspb.ListDatabasesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error)); ok {
@@ -2006,6 +2150,10 @@ func (_c *RootCoord_ListDatabases_Call) RunAndReturn(run func(context.Context, *
 func (_m *RootCoord) ListPolicy(_a0 context.Context, _a1 *internalpb.ListPolicyRequest) (*internalpb.ListPolicyResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListPolicy")
+	}
+
 	var r0 *internalpb.ListPolicyResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListPolicyRequest) (*internalpb.ListPolicyResponse, error)); ok {
@@ -2061,6 +2209,10 @@ func (_c *RootCoord_ListPolicy_Call) RunAndReturn(run func(context.Context, *int
 func (_m *RootCoord) OperatePrivilege(_a0 context.Context, _a1 *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for OperatePrivilege")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error)); ok {
@@ -2116,6 +2268,10 @@ func (_c *RootCoord_OperatePrivilege_Call) RunAndReturn(run func(context.Context
 func (_m *RootCoord) OperateUserRole(_a0 context.Context, _a1 *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for OperateUserRole")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error)); ok {
@@ -2171,6 +2327,10 @@ func (_c *RootCoord_OperateUserRole_Call) RunAndReturn(run func(context.Context,
 func (_m *RootCoord) Register() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Register")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -2212,6 +2372,10 @@ func (_c *RootCoord_Register_Call) RunAndReturn(run func() error) *RootCoord_Reg
 func (_m *RootCoord) RenameCollection(_a0 context.Context, _a1 *milvuspb.RenameCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RenameCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RenameCollectionRequest) (*commonpb.Status, error)); ok {
@@ -2267,6 +2431,10 @@ func (_c *RootCoord_RenameCollection_Call) RunAndReturn(run func(context.Context
 func (_m *RootCoord) RestoreRBAC(_a0 context.Context, _a1 *milvuspb.RestoreRBACMetaRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RestoreRBAC")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RestoreRBACMetaRequest) (*commonpb.Status, error)); ok {
@@ -2322,6 +2490,10 @@ func (_c *RootCoord_RestoreRBAC_Call) RunAndReturn(run func(context.Context, *mi
 func (_m *RootCoord) SelectGrant(_a0 context.Context, _a1 *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SelectGrant")
+	}
+
 	var r0 *milvuspb.SelectGrantResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error)); ok {
@@ -2377,6 +2549,10 @@ func (_c *RootCoord_SelectGrant_Call) RunAndReturn(run func(context.Context, *mi
 func (_m *RootCoord) SelectRole(_a0 context.Context, _a1 *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SelectRole")
+	}
+
 	var r0 *milvuspb.SelectRoleResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error)); ok {
@@ -2432,6 +2608,10 @@ func (_c *RootCoord_SelectRole_Call) RunAndReturn(run func(context.Context, *mil
 func (_m *RootCoord) SelectUser(_a0 context.Context, _a1 *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SelectUser")
+	}
+
 	var r0 *milvuspb.SelectUserResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error)); ok {
@@ -2520,6 +2700,10 @@ func (_c *RootCoord_SetAddress_Call) RunAndReturn(run func(string)) *RootCoord_S
 func (_m *RootCoord) SetDataCoordClient(dataCoord types.DataCoordClient) error {
 	ret := _m.Called(dataCoord)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetDataCoordClient")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(types.DataCoordClient) error); ok {
 		r0 = rf(dataCoord)
@@ -2628,6 +2812,10 @@ func (_c *RootCoord_SetProxyCreator_Call) RunAndReturn(run func(func(context.Con
 func (_m *RootCoord) SetQueryCoordClient(queryCoord types.QueryCoordClient) error {
 	ret := _m.Called(queryCoord)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetQueryCoordClient")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(types.QueryCoordClient) error); ok {
 		r0 = rf(queryCoord)
@@ -2703,6 +2891,10 @@ func (_c *RootCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client)) *R
 func (_m *RootCoord) ShowCollections(_a0 context.Context, _a1 *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowCollections")
+	}
+
 	var r0 *milvuspb.ShowCollectionsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error)); ok {
@@ -2758,6 +2950,10 @@ func (_c *RootCoord_ShowCollections_Call) RunAndReturn(run func(context.Context,
 func (_m *RootCoord) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok {
@@ -2813,6 +3009,10 @@ func (_c *RootCoord_ShowConfigurations_Call) RunAndReturn(run func(context.Conte
 func (_m *RootCoord) ShowPartitions(_a0 context.Context, _a1 *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowPartitions")
+	}
+
 	var r0 *milvuspb.ShowPartitionsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error)); ok {
@@ -2868,6 +3068,10 @@ func (_c *RootCoord_ShowPartitions_Call) RunAndReturn(run func(context.Context,
 func (_m *RootCoord) ShowPartitionsInternal(_a0 context.Context, _a1 *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowPartitionsInternal")
+	}
+
 	var r0 *milvuspb.ShowPartitionsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error)); ok {
@@ -2923,6 +3127,10 @@ func (_c *RootCoord_ShowPartitionsInternal_Call) RunAndReturn(run func(context.C
 func (_m *RootCoord) ShowSegments(_a0 context.Context, _a1 *milvuspb.ShowSegmentsRequest) (*milvuspb.ShowSegmentsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowSegments")
+	}
+
 	var r0 *milvuspb.ShowSegmentsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowSegmentsRequest) (*milvuspb.ShowSegmentsResponse, error)); ok {
@@ -2978,6 +3186,10 @@ func (_c *RootCoord_ShowSegments_Call) RunAndReturn(run func(context.Context, *m
 func (_m *RootCoord) Start() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Start")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -3019,6 +3231,10 @@ func (_c *RootCoord_Start_Call) RunAndReturn(run func() error) *RootCoord_Start_
 func (_m *RootCoord) Stop() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Stop")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -3060,6 +3276,10 @@ func (_c *RootCoord_Stop_Call) RunAndReturn(run func() error) *RootCoord_Stop_Ca
 func (_m *RootCoord) UpdateChannelTimeTick(_a0 context.Context, _a1 *internalpb.ChannelTimeTickMsg) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateChannelTimeTick")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ChannelTimeTickMsg) (*commonpb.Status, error)); ok {
@@ -3115,6 +3335,10 @@ func (_c *RootCoord_UpdateChannelTimeTick_Call) RunAndReturn(run func(context.Co
 func (_m *RootCoord) UpdateCredential(_a0 context.Context, _a1 *internalpb.CredentialInfo) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateCredential")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.CredentialInfo) (*commonpb.Status, error)); ok {

internal/mocks/mock_rootcoord_client.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -44,6 +44,10 @@ func (_m *MockRootCoordClient) AllocID(ctx context.Context, in *rootcoordpb.Allo
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocID")
+	}
+
 	var r0 *rootcoordpb.AllocIDResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AllocIDRequest, ...grpc.CallOption) (*rootcoordpb.AllocIDResponse, error)); ok {
@@ -114,6 +118,10 @@ func (_m *MockRootCoordClient) AllocTimestamp(ctx context.Context, in *rootcoord
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocTimestamp")
+	}
+
 	var r0 *rootcoordpb.AllocTimestampResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AllocTimestampRequest, ...grpc.CallOption) (*rootcoordpb.AllocTimestampResponse, error)); ok {
@@ -184,6 +192,10 @@ func (_m *MockRootCoordClient) AlterAlias(ctx context.Context, in *milvuspb.Alte
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AlterAlias")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterAliasRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -254,6 +266,10 @@ func (_m *MockRootCoordClient) AlterCollection(ctx context.Context, in *milvuspb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AlterCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -324,6 +340,10 @@ func (_m *MockRootCoordClient) AlterDatabase(ctx context.Context, in *rootcoordp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AlterDatabase")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AlterDatabaseRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -394,6 +414,10 @@ func (_m *MockRootCoordClient) BackupRBAC(ctx context.Context, in *milvuspb.Back
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for BackupRBAC")
+	}
+
 	var r0 *milvuspb.BackupRBACMetaResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.BackupRBACMetaRequest, ...grpc.CallOption) (*milvuspb.BackupRBACMetaResponse, error)); ok {
@@ -464,6 +488,10 @@ func (_m *MockRootCoordClient) CheckHealth(ctx context.Context, in *milvuspb.Che
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckHealth")
+	}
+
 	var r0 *milvuspb.CheckHealthResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest, ...grpc.CallOption) (*milvuspb.CheckHealthResponse, error)); ok {
@@ -527,6 +555,10 @@ func (_c *MockRootCoordClient_CheckHealth_Call) RunAndReturn(run func(context.Co
 func (_m *MockRootCoordClient) Close() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Close")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -575,6 +607,10 @@ func (_m *MockRootCoordClient) CreateAlias(ctx context.Context, in *milvuspb.Cre
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateAlias")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateAliasRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -645,6 +681,10 @@ func (_m *MockRootCoordClient) CreateCollection(ctx context.Context, in *milvusp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -715,6 +755,10 @@ func (_m *MockRootCoordClient) CreateCredential(ctx context.Context, in *interna
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateCredential")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.CredentialInfo, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -785,6 +829,10 @@ func (_m *MockRootCoordClient) CreateDatabase(ctx context.Context, in *milvuspb.
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateDatabase")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateDatabaseRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -855,6 +903,10 @@ func (_m *MockRootCoordClient) CreatePartition(ctx context.Context, in *milvuspb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreatePartition")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreatePartitionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -925,6 +977,10 @@ func (_m *MockRootCoordClient) CreateRole(ctx context.Context, in *milvuspb.Crea
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateRole")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRoleRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -995,6 +1051,10 @@ func (_m *MockRootCoordClient) DeleteCredential(ctx context.Context, in *milvusp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteCredential")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteCredentialRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1065,6 +1125,10 @@ func (_m *MockRootCoordClient) DescribeAlias(ctx context.Context, in *milvuspb.D
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeAlias")
+	}
+
 	var r0 *milvuspb.DescribeAliasResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeAliasRequest, ...grpc.CallOption) (*milvuspb.DescribeAliasResponse, error)); ok {
@@ -1135,6 +1199,10 @@ func (_m *MockRootCoordClient) DescribeCollection(ctx context.Context, in *milvu
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeCollection")
+	}
+
 	var r0 *milvuspb.DescribeCollectionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest, ...grpc.CallOption) (*milvuspb.DescribeCollectionResponse, error)); ok {
@@ -1205,6 +1273,10 @@ func (_m *MockRootCoordClient) DescribeCollectionInternal(ctx context.Context, i
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeCollectionInternal")
+	}
+
 	var r0 *milvuspb.DescribeCollectionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest, ...grpc.CallOption) (*milvuspb.DescribeCollectionResponse, error)); ok {
@@ -1275,6 +1347,10 @@ func (_m *MockRootCoordClient) DescribeDatabase(ctx context.Context, in *rootcoo
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeDatabase")
+	}
+
 	var r0 *rootcoordpb.DescribeDatabaseResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.DescribeDatabaseRequest, ...grpc.CallOption) (*rootcoordpb.DescribeDatabaseResponse, error)); ok {
@@ -1345,6 +1421,10 @@ func (_m *MockRootCoordClient) DropAlias(ctx context.Context, in *milvuspb.DropA
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropAlias")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropAliasRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1415,6 +1495,10 @@ func (_m *MockRootCoordClient) DropCollection(ctx context.Context, in *milvuspb.
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1485,6 +1569,10 @@ func (_m *MockRootCoordClient) DropDatabase(ctx context.Context, in *milvuspb.Dr
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropDatabase")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropDatabaseRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1555,6 +1643,10 @@ func (_m *MockRootCoordClient) DropPartition(ctx context.Context, in *milvuspb.D
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropPartition")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropPartitionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1625,6 +1717,10 @@ func (_m *MockRootCoordClient) DropRole(ctx context.Context, in *milvuspb.DropRo
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropRole")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRoleRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -1695,6 +1791,10 @@ func (_m *MockRootCoordClient) GetComponentStates(ctx context.Context, in *milvu
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok {
@@ -1765,6 +1865,10 @@ func (_m *MockRootCoordClient) GetCredential(ctx context.Context, in *rootcoordp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCredential")
+	}
+
 	var r0 *rootcoordpb.GetCredentialResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.GetCredentialRequest, ...grpc.CallOption) (*rootcoordpb.GetCredentialResponse, error)); ok {
@@ -1835,6 +1939,10 @@ func (_m *MockRootCoordClient) GetMetrics(ctx context.Context, in *milvuspb.GetM
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -1905,6 +2013,10 @@ func (_m *MockRootCoordClient) GetPChannelInfo(ctx context.Context, in *rootcoor
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPChannelInfo")
+	}
+
 	var r0 *rootcoordpb.GetPChannelInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.GetPChannelInfoRequest, ...grpc.CallOption) (*rootcoordpb.GetPChannelInfoResponse, error)); ok {
@@ -1975,6 +2087,10 @@ func (_m *MockRootCoordClient) GetStatisticsChannel(ctx context.Context, in *int
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -2045,6 +2161,10 @@ func (_m *MockRootCoordClient) GetTimeTickChannel(ctx context.Context, in *inter
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetTimeTickChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok {
@@ -2115,6 +2235,10 @@ func (_m *MockRootCoordClient) HasCollection(ctx context.Context, in *milvuspb.H
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for HasCollection")
+	}
+
 	var r0 *milvuspb.BoolResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasCollectionRequest, ...grpc.CallOption) (*milvuspb.BoolResponse, error)); ok {
@@ -2185,6 +2309,10 @@ func (_m *MockRootCoordClient) HasPartition(ctx context.Context, in *milvuspb.Ha
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for HasPartition")
+	}
+
 	var r0 *milvuspb.BoolResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasPartitionRequest, ...grpc.CallOption) (*milvuspb.BoolResponse, error)); ok {
@@ -2255,6 +2383,10 @@ func (_m *MockRootCoordClient) InvalidateCollectionMetaCache(ctx context.Context
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InvalidateCollectionMetaCache")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCollMetaCacheRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2325,6 +2457,10 @@ func (_m *MockRootCoordClient) ListAliases(ctx context.Context, in *milvuspb.Lis
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListAliases")
+	}
+
 	var r0 *milvuspb.ListAliasesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListAliasesRequest, ...grpc.CallOption) (*milvuspb.ListAliasesResponse, error)); ok {
@@ -2395,6 +2531,10 @@ func (_m *MockRootCoordClient) ListCredUsers(ctx context.Context, in *milvuspb.L
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListCredUsers")
+	}
+
 	var r0 *milvuspb.ListCredUsersResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListCredUsersRequest, ...grpc.CallOption) (*milvuspb.ListCredUsersResponse, error)); ok {
@@ -2465,6 +2605,10 @@ func (_m *MockRootCoordClient) ListDatabases(ctx context.Context, in *milvuspb.L
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListDatabases")
+	}
+
 	var r0 *milvuspb.ListDatabasesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListDatabasesRequest, ...grpc.CallOption) (*milvuspb.ListDatabasesResponse, error)); ok {
@@ -2535,6 +2679,10 @@ func (_m *MockRootCoordClient) ListPolicy(ctx context.Context, in *internalpb.Li
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListPolicy")
+	}
+
 	var r0 *internalpb.ListPolicyResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListPolicyRequest, ...grpc.CallOption) (*internalpb.ListPolicyResponse, error)); ok {
@@ -2605,6 +2753,10 @@ func (_m *MockRootCoordClient) OperatePrivilege(ctx context.Context, in *milvusp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for OperatePrivilege")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperatePrivilegeRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2675,6 +2827,10 @@ func (_m *MockRootCoordClient) OperateUserRole(ctx context.Context, in *milvuspb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for OperateUserRole")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperateUserRoleRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2745,6 +2901,10 @@ func (_m *MockRootCoordClient) RenameCollection(ctx context.Context, in *milvusp
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RenameCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RenameCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2815,6 +2975,10 @@ func (_m *MockRootCoordClient) RestoreRBAC(ctx context.Context, in *milvuspb.Res
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RestoreRBAC")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RestoreRBACMetaRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -2885,6 +3049,10 @@ func (_m *MockRootCoordClient) SelectGrant(ctx context.Context, in *milvuspb.Sel
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SelectGrant")
+	}
+
 	var r0 *milvuspb.SelectGrantResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectGrantRequest, ...grpc.CallOption) (*milvuspb.SelectGrantResponse, error)); ok {
@@ -2955,6 +3123,10 @@ func (_m *MockRootCoordClient) SelectRole(ctx context.Context, in *milvuspb.Sele
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SelectRole")
+	}
+
 	var r0 *milvuspb.SelectRoleResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectRoleRequest, ...grpc.CallOption) (*milvuspb.SelectRoleResponse, error)); ok {
@@ -3025,6 +3197,10 @@ func (_m *MockRootCoordClient) SelectUser(ctx context.Context, in *milvuspb.Sele
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SelectUser")
+	}
+
 	var r0 *milvuspb.SelectUserResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectUserRequest, ...grpc.CallOption) (*milvuspb.SelectUserResponse, error)); ok {
@@ -3095,6 +3271,10 @@ func (_m *MockRootCoordClient) ShowCollections(ctx context.Context, in *milvuspb
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowCollections")
+	}
+
 	var r0 *milvuspb.ShowCollectionsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowCollectionsRequest, ...grpc.CallOption) (*milvuspb.ShowCollectionsResponse, error)); ok {
@@ -3165,6 +3345,10 @@ func (_m *MockRootCoordClient) ShowConfigurations(ctx context.Context, in *inter
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest, ...grpc.CallOption) (*internalpb.ShowConfigurationsResponse, error)); ok {
@@ -3235,6 +3419,10 @@ func (_m *MockRootCoordClient) ShowPartitions(ctx context.Context, in *milvuspb.
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowPartitions")
+	}
+
 	var r0 *milvuspb.ShowPartitionsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest, ...grpc.CallOption) (*milvuspb.ShowPartitionsResponse, error)); ok {
@@ -3305,6 +3493,10 @@ func (_m *MockRootCoordClient) ShowPartitionsInternal(ctx context.Context, in *m
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowPartitionsInternal")
+	}
+
 	var r0 *milvuspb.ShowPartitionsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest, ...grpc.CallOption) (*milvuspb.ShowPartitionsResponse, error)); ok {
@@ -3375,6 +3567,10 @@ func (_m *MockRootCoordClient) ShowSegments(ctx context.Context, in *milvuspb.Sh
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowSegments")
+	}
+
 	var r0 *milvuspb.ShowSegmentsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowSegmentsRequest, ...grpc.CallOption) (*milvuspb.ShowSegmentsResponse, error)); ok {
@@ -3445,6 +3641,10 @@ func (_m *MockRootCoordClient) UpdateChannelTimeTick(ctx context.Context, in *in
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateChannelTimeTick")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ChannelTimeTickMsg, ...grpc.CallOption) (*commonpb.Status, error)); ok {
@@ -3515,6 +3715,10 @@ func (_m *MockRootCoordClient) UpdateCredential(ctx context.Context, in *interna
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateCredential")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.CredentialInfo, ...grpc.CallOption) (*commonpb.Status, error)); ok {

internal/mocks/streamingcoord/mock_client/mock_Client.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_client
 
@@ -24,6 +24,10 @@ func (_m *MockClient) EXPECT() *MockClient_Expecter {
 func (_m *MockClient) Assignment() client.AssignmentService {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Assignment")
+	}
+
 	var r0 client.AssignmentService
 	if rf, ok := ret.Get(0).(func() client.AssignmentService); ok {
 		r0 = rf()

internal/mocks/streamingcoord/server/mock_balancer/mock_Balancer.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_balancer
 
@@ -60,6 +60,10 @@ func (_c *MockBalancer_Close_Call) RunAndReturn(run func()) *MockBalancer_Close_
 func (_m *MockBalancer) MarkAsUnavailable(ctx context.Context, pChannels []types.PChannelInfo) error {
 	ret := _m.Called(ctx, pChannels)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MarkAsUnavailable")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, []types.PChannelInfo) error); ok {
 		r0 = rf(ctx, pChannels)
@@ -103,6 +107,10 @@ func (_c *MockBalancer_MarkAsUnavailable_Call) RunAndReturn(run func(context.Con
 func (_m *MockBalancer) Trigger(ctx context.Context) error {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Trigger")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
 		r0 = rf(ctx)
@@ -145,6 +153,10 @@ func (_c *MockBalancer_Trigger_Call) RunAndReturn(run func(context.Context) erro
 func (_m *MockBalancer) WatchChannelAssignments(ctx context.Context, cb func(typeutil.VersionInt64Pair, []types.PChannelInfoAssigned) error) error {
 	ret := _m.Called(ctx, cb)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WatchChannelAssignments")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, func(typeutil.VersionInt64Pair, []types.PChannelInfoAssigned) error) error); ok {
 		r0 = rf(ctx, cb)

internal/mocks/streamingnode/client/handler/mock_assignment/mock_Watcher.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_assignment
 
@@ -58,6 +58,10 @@ func (_c *MockWatcher_Close_Call) RunAndReturn(run func()) *MockWatcher_Close_Ca
 func (_m *MockWatcher) Get(ctx context.Context, channel string) *types.PChannelInfoAssigned {
 	ret := _m.Called(ctx, channel)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Get")
+	}
+
 	var r0 *types.PChannelInfoAssigned
 	if rf, ok := ret.Get(0).(func(context.Context, string) *types.PChannelInfoAssigned); ok {
 		r0 = rf(ctx, channel)
@@ -103,6 +107,10 @@ func (_c *MockWatcher_Get_Call) RunAndReturn(run func(context.Context, string) *
 func (_m *MockWatcher) Watch(ctx context.Context, channel string, previous *types.PChannelInfoAssigned) error {
 	ret := _m.Called(ctx, channel, previous)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Watch")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, *types.PChannelInfoAssigned) error); ok {
 		r0 = rf(ctx, channel, previous)

internal/mocks/streamingnode/client/handler/mock_consumer/mock_Consumer.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_consumer
 
@@ -53,6 +53,10 @@ func (_c *MockConsumer_Close_Call) RunAndReturn(run func()) *MockConsumer_Close_
 func (_m *MockConsumer) Done() <-chan struct{} {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Done")
+	}
+
 	var r0 <-chan struct{}
 	if rf, ok := ret.Get(0).(func() <-chan struct{}); ok {
 		r0 = rf()
@@ -96,6 +100,10 @@ func (_c *MockConsumer_Done_Call) RunAndReturn(run func() <-chan struct{}) *Mock
 func (_m *MockConsumer) Error() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Error")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()

internal/mocks/streamingnode/client/handler/mock_producer/mock_Producer.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_producer
 
@@ -28,6 +28,10 @@ func (_m *MockProducer) EXPECT() *MockProducer_Expecter {
 func (_m *MockProducer) Assignment() types.PChannelInfoAssigned {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Assignment")
+	}
+
 	var r0 types.PChannelInfoAssigned
 	if rf, ok := ret.Get(0).(func() types.PChannelInfoAssigned); ok {
 		r0 = rf()
@@ -69,6 +73,10 @@ func (_c *MockProducer_Assignment_Call) RunAndReturn(run func() types.PChannelIn
 func (_m *MockProducer) Available() <-chan struct{} {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Available")
+	}
+
 	var r0 <-chan struct{}
 	if rf, ok := ret.Get(0).(func() <-chan struct{}); ok {
 		r0 = rf()
@@ -144,6 +152,10 @@ func (_c *MockProducer_Close_Call) RunAndReturn(run func()) *MockProducer_Close_
 func (_m *MockProducer) IsAvailable() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsAvailable")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()
@@ -185,6 +197,10 @@ func (_c *MockProducer_IsAvailable_Call) RunAndReturn(run func() bool) *MockProd
 func (_m *MockProducer) Produce(ctx context.Context, msg message.MutableMessage) (*types.AppendResult, error) {
 	ret := _m.Called(ctx, msg)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Produce")
+	}
+
 	var r0 *types.AppendResult
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, message.MutableMessage) (*types.AppendResult, error)); ok {

internal/mocks/streamingnode/client/mock_handler/mock_HandlerClient.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_handler
 
@@ -63,6 +63,10 @@ func (_c *MockHandlerClient_Close_Call) RunAndReturn(run func()) *MockHandlerCli
 func (_m *MockHandlerClient) CreateConsumer(ctx context.Context, opts *handler.ConsumerOptions) (consumer.Consumer, error) {
 	ret := _m.Called(ctx, opts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateConsumer")
+	}
+
 	var r0 consumer.Consumer
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *handler.ConsumerOptions) (consumer.Consumer, error)); ok {
@@ -118,6 +122,10 @@ func (_c *MockHandlerClient_CreateConsumer_Call) RunAndReturn(run func(context.C
 func (_m *MockHandlerClient) CreateProducer(ctx context.Context, opts *handler.ProducerOptions) (producer.Producer, error) {
 	ret := _m.Called(ctx, opts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateProducer")
+	}
+
 	var r0 producer.Producer
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *handler.ProducerOptions) (producer.Producer, error)); ok {

internal/mocks/streamingnode/client/mock_manager/mock_ManagerClient.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_manager
 
@@ -27,6 +27,10 @@ func (_m *MockManagerClient) EXPECT() *MockManagerClient_Expecter {
 func (_m *MockManagerClient) Assign(ctx context.Context, pchannel types.PChannelInfoAssigned) error {
 	ret := _m.Called(ctx, pchannel)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Assign")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, types.PChannelInfoAssigned) error); ok {
 		r0 = rf(ctx, pchannel)
@@ -102,6 +106,10 @@ func (_c *MockManagerClient_Close_Call) RunAndReturn(run func()) *MockManagerCli
 func (_m *MockManagerClient) CollectAllStatus(ctx context.Context) (map[int64]*types.StreamingNodeStatus, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CollectAllStatus")
+	}
+
 	var r0 map[int64]*types.StreamingNodeStatus
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (map[int64]*types.StreamingNodeStatus, error)); ok {
@@ -156,6 +164,10 @@ func (_c *MockManagerClient_CollectAllStatus_Call) RunAndReturn(run func(context
 func (_m *MockManagerClient) Remove(ctx context.Context, pchannel types.PChannelInfoAssigned) error {
 	ret := _m.Called(ctx, pchannel)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Remove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, types.PChannelInfoAssigned) error); ok {
 		r0 = rf(ctx, pchannel)
@@ -199,6 +211,10 @@ func (_c *MockManagerClient_Remove_Call) RunAndReturn(run func(context.Context,
 func (_m *MockManagerClient) WatchNodeChanged(ctx context.Context) (<-chan struct{}, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WatchNodeChanged")
+	}
+
 	var r0 <-chan struct{}
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (<-chan struct{}, error)); ok {

internal/mocks/streamingnode/server/mock_flusher/mock_Flusher.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_flusher
 
@@ -24,6 +24,10 @@ func (_m *MockFlusher) EXPECT() *MockFlusher_Expecter {
 func (_m *MockFlusher) RegisterPChannel(pchannel string, w wal.WAL) error {
 	ret := _m.Called(pchannel, w)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RegisterPChannel")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, wal.WAL) error); ok {
 		r0 = rf(pchannel, w)

internal/mocks/streamingnode/server/mock_wal/mock_Opener.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_wal
 
@@ -58,6 +58,10 @@ func (_c *MockOpener_Close_Call) RunAndReturn(run func()) *MockOpener_Close_Call
 func (_m *MockOpener) Open(ctx context.Context, opt *wal.OpenOption) (wal.WAL, error) {
 	ret := _m.Called(ctx, opt)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Open")
+	}
+
 	var r0 wal.WAL
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *wal.OpenOption) (wal.WAL, error)); ok {

internal/mocks/streamingnode/server/mock_wal/mock_OpenerBuilder.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_wal
 
@@ -24,6 +24,10 @@ func (_m *MockOpenerBuilder) EXPECT() *MockOpenerBuilder_Expecter {
 func (_m *MockOpenerBuilder) Build() (wal.Opener, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Build")
+	}
+
 	var r0 wal.Opener
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (wal.Opener, error)); ok {
@@ -77,6 +81,10 @@ func (_c *MockOpenerBuilder_Build_Call) RunAndReturn(run func() (wal.Opener, err
 func (_m *MockOpenerBuilder) Name() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Name")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()

internal/mocks/streamingnode/server/mock_wal/mock_Scanner.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_wal
 
@@ -26,6 +26,10 @@ func (_m *MockScanner) EXPECT() *MockScanner_Expecter {
 func (_m *MockScanner) Chan() <-chan message.ImmutableMessage {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Chan")
+	}
+
 	var r0 <-chan message.ImmutableMessage
 	if rf, ok := ret.Get(0).(func() <-chan message.ImmutableMessage); ok {
 		r0 = rf()
@@ -69,6 +73,10 @@ func (_c *MockScanner_Chan_Call) RunAndReturn(run func() <-chan message.Immutabl
 func (_m *MockScanner) Channel() types.PChannelInfo {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Channel")
+	}
+
 	var r0 types.PChannelInfo
 	if rf, ok := ret.Get(0).(func() types.PChannelInfo); ok {
 		r0 = rf()
@@ -110,6 +118,10 @@ func (_c *MockScanner_Channel_Call) RunAndReturn(run func() types.PChannelInfo)
 func (_m *MockScanner) Close() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Close")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -151,6 +163,10 @@ func (_c *MockScanner_Close_Call) RunAndReturn(run func() error) *MockScanner_Cl
 func (_m *MockScanner) Done() <-chan struct{} {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Done")
+	}
+
 	var r0 <-chan struct{}
 	if rf, ok := ret.Get(0).(func() <-chan struct{}); ok {
 		r0 = rf()
@@ -194,6 +210,10 @@ func (_c *MockScanner_Done_Call) RunAndReturn(run func() <-chan struct{}) *MockS
 func (_m *MockScanner) Error() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Error")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()

internal/mocks/streamingnode/server/mock_wal/mock_WAL.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_wal
 
@@ -30,6 +30,10 @@ func (_m *MockWAL) EXPECT() *MockWAL_Expecter {
 func (_m *MockWAL) Append(ctx context.Context, msg message.MutableMessage) (*types.AppendResult, error) {
 	ret := _m.Called(ctx, msg)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Append")
+	}
+
 	var r0 *types.AppendResult
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, message.MutableMessage) (*types.AppendResult, error)); ok {
@@ -120,6 +124,10 @@ func (_c *MockWAL_AppendAsync_Call) RunAndReturn(run func(context.Context, messa
 func (_m *MockWAL) Available() <-chan struct{} {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Available")
+	}
+
 	var r0 <-chan struct{}
 	if rf, ok := ret.Get(0).(func() <-chan struct{}); ok {
 		r0 = rf()
@@ -163,6 +171,10 @@ func (_c *MockWAL_Available_Call) RunAndReturn(run func() <-chan struct{}) *Mock
 func (_m *MockWAL) Channel() types.PChannelInfo {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Channel")
+	}
+
 	var r0 types.PChannelInfo
 	if rf, ok := ret.Get(0).(func() types.PChannelInfo); ok {
 		r0 = rf()
@@ -236,6 +248,10 @@ func (_c *MockWAL_Close_Call) RunAndReturn(run func()) *MockWAL_Close_Call {
 func (_m *MockWAL) IsAvailable() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsAvailable")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()
@@ -277,6 +293,10 @@ func (_c *MockWAL_IsAvailable_Call) RunAndReturn(run func() bool) *MockWAL_IsAva
 func (_m *MockWAL) Read(ctx context.Context, deliverPolicy wal.ReadOption) (wal.Scanner, error) {
 	ret := _m.Called(ctx, deliverPolicy)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Read")
+	}
+
 	var r0 wal.Scanner
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, wal.ReadOption) (wal.Scanner, error)); ok {
@@ -332,6 +352,10 @@ func (_c *MockWAL_Read_Call) RunAndReturn(run func(context.Context, wal.ReadOpti
 func (_m *MockWAL) WALName() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for WALName")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()

internal/mocks/streamingnode/server/mock_walmanager/mock_Manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_walmanager
 
@@ -60,6 +60,10 @@ func (_c *MockManager_Close_Call) RunAndReturn(run func()) *MockManager_Close_Ca
 func (_m *MockManager) GetAllAvailableChannels() ([]types.PChannelInfo, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAllAvailableChannels")
+	}
+
 	var r0 []types.PChannelInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func() ([]types.PChannelInfo, error)); ok {
@@ -113,6 +117,10 @@ func (_c *MockManager_GetAllAvailableChannels_Call) RunAndReturn(run func() ([]t
 func (_m *MockManager) GetAvailableWAL(channel types.PChannelInfo) (wal.WAL, error) {
 	ret := _m.Called(channel)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAvailableWAL")
+	}
+
 	var r0 wal.WAL
 	var r1 error
 	if rf, ok := ret.Get(0).(func(types.PChannelInfo) (wal.WAL, error)); ok {
@@ -167,6 +175,10 @@ func (_c *MockManager_GetAvailableWAL_Call) RunAndReturn(run func(types.PChannel
 func (_m *MockManager) Open(ctx context.Context, channel types.PChannelInfo) error {
 	ret := _m.Called(ctx, channel)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Open")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, types.PChannelInfo) error); ok {
 		r0 = rf(ctx, channel)
@@ -210,6 +222,10 @@ func (_c *MockManager_Open_Call) RunAndReturn(run func(context.Context, types.PC
 func (_m *MockManager) Remove(ctx context.Context, channel types.PChannelInfo) error {
 	ret := _m.Called(ctx, channel)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Remove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, types.PChannelInfo) error); ok {
 		r0 = rf(ctx, channel)

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

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_inspector
 
@@ -29,6 +29,10 @@ func (_m *MockSealOperator) EXPECT() *MockSealOperator_Expecter {
 func (_m *MockSealOperator) Channel() types.PChannelInfo {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Channel")
+	}
+
 	var r0 types.PChannelInfo
 	if rf, ok := ret.Get(0).(func() types.PChannelInfo); ok {
 		r0 = rf()
@@ -70,6 +74,10 @@ func (_c *MockSealOperator_Channel_Call) RunAndReturn(run func() types.PChannelI
 func (_m *MockSealOperator) IsNoWaitSeal() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsNoWaitSeal")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()

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

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_inspector
 
@@ -28,6 +28,10 @@ func (_m *MockTimeTickSyncOperator) EXPECT() *MockTimeTickSyncOperator_Expecter
 func (_m *MockTimeTickSyncOperator) Channel() types.PChannelInfo {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Channel")
+	}
+
 	var r0 types.PChannelInfo
 	if rf, ok := ret.Get(0).(func() types.PChannelInfo); ok {
 		r0 = rf()
@@ -102,6 +106,10 @@ func (_c *MockTimeTickSyncOperator_Sync_Call) RunAndReturn(run func(context.Cont
 func (_m *MockTimeTickSyncOperator) TimeTickNotifier() *inspector.TimeTickNotifier {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for TimeTickNotifier")
+	}
+
 	var r0 *inspector.TimeTickNotifier
 	if rf, ok := ret.Get(0).(func() *inspector.TimeTickNotifier); ok {
 		r0 = rf()

internal/mocks/streamingnode/server/wal/mock_interceptors/mock_Interceptor.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_interceptors
 
@@ -59,6 +59,10 @@ func (_c *MockInterceptor_Close_Call) RunAndReturn(run func()) *MockInterceptor_
 func (_m *MockInterceptor) DoAppend(ctx context.Context, msg message.MutableMessage, append func(context.Context, message.MutableMessage) (message.MessageID, error)) (message.MessageID, error) {
 	ret := _m.Called(ctx, msg, append)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DoAppend")
+	}
+
 	var r0 message.MessageID
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, message.MutableMessage, func(context.Context, message.MutableMessage) (message.MessageID, error)) (message.MessageID, error)); ok {

internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorBuilder.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_interceptors
 
@@ -24,6 +24,10 @@ func (_m *MockInterceptorBuilder) EXPECT() *MockInterceptorBuilder_Expecter {
 func (_m *MockInterceptorBuilder) Build(param interceptors.InterceptorBuildParam) interceptors.Interceptor {
 	ret := _m.Called(param)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Build")
+	}
+
 	var r0 interceptors.Interceptor
 	if rf, ok := ret.Get(0).(func(interceptors.InterceptorBuildParam) interceptors.Interceptor); ok {
 		r0 = rf(param)

internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorWithReady.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_interceptors
 
@@ -59,6 +59,10 @@ func (_c *MockInterceptorWithReady_Close_Call) RunAndReturn(run func()) *MockInt
 func (_m *MockInterceptorWithReady) DoAppend(ctx context.Context, msg message.MutableMessage, append func(context.Context, message.MutableMessage) (message.MessageID, error)) (message.MessageID, error) {
 	ret := _m.Called(ctx, msg, append)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DoAppend")
+	}
+
 	var r0 message.MessageID
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, message.MutableMessage, func(context.Context, message.MutableMessage) (message.MessageID, error)) (message.MessageID, error)); ok {
@@ -115,6 +119,10 @@ func (_c *MockInterceptorWithReady_DoAppend_Call) RunAndReturn(run func(context.
 func (_m *MockInterceptorWithReady) Ready() <-chan struct{} {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Ready")
+	}
+
 	var r0 <-chan struct{}
 	if rf, ok := ret.Get(0).(func() <-chan struct{}); ok {
 		r0 = rf()

internal/mocks/util/streamingutil/service/mock_discoverer/mock_Discoverer.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_discoverer
 
@@ -26,6 +26,10 @@ func (_m *MockDiscoverer) EXPECT() *MockDiscoverer_Expecter {
 func (_m *MockDiscoverer) Discover(ctx context.Context, cb func(discoverer.VersionedState) error) error {
 	ret := _m.Called(ctx, cb)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Discover")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, func(discoverer.VersionedState) error) error); ok {
 		r0 = rf(ctx, cb)
@@ -69,6 +73,10 @@ func (_c *MockDiscoverer_Discover_Call) RunAndReturn(run func(context.Context, f
 func (_m *MockDiscoverer) NewVersionedState() discoverer.VersionedState {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for NewVersionedState")
+	}
+
 	var r0 discoverer.VersionedState
 	if rf, ok := ret.Get(0).(func() discoverer.VersionedState); ok {
 		r0 = rf()

internal/mocks/util/streamingutil/service/mock_lazygrpc/mock_Service.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_lazygrpc
 
@@ -59,6 +59,10 @@ func (_c *MockService_Close_Call[T]) RunAndReturn(run func()) *MockService_Close
 func (_m *MockService[T]) GetConn(ctx context.Context) (*grpc.ClientConn, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetConn")
+	}
+
 	var r0 *grpc.ClientConn
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (*grpc.ClientConn, error)); ok {
@@ -113,6 +117,10 @@ func (_c *MockService_GetConn_Call[T]) RunAndReturn(run func(context.Context) (*
 func (_m *MockService[T]) GetService(ctx context.Context) (T, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetService")
+	}
+
 	var r0 T
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (T, error)); ok {

internal/mocks/util/streamingutil/service/mock_resolver/mock_Builder.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_resolver
 
@@ -26,6 +26,10 @@ func (_m *MockBuilder) EXPECT() *MockBuilder_Expecter {
 func (_m *MockBuilder) Build(target resolver.Target, cc resolver.ClientConn, opts resolver.BuildOptions) (resolver.Resolver, error) {
 	ret := _m.Called(target, cc, opts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Build")
+	}
+
 	var r0 resolver.Resolver
 	var r1 error
 	if rf, ok := ret.Get(0).(func(resolver.Target, resolver.ClientConn, resolver.BuildOptions) (resolver.Resolver, error)); ok {
@@ -114,6 +118,10 @@ func (_c *MockBuilder_Close_Call) RunAndReturn(run func()) *MockBuilder_Close_Ca
 func (_m *MockBuilder) Resolver() serviceresolver.Resolver {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Resolver")
+	}
+
 	var r0 serviceresolver.Resolver
 	if rf, ok := ret.Get(0).(func() serviceresolver.Resolver); ok {
 		r0 = rf()
@@ -157,6 +165,10 @@ func (_c *MockBuilder_Resolver_Call) RunAndReturn(run func() serviceresolver.Res
 func (_m *MockBuilder) Scheme() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Scheme")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()

internal/mocks/util/streamingutil/service/mock_resolver/mock_Resolver.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_resolver
 
@@ -26,6 +26,10 @@ func (_m *MockResolver) EXPECT() *MockResolver_Expecter {
 func (_m *MockResolver) GetLatestState() discoverer.VersionedState {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetLatestState")
+	}
+
 	var r0 discoverer.VersionedState
 	if rf, ok := ret.Get(0).(func() discoverer.VersionedState); ok {
 		r0 = rf()
@@ -67,6 +71,10 @@ func (_c *MockResolver_GetLatestState_Call) RunAndReturn(run func() discoverer.V
 func (_m *MockResolver) Watch(ctx context.Context, cb func(discoverer.VersionedState) error) error {
 	ret := _m.Called(ctx, cb)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Watch")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, func(discoverer.VersionedState) error) error); ok {
 		r0 = rf(ctx, cb)

internal/proxy/mock_cache.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package proxy
 
@@ -28,6 +28,10 @@ func (_m *MockCache) EXPECT() *MockCache_Expecter {
 func (_m *MockCache) AllocID(ctx context.Context) (int64, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocID")
+	}
+
 	var r0 int64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (int64, error)); ok {
@@ -114,6 +118,10 @@ func (_c *MockCache_DeprecateShardCache_Call) RunAndReturn(run func(string, stri
 func (_m *MockCache) GetCollectionID(ctx context.Context, database string, collectionName string) (int64, error) {
 	ret := _m.Called(ctx, database, collectionName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollectionID")
+	}
+
 	var r0 int64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) (int64, error)); ok {
@@ -168,6 +176,10 @@ func (_c *MockCache_GetCollectionID_Call) RunAndReturn(run func(context.Context,
 func (_m *MockCache) GetCollectionInfo(ctx context.Context, database string, collectionName string, collectionID int64) (*collectionBasicInfo, error) {
 	ret := _m.Called(ctx, database, collectionName, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollectionInfo")
+	}
+
 	var r0 *collectionBasicInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, int64) (*collectionBasicInfo, error)); ok {
@@ -225,6 +237,10 @@ func (_c *MockCache_GetCollectionInfo_Call) RunAndReturn(run func(context.Contex
 func (_m *MockCache) GetCollectionName(ctx context.Context, database string, collectionID int64) (string, error) {
 	ret := _m.Called(ctx, database, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollectionName")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, int64) (string, error)); ok {
@@ -279,6 +295,10 @@ func (_c *MockCache_GetCollectionName_Call) RunAndReturn(run func(context.Contex
 func (_m *MockCache) GetCollectionSchema(ctx context.Context, database string, collectionName string) (*schemaInfo, error) {
 	ret := _m.Called(ctx, database, collectionName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollectionSchema")
+	}
+
 	var r0 *schemaInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) (*schemaInfo, error)); ok {
@@ -335,6 +355,10 @@ func (_c *MockCache_GetCollectionSchema_Call) RunAndReturn(run func(context.Cont
 func (_m *MockCache) GetCredentialInfo(ctx context.Context, username string) (*internalpb.CredentialInfo, error) {
 	ret := _m.Called(ctx, username)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCredentialInfo")
+	}
+
 	var r0 *internalpb.CredentialInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*internalpb.CredentialInfo, error)); ok {
@@ -390,6 +414,10 @@ func (_c *MockCache_GetCredentialInfo_Call) RunAndReturn(run func(context.Contex
 func (_m *MockCache) GetDatabaseInfo(ctx context.Context, database string) (*databaseInfo, error) {
 	ret := _m.Called(ctx, database)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDatabaseInfo")
+	}
+
 	var r0 *databaseInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*databaseInfo, error)); ok {
@@ -445,6 +473,10 @@ func (_c *MockCache_GetDatabaseInfo_Call) RunAndReturn(run func(context.Context,
 func (_m *MockCache) GetPartitionID(ctx context.Context, database string, collectionName string, partitionName string) (int64, error) {
 	ret := _m.Called(ctx, database, collectionName, partitionName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPartitionID")
+	}
+
 	var r0 int64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string) (int64, error)); ok {
@@ -500,6 +532,10 @@ func (_c *MockCache_GetPartitionID_Call) RunAndReturn(run func(context.Context,
 func (_m *MockCache) GetPartitionInfo(ctx context.Context, database string, collectionName string, partitionName string) (*partitionInfo, error) {
 	ret := _m.Called(ctx, database, collectionName, partitionName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPartitionInfo")
+	}
+
 	var r0 *partitionInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string) (*partitionInfo, error)); ok {
@@ -557,6 +593,10 @@ func (_c *MockCache_GetPartitionInfo_Call) RunAndReturn(run func(context.Context
 func (_m *MockCache) GetPartitions(ctx context.Context, database string, collectionName string) (map[string]int64, error) {
 	ret := _m.Called(ctx, database, collectionName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPartitions")
+	}
+
 	var r0 map[string]int64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) (map[string]int64, error)); ok {
@@ -613,6 +653,10 @@ func (_c *MockCache_GetPartitions_Call) RunAndReturn(run func(context.Context, s
 func (_m *MockCache) GetPartitionsIndex(ctx context.Context, database string, collectionName string) ([]string, error) {
 	ret := _m.Called(ctx, database, collectionName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPartitionsIndex")
+	}
+
 	var r0 []string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) ([]string, error)); ok {
@@ -669,6 +713,10 @@ func (_c *MockCache_GetPartitionsIndex_Call) RunAndReturn(run func(context.Conte
 func (_m *MockCache) GetPrivilegeInfo(ctx context.Context) []string {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPrivilegeInfo")
+	}
+
 	var r0 []string
 	if rf, ok := ret.Get(0).(func(context.Context) []string); ok {
 		r0 = rf(ctx)
@@ -713,6 +761,10 @@ func (_c *MockCache_GetPrivilegeInfo_Call) RunAndReturn(run func(context.Context
 func (_m *MockCache) GetShards(ctx context.Context, withCache bool, database string, collectionName string, collectionID int64) (map[string][]nodeInfo, error) {
 	ret := _m.Called(ctx, withCache, database, collectionName, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetShards")
+	}
+
 	var r0 map[string][]nodeInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, bool, string, string, int64) (map[string][]nodeInfo, error)); ok {
@@ -771,6 +823,10 @@ func (_c *MockCache_GetShards_Call) RunAndReturn(run func(context.Context, bool,
 func (_m *MockCache) GetUserRole(username string) []string {
 	ret := _m.Called(username)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetUserRole")
+	}
+
 	var r0 []string
 	if rf, ok := ret.Get(0).(func(string) []string); ok {
 		r0 = rf(username)
@@ -815,6 +871,10 @@ func (_c *MockCache_GetUserRole_Call) RunAndReturn(run func(string) []string) *M
 func (_m *MockCache) HasDatabase(ctx context.Context, database string) bool {
 	ret := _m.Called(ctx, database)
 
+	if len(ret) == 0 {
+		panic("no return value specified for HasDatabase")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(context.Context, string) bool); ok {
 		r0 = rf(ctx, database)
@@ -925,6 +985,10 @@ func (_c *MockCache_InvalidateShardLeaderCache_Call) RunAndReturn(run func([]int
 func (_m *MockCache) RefreshPolicyInfo(op typeutil.CacheOp) error {
 	ret := _m.Called(op)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RefreshPolicyInfo")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(typeutil.CacheOp) error); ok {
 		r0 = rf(op)
@@ -1002,6 +1066,10 @@ func (_c *MockCache_RemoveCollection_Call) RunAndReturn(run func(context.Context
 func (_m *MockCache) RemoveCollectionsByID(ctx context.Context, collectionID int64) []string {
 	ret := _m.Called(ctx, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemoveCollectionsByID")
+	}
+
 	var r0 []string
 	if rf, ok := ret.Get(0).(func(context.Context, int64) []string); ok {
 		r0 = rf(ctx, collectionID)

internal/proxy/mock_channels_manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package proxy
 
@@ -24,6 +24,10 @@ func (_m *MockChannelsMgr) EXPECT() *MockChannelsMgr_Expecter {
 func (_m *MockChannelsMgr) getChannels(collectionID int64) ([]string, error) {
 	ret := _m.Called(collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for getChannels")
+	}
+
 	var r0 []string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(int64) ([]string, error)); ok {
@@ -78,6 +82,10 @@ func (_c *MockChannelsMgr_getChannels_Call) RunAndReturn(run func(int64) ([]stri
 func (_m *MockChannelsMgr) getOrCreateDmlStream(collectionID int64) (msgstream.MsgStream, error) {
 	ret := _m.Called(collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for getOrCreateDmlStream")
+	}
+
 	var r0 msgstream.MsgStream
 	var r1 error
 	if rf, ok := ret.Get(0).(func(int64) (msgstream.MsgStream, error)); ok {
@@ -132,6 +140,10 @@ func (_c *MockChannelsMgr_getOrCreateDmlStream_Call) RunAndReturn(run func(int64
 func (_m *MockChannelsMgr) getVChannels(collectionID int64) ([]string, error) {
 	ret := _m.Called(collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for getVChannels")
+	}
+
 	var r0 []string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(int64) ([]string, error)); ok {

internal/proxy/mock_lb_balancer.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package proxy
 
@@ -92,6 +92,10 @@ func (_c *MockLBBalancer_Close_Call) RunAndReturn(run func()) *MockLBBalancer_Cl
 func (_m *MockLBBalancer) SelectNode(ctx context.Context, availableNodes []int64, nq int64) (int64, error) {
 	ret := _m.Called(ctx, availableNodes, nq)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SelectNode")
+	}
+
 	var r0 int64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, []int64, int64) (int64, error)); ok {

internal/proxy/mock_lb_policy.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package proxy
 
@@ -58,6 +58,10 @@ func (_c *MockLBPolicy_Close_Call) RunAndReturn(run func()) *MockLBPolicy_Close_
 func (_m *MockLBPolicy) Execute(ctx context.Context, workload CollectionWorkLoad) error {
 	ret := _m.Called(ctx, workload)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Execute")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, CollectionWorkLoad) error); ok {
 		r0 = rf(ctx, workload)
@@ -101,6 +105,10 @@ func (_c *MockLBPolicy_Execute_Call) RunAndReturn(run func(context.Context, Coll
 func (_m *MockLBPolicy) ExecuteWithRetry(ctx context.Context, workload ChannelWorkload) error {
 	ret := _m.Called(ctx, workload)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ExecuteWithRetry")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, ChannelWorkload) error); ok {
 		r0 = rf(ctx, workload)

internal/proxy/mock_shardclient_manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package proxy
 
@@ -58,6 +58,10 @@ func (_c *MockShardClientManager_Close_Call) RunAndReturn(run func()) *MockShard
 func (_m *MockShardClientManager) GetClient(ctx context.Context, nodeID int64) (types.QueryNodeClient, error) {
 	ret := _m.Called(ctx, nodeID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetClient")
+	}
+
 	var r0 types.QueryNodeClient
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) (types.QueryNodeClient, error)); ok {
@@ -146,6 +150,10 @@ func (_c *MockShardClientManager_SetClientCreatorFunc_Call) RunAndReturn(run fun
 func (_m *MockShardClientManager) UpdateShardLeaders(oldLeaders map[string][]nodeInfo, newLeaders map[string][]nodeInfo) error {
 	ret := _m.Called(oldLeaders, newLeaders)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateShardLeaders")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string][]nodeInfo, map[string][]nodeInfo) error); ok {
 		r0 = rf(oldLeaders, newLeaders)

internal/proxy/mock_tso_test.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package proxy
 
@@ -36,6 +36,10 @@ func (_m *mockTimestampAllocator) AllocTimestamp(ctx context.Context, req *rootc
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AllocTimestamp")
+	}
+
 	var r0 *rootcoordpb.AllocTimestampResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AllocTimestampRequest, ...grpc.CallOption) (*rootcoordpb.AllocTimestampResponse, error)); ok {

internal/querycoordv2/balance/mock_balancer.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package balance
 
@@ -24,6 +24,10 @@ func (_m *MockBalancer) EXPECT() *MockBalancer_Expecter {
 func (_m *MockBalancer) AssignChannel(channels []*meta.DmChannel, nodes []int64, manualBalance bool) []ChannelAssignPlan {
 	ret := _m.Called(channels, nodes, manualBalance)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AssignChannel")
+	}
+
 	var r0 []ChannelAssignPlan
 	if rf, ok := ret.Get(0).(func([]*meta.DmChannel, []int64, bool) []ChannelAssignPlan); ok {
 		r0 = rf(channels, nodes, manualBalance)
@@ -70,6 +74,10 @@ func (_c *MockBalancer_AssignChannel_Call) RunAndReturn(run func([]*meta.DmChann
 func (_m *MockBalancer) AssignSegment(collectionID int64, segments []*meta.Segment, nodes []int64, manualBalance bool) []SegmentAssignPlan {
 	ret := _m.Called(collectionID, segments, nodes, manualBalance)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AssignSegment")
+	}
+
 	var r0 []SegmentAssignPlan
 	if rf, ok := ret.Get(0).(func(int64, []*meta.Segment, []int64, bool) []SegmentAssignPlan); ok {
 		r0 = rf(collectionID, segments, nodes, manualBalance)
@@ -117,6 +125,10 @@ func (_c *MockBalancer_AssignSegment_Call) RunAndReturn(run func(int64, []*meta.
 func (_m *MockBalancer) BalanceReplica(replica *meta.Replica) ([]SegmentAssignPlan, []ChannelAssignPlan) {
 	ret := _m.Called(replica)
 
+	if len(ret) == 0 {
+		panic("no return value specified for BalanceReplica")
+	}
+
 	var r0 []SegmentAssignPlan
 	var r1 []ChannelAssignPlan
 	if rf, ok := ret.Get(0).(func(*meta.Replica) ([]SegmentAssignPlan, []ChannelAssignPlan)); ok {

internal/querycoordv2/dist/mock_controller.go

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

internal/querycoordv2/meta/mock_broker.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package meta
 
@@ -34,6 +34,10 @@ func (_m *MockBroker) EXPECT() *MockBroker_Expecter {
 func (_m *MockBroker) DescribeCollection(ctx context.Context, collectionID int64) (*milvuspb.DescribeCollectionResponse, error) {
 	ret := _m.Called(ctx, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeCollection")
+	}
+
 	var r0 *milvuspb.DescribeCollectionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) (*milvuspb.DescribeCollectionResponse, error)); ok {
@@ -89,6 +93,10 @@ func (_c *MockBroker_DescribeCollection_Call) RunAndReturn(run func(context.Cont
 func (_m *MockBroker) DescribeDatabase(ctx context.Context, dbName string) (*rootcoordpb.DescribeDatabaseResponse, error) {
 	ret := _m.Called(ctx, dbName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeDatabase")
+	}
+
 	var r0 *rootcoordpb.DescribeDatabaseResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*rootcoordpb.DescribeDatabaseResponse, error)); ok {
@@ -144,6 +152,10 @@ func (_c *MockBroker_DescribeDatabase_Call) RunAndReturn(run func(context.Contex
 func (_m *MockBroker) GetCollectionLoadInfo(ctx context.Context, collectionID int64) ([]string, int64, error) {
 	ret := _m.Called(ctx, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollectionLoadInfo")
+	}
+
 	var r0 []string
 	var r1 int64
 	var r2 error
@@ -206,6 +218,10 @@ func (_c *MockBroker_GetCollectionLoadInfo_Call) RunAndReturn(run func(context.C
 func (_m *MockBroker) GetIndexInfo(ctx context.Context, collectionID int64, segmentID int64) ([]*querypb.FieldIndexInfo, error) {
 	ret := _m.Called(ctx, collectionID, segmentID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetIndexInfo")
+	}
+
 	var r0 []*querypb.FieldIndexInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, int64) ([]*querypb.FieldIndexInfo, error)); ok {
@@ -262,6 +278,10 @@ func (_c *MockBroker_GetIndexInfo_Call) RunAndReturn(run func(context.Context, i
 func (_m *MockBroker) GetPartitions(ctx context.Context, collectionID int64) ([]int64, error) {
 	ret := _m.Called(ctx, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPartitions")
+	}
+
 	var r0 []int64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) ([]int64, error)); ok {
@@ -317,6 +337,10 @@ func (_c *MockBroker_GetPartitions_Call) RunAndReturn(run func(context.Context,
 func (_m *MockBroker) GetRecoveryInfo(ctx context.Context, collectionID int64, partitionID int64) ([]*datapb.VchannelInfo, []*datapb.SegmentBinlogs, error) {
 	ret := _m.Called(ctx, collectionID, partitionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetRecoveryInfo")
+	}
+
 	var r0 []*datapb.VchannelInfo
 	var r1 []*datapb.SegmentBinlogs
 	var r2 error
@@ -389,6 +413,10 @@ func (_m *MockBroker) GetRecoveryInfoV2(ctx context.Context, collectionID int64,
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetRecoveryInfoV2")
+	}
+
 	var r0 []*datapb.VchannelInfo
 	var r1 []*datapb.SegmentInfo
 	var r2 error
@@ -468,6 +496,10 @@ func (_m *MockBroker) GetSegmentInfo(ctx context.Context, segmentID ...int64) ([
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentInfo")
+	}
+
 	var r0 []*datapb.SegmentInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, ...int64) ([]*datapb.SegmentInfo, error)); ok {
@@ -530,6 +562,10 @@ func (_c *MockBroker_GetSegmentInfo_Call) RunAndReturn(run func(context.Context,
 func (_m *MockBroker) ListIndexes(ctx context.Context, collectionID int64) ([]*indexpb.IndexInfo, error) {
 	ret := _m.Called(ctx, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListIndexes")
+	}
+
 	var r0 []*indexpb.IndexInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) ([]*indexpb.IndexInfo, error)); ok {

internal/querycoordv2/meta/mock_target_manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package meta
 
@@ -28,6 +28,10 @@ func (_m *MockTargetManager) EXPECT() *MockTargetManager_Expecter {
 func (_m *MockTargetManager) CanSegmentBeMoved(collectionID int64, segmentID int64) bool {
 	ret := _m.Called(collectionID, segmentID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CanSegmentBeMoved")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(int64, int64) bool); ok {
 		r0 = rf(collectionID, segmentID)
@@ -71,6 +75,10 @@ func (_c *MockTargetManager_CanSegmentBeMoved_Call) RunAndReturn(run func(int64,
 func (_m *MockTargetManager) GetCollectionTargetVersion(collectionID int64, scope int32) int64 {
 	ret := _m.Called(collectionID, scope)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollectionTargetVersion")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func(int64, int32) int64); ok {
 		r0 = rf(collectionID, scope)
@@ -114,6 +122,10 @@ func (_c *MockTargetManager_GetCollectionTargetVersion_Call) RunAndReturn(run fu
 func (_m *MockTargetManager) GetDmChannel(collectionID int64, channel string, scope int32) *DmChannel {
 	ret := _m.Called(collectionID, channel, scope)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDmChannel")
+	}
+
 	var r0 *DmChannel
 	if rf, ok := ret.Get(0).(func(int64, string, int32) *DmChannel); ok {
 		r0 = rf(collectionID, channel, scope)
@@ -160,6 +172,10 @@ func (_c *MockTargetManager_GetDmChannel_Call) RunAndReturn(run func(int64, stri
 func (_m *MockTargetManager) GetDmChannelsByCollection(collectionID int64, scope int32) map[string]*DmChannel {
 	ret := _m.Called(collectionID, scope)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDmChannelsByCollection")
+	}
+
 	var r0 map[string]*DmChannel
 	if rf, ok := ret.Get(0).(func(int64, int32) map[string]*DmChannel); ok {
 		r0 = rf(collectionID, scope)
@@ -205,6 +221,10 @@ func (_c *MockTargetManager_GetDmChannelsByCollection_Call) RunAndReturn(run fun
 func (_m *MockTargetManager) GetDroppedSegmentsByChannel(collectionID int64, channelName string, scope int32) []int64 {
 	ret := _m.Called(collectionID, channelName, scope)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDroppedSegmentsByChannel")
+	}
+
 	var r0 []int64
 	if rf, ok := ret.Get(0).(func(int64, string, int32) []int64); ok {
 		r0 = rf(collectionID, channelName, scope)
@@ -251,6 +271,10 @@ func (_c *MockTargetManager_GetDroppedSegmentsByChannel_Call) RunAndReturn(run f
 func (_m *MockTargetManager) GetGrowingSegmentsByChannel(collectionID int64, channelName string, scope int32) typeutil.Set[int64] {
 	ret := _m.Called(collectionID, channelName, scope)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetGrowingSegmentsByChannel")
+	}
+
 	var r0 typeutil.Set[int64]
 	if rf, ok := ret.Get(0).(func(int64, string, int32) typeutil.Set[int64]); ok {
 		r0 = rf(collectionID, channelName, scope)
@@ -297,6 +321,10 @@ func (_c *MockTargetManager_GetGrowingSegmentsByChannel_Call) RunAndReturn(run f
 func (_m *MockTargetManager) GetGrowingSegmentsByCollection(collectionID int64, scope int32) typeutil.Set[int64] {
 	ret := _m.Called(collectionID, scope)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetGrowingSegmentsByCollection")
+	}
+
 	var r0 typeutil.Set[int64]
 	if rf, ok := ret.Get(0).(func(int64, int32) typeutil.Set[int64]); ok {
 		r0 = rf(collectionID, scope)
@@ -342,6 +370,10 @@ func (_c *MockTargetManager_GetGrowingSegmentsByCollection_Call) RunAndReturn(ru
 func (_m *MockTargetManager) GetSealedSegment(collectionID int64, id int64, scope int32) *datapb.SegmentInfo {
 	ret := _m.Called(collectionID, id, scope)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSealedSegment")
+	}
+
 	var r0 *datapb.SegmentInfo
 	if rf, ok := ret.Get(0).(func(int64, int64, int32) *datapb.SegmentInfo); ok {
 		r0 = rf(collectionID, id, scope)
@@ -388,6 +420,10 @@ func (_c *MockTargetManager_GetSealedSegment_Call) RunAndReturn(run func(int64,
 func (_m *MockTargetManager) GetSealedSegmentsByChannel(collectionID int64, channelName string, scope int32) map[int64]*datapb.SegmentInfo {
 	ret := _m.Called(collectionID, channelName, scope)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSealedSegmentsByChannel")
+	}
+
 	var r0 map[int64]*datapb.SegmentInfo
 	if rf, ok := ret.Get(0).(func(int64, string, int32) map[int64]*datapb.SegmentInfo); ok {
 		r0 = rf(collectionID, channelName, scope)
@@ -434,6 +470,10 @@ func (_c *MockTargetManager_GetSealedSegmentsByChannel_Call) RunAndReturn(run fu
 func (_m *MockTargetManager) GetSealedSegmentsByCollection(collectionID int64, scope int32) map[int64]*datapb.SegmentInfo {
 	ret := _m.Called(collectionID, scope)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSealedSegmentsByCollection")
+	}
+
 	var r0 map[int64]*datapb.SegmentInfo
 	if rf, ok := ret.Get(0).(func(int64, int32) map[int64]*datapb.SegmentInfo); ok {
 		r0 = rf(collectionID, scope)
@@ -479,6 +519,10 @@ func (_c *MockTargetManager_GetSealedSegmentsByCollection_Call) RunAndReturn(run
 func (_m *MockTargetManager) GetSealedSegmentsByPartition(collectionID int64, partitionID int64, scope int32) map[int64]*datapb.SegmentInfo {
 	ret := _m.Called(collectionID, partitionID, scope)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSealedSegmentsByPartition")
+	}
+
 	var r0 map[int64]*datapb.SegmentInfo
 	if rf, ok := ret.Get(0).(func(int64, int64, int32) map[int64]*datapb.SegmentInfo); ok {
 		r0 = rf(collectionID, partitionID, scope)
@@ -525,6 +569,10 @@ func (_c *MockTargetManager_GetSealedSegmentsByPartition_Call) RunAndReturn(run
 func (_m *MockTargetManager) IsCurrentTargetExist(collectionID int64, partitionID int64) bool {
 	ret := _m.Called(collectionID, partitionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsCurrentTargetExist")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(int64, int64) bool); ok {
 		r0 = rf(collectionID, partitionID)
@@ -568,6 +616,10 @@ func (_c *MockTargetManager_IsCurrentTargetExist_Call) RunAndReturn(run func(int
 func (_m *MockTargetManager) IsNextTargetExist(collectionID int64) bool {
 	ret := _m.Called(collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsNextTargetExist")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(int64) bool); ok {
 		r0 = rf(collectionID)
@@ -617,6 +669,10 @@ func (_m *MockTargetManager) PullNextTargetV1(broker Broker, collectionID int64,
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for PullNextTargetV1")
+	}
+
 	var r0 map[int64]*datapb.SegmentInfo
 	var r1 map[string]*DmChannel
 	var r2 error
@@ -696,6 +752,10 @@ func (_m *MockTargetManager) PullNextTargetV2(broker Broker, collectionID int64,
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for PullNextTargetV2")
+	}
+
 	var r0 map[int64]*datapb.SegmentInfo
 	var r1 map[string]*DmChannel
 	var r2 error
@@ -768,6 +828,10 @@ func (_c *MockTargetManager_PullNextTargetV2_Call) RunAndReturn(run func(Broker,
 func (_m *MockTargetManager) Recover(catalog metastore.QueryCoordCatalog) error {
 	ret := _m.Called(catalog)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Recover")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(metastore.QueryCoordCatalog) error); ok {
 		r0 = rf(catalog)
@@ -924,6 +988,10 @@ func (_c *MockTargetManager_SaveCurrentTarget_Call) RunAndReturn(run func(metast
 func (_m *MockTargetManager) UpdateCollectionCurrentTarget(collectionID int64) bool {
 	ret := _m.Called(collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateCollectionCurrentTarget")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(int64) bool); ok {
 		r0 = rf(collectionID)
@@ -966,6 +1034,10 @@ func (_c *MockTargetManager_UpdateCollectionCurrentTarget_Call) RunAndReturn(run
 func (_m *MockTargetManager) UpdateCollectionNextTarget(collectionID int64) error {
 	ret := _m.Called(collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateCollectionNextTarget")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(int64) error); ok {
 		r0 = rf(collectionID)

internal/querycoordv2/mocks/mock_querynode.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mocks
 
@@ -33,6 +33,10 @@ func (_m *MockQueryNodeServer) EXPECT() *MockQueryNodeServer_Expecter {
 func (_m *MockQueryNodeServer) Delete(_a0 context.Context, _a1 *querypb.DeleteRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Delete")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.DeleteRequest) (*commonpb.Status, error)); ok {
@@ -88,6 +92,10 @@ func (_c *MockQueryNodeServer_Delete_Call) RunAndReturn(run func(context.Context
 func (_m *MockQueryNodeServer) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok {
@@ -143,6 +151,10 @@ func (_c *MockQueryNodeServer_GetComponentStates_Call) RunAndReturn(run func(con
 func (_m *MockQueryNodeServer) GetDataDistribution(_a0 context.Context, _a1 *querypb.GetDataDistributionRequest) (*querypb.GetDataDistributionResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDataDistribution")
+	}
+
 	var r0 *querypb.GetDataDistributionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetDataDistributionRequest) (*querypb.GetDataDistributionResponse, error)); ok {
@@ -198,6 +210,10 @@ func (_c *MockQueryNodeServer_GetDataDistribution_Call) RunAndReturn(run func(co
 func (_m *MockQueryNodeServer) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -253,6 +269,10 @@ func (_c *MockQueryNodeServer_GetMetrics_Call) RunAndReturn(run func(context.Con
 func (_m *MockQueryNodeServer) GetSegmentInfo(_a0 context.Context, _a1 *querypb.GetSegmentInfoRequest) (*querypb.GetSegmentInfoResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentInfo")
+	}
+
 	var r0 *querypb.GetSegmentInfoResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetSegmentInfoRequest) (*querypb.GetSegmentInfoResponse, error)); ok {
@@ -308,6 +328,10 @@ func (_c *MockQueryNodeServer_GetSegmentInfo_Call) RunAndReturn(run func(context
 func (_m *MockQueryNodeServer) GetStatistics(_a0 context.Context, _a1 *querypb.GetStatisticsRequest) (*internalpb.GetStatisticsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatistics")
+	}
+
 	var r0 *internalpb.GetStatisticsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetStatisticsRequest) (*internalpb.GetStatisticsResponse, error)); ok {
@@ -363,6 +387,10 @@ func (_c *MockQueryNodeServer_GetStatistics_Call) RunAndReturn(run func(context.
 func (_m *MockQueryNodeServer) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatisticsChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -418,6 +446,10 @@ func (_c *MockQueryNodeServer_GetStatisticsChannel_Call) RunAndReturn(run func(c
 func (_m *MockQueryNodeServer) GetTimeTickChannel(_a0 context.Context, _a1 *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetTimeTickChannel")
+	}
+
 	var r0 *milvuspb.StringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error)); ok {
@@ -473,6 +505,10 @@ func (_c *MockQueryNodeServer_GetTimeTickChannel_Call) RunAndReturn(run func(con
 func (_m *MockQueryNodeServer) LoadPartitions(_a0 context.Context, _a1 *querypb.LoadPartitionsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadPartitions")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadPartitionsRequest) (*commonpb.Status, error)); ok {
@@ -528,6 +564,10 @@ func (_c *MockQueryNodeServer_LoadPartitions_Call) RunAndReturn(run func(context
 func (_m *MockQueryNodeServer) LoadSegments(_a0 context.Context, _a1 *querypb.LoadSegmentsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadSegmentsRequest) (*commonpb.Status, error)); ok {
@@ -583,6 +623,10 @@ func (_c *MockQueryNodeServer_LoadSegments_Call) RunAndReturn(run func(context.C
 func (_m *MockQueryNodeServer) Query(_a0 context.Context, _a1 *querypb.QueryRequest) (*internalpb.RetrieveResults, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Query")
+	}
+
 	var r0 *internalpb.RetrieveResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest) (*internalpb.RetrieveResults, error)); ok {
@@ -638,6 +682,10 @@ func (_c *MockQueryNodeServer_Query_Call) RunAndReturn(run func(context.Context,
 func (_m *MockQueryNodeServer) QuerySegments(_a0 context.Context, _a1 *querypb.QueryRequest) (*internalpb.RetrieveResults, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QuerySegments")
+	}
+
 	var r0 *internalpb.RetrieveResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest) (*internalpb.RetrieveResults, error)); ok {
@@ -693,6 +741,10 @@ func (_c *MockQueryNodeServer_QuerySegments_Call) RunAndReturn(run func(context.
 func (_m *MockQueryNodeServer) QueryStream(_a0 *querypb.QueryRequest, _a1 querypb.QueryNode_QueryStreamServer) error {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryStream")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*querypb.QueryRequest, querypb.QueryNode_QueryStreamServer) error); ok {
 		r0 = rf(_a0, _a1)
@@ -736,6 +788,10 @@ func (_c *MockQueryNodeServer_QueryStream_Call) RunAndReturn(run func(*querypb.Q
 func (_m *MockQueryNodeServer) QueryStreamSegments(_a0 *querypb.QueryRequest, _a1 querypb.QueryNode_QueryStreamSegmentsServer) error {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryStreamSegments")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*querypb.QueryRequest, querypb.QueryNode_QueryStreamSegmentsServer) error); ok {
 		r0 = rf(_a0, _a1)
@@ -779,6 +835,10 @@ func (_c *MockQueryNodeServer_QueryStreamSegments_Call) RunAndReturn(run func(*q
 func (_m *MockQueryNodeServer) ReleaseCollection(_a0 context.Context, _a1 *querypb.ReleaseCollectionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleaseCollection")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseCollectionRequest) (*commonpb.Status, error)); ok {
@@ -834,6 +894,10 @@ func (_c *MockQueryNodeServer_ReleaseCollection_Call) RunAndReturn(run func(cont
 func (_m *MockQueryNodeServer) ReleasePartitions(_a0 context.Context, _a1 *querypb.ReleasePartitionsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleasePartitions")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleasePartitionsRequest) (*commonpb.Status, error)); ok {
@@ -889,6 +953,10 @@ func (_c *MockQueryNodeServer_ReleasePartitions_Call) RunAndReturn(run func(cont
 func (_m *MockQueryNodeServer) ReleaseSegments(_a0 context.Context, _a1 *querypb.ReleaseSegmentsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleaseSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseSegmentsRequest) (*commonpb.Status, error)); ok {
@@ -944,6 +1012,10 @@ func (_c *MockQueryNodeServer_ReleaseSegments_Call) RunAndReturn(run func(contex
 func (_m *MockQueryNodeServer) Search(_a0 context.Context, _a1 *querypb.SearchRequest) (*internalpb.SearchResults, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Search")
+	}
+
 	var r0 *internalpb.SearchResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest) (*internalpb.SearchResults, error)); ok {
@@ -999,6 +1071,10 @@ func (_c *MockQueryNodeServer_Search_Call) RunAndReturn(run func(context.Context
 func (_m *MockQueryNodeServer) SearchSegments(_a0 context.Context, _a1 *querypb.SearchRequest) (*internalpb.SearchResults, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SearchSegments")
+	}
+
 	var r0 *internalpb.SearchResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest) (*internalpb.SearchResults, error)); ok {
@@ -1054,6 +1130,10 @@ func (_c *MockQueryNodeServer_SearchSegments_Call) RunAndReturn(run func(context
 func (_m *MockQueryNodeServer) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ShowConfigurations")
+	}
+
 	var r0 *internalpb.ShowConfigurationsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok {
@@ -1109,6 +1189,10 @@ func (_c *MockQueryNodeServer_ShowConfigurations_Call) RunAndReturn(run func(con
 func (_m *MockQueryNodeServer) SyncDistribution(_a0 context.Context, _a1 *querypb.SyncDistributionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SyncDistribution")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncDistributionRequest) (*commonpb.Status, error)); ok {
@@ -1164,6 +1248,10 @@ func (_c *MockQueryNodeServer_SyncDistribution_Call) RunAndReturn(run func(conte
 func (_m *MockQueryNodeServer) SyncReplicaSegments(_a0 context.Context, _a1 *querypb.SyncReplicaSegmentsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SyncReplicaSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncReplicaSegmentsRequest) (*commonpb.Status, error)); ok {
@@ -1219,6 +1307,10 @@ func (_c *MockQueryNodeServer_SyncReplicaSegments_Call) RunAndReturn(run func(co
 func (_m *MockQueryNodeServer) UnsubDmChannel(_a0 context.Context, _a1 *querypb.UnsubDmChannelRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UnsubDmChannel")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.UnsubDmChannelRequest) (*commonpb.Status, error)); ok {
@@ -1274,6 +1366,10 @@ func (_c *MockQueryNodeServer_UnsubDmChannel_Call) RunAndReturn(run func(context
 func (_m *MockQueryNodeServer) WatchDmChannels(_a0 context.Context, _a1 *querypb.WatchDmChannelsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WatchDmChannels")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.WatchDmChannelsRequest) (*commonpb.Status, error)); ok {

internal/querycoordv2/session/mock_cluster.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package session
 
@@ -31,6 +31,10 @@ func (_m *MockCluster) EXPECT() *MockCluster_Expecter {
 func (_m *MockCluster) GetComponentStates(ctx context.Context, nodeID int64) (*milvuspb.ComponentStates, error) {
 	ret := _m.Called(ctx, nodeID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetComponentStates")
+	}
+
 	var r0 *milvuspb.ComponentStates
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) (*milvuspb.ComponentStates, error)); ok {
@@ -86,6 +90,10 @@ func (_c *MockCluster_GetComponentStates_Call) RunAndReturn(run func(context.Con
 func (_m *MockCluster) GetDataDistribution(ctx context.Context, nodeID int64, req *querypb.GetDataDistributionRequest) (*querypb.GetDataDistributionResponse, error) {
 	ret := _m.Called(ctx, nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDataDistribution")
+	}
+
 	var r0 *querypb.GetDataDistributionResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.GetDataDistributionRequest) (*querypb.GetDataDistributionResponse, error)); ok {
@@ -142,6 +150,10 @@ func (_c *MockCluster_GetDataDistribution_Call) RunAndReturn(run func(context.Co
 func (_m *MockCluster) GetMetrics(ctx context.Context, nodeID int64, req *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) {
 	ret := _m.Called(ctx, nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetMetrics")
+	}
+
 	var r0 *milvuspb.GetMetricsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok {
@@ -198,6 +210,10 @@ func (_c *MockCluster_GetMetrics_Call) RunAndReturn(run func(context.Context, in
 func (_m *MockCluster) LoadPartitions(ctx context.Context, nodeID int64, req *querypb.LoadPartitionsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(ctx, nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadPartitions")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.LoadPartitionsRequest) (*commonpb.Status, error)); ok {
@@ -254,6 +270,10 @@ func (_c *MockCluster_LoadPartitions_Call) RunAndReturn(run func(context.Context
 func (_m *MockCluster) LoadSegments(ctx context.Context, nodeID int64, req *querypb.LoadSegmentsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(ctx, nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.LoadSegmentsRequest) (*commonpb.Status, error)); ok {
@@ -310,6 +330,10 @@ func (_c *MockCluster_LoadSegments_Call) RunAndReturn(run func(context.Context,
 func (_m *MockCluster) ReleasePartitions(ctx context.Context, nodeID int64, req *querypb.ReleasePartitionsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(ctx, nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleasePartitions")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.ReleasePartitionsRequest) (*commonpb.Status, error)); ok {
@@ -366,6 +390,10 @@ func (_c *MockCluster_ReleasePartitions_Call) RunAndReturn(run func(context.Cont
 func (_m *MockCluster) ReleaseSegments(ctx context.Context, nodeID int64, req *querypb.ReleaseSegmentsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(ctx, nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleaseSegments")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.ReleaseSegmentsRequest) (*commonpb.Status, error)); ok {
@@ -486,6 +514,10 @@ func (_c *MockCluster_Stop_Call) RunAndReturn(run func()) *MockCluster_Stop_Call
 func (_m *MockCluster) SyncDistribution(ctx context.Context, nodeID int64, req *querypb.SyncDistributionRequest) (*commonpb.Status, error) {
 	ret := _m.Called(ctx, nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SyncDistribution")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.SyncDistributionRequest) (*commonpb.Status, error)); ok {
@@ -542,6 +574,10 @@ func (_c *MockCluster_SyncDistribution_Call) RunAndReturn(run func(context.Conte
 func (_m *MockCluster) UnsubDmChannel(ctx context.Context, nodeID int64, req *querypb.UnsubDmChannelRequest) (*commonpb.Status, error) {
 	ret := _m.Called(ctx, nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UnsubDmChannel")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.UnsubDmChannelRequest) (*commonpb.Status, error)); ok {
@@ -598,6 +634,10 @@ func (_c *MockCluster_UnsubDmChannel_Call) RunAndReturn(run func(context.Context
 func (_m *MockCluster) WatchDmChannels(ctx context.Context, nodeID int64, req *querypb.WatchDmChannelsRequest) (*commonpb.Status, error) {
 	ret := _m.Called(ctx, nodeID, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WatchDmChannels")
+	}
+
 	var r0 *commonpb.Status
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.WatchDmChannelsRequest) (*commonpb.Status, error)); ok {

internal/querycoordv2/task/mock_scheduler.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package task
 
@@ -21,6 +21,10 @@ func (_m *MockScheduler) EXPECT() *MockScheduler_Expecter {
 func (_m *MockScheduler) Add(task Task) error {
 	ret := _m.Called(task)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Add")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(Task) error); ok {
 		r0 = rf(task)
@@ -129,6 +133,10 @@ func (_c *MockScheduler_Dispatch_Call) RunAndReturn(run func(int64)) *MockSchedu
 func (_m *MockScheduler) GetChannelTaskDelta(nodeID int64, collectionID int64) int {
 	ret := _m.Called(nodeID, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetChannelTaskDelta")
+	}
+
 	var r0 int
 	if rf, ok := ret.Get(0).(func(int64, int64) int); ok {
 		r0 = rf(nodeID, collectionID)
@@ -172,6 +180,10 @@ func (_c *MockScheduler_GetChannelTaskDelta_Call) RunAndReturn(run func(int64, i
 func (_m *MockScheduler) GetChannelTaskNum() int {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetChannelTaskNum")
+	}
+
 	var r0 int
 	if rf, ok := ret.Get(0).(func() int); ok {
 		r0 = rf()
@@ -213,6 +225,10 @@ func (_c *MockScheduler_GetChannelTaskNum_Call) RunAndReturn(run func() int) *Mo
 func (_m *MockScheduler) GetExecutedFlag(nodeID int64) <-chan struct{} {
 	ret := _m.Called(nodeID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetExecutedFlag")
+	}
+
 	var r0 <-chan struct{}
 	if rf, ok := ret.Get(0).(func(int64) <-chan struct{}); ok {
 		r0 = rf(nodeID)
@@ -257,6 +273,10 @@ func (_c *MockScheduler_GetExecutedFlag_Call) RunAndReturn(run func(int64) <-cha
 func (_m *MockScheduler) GetSegmentTaskDelta(nodeID int64, collectionID int64) int {
 	ret := _m.Called(nodeID, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentTaskDelta")
+	}
+
 	var r0 int
 	if rf, ok := ret.Get(0).(func(int64, int64) int); ok {
 		r0 = rf(nodeID, collectionID)
@@ -300,6 +320,10 @@ func (_c *MockScheduler_GetSegmentTaskDelta_Call) RunAndReturn(run func(int64, i
 func (_m *MockScheduler) GetSegmentTaskNum() int {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentTaskNum")
+	}
+
 	var r0 int
 	if rf, ok := ret.Get(0).(func() int); ok {
 		r0 = rf()

internal/querynodev2/cluster/mock_manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package cluster
 
@@ -25,6 +25,10 @@ func (_m *MockManager) EXPECT() *MockManager_Expecter {
 func (_m *MockManager) GetWorker(ctx context.Context, nodeID int64) (Worker, error) {
 	ret := _m.Called(ctx, nodeID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetWorker")
+	}
+
 	var r0 Worker
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) (Worker, error)); ok {

internal/querynodev2/cluster/mock_worker.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package cluster
 
@@ -30,6 +30,10 @@ func (_m *MockWorker) EXPECT() *MockWorker_Expecter {
 func (_m *MockWorker) Delete(ctx context.Context, req *querypb.DeleteRequest) error {
 	ret := _m.Called(ctx, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Delete")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.DeleteRequest) error); ok {
 		r0 = rf(ctx, req)
@@ -73,6 +77,10 @@ func (_c *MockWorker_Delete_Call) RunAndReturn(run func(context.Context, *queryp
 func (_m *MockWorker) GetStatistics(ctx context.Context, req *querypb.GetStatisticsRequest) (*internalpb.GetStatisticsResponse, error) {
 	ret := _m.Called(ctx, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatistics")
+	}
+
 	var r0 *internalpb.GetStatisticsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetStatisticsRequest) (*internalpb.GetStatisticsResponse, error)); ok {
@@ -128,6 +136,10 @@ func (_c *MockWorker_GetStatistics_Call) RunAndReturn(run func(context.Context,
 func (_m *MockWorker) IsHealthy() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsHealthy")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()
@@ -169,6 +181,10 @@ func (_c *MockWorker_IsHealthy_Call) RunAndReturn(run func() bool) *MockWorker_I
 func (_m *MockWorker) LoadSegments(_a0 context.Context, _a1 *querypb.LoadSegmentsRequest) error {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadSegments")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadSegmentsRequest) error); ok {
 		r0 = rf(_a0, _a1)
@@ -212,6 +228,10 @@ func (_c *MockWorker_LoadSegments_Call) RunAndReturn(run func(context.Context, *
 func (_m *MockWorker) QuerySegments(ctx context.Context, req *querypb.QueryRequest) (*internalpb.RetrieveResults, error) {
 	ret := _m.Called(ctx, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QuerySegments")
+	}
+
 	var r0 *internalpb.RetrieveResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest) (*internalpb.RetrieveResults, error)); ok {
@@ -267,6 +287,10 @@ func (_c *MockWorker_QuerySegments_Call) RunAndReturn(run func(context.Context,
 func (_m *MockWorker) QueryStreamSegments(ctx context.Context, req *querypb.QueryRequest, srv streamrpc.QueryStreamServer) error {
 	ret := _m.Called(ctx, req, srv)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryStreamSegments")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest, streamrpc.QueryStreamServer) error); ok {
 		r0 = rf(ctx, req, srv)
@@ -311,6 +335,10 @@ func (_c *MockWorker_QueryStreamSegments_Call) RunAndReturn(run func(context.Con
 func (_m *MockWorker) ReleaseSegments(_a0 context.Context, _a1 *querypb.ReleaseSegmentsRequest) error {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleaseSegments")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseSegmentsRequest) error); ok {
 		r0 = rf(_a0, _a1)
@@ -354,6 +382,10 @@ func (_c *MockWorker_ReleaseSegments_Call) RunAndReturn(run func(context.Context
 func (_m *MockWorker) SearchSegments(ctx context.Context, req *querypb.SearchRequest) (*internalpb.SearchResults, error) {
 	ret := _m.Called(ctx, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SearchSegments")
+	}
+
 	var r0 *internalpb.SearchResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest) (*internalpb.SearchResults, error)); ok {

internal/querynodev2/delegator/mock_delegator.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package delegator
 
@@ -97,6 +97,10 @@ func (_c *MockShardDelegator_Close_Call) RunAndReturn(run func()) *MockShardDele
 func (_m *MockShardDelegator) Collection() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Collection")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()
@@ -138,6 +142,10 @@ func (_c *MockShardDelegator_Collection_Call) RunAndReturn(run func() int64) *Mo
 func (_m *MockShardDelegator) GetDeleteBufferSize() (int64, int64) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDeleteBufferSize")
+	}
+
 	var r0 int64
 	var r1 int64
 	if rf, ok := ret.Get(0).(func() (int64, int64)); ok {
@@ -189,6 +197,10 @@ func (_c *MockShardDelegator_GetDeleteBufferSize_Call) RunAndReturn(run func() (
 func (_m *MockShardDelegator) GetPartitionStatsVersions(ctx context.Context) map[int64]int64 {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPartitionStatsVersions")
+	}
+
 	var r0 map[int64]int64
 	if rf, ok := ret.Get(0).(func(context.Context) map[int64]int64); ok {
 		r0 = rf(ctx)
@@ -233,6 +245,10 @@ func (_c *MockShardDelegator_GetPartitionStatsVersions_Call) RunAndReturn(run fu
 func (_m *MockShardDelegator) GetSegmentInfo(readable bool) ([]SnapshotItem, []SegmentEntry) {
 	ret := _m.Called(readable)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSegmentInfo")
+	}
+
 	var r0 []SnapshotItem
 	var r1 []SegmentEntry
 	if rf, ok := ret.Get(0).(func(bool) ([]SnapshotItem, []SegmentEntry)); ok {
@@ -289,6 +305,10 @@ func (_c *MockShardDelegator_GetSegmentInfo_Call) RunAndReturn(run func(bool) ([
 func (_m *MockShardDelegator) GetStatistics(ctx context.Context, req *querypb.GetStatisticsRequest) ([]*internalpb.GetStatisticsResponse, error) {
 	ret := _m.Called(ctx, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStatistics")
+	}
+
 	var r0 []*internalpb.GetStatisticsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetStatisticsRequest) ([]*internalpb.GetStatisticsResponse, error)); ok {
@@ -344,6 +364,10 @@ func (_c *MockShardDelegator_GetStatistics_Call) RunAndReturn(run func(context.C
 func (_m *MockShardDelegator) GetTargetVersion() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetTargetVersion")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()
@@ -385,6 +409,10 @@ func (_c *MockShardDelegator_GetTargetVersion_Call) RunAndReturn(run func() int6
 func (_m *MockShardDelegator) LoadGrowing(ctx context.Context, infos []*querypb.SegmentLoadInfo, version int64) error {
 	ret := _m.Called(ctx, infos, version)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadGrowing")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, []*querypb.SegmentLoadInfo, int64) error); ok {
 		r0 = rf(ctx, infos, version)
@@ -429,6 +457,10 @@ func (_c *MockShardDelegator_LoadGrowing_Call) RunAndReturn(run func(context.Con
 func (_m *MockShardDelegator) LoadSegments(ctx context.Context, req *querypb.LoadSegmentsRequest) error {
 	ret := _m.Called(ctx, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadSegments")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadSegmentsRequest) error); ok {
 		r0 = rf(ctx, req)
@@ -539,6 +571,10 @@ func (_c *MockShardDelegator_ProcessInsert_Call) RunAndReturn(run func(map[int64
 func (_m *MockShardDelegator) Query(ctx context.Context, req *querypb.QueryRequest) ([]*internalpb.RetrieveResults, error) {
 	ret := _m.Called(ctx, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Query")
+	}
+
 	var r0 []*internalpb.RetrieveResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest) ([]*internalpb.RetrieveResults, error)); ok {
@@ -594,6 +630,10 @@ func (_c *MockShardDelegator_Query_Call) RunAndReturn(run func(context.Context,
 func (_m *MockShardDelegator) QueryStream(ctx context.Context, req *querypb.QueryRequest, srv streamrpc.QueryStreamServer) error {
 	ret := _m.Called(ctx, req, srv)
 
+	if len(ret) == 0 {
+		panic("no return value specified for QueryStream")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest, streamrpc.QueryStreamServer) error); ok {
 		r0 = rf(ctx, req, srv)
@@ -638,6 +678,10 @@ func (_c *MockShardDelegator_QueryStream_Call) RunAndReturn(run func(context.Con
 func (_m *MockShardDelegator) ReleaseSegments(ctx context.Context, req *querypb.ReleaseSegmentsRequest, force bool) error {
 	ret := _m.Called(ctx, req, force)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReleaseSegments")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseSegmentsRequest, bool) error); ok {
 		r0 = rf(ctx, req, force)
@@ -682,6 +726,10 @@ func (_c *MockShardDelegator_ReleaseSegments_Call) RunAndReturn(run func(context
 func (_m *MockShardDelegator) Search(ctx context.Context, req *querypb.SearchRequest) ([]*internalpb.SearchResults, error) {
 	ret := _m.Called(ctx, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Search")
+	}
+
 	var r0 []*internalpb.SearchResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest) ([]*internalpb.SearchResults, error)); ok {
@@ -737,6 +785,10 @@ func (_c *MockShardDelegator_Search_Call) RunAndReturn(run func(context.Context,
 func (_m *MockShardDelegator) Serviceable() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Serviceable")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()
@@ -962,6 +1014,10 @@ func (_c *MockShardDelegator_TryCleanExcludedSegments_Call) RunAndReturn(run fun
 func (_m *MockShardDelegator) VerifyExcludedSegments(segmentID int64, ts uint64) bool {
 	ret := _m.Called(segmentID, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for VerifyExcludedSegments")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(int64, uint64) bool); ok {
 		r0 = rf(segmentID, ts)
@@ -1005,6 +1061,10 @@ func (_c *MockShardDelegator_VerifyExcludedSegments_Call) RunAndReturn(run func(
 func (_m *MockShardDelegator) Version() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Version")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()

internal/querynodev2/optimizers/mock_query_hook.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package optimizers
 
@@ -21,6 +21,10 @@ func (_m *MockQueryHook) EXPECT() *MockQueryHook_Expecter {
 func (_m *MockQueryHook) DeleteTuningConfig(_a0 string) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteTuningConfig")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(_a0)
@@ -63,6 +67,10 @@ func (_c *MockQueryHook_DeleteTuningConfig_Call) RunAndReturn(run func(string) e
 func (_m *MockQueryHook) Init(_a0 string) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Init")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(_a0)
@@ -105,6 +113,10 @@ func (_c *MockQueryHook_Init_Call) RunAndReturn(run func(string) error) *MockQue
 func (_m *MockQueryHook) InitTuningConfig(_a0 map[string]string) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InitTuningConfig")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string) error); ok {
 		r0 = rf(_a0)
@@ -147,6 +159,10 @@ func (_c *MockQueryHook_InitTuningConfig_Call) RunAndReturn(run func(map[string]
 func (_m *MockQueryHook) Run(_a0 map[string]interface{}) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Run")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]interface{}) error); ok {
 		r0 = rf(_a0)

internal/querynodev2/segments/mock_collection_manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package segments
 
@@ -27,6 +27,10 @@ func (_m *MockCollectionManager) EXPECT() *MockCollectionManager_Expecter {
 func (_m *MockCollectionManager) Get(collectionID int64) *Collection {
 	ret := _m.Called(collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Get")
+	}
+
 	var r0 *Collection
 	if rf, ok := ret.Get(0).(func(int64) *Collection); ok {
 		r0 = rf(collectionID)
@@ -71,6 +75,10 @@ func (_c *MockCollectionManager_Get_Call) RunAndReturn(run func(int64) *Collecti
 func (_m *MockCollectionManager) List() []int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for List")
+	}
+
 	var r0 []int64
 	if rf, ok := ret.Get(0).(func() []int64); ok {
 		r0 = rf()
@@ -150,6 +158,10 @@ func (_c *MockCollectionManager_PutOrRef_Call) RunAndReturn(run func(int64, *sch
 func (_m *MockCollectionManager) Ref(collectionID int64, count uint32) bool {
 	ret := _m.Called(collectionID, count)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Ref")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(int64, uint32) bool); ok {
 		r0 = rf(collectionID, count)
@@ -193,6 +205,10 @@ func (_c *MockCollectionManager_Ref_Call) RunAndReturn(run func(int64, uint32) b
 func (_m *MockCollectionManager) Unref(collectionID int64, count uint32) bool {
 	ret := _m.Called(collectionID, count)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Unref")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(int64, uint32) bool); ok {
 		r0 = rf(collectionID, count)

internal/querynodev2/segments/mock_loader.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package segments
 
@@ -44,6 +44,10 @@ func (_m *MockLoader) Load(ctx context.Context, collectionID int64, segmentType
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Load")
+	}
+
 	var r0 []Segment
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, commonpb.SegmentState, int64, ...*querypb.SegmentLoadInfo) ([]Segment, error)); ok {
@@ -116,6 +120,10 @@ func (_m *MockLoader) LoadBM25Stats(ctx context.Context, collectionID int64, inf
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadBM25Stats")
+	}
+
 	var r0 *typeutil.ConcurrentMap[int64, map[int64]*storage.BM25Stats]
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, ...*querypb.SegmentLoadInfo) (*typeutil.ConcurrentMap[int64, map[int64]*storage.BM25Stats], error)); ok {
@@ -186,6 +194,10 @@ func (_m *MockLoader) LoadBloomFilterSet(ctx context.Context, collectionID int64
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadBloomFilterSet")
+	}
+
 	var r0 []*pkoracle.BloomFilterSet
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, int64, ...*querypb.SegmentLoadInfo) ([]*pkoracle.BloomFilterSet, error)); ok {
@@ -250,6 +262,10 @@ func (_c *MockLoader_LoadBloomFilterSet_Call) RunAndReturn(run func(context.Cont
 func (_m *MockLoader) LoadDeltaLogs(ctx context.Context, segment Segment, deltaLogs []*datapb.FieldBinlog) error {
 	ret := _m.Called(ctx, segment, deltaLogs)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadDeltaLogs")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, Segment, []*datapb.FieldBinlog) error); ok {
 		r0 = rf(ctx, segment, deltaLogs)
@@ -294,6 +310,10 @@ func (_c *MockLoader_LoadDeltaLogs_Call) RunAndReturn(run func(context.Context,
 func (_m *MockLoader) LoadIndex(ctx context.Context, segment Segment, info *querypb.SegmentLoadInfo, version int64) error {
 	ret := _m.Called(ctx, segment, info, version)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadIndex")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, Segment, *querypb.SegmentLoadInfo, int64) error); ok {
 		r0 = rf(ctx, segment, info, version)
@@ -339,6 +359,10 @@ func (_c *MockLoader_LoadIndex_Call) RunAndReturn(run func(context.Context, Segm
 func (_m *MockLoader) LoadLazySegment(ctx context.Context, segment Segment, loadInfo *querypb.SegmentLoadInfo) error {
 	ret := _m.Called(ctx, segment, loadInfo)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadLazySegment")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, Segment, *querypb.SegmentLoadInfo) error); ok {
 		r0 = rf(ctx, segment, loadInfo)

internal/querynodev2/segments/mock_segment.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package segments
 
@@ -39,6 +39,10 @@ func (_m *MockSegment) EXPECT() *MockSegment_Expecter {
 func (_m *MockSegment) BatchPkExist(lc *storage.BatchLocationsCache) []bool {
 	ret := _m.Called(lc)
 
+	if len(ret) == 0 {
+		panic("no return value specified for BatchPkExist")
+	}
+
 	var r0 []bool
 	if rf, ok := ret.Get(0).(func(*storage.BatchLocationsCache) []bool); ok {
 		r0 = rf(lc)
@@ -83,6 +87,10 @@ func (_c *MockSegment_BatchPkExist_Call) RunAndReturn(run func(*storage.BatchLoc
 func (_m *MockSegment) CASVersion(_a0 int64, _a1 int64) bool {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CASVersion")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(int64, int64) bool); ok {
 		r0 = rf(_a0, _a1)
@@ -126,6 +134,10 @@ func (_c *MockSegment_CASVersion_Call) RunAndReturn(run func(int64, int64) bool)
 func (_m *MockSegment) Collection() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Collection")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()
@@ -167,6 +179,10 @@ func (_c *MockSegment_Collection_Call) RunAndReturn(run func() int64) *MockSegme
 func (_m *MockSegment) DatabaseName() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for DatabaseName")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()
@@ -208,6 +224,10 @@ func (_c *MockSegment_DatabaseName_Call) RunAndReturn(run func() string) *MockSe
 func (_m *MockSegment) Delete(ctx context.Context, primaryKeys []storage.PrimaryKey, timestamps []uint64) error {
 	ret := _m.Called(ctx, primaryKeys, timestamps)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Delete")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, []storage.PrimaryKey, []uint64) error); ok {
 		r0 = rf(ctx, primaryKeys, timestamps)
@@ -252,6 +272,10 @@ func (_c *MockSegment_Delete_Call) RunAndReturn(run func(context.Context, []stor
 func (_m *MockSegment) ExistIndex(fieldID int64) bool {
 	ret := _m.Called(fieldID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ExistIndex")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(int64) bool); ok {
 		r0 = rf(fieldID)
@@ -294,6 +318,10 @@ func (_c *MockSegment_ExistIndex_Call) RunAndReturn(run func(int64) bool) *MockS
 func (_m *MockSegment) GetBM25Stats() map[int64]*storage.BM25Stats {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetBM25Stats")
+	}
+
 	var r0 map[int64]*storage.BM25Stats
 	if rf, ok := ret.Get(0).(func() map[int64]*storage.BM25Stats); ok {
 		r0 = rf()
@@ -337,6 +365,10 @@ func (_c *MockSegment_GetBM25Stats_Call) RunAndReturn(run func() map[int64]*stor
 func (_m *MockSegment) GetIndex(fieldID int64) *IndexedFieldInfo {
 	ret := _m.Called(fieldID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetIndex")
+	}
+
 	var r0 *IndexedFieldInfo
 	if rf, ok := ret.Get(0).(func(int64) *IndexedFieldInfo); ok {
 		r0 = rf(fieldID)
@@ -381,6 +413,10 @@ func (_c *MockSegment_GetIndex_Call) RunAndReturn(run func(int64) *IndexedFieldI
 func (_m *MockSegment) HasRawData(fieldID int64) bool {
 	ret := _m.Called(fieldID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for HasRawData")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(int64) bool); ok {
 		r0 = rf(fieldID)
@@ -423,6 +459,10 @@ func (_c *MockSegment_HasRawData_Call) RunAndReturn(run func(int64) bool) *MockS
 func (_m *MockSegment) ID() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for ID")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()
@@ -464,6 +504,10 @@ func (_c *MockSegment_ID_Call) RunAndReturn(run func() int64) *MockSegment_ID_Ca
 func (_m *MockSegment) Indexes() []*IndexedFieldInfo {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Indexes")
+	}
+
 	var r0 []*IndexedFieldInfo
 	if rf, ok := ret.Get(0).(func() []*IndexedFieldInfo); ok {
 		r0 = rf()
@@ -507,6 +551,10 @@ func (_c *MockSegment_Indexes_Call) RunAndReturn(run func() []*IndexedFieldInfo)
 func (_m *MockSegment) Insert(ctx context.Context, rowIDs []int64, timestamps []uint64, record *segcorepb.InsertRecord) error {
 	ret := _m.Called(ctx, rowIDs, timestamps, record)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Insert")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, []int64, []uint64, *segcorepb.InsertRecord) error); ok {
 		r0 = rf(ctx, rowIDs, timestamps, record)
@@ -552,6 +600,10 @@ func (_c *MockSegment_Insert_Call) RunAndReturn(run func(context.Context, []int6
 func (_m *MockSegment) InsertCount() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for InsertCount")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()
@@ -593,6 +645,10 @@ func (_c *MockSegment_InsertCount_Call) RunAndReturn(run func() int64) *MockSegm
 func (_m *MockSegment) IsLazyLoad() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsLazyLoad")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()
@@ -634,6 +690,10 @@ func (_c *MockSegment_IsLazyLoad_Call) RunAndReturn(run func() bool) *MockSegmen
 func (_m *MockSegment) IsSorted() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsSorted")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()
@@ -675,6 +735,10 @@ func (_c *MockSegment_IsSorted_Call) RunAndReturn(run func() bool) *MockSegment_
 func (_m *MockSegment) LastDeltaTimestamp() uint64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for LastDeltaTimestamp")
+	}
+
 	var r0 uint64
 	if rf, ok := ret.Get(0).(func() uint64); ok {
 		r0 = rf()
@@ -716,6 +780,10 @@ func (_c *MockSegment_LastDeltaTimestamp_Call) RunAndReturn(run func() uint64) *
 func (_m *MockSegment) Level() datapb.SegmentLevel {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Level")
+	}
+
 	var r0 datapb.SegmentLevel
 	if rf, ok := ret.Get(0).(func() datapb.SegmentLevel); ok {
 		r0 = rf()
@@ -757,6 +825,10 @@ func (_c *MockSegment_Level_Call) RunAndReturn(run func() datapb.SegmentLevel) *
 func (_m *MockSegment) LoadDeltaData(ctx context.Context, deltaData *storage.DeleteData) error {
 	ret := _m.Called(ctx, deltaData)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadDeltaData")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *storage.DeleteData) error); ok {
 		r0 = rf(ctx, deltaData)
@@ -800,6 +872,10 @@ func (_c *MockSegment_LoadDeltaData_Call) RunAndReturn(run func(context.Context,
 func (_m *MockSegment) LoadInfo() *querypb.SegmentLoadInfo {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadInfo")
+	}
+
 	var r0 *querypb.SegmentLoadInfo
 	if rf, ok := ret.Get(0).(func() *querypb.SegmentLoadInfo); ok {
 		r0 = rf()
@@ -843,6 +919,10 @@ func (_c *MockSegment_LoadInfo_Call) RunAndReturn(run func() *querypb.SegmentLoa
 func (_m *MockSegment) MayPkExist(lc *storage.LocationsCache) bool {
 	ret := _m.Called(lc)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MayPkExist")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(*storage.LocationsCache) bool); ok {
 		r0 = rf(lc)
@@ -885,6 +965,10 @@ func (_c *MockSegment_MayPkExist_Call) RunAndReturn(run func(*storage.LocationsC
 func (_m *MockSegment) MemSize() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for MemSize")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()
@@ -926,6 +1010,10 @@ func (_c *MockSegment_MemSize_Call) RunAndReturn(run func() int64) *MockSegment_
 func (_m *MockSegment) NeedUpdatedVersion() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for NeedUpdatedVersion")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()
@@ -967,6 +1055,10 @@ func (_c *MockSegment_NeedUpdatedVersion_Call) RunAndReturn(run func() int64) *M
 func (_m *MockSegment) Partition() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Partition")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()
@@ -1008,6 +1100,10 @@ func (_c *MockSegment_Partition_Call) RunAndReturn(run func() int64) *MockSegmen
 func (_m *MockSegment) PinIfNotReleased() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for PinIfNotReleased")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -1097,6 +1193,10 @@ func (_c *MockSegment_Release_Call) RunAndReturn(run func(context.Context, ...re
 func (_m *MockSegment) RemoveUnusedFieldFiles() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemoveUnusedFieldFiles")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -1171,6 +1271,10 @@ func (_c *MockSegment_ResetIndexesLazyLoad_Call) RunAndReturn(run func(bool)) *M
 func (_m *MockSegment) ResourceGroup() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for ResourceGroup")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()
@@ -1212,6 +1316,10 @@ func (_c *MockSegment_ResourceGroup_Call) RunAndReturn(run func() string) *MockS
 func (_m *MockSegment) ResourceUsageEstimate() ResourceUsage {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for ResourceUsageEstimate")
+	}
+
 	var r0 ResourceUsage
 	if rf, ok := ret.Get(0).(func() ResourceUsage); ok {
 		r0 = rf()
@@ -1253,6 +1361,10 @@ func (_c *MockSegment_ResourceUsageEstimate_Call) RunAndReturn(run func() Resour
 func (_m *MockSegment) Retrieve(ctx context.Context, plan *RetrievePlan) (*segcorepb.RetrieveResults, error) {
 	ret := _m.Called(ctx, plan)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Retrieve")
+	}
+
 	var r0 *segcorepb.RetrieveResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *RetrievePlan) (*segcorepb.RetrieveResults, error)); ok {
@@ -1308,6 +1420,10 @@ func (_c *MockSegment_Retrieve_Call) RunAndReturn(run func(context.Context, *Ret
 func (_m *MockSegment) RetrieveByOffsets(ctx context.Context, plan *RetrievePlan, offsets []int64) (*segcorepb.RetrieveResults, error) {
 	ret := _m.Called(ctx, plan, offsets)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RetrieveByOffsets")
+	}
+
 	var r0 *segcorepb.RetrieveResults
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *RetrievePlan, []int64) (*segcorepb.RetrieveResults, error)); ok {
@@ -1364,6 +1480,10 @@ func (_c *MockSegment_RetrieveByOffsets_Call) RunAndReturn(run func(context.Cont
 func (_m *MockSegment) RowNum() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for RowNum")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()
@@ -1405,6 +1525,10 @@ func (_c *MockSegment_RowNum_Call) RunAndReturn(run func() int64) *MockSegment_R
 func (_m *MockSegment) Search(ctx context.Context, searchReq *SearchRequest) (*SearchResult, error) {
 	ret := _m.Called(ctx, searchReq)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Search")
+	}
+
 	var r0 *SearchResult
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *SearchRequest) (*SearchResult, error)); ok {
@@ -1460,6 +1584,10 @@ func (_c *MockSegment_Search_Call) RunAndReturn(run func(context.Context, *Searc
 func (_m *MockSegment) Shard() metautil.Channel {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Shard")
+	}
+
 	var r0 metautil.Channel
 	if rf, ok := ret.Get(0).(func() metautil.Channel); ok {
 		r0 = rf()
@@ -1501,6 +1629,10 @@ func (_c *MockSegment_Shard_Call) RunAndReturn(run func() metautil.Channel) *Moc
 func (_m *MockSegment) StartPosition() *msgpb.MsgPosition {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartPosition")
+	}
+
 	var r0 *msgpb.MsgPosition
 	if rf, ok := ret.Get(0).(func() *msgpb.MsgPosition); ok {
 		r0 = rf()
@@ -1544,6 +1676,10 @@ func (_c *MockSegment_StartPosition_Call) RunAndReturn(run func() *msgpb.MsgPosi
 func (_m *MockSegment) Type() commonpb.SegmentState {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Type")
+	}
+
 	var r0 commonpb.SegmentState
 	if rf, ok := ret.Get(0).(func() commonpb.SegmentState); ok {
 		r0 = rf()
@@ -1683,6 +1819,10 @@ func (_c *MockSegment_UpdateBloomFilter_Call) RunAndReturn(run func([]storage.Pr
 func (_m *MockSegment) Version() int64 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Version")
+	}
+
 	var r0 int64
 	if rf, ok := ret.Get(0).(func() int64); ok {
 		r0 = rf()

internal/querynodev2/segments/mock_segment_manager.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package segments
 
@@ -62,6 +62,10 @@ func (_c *MockSegmentManager_Clear_Call) RunAndReturn(run func(context.Context))
 func (_m *MockSegmentManager) Empty() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Empty")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()
@@ -103,6 +107,10 @@ func (_c *MockSegmentManager_Empty_Call) RunAndReturn(run func() bool) *MockSegm
 func (_m *MockSegmentManager) Exist(segmentID int64, typ commonpb.SegmentState) bool {
 	ret := _m.Called(segmentID, typ)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Exist")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(int64, commonpb.SegmentState) bool); ok {
 		r0 = rf(segmentID, typ)
@@ -146,6 +154,10 @@ func (_c *MockSegmentManager_Exist_Call) RunAndReturn(run func(int64, commonpb.S
 func (_m *MockSegmentManager) Get(segmentID int64) Segment {
 	ret := _m.Called(segmentID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Get")
+	}
+
 	var r0 Segment
 	if rf, ok := ret.Get(0).(func(int64) Segment); ok {
 		r0 = rf(segmentID)
@@ -197,6 +209,10 @@ func (_m *MockSegmentManager) GetAndPin(segments []int64, filters ...SegmentFilt
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAndPin")
+	}
+
 	var r0 []Segment
 	var r1 error
 	if rf, ok := ret.Get(0).(func([]int64, ...SegmentFilter) ([]Segment, error)); ok {
@@ -265,6 +281,10 @@ func (_m *MockSegmentManager) GetAndPinBy(filters ...SegmentFilter) ([]Segment,
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAndPinBy")
+	}
+
 	var r0 []Segment
 	var r1 error
 	if rf, ok := ret.Get(0).(func(...SegmentFilter) ([]Segment, error)); ok {
@@ -332,6 +352,10 @@ func (_m *MockSegmentManager) GetBy(filters ...SegmentFilter) []Segment {
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetBy")
+	}
+
 	var r0 []Segment
 	if rf, ok := ret.Get(0).(func(...SegmentFilter) []Segment); ok {
 		r0 = rf(filters...)
@@ -383,6 +407,10 @@ func (_c *MockSegmentManager_GetBy_Call) RunAndReturn(run func(...SegmentFilter)
 func (_m *MockSegmentManager) GetGrowing(segmentID int64) Segment {
 	ret := _m.Called(segmentID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetGrowing")
+	}
+
 	var r0 Segment
 	if rf, ok := ret.Get(0).(func(int64) Segment); ok {
 		r0 = rf(segmentID)
@@ -427,6 +455,10 @@ func (_c *MockSegmentManager_GetGrowing_Call) RunAndReturn(run func(int64) Segme
 func (_m *MockSegmentManager) GetSealed(segmentID int64) Segment {
 	ret := _m.Called(segmentID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSealed")
+	}
+
 	var r0 Segment
 	if rf, ok := ret.Get(0).(func(int64) Segment); ok {
 		r0 = rf(segmentID)
@@ -471,6 +503,10 @@ func (_c *MockSegmentManager_GetSealed_Call) RunAndReturn(run func(int64) Segmen
 func (_m *MockSegmentManager) GetWithType(segmentID int64, typ commonpb.SegmentState) Segment {
 	ret := _m.Called(segmentID, typ)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetWithType")
+	}
+
 	var r0 Segment
 	if rf, ok := ret.Get(0).(func(int64, commonpb.SegmentState) Segment); ok {
 		r0 = rf(segmentID, typ)
@@ -565,6 +601,10 @@ func (_c *MockSegmentManager_Put_Call) RunAndReturn(run func(context.Context, co
 func (_m *MockSegmentManager) Remove(ctx context.Context, segmentID int64, scope querypb.DataScope) (int, int) {
 	ret := _m.Called(ctx, segmentID, scope)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Remove")
+	}
+
 	var r0 int
 	var r1 int
 	if rf, ok := ret.Get(0).(func(context.Context, int64, querypb.DataScope) (int, int)); ok {
@@ -626,6 +666,10 @@ func (_m *MockSegmentManager) RemoveBy(ctx context.Context, filters ...SegmentFi
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemoveBy")
+	}
+
 	var r0 int
 	var r1 int
 	if rf, ok := ret.Get(0).(func(context.Context, ...SegmentFilter) (int, int)); ok {
@@ -726,6 +770,10 @@ func (_m *MockSegmentManager) UpdateBy(action SegmentAction, filters ...SegmentF
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateBy")
+	}
+
 	var r0 int
 	if rf, ok := ret.Get(0).(func(SegmentAction, ...SegmentFilter) int); ok {
 		r0 = rf(action, filters...)

internal/rootcoord/mocks/garbage_collector.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mockrootcoord
 
@@ -26,6 +26,10 @@ func (_m *GarbageCollector) EXPECT() *GarbageCollector_Expecter {
 func (_m *GarbageCollector) GcCollectionData(ctx context.Context, coll *model.Collection) (uint64, error) {
 	ret := _m.Called(ctx, coll)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GcCollectionData")
+	}
+
 	var r0 uint64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *model.Collection) (uint64, error)); ok {
@@ -79,6 +83,10 @@ func (_c *GarbageCollector_GcCollectionData_Call) RunAndReturn(run func(context.
 func (_m *GarbageCollector) GcPartitionData(ctx context.Context, pChannels []string, vchannels []string, partition *model.Partition) (uint64, error) {
 	ret := _m.Called(ctx, pChannels, vchannels, partition)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GcPartitionData")
+	}
+
 	var r0 uint64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, []string, []string, *model.Partition) (uint64, error)); ok {

internal/rootcoord/mocks/meta_table.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mockrootcoord
 
@@ -34,6 +34,10 @@ func (_m *IMetaTable) EXPECT() *IMetaTable_Expecter {
 func (_m *IMetaTable) AddCollection(ctx context.Context, coll *model.Collection) error {
 	ret := _m.Called(ctx, coll)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AddCollection")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *model.Collection) error); ok {
 		r0 = rf(ctx, coll)
@@ -77,6 +81,10 @@ func (_c *IMetaTable_AddCollection_Call) RunAndReturn(run func(context.Context,
 func (_m *IMetaTable) AddCredential(credInfo *internalpb.CredentialInfo) error {
 	ret := _m.Called(credInfo)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AddCredential")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*internalpb.CredentialInfo) error); ok {
 		r0 = rf(credInfo)
@@ -119,6 +127,10 @@ func (_c *IMetaTable_AddCredential_Call) RunAndReturn(run func(*internalpb.Crede
 func (_m *IMetaTable) AddPartition(ctx context.Context, partition *model.Partition) error {
 	ret := _m.Called(ctx, partition)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AddPartition")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *model.Partition) error); ok {
 		r0 = rf(ctx, partition)
@@ -162,6 +174,10 @@ func (_c *IMetaTable_AddPartition_Call) RunAndReturn(run func(context.Context, *
 func (_m *IMetaTable) AlterAlias(ctx context.Context, dbName string, alias string, collectionName string, ts uint64) error {
 	ret := _m.Called(ctx, dbName, alias, collectionName, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AlterAlias")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, uint64) error); ok {
 		r0 = rf(ctx, dbName, alias, collectionName, ts)
@@ -208,6 +224,10 @@ func (_c *IMetaTable_AlterAlias_Call) RunAndReturn(run func(context.Context, str
 func (_m *IMetaTable) AlterCollection(ctx context.Context, oldColl *model.Collection, newColl *model.Collection, ts uint64) error {
 	ret := _m.Called(ctx, oldColl, newColl, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AlterCollection")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *model.Collection, *model.Collection, uint64) error); ok {
 		r0 = rf(ctx, oldColl, newColl, ts)
@@ -253,6 +273,10 @@ func (_c *IMetaTable_AlterCollection_Call) RunAndReturn(run func(context.Context
 func (_m *IMetaTable) AlterCredential(credInfo *internalpb.CredentialInfo) error {
 	ret := _m.Called(credInfo)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AlterCredential")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*internalpb.CredentialInfo) error); ok {
 		r0 = rf(credInfo)
@@ -295,6 +319,10 @@ func (_c *IMetaTable_AlterCredential_Call) RunAndReturn(run func(*internalpb.Cre
 func (_m *IMetaTable) AlterDatabase(ctx context.Context, oldDB *model.Database, newDB *model.Database, ts uint64) error {
 	ret := _m.Called(ctx, oldDB, newDB, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AlterDatabase")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *model.Database, *model.Database, uint64) error); ok {
 		r0 = rf(ctx, oldDB, newDB, ts)
@@ -340,6 +368,10 @@ func (_c *IMetaTable_AlterDatabase_Call) RunAndReturn(run func(context.Context,
 func (_m *IMetaTable) BackupRBAC(ctx context.Context, tenant string) (*milvuspb.RBACMeta, error) {
 	ret := _m.Called(ctx, tenant)
 
+	if len(ret) == 0 {
+		panic("no return value specified for BackupRBAC")
+	}
+
 	var r0 *milvuspb.RBACMeta
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*milvuspb.RBACMeta, error)); ok {
@@ -395,6 +427,10 @@ func (_c *IMetaTable_BackupRBAC_Call) RunAndReturn(run func(context.Context, str
 func (_m *IMetaTable) ChangeCollectionState(ctx context.Context, collectionID int64, state etcdpb.CollectionState, ts uint64) error {
 	ret := _m.Called(ctx, collectionID, state, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ChangeCollectionState")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, etcdpb.CollectionState, uint64) error); ok {
 		r0 = rf(ctx, collectionID, state, ts)
@@ -440,6 +476,10 @@ func (_c *IMetaTable_ChangeCollectionState_Call) RunAndReturn(run func(context.C
 func (_m *IMetaTable) ChangePartitionState(ctx context.Context, collectionID int64, partitionID int64, state etcdpb.PartitionState, ts uint64) error {
 	ret := _m.Called(ctx, collectionID, partitionID, state, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ChangePartitionState")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, int64, etcdpb.PartitionState, uint64) error); ok {
 		r0 = rf(ctx, collectionID, partitionID, state, ts)
@@ -486,6 +526,10 @@ func (_c *IMetaTable_ChangePartitionState_Call) RunAndReturn(run func(context.Co
 func (_m *IMetaTable) CreateAlias(ctx context.Context, dbName string, alias string, collectionName string, ts uint64) error {
 	ret := _m.Called(ctx, dbName, alias, collectionName, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateAlias")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, uint64) error); ok {
 		r0 = rf(ctx, dbName, alias, collectionName, ts)
@@ -532,6 +576,10 @@ func (_c *IMetaTable_CreateAlias_Call) RunAndReturn(run func(context.Context, st
 func (_m *IMetaTable) CreateDatabase(ctx context.Context, db *model.Database, ts uint64) error {
 	ret := _m.Called(ctx, db, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateDatabase")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *model.Database, uint64) error); ok {
 		r0 = rf(ctx, db, ts)
@@ -576,6 +624,10 @@ func (_c *IMetaTable_CreateDatabase_Call) RunAndReturn(run func(context.Context,
 func (_m *IMetaTable) CreateRole(tenant string, entity *milvuspb.RoleEntity) error {
 	ret := _m.Called(tenant, entity)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateRole")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, *milvuspb.RoleEntity) error); ok {
 		r0 = rf(tenant, entity)
@@ -619,6 +671,10 @@ func (_c *IMetaTable_CreateRole_Call) RunAndReturn(run func(string, *milvuspb.Ro
 func (_m *IMetaTable) DeleteCredential(username string) error {
 	ret := _m.Called(username)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteCredential")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(username)
@@ -661,6 +717,10 @@ func (_c *IMetaTable_DeleteCredential_Call) RunAndReturn(run func(string) error)
 func (_m *IMetaTable) DescribeAlias(ctx context.Context, dbName string, alias string, ts uint64) (string, error) {
 	ret := _m.Called(ctx, dbName, alias, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DescribeAlias")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, uint64) (string, error)); ok {
@@ -716,6 +776,10 @@ func (_c *IMetaTable_DescribeAlias_Call) RunAndReturn(run func(context.Context,
 func (_m *IMetaTable) DropAlias(ctx context.Context, dbName string, alias string, ts uint64) error {
 	ret := _m.Called(ctx, dbName, alias, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropAlias")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, uint64) error); ok {
 		r0 = rf(ctx, dbName, alias, ts)
@@ -761,6 +825,10 @@ func (_c *IMetaTable_DropAlias_Call) RunAndReturn(run func(context.Context, stri
 func (_m *IMetaTable) DropDatabase(ctx context.Context, dbName string, ts uint64) error {
 	ret := _m.Called(ctx, dbName, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropDatabase")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, uint64) error); ok {
 		r0 = rf(ctx, dbName, ts)
@@ -805,6 +873,10 @@ func (_c *IMetaTable_DropDatabase_Call) RunAndReturn(run func(context.Context, s
 func (_m *IMetaTable) DropGrant(tenant string, role *milvuspb.RoleEntity) error {
 	ret := _m.Called(tenant, role)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropGrant")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, *milvuspb.RoleEntity) error); ok {
 		r0 = rf(tenant, role)
@@ -848,6 +920,10 @@ func (_c *IMetaTable_DropGrant_Call) RunAndReturn(run func(string, *milvuspb.Rol
 func (_m *IMetaTable) DropRole(tenant string, roleName string) error {
 	ret := _m.Called(tenant, roleName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DropRole")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, string) error); ok {
 		r0 = rf(tenant, roleName)
@@ -891,6 +967,10 @@ func (_c *IMetaTable_DropRole_Call) RunAndReturn(run func(string, string) error)
 func (_m *IMetaTable) GetCollectionByID(ctx context.Context, dbName string, collectionID int64, ts uint64, allowUnavailable bool) (*model.Collection, error) {
 	ret := _m.Called(ctx, dbName, collectionID, ts, allowUnavailable)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollectionByID")
+	}
+
 	var r0 *model.Collection
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, int64, uint64, bool) (*model.Collection, error)); ok {
@@ -949,6 +1029,10 @@ func (_c *IMetaTable_GetCollectionByID_Call) RunAndReturn(run func(context.Conte
 func (_m *IMetaTable) GetCollectionByIDWithMaxTs(ctx context.Context, collectionID int64) (*model.Collection, error) {
 	ret := _m.Called(ctx, collectionID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollectionByIDWithMaxTs")
+	}
+
 	var r0 *model.Collection
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) (*model.Collection, error)); ok {
@@ -1004,6 +1088,10 @@ func (_c *IMetaTable_GetCollectionByIDWithMaxTs_Call) RunAndReturn(run func(cont
 func (_m *IMetaTable) GetCollectionByName(ctx context.Context, dbName string, collectionName string, ts uint64) (*model.Collection, error) {
 	ret := _m.Called(ctx, dbName, collectionName, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollectionByName")
+	}
+
 	var r0 *model.Collection
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, uint64) (*model.Collection, error)); ok {
@@ -1061,6 +1149,10 @@ func (_c *IMetaTable_GetCollectionByName_Call) RunAndReturn(run func(context.Con
 func (_m *IMetaTable) GetCollectionVirtualChannels(colID int64) []string {
 	ret := _m.Called(colID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCollectionVirtualChannels")
+	}
+
 	var r0 []string
 	if rf, ok := ret.Get(0).(func(int64) []string); ok {
 		r0 = rf(colID)
@@ -1105,6 +1197,10 @@ func (_c *IMetaTable_GetCollectionVirtualChannels_Call) RunAndReturn(run func(in
 func (_m *IMetaTable) GetCredential(username string) (*internalpb.CredentialInfo, error) {
 	ret := _m.Called(username)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetCredential")
+	}
+
 	var r0 *internalpb.CredentialInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (*internalpb.CredentialInfo, error)); ok {
@@ -1159,6 +1255,10 @@ func (_c *IMetaTable_GetCredential_Call) RunAndReturn(run func(string) (*interna
 func (_m *IMetaTable) GetDatabaseByID(ctx context.Context, dbID int64, ts uint64) (*model.Database, error) {
 	ret := _m.Called(ctx, dbID, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDatabaseByID")
+	}
+
 	var r0 *model.Database
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, uint64) (*model.Database, error)); ok {
@@ -1215,6 +1315,10 @@ func (_c *IMetaTable_GetDatabaseByID_Call) RunAndReturn(run func(context.Context
 func (_m *IMetaTable) GetDatabaseByName(ctx context.Context, dbName string, ts uint64) (*model.Database, error) {
 	ret := _m.Called(ctx, dbName, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDatabaseByName")
+	}
+
 	var r0 *model.Database
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, uint64) (*model.Database, error)); ok {
@@ -1271,6 +1375,10 @@ func (_c *IMetaTable_GetDatabaseByName_Call) RunAndReturn(run func(context.Conte
 func (_m *IMetaTable) GetPChannelInfo(pchannel string) *rootcoordpb.GetPChannelInfoResponse {
 	ret := _m.Called(pchannel)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPChannelInfo")
+	}
+
 	var r0 *rootcoordpb.GetPChannelInfoResponse
 	if rf, ok := ret.Get(0).(func(string) *rootcoordpb.GetPChannelInfoResponse); ok {
 		r0 = rf(pchannel)
@@ -1315,6 +1423,10 @@ func (_c *IMetaTable_GetPChannelInfo_Call) RunAndReturn(run func(string) *rootco
 func (_m *IMetaTable) IsAlias(db string, name string) bool {
 	ret := _m.Called(db, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsAlias")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(string, string) bool); ok {
 		r0 = rf(db, name)
@@ -1358,6 +1470,10 @@ func (_c *IMetaTable_IsAlias_Call) RunAndReturn(run func(string, string) bool) *
 func (_m *IMetaTable) ListAliases(ctx context.Context, dbName string, collectionName string, ts uint64) ([]string, error) {
 	ret := _m.Called(ctx, dbName, collectionName, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListAliases")
+	}
+
 	var r0 []string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, uint64) ([]string, error)); ok {
@@ -1415,6 +1531,10 @@ func (_c *IMetaTable_ListAliases_Call) RunAndReturn(run func(context.Context, st
 func (_m *IMetaTable) ListAliasesByID(collID int64) []string {
 	ret := _m.Called(collID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListAliasesByID")
+	}
+
 	var r0 []string
 	if rf, ok := ret.Get(0).(func(int64) []string); ok {
 		r0 = rf(collID)
@@ -1459,6 +1579,10 @@ func (_c *IMetaTable_ListAliasesByID_Call) RunAndReturn(run func(int64) []string
 func (_m *IMetaTable) ListAllAvailCollections(ctx context.Context) map[int64][]int64 {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListAllAvailCollections")
+	}
+
 	var r0 map[int64][]int64
 	if rf, ok := ret.Get(0).(func(context.Context) map[int64][]int64); ok {
 		r0 = rf(ctx)
@@ -1503,6 +1627,10 @@ func (_c *IMetaTable_ListAllAvailCollections_Call) RunAndReturn(run func(context
 func (_m *IMetaTable) ListCollectionPhysicalChannels() map[int64][]string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListCollectionPhysicalChannels")
+	}
+
 	var r0 map[int64][]string
 	if rf, ok := ret.Get(0).(func() map[int64][]string); ok {
 		r0 = rf()
@@ -1546,6 +1674,10 @@ func (_c *IMetaTable_ListCollectionPhysicalChannels_Call) RunAndReturn(run func(
 func (_m *IMetaTable) ListCollections(ctx context.Context, dbName string, ts uint64, onlyAvail bool) ([]*model.Collection, error) {
 	ret := _m.Called(ctx, dbName, ts, onlyAvail)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListCollections")
+	}
+
 	var r0 []*model.Collection
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, uint64, bool) ([]*model.Collection, error)); ok {
@@ -1603,6 +1735,10 @@ func (_c *IMetaTable_ListCollections_Call) RunAndReturn(run func(context.Context
 func (_m *IMetaTable) ListCredentialUsernames() (*milvuspb.ListCredUsersResponse, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListCredentialUsernames")
+	}
+
 	var r0 *milvuspb.ListCredUsersResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (*milvuspb.ListCredUsersResponse, error)); ok {
@@ -1656,6 +1792,10 @@ func (_c *IMetaTable_ListCredentialUsernames_Call) RunAndReturn(run func() (*mil
 func (_m *IMetaTable) ListDatabases(ctx context.Context, ts uint64) ([]*model.Database, error) {
 	ret := _m.Called(ctx, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListDatabases")
+	}
+
 	var r0 []*model.Database
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, uint64) ([]*model.Database, error)); ok {
@@ -1711,6 +1851,10 @@ func (_c *IMetaTable_ListDatabases_Call) RunAndReturn(run func(context.Context,
 func (_m *IMetaTable) ListPolicy(tenant string) ([]string, error) {
 	ret := _m.Called(tenant)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListPolicy")
+	}
+
 	var r0 []string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) ([]string, error)); ok {
@@ -1765,6 +1909,10 @@ func (_c *IMetaTable_ListPolicy_Call) RunAndReturn(run func(string) ([]string, e
 func (_m *IMetaTable) ListUserRole(tenant string) ([]string, error) {
 	ret := _m.Called(tenant)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListUserRole")
+	}
+
 	var r0 []string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) ([]string, error)); ok {
@@ -1819,6 +1967,10 @@ func (_c *IMetaTable_ListUserRole_Call) RunAndReturn(run func(string) ([]string,
 func (_m *IMetaTable) OperatePrivilege(tenant string, entity *milvuspb.GrantEntity, operateType milvuspb.OperatePrivilegeType) error {
 	ret := _m.Called(tenant, entity, operateType)
 
+	if len(ret) == 0 {
+		panic("no return value specified for OperatePrivilege")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, *milvuspb.GrantEntity, milvuspb.OperatePrivilegeType) error); ok {
 		r0 = rf(tenant, entity, operateType)
@@ -1863,6 +2015,10 @@ func (_c *IMetaTable_OperatePrivilege_Call) RunAndReturn(run func(string, *milvu
 func (_m *IMetaTable) OperateUserRole(tenant string, userEntity *milvuspb.UserEntity, roleEntity *milvuspb.RoleEntity, operateType milvuspb.OperateUserRoleType) error {
 	ret := _m.Called(tenant, userEntity, roleEntity, operateType)
 
+	if len(ret) == 0 {
+		panic("no return value specified for OperateUserRole")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, *milvuspb.UserEntity, *milvuspb.RoleEntity, milvuspb.OperateUserRoleType) error); ok {
 		r0 = rf(tenant, userEntity, roleEntity, operateType)
@@ -1908,6 +2064,10 @@ func (_c *IMetaTable_OperateUserRole_Call) RunAndReturn(run func(string, *milvus
 func (_m *IMetaTable) RemoveCollection(ctx context.Context, collectionID int64, ts uint64) error {
 	ret := _m.Called(ctx, collectionID, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemoveCollection")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, uint64) error); ok {
 		r0 = rf(ctx, collectionID, ts)
@@ -1952,6 +2112,10 @@ func (_c *IMetaTable_RemoveCollection_Call) RunAndReturn(run func(context.Contex
 func (_m *IMetaTable) RemovePartition(ctx context.Context, dbID int64, collectionID int64, partitionID int64, ts uint64) error {
 	ret := _m.Called(ctx, dbID, collectionID, partitionID, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemovePartition")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64, int64, int64, uint64) error); ok {
 		r0 = rf(ctx, dbID, collectionID, partitionID, ts)
@@ -1998,6 +2162,10 @@ func (_c *IMetaTable_RemovePartition_Call) RunAndReturn(run func(context.Context
 func (_m *IMetaTable) RenameCollection(ctx context.Context, dbName string, oldName string, newDBName string, newName string, ts uint64) error {
 	ret := _m.Called(ctx, dbName, oldName, newDBName, newName, ts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RenameCollection")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string, uint64) error); ok {
 		r0 = rf(ctx, dbName, oldName, newDBName, newName, ts)
@@ -2045,6 +2213,10 @@ func (_c *IMetaTable_RenameCollection_Call) RunAndReturn(run func(context.Contex
 func (_m *IMetaTable) RestoreRBAC(ctx context.Context, tenant string, meta *milvuspb.RBACMeta) error {
 	ret := _m.Called(ctx, tenant, meta)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RestoreRBAC")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, *milvuspb.RBACMeta) error); ok {
 		r0 = rf(ctx, tenant, meta)
@@ -2089,6 +2261,10 @@ func (_c *IMetaTable_RestoreRBAC_Call) RunAndReturn(run func(context.Context, st
 func (_m *IMetaTable) SelectGrant(tenant string, entity *milvuspb.GrantEntity) ([]*milvuspb.GrantEntity, error) {
 	ret := _m.Called(tenant, entity)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SelectGrant")
+	}
+
 	var r0 []*milvuspb.GrantEntity
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string, *milvuspb.GrantEntity) ([]*milvuspb.GrantEntity, error)); ok {
@@ -2144,6 +2320,10 @@ func (_c *IMetaTable_SelectGrant_Call) RunAndReturn(run func(string, *milvuspb.G
 func (_m *IMetaTable) SelectRole(tenant string, entity *milvuspb.RoleEntity, includeUserInfo bool) ([]*milvuspb.RoleResult, error) {
 	ret := _m.Called(tenant, entity, includeUserInfo)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SelectRole")
+	}
+
 	var r0 []*milvuspb.RoleResult
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string, *milvuspb.RoleEntity, bool) ([]*milvuspb.RoleResult, error)); ok {
@@ -2200,6 +2380,10 @@ func (_c *IMetaTable_SelectRole_Call) RunAndReturn(run func(string, *milvuspb.Ro
 func (_m *IMetaTable) SelectUser(tenant string, entity *milvuspb.UserEntity, includeRoleInfo bool) ([]*milvuspb.UserResult, error) {
 	ret := _m.Called(tenant, entity, includeRoleInfo)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SelectUser")
+	}
+
 	var r0 []*milvuspb.UserResult
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string, *milvuspb.UserEntity, bool) ([]*milvuspb.UserResult, error)); ok {

pkg/eventlog/mock_logger.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package eventlog
 
@@ -21,6 +21,10 @@ func (_m *MockLogger) EXPECT() *MockLogger_Expecter {
 func (_m *MockLogger) Flush() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Flush")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()

pkg/kv/predicates/mock_predicate.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package predicates
 
@@ -21,6 +21,10 @@ func (_m *MockPredicate) EXPECT() *MockPredicate_Expecter {
 func (_m *MockPredicate) IsTrue(_a0 interface{}) bool {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsTrue")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(interface{}) bool); ok {
 		r0 = rf(_a0)
@@ -63,6 +67,10 @@ func (_c *MockPredicate_IsTrue_Call) RunAndReturn(run func(interface{}) bool) *M
 func (_m *MockPredicate) Key() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Key")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()
@@ -104,6 +112,10 @@ func (_c *MockPredicate_Key_Call) RunAndReturn(run func() string) *MockPredicate
 func (_m *MockPredicate) Target() PredicateTarget {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Target")
+	}
+
 	var r0 PredicateTarget
 	if rf, ok := ret.Get(0).(func() PredicateTarget); ok {
 		r0 = rf()
@@ -145,6 +157,10 @@ func (_c *MockPredicate_Target_Call) RunAndReturn(run func() PredicateTarget) *M
 func (_m *MockPredicate) TargetValue() interface{} {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for TargetValue")
+	}
+
 	var r0 interface{}
 	if rf, ok := ret.Get(0).(func() interface{}); ok {
 		r0 = rf()
@@ -188,6 +204,10 @@ func (_c *MockPredicate_TargetValue_Call) RunAndReturn(run func() interface{}) *
 func (_m *MockPredicate) Type() PredicateType {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Type")
+	}
+
 	var r0 PredicateType
 	if rf, ok := ret.Get(0).(func() PredicateType); ok {
 		r0 = rf()

pkg/mocks/mock_kv/mock_MetaKv.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_kv
 
@@ -56,6 +56,10 @@ func (_c *MockMetaKv_Close_Call) RunAndReturn(run func()) *MockMetaKv_Close_Call
 func (_m *MockMetaKv) CompareVersionAndSwap(key string, version int64, target string) (bool, error) {
 	ret := _m.Called(key, version, target)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CompareVersionAndSwap")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string, int64, string) (bool, error)); ok {
@@ -110,6 +114,10 @@ func (_c *MockMetaKv_CompareVersionAndSwap_Call) RunAndReturn(run func(string, i
 func (_m *MockMetaKv) GetPath(key string) string {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPath")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func(string) string); ok {
 		r0 = rf(key)
@@ -152,6 +160,10 @@ func (_c *MockMetaKv_GetPath_Call) RunAndReturn(run func(string) string) *MockMe
 func (_m *MockMetaKv) Has(key string) (bool, error) {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Has")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (bool, error)); ok {
@@ -204,6 +216,10 @@ func (_c *MockMetaKv_Has_Call) RunAndReturn(run func(string) (bool, error)) *Moc
 func (_m *MockMetaKv) HasPrefix(prefix string) (bool, error) {
 	ret := _m.Called(prefix)
 
+	if len(ret) == 0 {
+		panic("no return value specified for HasPrefix")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (bool, error)); ok {
@@ -256,6 +272,10 @@ func (_c *MockMetaKv_HasPrefix_Call) RunAndReturn(run func(string) (bool, error)
 func (_m *MockMetaKv) Load(key string) (string, error) {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Load")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (string, error)); ok {
@@ -308,6 +328,10 @@ func (_c *MockMetaKv_Load_Call) RunAndReturn(run func(string) (string, error)) *
 func (_m *MockMetaKv) LoadWithPrefix(key string) ([]string, []string, error) {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LoadWithPrefix")
+	}
+
 	var r0 []string
 	var r1 []string
 	var r2 error
@@ -371,6 +395,10 @@ func (_c *MockMetaKv_LoadWithPrefix_Call) RunAndReturn(run func(string) ([]strin
 func (_m *MockMetaKv) MultiLoad(keys []string) ([]string, error) {
 	ret := _m.Called(keys)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiLoad")
+	}
+
 	var r0 []string
 	var r1 error
 	if rf, ok := ret.Get(0).(func([]string) ([]string, error)); ok {
@@ -425,6 +453,10 @@ func (_c *MockMetaKv_MultiLoad_Call) RunAndReturn(run func([]string) ([]string,
 func (_m *MockMetaKv) MultiRemove(keys []string) error {
 	ret := _m.Called(keys)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiRemove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func([]string) error); ok {
 		r0 = rf(keys)
@@ -467,6 +499,10 @@ func (_c *MockMetaKv_MultiRemove_Call) RunAndReturn(run func([]string) error) *M
 func (_m *MockMetaKv) MultiSave(kvs map[string]string) error {
 	ret := _m.Called(kvs)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSave")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string) error); ok {
 		r0 = rf(kvs)
@@ -516,6 +552,10 @@ func (_m *MockMetaKv) MultiSaveAndRemove(saves map[string]string, removals []str
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSaveAndRemove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok {
 		r0 = rf(saves, removals, preds...)
@@ -574,6 +614,10 @@ func (_m *MockMetaKv) MultiSaveAndRemoveWithPrefix(saves map[string]string, remo
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for MultiSaveAndRemoveWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok {
 		r0 = rf(saves, removals, preds...)
@@ -625,6 +669,10 @@ func (_c *MockMetaKv_MultiSaveAndRemoveWithPrefix_Call) RunAndReturn(run func(ma
 func (_m *MockMetaKv) Remove(key string) error {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Remove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(key)
@@ -667,6 +715,10 @@ func (_c *MockMetaKv_Remove_Call) RunAndReturn(run func(string) error) *MockMeta
 func (_m *MockMetaKv) RemoveWithPrefix(key string) error {
 	ret := _m.Called(key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemoveWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(key)
@@ -709,6 +761,10 @@ func (_c *MockMetaKv_RemoveWithPrefix_Call) RunAndReturn(run func(string) error)
 func (_m *MockMetaKv) Save(key string, value string) error {
 	ret := _m.Called(key, value)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Save")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, string) error); ok {
 		r0 = rf(key, value)
@@ -752,6 +808,10 @@ func (_c *MockMetaKv_Save_Call) RunAndReturn(run func(string, string) error) *Mo
 func (_m *MockMetaKv) WalkWithPrefix(prefix string, paginationSize int, fn func([]byte, []byte) error) error {
 	ret := _m.Called(prefix, paginationSize, fn)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WalkWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, int, func([]byte, []byte) error) error); ok {
 		r0 = rf(prefix, paginationSize, fn)

pkg/mocks/streaming/mock_walimpls/mock_OpenerBuilderImpls.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_walimpls
 
@@ -24,6 +24,10 @@ func (_m *MockOpenerBuilderImpls) EXPECT() *MockOpenerBuilderImpls_Expecter {
 func (_m *MockOpenerBuilderImpls) Build() (walimpls.OpenerImpls, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Build")
+	}
+
 	var r0 walimpls.OpenerImpls
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (walimpls.OpenerImpls, error)); ok {
@@ -77,6 +81,10 @@ func (_c *MockOpenerBuilderImpls_Build_Call) RunAndReturn(run func() (walimpls.O
 func (_m *MockOpenerBuilderImpls) Name() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Name")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()

pkg/mocks/streaming/mock_walimpls/mock_OpenerImpls.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_walimpls
 
@@ -58,6 +58,10 @@ func (_c *MockOpenerImpls_Close_Call) RunAndReturn(run func()) *MockOpenerImpls_
 func (_m *MockOpenerImpls) Open(ctx context.Context, opt *walimpls.OpenOption) (walimpls.WALImpls, error) {
 	ret := _m.Called(ctx, opt)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Open")
+	}
+
 	var r0 walimpls.WALImpls
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *walimpls.OpenOption) (walimpls.WALImpls, error)); ok {

pkg/mocks/streaming/mock_walimpls/mock_ScannerImpls.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_walimpls
 
@@ -24,6 +24,10 @@ func (_m *MockScannerImpls) EXPECT() *MockScannerImpls_Expecter {
 func (_m *MockScannerImpls) Chan() <-chan message.ImmutableMessage {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Chan")
+	}
+
 	var r0 <-chan message.ImmutableMessage
 	if rf, ok := ret.Get(0).(func() <-chan message.ImmutableMessage); ok {
 		r0 = rf()
@@ -67,6 +71,10 @@ func (_c *MockScannerImpls_Chan_Call) RunAndReturn(run func() <-chan message.Imm
 func (_m *MockScannerImpls) Close() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Close")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -108,6 +116,10 @@ func (_c *MockScannerImpls_Close_Call) RunAndReturn(run func() error) *MockScann
 func (_m *MockScannerImpls) Done() <-chan struct{} {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Done")
+	}
+
 	var r0 <-chan struct{}
 	if rf, ok := ret.Get(0).(func() <-chan struct{}); ok {
 		r0 = rf()
@@ -151,6 +163,10 @@ func (_c *MockScannerImpls_Done_Call) RunAndReturn(run func() <-chan struct{}) *
 func (_m *MockScannerImpls) Error() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Error")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -192,6 +208,10 @@ func (_c *MockScannerImpls_Error_Call) RunAndReturn(run func() error) *MockScann
 func (_m *MockScannerImpls) Name() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Name")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()

pkg/mocks/streaming/mock_walimpls/mock_WALImpls.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_walimpls
 
@@ -30,6 +30,10 @@ func (_m *MockWALImpls) EXPECT() *MockWALImpls_Expecter {
 func (_m *MockWALImpls) Append(ctx context.Context, msg message.MutableMessage) (message.MessageID, error) {
 	ret := _m.Called(ctx, msg)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Append")
+	}
+
 	var r0 message.MessageID
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, message.MutableMessage) (message.MessageID, error)); ok {
@@ -85,6 +89,10 @@ func (_c *MockWALImpls_Append_Call) RunAndReturn(run func(context.Context, messa
 func (_m *MockWALImpls) Channel() types.PChannelInfo {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Channel")
+	}
+
 	var r0 types.PChannelInfo
 	if rf, ok := ret.Get(0).(func() types.PChannelInfo); ok {
 		r0 = rf()
@@ -158,6 +166,10 @@ func (_c *MockWALImpls_Close_Call) RunAndReturn(run func()) *MockWALImpls_Close_
 func (_m *MockWALImpls) Read(ctx context.Context, opts walimpls.ReadOption) (walimpls.ScannerImpls, error) {
 	ret := _m.Called(ctx, opts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Read")
+	}
+
 	var r0 walimpls.ScannerImpls
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, walimpls.ReadOption) (walimpls.ScannerImpls, error)); ok {
@@ -213,6 +225,10 @@ func (_c *MockWALImpls_Read_Call) RunAndReturn(run func(context.Context, walimpl
 func (_m *MockWALImpls) WALName() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for WALName")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()

pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_streamingpb
 
@@ -36,6 +36,10 @@ func (_m *MockStreamingCoordAssignmentServiceClient) AssignmentDiscover(ctx cont
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AssignmentDiscover")
+	}
+
 	var r0 streamingpb.StreamingCoordAssignmentService_AssignmentDiscoverClient
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, ...grpc.CallOption) (streamingpb.StreamingCoordAssignmentService_AssignmentDiscoverClient, error)); ok {

pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_streamingpb
 
@@ -28,6 +28,10 @@ func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) EXPECT()
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) CloseSend() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for CloseSend")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -69,6 +73,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_CloseSend
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Context() context.Context {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Context")
+	}
+
 	var r0 context.Context
 	if rf, ok := ret.Get(0).(func() context.Context); ok {
 		r0 = rf()
@@ -112,6 +120,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Context_C
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Header() (metadata.MD, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Header")
+	}
+
 	var r0 metadata.MD
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (metadata.MD, error)); ok {
@@ -165,6 +177,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Header_Ca
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Recv() (*streamingpb.AssignmentDiscoverResponse, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Recv")
+	}
+
 	var r0 *streamingpb.AssignmentDiscoverResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (*streamingpb.AssignmentDiscoverResponse, error)); ok {
@@ -218,6 +234,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Recv_Call
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) RecvMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RecvMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -260,6 +280,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_RecvMsg_C
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Send(_a0 *streamingpb.AssignmentDiscoverRequest) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Send")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*streamingpb.AssignmentDiscoverRequest) error); ok {
 		r0 = rf(_a0)
@@ -302,6 +326,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Send_Call
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) SendMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SendMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -344,6 +372,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_SendMsg_C
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Trailer() metadata.MD {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Trailer")
+	}
+
 	var r0 metadata.MD
 	if rf, ok := ret.Get(0).(func() metadata.MD); ok {
 		r0 = rf()

pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_streamingpb
 
@@ -28,6 +28,10 @@ func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) EXPECT()
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) Context() context.Context {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Context")
+	}
+
 	var r0 context.Context
 	if rf, ok := ret.Get(0).(func() context.Context); ok {
 		r0 = rf()
@@ -71,6 +75,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_Context_C
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) Recv() (*streamingpb.AssignmentDiscoverRequest, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Recv")
+	}
+
 	var r0 *streamingpb.AssignmentDiscoverRequest
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (*streamingpb.AssignmentDiscoverRequest, error)); ok {
@@ -124,6 +132,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_Recv_Call
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) RecvMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RecvMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -166,6 +178,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_RecvMsg_C
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) Send(_a0 *streamingpb.AssignmentDiscoverResponse) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Send")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*streamingpb.AssignmentDiscoverResponse) error); ok {
 		r0 = rf(_a0)
@@ -208,6 +224,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_Send_Call
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) SendHeader(_a0 metadata.MD) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SendHeader")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(metadata.MD) error); ok {
 		r0 = rf(_a0)
@@ -250,6 +270,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_SendHeade
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) SendMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SendMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -292,6 +316,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_SendMsg_C
 func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) SetHeader(_a0 metadata.MD) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetHeader")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(metadata.MD) error); ok {
 		r0 = rf(_a0)

pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_streamingpb
 
@@ -36,6 +36,10 @@ func (_m *MockStreamingNodeHandlerServiceClient) Consume(ctx context.Context, op
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Consume")
+	}
+
 	var r0 streamingpb.StreamingNodeHandlerService_ConsumeClient
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, ...grpc.CallOption) (streamingpb.StreamingNodeHandlerService_ConsumeClient, error)); ok {
@@ -105,6 +109,10 @@ func (_m *MockStreamingNodeHandlerServiceClient) Produce(ctx context.Context, op
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Produce")
+	}
+
 	var r0 streamingpb.StreamingNodeHandlerService_ProduceClient
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, ...grpc.CallOption) (streamingpb.StreamingNodeHandlerService_ProduceClient, error)); ok {

pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_streamingpb
 
@@ -28,6 +28,10 @@ func (_m *MockStreamingNodeHandlerService_ConsumeClient) EXPECT() *MockStreaming
 func (_m *MockStreamingNodeHandlerService_ConsumeClient) CloseSend() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for CloseSend")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -69,6 +73,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_CloseSend_Call) RunAndRe
 func (_m *MockStreamingNodeHandlerService_ConsumeClient) Context() context.Context {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Context")
+	}
+
 	var r0 context.Context
 	if rf, ok := ret.Get(0).(func() context.Context); ok {
 		r0 = rf()
@@ -112,6 +120,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_Context_Call) RunAndRetu
 func (_m *MockStreamingNodeHandlerService_ConsumeClient) Header() (metadata.MD, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Header")
+	}
+
 	var r0 metadata.MD
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (metadata.MD, error)); ok {
@@ -165,6 +177,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_Header_Call) RunAndRetur
 func (_m *MockStreamingNodeHandlerService_ConsumeClient) Recv() (*streamingpb.ConsumeResponse, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Recv")
+	}
+
 	var r0 *streamingpb.ConsumeResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (*streamingpb.ConsumeResponse, error)); ok {
@@ -218,6 +234,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_Recv_Call) RunAndReturn(
 func (_m *MockStreamingNodeHandlerService_ConsumeClient) RecvMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RecvMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -260,6 +280,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_RecvMsg_Call) RunAndRetu
 func (_m *MockStreamingNodeHandlerService_ConsumeClient) Send(_a0 *streamingpb.ConsumeRequest) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Send")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*streamingpb.ConsumeRequest) error); ok {
 		r0 = rf(_a0)
@@ -302,6 +326,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_Send_Call) RunAndReturn(
 func (_m *MockStreamingNodeHandlerService_ConsumeClient) SendMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SendMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -344,6 +372,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_SendMsg_Call) RunAndRetu
 func (_m *MockStreamingNodeHandlerService_ConsumeClient) Trailer() metadata.MD {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Trailer")
+	}
+
 	var r0 metadata.MD
 	if rf, ok := ret.Get(0).(func() metadata.MD); ok {
 		r0 = rf()

pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_streamingpb
 
@@ -28,6 +28,10 @@ func (_m *MockStreamingNodeHandlerService_ConsumeServer) EXPECT() *MockStreaming
 func (_m *MockStreamingNodeHandlerService_ConsumeServer) Context() context.Context {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Context")
+	}
+
 	var r0 context.Context
 	if rf, ok := ret.Get(0).(func() context.Context); ok {
 		r0 = rf()
@@ -71,6 +75,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_Context_Call) RunAndRetu
 func (_m *MockStreamingNodeHandlerService_ConsumeServer) Recv() (*streamingpb.ConsumeRequest, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Recv")
+	}
+
 	var r0 *streamingpb.ConsumeRequest
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (*streamingpb.ConsumeRequest, error)); ok {
@@ -124,6 +132,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_Recv_Call) RunAndReturn(
 func (_m *MockStreamingNodeHandlerService_ConsumeServer) RecvMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RecvMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -166,6 +178,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_RecvMsg_Call) RunAndRetu
 func (_m *MockStreamingNodeHandlerService_ConsumeServer) Send(_a0 *streamingpb.ConsumeResponse) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Send")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*streamingpb.ConsumeResponse) error); ok {
 		r0 = rf(_a0)
@@ -208,6 +224,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_Send_Call) RunAndReturn(
 func (_m *MockStreamingNodeHandlerService_ConsumeServer) SendHeader(_a0 metadata.MD) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SendHeader")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(metadata.MD) error); ok {
 		r0 = rf(_a0)
@@ -250,6 +270,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_SendHeader_Call) RunAndR
 func (_m *MockStreamingNodeHandlerService_ConsumeServer) SendMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SendMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -292,6 +316,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_SendMsg_Call) RunAndRetu
 func (_m *MockStreamingNodeHandlerService_ConsumeServer) SetHeader(_a0 metadata.MD) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetHeader")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(metadata.MD) error); ok {
 		r0 = rf(_a0)

pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_streamingpb
 
@@ -28,6 +28,10 @@ func (_m *MockStreamingNodeHandlerService_ProduceClient) EXPECT() *MockStreaming
 func (_m *MockStreamingNodeHandlerService_ProduceClient) CloseSend() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for CloseSend")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -69,6 +73,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_CloseSend_Call) RunAndRe
 func (_m *MockStreamingNodeHandlerService_ProduceClient) Context() context.Context {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Context")
+	}
+
 	var r0 context.Context
 	if rf, ok := ret.Get(0).(func() context.Context); ok {
 		r0 = rf()
@@ -112,6 +120,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_Context_Call) RunAndRetu
 func (_m *MockStreamingNodeHandlerService_ProduceClient) Header() (metadata.MD, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Header")
+	}
+
 	var r0 metadata.MD
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (metadata.MD, error)); ok {
@@ -165,6 +177,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_Header_Call) RunAndRetur
 func (_m *MockStreamingNodeHandlerService_ProduceClient) Recv() (*streamingpb.ProduceResponse, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Recv")
+	}
+
 	var r0 *streamingpb.ProduceResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (*streamingpb.ProduceResponse, error)); ok {
@@ -218,6 +234,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_Recv_Call) RunAndReturn(
 func (_m *MockStreamingNodeHandlerService_ProduceClient) RecvMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RecvMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -260,6 +280,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_RecvMsg_Call) RunAndRetu
 func (_m *MockStreamingNodeHandlerService_ProduceClient) Send(_a0 *streamingpb.ProduceRequest) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Send")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*streamingpb.ProduceRequest) error); ok {
 		r0 = rf(_a0)
@@ -302,6 +326,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_Send_Call) RunAndReturn(
 func (_m *MockStreamingNodeHandlerService_ProduceClient) SendMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SendMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -344,6 +372,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_SendMsg_Call) RunAndRetu
 func (_m *MockStreamingNodeHandlerService_ProduceClient) Trailer() metadata.MD {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Trailer")
+	}
+
 	var r0 metadata.MD
 	if rf, ok := ret.Get(0).(func() metadata.MD); ok {
 		r0 = rf()

pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go

@@ -1,4 +1,4 @@
-// Code generated by mockery v2.32.4. DO NOT EDIT.
+// Code generated by mockery v2.46.0. DO NOT EDIT.
 
 package mock_streamingpb
 
@@ -28,6 +28,10 @@ func (_m *MockStreamingNodeHandlerService_ProduceServer) EXPECT() *MockStreaming
 func (_m *MockStreamingNodeHandlerService_ProduceServer) Context() context.Context {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Context")
+	}
+
 	var r0 context.Context
 	if rf, ok := ret.Get(0).(func() context.Context); ok {
 		r0 = rf()
@@ -71,6 +75,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_Context_Call) RunAndRetu
 func (_m *MockStreamingNodeHandlerService_ProduceServer) Recv() (*streamingpb.ProduceRequest, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Recv")
+	}
+
 	var r0 *streamingpb.ProduceRequest
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (*streamingpb.ProduceRequest, error)); ok {
@@ -124,6 +132,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_Recv_Call) RunAndReturn(
 func (_m *MockStreamingNodeHandlerService_ProduceServer) RecvMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RecvMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -166,6 +178,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_RecvMsg_Call) RunAndRetu
 func (_m *MockStreamingNodeHandlerService_ProduceServer) Send(_a0 *streamingpb.ProduceResponse) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Send")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*streamingpb.ProduceResponse) error); ok {
 		r0 = rf(_a0)
@@ -208,6 +224,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_Send_Call) RunAndReturn(
 func (_m *MockStreamingNodeHandlerService_ProduceServer) SendHeader(_a0 metadata.MD) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SendHeader")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(metadata.MD) error); ok {
 		r0 = rf(_a0)
@@ -250,6 +270,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_SendHeader_Call) RunAndR
 func (_m *MockStreamingNodeHandlerService_ProduceServer) SendMsg(m interface{}) error {
 	ret := _m.Called(m)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SendMsg")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(interface{}) error); ok {
 		r0 = rf(m)
@@ -292,6 +316,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_SendMsg_Call) RunAndRetu
 func (_m *MockStreamingNodeHandlerService_ProduceServer) SetHeader(_a0 metadata.MD) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetHeader")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(metadata.MD) error); ok {
 		r0 = rf(_a0)