proxyutil

package
v0.10.3-0...-a99c92e Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Jan 19, 2025 License: Apache-2.0 Imports: 27 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func DefaultProxyCreator 

func DefaultProxyCreator(ctx context.Context, addr string, nodeID int64) (types.ProxyClient, error)

Types

type ExpireCacheConfig 

type ExpireCacheConfig struct {
	// contains filtered or unexported fields
}

func DefaultExpireCacheConfig 

func DefaultExpireCacheConfig() ExpireCacheConfig

func (ExpireCacheConfig) Apply 

func (c ExpireCacheConfig) Apply(req *proxypb.InvalidateCollMetaCacheRequest)

type ExpireCacheOpt 

type ExpireCacheOpt func(c *ExpireCacheConfig)

func SetMsgType 

func SetMsgType(msgType commonpb.MsgType) ExpireCacheOpt

type MockProxyClientManager 

type MockProxyClientManager struct {
	mock.Mock
}

MockProxyClientManager is an autogenerated mock type for the ProxyClientManagerInterface type

func NewMockProxyClientManager 

func NewMockProxyClientManager(t interface {
	mock.TestingT
	Cleanup(func())
}) *MockProxyClientManager

NewMockProxyClientManager creates a new instance of MockProxyClientManager. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. The first argument is typically a *testing.T value.

func (*MockProxyClientManager) AddProxyClient 

func (_m *MockProxyClientManager) AddProxyClient(session *sessionutil.Session)

AddProxyClient provides a mock function with given fields: session

func (*MockProxyClientManager) AddProxyClients 

func (_m *MockProxyClientManager) AddProxyClients(session []*sessionutil.Session)

AddProxyClients provides a mock function with given fields: session

func (*MockProxyClientManager) DelProxyClient 

func (_m *MockProxyClientManager) DelProxyClient(s *sessionutil.Session)

DelProxyClient provides a mock function with given fields: s

func (*MockProxyClientManager) EXPECT 

func (_m *MockProxyClientManager) EXPECT() *MockProxyClientManager_Expecter

func (*MockProxyClientManager) GetComponentStates 

func (_m *MockProxyClientManager) GetComponentStates(ctx context.Context) (map[int64]*milvuspb.ComponentStates, error)

GetComponentStates provides a mock function with given fields: ctx

func (*MockProxyClientManager) GetProxyClients 

func (_m *MockProxyClientManager) GetProxyClients() *typeutil.ConcurrentMap[int64, types.ProxyClient]

GetProxyClients provides a mock function with given fields:

func (*MockProxyClientManager) GetProxyCount 

func (_m *MockProxyClientManager) GetProxyCount() int

GetProxyCount provides a mock function with given fields:

func (*MockProxyClientManager) GetProxyMetrics 

func (_m *MockProxyClientManager) GetProxyMetrics(ctx context.Context) ([]*milvuspb.GetMetricsResponse, error)

GetProxyMetrics provides a mock function with given fields: ctx

func (*MockProxyClientManager) InvalidateCollectionMetaCache 

func (_m *MockProxyClientManager) InvalidateCollectionMetaCache(ctx context.Context, request *proxypb.InvalidateCollMetaCacheRequest, opts ...ExpireCacheOpt) error

InvalidateCollectionMetaCache provides a mock function with given fields: ctx, request, opts

func (*MockProxyClientManager) InvalidateCredentialCache 

func (_m *MockProxyClientManager) InvalidateCredentialCache(ctx context.Context, request *proxypb.InvalidateCredCacheRequest) error

InvalidateCredentialCache provides a mock function with given fields: ctx, request

func (*MockProxyClientManager) InvalidateShardLeaderCache 

func (_m *MockProxyClientManager) InvalidateShardLeaderCache(ctx context.Context, request *proxypb.InvalidateShardLeaderCacheRequest) error

InvalidateShardLeaderCache provides a mock function with given fields: ctx, request

func (*MockProxyClientManager) RefreshPolicyInfoCache 

func (_m *MockProxyClientManager) RefreshPolicyInfoCache(ctx context.Context, req *proxypb.RefreshPolicyInfoCacheRequest) error

RefreshPolicyInfoCache provides a mock function with given fields: ctx, req

func (*MockProxyClientManager) SetRates 

func (_m *MockProxyClientManager) SetRates(ctx context.Context, request *proxypb.SetRatesRequest) error

