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

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

View File

@@ -188,6 +188,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
@@ -206,8 +207,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.age; value != nil {
updater.Set(user.FieldAge, *value)
}
@@ -418,6 +420,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
@@ -441,8 +444,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.age; value != nil {
updater.Set(user.FieldAge, *value)
u.Age = *value

View File

@@ -42,6 +42,7 @@ func (gc *GroupCreate) sqlSave(ctx context.Context) (*Group, error) {
return nil, err
}
insert := builder.Insert(group.Table).Default()
id, err := insertLastID(ctx, tx, insert.Returning(group.FieldID))
if err != nil {
return nil, rollback(tx, err)

View File

@@ -68,6 +68,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
@@ -135,6 +136,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

View File

@@ -42,6 +42,7 @@ func (pc *PetCreate) sqlSave(ctx context.Context) (*Pet, error) {
return nil, err
}
insert := builder.Insert(pet.Table).Default()
id, err := insertLastID(ctx, tx, insert.Returning(pet.FieldID))
if err != nil {
return nil, rollback(tx, err)

View File

@@ -68,6 +68,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
@@ -135,6 +136,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

View File

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

View File

@@ -205,6 +205,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
@@ -223,8 +224,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.age; value != nil {
updater.Set(user.FieldAge, *value)
}
@@ -458,6 +460,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
@@ -481,8 +484,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.age; value != nil {
updater.Set(user.FieldAge, *value)
u.Age = *value