segments

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

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

 Go to latest Published: Sep 30, 2025 License: Apache-2.0 Imports: 69 Imported by: 0

Documentation

Index

Constants

View Source 
const (
	SegmentTypeGrowing = commonpb.SegmentState_Growing
	SegmentTypeSealed  = commonpb.SegmentState_Sealed
)
View Source 
const (
	UsedDiskMemoryRatio = 4
)

Variables

View Source 
var ErrSegmentUnhealthy = errors.New("segment unhealthy")

Functions

func CreateInternalReducer 

func CreateInternalReducer(req *querypb.QueryRequest, schema *schemapb.CollectionSchema) internalReducer

func CreateSegCoreReducer 

func CreateSegCoreReducer(req *querypb.QueryRequest, schema *schemapb.CollectionSchema, manager *Manager) segCoreReducer

func DecodeSearchResults 

func DecodeSearchResults(ctx context.Context, searchResults []*internalpb.SearchResults) ([]*schemapb.SearchResultData, error)

func DeleteCollection 

func DeleteCollection(collection *Collection)

deleteCollection delete collection and free the collection memory

func DeleteSearchResults 

func DeleteSearchResults(results []*SearchResult)

func DoubleMemoryDataType 

func DoubleMemoryDataType(dataType schemapb.DataType) bool

func DoubleMemorySystemField 

func DoubleMemorySystemField(fieldID int64) bool

func EncodeSearchResultData 

func EncodeSearchResultData(ctx context.Context, searchResultData *schemapb.SearchResultData,
	nq int64, topk int64, metricType string,
) (searchResults *internalpb.SearchResults, err error)

func FilterZeroValuesFromSlice 

func FilterZeroValuesFromSlice(intVals []int64) []int64

func GetBFApplyPool 

func GetBFApplyPool() *conc.Pool[any]

func GetCLoadInfoWithFunc 

func GetCLoadInfoWithFunc(ctx context.Context,
	fieldSchema *schemapb.FieldSchema,
	loadInfo *querypb.SegmentLoadInfo,
	indexInfo *querypb.FieldIndexInfo,
	f func(c *LoadIndexInfo) error,
) error

func GetDeletePool 

func GetDeletePool() *conc.Pool[struct{}]

func GetDynamicPool 

func GetDynamicPool() *conc.Pool[any]

GetDynamicPool returns the singleton pool for dynamic cgo operations.

func GetLoadPool 

func GetLoadPool() *conc.Pool[any]

func GetPkField 

func GetPkField(schema *schemapb.CollectionSchema) *schemapb.FieldSchema

func GetPrimaryKeys 

func GetPrimaryKeys(msg *msgstream.InsertMsg, schema *schemapb.CollectionSchema) ([]storage.PrimaryKey, error)

TODO: remove this function to proper file GetPrimaryKeys would get primary keys by insert messages

func GetSQPool 

func GetSQPool() *conc.Pool[any]

GetSQPool returns the singleton pool instance for search/query operations.

func GetSegmentRelatedDataSize 

func GetSegmentRelatedDataSize(segment Segment) int64

func GetWarmupPool 

func GetWarmupPool() *conc.Pool[any]

func HandleCStatus 

func HandleCStatus(ctx context.Context, status *C.CStatus, extraInfo string, fields ...zap.Field) error

HandleCStatus deals with the error returned from CGO

func JoinIDPath 

func JoinIDPath(ids ...int64) string

JoinIDPath joins ids to path format.

func MergeInternalRetrieveResult 

func MergeInternalRetrieveResult(ctx context.Context, retrieveResults []*internalpb.RetrieveResults, param *mergeParam) (*internalpb.RetrieveResults, error)

func MergeSegcoreRetrieveResults 

func MergeSegcoreRetrieveResults(ctx context.Context, retrieveResults []*segcorepb.RetrieveResults, param *mergeParam, segments []Segment, plan *RetrievePlan, manager *Manager) (*segcorepb.RetrieveResults, error)

func NewCollectionManager 

func NewCollectionManager() *collectionManager

func NewDiskUsageFetcher 

func NewDiskUsageFetcher(ctx context.Context) *diskUsageFetcher

func NewLoader 

func NewLoader(
	ctx context.Context,
	manager *Manager,
	cm storage.ChunkManager,
) *segmentLoader

func NewMergeParam 

func NewMergeParam(limit int64, outputFieldsId []int64, schema *schemapb.CollectionSchema, reduceType reduce.IReduceType) *mergeParam

func NewSegmentManager 

func NewSegmentManager() *segmentManager

func ReduceAdvancedSearchResults 

func ReduceAdvancedSearchResults(ctx context.Context, results []*internalpb.SearchResults) (*internalpb.SearchResults, error)

func ReduceSearchOnQueryNode 

func ReduceSearchOnQueryNode(ctx context.Context, results []*internalpb.SearchResults, info *reduce.ResultInfo) (*internalpb.SearchResults, error)

func ReduceSearchResults 

func ReduceSearchResults(ctx context.Context, results []*internalpb.SearchResults, info *reduce.ResultInfo) (*internalpb.SearchResults, error)

func ResizeBFApplyPool 

func ResizeBFApplyPool(evt *config.Event)

func ResizeLoadPool 

func ResizeLoadPool(evt *config.Event)

func ResizeSQPool 

func ResizeSQPool(evt *config.Event)

func ResizeWarmupPool 

func ResizeWarmupPool(evt *config.Event)

func Retrieve 

func Retrieve(ctx context.Context, manager *Manager, plan *RetrievePlan, req *querypb.QueryRequest, queryPlan *planpb.PlanNode) ([]RetrieveSegmentResult, []Segment, error)

retrieve will retrieve all the validate target segments

func SearchHistorical 

func SearchHistorical(ctx context.Context, manager *Manager, searchReq *SearchRequest, collID int64, partIDs []int64, segIDs []int64) ([]*SearchResult, []Segment, error)

search will search on the historical segments the target segments in historical. if segIDs is not specified, it will search on all the historical segments speficied by partIDs. if segIDs is specified, it will only search on the segments specified by the segIDs. if partIDs is empty, it means all the partitions of the loaded collection or all the partitions loaded.

func SearchStreaming 

func SearchStreaming(ctx context.Context, manager *Manager, searchReq *SearchRequest, collID int64, partIDs []int64, segIDs []int64) ([]*SearchResult, []Segment, error)

searchStreaming will search all the target segments in streaming if partIDs is empty, it means all the partitions of the loaded collection or all the partitions loaded.

func SelectSearchResultData 

func SelectSearchResultData(dataArray []*schemapb.SearchResultData, resultOffsets [][]int64, offsets []int64, qi int64) int

func StatisticStreaming 

func StatisticStreaming(ctx context.Context, manager *Manager, collID int64, partIDs []int64, segIDs []int64) ([]SegmentStats, []Segment, error)

StatisticStreaming will do statistics all the target segments in streaming if partIDs is empty, it means all the partitions of the loaded collection or all the partitions loaded.

func StatisticsHistorical 

func StatisticsHistorical(ctx context.Context, manager *Manager, collID int64, partIDs []int64, segIDs []int64) ([]SegmentStats, []Segment, error)

statistic will do statistics on the historical segments the target segments in historical. if segIDs is not specified, it will search on all the historical segments specified by partIDs. if segIDs is specified, it will only search on the segments specified by the segIDs. if partIDs is empty, it means all the partitions of the loaded collection or all the partitions loaded.

func SupportInterimIndexDataType 

func SupportInterimIndexDataType(dataType schemapb.DataType) bool

func WithReleaseScope 

func WithReleaseScope(scope ReleaseScope) releaseOption

Types

type CTraceContext 

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

CTraceContext is the wrapper for `C.CTraceContext` it stores the internal C.CTraceContext and

func ParseCTraceContext 

func ParseCTraceContext(ctx context.Context) *CTraceContext

ParseCTraceContext parses tracing span and convert it into `C.CTraceContext`.

type Collection 

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

Collection is a wrapper of the underlying C-structure C.CCollection In a query node, `Collection` is a replica info of a collection in these query node.

func NewCollection 

func NewCollection(collectionID int64, schema *schemapb.CollectionSchema, indexMeta *segcorepb.CollectionIndexMeta, loadMetaInfo *querypb.LoadMetaInfo) (*Collection, error)

newCollection returns a new Collection

func NewCollectionWithoutSegcoreForTest 

func NewCollectionWithoutSegcoreForTest(collectionID int64, schema *schemapb.CollectionSchema) *Collection

new collection without segcore prepare ONLY FOR TEST

func NewTestCollection 

func NewTestCollection(collectionID int64, loadType querypb.LoadType, schema *schemapb.CollectionSchema) *Collection

Only for test

func (*Collection) AddPartition 

func (c *Collection) AddPartition(partitions ...int64)

addPartitionID would add a partition id to partition id list of collection

func (*Collection) ExistPartition 

func (c *Collection) ExistPartition(partitionIDs ...int64) bool

func (*Collection) GetCCollection 

func (c *Collection) GetCCollection() *segcore.CCollection

GetCCollection returns the CCollection of collection

func (*Collection) GetDBName 

func (c *Collection) GetDBName() string

GetDBName returns the database name of collection.

func (*Collection) GetDBProperties 

func (c *Collection) GetDBProperties() []*commonpb.KeyValuePair

func (*Collection) GetLoadType 

func (c *Collection) GetLoadType() querypb.LoadType

getLoadType get the loadType of collection, which is loadTypeCollection or loadTypePartition

func (*Collection) GetPartitions 

func (c *Collection) GetPartitions() []int64

getPartitionIDs return partitionIDs of collection

func (*Collection) GetResourceGroup 

func (c *Collection) GetResourceGroup() string

GetResourceGroup returns the resource group of collection.

func (*Collection) ID 

func (c *Collection) ID() int64

ID returns collection id

func (*Collection) IsGpuIndex 

func (c *Collection) IsGpuIndex() bool

IsGpuIndex returns a boolean value indicating whether the collection is using a GPU index.

func (*Collection) Ref 

func (c *Collection) Ref(count uint32) uint32

func (*Collection) RemovePartition 

func (c *Collection) RemovePartition(partitionID int64)

removePartitionID removes the partition id from partition id list of collection

func (*Collection) Schema 

func (c *Collection) Schema() *schemapb.CollectionSchema

Schema returns the schema of collection

func (*Collection) Unref 

func (c *Collection) Unref(count uint32) uint32

type CollectionManager 

type CollectionManager interface {
	List() []int64
	ListWithName() map[int64]string
	Get(collectionID int64) *Collection
	PutOrRef(collectionID int64, schema *schemapb.CollectionSchema, meta *segcorepb.CollectionIndexMeta, loadMeta *querypb.LoadMetaInfo) error
	Ref(collectionID int64, count uint32) bool
	// unref the collection,
	// returns true if the collection ref count goes 0, or the collection not exists,
	// return false otherwise
	Unref(collectionID int64, count uint32) bool
	// UpdateSchema update the underlying collection schema of the provided collection.
	UpdateSchema(collectionID int64, schema *schemapb.CollectionSchema, version uint64) error
}

type FieldInfo 

type FieldInfo struct {
	*datapb.FieldBinlog
	RowCount int64
}

type IndexAttrCache 

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

IndexAttrCache index meta cache stores calculated attribute.

func NewIndexAttrCache 

func NewIndexAttrCache() *IndexAttrCache

func (*IndexAttrCache) GetIndexResourceUsage 

func (c *IndexAttrCache) GetIndexResourceUsage(indexInfo *querypb.FieldIndexInfo, memoryIndexLoadPredictMemoryUsageFactor float64, fieldBinlog *datapb.FieldBinlog) (memory uint64, disk uint64, err error)

type IndexedFieldInfo 

type IndexedFieldInfo struct {
	FieldBinlog *datapb.FieldBinlog
	IndexInfo   *querypb.FieldIndexInfo
	IsLoaded    bool
}

IndexedFieldInfo contains binlog info of vector field

type L0Segment 

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

func (*L0Segment) BatchPkExist 

func (s *L0Segment) BatchPkExist(lc *storage.BatchLocationsCache) []bool

func (*L0Segment) BloomFilterExist 

func (s *L0Segment) BloomFilterExist() bool

func (*L0Segment) CASVersion 

func (s *L0Segment) CASVersion(old, newVersion int64) bool

func (*L0Segment) Collection 

func (s *L0Segment) Collection() int64

func (*L0Segment) DatabaseName 

func (s *L0Segment) DatabaseName() string

func (*L0Segment) Delete 

func (s *L0Segment) Delete(ctx context.Context, primaryKeys storage.PrimaryKeys, timestamps []typeutil.Timestamp) error

func (*L0Segment) DeleteRecords 

func (s *L0Segment) DeleteRecords() ([]storage.PrimaryKey, []uint64)

func (*L0Segment) DropIndex 

func (s *L0Segment) DropIndex(ctx context.Context, indexID int64) error

func (*L0Segment) ExistIndex 

func (s *L0Segment) ExistIndex(fieldID int64) bool

func (*L0Segment) FinishLoad 

func (s *L0Segment) FinishLoad() error

func (*L0Segment) GetBM25Stats 

func (s *L0Segment) GetBM25Stats() map[int64]*storage.BM25Stats

func (*L0Segment) GetCollection 

func (s *L0Segment) GetCollection() *Collection

func (*L0Segment) GetFieldJSONIndexStats 

func (s *L0Segment) GetFieldJSONIndexStats() map[int64]*querypb.JsonStatsInfo

func (*L0Segment) GetIndex 

func (s *L0Segment) GetIndex(fieldID int64) []*IndexedFieldInfo

func (*L0Segment) GetIndexByID 

func (s *L0Segment) GetIndexByID(indexID int64) *IndexedFieldInfo

func (*L0Segment) GetMaxPk 

func (s *L0Segment) GetMaxPk() *storage.PrimaryKey

func (*L0Segment) GetMinPk 

func (s *L0Segment) GetMinPk() *storage.PrimaryKey

func (*L0Segment) HasRawData 

func (s *L0Segment) HasRawData(fieldID int64) bool

func (*L0Segment) ID 

func (s *L0Segment) ID() int64

ID returns the identity number.

func (*L0Segment) Indexes 

func (s *L0Segment) Indexes() []*IndexedFieldInfo

func (*L0Segment) Insert 

func (s *L0Segment) Insert(ctx context.Context, rowIDs []int64, timestamps []typeutil.Timestamp, record *segcorepb.InsertRecord) error

func (*L0Segment) InsertCount 

func (s *L0Segment) InsertCount() int64

func (*L0Segment) IsLazyLoad 

func (s *L0Segment) IsLazyLoad() bool

func (*L0Segment) IsSorted 

func (s *L0Segment) IsSorted() bool

func (*L0Segment) LastDeltaTimestamp 

func (s *L0Segment) LastDeltaTimestamp() uint64

func (*L0Segment) Level 

func (s *L0Segment) Level() datapb.SegmentLevel

func (*L0Segment) LoadDeltaData 

func (s *L0Segment) LoadDeltaData(ctx context.Context, deltaData *storage.DeltaData) error

func (*L0Segment) LoadInfo 

func (s *L0Segment) LoadInfo() *querypb.SegmentLoadInfo

func (*L0Segment) MayPkExist 

func (s *L0Segment) MayPkExist(pk *storage.LocationsCache) bool

MayPkExist returns true if the given PK exists in the PK range and being positive through the bloom filter, false otherwise, may returns true even the PK doesn't exist actually

func (*L0Segment) MemSize 

func (s *L0Segment) MemSize() int64

func (*L0Segment) NeedUpdatedVersion 

func (s *L0Segment) NeedUpdatedVersion() int64

func (*L0Segment) Partition 

func (s *L0Segment) Partition() int64

func (*L0Segment) PinIfNotReleased 

func (s *L0Segment) PinIfNotReleased() error

func (*L0Segment) Release 

func (s *L0Segment) Release(ctx context.Context, opts ...releaseOption)

func (*L0Segment) RemoveUnusedFieldFiles 

func (s *L0Segment) RemoveUnusedFieldFiles() error

func (*L0Segment) ResetIndexesLazyLoad 

func (s *L0Segment) ResetIndexesLazyLoad(lazyState bool)

func (*L0Segment) ResourceGroup 

func (s *L0Segment) ResourceGroup() string

func (*L0Segment) ResourceUsageEstimate 

func (s *L0Segment) ResourceUsageEstimate() ResourceUsage

ResourceUsageEstimate returns the final estimated resource usage of the segment.

func (*L0Segment) Retrieve 

func (s *L0Segment) Retrieve(ctx context.Context, plan *segcore.RetrievePlan) (*segcorepb.RetrieveResults, error)

func (*L0Segment) RetrieveByOffsets 

func (s *L0Segment) RetrieveByOffsets(ctx context.Context, plan *segcore.RetrievePlanWithOffsets) (*segcorepb.RetrieveResults, error)

func (*L0Segment) RowNum 

func (s *L0Segment) RowNum() int64