SetRates provides a mock function with given fields: ctx, request

func (*MockProxyClientManager) UpdateCredentialCache 

func (_m *MockProxyClientManager) UpdateCredentialCache(ctx context.Context, request *proxypb.UpdateCredCacheRequest) error

UpdateCredentialCache provides a mock function with given fields: ctx, request

type MockProxyClientManager_AddProxyClient_Call 

type MockProxyClientManager_AddProxyClient_Call struct {
	*mock.Call
}

MockProxyClientManager_AddProxyClient_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AddProxyClient'

func (*MockProxyClientManager_AddProxyClient_Call) Return 

func (_c *MockProxyClientManager_AddProxyClient_Call) Return() *MockProxyClientManager_AddProxyClient_Call

func (*MockProxyClientManager_AddProxyClient_Call) Run 

func (_c *MockProxyClientManager_AddProxyClient_Call) Run(run func(session *sessionutil.Session)) *MockProxyClientManager_AddProxyClient_Call

func (*MockProxyClientManager_AddProxyClient_Call) RunAndReturn 

func (_c *MockProxyClientManager_AddProxyClient_Call) RunAndReturn(run func(*sessionutil.Session)) *MockProxyClientManager_AddProxyClient_Call

type MockProxyClientManager_AddProxyClients_Call 

type MockProxyClientManager_AddProxyClients_Call struct {
	*mock.Call
}

MockProxyClientManager_AddProxyClients_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AddProxyClients'

func (*MockProxyClientManager_AddProxyClients_Call) Return 

func (_c *MockProxyClientManager_AddProxyClients_Call) Return() *MockProxyClientManager_AddProxyClients_Call

func (*MockProxyClientManager_AddProxyClients_Call) Run 

func (_c *MockProxyClientManager_AddProxyClients_Call) Run(run func(session []*sessionutil.Session)) *MockProxyClientManager_AddProxyClients_Call

func (*MockProxyClientManager_AddProxyClients_Call) RunAndReturn 

func (_c *MockProxyClientManager_AddProxyClients_Call) RunAndReturn(run func([]*sessionutil.Session)) *MockProxyClientManager_AddProxyClients_Call

type MockProxyClientManager_DelProxyClient_Call 

type MockProxyClientManager_DelProxyClient_Call struct {
	*mock.Call
}

MockProxyClientManager_DelProxyClient_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DelProxyClient'

func (*MockProxyClientManager_DelProxyClient_Call) Return 

func (_c *MockProxyClientManager_DelProxyClient_Call) Return() *MockProxyClientManager_DelProxyClient_Call

func (*MockProxyClientManager_DelProxyClient_Call) Run 

func (_c *MockProxyClientManager_DelProxyClient_Call) Run(run func(s *sessionutil.Session)) *MockProxyClientManager_DelProxyClient_Call

func (*MockProxyClientManager_DelProxyClient_Call) RunAndReturn 

func (_c *MockProxyClientManager_DelProxyClient_Call) RunAndReturn(run func(*sessionutil.Session)) *MockProxyClientManager_DelProxyClient_Call

type MockProxyClientManager_Expecter 

type MockProxyClientManager_Expecter struct {
	// contains filtered or unexported fields
}

func (*MockProxyClientManager_Expecter) AddProxyClient 

func (_e *MockProxyClientManager_Expecter) AddProxyClient(session interface{}) *MockProxyClientManager_AddProxyClient_Call

AddProxyClient is a helper method to define mock.On call

  • session *sessionutil.Session

func (*MockProxyClientManager_Expecter) AddProxyClients 

func (_e *MockProxyClientManager_Expecter) AddProxyClients(session interface{}) *MockProxyClientManager_AddProxyClients_Call

AddProxyClients is a helper method to define mock.On call

  • session []*sessionutil.Session

func (*MockProxyClientManager_Expecter) DelProxyClient 

func (_e *MockProxyClientManager_Expecter) DelProxyClient(s interface{}) *MockProxyClientManager_DelProxyClient_Call

DelProxyClient is a helper method to define mock.On call

  • s *sessionutil.Session

func (*MockProxyClientManager_Expecter) GetComponentStates 

func (_e *MockProxyClientManager_Expecter) GetComponentStates(ctx interface{}) *MockProxyClientManager_GetComponentStates_Call

GetComponentStates is a helper method to define mock.On call

  • ctx context.Context

