golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-24 09:31:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

entc/gen/sql: move spec creation to a function (#604)

Browse Source 
To be shared with batch-create builder
This commit is contained in:
Ariel Mashraki
2020-07-13 13:27:36 +03:00
committed by GitHub
parent f7d4a52e68
commit 399883a71a
55 changed files with 793 additions and 523 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
examples/o2m2types/ent/pet_create.go
View File

@@ -96,6 +96,19 @@ func (pc *PetCreate) SaveX(ctx context.Context) *Pet {
}
func (pc *PetCreate) sqlSave(ctx context.Context) (*Pet, error) {
pe, _spec := pc.createSpec()
if err := sqlgraph.CreateNode(ctx, pc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
pe.ID = int(id)
return pe, nil
}
func (pc *PetCreate) createSpec() (*Pet, *sqlgraph.CreateSpec) {
var (
pe = &Pet{config: pc.config}
_spec = &sqlgraph.CreateSpec{
@@ -133,13 +146,5 @@ func (pc *PetCreate) sqlSave(ctx context.Context) (*Pet, error) {
}
_spec.Edges = append(_spec.Edges, edge)
}
if err := sqlgraph.CreateNode(ctx, pc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
pe.ID = int(id)
return pe, nil
return pe, _spec
}

23
examples/o2m2types/ent/user_create.go
View File

@@ -101,6 +101,19 @@ func (uc *UserCreate) SaveX(ctx context.Context) *User {
}
func (uc *UserCreate) sqlSave(ctx context.Context) (*User, error) {
u, _spec := uc.createSpec()
if err := sqlgraph.CreateNode(ctx, uc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
u.ID = int(id)
return u, nil
}
func (uc *UserCreate) createSpec() (*User, *sqlgraph.CreateSpec) {
var (
u = &User{config: uc.config}
_spec = &sqlgraph.CreateSpec{
@@ -146,13 +159,5 @@ func (uc *UserCreate) sqlSave(ctx context.Context) (*User, error) {
}
_spec.Edges = append(_spec.Edges, edge)
}
if err := sqlgraph.CreateNode(ctx, uc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
u.ID = int(id)
return u, nil
return u, _spec
}