func (*L0Segment) Search 

func (s *L0Segment) Search(ctx context.Context, searchReq *segcore.SearchRequest) (*segcore.SearchResult, error)

func (*L0Segment) SetBloomFilter 

func (s *L0Segment) SetBloomFilter(bf *pkoracle.BloomFilterSet)

func (*L0Segment) SetLoadInfo 

func (s *L0Segment) SetLoadInfo(loadInfo *querypb.SegmentLoadInfo)

func (*L0Segment) SetNeedUpdatedVersion 

func (s *L0Segment) SetNeedUpdatedVersion(version int64)

func (*L0Segment) Shard 

func (s *L0Segment) Shard() metautil.Channel

func (*L0Segment) StartPosition 

func (s *L0Segment) StartPosition() *msgpb.MsgPosition

func (*L0Segment) Type 

func (s *L0Segment) Type() SegmentType

func (*L0Segment) Unpin 

func (s *L0Segment) Unpin()

func (*L0Segment) UpdateBM25Stats 

func (s *L0Segment) UpdateBM25Stats(stats map[int64]*storage.BM25Stats)

func (*L0Segment) UpdateBloomFilter 

func (s *L0Segment) UpdateBloomFilter(pks []storage.PrimaryKey)

func (*L0Segment) Version 

func (s *L0Segment) Version() int64

type LoadIndexInfo 

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

LoadIndexInfo is a wrapper of the underlying C-structure C.CLoadIndexInfo

type LoadResource 

type LoadResource struct {
	MemorySize uint64
	DiskSize   uint64
}

func (*LoadResource) Add 

func (r *LoadResource) Add(resource LoadResource)

func (*LoadResource) IsZero 

func (r *LoadResource) IsZero() bool

func (*LoadResource) Sub 

func (r *LoadResource) Sub(resource LoadResource)

type Loader 

type Loader interface {
	// Load loads binlogs, and spawn segments,
	// NOTE: make sure the ref count of the corresponding collection will never go down to 0 during this
	Load(ctx context.Context, collectionID int64, segmentType SegmentType, version int64, segments ...*querypb.SegmentLoadInfo) ([]Segment, error)

	// LoadDeltaLogs load deltalog and write delta data into provided segment.
	// it also executes resource protection logic in case of OOM.
	LoadDeltaLogs(ctx context.Context, segment Segment, deltaLogs []*datapb.FieldBinlog) error

	// LoadBloomFilterSet loads needed statslog for RemoteSegment.
	LoadBloomFilterSet(ctx context.Context, collectionID int64, infos ...*querypb.SegmentLoadInfo) ([]*pkoracle.BloomFilterSet, error)

	// LoadBM25Stats loads BM25 statslog for RemoteSegment
	LoadBM25Stats(ctx context.Context, collectionID int64, infos ...*querypb.SegmentLoadInfo) (*typeutil.ConcurrentMap[int64, map[int64]*storage.BM25Stats], error)

	// LoadIndex append index for segment and remove vector binlogs.
	LoadIndex(ctx context.Context,
		segment Segment,
		info *querypb.SegmentLoadInfo,
		version int64) error

	LoadLazySegment(ctx context.Context,
		segment Segment,
		loadInfo *querypb.SegmentLoadInfo,
	) error

	LoadJSONIndex(ctx context.Context,
		segment Segment,
		info *querypb.SegmentLoadInfo) error
}

type LocalSegment 

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

Segment is a wrapper of the underlying C-structure segment.

func (*LocalSegment) AddFieldDataInfo 

func (s *LocalSegment) AddFieldDataInfo(ctx context.Context, rowCount int64, fields []*datapb.FieldBinlog) error

func (*LocalSegment) BatchPkExist 

func (s *LocalSegment) BatchPkExist(lc *storage.BatchLocationsCache) []bool

func (*LocalSegment) BloomFilterExist 

func (s *LocalSegment) BloomFilterExist() bool

func (*LocalSegment) CASVersion 

func (s *LocalSegment) CASVersion(old, newVersion int64) bool

func (*LocalSegment) Collection 

func (s *LocalSegment) Collection() int64

func (*LocalSegment) CreateTextIndex 

func (s *LocalSegment) CreateTextIndex(ctx context.Context, fieldID int64) error

func (*LocalSegment) DatabaseName 

func (s *LocalSegment) DatabaseName() string

func (*LocalSegment) Delete 

func (s *LocalSegment) Delete(ctx context.Context, primaryKeys storage.PrimaryKeys, timestamps []typeutil.Timestamp) error

func (*LocalSegment) DropIndex 

func (s *LocalSegment) DropIndex(ctx context.Context, indexID int64) error

func (*LocalSegment) ExistIndex 

func (s *LocalSegment) ExistIndex(fieldID int64) bool

func (*LocalSegment) FinishLoad 

func (s *LocalSegment) FinishLoad() error

func (*LocalSegment) GetBM25Stats 

func (s *LocalSegment) GetBM25Stats() map[int64]*storage.BM25Stats

func (*LocalSegment) GetCollection 

func (s *LocalSegment) GetCollection() *Collection

func (*LocalSegment) GetFieldJSONIndexStats 

func (s *LocalSegment) GetFieldJSONIndexStats() map[int64]*querypb.JsonStatsInfo

func (*LocalSegment) GetIndex 

func (s *LocalSegment) GetIndex(fieldID int64) []*IndexedFieldInfo

func (*LocalSegment) GetIndexByID 

func (s *LocalSegment) GetIndexByID(indexID int64) *IndexedFieldInfo

func (*LocalSegment) GetMaxPk 

func (s *LocalSegment) GetMaxPk() *storage.PrimaryKey

func (*LocalSegment) GetMinPk 

func (s *LocalSegment) GetMinPk() *storage.PrimaryKey

func (*LocalSegment) HasFieldData 

func (s *LocalSegment) HasFieldData(fieldID int64) bool

func (*LocalSegment) HasRawData 

func (s *LocalSegment) HasRawData(fieldID int64) bool

func (*LocalSegment) ID 

func (s *LocalSegment) ID() int64

ID returns the identity number.

func (*LocalSegment) Indexes 

func (s *LocalSegment) Indexes() []*IndexedFieldInfo

func (*LocalSegment) Insert 

func (s *LocalSegment) Insert(ctx context.Context, rowIDs []int64, timestamps []typeutil.Timestamp, record *segcorepb.InsertRecord) error

func (*LocalSegment) InsertCount 

func (s *LocalSegment) InsertCount() int64

func (*LocalSegment) IsLazyLoad 

func (s *LocalSegment) IsLazyLoad() bool

func (*LocalSegment) IsSorted 

func (s *LocalSegment) IsSorted() bool

func (*LocalSegment) LastDeltaTimestamp 

func (s *LocalSegment) LastDeltaTimestamp() uint64

func (*LocalSegment) Level 

func (s *LocalSegment) Level() datapb.SegmentLevel

func (*LocalSegment) LoadDeltaData 

func (s *LocalSegment) LoadDeltaData(ctx context.Context, deltaData *storage.DeltaData) error

func (*LocalSegment) LoadFieldData 

func (s *LocalSegment) LoadFieldData(ctx context.Context, fieldID int64, rowCount int64, field *datapb.FieldBinlog, warmupPolicy ...string) error

func (*LocalSegment) LoadIndex 

func (s *LocalSegment) LoadIndex(ctx context.Context, indexInfo *querypb.FieldIndexInfo, fieldType schemapb.DataType) error

func (*LocalSegment) LoadInfo 

func (s *LocalSegment) LoadInfo() *querypb.SegmentLoadInfo

func (*LocalSegment) LoadJSONKeyIndex 

func (s *LocalSegment) LoadJSONKeyIndex(ctx context.Context, jsonKeyStats *datapb.JsonKeyStats, schemaHelper *typeutil.SchemaHelper) error

func (*LocalSegment) LoadMultiFieldData 

func (s *LocalSegment) LoadMultiFieldData(ctx context.Context) error

-------------------------------------------------------------------------------------- interfaces for sealed segment

func (*LocalSegment) LoadTextIndex 

func (s *LocalSegment) LoadTextIndex(ctx context.Context, textLogs *datapb.TextIndexStats, schemaHelper *typeutil.SchemaHelper) error

func (*LocalSegment) MayPkExist 

func (s *LocalSegment) MayPkExist(pk *storage.LocationsCache) bool

MayPkExist returns true if the given PK exists in the PK range and being positive through the bloom filter, false otherwise, may returns true even the PK doesn't exist actually

func (*LocalSegment) MemSize 

func (s *LocalSegment) MemSize() int64

func (*LocalSegment) NeedUpdatedVersion 

func (s *LocalSegment) NeedUpdatedVersion() int64

func (*LocalSegment) Partition 

func (s *LocalSegment) Partition() int64

func (*LocalSegment) PinIfNotReleased 

func (s *LocalSegment) PinIfNotReleased() error

PinIfNotReleased acquires the `ptrLock` and returns true if the pointer is valid Provide ONLY the read lock operations, don't make `ptrLock` public to avoid abusing of the mutex.

func (*LocalSegment) Release 

func (s *LocalSegment) Release(ctx context.Context, opts ...releaseOption)

func (*LocalSegment) ReleaseSegmentData 

func (s *LocalSegment) ReleaseSegmentData()

ReleaseSegmentData releases the segment data.

func (*LocalSegment) RemoveFieldFile 

func (s *LocalSegment) RemoveFieldFile(fieldId int64)

func (*LocalSegment) RemoveUnusedFieldFiles 

func (s *LocalSegment) RemoveUnusedFieldFiles() error

func (*LocalSegment) ResetIndexesLazyLoad 

func (s *LocalSegment) ResetIndexesLazyLoad(lazyState bool)

func (*LocalSegment) ResourceGroup 

func (s *LocalSegment) ResourceGroup() string

func (*LocalSegment) ResourceUsageEstimate 

func (s *LocalSegment) ResourceUsageEstimate() ResourceUsage

ResourceUsageEstimate returns the final estimated resource usage of the segment.

func (*LocalSegment) Retrieve 

func (s *LocalSegment) Retrieve(ctx context.Context, plan *segcore.RetrievePlan) (*segcorepb.RetrieveResults, error)

func (*LocalSegment) RetrieveByOffsets 

func (s *LocalSegment) RetrieveByOffsets(ctx context.Context, plan *segcore.RetrievePlanWithOffsets) (*segcorepb.RetrieveResults, error)

func (*LocalSegment) RowNum 

func (s *LocalSegment) RowNum() int64

func (*LocalSegment) Search 

func (s *LocalSegment) Search(ctx context.Context, searchReq *segcore.SearchRequest) (*segcore.SearchResult, error)

func (*LocalSegment) SetBloomFilter 

func (s *LocalSegment) SetBloomFilter(bf *pkoracle.BloomFilterSet)

func (*LocalSegment) SetLoadInfo 

func (s *LocalSegment) SetLoadInfo(loadInfo *querypb.SegmentLoadInfo)

func (*LocalSegment) SetNeedUpdatedVersion 

func (s *LocalSegment) SetNeedUpdatedVersion(version int64)

func (*LocalSegment) Shard 

func (s *LocalSegment) Shard() metautil.Channel

func (*LocalSegment) StartLoadData 

func (s *LocalSegment) StartLoadData() (state.LoadStateLockGuard, error)

StartLoadData starts the loading process of the segment.

func (*LocalSegment) StartPosition 

func (s *LocalSegment) StartPosition() *msgpb.MsgPosition

func (*LocalSegment) Type 

func (s *LocalSegment) Type() SegmentType

func (*LocalSegment) Unpin 

func (s *LocalSegment) Unpin()

func (*LocalSegment) UpdateBM25Stats 

func (s *LocalSegment) UpdateBM25Stats(stats map[int64]*storage.BM25Stats)

func (*LocalSegment) UpdateBloomFilter 

func (s *LocalSegment) UpdateBloomFilter(pks []storage.PrimaryKey)

func (*LocalSegment) UpdateFieldRawDataSize 

func (s *LocalSegment) UpdateFieldRawDataSize(ctx context.Context, numRows int64, fieldBinlog *datapb.FieldBinlog) error

func (*LocalSegment) UpdateIndexInfo 

func (s *LocalSegment) UpdateIndexInfo(ctx context.Context, indexInfo *querypb.FieldIndexInfo, info *LoadIndexInfo) error

func (*LocalSegment) Version 

func (s *LocalSegment) Version() int64

type Manager 

type Manager struct {
	Collection CollectionManager
	Segment    SegmentManager
	DiskCache  cache.Cache[int64, Segment]
	Loader     Loader
}

func NewManager 

func NewManager() *Manager

func (*Manager) SetLoader 

func (mgr *Manager) SetLoader(loader Loader)

type MockCollectionManager 

type MockCollectionManager struct {
	mock.Mock
}

MockCollectionManager is an autogenerated mock type for the CollectionManager type

func NewMockCollectionManager 

func NewMockCollectionManager(t interface {
	mock.TestingT
	Cleanup(func())
}) *MockCollectionManager

NewMockCollectionManager creates a new instance of MockCollectionManager. 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 (*MockCollectionManager) EXPECT 

func (_m *MockCollectionManager) EXPECT() *MockCollectionManager_Expecter

func (*MockCollectionManager) Get 

func (_m *MockCollectionManager) Get(collectionID int64) *Collection

Get provides a mock function with given fields: collectionID

func (*MockCollectionManager) List 

func (_m *MockCollectionManager) List() []int64

List provides a mock function with no fields

func (*MockCollectionManager) ListWithName 

func (_m *MockCollectionManager) ListWithName() map[int64]string

ListWithName provides a mock function with no fields

func (*MockCollectionManager) PutOrRef 

func (_m *MockCollectionManager) PutOrRef(collectionID int64, schema *schemapb.CollectionSchema, meta *segcorepb.CollectionIndexMeta, loadMeta *querypb.LoadMetaInfo) error

PutOrRef provides a mock function with given fields: collectionID, schema, meta, loadMeta

func (*MockCollectionManager) Ref 

func (_m *MockCollectionManager) Ref(collectionID int64, count uint32) bool

Ref provides a mock function with given fields: collectionID, count

func (*MockCollectionManager) Unref 

func (_m *MockCollectionManager) Unref(collectionID int64, count uint32) bool

Unref provides a mock function with given fields: collectionID, count

func (*MockCollectionManager) UpdateSchema 

func (_m *MockCollectionManager) UpdateSchema(collectionID int64, schema *schemapb.CollectionSchema, version uint64) error

UpdateSchema provides a mock function with given fields: collectionID, schema, version

type MockCollectionManager_Expecter 

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

func (*MockCollectionManager_Expecter) Get 

func (_e *MockCollectionManager_Expecter) Get(collectionID interface{}) *MockCollectionManager_Get_Call

Get is a helper method to define mock.On call

  • collectionID int64

func (*MockCollectionManager_Expecter) List 

func (_e *MockCollectionManager_Expecter) List() *MockCollectionManager_List_Call

List is a helper method to define mock.On call

func (*MockCollectionManager_Expecter) ListWithName 

func (_e *MockCollectionManager_Expecter) ListWithName() *MockCollectionManager_ListWithName_Call

ListWithName is a helper method to define mock.On call

func (*MockCollectionManager_Expecter) PutOrRef 

func (_e *MockCollectionManager_Expecter) PutOrRef(collectionID interface{}, schema interface{}, meta interface{}, loadMeta interface{}) *MockCollectionManager_PutOrRef_Call

PutOrRef is a helper method to define mock.On call

  • collectionID int64
  • schema *schemapb.CollectionSchema
  • meta *segcorepb.CollectionIndexMeta
  • loadMeta *querypb.LoadMetaInfo

func (*MockCollectionManager_Expecter) Ref 

func (_e *MockCollectionManager_Expecter) Ref(collectionID interface{}, count interface{}) *MockCollectionManager_Ref_Call

Ref is a helper method to define mock.On call

  • collectionID int64
  • count uint32

func (*MockCollectionManager_Expecter) Unref 

func (_e *MockCollectionManager_Expecter) Unref(collectionID interface{}, count interface{}) *MockCollectionManager_Unref_Call

Unref is a helper method to define mock.On call

  • collectionID int64
  • count uint32

func (*MockCollectionManager_Expecter) UpdateSchema 

func (_e *MockCollectionManager_Expecter) UpdateSchema(collectionID interface{}, schema interface{}, version interface{}) *MockCollectionManager_UpdateSchema_Call

UpdateSchema is a helper method to define mock.On call

  • collectionID int64
  • schema *schemapb.CollectionSchema
  • version uint64

type MockCollectionManager_Get_Call 

type MockCollectionManager_Get_Call struct {
	*mock.Call
}

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

func (*MockCollectionManager_Get_Call) Return 

func (_c *MockCollectionManager_Get_Call) Return(_a0 *Collection) *MockCollectionManager_Get_Call

func (*MockCollectionManager_Get_Call) Run 