func (*MockProxyClientManager_Expecter) GetProxyClients 

func (_e *MockProxyClientManager_Expecter) GetProxyClients() *MockProxyClientManager_GetProxyClients_Call

GetProxyClients is a helper method to define mock.On call

func (*MockProxyClientManager_Expecter) GetProxyCount 

func (_e *MockProxyClientManager_Expecter) GetProxyCount() *MockProxyClientManager_GetProxyCount_Call

GetProxyCount is a helper method to define mock.On call

func (*MockProxyClientManager_Expecter) GetProxyMetrics 

func (_e *MockProxyClientManager_Expecter) GetProxyMetrics(ctx interface{}) *MockProxyClientManager_GetProxyMetrics_Call

GetProxyMetrics is a helper method to define mock.On call

  • ctx context.Context

func (*MockProxyClientManager_Expecter) InvalidateCollectionMetaCache 

func (_e *MockProxyClientManager_Expecter) InvalidateCollectionMetaCache(ctx interface{}, request interface{}, opts ...interface{}) *MockProxyClientManager_InvalidateCollectionMetaCache_Call

InvalidateCollectionMetaCache is a helper method to define mock.On call

  • ctx context.Context
  • request *proxypb.InvalidateCollMetaCacheRequest
  • opts ...ExpireCacheOpt

func (*MockProxyClientManager_Expecter) InvalidateCredentialCache 

func (_e *MockProxyClientManager_Expecter) InvalidateCredentialCache(ctx interface{}, request interface{}) *MockProxyClientManager_InvalidateCredentialCache_Call

InvalidateCredentialCache is a helper method to define mock.On call

  • ctx context.Context
  • request *proxypb.InvalidateCredCacheRequest

func (*MockProxyClientManager_Expecter) InvalidateShardLeaderCache 

func (_e *MockProxyClientManager_Expecter) InvalidateShardLeaderCache(ctx interface{}, request interface{}) *MockProxyClientManager_InvalidateShardLeaderCache_Call

InvalidateShardLeaderCache is a helper method to define mock.On call

  • ctx context.Context
  • request *proxypb.InvalidateShardLeaderCacheRequest

func (*MockProxyClientManager_Expecter) RefreshPolicyInfoCache 

func (_e *MockProxyClientManager_Expecter) RefreshPolicyInfoCache(ctx interface{}, req interface{}) *MockProxyClientManager_RefreshPolicyInfoCache_Call

RefreshPolicyInfoCache is a helper method to define mock.On call

  • ctx context.Context
  • req *proxypb.RefreshPolicyInfoCacheRequest

func (*MockProxyClientManager_Expecter) SetRates 

func (_e *MockProxyClientManager_Expecter) SetRates(ctx interface{}, request interface{}) *MockProxyClientManager_SetRates_Call

SetRates is a helper method to define mock.On call

  • ctx context.Context
  • request *proxypb.SetRatesRequest

func (*MockProxyClientManager_Expecter) UpdateCredentialCache 

func (_e *MockProxyClientManager_Expecter) UpdateCredentialCache(ctx interface{}, request interface{}) *MockProxyClientManager_UpdateCredentialCache_Call

UpdateCredentialCache is a helper method to define mock.On call

  • ctx context.Context
  • request *proxypb.UpdateCredCacheRequest

type MockProxyClientManager_GetComponentStates_Call 

type MockProxyClientManager_GetComponentStates_Call struct {
	*mock.Call
}

MockProxyClientManager_GetComponentStates_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetComponentStates'

func (*MockProxyClientManager_GetComponentStates_Call) Return 

func (_c *MockProxyClientManager_GetComponentStates_Call) Return(_a0 map[int64]*milvuspb.ComponentStates, _a1 error) *MockProxyClientManager_GetComponentStates_Call

func (*MockProxyClientManager_GetComponentStates_Call) Run 

func (_c *MockProxyClientManager_GetComponentStates_Call) Run(run func(ctx context.Context)) *MockProxyClientManager_GetComponentStates_Call

func (*MockProxyClientManager_GetComponentStates_Call) RunAndReturn 

func (_c *MockProxyClientManager_GetComponentStates_Call) RunAndReturn(run func(context.Context) (map[int64]*milvuspb.ComponentStates, error)) *MockProxyClientManager_GetComponentStates_Call

type MockProxyClientManager_GetProxyClients_Call 

