entc/gen: initial work for supporting uuid fields in codegen

Summary: Pull Request resolved: https://github.com/facebookincubator/ent/pull/182

Reviewed By: alexsn

Differential Revision: D18638199

fbshipit-source-id: 0de79c78b51e544486c07a004c3c8ea82e5c3398
This commit is contained in:
Ariel Mashraki
2019-11-24 07:00:37 -08:00
committed by Facebook Github Bot
parent e6c6442c84
commit 67c3fd2db9
81 changed files with 1914 additions and 94 deletions

View File

@@ -57,6 +57,7 @@ func (gc *GroupCreate) sqlSave(ctx context.Context) (*Group, error) {
insert.Set(group.FieldMaxUsers, *value)
gr.MaxUsers = *value
}
id, err := insertLastID(ctx, tx, insert.Returning(group.FieldID))
if err != nil {
return nil, rollback(tx, err)

View File

@@ -87,6 +87,7 @@ func (gu *GroupUpdate) sqlSave(ctx context.Context) (n int, err error) {
return 0, err
}
defer rows.Close()
var ids []int
for rows.Next() {
var id int
@@ -105,8 +106,9 @@ func (gu *GroupUpdate) sqlSave(ctx context.Context) (n int, err error) {
}
var (
res sql.Result
updater = builder.Update(group.Table).Where(sql.InInts(group.FieldID, ids...))
updater = builder.Update(group.Table)
)
updater = updater.Where(sql.InInts(group.FieldID, ids...))
if value := gu.max_users; value != nil {
updater.Set(group.FieldMaxUsers, *value)
}
@@ -189,6 +191,7 @@ func (guo *GroupUpdateOne) sqlSave(ctx context.Context) (gr *Group, err error) {
return nil, err
}
defer rows.Close()
var ids []int
for rows.Next() {
var id int
@@ -212,8 +215,9 @@ func (guo *GroupUpdateOne) sqlSave(ctx context.Context) (gr *Group, err error) {
}
var (
res sql.Result
updater = builder.Update(group.Table).Where(sql.InInts(group.FieldID, ids...))
updater = builder.Update(group.Table)
)
updater = updater.Where(sql.InInts(group.FieldID, ids...))
if value := guo.max_users; value != nil {
updater.Set(group.FieldMaxUsers, *value)
gr.MaxUsers = *value

View File

@@ -104,6 +104,7 @@ func (pc *PetCreate) sqlSave(ctx context.Context) (*Pet, error) {
insert.Set(pet.FieldLicensedAt, *value)
pe.LicensedAt = value
}
id, err := insertLastID(ctx, tx, insert.Returning(pet.FieldID))
if err != nil {
return nil, rollback(tx, err)

View File

@@ -146,6 +146,7 @@ func (pu *PetUpdate) sqlSave(ctx context.Context) (n int, err error) {
return 0, err
}
defer rows.Close()
var ids []int
for rows.Next() {
var id int
@@ -164,8 +165,9 @@ func (pu *PetUpdate) sqlSave(ctx context.Context) (n int, err error) {
}
var (
res sql.Result
updater = builder.Update(pet.Table).Where(sql.InInts(pet.FieldID, ids...))
updater = builder.Update(pet.Table)
)
updater = updater.Where(sql.InInts(pet.FieldID, ids...))
if value := pu.age; value != nil {
updater.Set(pet.FieldAge, *value)
}
@@ -330,6 +332,7 @@ func (puo *PetUpdateOne) sqlSave(ctx context.Context) (pe *Pet, err error) {
return nil, err
}
defer rows.Close()
var ids []int
for rows.Next() {
var id int
@@ -353,8 +356,9 @@ func (puo *PetUpdateOne) sqlSave(ctx context.Context) (pe *Pet, err error) {
}
var (
res sql.Result
updater = builder.Update(pet.Table).Where(sql.InInts(pet.FieldID, ids...))
updater = builder.Update(pet.Table)
)
updater = updater.Where(sql.InInts(pet.FieldID, ids...))
if value := puo.age; value != nil {
updater.Set(pet.FieldAge, *value)
pe.Age = *value

View File

@@ -102,6 +102,7 @@ func (uc *UserCreate) sqlSave(ctx context.Context) (*User, error) {
insert.Set(user.FieldName, *value)
u.Name = *value
}
id, err := insertLastID(ctx, tx, insert.Returning(user.FieldID))
if err != nil {
return nil, rollback(tx, err)

View File

@@ -160,6 +160,7 @@ func (uu *UserUpdate) sqlSave(ctx context.Context) (n int, err error) {
return 0, err
}
defer rows.Close()
var ids []int
for rows.Next() {
var id int
@@ -178,8 +179,9 @@ func (uu *UserUpdate) sqlSave(ctx context.Context) (n int, err error) {
}
var (
res sql.Result
updater = builder.Update(user.Table).Where(sql.InInts(user.FieldID, ids...))
updater = builder.Update(user.Table)
)
updater = updater.Where(sql.InInts(user.FieldID, ids...))
if value := uu.name; value != nil {
updater.Set(user.FieldName, *value)
}
@@ -404,6 +406,7 @@ func (uuo *UserUpdateOne) sqlSave(ctx context.Context) (u *User, err error) {
return nil, err
}
defer rows.Close()
var ids []int
for rows.Next() {
var id int
@@ -427,8 +430,9 @@ func (uuo *UserUpdateOne) sqlSave(ctx context.Context) (u *User, err error) {
}
var (
res sql.Result
updater = builder.Update(user.Table).Where(sql.InInts(user.FieldID, ids...))
updater = builder.Update(user.Table)
)
updater = updater.Where(sql.InInts(user.FieldID, ids...))
if value := uuo.name; value != nil {
updater.Set(user.FieldName, *value)
u.Name = *value