admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add collectonScheam when calling load_collection and load_partition

Browse Source 
Signed-off-by: zhenshan.cao <zhenshan.cao@zilliz.com>
This commit is contained in:
zhenshan.cao 2021-02-06 21:17:18 +08:00 committed by yefu.chen
parent a12b184e02
commit 39161cac9a
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
internal/proxynode/task.go
View File

@ -1625,6 +1625,11 @@ func (lct *LoadCollectionTask) Execute() (err error) {
if err != nil {
return err
}
collSchema, err := globalMetaCache.GetCollectionSchema(lct.CollectionName)
if err != nil {
return err
}
request := &querypb.LoadCollectionRequest{
Base: &commonpb.MsgBase{
MsgType: commonpb.MsgType_kLoadCollection,
@ -1634,6 +1639,7 @@ func (lct *LoadCollectionTask) Execute() (err error) {
},
DbID: 0,
CollectionID: collID,
Schema: collSchema,
}
lct.result, err = lct.queryserviceClient.LoadCollection(request)
return err
@ -1770,6 +1776,10 @@ func (lpt *LoadPartitionTask) Execute() (err error) {
if err != nil {
return err
}
collSchema, err := globalMetaCache.GetCollectionSchema(lpt.CollectionName)
if err != nil {
return err
}
for _, partitionName := range lpt.PartitionNames {
partitionID, err := globalMetaCache.GetPartitionID(lpt.CollectionName, partitionName)
if err != nil {
@ -1787,6 +1797,7 @@ func (lpt *LoadPartitionTask) Execute() (err error) {
DbID: 0,
CollectionID: collID,
PartitionIDs: partitionIDs,
Schema: collSchema,
}
lpt.result, err = lpt.queryserviceClient.LoadPartitions(request)
return err