type MockProxyClientManager_GetProxyClients_Call struct {
	*mock.Call
}

MockProxyClientManager_GetProxyClients_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetProxyClients'

func (*MockProxyClientManager_GetProxyClients_Call) Return 

func (_c *MockProxyClientManager_GetProxyClients_Call) Return(_a0 *typeutil.ConcurrentMap[int64, types.ProxyClient]) *MockProxyClientManager_GetProxyClients_Call

func (*MockProxyClientManager_GetProxyClients_Call) Run 

func (_c *MockProxyClientManager_GetProxyClients_Call) Run(run func()) *MockProxyClientManager_GetProxyClients_Call

func (*MockProxyClientManager_GetProxyClients_Call) RunAndReturn 

func (_c *MockProxyClientManager_GetProxyClients_Call) RunAndReturn(run func() *typeutil.ConcurrentMap[int64, types.ProxyClient]) *MockProxyClientManager_GetProxyClients_Call

type MockProxyClientManager_GetProxyCount_Call 

type MockProxyClientManager_GetProxyCount_Call struct {
	*mock.Call
}

MockProxyClientManager_GetProxyCount_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetProxyCount'

func (*MockProxyClientManager_GetProxyCount_Call) Return 

func (_c *MockProxyClientManager_GetProxyCount_Call) Return(_a0 int) *MockProxyClientManager_GetProxyCount_Call

func (*MockProxyClientManager_GetProxyCount_Call) Run 

func (_c *MockProxyClientManager_GetProxyCount_Call) Run(run func()) *MockProxyClientManager_GetProxyCount_Call

func (*MockProxyClientManager_GetProxyCount_Call) RunAndReturn 

func (_c *MockProxyClientManager_GetProxyCount_Call) RunAndReturn(run func() int) *MockProxyClientManager_GetProxyCount_Call

type MockProxyClientManager_GetProxyMetrics_Call 

type MockProxyClientManager_GetProxyMetrics_Call struct {
	*mock.Call
}

MockProxyClientManager_GetProxyMetrics_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetProxyMetrics'

func (*MockProxyClientManager_GetProxyMetrics_Call) Return 

func (_c *MockProxyClientManager_GetProxyMetrics_Call) Return(_a0 []*milvuspb.GetMetricsResponse, _a1 error) *MockProxyClientManager_GetProxyMetrics_Call

func (*MockProxyClientManager_GetProxyMetrics_Call) Run 

func (_c *MockProxyClientManager_GetProxyMetrics_Call) Run(run func(ctx context.Context)) *MockProxyClientManager_GetProxyMetrics_Call

func (*MockProxyClientManager_GetProxyMetrics_Call) RunAndReturn 

func (_c *MockProxyClientManager_GetProxyMetrics_Call) RunAndReturn(run func(context.Context) ([]*milvuspb.GetMetricsResponse, error)) *MockProxyClientManager_GetProxyMetrics_Call

type MockProxyClientManager_InvalidateCollectionMetaCache_Call 

type MockProxyClientManager_InvalidateCollectionMetaCache_Call struct {
	*mock.Call
}

MockProxyClientManager_InvalidateCollectionMetaCache_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'InvalidateCollectionMetaCache'

func (*MockProxyClientManager_InvalidateCollectionMetaCache_Call) Return 

func (_c *MockProxyClientManager_InvalidateCollectionMetaCache_Call) Return(_a0 error) *MockProxyClientManager_InvalidateCollectionMetaCache_Call

func (*MockProxyClientManager_InvalidateCollectionMetaCache_Call) Run 

func (_c *MockProxyClientManager_InvalidateCollectionMetaCache_Call) Run(run func(ctx context.Context, request *proxypb.InvalidateCollMetaCacheRequest, opts ...ExpireCacheOpt)) *MockProxyClientManager_InvalidateCollectionMetaCache_Call

func (*MockProxyClientManager_InvalidateCollectionMetaCache_Call) RunAndReturn 

func (_c *MockProxyClientManager_InvalidateCollectionMetaCache_Call) RunAndReturn(run func(context.Context, *proxypb.InvalidateCollMetaCacheRequest, ...ExpireCacheOpt) error) *MockProxyClientManager_InvalidateCollectionMetaCache_Call

type MockProxyClientManager_InvalidateCredentialCache_Call 

type MockProxyClientManager_InvalidateCredentialCache_Call struct {
	*mock.Call
}

