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: move node creation to sqlgraph

Browse Source
This commit is contained in:
Ariel Mashraki
2019-12-16 16:55:20 +02:00
parent 5e9d13be97
commit c6800a3869
71 changed files with 2524 additions and 1838 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -9,11 +9,12 @@ package ent
import (
"context"
"errors"
"fmt"
"time"
"github.com/facebookincubator/ent/dialect/sql"
"github.com/facebookincubator/ent/dialect/sql/sqlgraph"
"github.com/facebookincubator/ent/examples/o2o2types/ent/card"
"github.com/facebookincubator/ent/examples/o2o2types/ent/user"
"github.com/facebookincubator/ent/schema/field"
)
// CardCreate is the builder for creating a Card entity.
@@ -78,48 +79,59 @@ func (cc *CardCreate) SaveX(ctx context.Context) *Card {
func (cc *CardCreate) sqlSave(ctx context.Context) (*Card, error) {
var (
res sql.Result
builder = sql.Dialect(cc.driver.Dialect())
c = &Card{config: cc.config}
c = &Card{config: cc.config}
spec = &sqlgraph.CreateSpec{
Table: card.Table,
ID: &sqlgraph.FieldSpec{
Type: field.TypeInt,
Column: card.FieldID,
},
}
)
tx, err := cc.driver.Tx(ctx)
if err != nil {
return nil, err
}
insert := builder.Insert(card.Table).Default()
if value := cc.expired; value != nil {
insert.Set(card.FieldExpired, *value)
spec.Fields = append(spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeTime,
Value: *value,
Column: card.FieldExpired,
})
c.Expired = *value
}
if value := cc.number; value != nil {
insert.Set(card.FieldNumber, *value)
spec.Fields = append(spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeString,
Value: *value,
Column: card.FieldNumber,
})
c.Number = *value
}
id, err := insertLastID(ctx, tx, insert.Returning(card.FieldID))
if err != nil {
return nil, rollback(tx, err)
if nodes := cc.owner; len(nodes) > 0 {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.O2O,
Inverse: true,
Table: card.OwnerTable,
Columns: []string{card.OwnerColumn},
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: &sqlgraph.FieldSpec{
Type: field.TypeInt,
Column: user.FieldID,
},
},
}
for k, _ := range nodes {
edge.Target.Nodes = append(edge.Target.Nodes, k)
}
spec.Edges = append(spec.Edges, edge)
}
c.ID = int(id)
if len(cc.owner) > 0 {
eid := keys(cc.owner)[0]
query, args := builder.Update(card.OwnerTable).
Set(card.OwnerColumn, eid).
Where(sql.EQ(card.FieldID, id).And().IsNull(card.OwnerColumn)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
if err := sqlgraph.CreateNode(ctx, cc.driver, spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
affected, err := res.RowsAffected()
if err != nil {
return nil, rollback(tx, err)
}
if int(affected) < len(cc.owner) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"owner\" %v already connected to a different \"Card\"", keys(cc.owner))})
}
}
if err := tx.Commit(); err != nil {
return nil, err
}
id := spec.ID.Value.(int64)
c.ID = int(id)
return c, nil
}

4
examples/o2o2types/ent/ent.go
View File

@@ -13,6 +13,7 @@ import (
"github.com/facebookincubator/ent/dialect"
"github.com/facebookincubator/ent/dialect/sql"
"github.com/facebookincubator/ent/dialect/sql/sqlgraph"
)
// Order applies an ordering on either graph traversal or sql selector.
@@ -159,6 +160,9 @@ func isSQLConstraintError(err error) (*ErrConstraintFailed, bool) {
"duplicate key value violates unique constraint", // PostgreSQL.
}
)
if _, ok := err.(*sqlgraph.ConstraintError); ok {
return &ErrConstraintFailed{msg, err}, true
}
for i := range errors {
if strings.Contains(msg, errors[i]) {
return &ErrConstraintFailed{msg, err}, true

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

@@ -9,11 +9,11 @@ package ent
import (
"context"
"errors"
"fmt"
"github.com/facebookincubator/ent/dialect/sql"
"github.com/facebookincubator/ent/dialect/sql/sqlgraph"
"github.com/facebookincubator/ent/examples/o2o2types/ent/card"
"github.com/facebookincubator/ent/examples/o2o2types/ent/user"
"github.com/facebookincubator/ent/schema/field"
)
// UserCreate is the builder for creating a User entity.
@@ -83,48 +83,59 @@ func (uc *UserCreate) SaveX(ctx context.Context) *User {
func (uc *UserCreate) sqlSave(ctx context.Context) (*User, error) {
var (
res sql.Result
builder = sql.Dialect(uc.driver.Dialect())
u = &User{config: uc.config}
u = &User{config: uc.config}
spec = &sqlgraph.CreateSpec{
Table: user.Table,
ID: &sqlgraph.FieldSpec{
Type: field.TypeInt,
Column: user.FieldID,
},
}
)
tx, err := uc.driver.Tx(ctx)
if err != nil {
return nil, err
}
insert := builder.Insert(user.Table).Default()
if value := uc.age; value != nil {
insert.Set(user.FieldAge, *value)
spec.Fields = append(spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeInt,
Value: *value,
Column: user.FieldAge,
})
u.Age = *value
}
if value := uc.name; value != nil {
insert.Set(user.FieldName, *value)
spec.Fields = append(spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeString,
Value: *value,
Column: user.FieldName,
})
u.Name = *value
}
id, err := insertLastID(ctx, tx, insert.Returning(user.FieldID))
if err != nil {
return nil, rollback(tx, err)
if nodes := uc.card; len(nodes) > 0 {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.O2O,
Inverse: false,
Table: user.CardTable,
Columns: []string{user.CardColumn},
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: &sqlgraph.FieldSpec{
Type: field.TypeInt,
Column: card.FieldID,
},
},
}
for k, _ := range nodes {
edge.Target.Nodes = append(edge.Target.Nodes, k)
}
spec.Edges = append(spec.Edges, edge)
}
u.ID = int(id)
if len(uc.card) > 0 {
eid := keys(uc.card)[0]
query, args := builder.Update(user.CardTable).
Set(user.CardColumn, id).
Where(sql.EQ(card.FieldID, eid).And().IsNull(user.CardColumn)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
if err := sqlgraph.CreateNode(ctx, uc.driver, spec); err != nil {
if cerr, ok := isSQLConstraintError(err); ok {
err = cerr
}
affected, err := res.RowsAffected()
if err != nil {
return nil, rollback(tx, err)
}
if int(affected) < len(uc.card) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"card\" %v already connected to a different \"User\"", keys(uc.card))})
}
}
if err := tx.Commit(); err != nil {
return nil, err
}
id := spec.ID.Value.(int64)
u.ID = int(id)
return u, nil
}