From 890de2ba6c4da736cf53676cd912390242cb7810 Mon Sep 17 00:00:00 2001 From: Congqi Xia Date: Wed, 17 Jan 2024 12:21:30 +0800 Subject: [PATCH] fix: deduce AutoID from field info Resolves: #654 Signed-off-by: Congqi Xia --- entity/schema.go | 4 +++- entity/schema_test.go | 1 - 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/entity/schema.go b/entity/schema.go index ee7179abf..08e6a6c54 100644 --- a/entity/schema.go +++ b/entity/schema.go @@ -107,11 +107,13 @@ func (s *Schema) ProtoMessage() *schema.CollectionSchema { // ReadProto parses proto Collection Schema func (s *Schema) ReadProto(p *schema.CollectionSchema) *Schema { - s.AutoID = p.GetAutoID() s.Description = p.GetDescription() s.CollectionName = p.GetName() s.Fields = make([]*Field, 0, len(p.GetFields())) for _, fp := range p.GetFields() { + if fp.GetAutoID() { + s.AutoID = true + } s.Fields = append(s.Fields, NewField().ReadProto(fp)) } s.EnableDynamicField = p.GetEnableDynamicField() diff --git a/entity/schema_test.go b/entity/schema_test.go index 07d07de6f..471175278 100644 --- a/entity/schema_test.go +++ b/entity/schema_test.go @@ -107,7 +107,6 @@ func (s *SchemaSuite) TestBasic() { nsch = nsch.ReadProto(p) s.Equal(sch.CollectionName, nsch.CollectionName) - s.Equal(sch.AutoID, nsch.AutoID) s.Equal(sch.Description, nsch.Description) s.Equal(sch.EnableDynamicField, nsch.EnableDynamicField) s.Equal(len(sch.Fields), len(nsch.Fields))