MockProxyClientManager_InvalidateCredentialCache_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'InvalidateCredentialCache'

func (*MockProxyClientManager_InvalidateCredentialCache_Call) Return 

func (_c *MockProxyClientManager_InvalidateCredentialCache_Call) Return(_a0 error) *MockProxyClientManager_InvalidateCredentialCache_Call

func (*MockProxyClientManager_InvalidateCredentialCache_Call) Run 

func (_c *MockProxyClientManager_InvalidateCredentialCache_Call) Run(run func(ctx context.Context, request *proxypb.InvalidateCredCacheRequest)) *MockProxyClientManager_InvalidateCredentialCache_Call

func (*MockProxyClientManager_InvalidateCredentialCache_Call) RunAndReturn 

func (_c *MockProxyClientManager_InvalidateCredentialCache_Call) RunAndReturn(run func(context.Context, *proxypb.InvalidateCredCacheRequest) error) *MockProxyClientManager_InvalidateCredentialCache_Call

type MockProxyClientManager_InvalidateShardLeaderCache_Call 

type MockProxyClientManager_InvalidateShardLeaderCache_Call struct {
	*mock.Call
}

MockProxyClientManager_InvalidateShardLeaderCache_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'InvalidateShardLeaderCache'

func (*MockProxyClientManager_InvalidateShardLeaderCache_Call) Return 

func (_c *MockProxyClientManager_InvalidateShardLeaderCache_Call) Return(_a0 error) *MockProxyClientManager_InvalidateShardLeaderCache_Call

func (*MockProxyClientManager_InvalidateShardLeaderCache_Call) Run 

func (_c *MockProxyClientManager_InvalidateShardLeaderCache_Call) Run(run func(ctx context.Context, request *proxypb.InvalidateShardLeaderCacheRequest)) *MockProxyClientManager_InvalidateShardLeaderCache_Call

func (*MockProxyClientManager_InvalidateShardLeaderCache_Call) RunAndReturn 

func (_c *MockProxyClientManager_InvalidateShardLeaderCache_Call) RunAndReturn(run func(context.Context, *proxypb.InvalidateShardLeaderCacheRequest) error) *MockProxyClientManager_InvalidateShardLeaderCache_Call

type MockProxyClientManager_RefreshPolicyInfoCache_Call 

type MockProxyClientManager_RefreshPolicyInfoCache_Call struct {
	*mock.Call
}

MockProxyClientManager_RefreshPolicyInfoCache_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RefreshPolicyInfoCache'

func (*MockProxyClientManager_RefreshPolicyInfoCache_Call) Return 

func (_c *MockProxyClientManager_RefreshPolicyInfoCache_Call) Return(_a0 error) *MockProxyClientManager_RefreshPolicyInfoCache_Call

func (*MockProxyClientManager_RefreshPolicyInfoCache_Call) Run 

func (_c *MockProxyClientManager_RefreshPolicyInfoCache_Call) Run(run func(ctx context.Context, req *proxypb.RefreshPolicyInfoCacheRequest)) *MockProxyClientManager_RefreshPolicyInfoCache_Call

func (*MockProxyClientManager_RefreshPolicyInfoCache_Call) RunAndReturn 

func (_c *MockProxyClientManager_RefreshPolicyInfoCache_Call) RunAndReturn(run func(context.Context, *proxypb.RefreshPolicyInfoCacheRequest) error) *MockProxyClientManager_RefreshPolicyInfoCache_Call

type MockProxyClientManager_SetRates_Call 

type MockProxyClientManager_SetRates_Call struct {
	*mock.Call
}

MockProxyClientManager_SetRates_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetRates'

func (*MockProxyClientManager_SetRates_Call) Return 

func (_c *MockProxyClientManager_SetRates_Call) Return(_a0 error) *MockProxyClientManager_SetRates_Call

func (*MockProxyClientManager_SetRates_Call) Run 

func (_c *MockProxyClientManager_SetRates_Call) Run(run func(ctx context.Context, request *proxypb.SetRatesRequest)) *MockProxyClientManager_SetRates_Call

func (*MockProxyClientManager_SetRates_Call) RunAndReturn 

func (_c *MockProxyClientManager_SetRates_Call) RunAndReturn(run func(context.Context, *proxypb.SetRatesRequest) error) *MockProxyClientManager_SetRates_Call

type MockProxyClientManager_UpdateCredentialCache_Call 