func (_c *MockCollectionManager_Get_Call) Run(run func(collectionID int64)) *MockCollectionManager_Get_Call

func (*MockCollectionManager_Get_Call) RunAndReturn 

func (_c *MockCollectionManager_Get_Call) RunAndReturn(run func(int64) *Collection) *MockCollectionManager_Get_Call

type MockCollectionManager_ListWithName_Call 

type MockCollectionManager_ListWithName_Call struct {
	*mock.Call
}

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

func (*MockCollectionManager_ListWithName_Call) Return 

func (_c *MockCollectionManager_ListWithName_Call) Return(_a0 map[int64]string) *MockCollectionManager_ListWithName_Call

func (*MockCollectionManager_ListWithName_Call) Run 

func (_c *MockCollectionManager_ListWithName_Call) Run(run func()) *MockCollectionManager_ListWithName_Call

func (*MockCollectionManager_ListWithName_Call) RunAndReturn 

func (_c *MockCollectionManager_ListWithName_Call) RunAndReturn(run func() map[int64]string) *MockCollectionManager_ListWithName_Call

type MockCollectionManager_List_Call 

type MockCollectionManager_List_Call struct {
	*mock.Call
}

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

func (*MockCollectionManager_List_Call) Return 

func (_c *MockCollectionManager_List_Call) Return(_a0 []int64) *MockCollectionManager_List_Call

func (*MockCollectionManager_List_Call) Run 

func (_c *MockCollectionManager_List_Call) Run(run func()) *MockCollectionManager_List_Call

func (*MockCollectionManager_List_Call) RunAndReturn 

func (_c *MockCollectionManager_List_Call) RunAndReturn(run func() []int64) *MockCollectionManager_List_Call

type MockCollectionManager_PutOrRef_Call 

type MockCollectionManager_PutOrRef_Call struct {
	*mock.Call
}

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

func (*MockCollectionManager_PutOrRef_Call) Return 

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

func (*MockCollectionManager_PutOrRef_Call) Run 

func (_c *MockCollectionManager_PutOrRef_Call) Run(run func(collectionID int64, schema *schemapb.CollectionSchema, meta *segcorepb.CollectionIndexMeta, loadMeta *querypb.LoadMetaInfo)) *MockCollectionManager_PutOrRef_Call

func (*MockCollectionManager_PutOrRef_Call) RunAndReturn 

func (_c *MockCollectionManager_PutOrRef_Call) RunAndReturn(run func(int64, *schemapb.CollectionSchema, *segcorepb.CollectionIndexMeta, *querypb.LoadMetaInfo) error) *MockCollectionManager_PutOrRef_Call

type MockCollectionManager_Ref_Call 

type MockCollectionManager_Ref_Call struct {
	*mock.Call
}

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

func (*MockCollectionManager_Ref_Call) Return 

func (_c *MockCollectionManager_Ref_Call) Return(_a0 bool) *MockCollectionManager_Ref_Call

func (*MockCollectionManager_Ref_Call) Run 

func (_c *MockCollectionManager_Ref_Call) Run(run func(collectionID int64, count uint32)) *MockCollectionManager_Ref_Call

func (*MockCollectionManager_Ref_Call) RunAndReturn 

func (_c *MockCollectionManager_Ref_Call) RunAndReturn(run func(int64, uint32) bool) *MockCollectionManager_Ref_Call

type MockCollectionManager_Unref_Call 

type MockCollectionManager_Unref_Call struct {
	*mock.Call
}

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

func (*MockCollectionManager_Unref_Call) Return 

func (_c *MockCollectionManager_Unref_Call) Return(_a0 bool) *MockCollectionManager_Unref_Call

func (*MockCollectionManager_Unref_Call) Run 

func (_c *MockCollectionManager_Unref_Call) Run(run func(collectionID int64, count uint32)) *MockCollectionManager_Unref_Call

func (*MockCollectionManager_Unref_Call) RunAndReturn 

func (_c *MockCollectionManager_Unref_Call) RunAndReturn(run func(int64, uint32) bool) *MockCollectionManager_Unref_Call

type MockCollectionManager_UpdateSchema_Call 

type MockCollectionManager_UpdateSchema_Call struct {
	*mock.Call
}

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

func (*MockCollectionManager_UpdateSchema_Call) Return 

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

func (*MockCollectionManager_UpdateSchema_Call) Run 

func (_c *MockCollectionManager_UpdateSchema_Call) Run(run func(collectionID int64, schema *schemapb.CollectionSchema, version uint64)) *MockCollectionManager_UpdateSchema_Call

func (*MockCollectionManager_UpdateSchema_Call) RunAndReturn 

func (_c *MockCollectionManager_UpdateSchema_Call) RunAndReturn(run func(int64, *schemapb.CollectionSchema, uint64) error) *MockCollectionManager_UpdateSchema_Call

type MockLoader 

type MockLoader struct {
	mock.Mock
}

MockLoader is an autogenerated mock type for the Loader type

func NewMockLoader 

func NewMockLoader(t interface {
	mock.TestingT
	Cleanup(func())
}) *MockLoader

NewMockLoader creates a new instance of MockLoader. 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 (*MockLoader) EXPECT 

func (_m *MockLoader) EXPECT() *MockLoader_Expecter

func (*MockLoader) Load 

func (_m *MockLoader) Load(ctx context.Context, collectionID int64, segmentType commonpb.SegmentState, version int64, _a4 ...*querypb.SegmentLoadInfo) ([]Segment, error)

Load provides a mock function with given fields: ctx, collectionID, segmentType, version, _a4

func (*MockLoader) LoadBM25Stats 

func (_m *MockLoader) LoadBM25Stats(ctx context.Context, collectionID int64, infos ...*querypb.SegmentLoadInfo) (*typeutil.ConcurrentMap[int64, map[int64]*storage.BM25Stats], error)

LoadBM25Stats provides a mock function with given fields: ctx, collectionID, infos

func (*MockLoader) LoadBloomFilterSet 

func (_m *MockLoader) LoadBloomFilterSet(ctx context.Context, collectionID int64, infos ...*querypb.SegmentLoadInfo) ([]*pkoracle.BloomFilterSet, error)

LoadBloomFilterSet provides a mock function with given fields: ctx, collectionID, infos

func (*MockLoader) LoadDeltaLogs 

func (_m *MockLoader) LoadDeltaLogs(ctx context.Context, segment Segment, deltaLogs []*datapb.FieldBinlog) error

LoadDeltaLogs provides a mock function with given fields: ctx, segment, deltaLogs

func (*MockLoader) LoadIndex 

func (_m *MockLoader) LoadIndex(ctx context.Context, segment Segment, info *querypb.SegmentLoadInfo, version int64) error

LoadIndex provides a mock function with given fields: ctx, segment, info, version

func (*MockLoader) LoadJSONIndex 

func (_m *MockLoader) LoadJSONIndex(ctx context.Context, segment Segment, info *querypb.SegmentLoadInfo) error

LoadJSONIndex provides a mock function with given fields: ctx, segment, info

func (*MockLoader) LoadLazySegment 

func (_m *MockLoader) LoadLazySegment(ctx context.Context, segment Segment, loadInfo *querypb.SegmentLoadInfo) error

LoadLazySegment provides a mock function with given fields: ctx, segment, loadInfo

type MockLoader_Expecter 

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

func (*MockLoader_Expecter) Load 

func (_e *MockLoader_Expecter) Load(ctx interface{}, collectionID interface{}, segmentType interface{}, version interface{}, _a4 ...interface{}) *MockLoader_Load_Call

Load is a helper method to define mock.On call

  • ctx context.Context
  • collectionID int64
  • segmentType commonpb.SegmentState
  • version int64
  • _a4 ...*querypb.SegmentLoadInfo

func (*MockLoader_Expecter) LoadBM25Stats 

func (_e *MockLoader_Expecter) LoadBM25Stats(ctx interface{}, collectionID interface{}, infos ...interface{}) *MockLoader_LoadBM25Stats_Call

LoadBM25Stats is a helper method to define mock.On call

  • ctx context.Context
  • collectionID int64
  • infos ...*querypb.SegmentLoadInfo

func (*MockLoader_Expecter) LoadBloomFilterSet 

func (_e *MockLoader_Expecter) LoadBloomFilterSet(ctx interface{}, collectionID interface{}, infos ...interface{}) *MockLoader_LoadBloomFilterSet_Call

LoadBloomFilterSet is a helper method to define mock.On call

  • ctx context.Context
  • collectionID int64
  • infos ...*querypb.SegmentLoadInfo

func (*MockLoader_Expecter) LoadDeltaLogs 

func (_e *MockLoader_Expecter) LoadDeltaLogs(ctx interface{}, segment interface{}, deltaLogs interface{}) *MockLoader_LoadDeltaLogs_Call

LoadDeltaLogs is a helper method to define mock.On call

  • ctx context.Context
  • segment Segment
  • deltaLogs []*datapb.FieldBinlog

func (*MockLoader_Expecter) LoadIndex 

func (_e *MockLoader_Expecter) LoadIndex(ctx interface{}, segment interface{}, info interface{}, version interface{}) *MockLoader_LoadIndex_Call

LoadIndex is a helper method to define mock.On call

  • ctx context.Context
  • segment Segment
  • info *querypb.SegmentLoadInfo
  • version int64

func (*MockLoader_Expecter) LoadJSONIndex 

func (_e *MockLoader_Expecter) LoadJSONIndex(ctx interface{}, segment interface{}, info interface{}) *MockLoader_LoadJSONIndex_Call

LoadJSONIndex is a helper method to define mock.On call

  • ctx context.Context
  • segment Segment
  • info *querypb.SegmentLoadInfo

func (*MockLoader_Expecter) LoadLazySegment 

func (_e *MockLoader_Expecter) LoadLazySegment(ctx interface{}, segment interface{}, loadInfo interface{}) *MockLoader_LoadLazySegment_Call

LoadLazySegment is a helper method to define mock.On call

  • ctx context.Context
  • segment Segment
  • loadInfo *querypb.SegmentLoadInfo

type MockLoader_LoadBM25Stats_Call 

type MockLoader_LoadBM25Stats_Call struct {
	*mock.Call
}

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

func (*MockLoader_LoadBM25Stats_Call) Return 

func (_c *MockLoader_LoadBM25Stats_Call) Return(_a0 *typeutil.ConcurrentMap[int64, map[int64]*storage.BM25Stats], _a1 error) *MockLoader_LoadBM25Stats_Call

func (*MockLoader_LoadBM25Stats_Call) Run 

func (_c *MockLoader_LoadBM25Stats_Call) Run(run func(ctx context.Context, collectionID int64, infos ...*querypb.SegmentLoadInfo)) *MockLoader_LoadBM25Stats_Call

func (*MockLoader_LoadBM25Stats_Call) RunAndReturn 

func (_c *MockLoader_LoadBM25Stats_Call) RunAndReturn(run func(context.Context, int64, ...*querypb.SegmentLoadInfo) (*typeutil.ConcurrentMap[int64, map[int64]*storage.BM25Stats], error)) *MockLoader_LoadBM25Stats_Call

type MockLoader_LoadBloomFilterSet_Call 

type MockLoader_LoadBloomFilterSet_Call struct {
	*mock.Call
}

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

func (*MockLoader_LoadBloomFilterSet_Call) Return 

func (_c *MockLoader_LoadBloomFilterSet_Call) Return(_a0 []*pkoracle.BloomFilterSet, _a1 error) *MockLoader_LoadBloomFilterSet_Call

func (*MockLoader_LoadBloomFilterSet_Call) Run 

func (_c *MockLoader_LoadBloomFilterSet_Call) Run(run func(ctx context.Context, collectionID int64, infos ...*querypb.SegmentLoadInfo)) *MockLoader_LoadBloomFilterSet_Call

func (*MockLoader_LoadBloomFilterSet_Call) RunAndReturn 

func (_c *MockLoader_LoadBloomFilterSet_Call) RunAndReturn(run func(context.Context, int64, ...*querypb.SegmentLoadInfo) ([]*pkoracle.BloomFilterSet, error)) *MockLoader_LoadBloomFilterSet_Call

type MockLoader_LoadDeltaLogs_Call 

type MockLoader_LoadDeltaLogs_Call struct {
	*mock.Call
}

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

func (*MockLoader_LoadDeltaLogs_Call) Return 

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

func (*MockLoader_LoadDeltaLogs_Call) Run 

func (_c *MockLoader_LoadDeltaLogs_Call) Run(run func(ctx context.Context, segment Segment, deltaLogs []*datapb.FieldBinlog)) *MockLoader_LoadDeltaLogs_Call

func (*MockLoader_LoadDeltaLogs_Call) RunAndReturn 

func (_c *MockLoader_LoadDeltaLogs_Call) RunAndReturn(run func(context.Context, Segment, []*datapb.FieldBinlog) error) *MockLoader_LoadDeltaLogs_Call

type MockLoader_LoadIndex_Call 

type MockLoader_LoadIndex_Call struct {
	*mock.Call
}

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

func (*MockLoader_LoadIndex_Call) Return 

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

func (*MockLoader_LoadIndex_Call) Run 

func (_c *MockLoader_LoadIndex_Call) Run(run func(ctx context.Context, segment Segment, info *querypb.SegmentLoadInfo, version int64)) *MockLoader_LoadIndex_Call

func (*MockLoader_LoadIndex_Call) RunAndReturn 

func (_c *MockLoader_LoadIndex_Call) RunAndReturn(run func(context.Context, Segment, *querypb.SegmentLoadInfo, int64) error) *MockLoader_LoadIndex_Call

type MockLoader_LoadJSONIndex_Call 

type MockLoader_LoadJSONIndex_Call struct {
	*mock.Call
}

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

func (*MockLoader_LoadJSONIndex_Call) Return 

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

func (*MockLoader_LoadJSONIndex_Call) Run 

func (_c *MockLoader_LoadJSONIndex_Call) Run(run func(ctx context.Context, segment Segment, info *querypb.SegmentLoadInfo)) *MockLoader_LoadJSONIndex_Call

func (*MockLoader_LoadJSONIndex_Call) RunAndReturn 

func (_c *MockLoader_LoadJSONIndex_Call) RunAndReturn(run func(context.Context, Segment, *querypb.SegmentLoadInfo) error) *MockLoader_LoadJSONIndex_Call

type MockLoader_LoadLazySegment_Call 

type MockLoader_LoadLazySegment_Call struct {
	*mock.Call
}

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

func (*MockLoader_LoadLazySegment_Call) Return 

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

func (*MockLoader_LoadLazySegment_Call) Run 

func (_c *MockLoader_LoadLazySegment_Call) Run(run func(ctx context.Context, segment Segment, loadInfo *querypb.SegmentLoadInfo)) *MockLoader_LoadLazySegment_Call

func (*MockLoader_LoadLazySegment_Call) RunAndReturn 

func (_c *MockLoader_LoadLazySegment_Call) RunAndReturn(run func(context.Context, Segment, *querypb.SegmentLoadInfo) error) *MockLoader_LoadLazySegment_Call

type MockLoader_Load_Call 

type MockLoader_Load_Call struct {
	*mock.Call
}

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

func (*MockLoader_Load_Call) Return 

func (_c *MockLoader_Load_Call) Return(_a0 []Segment, _a1 error) *MockLoader_Load_Call

func (*MockLoader_Load_Call) Run 

func (_c *MockLoader_Load_Call) Run(run func(ctx context.Context, collectionID int64, segmentType commonpb.SegmentState, version int64, _a4 ...*querypb.SegmentLoadInfo)) *MockLoader_Load_Call

func (*MockLoader_Load_Call) RunAndReturn 

func (_c *MockLoader_Load_Call) RunAndReturn(run func(context.Context, int64, commonpb.SegmentState, int64, ...*querypb.SegmentLoadInfo) ([]Segment, error)) *MockLoader_Load_Call

type MockSegment 

type MockSegment struct {
	mock.Mock
}

MockSegment is an autogenerated mock type for the Segment type

func NewMockSegment 

func NewMockSegment(t interface {
	mock.TestingT
	Cleanup(func())
}) *MockSegment

NewMockSegment creates a new instance of MockSegment. 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 (*MockSegment) BatchPkExist 

func (_m *MockSegment) BatchPkExist(lc *storage.BatchLocationsCache) []bool

BatchPkExist provides a mock function with given fields: lc

func (*MockSegment) BloomFilterExist 

func (_m *MockSegment) BloomFilterExist() bool

BloomFilterExist provides a mock function with no fields

func (*MockSegment) CASVersion 

func (_m *MockSegment) CASVersion(_a0 int64, _a1 int64) bool

CASVersion provides a mock function with given fields: _a0, _a1

func (*MockSegment) Collection 

func (_m *MockSegment) Collection() int64

Collection provides a mock function with no fields

func (*MockSegment) DatabaseName 

func (_m *MockSegment) DatabaseName() string

DatabaseName provides a mock function with no fields

func (*MockSegment) Delete 

func (_m *MockSegment) Delete(ctx context.Context, primaryKeys storage.PrimaryKeys, timestamps []uint64) error

