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/edgeindex/ent/city_create.go
View File

@@ -92,6 +92,19 @@ func (cc *CityCreate) SaveX(ctx context.Context) *City {
}
func (cc *CityCreate) sqlSave(ctx context.Context) (*City, error) {
c, _spec := cc.createSpec()
if err := sqlgraph.CreateNode(ctx, cc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
c.ID = int(id)
return c, nil
}
func (cc *CityCreate) createSpec() (*City, *sqlgraph.CreateSpec) {
var (
c = &City{config: cc.config}
_spec = &sqlgraph.CreateSpec{
@@ -129,13 +142,5 @@ func (cc *CityCreate) sqlSave(ctx context.Context) (*City, error) {
}
_spec.Edges = append(_spec.Edges, edge)
}
if err := sqlgraph.CreateNode(ctx, cc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
c.ID = int(id)
return c, nil
return c, _spec
}

23
examples/edgeindex/ent/street_create.go
View File

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

25
examples/entcpkg/ent/user_create.go
View File

@@ -66,16 +66,7 @@ func (uc *UserCreate) SaveX(ctx context.Context) *User {
}
func (uc *UserCreate) sqlSave(ctx context.Context) (*User, error) {
var (
u = &User{config: uc.config}
_spec = &sqlgraph.CreateSpec{
Table: user.Table,
ID: &sqlgraph.FieldSpec{
Type: field.TypeInt,
Column: user.FieldID,
},
}
)
u, _spec := uc.createSpec()
if err := sqlgraph.CreateNode(ctx, uc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
@@ -86,3 +77,17 @@ func (uc *UserCreate) sqlSave(ctx context.Context) (*User, error) {
u.ID = int(id)
return u, nil
}
func (uc *UserCreate) createSpec() (*User, *sqlgraph.CreateSpec) {
var (
u = &User{config: uc.config}
_spec = &sqlgraph.CreateSpec{
Table: user.Table,
ID: &sqlgraph.FieldSpec{
Type: field.TypeInt,
Column: user.FieldID,
},
}
)
return u, _spec
}

23
examples/m2m2types/ent/group_create.go
View File

@@ -92,6 +92,19 @@ func (gc *GroupCreate) SaveX(ctx context.Context) *Group {
}
func (gc *GroupCreate) sqlSave(ctx context.Context) (*Group, error) {
gr, _spec := gc.createSpec()
if err := sqlgraph.CreateNode(ctx, gc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
gr.ID = int(id)
return gr, nil
}
func (gc *GroupCreate) createSpec() (*Group, *sqlgraph.CreateSpec) {
var (
gr = &Group{config: gc.config}
_spec = &sqlgraph.CreateSpec{
@@ -129,13 +142,5 @@ func (gc *GroupCreate) sqlSave(ctx context.Context) (*Group, error) {
}
_spec.Edges = append(_spec.Edges, edge)
}
if err := sqlgraph.CreateNode(ctx, gc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
gr.ID = int(id)
return gr, nil
return gr, _spec
}

23
examples/m2m2types/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
}

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

@@ -100,6 +100,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{
@@ -145,13 +158,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
}

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

@@ -115,6 +115,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{
@@ -179,13 +192,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
}

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
}

23
examples/o2mrecur/ent/node_create.go
View File

@@ -110,6 +110,19 @@ func (nc *NodeCreate) SaveX(ctx context.Context) *Node {
}
func (nc *NodeCreate) sqlSave(ctx context.Context) (*Node, error) {
n, _spec := nc.createSpec()
if err := sqlgraph.CreateNode(ctx, nc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
n.ID = int(id)
return n, nil
}
func (nc *NodeCreate) createSpec() (*Node, *sqlgraph.CreateSpec) {
var (
n = &Node{config: nc.config}
_spec = &sqlgraph.CreateSpec{
@@ -166,13 +179,5 @@ func (nc *NodeCreate) sqlSave(ctx context.Context) (*Node, error) {
}
_spec.Edges = append(_spec.Edges, edge)
}
if err := sqlgraph.CreateNode(ctx, nc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
n.ID = int(id)
return n, nil
return n, _spec
}

23
examples/o2o2types/ent/card_create.go
View File

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

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

@@ -105,6 +105,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{
@@ -150,13 +163,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
}

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

@@ -104,6 +104,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{
@@ -149,13 +162,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
}

23
examples/o2orecur/ent/node_create.go
View File

@@ -114,6 +114,19 @@ func (nc *NodeCreate) SaveX(ctx context.Context) *Node {
}
func (nc *NodeCreate) sqlSave(ctx context.Context) (*Node, error) {
n, _spec := nc.createSpec()
if err := sqlgraph.CreateNode(ctx, nc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
n.ID = int(id)
return n, nil
}
func (nc *NodeCreate) createSpec() (*Node, *sqlgraph.CreateSpec) {
var (
n = &Node{config: nc.config}
_spec = &sqlgraph.CreateSpec{
@@ -170,13 +183,5 @@ func (nc *NodeCreate) sqlSave(ctx context.Context) (*Node, error) {
}
_spec.Edges = append(_spec.Edges, edge)
}
if err := sqlgraph.CreateNode(ctx, nc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
n.ID = int(id)
return n, nil
return n, _spec
}

23
examples/start/ent/car_create.go
View File

@@ -106,6 +106,19 @@ func (cc *CarCreate) SaveX(ctx context.Context) *Car {
}
func (cc *CarCreate) sqlSave(ctx context.Context) (*Car, error) {
c, _spec := cc.createSpec()
if err := sqlgraph.CreateNode(ctx, cc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
c.ID = int(id)
return c, nil
}
func (cc *CarCreate) createSpec() (*Car, *sqlgraph.CreateSpec) {
var (
c = &Car{config: cc.config}
_spec = &sqlgraph.CreateSpec{
@@ -151,13 +164,5 @@ func (cc *CarCreate) sqlSave(ctx context.Context) (*Car, error) {
}
_spec.Edges = append(_spec.Edges, edge)
}
if err := sqlgraph.CreateNode(ctx, cc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
c.ID = int(id)
return c, nil
return c, _spec
}

23
examples/start/ent/group_create.go
View File

@@ -97,6 +97,19 @@ func (gc *GroupCreate) SaveX(ctx context.Context) *Group {
}
func (gc *GroupCreate) sqlSave(ctx context.Context) (*Group, error) {
gr, _spec := gc.createSpec()
if err := sqlgraph.CreateNode(ctx, gc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
gr.ID = int(id)
return gr, nil
}
func (gc *GroupCreate) createSpec() (*Group, *sqlgraph.CreateSpec) {
var (
gr = &Group{config: gc.config}
_spec = &sqlgraph.CreateSpec{
@@ -134,13 +147,5 @@ func (gc *GroupCreate) sqlSave(ctx context.Context) (*Group, error) {
}
_spec.Edges = append(_spec.Edges, edge)
}
if err := sqlgraph.CreateNode(ctx, gc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
gr.ID = int(id)
return gr, nil
return gr, _spec
}

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

@@ -131,6 +131,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{
@@ -195,13 +208,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
}

23
examples/traversal/ent/group_create.go
View File

@@ -111,6 +111,19 @@ func (gc *GroupCreate) SaveX(ctx context.Context) *Group {
}
func (gc *GroupCreate) sqlSave(ctx context.Context) (*Group, error) {
gr, _spec := gc.createSpec()
if err := sqlgraph.CreateNode(ctx, gc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
gr.ID = int(id)
return gr, nil
}
func (gc *GroupCreate) createSpec() (*Group, *sqlgraph.CreateSpec) {
var (
gr = &Group{config: gc.config}
_spec = &sqlgraph.CreateSpec{
@@ -167,13 +180,5 @@ func (gc *GroupCreate) sqlSave(ctx context.Context) (*Group, error) {
}
_spec.Edges = append(_spec.Edges, edge)
}
if err := sqlgraph.CreateNode(ctx, gc.driver, _spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
return nil, err
}
id := _spec.ID.Value.(int64)
gr.ID = int(id)
return gr, nil
return gr, _spec
}

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

@@ -111,6 +111,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{
@@ -167,13 +180,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/traversal/ent/user_create.go
View File

@@ -147,6 +147,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{
@@ -249,13 +262,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
}