type MockProxyClientManager_UpdateCredentialCache_Call struct {
	*mock.Call
}

MockProxyClientManager_UpdateCredentialCache_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'UpdateCredentialCache'

func (*MockProxyClientManager_UpdateCredentialCache_Call) Return 

func (_c *MockProxyClientManager_UpdateCredentialCache_Call) Return(_a0 error) *MockProxyClientManager_UpdateCredentialCache_Call

func (*MockProxyClientManager_UpdateCredentialCache_Call) Run 

func (_c *MockProxyClientManager_UpdateCredentialCache_Call) Run(run func(ctx context.Context, request *proxypb.UpdateCredCacheRequest)) *MockProxyClientManager_UpdateCredentialCache_Call

func (*MockProxyClientManager_UpdateCredentialCache_Call) RunAndReturn 

func (_c *MockProxyClientManager_UpdateCredentialCache_Call) RunAndReturn(run func(context.Context, *proxypb.UpdateCredCacheRequest) error) *MockProxyClientManager_UpdateCredentialCache_Call

type MockProxyWatcher 

type MockProxyWatcher struct {
	mock.Mock
}

MockProxyWatcher is an autogenerated mock type for the ProxyWatcherInterface type

func NewMockProxyWatcher 

func NewMockProxyWatcher(t interface {
	mock.TestingT
	Cleanup(func())
}) *MockProxyWatcher

NewMockProxyWatcher creates a new instance of MockProxyWatcher. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. The first argument is typically a *testing.T value.

func (*MockProxyWatcher) AddSessionFunc 

func (_m *MockProxyWatcher) AddSessionFunc(fns ...func(*sessionutil.Session))

AddSessionFunc provides a mock function with given fields: fns

func (*MockProxyWatcher) DelSessionFunc 

func (_m *MockProxyWatcher) DelSessionFunc(fns ...func(*sessionutil.Session))

DelSessionFunc provides a mock function with given fields: fns

func (*MockProxyWatcher) EXPECT 

func (_m *MockProxyWatcher) EXPECT() *MockProxyWatcher_Expecter

func (*MockProxyWatcher) Stop 

func (_m *MockProxyWatcher) Stop()

Stop provides a mock function with given fields:

func (*MockProxyWatcher) WatchProxy 

func (_m *MockProxyWatcher) WatchProxy(ctx context.Context) error

WatchProxy provides a mock function with given fields: ctx

type MockProxyWatcher_AddSessionFunc_Call 

type MockProxyWatcher_AddSessionFunc_Call struct {
	*mock.Call
}

MockProxyWatcher_AddSessionFunc_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AddSessionFunc'

func (*MockProxyWatcher_AddSessionFunc_Call) Return 

func (_c *MockProxyWatcher_AddSessionFunc_Call) Return() *MockProxyWatcher_AddSessionFunc_Call

func (*MockProxyWatcher_AddSessionFunc_Call) Run 

func (_c *MockProxyWatcher_AddSessionFunc_Call) Run(run func(fns ...func(*sessionutil.Session))) *MockProxyWatcher_AddSessionFunc_Call

func (*MockProxyWatcher_AddSessionFunc_Call) RunAndReturn 

func (_c *MockProxyWatcher_AddSessionFunc_Call) RunAndReturn(run func(...func(*sessionutil.Session))) *MockProxyWatcher_AddSessionFunc_Call

type MockProxyWatcher_DelSessionFunc_Call 

type MockProxyWatcher_DelSessionFunc_Call struct {
	*mock.Call
}

MockProxyWatcher_DelSessionFunc_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DelSessionFunc'

func (*MockProxyWatcher_DelSessionFunc_Call) Return 

func (_c *MockProxyWatcher_DelSessionFunc_Call) Return() *MockProxyWatcher_DelSessionFunc_Call

func (*MockProxyWatcher_DelSessionFunc_Call) Run 

func (_c *MockProxyWatcher_DelSessionFunc_Call) Run(run func(fns ...func(*sessionutil.Session))) *MockProxyWatcher_DelSessionFunc_Call

func (*MockProxyWatcher_DelSessionFunc_Call) RunAndReturn 

func (_c *MockProxyWatcher_DelSessionFunc_Call) RunAndReturn(run func(...func(*sessionutil.Session))) *MockProxyWatcher_DelSessionFunc_Call

type MockProxyWatcher_Expecter 

