From 2d9361e0bae2c459a3ba4ba15ebfdcacaa7a5cc2 Mon Sep 17 00:00:00 2001 From: dragondriver Date: Sat, 17 Jul 2021 10:53:30 +0800 Subject: [PATCH] Fix bug: remove duplicated primary field id when retrieve entities (#6577) Signed-off-by: dragondriver --- internal/proxy/task.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/internal/proxy/task.go b/internal/proxy/task.go index ff7145f92e..fc985cbc41 100644 --- a/internal/proxy/task.go +++ b/internal/proxy/task.go @@ -1420,6 +1420,7 @@ func (st *SearchTask) PreExecute(ctx context.Context) error { if err != nil { return err } + log.Debug("translate output fields", zap.Any("OutputFields", outputFields)) st.query.OutputFields = outputFields if st.query.GetDslType() == commonpb.DslType_BoolExprV1 { @@ -2102,6 +2103,7 @@ func (rt *RetrieveTask) PreExecute(ctx context.Context) error { if err != nil { return err } + log.Debug("translate output fields", zap.Any("OutputFields", rt.retrieve.OutputFields)) if len(rt.retrieve.OutputFields) == 0 { for _, field := range schema.Fields { if field.FieldID >= 100 && field.DataType != schemapb.DataType_FloatVector && field.DataType != schemapb.DataType_BinaryVector { @@ -2109,9 +2111,9 @@ func (rt *RetrieveTask) PreExecute(ctx context.Context) error { } } } else { + addPrimaryKey := false for _, reqField := range rt.retrieve.OutputFields { findField := false - addPrimaryKey := false for _, field := range schema.Fields { if reqField == field.Name { if field.IsPrimaryKey { @@ -2132,6 +2134,7 @@ func (rt *RetrieveTask) PreExecute(ctx context.Context) error { } } } + log.Debug("translate output fields to field ids", zap.Any("OutputFieldsID", rt.OutputFieldsId)) travelTimestamp := rt.retrieve.TravelTimestamp if travelTimestamp == 0 {