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)) diff --git a/test/testcases/collection_test.go b/test/testcases/collection_test.go index 4c32d146e..2cd6cb48c 100644 --- a/test/testcases/collection_test.go +++ b/test/testcases/collection_test.go @@ -419,7 +419,7 @@ func TestCreateJsonCollection(t *testing.T) { // schema collName := common.GenRandomString(6) - schema := common.GenSchema(collName, false, fields) + schema := common.GenSchema(collName, true, fields) // create collection err := mc.CreateCollection(ctx, schema, common.DefaultShards)