type MockProxyWatcher_Expecter struct {
	// contains filtered or unexported fields
}

func (*MockProxyWatcher_Expecter) AddSessionFunc 

func (_e *MockProxyWatcher_Expecter) AddSessionFunc(fns ...interface{}) *MockProxyWatcher_AddSessionFunc_Call

AddSessionFunc is a helper method to define mock.On call

  • fns ...func(*sessionutil.Session)

func (*MockProxyWatcher_Expecter) DelSessionFunc 

func (_e *MockProxyWatcher_Expecter) DelSessionFunc(fns ...interface{}) *MockProxyWatcher_DelSessionFunc_Call

DelSessionFunc is a helper method to define mock.On call

  • fns ...func(*sessionutil.Session)

func (*MockProxyWatcher_Expecter) Stop 

func (_e *MockProxyWatcher_Expecter) Stop() *MockProxyWatcher_Stop_Call

Stop is a helper method to define mock.On call

func (*MockProxyWatcher_Expecter) WatchProxy 

func (_e *MockProxyWatcher_Expecter) WatchProxy(ctx interface{}) *MockProxyWatcher_WatchProxy_Call

WatchProxy is a helper method to define mock.On call

  • ctx context.Context

type MockProxyWatcher_Stop_Call 

type MockProxyWatcher_Stop_Call struct {
	*mock.Call
}

MockProxyWatcher_Stop_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Stop'

func (*MockProxyWatcher_Stop_Call) Return 

func (_c *MockProxyWatcher_Stop_Call) Return() *MockProxyWatcher_Stop_Call

func (*MockProxyWatcher_Stop_Call) Run 

func (_c *MockProxyWatcher_Stop_Call) Run(run func()) *MockProxyWatcher_Stop_Call

func (*MockProxyWatcher_Stop_Call) RunAndReturn 

func (_c *MockProxyWatcher_Stop_Call) RunAndReturn(run func()) *MockProxyWatcher_Stop_Call

type MockProxyWatcher_WatchProxy_Call 

type MockProxyWatcher_WatchProxy_Call struct {
	*mock.Call
}

MockProxyWatcher_WatchProxy_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'WatchProxy'

func (*MockProxyWatcher_WatchProxy_Call) Return 

func (_c *MockProxyWatcher_WatchProxy_Call) Return(_a0 error) *MockProxyWatcher_WatchProxy_Call

func (*MockProxyWatcher_WatchProxy_Call) Run 

func (_c *MockProxyWatcher_WatchProxy_Call) Run(run func(ctx context.Context)) *MockProxyWatcher_WatchProxy_Call

func (*MockProxyWatcher_WatchProxy_Call) RunAndReturn 

func (_c *MockProxyWatcher_WatchProxy_Call) RunAndReturn(run func(context.Context) error) *MockProxyWatcher_WatchProxy_Call

type ProxyClientManager 

type ProxyClientManager struct {
	// contains filtered or unexported fields
}

func NewProxyClientManager 

func NewProxyClientManager(creator ProxyCreator) *ProxyClientManager

func (*ProxyClientManager) AddProxyClient 

func (p *ProxyClientManager) AddProxyClient(session *sessionutil.Session)

func (*ProxyClientManager) AddProxyClients 

func (p *ProxyClientManager) AddProxyClients(sessions []*sessionutil.Session)

func (*ProxyClientManager) DelProxyClient 

func (p *ProxyClientManager) DelProxyClient(s *sessionutil.Session)

func (*ProxyClientManager) GetComponentStates 

func (p *ProxyClientManager) GetComponentStates(ctx context.Context) (map[int64]*milvuspb.ComponentStates, error)

func (*ProxyClientManager) GetProxyClients 

func (p *ProxyClientManager) GetProxyClients() *typeutil.ConcurrentMap[int64, types.ProxyClient]

func (*ProxyClientManager) GetProxyCount 

func (p *ProxyClientManager) GetProxyCount() int

GetProxyCount returns number of proxy clients.

func (*ProxyClientManager) GetProxyMetrics 

func (p *ProxyClientManager) GetProxyMetrics(ctx context.Context) ([]*milvuspb.GetMetricsResponse, error)

GetProxyMetrics sends requests to proxies to get metrics.

func (*ProxyClientManager) InvalidateCollectionMetaCache 