Delete provides a mock function with given fields: ctx, primaryKeys, timestamps

func (*MockSegment) DropIndex 

func (_m *MockSegment) DropIndex(ctx context.Context, indexID int64) error

DropIndex provides a mock function with given fields: ctx, indexID

func (*MockSegment) EXPECT 

func (_m *MockSegment) EXPECT() *MockSegment_Expecter

func (*MockSegment) ExistIndex 

func (_m *MockSegment) ExistIndex(fieldID int64) bool

ExistIndex provides a mock function with given fields: fieldID

func (*MockSegment) FinishLoad 

func (_m *MockSegment) FinishLoad() error

FinishLoad provides a mock function with no fields

func (*MockSegment) GetBM25Stats 

func (_m *MockSegment) GetBM25Stats() map[int64]*storage.BM25Stats

GetBM25Stats provides a mock function with no fields

func (*MockSegment) GetFieldJSONIndexStats 

func (_m *MockSegment) GetFieldJSONIndexStats() map[int64]*querypb.JsonStatsInfo

GetFieldJSONIndexStats provides a mock function with no fields

func (*MockSegment) GetIndex 

func (_m *MockSegment) GetIndex(fieldID int64) []*IndexedFieldInfo

GetIndex provides a mock function with given fields: fieldID

func (*MockSegment) GetIndexByID 

func (_m *MockSegment) GetIndexByID(indexID int64) *IndexedFieldInfo

GetIndexByID provides a mock function with given fields: indexID

func (*MockSegment) GetMaxPk 

func (_m *MockSegment) GetMaxPk() *storage.PrimaryKey

GetMaxPk provides a mock function with no fields

func (*MockSegment) GetMinPk 

func (_m *MockSegment) GetMinPk() *storage.PrimaryKey

GetMinPk provides a mock function with no fields

func (*MockSegment) HasRawData 

func (_m *MockSegment) HasRawData(fieldID int64) bool

HasRawData provides a mock function with given fields: fieldID

func (*MockSegment) ID 

func (_m *MockSegment) ID() int64

ID provides a mock function with no fields

func (*MockSegment) Indexes 

func (_m *MockSegment) Indexes() []*IndexedFieldInfo

Indexes provides a mock function with no fields

func (*MockSegment) Insert 

func (_m *MockSegment) Insert(ctx context.Context, rowIDs []int64, timestamps []uint64, record *segcorepb.InsertRecord) error

Insert provides a mock function with given fields: ctx, rowIDs, timestamps, record

func (*MockSegment) InsertCount 

func (_m *MockSegment) InsertCount() int64

InsertCount provides a mock function with no fields

func (*MockSegment) IsLazyLoad 

func (_m *MockSegment) IsLazyLoad() bool

IsLazyLoad provides a mock function with no fields

func (*MockSegment) IsSorted 

func (_m *MockSegment) IsSorted() bool

IsSorted provides a mock function with no fields

func (*MockSegment) LastDeltaTimestamp 

func (_m *MockSegment) LastDeltaTimestamp() uint64

LastDeltaTimestamp provides a mock function with no fields

func (*MockSegment) Level 

func (_m *MockSegment) Level() datapb.SegmentLevel

Level provides a mock function with no fields

func (*MockSegment) LoadDeltaData 

func (_m *MockSegment) LoadDeltaData(ctx context.Context, deltaData *storage.DeltaData) error

LoadDeltaData provides a mock function with given fields: ctx, deltaData

func (*MockSegment) LoadInfo 

func (_m *MockSegment) LoadInfo() *querypb.SegmentLoadInfo

LoadInfo provides a mock function with no fields

func (*MockSegment) MayPkExist 

func (_m *MockSegment) MayPkExist(lc *storage.LocationsCache) bool

MayPkExist provides a mock function with given fields: lc

func (*MockSegment) MemSize 

func (_m *MockSegment) MemSize() int64

MemSize provides a mock function with no fields

func (*MockSegment) NeedUpdatedVersion 

func (_m *MockSegment) NeedUpdatedVersion() int64

NeedUpdatedVersion provides a mock function with no fields

func (*MockSegment) Partition 

func (_m *MockSegment) Partition() int64

Partition provides a mock function with no fields

func (*MockSegment) PinIfNotReleased 

func (_m *MockSegment) PinIfNotReleased() error

PinIfNotReleased provides a mock function with no fields

func (*MockSegment) Release 

func (_m *MockSegment) Release(ctx context.Context, opts ...releaseOption)

Release provides a mock function with given fields: ctx, opts

func (*MockSegment) RemoveUnusedFieldFiles 

func (_m *MockSegment) RemoveUnusedFieldFiles() error

RemoveUnusedFieldFiles provides a mock function with no fields

func (*MockSegment) ResetIndexesLazyLoad 

func (_m *MockSegment) ResetIndexesLazyLoad(lazyState bool)

ResetIndexesLazyLoad provides a mock function with given fields: lazyState

func (*MockSegment) ResourceGroup 

func (_m *MockSegment) ResourceGroup() string

ResourceGroup provides a mock function with no fields

func (*MockSegment) ResourceUsageEstimate 

func (_m *MockSegment) ResourceUsageEstimate() ResourceUsage

ResourceUsageEstimate provides a mock function with no fields

func (*MockSegment) Retrieve 

func (_m *MockSegment) Retrieve(ctx context.Context, plan *segcore.RetrievePlan) (*segcorepb.RetrieveResults, error)

Retrieve provides a mock function with given fields: ctx, plan

func (*MockSegment) RetrieveByOffsets 

func (_m *MockSegment) RetrieveByOffsets(ctx context.Context, plan *segcore.RetrievePlanWithOffsets) (*segcorepb.RetrieveResults, error)

RetrieveByOffsets provides a mock function with given fields: ctx, plan

func (*MockSegment) RowNum 

func (_m *MockSegment) RowNum() int64

RowNum provides a mock function with no fields

func (*MockSegment) Search 

func (_m *MockSegment) Search(ctx context.Context, searchReq *segcore.SearchRequest) (*segcore.SearchResult, error)

Search provides a mock function with given fields: ctx, searchReq

func (*MockSegment) SetBloomFilter 

func (_m *MockSegment) SetBloomFilter(bf *pkoracle.BloomFilterSet)

SetBloomFilter provides a mock function with given fields: bf

func (*MockSegment) Shard 

func (_m *MockSegment) Shard() metautil.Channel

Shard provides a mock function with no fields

func (*MockSegment) StartPosition 

func (_m *MockSegment) StartPosition() *msgpb.MsgPosition

StartPosition provides a mock function with no fields

func (*MockSegment) Type 

func (_m *MockSegment) Type() commonpb.SegmentState

Type provides a mock function with no fields

func (*MockSegment) Unpin 

func (_m *MockSegment) Unpin()

Unpin provides a mock function with no fields

func (*MockSegment) UpdateBM25Stats 

func (_m *MockSegment) UpdateBM25Stats(stats map[int64]*storage.BM25Stats)

UpdateBM25Stats provides a mock function with given fields: stats

func (*MockSegment) UpdateBloomFilter 

func (_m *MockSegment) UpdateBloomFilter(pks []storage.PrimaryKey)

UpdateBloomFilter provides a mock function with given fields: pks

func (*MockSegment) Version 

func (_m *MockSegment) Version() int64

Version provides a mock function with no fields

type MockSegmentManager 

type MockSegmentManager struct {
	mock.Mock
}

MockSegmentManager is an autogenerated mock type for the SegmentManager type

func NewMockSegmentManager 

func NewMockSegmentManager(t interface {
	mock.TestingT
	Cleanup(func())
}) *MockSegmentManager

NewMockSegmentManager creates a new instance of MockSegmentManager. 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 (*MockSegmentManager) AddLogicalResource 

func (_m *MockSegmentManager) AddLogicalResource(usage ResourceUsage)

AddLogicalResource provides a mock function with given fields: usage

func (*MockSegmentManager) Clear 

func (_m *MockSegmentManager) Clear(ctx context.Context)

Clear provides a mock function with given fields: ctx

func (*MockSegmentManager) EXPECT 

func (_m *MockSegmentManager) EXPECT() *MockSegmentManager_Expecter

func (*MockSegmentManager) Empty 

func (_m *MockSegmentManager) Empty() bool

Empty provides a mock function with no fields

func (*MockSegmentManager) Exist 

func (_m *MockSegmentManager) Exist(segmentID int64, typ commonpb.SegmentState) bool

Exist provides a mock function with given fields: segmentID, typ

func (*MockSegmentManager) Get 

func (_m *MockSegmentManager) Get(segmentID int64) Segment

Get provides a mock function with given fields: segmentID

func (*MockSegmentManager) GetAndPin 

func (_m *MockSegmentManager) GetAndPin(_a0 []int64, filters ...SegmentFilter) ([]Segment, error)

GetAndPin provides a mock function with given fields: _a0, filters

func (*MockSegmentManager) GetAndPinBy 

func (_m *MockSegmentManager) GetAndPinBy(filters ...SegmentFilter) ([]Segment, error)

GetAndPinBy provides a mock function with given fields: filters

func (*MockSegmentManager) GetBy 

func (_m *MockSegmentManager) GetBy(filters ...SegmentFilter) []Segment

GetBy provides a mock function with given fields: filters

func (*MockSegmentManager) GetGrowing 

func (_m *MockSegmentManager) GetGrowing(segmentID int64) Segment

GetGrowing provides a mock function with given fields: segmentID

func (*MockSegmentManager) GetLogicalResource 

func (_m *MockSegmentManager) GetLogicalResource() ResourceUsage

GetLogicalResource provides a mock function with no fields

func (*MockSegmentManager) GetSealed 

func (_m *MockSegmentManager) GetSealed(segmentID int64) Segment

GetSealed provides a mock function with given fields: segmentID

func (*MockSegmentManager) GetWithType 

func (_m *MockSegmentManager) GetWithType(segmentID int64, typ commonpb.SegmentState) Segment

GetWithType provides a mock function with given fields: segmentID, typ

func (*MockSegmentManager) Put 

func (_m *MockSegmentManager) Put(ctx context.Context, segmentType commonpb.SegmentState, _a2 ...Segment)

Put provides a mock function with given fields: ctx, segmentType, _a2

func (*MockSegmentManager) Remove 

func (_m *MockSegmentManager) Remove(ctx context.Context, segmentID int64, scope querypb.DataScope) (int, int)

Remove provides a mock function with given fields: ctx, segmentID, scope

func (*MockSegmentManager) RemoveBy 

func (_m *MockSegmentManager) RemoveBy(ctx context.Context, filters ...SegmentFilter) (int, int)

RemoveBy provides a mock function with given fields: ctx, filters

func (*MockSegmentManager) SubLogicalResource 

func (_m *MockSegmentManager) SubLogicalResource(usage ResourceUsage)

SubLogicalResource provides a mock function with given fields: usage

func (*MockSegmentManager) Unpin 

func (_m *MockSegmentManager) Unpin(_a0 []Segment)

Unpin provides a mock function with given fields: _a0

func (*MockSegmentManager) UpdateBy 

func (_m *MockSegmentManager) UpdateBy(action SegmentAction, filters ...SegmentFilter) int

UpdateBy provides a mock function with given fields: action, filters

type MockSegmentManager_AddLogicalResource_Call 

type MockSegmentManager_AddLogicalResource_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_AddLogicalResource_Call) Return 

func (_c *MockSegmentManager_AddLogicalResource_Call) Return() *MockSegmentManager_AddLogicalResource_Call

func (*MockSegmentManager_AddLogicalResource_Call) Run 

func (_c *MockSegmentManager_AddLogicalResource_Call) Run(run func(usage ResourceUsage)) *MockSegmentManager_AddLogicalResource_Call

func (*MockSegmentManager_AddLogicalResource_Call) RunAndReturn 

func (_c *MockSegmentManager_AddLogicalResource_Call) RunAndReturn(run func(ResourceUsage)) *MockSegmentManager_AddLogicalResource_Call

type MockSegmentManager_Clear_Call 

type MockSegmentManager_Clear_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_Clear_Call) Return 

func (_c *MockSegmentManager_Clear_Call) Return() *MockSegmentManager_Clear_Call

func (*MockSegmentManager_Clear_Call) Run 

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

func (*MockSegmentManager_Clear_Call) RunAndReturn 

func (_c *MockSegmentManager_Clear_Call) RunAndReturn(run func(context.Context)) *MockSegmentManager_Clear_Call

type MockSegmentManager_Empty_Call 

type MockSegmentManager_Empty_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_Empty_Call) Return 

func (_c *MockSegmentManager_Empty_Call) Return(_a0 bool) *MockSegmentManager_Empty_Call

func (*MockSegmentManager_Empty_Call) Run 

func (_c *MockSegmentManager_Empty_Call) Run(run func()) *MockSegmentManager_Empty_Call

func (*MockSegmentManager_Empty_Call) RunAndReturn 

func (_c *MockSegmentManager_Empty_Call) RunAndReturn(run func() bool) *MockSegmentManager_Empty_Call

type MockSegmentManager_Exist_Call 

type MockSegmentManager_Exist_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_Exist_Call) Return 

func (_c *MockSegmentManager_Exist_Call) Return(_a0 bool) *MockSegmentManager_Exist_Call

func (*MockSegmentManager_Exist_Call) Run 

func (_c *MockSegmentManager_Exist_Call) Run(run func(segmentID int64, typ commonpb.SegmentState)) *MockSegmentManager_Exist_Call

func (*MockSegmentManager_Exist_Call) RunAndReturn 

func (_c *MockSegmentManager_Exist_Call) RunAndReturn(run func(int64, commonpb.SegmentState) bool) *MockSegmentManager_Exist_Call

type MockSegmentManager_Expecter 

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

func (*MockSegmentManager_Expecter) AddLogicalResource 

func (_e *MockSegmentManager_Expecter) AddLogicalResource(usage interface{}) *MockSegmentManager_AddLogicalResource_Call

AddLogicalResource is a helper method to define mock.On call

  • usage ResourceUsage

func (*MockSegmentManager_Expecter) Clear 

func (_e *MockSegmentManager_Expecter) Clear(ctx interface{}) *MockSegmentManager_Clear_Call

Clear is a helper method to define mock.On call

  • ctx context.Context

func (*MockSegmentManager_Expecter) Empty 

func (_e *MockSegmentManager_Expecter) Empty() *MockSegmentManager_Empty_Call

Empty is a helper method to define mock.On call

func (*MockSegmentManager_Expecter) Exist 

func (_e *MockSegmentManager_Expecter) Exist(segmentID interface{}, typ interface{}) *MockSegmentManager_Exist_Call

Exist is a helper method to define mock.On call

  • segmentID int64
  • typ commonpb.SegmentState

func (*MockSegmentManager_Expecter) Get 

func (_e *MockSegmentManager_Expecter) Get(segmentID interface{}) *MockSegmentManager_Get_Call

Get is a helper method to define mock.On call

  • segmentID int64

func (*MockSegmentManager_Expecter) GetAndPin 

func (_e *MockSegmentManager_Expecter) GetAndPin(_a0 interface{}, filters ...interface{}) *MockSegmentManager_GetAndPin_Call

GetAndPin is a helper method to define mock.On call

  • _a0 []int64
  • filters ...SegmentFilter

func (*MockSegmentManager_Expecter) GetAndPinBy 

func (_e *MockSegmentManager_Expecter) GetAndPinBy(filters ...interface{}) *MockSegmentManager_GetAndPinBy_Call

GetAndPinBy is a helper method to define mock.On call

  • filters ...SegmentFilter

func (*MockSegmentManager_Expecter) GetBy 

func (_e *MockSegmentManager_Expecter) GetBy(filters ...interface{}) *MockSegmentManager_GetBy_Call

GetBy is a helper method to define mock.On call

  • filters ...SegmentFilter

func (*MockSegmentManager_Expecter) GetGrowing 

func (_e *MockSegmentManager_Expecter) GetGrowing(segmentID interface{}) *MockSegmentManager_GetGrowing_Call

GetGrowing is a helper method to define mock.On call

  • segmentID int64

func (*MockSegmentManager_Expecter) GetLogicalResource 

func (_e *MockSegmentManager_Expecter) GetLogicalResource() *MockSegmentManager_GetLogicalResource_Call

GetLogicalResource is a helper method to define mock.On call

func (*MockSegmentManager_Expecter) GetSealed 

func (_e *MockSegmentManager_Expecter) GetSealed(segmentID interface{}) *MockSegmentManager_GetSealed_Call

GetSealed is a helper method to define mock.On call

  • segmentID int64

func (*MockSegmentManager_Expecter) GetWithType 

func (_e *MockSegmentManager_Expecter) GetWithType(segmentID interface{}, typ interface{}) *MockSegmentManager_GetWithType_Call

GetWithType is a helper method to define mock.On call

  • segmentID int64
  • typ commonpb.SegmentState

func (*MockSegmentManager_Expecter) Put 

