From 39161cac9ae5e39e2dcdd408027ee9c1ddf05b4c Mon Sep 17 00:00:00 2001
From: "zhenshan.cao" <zhenshan.cao@zilliz.com>
Date: Sat, 6 Feb 2021 21:17:18 +0800
Subject: [PATCH] Add collectonScheam when calling load_collection and
 load_partition

Signed-off-by: zhenshan.cao <zhenshan.cao@zilliz.com>
---
 internal/proxynode/task.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/internal/proxynode/task.go b/internal/proxynode/task.go
index 9aff4e2578..3d11536de0 100644
--- a/internal/proxynode/task.go
+++ b/internal/proxynode/task.go
@@ -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