func (p *ProxyClientManager) InvalidateCollectionMetaCache(ctx context.Context, request *proxypb.InvalidateCollMetaCacheRequest, opts ...ExpireCacheOpt) error

func (*ProxyClientManager) InvalidateCredentialCache 

func (p *ProxyClientManager) InvalidateCredentialCache(ctx context.Context, request *proxypb.InvalidateCredCacheRequest) error

InvalidateCredentialCache TODO: too many codes similar to InvalidateCollectionMetaCache.

func (*ProxyClientManager) InvalidateShardLeaderCache 

func (p *ProxyClientManager) InvalidateShardLeaderCache(ctx context.Context, request *proxypb.InvalidateShardLeaderCacheRequest) error

func (*ProxyClientManager) RefreshPolicyInfoCache 

func (p *ProxyClientManager) RefreshPolicyInfoCache(ctx context.Context, req *proxypb.RefreshPolicyInfoCacheRequest) error

RefreshPolicyInfoCache TODO: too many codes similar to InvalidateCollectionMetaCache.

func (*ProxyClientManager) SetRates 

func (p *ProxyClientManager) SetRates(ctx context.Context, request *proxypb.SetRatesRequest) error

SetRates notifies Proxy to limit rates of requests.

func (*ProxyClientManager) UpdateCredentialCache 

func (p *ProxyClientManager) UpdateCredentialCache(ctx context.Context, request *proxypb.UpdateCredCacheRequest) error

UpdateCredentialCache TODO: too many codes similar to InvalidateCollectionMetaCache.

type ProxyClientManagerHelper 

type ProxyClientManagerHelper struct {
	// contains filtered or unexported fields
}

type ProxyClientManagerInterface 

type ProxyClientManagerInterface interface {
	AddProxyClient(session *sessionutil.Session)
	AddProxyClients(session []*sessionutil.Session)
	GetProxyClients() *typeutil.ConcurrentMap[int64, types.ProxyClient]
	DelProxyClient(s *sessionutil.Session)
	GetProxyCount() int

	InvalidateCollectionMetaCache(ctx context.Context, request *proxypb.InvalidateCollMetaCacheRequest, opts ...ExpireCacheOpt) error
	InvalidateShardLeaderCache(ctx context.Context, request *proxypb.InvalidateShardLeaderCacheRequest) error
	InvalidateCredentialCache(ctx context.Context, request *proxypb.InvalidateCredCacheRequest) error
	UpdateCredentialCache(ctx context.Context, request *proxypb.UpdateCredCacheRequest) error
	RefreshPolicyInfoCache(ctx context.Context, req *proxypb.RefreshPolicyInfoCacheRequest) error
	GetProxyMetrics(ctx context.Context) ([]*milvuspb.GetMetricsResponse, error)
	SetRates(ctx context.Context, request *proxypb.SetRatesRequest) error
	GetComponentStates(ctx context.Context) (map[int64]*milvuspb.ComponentStates, error)
}

type ProxyCreator 

type ProxyCreator func(ctx context.Context, addr string, nodeID int64) (types.ProxyClient, error)

type ProxyWatcher 

type ProxyWatcher struct {
	// contains filtered or unexported fields
}

ProxyWatcher manages proxy clients

func NewProxyWatcher 

func NewProxyWatcher(client *clientv3.Client, fns ...func([]*sessionutil.Session)) *ProxyWatcher

NewProxyWatcher helper function to create a proxyWatcher fns are the custom getSessions function list

func (*ProxyWatcher) AddSessionFunc 

func (p *ProxyWatcher) AddSessionFunc(fns ...func(*sessionutil.Session))

AddSessionFunc adds functions to addSessions function list

func (*ProxyWatcher) DelSessionFunc 

func (p *ProxyWatcher) DelSessionFunc(fns ...func(*sessionutil.Session))

DelSessionFunc add functions to delSessions function list

func (*ProxyWatcher) Stop 

func (p *ProxyWatcher) Stop()

Stop stops the ProxyManager

func (*ProxyWatcher) WatchProxy 

func (p *ProxyWatcher) WatchProxy(ctx context.Context) error

WatchProxy starts a goroutine to watch proxy session changes on etcd

type ProxyWatcherInterface 

type ProxyWatcherInterface interface {
	AddSessionFunc(fns ...func(*sessionutil.Session))
	DelSessionFunc(fns ...func(*sessionutil.Session))

	WatchProxy(ctx context.Context) error
	Stop()
}

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.