func (_e *MockSegmentManager_Expecter) Put(ctx interface{}, segmentType interface{}, _a2 ...interface{}) *MockSegmentManager_Put_Call

Put is a helper method to define mock.On call

  • ctx context.Context
  • segmentType commonpb.SegmentState
  • _a2 ...Segment

func (*MockSegmentManager_Expecter) Remove 

func (_e *MockSegmentManager_Expecter) Remove(ctx interface{}, segmentID interface{}, scope interface{}) *MockSegmentManager_Remove_Call

Remove is a helper method to define mock.On call

  • ctx context.Context
  • segmentID int64
  • scope querypb.DataScope

func (*MockSegmentManager_Expecter) RemoveBy 

func (_e *MockSegmentManager_Expecter) RemoveBy(ctx interface{}, filters ...interface{}) *MockSegmentManager_RemoveBy_Call

RemoveBy is a helper method to define mock.On call

  • ctx context.Context
  • filters ...SegmentFilter

func (*MockSegmentManager_Expecter) SubLogicalResource 

func (_e *MockSegmentManager_Expecter) SubLogicalResource(usage interface{}) *MockSegmentManager_SubLogicalResource_Call

SubLogicalResource is a helper method to define mock.On call

  • usage ResourceUsage

func (*MockSegmentManager_Expecter) Unpin 

func (_e *MockSegmentManager_Expecter) Unpin(_a0 interface{}) *MockSegmentManager_Unpin_Call

Unpin is a helper method to define mock.On call

  • _a0 []Segment

func (*MockSegmentManager_Expecter) UpdateBy 

func (_e *MockSegmentManager_Expecter) UpdateBy(action interface{}, filters ...interface{}) *MockSegmentManager_UpdateBy_Call

UpdateBy is a helper method to define mock.On call

  • action SegmentAction
  • filters ...SegmentFilter

type MockSegmentManager_GetAndPinBy_Call 

type MockSegmentManager_GetAndPinBy_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_GetAndPinBy_Call) Return 

func (_c *MockSegmentManager_GetAndPinBy_Call) Return(_a0 []Segment, _a1 error) *MockSegmentManager_GetAndPinBy_Call

func (*MockSegmentManager_GetAndPinBy_Call) Run 

func (_c *MockSegmentManager_GetAndPinBy_Call) Run(run func(filters ...SegmentFilter)) *MockSegmentManager_GetAndPinBy_Call

func (*MockSegmentManager_GetAndPinBy_Call) RunAndReturn 

func (_c *MockSegmentManager_GetAndPinBy_Call) RunAndReturn(run func(...SegmentFilter) ([]Segment, error)) *MockSegmentManager_GetAndPinBy_Call

type MockSegmentManager_GetAndPin_Call 

type MockSegmentManager_GetAndPin_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_GetAndPin_Call) Return 

func (_c *MockSegmentManager_GetAndPin_Call) Return(_a0 []Segment, _a1 error) *MockSegmentManager_GetAndPin_Call

func (*MockSegmentManager_GetAndPin_Call) Run 

func (_c *MockSegmentManager_GetAndPin_Call) Run(run func(_a0 []int64, filters ...SegmentFilter)) *MockSegmentManager_GetAndPin_Call

func (*MockSegmentManager_GetAndPin_Call) RunAndReturn 

func (_c *MockSegmentManager_GetAndPin_Call) RunAndReturn(run func([]int64, ...SegmentFilter) ([]Segment, error)) *MockSegmentManager_GetAndPin_Call

type MockSegmentManager_GetBy_Call 

type MockSegmentManager_GetBy_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_GetBy_Call) Return 

func (_c *MockSegmentManager_GetBy_Call) Return(_a0 []Segment) *MockSegmentManager_GetBy_Call

func (*MockSegmentManager_GetBy_Call) Run 

func (_c *MockSegmentManager_GetBy_Call) Run(run func(filters ...SegmentFilter)) *MockSegmentManager_GetBy_Call

func (*MockSegmentManager_GetBy_Call) RunAndReturn 

func (_c *MockSegmentManager_GetBy_Call) RunAndReturn(run func(...SegmentFilter) []Segment) *MockSegmentManager_GetBy_Call

type MockSegmentManager_GetGrowing_Call 

type MockSegmentManager_GetGrowing_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_GetGrowing_Call) Return 

func (_c *MockSegmentManager_GetGrowing_Call) Return(_a0 Segment) *MockSegmentManager_GetGrowing_Call

func (*MockSegmentManager_GetGrowing_Call) Run 

func (_c *MockSegmentManager_GetGrowing_Call) Run(run func(segmentID int64)) *MockSegmentManager_GetGrowing_Call

func (*MockSegmentManager_GetGrowing_Call) RunAndReturn 

func (_c *MockSegmentManager_GetGrowing_Call) RunAndReturn(run func(int64) Segment) *MockSegmentManager_GetGrowing_Call

type MockSegmentManager_GetLogicalResource_Call 

type MockSegmentManager_GetLogicalResource_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_GetLogicalResource_Call) Return 

func (_c *MockSegmentManager_GetLogicalResource_Call) Return(_a0 ResourceUsage) *MockSegmentManager_GetLogicalResource_Call

func (*MockSegmentManager_GetLogicalResource_Call) Run 

func (_c *MockSegmentManager_GetLogicalResource_Call) Run(run func()) *MockSegmentManager_GetLogicalResource_Call

func (*MockSegmentManager_GetLogicalResource_Call) RunAndReturn 

func (_c *MockSegmentManager_GetLogicalResource_Call) RunAndReturn(run func() ResourceUsage) *MockSegmentManager_GetLogicalResource_Call

type MockSegmentManager_GetSealed_Call 

type MockSegmentManager_GetSealed_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_GetSealed_Call) Return 

func (_c *MockSegmentManager_GetSealed_Call) Return(_a0 Segment) *MockSegmentManager_GetSealed_Call

func (*MockSegmentManager_GetSealed_Call) Run 

func (_c *MockSegmentManager_GetSealed_Call) Run(run func(segmentID int64)) *MockSegmentManager_GetSealed_Call

func (*MockSegmentManager_GetSealed_Call) RunAndReturn 

func (_c *MockSegmentManager_GetSealed_Call) RunAndReturn(run func(int64) Segment) *MockSegmentManager_GetSealed_Call

type MockSegmentManager_GetWithType_Call 

type MockSegmentManager_GetWithType_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_GetWithType_Call) Return 

func (_c *MockSegmentManager_GetWithType_Call) Return(_a0 Segment) *MockSegmentManager_GetWithType_Call

func (*MockSegmentManager_GetWithType_Call) Run 

func (_c *MockSegmentManager_GetWithType_Call) Run(run func(segmentID int64, typ commonpb.SegmentState)) *MockSegmentManager_GetWithType_Call

func (*MockSegmentManager_GetWithType_Call) RunAndReturn 

func (_c *MockSegmentManager_GetWithType_Call) RunAndReturn(run func(int64, commonpb.SegmentState) Segment) *MockSegmentManager_GetWithType_Call

type MockSegmentManager_Get_Call 

type MockSegmentManager_Get_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_Get_Call) Return 

func (_c *MockSegmentManager_Get_Call) Return(_a0 Segment) *MockSegmentManager_Get_Call

func (*MockSegmentManager_Get_Call) Run 

func (_c *MockSegmentManager_Get_Call) Run(run func(segmentID int64)) *MockSegmentManager_Get_Call

func (*MockSegmentManager_Get_Call) RunAndReturn 

func (_c *MockSegmentManager_Get_Call) RunAndReturn(run func(int64) Segment) *MockSegmentManager_Get_Call

type MockSegmentManager_Put_Call 

type MockSegmentManager_Put_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_Put_Call) Return 

func (_c *MockSegmentManager_Put_Call) Return() *MockSegmentManager_Put_Call

func (*MockSegmentManager_Put_Call) Run 

func (_c *MockSegmentManager_Put_Call) Run(run func(ctx context.Context, segmentType commonpb.SegmentState, _a2 ...Segment)) *MockSegmentManager_Put_Call

func (*MockSegmentManager_Put_Call) RunAndReturn 

func (_c *MockSegmentManager_Put_Call) RunAndReturn(run func(context.Context, commonpb.SegmentState, ...Segment)) *MockSegmentManager_Put_Call

type MockSegmentManager_RemoveBy_Call 

type MockSegmentManager_RemoveBy_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_RemoveBy_Call) Return 

func (_c *MockSegmentManager_RemoveBy_Call) Return(_a0 int, _a1 int) *MockSegmentManager_RemoveBy_Call

func (*MockSegmentManager_RemoveBy_Call) Run 

func (_c *MockSegmentManager_RemoveBy_Call) Run(run func(ctx context.Context, filters ...SegmentFilter)) *MockSegmentManager_RemoveBy_Call

func (*MockSegmentManager_RemoveBy_Call) RunAndReturn 

func (_c *MockSegmentManager_RemoveBy_Call) RunAndReturn(run func(context.Context, ...SegmentFilter) (int, int)) *MockSegmentManager_RemoveBy_Call

type MockSegmentManager_Remove_Call 

type MockSegmentManager_Remove_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_Remove_Call) Return 

func (_c *MockSegmentManager_Remove_Call) Return(_a0 int, _a1 int) *MockSegmentManager_Remove_Call

func (*MockSegmentManager_Remove_Call) Run 

func (_c *MockSegmentManager_Remove_Call) Run(run func(ctx context.Context, segmentID int64, scope querypb.DataScope)) *MockSegmentManager_Remove_Call

func (*MockSegmentManager_Remove_Call) RunAndReturn 

func (_c *MockSegmentManager_Remove_Call) RunAndReturn(run func(context.Context, int64, querypb.DataScope) (int, int)) *MockSegmentManager_Remove_Call

type MockSegmentManager_SubLogicalResource_Call 

type MockSegmentManager_SubLogicalResource_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_SubLogicalResource_Call) Return 

func (_c *MockSegmentManager_SubLogicalResource_Call) Return() *MockSegmentManager_SubLogicalResource_Call

func (*MockSegmentManager_SubLogicalResource_Call) Run 

func (_c *MockSegmentManager_SubLogicalResource_Call) Run(run func(usage ResourceUsage)) *MockSegmentManager_SubLogicalResource_Call

func (*MockSegmentManager_SubLogicalResource_Call) RunAndReturn 

func (_c *MockSegmentManager_SubLogicalResource_Call) RunAndReturn(run func(ResourceUsage)) *MockSegmentManager_SubLogicalResource_Call

type MockSegmentManager_Unpin_Call 

type MockSegmentManager_Unpin_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_Unpin_Call) Return 

func (_c *MockSegmentManager_Unpin_Call) Return() *MockSegmentManager_Unpin_Call

func (*MockSegmentManager_Unpin_Call) Run 

func (_c *MockSegmentManager_Unpin_Call) Run(run func(_a0 []Segment)) *MockSegmentManager_Unpin_Call

func (*MockSegmentManager_Unpin_Call) RunAndReturn 

func (_c *MockSegmentManager_Unpin_Call) RunAndReturn(run func([]Segment)) *MockSegmentManager_Unpin_Call

type MockSegmentManager_UpdateBy_Call 

type MockSegmentManager_UpdateBy_Call struct {
	*mock.Call
}

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

func (*MockSegmentManager_UpdateBy_Call) Return 

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

func (*MockSegmentManager_UpdateBy_Call) Run 

func (_c *MockSegmentManager_UpdateBy_Call) Run(run func(action SegmentAction, filters ...SegmentFilter)) *MockSegmentManager_UpdateBy_Call

func (*MockSegmentManager_UpdateBy_Call) RunAndReturn 

func (_c *MockSegmentManager_UpdateBy_Call) RunAndReturn(run func(SegmentAction, ...SegmentFilter) int) *MockSegmentManager_UpdateBy_Call

type MockSegment_BatchPkExist_Call 

type MockSegment_BatchPkExist_Call struct {
	*mock.Call
}

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

func (*MockSegment_BatchPkExist_Call) Return 

func (_c *MockSegment_BatchPkExist_Call) Return(_a0 []bool) *MockSegment_BatchPkExist_Call

func (*MockSegment_BatchPkExist_Call) Run 

func (_c *MockSegment_BatchPkExist_Call) Run(run func(lc *storage.BatchLocationsCache)) *MockSegment_BatchPkExist_Call

func (*MockSegment_BatchPkExist_Call) RunAndReturn 

func (_c *MockSegment_BatchPkExist_Call) RunAndReturn(run func(*storage.BatchLocationsCache) []bool) *MockSegment_BatchPkExist_Call

type MockSegment_BloomFilterExist_Call 

type MockSegment_BloomFilterExist_Call struct {
	*mock.Call
}

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

func (*MockSegment_BloomFilterExist_Call) Return 

func (_c *MockSegment_BloomFilterExist_Call) Return(_a0 bool) *MockSegment_BloomFilterExist_Call

func (*MockSegment_BloomFilterExist_Call) Run 

func (_c *MockSegment_BloomFilterExist_Call) Run(run func()) *MockSegment_BloomFilterExist_Call

func (*MockSegment_BloomFilterExist_Call) RunAndReturn 

func (_c *MockSegment_BloomFilterExist_Call) RunAndReturn(run func() bool) *MockSegment_BloomFilterExist_Call

type MockSegment_CASVersion_Call 

type MockSegment_CASVersion_Call struct {
	*mock.Call
}

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

func (*MockSegment_CASVersion_Call) Return 

func (_c *MockSegment_CASVersion_Call) Return(_a0 bool) *MockSegment_CASVersion_Call

func (*MockSegment_CASVersion_Call) Run 

func (_c *MockSegment_CASVersion_Call) Run(run func(_a0 int64, _a1 int64)) *MockSegment_CASVersion_Call

func (*MockSegment_CASVersion_Call) RunAndReturn 

func (_c *MockSegment_CASVersion_Call) RunAndReturn(run func(int64, int64) bool) *MockSegment_CASVersion_Call

type MockSegment_Collection_Call 

type MockSegment_Collection_Call struct {
	*mock.Call
}

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

func (*MockSegment_Collection_Call) Return 

func (_c *MockSegment_Collection_Call) Return(_a0 int64) *MockSegment_Collection_Call

func (*MockSegment_Collection_Call) Run 

func (_c *MockSegment_Collection_Call) Run(run func()) *MockSegment_Collection_Call

func (*MockSegment_Collection_Call) RunAndReturn 

func (_c *MockSegment_Collection_Call) RunAndReturn(run func() int64) *MockSegment_Collection_Call

type MockSegment_DatabaseName_Call 

type MockSegment_DatabaseName_Call struct {
	*mock.Call
}

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

func (*MockSegment_DatabaseName_Call) Return 

func (_c *MockSegment_DatabaseName_Call) Return(_a0 string) *MockSegment_DatabaseName_Call

func (*MockSegment_DatabaseName_Call) Run 

func (_c *MockSegment_DatabaseName_Call) Run(run func()) *MockSegment_DatabaseName_Call

func (*MockSegment_DatabaseName_Call) RunAndReturn 

func (_c *MockSegment_DatabaseName_Call) RunAndReturn(run func() string) *MockSegment_DatabaseName_Call

type MockSegment_Delete_Call 

type MockSegment_Delete_Call struct {
	*mock.Call
}

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

func (*MockSegment_Delete_Call) Return 

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

func (*MockSegment_Delete_Call) Run 

func (_c *MockSegment_Delete_Call) Run(run func(ctx context.Context, primaryKeys storage.PrimaryKeys, timestamps []uint64)) *MockSegment_Delete_Call

func (*MockSegment_Delete_Call) RunAndReturn 

func (_c *MockSegment_Delete_Call) RunAndReturn(run func(context.Context, storage.PrimaryKeys, []uint64) error) *MockSegment_Delete_Call

type MockSegment_DropIndex_Call 

type MockSegment_DropIndex_Call struct {
	*mock.Call
}

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

func (*MockSegment_DropIndex_Call) Return 

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

func (*MockSegment_DropIndex_Call) Run 

func (_c *MockSegment_DropIndex_Call) Run(run func(ctx context.Context, indexID int64)) *MockSegment_DropIndex_Call

func (*MockSegment_DropIndex_Call) RunAndReturn 

func (_c *MockSegment_DropIndex_Call) RunAndReturn(run func(context.Context, int64) error) *MockSegment_DropIndex_Call

type MockSegment_ExistIndex_Call 

type MockSegment_ExistIndex_Call struct {
	*mock.Call
}

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

func (*MockSegment_ExistIndex_Call) Return 

func (_c *MockSegment_ExistIndex_Call) Return(_a0 bool) *MockSegment_ExistIndex_Call

func (*MockSegment_ExistIndex_Call) Run 

func (_c *MockSegment_ExistIndex_Call) Run(run func(fieldID int64)) *MockSegment_ExistIndex_Call

func (*MockSegment_ExistIndex_Call) RunAndReturn 

