admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-06 17:18:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

enhance: [StorageV2] Add storage version in Data/Query view resp (#43348)

Browse Source 
Related to #39173

Add `storage_version` in data/query view segment info response

---------

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
This commit is contained in:
congqixia 2025-07-16 15:52:51 +08:00 committed by GitHub
parent 58a9e49066
commit 5d90b65342
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
14 changed files with 2089 additions and 1248 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
client/go.mod
View File

@ -6,13 +6,15 @@ require (
github.com/blang/semver/v4 v4.0.0
github.com/cockroachdb/errors v1.9.1
github.com/grpc-ecosystem/go-grpc-middleware v1.3.0
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1.0.20250716031043-88051c3893ce
github.com/milvus-io/milvus/pkg/v2 v2.0.0-20250319085209-5a6b4e56d59e
github.com/quasilyte/go-ruleguard/dsl v0.3.22
github.com/samber/lo v1.27.0
github.com/stretchr/testify v1.10.0
github.com/tidwall/gjson v1.17.1
go.opentelemetry.io/otel v1.28.0
go.uber.org/atomic v1.11.0
golang.org/x/exp v0.0.0-20240506185415-9bf2ced13842
google.golang.org/grpc v1.65.0
google.golang.org/protobuf v1.34.2
)
@ -87,7 +89,6 @@ require (
go.etcd.io/etcd/raft/v3 v3.5.5 // indirect
go.etcd.io/etcd/server/v3 v3.5.5 // indirect
go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.49.0 // indirect
go.opentelemetry.io/otel v1.28.0 // indirect
go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.20.0 // indirect
go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc v1.20.0 // indirect
go.opentelemetry.io/otel/metric v1.28.0 // indirect
@ -98,7 +99,6 @@ require (
go.uber.org/multierr v1.11.0 // indirect
go.uber.org/zap v1.27.0 // indirect
golang.org/x/crypto v0.36.0 // indirect
golang.org/x/exp v0.0.0-20240506185415-9bf2ced13842 // indirect
golang.org/x/net v0.38.0 // indirect
golang.org/x/sync v0.12.0 // indirect
golang.org/x/sys v0.31.0 // indirect

4
client/go.sum
View File

@ -318,8 +318,8 @@ github.com/matttproud/golang_protobuf_extensions v1.0.4/go.mod h1:BSXmuO+STAnVfr
github.com/mediocregopher/radix/v3 v3.4.2/go.mod h1:8FL3F6UQRXHXIBSPUs5h0RybMF8i4n7wVopoX3x7Bv8=
github.com/microcosm-cc/bluemonday v1.0.2/go.mod h1:iVP4YcDBq+n/5fb23BhYFvIMq/leAFZyRl6bYmGDlGc=
github.com/miekg/dns v1.0.14/go.mod h1:W1PPwlIAgtquWBMBEV9nkV9Cazfe8ScdGz/Lj7v3Nrg=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1 h1:gHSNIaJ9DY/eBziskOH2m+Xp7HE3XxR0gsq3G8LwBlI=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1/go.mod h1:/6UT4zZl6awVeXLeE7UGDWZvXj3IWkRsh3mqsn0DiAs=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1.0.20250716031043-88051c3893ce h1:8cIC7rG5/hJQTsBH61HPK75gTKVlJyw4qW9qAiA9WmQ=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1.0.20250716031043-88051c3893ce/go.mod h1:/6UT4zZl6awVeXLeE7UGDWZvXj3IWkRsh3mqsn0DiAs=
github.com/milvus-io/milvus/pkg/v2 v2.0.0-20250319085209-5a6b4e56d59e h1:VCr43pG4efacDbM4au70fh8/5hNTftoWzm1iEumvDWM=
github.com/milvus-io/milvus/pkg/v2 v2.0.0-20250319085209-5a6b4e56d59e/go.mod h1:37AWzxVs2NS4QUJrkcbeLUwi+4Av0h5mEdjLI62EANU=
github.com/mitchellh/cli v1.0.0/go.mod h1:hNIlj7HEI86fIcpObd7a0FcrxTWetlwJDGcceTlRvqc=

415
client/milvusclient/mock_milvus_server_test.go
View File

@ -1,4 +1,4 @@
// Code generated by mockery v2.40.1. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package milvusclient
@ -86,6 +86,65 @@ func (_c *MilvusServiceServer_AddCollectionField_Call) RunAndReturn(run func(con
return _c
}
// AddUserTags provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) AddUserTags(_a0 context.Context, _a1 *milvuspb.AddUserTagsRequest) (*commonpb.Status, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for AddUserTags")
}
var r0 *commonpb.Status
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AddUserTagsRequest) (*commonpb.Status, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AddUserTagsRequest) *commonpb.Status); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*commonpb.Status)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.AddUserTagsRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MilvusServiceServer_AddUserTags_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AddUserTags'
type MilvusServiceServer_AddUserTags_Call struct {
*mock.Call
}
// AddUserTags is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.AddUserTagsRequest
func (_e *MilvusServiceServer_Expecter) AddUserTags(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_AddUserTags_Call {
return &MilvusServiceServer_AddUserTags_Call{Call: _e.mock.On("AddUserTags", _a0, _a1)}
}
func (_c *MilvusServiceServer_AddUserTags_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.AddUserTagsRequest)) *MilvusServiceServer_AddUserTags_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.AddUserTagsRequest))
})
return _c
}
func (_c *MilvusServiceServer_AddUserTags_Call) Return(_a0 *commonpb.Status, _a1 error) *MilvusServiceServer_AddUserTags_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MilvusServiceServer_AddUserTags_Call) RunAndReturn(run func(context.Context, *milvuspb.AddUserTagsRequest) (*commonpb.Status, error)) *MilvusServiceServer_AddUserTags_Call {
_c.Call.Return(run)
return _c
}
// AllocTimestamp provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) AllocTimestamp(_a0 context.Context, _a1 *milvuspb.AllocTimestampRequest) (*milvuspb.AllocTimestampResponse, error) {
ret := _m.Called(_a0, _a1)
@ -1207,6 +1266,65 @@ func (_c *MilvusServiceServer_CreateRole_Call) RunAndReturn(run func(context.Con
return _c
}
// CreateRowPolicy provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) CreateRowPolicy(_a0 context.Context, _a1 *milvuspb.CreateRowPolicyRequest) (*commonpb.Status, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for CreateRowPolicy")
}
var r0 *commonpb.Status
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRowPolicyRequest) (*commonpb.Status, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRowPolicyRequest) *commonpb.Status); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*commonpb.Status)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateRowPolicyRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MilvusServiceServer_CreateRowPolicy_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CreateRowPolicy'
type MilvusServiceServer_CreateRowPolicy_Call struct {
*mock.Call
}
// CreateRowPolicy is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.CreateRowPolicyRequest
func (_e *MilvusServiceServer_Expecter) CreateRowPolicy(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_CreateRowPolicy_Call {
return &MilvusServiceServer_CreateRowPolicy_Call{Call: _e.mock.On("CreateRowPolicy", _a0, _a1)}
}
func (_c *MilvusServiceServer_CreateRowPolicy_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.CreateRowPolicyRequest)) *MilvusServiceServer_CreateRowPolicy_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.CreateRowPolicyRequest))
})
return _c
}
func (_c *MilvusServiceServer_CreateRowPolicy_Call) Return(_a0 *commonpb.Status, _a1 error) *MilvusServiceServer_CreateRowPolicy_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MilvusServiceServer_CreateRowPolicy_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateRowPolicyRequest) (*commonpb.Status, error)) *MilvusServiceServer_CreateRowPolicy_Call {
_c.Call.Return(run)
return _c
}
// Delete provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) Delete(_a0 context.Context, _a1 *milvuspb.DeleteRequest) (*milvuspb.MutationResult, error) {
ret := _m.Called(_a0, _a1)
@ -1325,6 +1443,65 @@ func (_c *MilvusServiceServer_DeleteCredential_Call) RunAndReturn(run func(conte
return _c
}
// DeleteUserTags provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) DeleteUserTags(_a0 context.Context, _a1 *milvuspb.DeleteUserTagsRequest) (*commonpb.Status, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for DeleteUserTags")
}
var r0 *commonpb.Status
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteUserTagsRequest) (*commonpb.Status, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteUserTagsRequest) *commonpb.Status); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*commonpb.Status)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DeleteUserTagsRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MilvusServiceServer_DeleteUserTags_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DeleteUserTags'
type MilvusServiceServer_DeleteUserTags_Call struct {
*mock.Call
}
// DeleteUserTags is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.DeleteUserTagsRequest
func (_e *MilvusServiceServer_Expecter) DeleteUserTags(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DeleteUserTags_Call {
return &MilvusServiceServer_DeleteUserTags_Call{Call: _e.mock.On("DeleteUserTags", _a0, _a1)}
}
func (_c *MilvusServiceServer_DeleteUserTags_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.DeleteUserTagsRequest)) *MilvusServiceServer_DeleteUserTags_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.DeleteUserTagsRequest))
})
return _c
}
func (_c *MilvusServiceServer_DeleteUserTags_Call) Return(_a0 *commonpb.Status, _a1 error) *MilvusServiceServer_DeleteUserTags_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MilvusServiceServer_DeleteUserTags_Call) RunAndReturn(run func(context.Context, *milvuspb.DeleteUserTagsRequest) (*commonpb.Status, error)) *MilvusServiceServer_DeleteUserTags_Call {
_c.Call.Return(run)
return _c
}
// DescribeAlias provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) DescribeAlias(_a0 context.Context, _a1 *milvuspb.DescribeAliasRequest) (*milvuspb.DescribeAliasResponse, error) {
ret := _m.Called(_a0, _a1)
@ -2151,6 +2328,65 @@ func (_c *MilvusServiceServer_DropRole_Call) RunAndReturn(run func(context.Conte
return _c
}
// DropRowPolicy provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) DropRowPolicy(_a0 context.Context, _a1 *milvuspb.DropRowPolicyRequest) (*commonpb.Status, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for DropRowPolicy")
}
var r0 *commonpb.Status
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRowPolicyRequest) (*commonpb.Status, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRowPolicyRequest) *commonpb.Status); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*commonpb.Status)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropRowPolicyRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MilvusServiceServer_DropRowPolicy_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DropRowPolicy'
type MilvusServiceServer_DropRowPolicy_Call struct {
*mock.Call
}
// DropRowPolicy is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.DropRowPolicyRequest
func (_e *MilvusServiceServer_Expecter) DropRowPolicy(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_DropRowPolicy_Call {
return &MilvusServiceServer_DropRowPolicy_Call{Call: _e.mock.On("DropRowPolicy", _a0, _a1)}
}
func (_c *MilvusServiceServer_DropRowPolicy_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.DropRowPolicyRequest)) *MilvusServiceServer_DropRowPolicy_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.DropRowPolicyRequest))
})
return _c
}
func (_c *MilvusServiceServer_DropRowPolicy_Call) Return(_a0 *commonpb.Status, _a1 error) *MilvusServiceServer_DropRowPolicy_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MilvusServiceServer_DropRowPolicy_Call) RunAndReturn(run func(context.Context, *milvuspb.DropRowPolicyRequest) (*commonpb.Status, error)) *MilvusServiceServer_DropRowPolicy_Call {
_c.Call.Return(run)
return _c
}
// Dummy provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) Dummy(_a0 context.Context, _a1 *milvuspb.DummyRequest) (*milvuspb.DummyResponse, error) {
ret := _m.Called(_a0, _a1)
@ -3331,6 +3567,65 @@ func (_c *MilvusServiceServer_GetReplicas_Call) RunAndReturn(run func(context.Co
return _c
}
// GetUserTags provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) GetUserTags(_a0 context.Context, _a1 *milvuspb.GetUserTagsRequest) (*milvuspb.GetUserTagsResponse, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for GetUserTags")
}
var r0 *milvuspb.GetUserTagsResponse
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetUserTagsRequest) (*milvuspb.GetUserTagsResponse, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetUserTagsRequest) *milvuspb.GetUserTagsResponse); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*milvuspb.GetUserTagsResponse)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetUserTagsRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MilvusServiceServer_GetUserTags_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetUserTags'
type MilvusServiceServer_GetUserTags_Call struct {
*mock.Call
}
// GetUserTags is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.GetUserTagsRequest
func (_e *MilvusServiceServer_Expecter) GetUserTags(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_GetUserTags_Call {
return &MilvusServiceServer_GetUserTags_Call{Call: _e.mock.On("GetUserTags", _a0, _a1)}
}
func (_c *MilvusServiceServer_GetUserTags_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.GetUserTagsRequest)) *MilvusServiceServer_GetUserTags_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.GetUserTagsRequest))
})
return _c
}
func (_c *MilvusServiceServer_GetUserTags_Call) Return(_a0 *milvuspb.GetUserTagsResponse, _a1 error) *MilvusServiceServer_GetUserTags_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MilvusServiceServer_GetUserTags_Call) RunAndReturn(run func(context.Context, *milvuspb.GetUserTagsRequest) (*milvuspb.GetUserTagsResponse, error)) *MilvusServiceServer_GetUserTags_Call {
_c.Call.Return(run)
return _c
}
// GetVersion provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) GetVersion(_a0 context.Context, _a1 *milvuspb.GetVersionRequest) (*milvuspb.GetVersionResponse, error) {
ret := _m.Called(_a0, _a1)
@ -4098,6 +4393,124 @@ func (_c *MilvusServiceServer_ListResourceGroups_Call) RunAndReturn(run func(con
return _c
}
// ListRowPolicies provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) ListRowPolicies(_a0 context.Context, _a1 *milvuspb.ListRowPoliciesRequest) (*milvuspb.ListRowPoliciesResponse, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for ListRowPolicies")
}
var r0 *milvuspb.ListRowPoliciesResponse
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListRowPoliciesRequest) (*milvuspb.ListRowPoliciesResponse, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListRowPoliciesRequest) *milvuspb.ListRowPoliciesResponse); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*milvuspb.ListRowPoliciesResponse)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListRowPoliciesRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MilvusServiceServer_ListRowPolicies_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListRowPolicies'
type MilvusServiceServer_ListRowPolicies_Call struct {
*mock.Call
}
// ListRowPolicies is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.ListRowPoliciesRequest
func (_e *MilvusServiceServer_Expecter) ListRowPolicies(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_ListRowPolicies_Call {
return &MilvusServiceServer_ListRowPolicies_Call{Call: _e.mock.On("ListRowPolicies", _a0, _a1)}
}
func (_c *MilvusServiceServer_ListRowPolicies_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.ListRowPoliciesRequest)) *MilvusServiceServer_ListRowPolicies_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.ListRowPoliciesRequest))
})
return _c
}
func (_c *MilvusServiceServer_ListRowPolicies_Call) Return(_a0 *milvuspb.ListRowPoliciesResponse, _a1 error) *MilvusServiceServer_ListRowPolicies_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MilvusServiceServer_ListRowPolicies_Call) RunAndReturn(run func(context.Context, *milvuspb.ListRowPoliciesRequest) (*milvuspb.ListRowPoliciesResponse, error)) *MilvusServiceServer_ListRowPolicies_Call {
_c.Call.Return(run)
return _c
}
// ListUsersWithTag provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) ListUsersWithTag(_a0 context.Context, _a1 *milvuspb.ListUsersWithTagRequest) (*milvuspb.ListUsersWithTagResponse, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for ListUsersWithTag")
}
var r0 *milvuspb.ListUsersWithTagResponse
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListUsersWithTagRequest) (*milvuspb.ListUsersWithTagResponse, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListUsersWithTagRequest) *milvuspb.ListUsersWithTagResponse); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*milvuspb.ListUsersWithTagResponse)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListUsersWithTagRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MilvusServiceServer_ListUsersWithTag_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListUsersWithTag'
type MilvusServiceServer_ListUsersWithTag_Call struct {
*mock.Call
}
// ListUsersWithTag is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.ListUsersWithTagRequest
func (_e *MilvusServiceServer_Expecter) ListUsersWithTag(_a0 interface{}, _a1 interface{}) *MilvusServiceServer_ListUsersWithTag_Call {
return &MilvusServiceServer_ListUsersWithTag_Call{Call: _e.mock.On("ListUsersWithTag", _a0, _a1)}
}
func (_c *MilvusServiceServer_ListUsersWithTag_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.ListUsersWithTagRequest)) *MilvusServiceServer_ListUsersWithTag_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.ListUsersWithTagRequest))
})
return _c
}
func (_c *MilvusServiceServer_ListUsersWithTag_Call) Return(_a0 *milvuspb.ListUsersWithTagResponse, _a1 error) *MilvusServiceServer_ListUsersWithTag_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MilvusServiceServer_ListUsersWithTag_Call) RunAndReturn(run func(context.Context, *milvuspb.ListUsersWithTagRequest) (*milvuspb.ListUsersWithTagResponse, error)) *MilvusServiceServer_ListUsersWithTag_Call {
_c.Call.Return(run)
return _c
}
// LoadBalance provides a mock function with given fields: _a0, _a1
func (_m *MilvusServiceServer) LoadBalance(_a0 context.Context, _a1 *milvuspb.LoadBalanceRequest) (*commonpb.Status, error) {
ret := _m.Called(_a0, _a1)

2
go.mod
View File

@ -21,7 +21,7 @@ require (
github.com/grpc-ecosystem/go-grpc-middleware v1.3.0
github.com/klauspost/compress v1.17.9
github.com/mgutz/ansi v0.0.0-20200706080929-d51e80ef957d
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1.0.20250716031043-88051c3893ce
github.com/minio/minio-go/v7 v7.0.73
github.com/panjf2000/ants/v2 v2.11.3 // indirect
github.com/pingcap/log v1.1.1-0.20221015072633-39906604fb81

4
go.sum
View File

@ -740,8 +740,8 @@ github.com/milvus-io/cgosymbolizer v0.0.0-20250318084424-114f4050c3a6 h1:YHMFI6L
github.com/milvus-io/cgosymbolizer v0.0.0-20250318084424-114f4050c3a6/go.mod h1:DvXTE/K/RtHehxU8/GtDs4vFtfw64jJ3PaCnFri8CRg=
github.com/milvus-io/gorocksdb v0.0.0-20220624081344-8c5f4212846b h1:TfeY0NxYxZzUfIfYe5qYDBzt4ZYRqzUjTR6CvUzjat8=
github.com/milvus-io/gorocksdb v0.0.0-20220624081344-8c5f4212846b/go.mod h1:iwW+9cWfIzzDseEBCCeDSN5SD16Tidvy8cwQ7ZY8Qj4=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1 h1:gHSNIaJ9DY/eBziskOH2m+Xp7HE3XxR0gsq3G8LwBlI=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1/go.mod h1:/6UT4zZl6awVeXLeE7UGDWZvXj3IWkRsh3mqsn0DiAs=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1.0.20250716031043-88051c3893ce h1:8cIC7rG5/hJQTsBH61HPK75gTKVlJyw4qW9qAiA9WmQ=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1.0.20250716031043-88051c3893ce/go.mod h1:/6UT4zZl6awVeXLeE7UGDWZvXj3IWkRsh3mqsn0DiAs=
github.com/milvus-io/pulsar-client-go v0.12.1 h1:O2JZp1tsYiO7C0MQ4hrUY/aJXnn2Gry6hpm7UodghmE=
github.com/milvus-io/pulsar-client-go v0.12.1/go.mod h1:dkutuH4oS2pXiGm+Ti7fQZ4MRjrMPZ8IJeEGAWMeckk=
github.com/minio/asm2plan9s v0.0.0-20200509001527-cdd76441f9d8 h1:AMFGa4R4MiIpspGNG7Z948v4n35fFGB3RR3G/ry4FWs=

413
internal/mocks/mock_proxy.go
View File

@ -93,6 +93,65 @@ func (_c *MockProxy_AddCollectionField_Call) RunAndReturn(run func(context.Conte
return _c
}
// AddUserTags provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) AddUserTags(_a0 context.Context, _a1 *milvuspb.AddUserTagsRequest) (*commonpb.Status, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for AddUserTags")
}
var r0 *commonpb.Status
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AddUserTagsRequest) (*commonpb.Status, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AddUserTagsRequest) *commonpb.Status); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*commonpb.Status)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.AddUserTagsRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MockProxy_AddUserTags_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AddUserTags'
type MockProxy_AddUserTags_Call struct {
*mock.Call
}
// AddUserTags is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.AddUserTagsRequest
func (_e *MockProxy_Expecter) AddUserTags(_a0 interface{}, _a1 interface{}) *MockProxy_AddUserTags_Call {
return &MockProxy_AddUserTags_Call{Call: _e.mock.On("AddUserTags", _a0, _a1)}
}
func (_c *MockProxy_AddUserTags_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.AddUserTagsRequest)) *MockProxy_AddUserTags_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.AddUserTagsRequest))
})
return _c
}
func (_c *MockProxy_AddUserTags_Call) Return(_a0 *commonpb.Status, _a1 error) *MockProxy_AddUserTags_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MockProxy_AddUserTags_Call) RunAndReturn(run func(context.Context, *milvuspb.AddUserTagsRequest) (*commonpb.Status, error)) *MockProxy_AddUserTags_Call {
_c.Call.Return(run)
return _c
}
// AllocTimestamp provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) AllocTimestamp(_a0 context.Context, _a1 *milvuspb.AllocTimestampRequest) (*milvuspb.AllocTimestampResponse, error) {
ret := _m.Called(_a0, _a1)
@ -1214,6 +1273,65 @@ func (_c *MockProxy_CreateRole_Call) RunAndReturn(run func(context.Context, *mil
return _c
}
// CreateRowPolicy provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) CreateRowPolicy(_a0 context.Context, _a1 *milvuspb.CreateRowPolicyRequest) (*commonpb.Status, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for CreateRowPolicy")
}
var r0 *commonpb.Status
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRowPolicyRequest) (*commonpb.Status, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRowPolicyRequest) *commonpb.Status); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*commonpb.Status)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateRowPolicyRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MockProxy_CreateRowPolicy_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CreateRowPolicy'
type MockProxy_CreateRowPolicy_Call struct {
*mock.Call
}
// CreateRowPolicy is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.CreateRowPolicyRequest
func (_e *MockProxy_Expecter) CreateRowPolicy(_a0 interface{}, _a1 interface{}) *MockProxy_CreateRowPolicy_Call {
return &MockProxy_CreateRowPolicy_Call{Call: _e.mock.On("CreateRowPolicy", _a0, _a1)}
}
func (_c *MockProxy_CreateRowPolicy_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.CreateRowPolicyRequest)) *MockProxy_CreateRowPolicy_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.CreateRowPolicyRequest))
})
return _c
}
func (_c *MockProxy_CreateRowPolicy_Call) Return(_a0 *commonpb.Status, _a1 error) *MockProxy_CreateRowPolicy_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MockProxy_CreateRowPolicy_Call) RunAndReturn(run func(context.Context, *milvuspb.CreateRowPolicyRequest) (*commonpb.Status, error)) *MockProxy_CreateRowPolicy_Call {
_c.Call.Return(run)
return _c
}
// Delete provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) Delete(_a0 context.Context, _a1 *milvuspb.DeleteRequest) (*milvuspb.MutationResult, error) {
ret := _m.Called(_a0, _a1)
@ -1332,6 +1450,65 @@ func (_c *MockProxy_DeleteCredential_Call) RunAndReturn(run func(context.Context
return _c
}
// DeleteUserTags provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) DeleteUserTags(_a0 context.Context, _a1 *milvuspb.DeleteUserTagsRequest) (*commonpb.Status, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for DeleteUserTags")
}
var r0 *commonpb.Status
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteUserTagsRequest) (*commonpb.Status, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteUserTagsRequest) *commonpb.Status); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*commonpb.Status)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DeleteUserTagsRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MockProxy_DeleteUserTags_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DeleteUserTags'
type MockProxy_DeleteUserTags_Call struct {
*mock.Call
}
// DeleteUserTags is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.DeleteUserTagsRequest
func (_e *MockProxy_Expecter) DeleteUserTags(_a0 interface{}, _a1 interface{}) *MockProxy_DeleteUserTags_Call {
return &MockProxy_DeleteUserTags_Call{Call: _e.mock.On("DeleteUserTags", _a0, _a1)}
}
func (_c *MockProxy_DeleteUserTags_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.DeleteUserTagsRequest)) *MockProxy_DeleteUserTags_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.DeleteUserTagsRequest))
})
return _c
}
func (_c *MockProxy_DeleteUserTags_Call) Return(_a0 *commonpb.Status, _a1 error) *MockProxy_DeleteUserTags_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MockProxy_DeleteUserTags_Call) RunAndReturn(run func(context.Context, *milvuspb.DeleteUserTagsRequest) (*commonpb.Status, error)) *MockProxy_DeleteUserTags_Call {
_c.Call.Return(run)
return _c
}
// DescribeAlias provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) DescribeAlias(_a0 context.Context, _a1 *milvuspb.DescribeAliasRequest) (*milvuspb.DescribeAliasResponse, error) {
ret := _m.Called(_a0, _a1)
@ -2158,6 +2335,65 @@ func (_c *MockProxy_DropRole_Call) RunAndReturn(run func(context.Context, *milvu
return _c
}
// DropRowPolicy provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) DropRowPolicy(_a0 context.Context, _a1 *milvuspb.DropRowPolicyRequest) (*commonpb.Status, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for DropRowPolicy")
}
var r0 *commonpb.Status
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRowPolicyRequest) (*commonpb.Status, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRowPolicyRequest) *commonpb.Status); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*commonpb.Status)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropRowPolicyRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MockProxy_DropRowPolicy_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DropRowPolicy'
type MockProxy_DropRowPolicy_Call struct {
*mock.Call
}
// DropRowPolicy is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.DropRowPolicyRequest
func (_e *MockProxy_Expecter) DropRowPolicy(_a0 interface{}, _a1 interface{}) *MockProxy_DropRowPolicy_Call {
return &MockProxy_DropRowPolicy_Call{Call: _e.mock.On("DropRowPolicy", _a0, _a1)}
}
func (_c *MockProxy_DropRowPolicy_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.DropRowPolicyRequest)) *MockProxy_DropRowPolicy_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.DropRowPolicyRequest))
})
return _c
}
func (_c *MockProxy_DropRowPolicy_Call) Return(_a0 *commonpb.Status, _a1 error) *MockProxy_DropRowPolicy_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MockProxy_DropRowPolicy_Call) RunAndReturn(run func(context.Context, *milvuspb.DropRowPolicyRequest) (*commonpb.Status, error)) *MockProxy_DropRowPolicy_Call {
_c.Call.Return(run)
return _c
}
// Dummy provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) Dummy(_a0 context.Context, _a1 *milvuspb.DummyRequest) (*milvuspb.DummyResponse, error) {
ret := _m.Called(_a0, _a1)
@ -3794,6 +4030,65 @@ func (_c *MockProxy_GetStatisticsChannel_Call) RunAndReturn(run func(context.Con
return _c
}
// GetUserTags provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) GetUserTags(_a0 context.Context, _a1 *milvuspb.GetUserTagsRequest) (*milvuspb.GetUserTagsResponse, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for GetUserTags")
}
var r0 *milvuspb.GetUserTagsResponse
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetUserTagsRequest) (*milvuspb.GetUserTagsResponse, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetUserTagsRequest) *milvuspb.GetUserTagsResponse); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*milvuspb.GetUserTagsResponse)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetUserTagsRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MockProxy_GetUserTags_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetUserTags'
type MockProxy_GetUserTags_Call struct {
*mock.Call
}
// GetUserTags is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.GetUserTagsRequest
func (_e *MockProxy_Expecter) GetUserTags(_a0 interface{}, _a1 interface{}) *MockProxy_GetUserTags_Call {
return &MockProxy_GetUserTags_Call{Call: _e.mock.On("GetUserTags", _a0, _a1)}
}
func (_c *MockProxy_GetUserTags_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.GetUserTagsRequest)) *MockProxy_GetUserTags_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.GetUserTagsRequest))
})
return _c
}
func (_c *MockProxy_GetUserTags_Call) Return(_a0 *milvuspb.GetUserTagsResponse, _a1 error) *MockProxy_GetUserTags_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MockProxy_GetUserTags_Call) RunAndReturn(run func(context.Context, *milvuspb.GetUserTagsRequest) (*milvuspb.GetUserTagsResponse, error)) *MockProxy_GetUserTags_Call {
_c.Call.Return(run)
return _c
}
// GetVersion provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) GetVersion(_a0 context.Context, _a1 *milvuspb.GetVersionRequest) (*milvuspb.GetVersionResponse, error) {
ret := _m.Called(_a0, _a1)
@ -4960,6 +5255,124 @@ func (_c *MockProxy_ListResourceGroups_Call) RunAndReturn(run func(context.Conte
return _c
}
// ListRowPolicies provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) ListRowPolicies(_a0 context.Context, _a1 *milvuspb.ListRowPoliciesRequest) (*milvuspb.ListRowPoliciesResponse, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for ListRowPolicies")
}
var r0 *milvuspb.ListRowPoliciesResponse
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListRowPoliciesRequest) (*milvuspb.ListRowPoliciesResponse, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListRowPoliciesRequest) *milvuspb.ListRowPoliciesResponse); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*milvuspb.ListRowPoliciesResponse)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListRowPoliciesRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MockProxy_ListRowPolicies_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListRowPolicies'
type MockProxy_ListRowPolicies_Call struct {
*mock.Call
}
// ListRowPolicies is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.ListRowPoliciesRequest
func (_e *MockProxy_Expecter) ListRowPolicies(_a0 interface{}, _a1 interface{}) *MockProxy_ListRowPolicies_Call {
return &MockProxy_ListRowPolicies_Call{Call: _e.mock.On("ListRowPolicies", _a0, _a1)}
}
func (_c *MockProxy_ListRowPolicies_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.ListRowPoliciesRequest)) *MockProxy_ListRowPolicies_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.ListRowPoliciesRequest))
})
return _c
}
func (_c *MockProxy_ListRowPolicies_Call) Return(_a0 *milvuspb.ListRowPoliciesResponse, _a1 error) *MockProxy_ListRowPolicies_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MockProxy_ListRowPolicies_Call) RunAndReturn(run func(context.Context, *milvuspb.ListRowPoliciesRequest) (*milvuspb.ListRowPoliciesResponse, error)) *MockProxy_ListRowPolicies_Call {
_c.Call.Return(run)
return _c
}
// ListUsersWithTag provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) ListUsersWithTag(_a0 context.Context, _a1 *milvuspb.ListUsersWithTagRequest) (*milvuspb.ListUsersWithTagResponse, error) {
ret := _m.Called(_a0, _a1)
if len(ret) == 0 {
panic("no return value specified for ListUsersWithTag")
}
var r0 *milvuspb.ListUsersWithTagResponse
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListUsersWithTagRequest) (*milvuspb.ListUsersWithTagResponse, error)); ok {
return rf(_a0, _a1)
}
if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListUsersWithTagRequest) *milvuspb.ListUsersWithTagResponse); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*milvuspb.ListUsersWithTagResponse)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListUsersWithTagRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MockProxy_ListUsersWithTag_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListUsersWithTag'
type MockProxy_ListUsersWithTag_Call struct {
*mock.Call
}
// ListUsersWithTag is a helper method to define mock.On call
// - _a0 context.Context
// - _a1 *milvuspb.ListUsersWithTagRequest
func (_e *MockProxy_Expecter) ListUsersWithTag(_a0 interface{}, _a1 interface{}) *MockProxy_ListUsersWithTag_Call {
return &MockProxy_ListUsersWithTag_Call{Call: _e.mock.On("ListUsersWithTag", _a0, _a1)}
}
func (_c *MockProxy_ListUsersWithTag_Call) Run(run func(_a0 context.Context, _a1 *milvuspb.ListUsersWithTagRequest)) *MockProxy_ListUsersWithTag_Call {
_c.Call.Run(func(args mock.Arguments) {
run(args[0].(context.Context), args[1].(*milvuspb.ListUsersWithTagRequest))
})
return _c
}
func (_c *MockProxy_ListUsersWithTag_Call) Return(_a0 *milvuspb.ListUsersWithTagResponse, _a1 error) *MockProxy_ListUsersWithTag_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MockProxy_ListUsersWithTag_Call) RunAndReturn(run func(context.Context, *milvuspb.ListUsersWithTagRequest) (*milvuspb.ListUsersWithTagResponse, error)) *MockProxy_ListUsersWithTag_Call {
_c.Call.Return(run)
return _c
}
// LoadBalance provides a mock function with given fields: _a0, _a1
func (_m *MockProxy) LoadBalance(_a0 context.Context, _a1 *milvuspb.LoadBalanceRequest) (*commonpb.Status, error) {
ret := _m.Called(_a0, _a1)

38
internal/proxy/impl.go
View File

@ -4333,13 +4333,14 @@ func (node *Proxy) GetPersistentSegmentInfo(ctx context.Context, req *milvuspb.G
persistentInfos := make([]*milvuspb.PersistentSegmentInfo, len(infoResp.Infos))
for i, info := range infoResp.Infos {
persistentInfos[i] = &milvuspb.PersistentSegmentInfo{
SegmentID: info.ID,
CollectionID: info.CollectionID,
PartitionID: info.PartitionID,
NumRows: info.NumOfRows,
State: info.State,
Level: commonpb.SegmentLevel(info.Level),
IsSorted: info.GetIsSorted(),
SegmentID: info.ID,
CollectionID: info.CollectionID,
PartitionID: info.PartitionID,
NumRows: info.NumOfRows,
State: info.State,
Level: commonpb.SegmentLevel(info.Level),
IsSorted: info.GetIsSorted(),
StorageVersion: info.GetStorageVersion(),
}
}
metrics.ProxyFunctionCall.WithLabelValues(strconv.FormatInt(paramtable.GetNodeID(), 10), method,
@ -4489,17 +4490,18 @@ func (node *Proxy) GetQuerySegmentInfo(ctx context.Context, req *milvuspb.GetQue
queryInfos := make([]*milvuspb.QuerySegmentInfo, len(infoResp.Infos))
for i, info := range infoResp.Infos {
queryInfos[i] = &milvuspb.QuerySegmentInfo{
SegmentID: info.SegmentID,
CollectionID: info.CollectionID,
PartitionID: info.PartitionID,
NumRows: info.NumRows,
MemSize: info.MemSize,
IndexName: info.IndexName,
IndexID: info.IndexID,
State: info.SegmentState,
NodeIds: info.NodeIds,
Level: commonpb.SegmentLevel(info.Level),
IsSorted: info.GetIsSorted(),
SegmentID: info.SegmentID,
CollectionID: info.CollectionID,
PartitionID: info.PartitionID,
NumRows: info.NumRows,
MemSize: info.MemSize,
IndexName: info.IndexName,
IndexID: info.IndexID,
State: info.SegmentState,
NodeIds: info.NodeIds,
Level: commonpb.SegmentLevel(info.Level),
IsSorted: info.GetIsSorted(),
StorageVersion: info.GetStorageVersion(),
}
}

25
internal/querynodev2/services.go
View File

@ -714,18 +714,19 @@ func (node *QueryNode) GetSegmentInfo(ctx context.Context, in *querypb.GetSegmen
}
info := &querypb.SegmentInfo{
SegmentID: segment.ID(),
SegmentState: segment.Type(),
DmChannel: segment.Shard().VirtualName(),
PartitionID: segment.Partition(),
CollectionID: segment.Collection(),
NodeID: node.GetNodeID(),
NodeIds: []int64{node.GetNodeID()},
MemSize: segment.MemSize(),
NumRows: segment.InsertCount(),
IndexName: indexName,
IndexID: indexID,
IndexInfos: indexInfos,
SegmentID: segment.ID(),
SegmentState: segment.Type(),
DmChannel: segment.Shard().VirtualName(),
PartitionID: segment.Partition(),
CollectionID: segment.Collection(),
NodeID: node.GetNodeID(),
NodeIds: []int64{node.GetNodeID()},
MemSize: segment.MemSize(),
NumRows: segment.InsertCount(),
IndexName: indexName,
IndexID: indexID,
IndexInfos: indexInfos,
StorageVersion: segment.LoadInfo().GetStorageVersion(),
}
segmentInfos = append(segmentInfos, info)
}

2
pkg/go.mod
View File

@ -20,7 +20,7 @@ require (
github.com/jolestar/go-commons-pool/v2 v2.1.2
github.com/json-iterator/go v1.1.12
github.com/klauspost/compress v1.17.9
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1.0.20250716031043-88051c3893ce
github.com/minio/minio-go/v7 v7.0.73
github.com/panjf2000/ants/v2 v2.11.3
github.com/prometheus/client_golang v1.14.0

4
pkg/go.sum
View File

@ -557,8 +557,8 @@ github.com/milvus-io/cgosymbolizer v0.0.0-20250318084424-114f4050c3a6 h1:YHMFI6L
github.com/milvus-io/cgosymbolizer v0.0.0-20250318084424-114f4050c3a6/go.mod h1:DvXTE/K/RtHehxU8/GtDs4vFtfw64jJ3PaCnFri8CRg=
github.com/milvus-io/gorocksdb v0.0.0-20220624081344-8c5f4212846b h1:TfeY0NxYxZzUfIfYe5qYDBzt4ZYRqzUjTR6CvUzjat8=
github.com/milvus-io/gorocksdb v0.0.0-20220624081344-8c5f4212846b/go.mod h1:iwW+9cWfIzzDseEBCCeDSN5SD16Tidvy8cwQ7ZY8Qj4=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1 h1:gHSNIaJ9DY/eBziskOH2m+Xp7HE3XxR0gsq3G8LwBlI=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1/go.mod h1:/6UT4zZl6awVeXLeE7UGDWZvXj3IWkRsh3mqsn0DiAs=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1.0.20250716031043-88051c3893ce h1:8cIC7rG5/hJQTsBH61HPK75gTKVlJyw4qW9qAiA9WmQ=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1.0.20250716031043-88051c3893ce/go.mod h1:/6UT4zZl6awVeXLeE7UGDWZvXj3IWkRsh3mqsn0DiAs=
github.com/milvus-io/pulsar-client-go v0.12.1 h1:O2JZp1tsYiO7C0MQ4hrUY/aJXnn2Gry6hpm7UodghmE=
github.com/milvus-io/pulsar-client-go v0.12.1/go.mod h1:dkutuH4oS2pXiGm+Ti7fQZ4MRjrMPZ8IJeEGAWMeckk=
github.com/minio/md5-simd v1.1.2 h1:Gdi1DZK69+ZVMoNHRXJyNcxrMA4dSxoYHZSQbirFg34=

1
pkg/proto/query_coord.proto
View File

@ -569,6 +569,7 @@ message SegmentInfo {
bool is_fake = 17;
data.SegmentLevel level = 18;
bool is_sorted = 19;
int64 storage_version = 20;
}
message CollectionInfo {

2417
pkg/proto/querypb/query_coord.pb.go
View File

File diff suppressed because it is too large Load Diff

2
tests/go_client/go.mod
View File

@ -51,7 +51,7 @@ require (
github.com/kr/text v0.2.0 // indirect
github.com/lufia/plan9stats v0.0.0-20211012122336-39d0f177ccd0 // indirect
github.com/matttproud/golang_protobuf_extensions v1.0.4 // indirect
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1 // indirect
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1.0.20250716031043-88051c3893ce // indirect
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
github.com/modern-go/reflect2 v1.0.2 // indirect
github.com/opencontainers/runtime-spec v1.0.2 // indirect

4
tests/go_client/go.sum
View File

@ -318,8 +318,8 @@ github.com/matttproud/golang_protobuf_extensions v1.0.4/go.mod h1:BSXmuO+STAnVfr
github.com/mediocregopher/radix/v3 v3.4.2/go.mod h1:8FL3F6UQRXHXIBSPUs5h0RybMF8i4n7wVopoX3x7Bv8=
github.com/microcosm-cc/bluemonday v1.0.2/go.mod h1:iVP4YcDBq+n/5fb23BhYFvIMq/leAFZyRl6bYmGDlGc=
github.com/miekg/dns v1.0.14/go.mod h1:W1PPwlIAgtquWBMBEV9nkV9Cazfe8ScdGz/Lj7v3Nrg=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1 h1:gHSNIaJ9DY/eBziskOH2m+Xp7HE3XxR0gsq3G8LwBlI=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1/go.mod h1:/6UT4zZl6awVeXLeE7UGDWZvXj3IWkRsh3mqsn0DiAs=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1.0.20250716031043-88051c3893ce h1:8cIC7rG5/hJQTsBH61HPK75gTKVlJyw4qW9qAiA9WmQ=
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.0-rc.1.0.20250716031043-88051c3893ce/go.mod h1:/6UT4zZl6awVeXLeE7UGDWZvXj3IWkRsh3mqsn0DiAs=
github.com/milvus-io/milvus/pkg/v2 v2.0.0-20250319085209-5a6b4e56d59e h1:VCr43pG4efacDbM4au70fh8/5hNTftoWzm1iEumvDWM=
github.com/milvus-io/milvus/pkg/v2 v2.0.0-20250319085209-5a6b4e56d59e/go.mod h1:37AWzxVs2NS4QUJrkcbeLUwi+4Av0h5mEdjLI62EANU=
github.com/mitchellh/cli v1.0.0/go.mod h1:hNIlj7HEI86fIcpObd7a0FcrxTWetlwJDGcceTlRvqc=