admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-07 01:28:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: [GoSDK] reset cache after UsingDatabase (#35638)

Browse Source 
Related to milvus-io/milvus-sdk-go#809

---------

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
This commit is contained in:
congqixia 2024-08-29 19:41:02 +08:00 committed by GitHub
parent 69265978bf
commit 4373c6994f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
client/client.go
View File

@ -202,6 +202,9 @@ func (c *Client) connectInternal(ctx context.Context) error {
c.config.setServerInfo(resp.GetServerInfo().GetBuildTags()) c.config.setServerInfo(resp.GetServerInfo().GetBuildTags())
c.setIdentifier(strconv.FormatInt(resp.GetIdentifier(), 10)) c.setIdentifier(strconv.FormatInt(resp.GetIdentifier(), 10))
if c.collCache != nil {
c.collCache.Reset()
}
return nil return nil
} }

5
client/common.go
View File

@ -32,6 +32,11 @@ func (c *CollectionCache) GetCollection(ctx context.Context, collName string) (*
return coll, err return coll, err
} }
// Reset clears all cached info, used when client switching env.
func (c *CollectionCache) Reset() {
c.collections = typeutil.NewConcurrentMap[string, *entity.Collection]()
}
func NewCollectionCache(fetcher func(context.Context, string) (*entity.Collection, error)) *CollectionCache { func NewCollectionCache(fetcher func(context.Context, string) (*entity.Collection, error)) *CollectionCache {
return &CollectionCache{ return &CollectionCache{
collections: typeutil.NewConcurrentMap[string, *entity.Collection](), collections: typeutil.NewConcurrentMap[string, *entity.Collection](),

4
client/index/hnsw.go
View File

@ -41,13 +41,13 @@ func (idx hnswIndex) Params() map[string]string {
} }
} }
func NewHNSWIndex(metricType MetricType, M int, efConstruction int) Index { func NewHNSWIndex(metricType MetricType, m int, efConstruction int) Index {
return hnswIndex{ return hnswIndex{
baseIndex: baseIndex{ baseIndex: baseIndex{
metricType: metricType, metricType: metricType,
indexType: HNSW, indexType: HNSW,
}, },
m: M, m: m,
efConstruction: efConstruction, efConstruction: efConstruction,
} }
} }