func (_c *MockSegment_ExistIndex_Call) RunAndReturn(run func(int64) bool) *MockSegment_ExistIndex_Call

type MockSegment_Expecter 

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

func (*MockSegment_Expecter) BatchPkExist 

func (_e *MockSegment_Expecter) BatchPkExist(lc interface{}) *MockSegment_BatchPkExist_Call

BatchPkExist is a helper method to define mock.On call

  • lc *storage.BatchLocationsCache

func (*MockSegment_Expecter) BloomFilterExist 

func (_e *MockSegment_Expecter) BloomFilterExist() *MockSegment_BloomFilterExist_Call

BloomFilterExist is a helper method to define mock.On call

func (*MockSegment_Expecter) CASVersion 

func (_e *MockSegment_Expecter) CASVersion(_a0 interface{}, _a1 interface{}) *MockSegment_CASVersion_Call

CASVersion is a helper method to define mock.On call

  • _a0 int64
  • _a1 int64

func (*MockSegment_Expecter) Collection 

func (_e *MockSegment_Expecter) Collection() *MockSegment_Collection_Call

Collection is a helper method to define mock.On call

func (*MockSegment_Expecter) DatabaseName 

func (_e *MockSegment_Expecter) DatabaseName() *MockSegment_DatabaseName_Call

DatabaseName is a helper method to define mock.On call

func (*MockSegment_Expecter) Delete 

func (_e *MockSegment_Expecter) Delete(ctx interface{}, primaryKeys interface{}, timestamps interface{}) *MockSegment_Delete_Call

Delete is a helper method to define mock.On call

  • ctx context.Context
  • primaryKeys storage.PrimaryKeys
  • timestamps []uint64

func (*MockSegment_Expecter) DropIndex 

func (_e *MockSegment_Expecter) DropIndex(ctx interface{}, indexID interface{}) *MockSegment_DropIndex_Call

DropIndex is a helper method to define mock.On call

  • ctx context.Context
  • indexID int64

func (*MockSegment_Expecter) ExistIndex 

func (_e *MockSegment_Expecter) ExistIndex(fieldID interface{}) *MockSegment_ExistIndex_Call

ExistIndex is a helper method to define mock.On call

  • fieldID int64

func (*MockSegment_Expecter) FinishLoad 

func (_e *MockSegment_Expecter) FinishLoad() *MockSegment_FinishLoad_Call

FinishLoad is a helper method to define mock.On call

func (*MockSegment_Expecter) GetBM25Stats 

func (_e *MockSegment_Expecter) GetBM25Stats() *MockSegment_GetBM25Stats_Call

GetBM25Stats is a helper method to define mock.On call

func (*MockSegment_Expecter) GetFieldJSONIndexStats 

func (_e *MockSegment_Expecter) GetFieldJSONIndexStats() *MockSegment_GetFieldJSONIndexStats_Call

GetFieldJSONIndexStats is a helper method to define mock.On call

func (*MockSegment_Expecter) GetIndex 

func (_e *MockSegment_Expecter) GetIndex(fieldID interface{}) *MockSegment_GetIndex_Call

GetIndex is a helper method to define mock.On call

  • fieldID int64

func (*MockSegment_Expecter) GetIndexByID 

func (_e *MockSegment_Expecter) GetIndexByID(indexID interface{}) *MockSegment_GetIndexByID_Call

GetIndexByID is a helper method to define mock.On call

  • indexID int64

func (*MockSegment_Expecter) GetMaxPk 

func (_e *MockSegment_Expecter) GetMaxPk() *MockSegment_GetMaxPk_Call

GetMaxPk is a helper method to define mock.On call

func (*MockSegment_Expecter) GetMinPk 

func (_e *MockSegment_Expecter) GetMinPk() *MockSegment_GetMinPk_Call

GetMinPk is a helper method to define mock.On call

func (*MockSegment_Expecter) HasRawData 

func (_e *MockSegment_Expecter) HasRawData(fieldID interface{}) *MockSegment_HasRawData_Call

HasRawData is a helper method to define mock.On call

  • fieldID int64

func (*MockSegment_Expecter) ID 

func (_e *MockSegment_Expecter) ID() *MockSegment_ID_Call

ID is a helper method to define mock.On call

func (*MockSegment_Expecter) Indexes 

func (_e *MockSegment_Expecter) Indexes() *MockSegment_Indexes_Call

Indexes is a helper method to define mock.On call

func (*MockSegment_Expecter) Insert 

func (_e *MockSegment_Expecter) Insert(ctx interface{}, rowIDs interface{}, timestamps interface{}, record interface{}) *MockSegment_Insert_Call

Insert is a helper method to define mock.On call

  • ctx context.Context
  • rowIDs []int64
  • timestamps []uint64
  • record *segcorepb.InsertRecord

func (*MockSegment_Expecter) InsertCount 

func (_e *MockSegment_Expecter) InsertCount() *MockSegment_InsertCount_Call

InsertCount is a helper method to define mock.On call

func (*MockSegment_Expecter) IsLazyLoad 

func (_e *MockSegment_Expecter) IsLazyLoad() *MockSegment_IsLazyLoad_Call

IsLazyLoad is a helper method to define mock.On call

func (*MockSegment_Expecter) IsSorted 

func (_e *MockSegment_Expecter) IsSorted() *MockSegment_IsSorted_Call

IsSorted is a helper method to define mock.On call

func (*MockSegment_Expecter) LastDeltaTimestamp 

func (_e *MockSegment_Expecter) LastDeltaTimestamp() *MockSegment_LastDeltaTimestamp_Call

LastDeltaTimestamp is a helper method to define mock.On call

func (*MockSegment_Expecter) Level 

func (_e *MockSegment_Expecter) Level() *MockSegment_Level_Call

Level is a helper method to define mock.On call

func (*MockSegment_Expecter) LoadDeltaData 

func (_e *MockSegment_Expecter) LoadDeltaData(ctx interface{}, deltaData interface{}) *MockSegment_LoadDeltaData_Call

LoadDeltaData is a helper method to define mock.On call

  • ctx context.Context
  • deltaData *storage.DeltaData

func (*MockSegment_Expecter) LoadInfo 

func (_e *MockSegment_Expecter) LoadInfo() *MockSegment_LoadInfo_Call

LoadInfo is a helper method to define mock.On call

func (*MockSegment_Expecter) MayPkExist 

func (_e *MockSegment_Expecter) MayPkExist(lc interface{}) *MockSegment_MayPkExist_Call

MayPkExist is a helper method to define mock.On call

  • lc *storage.LocationsCache

func (*MockSegment_Expecter) MemSize 

func (_e *MockSegment_Expecter) MemSize() *MockSegment_MemSize_Call

MemSize is a helper method to define mock.On call

func (*MockSegment_Expecter) NeedUpdatedVersion 

func (_e *MockSegment_Expecter) NeedUpdatedVersion() *MockSegment_NeedUpdatedVersion_Call

NeedUpdatedVersion is a helper method to define mock.On call

func (*MockSegment_Expecter) Partition 

func (_e *MockSegment_Expecter) Partition() *MockSegment_Partition_Call

Partition is a helper method to define mock.On call

func (*MockSegment_Expecter) PinIfNotReleased 

func (_e *MockSegment_Expecter) PinIfNotReleased() *MockSegment_PinIfNotReleased_Call

PinIfNotReleased is a helper method to define mock.On call

func (*MockSegment_Expecter) Release 

func (_e *MockSegment_Expecter) Release(ctx interface{}, opts ...interface{}) *MockSegment_Release_Call

Release is a helper method to define mock.On call

  • ctx context.Context
  • opts ...releaseOption

func (*MockSegment_Expecter) RemoveUnusedFieldFiles 

func (_e *MockSegment_Expecter) RemoveUnusedFieldFiles() *MockSegment_RemoveUnusedFieldFiles_Call

RemoveUnusedFieldFiles is a helper method to define mock.On call

func (*MockSegment_Expecter) ResetIndexesLazyLoad 

func (_e *MockSegment_Expecter) ResetIndexesLazyLoad(lazyState interface{}) *MockSegment_ResetIndexesLazyLoad_Call

ResetIndexesLazyLoad is a helper method to define mock.On call

  • lazyState bool

func (*MockSegment_Expecter) ResourceGroup 

func (_e *MockSegment_Expecter) ResourceGroup() *MockSegment_ResourceGroup_Call

ResourceGroup is a helper method to define mock.On call

func (*MockSegment_Expecter) ResourceUsageEstimate 

func (_e *MockSegment_Expecter) ResourceUsageEstimate() *MockSegment_ResourceUsageEstimate_Call

ResourceUsageEstimate is a helper method to define mock.On call

func (*MockSegment_Expecter) Retrieve 

func (_e *MockSegment_Expecter) Retrieve(ctx interface{}, plan interface{}) *MockSegment_Retrieve_Call

Retrieve is a helper method to define mock.On call

  • ctx context.Context
  • plan *segcore.RetrievePlan

func (*MockSegment_Expecter) RetrieveByOffsets 

func (_e *MockSegment_Expecter) RetrieveByOffsets(ctx interface{}, plan interface{}) *MockSegment_RetrieveByOffsets_Call

RetrieveByOffsets is a helper method to define mock.On call

  • ctx context.Context
  • plan *segcore.RetrievePlanWithOffsets

func (*MockSegment_Expecter) RowNum 

func (_e *MockSegment_Expecter) RowNum() *MockSegment_RowNum_Call

RowNum is a helper method to define mock.On call

func (*MockSegment_Expecter) Search 

func (_e *MockSegment_Expecter) Search(ctx interface{}, searchReq interface{}) *MockSegment_Search_Call

Search is a helper method to define mock.On call

  • ctx context.Context
  • searchReq *segcore.SearchRequest

func (*MockSegment_Expecter) SetBloomFilter 

func (_e *MockSegment_Expecter) SetBloomFilter(bf interface{}) *MockSegment_SetBloomFilter_Call

SetBloomFilter is a helper method to define mock.On call

  • bf *pkoracle.BloomFilterSet

func (*MockSegment_Expecter) Shard 

func (_e *MockSegment_Expecter) Shard() *MockSegment_Shard_Call

Shard is a helper method to define mock.On call

func (*MockSegment_Expecter) StartPosition 

func (_e *MockSegment_Expecter) StartPosition() *MockSegment_StartPosition_Call

StartPosition is a helper method to define mock.On call

func (*MockSegment_Expecter) Type 

func (_e *MockSegment_Expecter) Type() *MockSegment_Type_Call

Type is a helper method to define mock.On call

func (*MockSegment_Expecter) Unpin 

func (_e *MockSegment_Expecter) Unpin() *MockSegment_Unpin_Call

Unpin is a helper method to define mock.On call

func (*MockSegment_Expecter) UpdateBM25Stats 

func (_e *MockSegment_Expecter) UpdateBM25Stats(stats interface{}) *MockSegment_UpdateBM25Stats_Call

UpdateBM25Stats is a helper method to define mock.On call

  • stats map[int64]*storage.BM25Stats

func (*MockSegment_Expecter) UpdateBloomFilter 

func (_e *MockSegment_Expecter) UpdateBloomFilter(pks interface{}) *MockSegment_UpdateBloomFilter_Call

UpdateBloomFilter is a helper method to define mock.On call

  • pks []storage.PrimaryKey

func (*MockSegment_Expecter) Version 

func (_e *MockSegment_Expecter) Version() *MockSegment_Version_Call

Version is a helper method to define mock.On call

type MockSegment_FinishLoad_Call 

type MockSegment_FinishLoad_Call struct {
	*mock.Call
}

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

func (*MockSegment_FinishLoad_Call) Return 

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

func (*MockSegment_FinishLoad_Call) Run 

func (_c *MockSegment_FinishLoad_Call) Run(run func()) *MockSegment_FinishLoad_Call

func (*MockSegment_FinishLoad_Call) RunAndReturn 

func (_c *MockSegment_FinishLoad_Call) RunAndReturn(run func() error) *MockSegment_FinishLoad_Call

type MockSegment_GetBM25Stats_Call 

type MockSegment_GetBM25Stats_Call struct {
	*mock.Call
}

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

func (*MockSegment_GetBM25Stats_Call) Return 

func (_c *MockSegment_GetBM25Stats_Call) Return(_a0 map[int64]*storage.BM25Stats) *MockSegment_GetBM25Stats_Call

func (*MockSegment_GetBM25Stats_Call) Run 

func (_c *MockSegment_GetBM25Stats_Call) Run(run func()) *MockSegment_GetBM25Stats_Call

func (*MockSegment_GetBM25Stats_Call) RunAndReturn 

func (_c *MockSegment_GetBM25Stats_Call) RunAndReturn(run func() map[int64]*storage.BM25Stats) *MockSegment_GetBM25Stats_Call

type MockSegment_GetFieldJSONIndexStats_Call 

type MockSegment_GetFieldJSONIndexStats_Call struct {
	*mock.Call
}

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

func (*MockSegment_GetFieldJSONIndexStats_Call) Return 

func (_c *MockSegment_GetFieldJSONIndexStats_Call) Return(_a0 map[int64]*querypb.JsonStatsInfo) *MockSegment_GetFieldJSONIndexStats_Call

func (*MockSegment_GetFieldJSONIndexStats_Call) Run 

func (_c *MockSegment_GetFieldJSONIndexStats_Call) Run(run func()) *MockSegment_GetFieldJSONIndexStats_Call

func (*MockSegment_GetFieldJSONIndexStats_Call) RunAndReturn 

func (_c *MockSegment_GetFieldJSONIndexStats_Call) RunAndReturn(run func() map[int64]*querypb.JsonStatsInfo) *MockSegment_GetFieldJSONIndexStats_Call

type MockSegment_GetIndexByID_Call 

type MockSegment_GetIndexByID_Call struct {
	*mock.Call
}

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

func (*MockSegment_GetIndexByID_Call) Return 

func (_c *MockSegment_GetIndexByID_Call) Return(_a0 *IndexedFieldInfo) *MockSegment_GetIndexByID_Call

func (*MockSegment_GetIndexByID_Call) Run 

func (_c *MockSegment_GetIndexByID_Call) Run(run func(indexID int64)) *MockSegment_GetIndexByID_Call

func (*MockSegment_GetIndexByID_Call) RunAndReturn 

func (_c *MockSegment_GetIndexByID_Call) RunAndReturn(run func(int64) *IndexedFieldInfo) *MockSegment_GetIndexByID_Call

type MockSegment_GetIndex_Call 

type MockSegment_GetIndex_Call struct {
	*mock.Call
}

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

func (*MockSegment_GetIndex_Call) Return 

func (_c *MockSegment_GetIndex_Call) Return(_a0 []*IndexedFieldInfo) *MockSegment_GetIndex_Call

func (*MockSegment_GetIndex_Call) Run 

func (_c *MockSegment_GetIndex_Call) Run(run func(fieldID int64)) *MockSegment_GetIndex_Call

func (*MockSegment_GetIndex_Call) RunAndReturn 

func (_c *MockSegment_GetIndex_Call) RunAndReturn(run func(int64) []*IndexedFieldInfo) *MockSegment_GetIndex_Call

type MockSegment_GetMaxPk_Call 

type MockSegment_GetMaxPk_Call struct {
	*mock.Call
}

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

func (*MockSegment_GetMaxPk_Call) Return 

func (_c *MockSegment_GetMaxPk_Call) Return(_a0 *storage.PrimaryKey) *MockSegment_GetMaxPk_Call

func (*MockSegment_GetMaxPk_Call) Run 

func (_c *MockSegment_GetMaxPk_Call) Run(run func()) *MockSegment_GetMaxPk_Call

func (*MockSegment_GetMaxPk_Call) RunAndReturn 

func (_c *MockSegment_GetMaxPk_Call) RunAndReturn(run func() *storage.PrimaryKey) *MockSegment_GetMaxPk_Call

type MockSegment_GetMinPk_Call 

type MockSegment_GetMinPk_Call struct {
	*mock.Call
}

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

func (*MockSegment_GetMinPk_Call) Return 

func (_c *MockSegment_GetMinPk_Call) Return(_a0 *storage.PrimaryKey) *MockSegment_GetMinPk_Call

func (*MockSegment_GetMinPk_Call) Run 

func (_c *MockSegment_GetMinPk_Call) Run(run func()) *MockSegment_GetMinPk_Call

func (*MockSegment_GetMinPk_Call) RunAndReturn 

func (_c *MockSegment_GetMinPk_Call) RunAndReturn(run func() *storage.PrimaryKey) *MockSegment_GetMinPk_Call

type MockSegment_HasRawData_Call 

type MockSegment_HasRawData_Call struct {
	*mock.Call
}

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

func (*MockSegment_HasRawData_Call) Return 

func (_c *MockSegment_HasRawData_Call) Return(_a0 bool) *MockSegment_HasRawData_Call

func (*MockSegment_HasRawData_Call) Run 

func (_c *MockSegment_HasRawData_Call) Run(run func(fieldID int64)) *MockSegment_HasRawData_Call

func (*MockSegment_HasRawData_Call) RunAndReturn 

func (_c *MockSegment_HasRawData_Call) RunAndReturn(run func(int64) bool) *MockSegment_HasRawData_Call

type MockSegment_ID_Call 

type MockSegment_ID_Call struct {
	*mock.Call
}

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

func (*MockSegment_ID_Call) Return 

func (_c *MockSegment_ID_Call) Return(_a0 int64) *MockSegment_ID_Call

func (*MockSegment_ID_Call) Run 

func (_c *MockSegment_ID_Call) Run(run func()) *MockSegment_ID_Call

func (*MockSegment_ID_Call) RunAndReturn 

func (_c *MockSegment_ID_Call) RunAndReturn(run func() int64) *MockSegment_ID_Call

type MockSegment_Indexes_Call 

type MockSegment_Indexes_Call struct {
	*mock.Call
}

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

func (*MockSegment_Indexes_Call) Return 

func (_c *MockSegment_Indexes_Call) Return(_a0 []*IndexedFieldInfo) *MockSegment_Indexes_Call

func (*MockSegment_Indexes_Call) Run 

func (_c *MockSegment_Indexes_Call) Run(run func()) *MockSegment_Indexes_Call

func (*MockSegment_Indexes_Call) RunAndReturn 

func (_c *MockSegment_Indexes_Call) RunAndReturn(run func() []*IndexedFieldInfo) *MockSegment_Indexes_Call

type MockSegment_InsertCount_Call 

type MockSegment_InsertCount_Call struct {
	*mock.Call
}

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

func (*MockSegment_InsertCount_Call) Return 

func (_c *MockSegment_InsertCount_Call) Return(_a0 int64) *MockSegment_InsertCount_Call

func (*MockSegment_InsertCount_Call) Run 

func (_c *MockSegment_InsertCount_Call) Run(run func()) *MockSegment_InsertCount_Call

func (*MockSegment_InsertCount_Call) RunAndReturn 

func (_c *MockSegment_InsertCount_Call) RunAndReturn(run func() int64) *MockSegment_InsertCount_Call

type MockSegment_Insert_Call 

type MockSegment_Insert_Call struct {
	*mock.Call
}

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

func (*MockSegment_Insert_Call) Return 

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

func (*MockSegment_Insert_Call) Run 

func (_c *MockSegment_Insert_Call) Run(run func(ctx context.Context, rowIDs []int64, timestamps []uint64, record *segcorepb.InsertRecord)) *MockSegment_Insert_Call

func (*MockSegment_Insert_Call) RunAndReturn 

func (_c *MockSegment_Insert_Call) RunAndReturn(run func(context.Context, []int64, []uint64, *segcorepb.InsertRecord) error) *MockSegment_Insert_Call

type MockSegment_IsLazyLoad_Call 

type MockSegment_IsLazyLoad_Call struct {
	*mock.Call
}

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

func (*MockSegment_IsLazyLoad_Call) Return 

func (_c *MockSegment_IsLazyLoad_Call) Return(_a0 bool) *MockSegment_IsLazyLoad_Call

func (*MockSegment_IsLazyLoad_Call) Run 

func (_c *MockSegment_IsLazyLoad_Call) Run(run func()) *MockSegment_IsLazyLoad_Call

func (*MockSegment_IsLazyLoad_Call) RunAndReturn 

func (_c *MockSegment_IsLazyLoad_Call) RunAndReturn(run func() bool) *MockSegment_IsLazyLoad_Call

type MockSegment_IsSorted_Call 

type MockSegment_IsSorted_Call struct {
	*mock.Call
}

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

func (*MockSegment_IsSorted_Call) Return 

func (_c *MockSegment_IsSorted_Call) Return(_a0 bool) *MockSegment_IsSorted_Call

func (*MockSegment_IsSorted_Call) Run 

func (_c *MockSegment_IsSorted_Call) Run(run func()) *MockSegment_IsSorted_Call

func (*MockSegment_IsSorted_Call) RunAndReturn 

func (_c *MockSegment_IsSorted_Call) RunAndReturn(run func() bool) *MockSegment_IsSorted_Call

type MockSegment_LastDeltaTimestamp_Call 

type MockSegment_LastDeltaTimestamp_Call struct {
	*mock.Call
}

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

func (*MockSegment_LastDeltaTimestamp_Call) Return 

func (_c *MockSegment_LastDeltaTimestamp_Call) Return(_a0 uint64) *MockSegment_LastDeltaTimestamp_Call

func (*MockSegment_LastDeltaTimestamp_Call) Run 

func (_c *MockSegment_LastDeltaTimestamp_Call) Run(run func()) *MockSegment_LastDeltaTimestamp_Call

func (*MockSegment_LastDeltaTimestamp_Call) RunAndReturn 

func (_c *MockSegment_LastDeltaTimestamp_Call) RunAndReturn(run func() uint64) *MockSegment_LastDeltaTimestamp_Call

type MockSegment_Level_Call 

type MockSegment_Level_Call struct {
	*mock.Call
}

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

func (*MockSegment_Level_Call) Return 

func (_c *MockSegment_Level_Call) Return(_a0 datapb.SegmentLevel) *MockSegment_Level_Call

func (*MockSegment_Level_Call) Run 

func (_c *MockSegment_Level_Call) Run(run func()) *MockSegment_Level_Call

func (*MockSegment_Level_Call) RunAndReturn 

func (_c *MockSegment_Level_Call) RunAndReturn(run func() datapb.SegmentLevel) *MockSegment_Level_Call

type MockSegment_LoadDeltaData_Call 

type MockSegment_LoadDeltaData_Call struct {
	*mock.Call
}

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

func (*MockSegment_LoadDeltaData_Call) Return 

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

func (*MockSegment_LoadDeltaData_Call) Run 

func (_c *MockSegment_LoadDeltaData_Call) Run(run func(ctx context.Context, deltaData *storage.DeltaData)) *MockSegment_LoadDeltaData_Call

func (*MockSegment_LoadDeltaData_Call) RunAndReturn 

func (_c *MockSegment_LoadDeltaData_Call) RunAndReturn(run func(context.Context, *storage.DeltaData) error) *MockSegment_LoadDeltaData_Call

type MockSegment_LoadInfo_Call 

type MockSegment_LoadInfo_Call struct {
	*mock.Call
}

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

func (*MockSegment_LoadInfo_Call) Return 

func (_c *MockSegment_LoadInfo_Call) Return(_a0 *querypb.SegmentLoadInfo) *MockSegment_LoadInfo_Call

func (*MockSegment_LoadInfo_Call) Run 

func (_c *MockSegment_LoadInfo_Call) Run(run func()) *MockSegment_LoadInfo_Call

func (*MockSegment_LoadInfo_Call) RunAndReturn 

func (_c *MockSegment_LoadInfo_Call) RunAndReturn(run func() *querypb.SegmentLoadInfo) *MockSegment_LoadInfo_Call

type MockSegment_MayPkExist_Call 

type MockSegment_MayPkExist_Call struct {
	*mock.Call
}

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

func (*MockSegment_MayPkExist_Call) Return 

func (_c *MockSegment_MayPkExist_Call) Return(_a0 bool) *MockSegment_MayPkExist_Call

func (*MockSegment_MayPkExist_Call) Run 

func (_c *MockSegment_MayPkExist_Call) Run(run func(lc *storage.LocationsCache)) *MockSegment_MayPkExist_Call

func (*MockSegment_MayPkExist_Call) RunAndReturn 

func (_c *MockSegment_MayPkExist_Call) RunAndReturn(run func(*storage.LocationsCache) bool) *MockSegment_MayPkExist_Call

type MockSegment_MemSize_Call 

type MockSegment_MemSize_Call struct {
	*mock.Call
}

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

func (*MockSegment_MemSize_Call) Return 

func (_c *MockSegment_MemSize_Call) Return(_a0 int64) *MockSegment_MemSize_Call

func (*MockSegment_MemSize_Call) Run 

func (_c *MockSegment_MemSize_Call) Run(run func()) *MockSegment_MemSize_Call

func (*MockSegment_MemSize_Call) RunAndReturn 

func (_c *MockSegment_MemSize_Call) RunAndReturn(run func() int64) *MockSegment_MemSize_Call

type MockSegment_NeedUpdatedVersion_Call 

type MockSegment_NeedUpdatedVersion_Call struct {
	*mock.Call
}

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

func (*MockSegment_NeedUpdatedVersion_Call) Return 

func (_c *MockSegment_NeedUpdatedVersion_Call) Return(_a0 int64) *MockSegment_NeedUpdatedVersion_Call

func (*MockSegment_NeedUpdatedVersion_Call) Run 

func (_c *MockSegment_NeedUpdatedVersion_Call) Run(run func()) *MockSegment_NeedUpdatedVersion_Call

func (*MockSegment_NeedUpdatedVersion_Call) RunAndReturn 

func (_c *MockSegment_NeedUpdatedVersion_Call) RunAndReturn(run func() int64) *MockSegment_NeedUpdatedVersion_Call

type MockSegment_Partition_Call 

type MockSegment_Partition_Call struct {
	*mock.Call
}

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

func (*MockSegment_Partition_Call) Return 

func (_c *MockSegment_Partition_Call) Return(_a0 int64) *MockSegment_Partition_Call

func (*MockSegment_Partition_Call) Run 

func (_c *MockSegment_Partition_Call) Run(run func()) *MockSegment_Partition_Call

func (*MockSegment_Partition_Call) RunAndReturn 

func (_c *MockSegment_Partition_Call) RunAndReturn(run func() int64) *MockSegment_Partition_Call

type MockSegment_PinIfNotReleased_Call 

type MockSegment_PinIfNotReleased_Call struct {
	*mock.Call
}

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

func (*MockSegment_PinIfNotReleased_Call) Return 

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

func (*MockSegment_PinIfNotReleased_Call) Run 

func (_c *MockSegment_PinIfNotReleased_Call) Run(run func()) *MockSegment_PinIfNotReleased_Call

func (*MockSegment_PinIfNotReleased_Call) RunAndReturn 

func (_c *MockSegment_PinIfNotReleased_Call) RunAndReturn(run func() error) *MockSegment_PinIfNotReleased_Call

type MockSegment_Release_Call 

type MockSegment_Release_Call struct {
	*mock.Call
}

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

func (*MockSegment_Release_Call) Return 

func (_c *MockSegment_Release_Call) Return() *MockSegment_Release_Call

func (*MockSegment_Release_Call) Run 

func (_c *MockSegment_Release_Call) Run(run func(ctx context.Context, opts ...releaseOption)) *MockSegment_Release_Call

func (*MockSegment_Release_Call) RunAndReturn 

func (_c *MockSegment_Release_Call) RunAndReturn(run func(context.Context, ...releaseOption)) *MockSegment_Release_Call

type MockSegment_RemoveUnusedFieldFiles_Call 

type MockSegment_RemoveUnusedFieldFiles_Call struct {
	*mock.Call
}

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

func (*MockSegment_RemoveUnusedFieldFiles_Call) Return 

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

func (*MockSegment_RemoveUnusedFieldFiles_Call) Run 

func (_c *MockSegment_RemoveUnusedFieldFiles_Call) Run(run func()) *MockSegment_RemoveUnusedFieldFiles_Call

func (*MockSegment_RemoveUnusedFieldFiles_Call) RunAndReturn 

func (_c *MockSegment_RemoveUnusedFieldFiles_Call) RunAndReturn(run func() error) *MockSegment_RemoveUnusedFieldFiles_Call

type MockSegment_ResetIndexesLazyLoad_Call 

type MockSegment_ResetIndexesLazyLoad_Call struct {
	*mock.Call
}

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

func (*MockSegment_ResetIndexesLazyLoad_Call) Return 

func (_c *MockSegment_ResetIndexesLazyLoad_Call) Return() *MockSegment_ResetIndexesLazyLoad_Call

func (*MockSegment_ResetIndexesLazyLoad_Call) Run 

func (_c *MockSegment_ResetIndexesLazyLoad_Call) Run(run func(lazyState bool)) *MockSegment_ResetIndexesLazyLoad_Call

func (*MockSegment_ResetIndexesLazyLoad_Call) RunAndReturn 

func (_c *MockSegment_ResetIndexesLazyLoad_Call) RunAndReturn(run func(bool)) *MockSegment_ResetIndexesLazyLoad_Call

type MockSegment_ResourceGroup_Call 

type MockSegment_ResourceGroup_Call struct {
	*mock.Call
}

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

func (*MockSegment_ResourceGroup_Call) Return 

func (_c *MockSegment_ResourceGroup_Call) Return(_a0 string) *MockSegment_ResourceGroup_Call

func (*MockSegment_ResourceGroup_Call) Run 

func (_c *MockSegment_ResourceGroup_Call) Run(run func()) *MockSegment_ResourceGroup_Call

func (*MockSegment_ResourceGroup_Call) RunAndReturn 

func (_c *MockSegment_ResourceGroup_Call) RunAndReturn(run func() string) *MockSegment_ResourceGroup_Call

type MockSegment_ResourceUsageEstimate_Call 

type MockSegment_ResourceUsageEstimate_Call struct {
	*mock.Call
}

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

func (*MockSegment_ResourceUsageEstimate_Call) Return 

func (_c *MockSegment_ResourceUsageEstimate_Call) Return(_a0 ResourceUsage) *MockSegment_ResourceUsageEstimate_Call

func (*MockSegment_ResourceUsageEstimate_Call) Run 

func (_c *MockSegment_ResourceUsageEstimate_Call) Run(run func()) *MockSegment_ResourceUsageEstimate_Call

func (*MockSegment_ResourceUsageEstimate_Call) RunAndReturn 

func (_c *MockSegment_ResourceUsageEstimate_Call) RunAndReturn(run func() ResourceUsage) *MockSegment_ResourceUsageEstimate_Call

type MockSegment_RetrieveByOffsets_Call 

type MockSegment_RetrieveByOffsets_Call struct {
	*mock.Call
}

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

func (*MockSegment_RetrieveByOffsets_Call) Return 

func (_c *MockSegment_RetrieveByOffsets_Call) Return(_a0 *segcorepb.RetrieveResults, _a1 error) *MockSegment_RetrieveByOffsets_Call

func (*MockSegment_RetrieveByOffsets_Call) Run 

func (_c *MockSegment_RetrieveByOffsets_Call) Run(run func(ctx context.Context, plan *segcore.RetrievePlanWithOffsets)) *MockSegment_RetrieveByOffsets_Call

func (*MockSegment_RetrieveByOffsets_Call) RunAndReturn 

func (_c *MockSegment_RetrieveByOffsets_Call) RunAndReturn(run func(context.Context, *segcore.RetrievePlanWithOffsets) (*segcorepb.RetrieveResults, error)) *MockSegment_RetrieveByOffsets_Call

type MockSegment_Retrieve_Call 

type MockSegment_Retrieve_Call struct {
	*mock.Call
}

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

func (*MockSegment_Retrieve_Call) Return 

func (_c *MockSegment_Retrieve_Call) Return(_a0 *segcorepb.RetrieveResults, _a1 error) *MockSegment_Retrieve_Call

func (*MockSegment_Retrieve_Call) Run 

func (_c *MockSegment_Retrieve_Call) Run(run func(ctx context.Context, plan *segcore.RetrievePlan)) *MockSegment_Retrieve_Call

func (*MockSegment_Retrieve_Call) RunAndReturn 

func (_c *MockSegment_Retrieve_Call) RunAndReturn(run func(context.Context, *segcore.RetrievePlan) (*segcorepb.RetrieveResults, error)) *MockSegment_Retrieve_Call

type MockSegment_RowNum_Call 

type MockSegment_RowNum_Call struct {
	*mock.Call
}

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

func (*MockSegment_RowNum_Call) Return 

func (_c *MockSegment_RowNum_Call) Return(_a0 int64) *MockSegment_RowNum_Call

func (*MockSegment_RowNum_Call) Run 

func (_c *MockSegment_RowNum_Call) Run(run func()) *MockSegment_RowNum_Call

func (*MockSegment_RowNum_Call) RunAndReturn 

func (_c *MockSegment_RowNum_Call) RunAndReturn(run func() int64) *MockSegment_RowNum_Call

type MockSegment_Search_Call 

type MockSegment_Search_Call struct {
	*mock.Call
}

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

func (*MockSegment_Search_Call) Return 

func (_c *MockSegment_Search_Call) Return(_a0 *segcore.SearchResult, _a1 error) *MockSegment_Search_Call

func (*MockSegment_Search_Call) Run 

func (_c *MockSegment_Search_Call) Run(run func(ctx context.Context, searchReq *segcore.SearchRequest)) *MockSegment_Search_Call

func (*MockSegment_Search_Call) RunAndReturn 

func (_c *MockSegment_Search_Call) RunAndReturn(run func(context.Context, *segcore.SearchRequest) (*segcore.SearchResult, error)) *MockSegment_Search_Call

type MockSegment_SetBloomFilter_Call 

type MockSegment_SetBloomFilter_Call struct {
	*mock.Call
}

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

func (*MockSegment_SetBloomFilter_Call) Return 

func (_c *MockSegment_SetBloomFilter_Call) Return() *MockSegment_SetBloomFilter_Call

func (*MockSegment_SetBloomFilter_Call) Run 

func (_c *MockSegment_SetBloomFilter_Call) Run(run func(bf *pkoracle.BloomFilterSet)) *MockSegment_SetBloomFilter_Call

func (*MockSegment_SetBloomFilter_Call) RunAndReturn 

func (_c *MockSegment_SetBloomFilter_Call) RunAndReturn(run func(*pkoracle.BloomFilterSet)) *MockSegment_SetBloomFilter_Call

type MockSegment_Shard_Call 

type MockSegment_Shard_Call struct {
	*mock.Call
}

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

func (*MockSegment_Shard_Call) Return 

func (_c *MockSegment_Shard_Call) Return(_a0 metautil.Channel) *MockSegment_Shard_Call

func (*MockSegment_Shard_Call) Run 

func (_c *MockSegment_Shard_Call) Run(run func()) *MockSegment_Shard_Call

func (*MockSegment_Shard_Call) RunAndReturn 

func (_c *MockSegment_Shard_Call) RunAndReturn(run func() metautil.Channel) *MockSegment_Shard_Call

type MockSegment_StartPosition_Call 

type MockSegment_StartPosition_Call struct {
	*mock.Call
}

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

func (*MockSegment_StartPosition_Call) Return 

func (_c *MockSegment_StartPosition_Call) Return(_a0 *msgpb.MsgPosition) *MockSegment_StartPosition_Call

func (*MockSegment_StartPosition_Call) Run 

func (_c *MockSegment_StartPosition_Call) Run(run func()) *MockSegment_StartPosition_Call

func (*MockSegment_StartPosition_Call) RunAndReturn 

func (_c *MockSegment_StartPosition_Call) RunAndReturn(run func() *msgpb.MsgPosition) *MockSegment_StartPosition_Call

type MockSegment_Type_Call 

type MockSegment_Type_Call struct {
	*mock.Call
}

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

func (*MockSegment_Type_Call) Return 

func (_c *MockSegment_Type_Call) Return(_a0 commonpb.SegmentState) *MockSegment_Type_Call

func (*MockSegment_Type_Call) Run 

func (_c *MockSegment_Type_Call) Run(run func()) *MockSegment_Type_Call

func (*MockSegment_Type_Call) RunAndReturn 

func (_c *MockSegment_Type_Call) RunAndReturn(run func() commonpb.SegmentState) *MockSegment_Type_Call

type MockSegment_Unpin_Call 

type MockSegment_Unpin_Call struct {
	*mock.Call
}

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

func (*MockSegment_Unpin_Call) Return 

func (_c *MockSegment_Unpin_Call) Return() *MockSegment_Unpin_Call

func (*MockSegment_Unpin_Call) Run 

func (_c *MockSegment_Unpin_Call) Run(run func()) *MockSegment_Unpin_Call

func (*MockSegment_Unpin_Call) RunAndReturn 

func (_c *MockSegment_Unpin_Call) RunAndReturn(run func()) *MockSegment_Unpin_Call

type MockSegment_UpdateBM25Stats_Call 

type MockSegment_UpdateBM25Stats_Call struct {
	*mock.Call
}

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

func (*MockSegment_UpdateBM25Stats_Call) Return 

func (_c *MockSegment_UpdateBM25Stats_Call) Return() *MockSegment_UpdateBM25Stats_Call

func (*MockSegment_UpdateBM25Stats_Call) Run 

func (_c *MockSegment_UpdateBM25Stats_Call) Run(run func(stats map[int64]*storage.BM25Stats)) *MockSegment_UpdateBM25Stats_Call

func (*MockSegment_UpdateBM25Stats_Call) RunAndReturn 

func (_c *MockSegment_UpdateBM25Stats_Call) RunAndReturn(run func(map[int64]*storage.BM25Stats)) *MockSegment_UpdateBM25Stats_Call

type MockSegment_UpdateBloomFilter_Call 

type MockSegment_UpdateBloomFilter_Call struct {
	*mock.Call
}

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

func (*MockSegment_UpdateBloomFilter_Call) Return 

func (_c *MockSegment_UpdateBloomFilter_Call) Return() *MockSegment_UpdateBloomFilter_Call

func (*MockSegment_UpdateBloomFilter_Call) Run 

func (_c *MockSegment_UpdateBloomFilter_Call) Run(run func(pks []storage.PrimaryKey)) *MockSegment_UpdateBloomFilter_Call

func (*MockSegment_UpdateBloomFilter_Call) RunAndReturn 

func (_c *MockSegment_UpdateBloomFilter_Call) RunAndReturn(run func([]storage.PrimaryKey)) *MockSegment_UpdateBloomFilter_Call

type MockSegment_Version_Call 

type MockSegment_Version_Call struct {
	*mock.Call
}

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

func (*MockSegment_Version_Call) Return 

func (_c *MockSegment_Version_Call) Return(_a0 int64) *MockSegment_Version_Call

func (*MockSegment_Version_Call) Run 

func (_c *MockSegment_Version_Call) Run(run func()) *MockSegment_Version_Call

func (*MockSegment_Version_Call) RunAndReturn 

func (_c *MockSegment_Version_Call) RunAndReturn(run func() int64) *MockSegment_Version_Call

type ReleaseScope 

type ReleaseScope int
const (
	ReleaseScopeAll ReleaseScope = iota
	ReleaseScopeData
)

type ResourceEstimate 

type ResourceEstimate struct {
	MaxMemoryCost   uint64
	MaxDiskCost     uint64
	FinalMemoryCost uint64
	FinalDiskCost   uint64
	HasRawData      bool
}

func GetResourceEstimate 

func GetResourceEstimate(estimate *C.LoadResourceRequest) ResourceEstimate

type ResourceUsage 

type ResourceUsage struct {
	MemorySize         uint64
	DiskSize           uint64
	MmapFieldCount     int
	FieldGpuMemorySize []uint64
}

ResourceUsage is used to estimate the resource usage of a sealed segment.

type RetrievePlan 

type RetrievePlan = segcore.RetrievePlan

type RetrieveSegmentResult 

type RetrieveSegmentResult struct {
	Result  *segcorepb.RetrieveResults
	Segment Segment
}

type SearchCommonReduce 

type SearchCommonReduce struct{}

func (*SearchCommonReduce) ReduceSearchResultData 

func (scr *SearchCommonReduce) ReduceSearchResultData(ctx context.Context, searchResultData []*schemapb.SearchResultData, info *reduce.ResultInfo) (*schemapb.SearchResultData, error)

type SearchGroupByReduce 

type SearchGroupByReduce struct{}

func (*SearchGroupByReduce) ReduceSearchResultData 

func (sbr *SearchGroupByReduce) ReduceSearchResultData(ctx context.Context, searchResultData []*schemapb.SearchResultData, info *reduce.ResultInfo) (*schemapb.SearchResultData, error)

type SearchPlan 

type SearchPlan = segcore.SearchPlan

type SearchReduce 

type SearchReduce interface {
	ReduceSearchResultData(ctx context.Context, searchResultData []*schemapb.SearchResultData, info *reduce.ResultInfo) (*schemapb.SearchResultData, error)
}

func InitSearchReducer 

func InitSearchReducer(info *reduce.ResultInfo) SearchReduce

type SearchRequest 

type SearchRequest = segcore.SearchRequest

type SearchResult 

type SearchResult = segcore.SearchResult

type Segment 

type Segment interface {

	// Properties
	ID() int64
	DatabaseName() string
	ResourceGroup() string
	Collection() int64
	Partition() int64
	Shard() metautil.Channel
	Version() int64
	CASVersion(int64, int64) bool
	StartPosition() *msgpb.MsgPosition
	Type() SegmentType
	Level() datapb.SegmentLevel
	IsSorted() bool
	LoadInfo() *querypb.SegmentLoadInfo
	// PinIfNotReleased the segment to prevent it from being released
	PinIfNotReleased() error
	// Unpin the segment to allow it to be released
	Unpin()

	// Stats related
	// InsertCount returns the number of inserted rows, not effected by deletion
	InsertCount() int64
	// RowNum returns the number of rows, it's slow, so DO NOT call it in a loop
	RowNum() int64
	MemSize() int64
	// ResourceUsageEstimate returns the estimated resource usage of the segment
	ResourceUsageEstimate() ResourceUsage

	// Index related
	GetIndexByID(indexID int64) *IndexedFieldInfo
	GetIndex(fieldID int64) []*IndexedFieldInfo
	ExistIndex(fieldID int64) bool
	Indexes() []*IndexedFieldInfo
	HasRawData(fieldID int64) bool
	DropIndex(ctx context.Context, indexID int64) error

	// Modification related
	Insert(ctx context.Context, rowIDs []int64, timestamps []typeutil.Timestamp, record *segcorepb.InsertRecord) error
	Delete(ctx context.Context, primaryKeys storage.PrimaryKeys, timestamps []typeutil.Timestamp) error
	LoadDeltaData(ctx context.Context, deltaData *storage.DeltaData) error
	LastDeltaTimestamp() uint64
	FinishLoad() error
	Release(ctx context.Context, opts ...releaseOption)

	// Bloom filter related
	SetBloomFilter(bf *pkoracle.BloomFilterSet)
	BloomFilterExist() bool
	UpdateBloomFilter(pks []storage.PrimaryKey)
	MayPkExist(lc *storage.LocationsCache) bool
	BatchPkExist(lc *storage.BatchLocationsCache) []bool

	// Get min/max
	GetMinPk() *storage.PrimaryKey
	GetMaxPk() *storage.PrimaryKey

	// BM25 stats
	UpdateBM25Stats(stats map[int64]*storage.BM25Stats)
	GetBM25Stats() map[int64]*storage.BM25Stats

	// Read operations
	Search(ctx context.Context, searchReq *segcore.SearchRequest) (*segcore.SearchResult, error)
	Retrieve(ctx context.Context, plan *segcore.RetrievePlan) (*segcorepb.RetrieveResults, error)
	RetrieveByOffsets(ctx context.Context, plan *segcore.RetrievePlanWithOffsets) (*segcorepb.RetrieveResults, error)
	IsLazyLoad() bool
	ResetIndexesLazyLoad(lazyState bool)

	// lazy load related
	NeedUpdatedVersion() int64
	RemoveUnusedFieldFiles() error

	GetFieldJSONIndexStats() map[int64]*querypb.JsonStatsInfo
}

Segment is the interface of a segment implementation. Some methods can not apply to all segment types，such as LoadInfo, ResourceUsageEstimate. Add more interface to represent different segment types is a better implementation.

func NewL0Segment 

func NewL0Segment(collection *Collection,
	segmentType SegmentType,
	version int64,
	loadInfo *querypb.SegmentLoadInfo,
) (Segment, error)

func NewSegment 

func NewSegment(ctx context.Context,
	collection *Collection,
	manager SegmentManager,
	segmentType SegmentType,
	version int64,
	loadInfo *querypb.SegmentLoadInfo,
) (Segment, error)

func RetrieveStream 

func RetrieveStream(ctx context.Context, manager *Manager, plan *RetrievePlan, req *querypb.QueryRequest, srv streamrpc.QueryStreamServer) ([]Segment, error)

retrieveStreaming will retrieve all the validate target segments and return by stream

func SearchHistoricalStreamly 

func SearchHistoricalStreamly(ctx context.Context, manager *Manager, searchReq *SearchRequest,
	collID int64, partIDs []int64, segIDs []int64, streamReduce func(result *SearchResult) error,
) ([]Segment, error)

type SegmentAction 

type SegmentAction func(segment Segment) bool

func IncreaseVersion 

func IncreaseVersion(version int64) SegmentAction

type SegmentFilter 

type SegmentFilter interface {
	Match(segment Segment) bool
	AddFilter(*segmentCriterion)
}

SegmentFilter is the interface for segment selection criteria.

func WithChannel 

func WithChannel(channel string) SegmentFilter

func WithID 

func WithID(id int64) SegmentFilter

func WithIDs 

func WithIDs(ids ...int64) SegmentFilter

func WithLevel 

func WithLevel(level datapb.SegmentLevel) SegmentFilter

func WithPartition 

func WithPartition(partitionID typeutil.UniqueID) SegmentFilter

func WithSkipEmpty 

func WithSkipEmpty() SegmentFilter

func WithSparseFilter 

func WithSparseFilter(plan *planpb.PlanNode) SegmentFilter

func WithType 

func WithType(typ SegmentType) SegmentFilter

func WithoutLevel 

func WithoutLevel(level datapb.SegmentLevel) SegmentFilter

WithoutLevel is the segment filter for without segment level.

type SegmentFilterFunc 

type SegmentFilterFunc func(segment Segment) bool

SegmentFilterFunc is a type wrapper for `func(Segment) bool` to SegmentFilter.

func (SegmentFilterFunc) AddFilter 

func (f SegmentFilterFunc) AddFilter(c *segmentCriterion)

func (SegmentFilterFunc) Match 

func (f SegmentFilterFunc) Match(segment Segment) bool

type SegmentIDFilter 

type SegmentIDFilter int64

SegmentIDFilter is the specific segment filter for SegmentID only.

func (SegmentIDFilter) AddFilter 

func (f SegmentIDFilter) AddFilter(c *segmentCriterion)

func (SegmentIDFilter) Match 

func (f SegmentIDFilter) Match(segment Segment) bool

type SegmentIDsFilter 

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

func (SegmentIDsFilter) AddFilter 

func (f SegmentIDsFilter) AddFilter(c *segmentCriterion)

func (SegmentIDsFilter) Match 

func (f SegmentIDsFilter) Match(segment Segment) bool

type SegmentManager 

type SegmentManager interface {
	// Put puts the given segments in,
	// and increases the ref count of the corresponding collection,
	// dup segments will not increase the ref count
	Put(ctx context.Context, segmentType SegmentType, segments ...Segment)
	UpdateBy(action SegmentAction, filters ...SegmentFilter) int
	Get(segmentID typeutil.UniqueID) Segment
	GetWithType(segmentID typeutil.UniqueID, typ SegmentType) Segment
	GetBy(filters ...SegmentFilter) []Segment
	// Get segments and acquire the read locks
	GetAndPinBy(filters ...SegmentFilter) ([]Segment, error)
	GetAndPin(segments []int64, filters ...SegmentFilter) ([]Segment, error)
	Unpin(segments []Segment)

	GetSealed(segmentID typeutil.UniqueID) Segment
	GetGrowing(segmentID typeutil.UniqueID) Segment
	Empty() bool

	// Remove removes the given segment,
	// and decreases the ref count of the corresponding collection,
	// will not decrease the ref count if the given segment not exists
	Remove(ctx context.Context, segmentID typeutil.UniqueID, scope querypb.DataScope) (int, int)
	RemoveBy(ctx context.Context, filters ...SegmentFilter) (int, int)
	Clear(ctx context.Context)

	// Deprecated: quick fix critical issue: #30857
	// TODO: All Segment assigned to querynode should be managed by SegmentManager, including loading or releasing to perform a transaction.
	Exist(segmentID typeutil.UniqueID, typ SegmentType) bool

	AddLogicalResource(usage ResourceUsage)
	SubLogicalResource(usage ResourceUsage)
	GetLogicalResource() ResourceUsage
}

type SegmentSparseFilter 

type SegmentSparseFilter SegmentType

type SegmentStats 

type SegmentStats struct {
	SegmentID int64
	RowCount  int64
}

SegmentStats struct for segment statistics.

type SegmentType 

type SegmentType = commonpb.SegmentState

type SegmentTypeFilter 

type SegmentTypeFilter SegmentType

func (SegmentTypeFilter) AddFilter 

func (f SegmentTypeFilter) AddFilter(c *segmentCriterion)

func (SegmentTypeFilter) Match 

func (f SegmentTypeFilter) Match(segment Segment) bool

type TimestampedRetrieveResult 

type TimestampedRetrieveResult[T interface {
	typeutil.ResultWithID
	GetFieldsData() []*schemapb.FieldData
}] struct {
	Result     T
	Timestamps []int64
}

func NewTimestampedRetrieveResult 

func NewTimestampedRetrieveResult[T interface {
	typeutil.ResultWithID
	GetFieldsData() []*schemapb.FieldData
}](result T) (*TimestampedRetrieveResult[T], error)

func (*TimestampedRetrieveResult[T]) GetHasMoreResult 

func (r *TimestampedRetrieveResult[T]) GetHasMoreResult() bool

func (*TimestampedRetrieveResult[T]) GetIds 

func (r *TimestampedRetrieveResult[T]) GetIds() *schemapb.IDs

func (*TimestampedRetrieveResult[T]) GetTimestamps 

func (r *TimestampedRetrieveResult[T]) GetTimestamps() []int64

Source Files

View all Source files

Directories

Path Synopsis

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.