mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
entc/gen: move sql-update-one to sqlgraph
This commit is contained in:
Ariel Mashraki
@@ -12,8 +12,10 @@ import (
|
||||
"fmt"
|
||||
|
||||
"github.com/facebookincubator/ent/dialect/sql"
|
||||
"github.com/facebookincubator/ent/dialect/sql/sqlgraph"
|
||||
"github.com/facebookincubator/ent/entc/integration/idtype/ent/predicate"
|
||||
"github.com/facebookincubator/ent/entc/integration/idtype/ent/user"
|
||||
"github.com/facebookincubator/ent/schema/field"
|
||||
)
|
||||
|
||||
// UserUpdate is the builder for updating User entities.
|
||||
@@ -180,161 +182,148 @@ func (uu *UserUpdate) ExecX(ctx context.Context) {
|
||||
}
|
||||
|
||||
func (uu *UserUpdate) sqlSave(ctx context.Context) (n int, err error) {
|
||||
var (
|
||||
builder = sql.Dialect(uu.driver.Dialect())
|
||||
selector = builder.Select(user.FieldID).From(builder.Table(user.Table))
|
||||
)
|
||||
for _, p := range uu.predicates {
|
||||
p(selector)
|
||||
spec := &sqlgraph.UpdateSpec{
|
||||
Node: &sqlgraph.NodeSpec{
|
||||
Table: user.Table,
|
||||
Columns: user.Columns,
|
||||
ID: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
rows := &sql.Rows{}
|
||||
query, args := selector.Query()
|
||||
if err = uu.driver.Query(ctx, query, args, rows); err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var ids []int
|
||||
for rows.Next() {
|
||||
var id int
|
||||
if err := rows.Scan(&id); err != nil {
|
||||
return 0, fmt.Errorf("ent: failed reading id: %v", err)
|
||||
if ps := uu.predicates; len(ps) > 0 {
|
||||
spec.Predicate = func(selector *sql.Selector) {
|
||||
for i := range ps {
|
||||
ps[i](selector)
|
||||
}
|
||||
}
|
||||
ids = append(ids, id)
|
||||
}
|
||||
if len(ids) == 0 {
|
||||
return 0, nil
|
||||
}
|
||||
|
||||
tx, err := uu.driver.Tx(ctx)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
var (
|
||||
res sql.Result
|
||||
updater = builder.Update(user.Table)
|
||||
)
|
||||
updater = updater.Where(sql.InInts(user.FieldID, ids...))
|
||||
if value := uu.name; value != nil {
|
||||
updater.Set(user.FieldName, *value)
|
||||
}
|
||||
if !updater.Empty() {
|
||||
query, args := updater.Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return 0, rollback(tx, err)
|
||||
}
|
||||
spec.Fields.Set = append(spec.Fields.Set, &sqlgraph.FieldSpec{
|
||||
Type: field.TypeString,
|
||||
Value: *value,
|
||||
Column: user.FieldName,
|
||||
})
|
||||
}
|
||||
if uu.clearedSpouse {
|
||||
query, args := builder.Update(user.SpouseTable).
|
||||
SetNull(user.SpouseColumn).
|
||||
Where(sql.InInts(user.FieldID, ids...)).
|
||||
Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return 0, rollback(tx, err)
|
||||
}
|
||||
query, args = builder.Update(user.SpouseTable).
|
||||
SetNull(user.SpouseColumn).
|
||||
Where(sql.InInts(user.SpouseColumn, ids...)).
|
||||
Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return 0, rollback(tx, err)
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
Rel: sqlgraph.O2O,
|
||||
Inverse: false,
|
||||
Table: user.SpouseTable,
|
||||
Columns: []string{user.SpouseColumn},
|
||||
Bidi: true,
|
||||
Target: &sqlgraph.EdgeTarget{
|
||||
IDSpec: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
||||
}
|
||||
if len(uu.spouse) > 0 {
|
||||
if n := len(ids); n > 1 {
|
||||
return 0, rollback(tx, fmt.Errorf("ent: can't link O2O edge \"spouse\" to %d vertices (> 1)", n))
|
||||
if nodes := uu.spouse; len(nodes) > 0 {
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
Rel: sqlgraph.O2O,
|
||||
Inverse: false,
|
||||
Table: user.SpouseTable,
|
||||
Columns: []string{user.SpouseColumn},
|
||||
Bidi: true,
|
||||
Target: &sqlgraph.EdgeTarget{
|
||||
IDSpec: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
for eid := range uu.spouse {
|
||||
eid := int(eid)
|
||||
query, args := builder.Update(user.SpouseTable).
|
||||
Set(user.SpouseColumn, eid).
|
||||
Where(sql.EQ(user.FieldID, ids[0])).Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return 0, rollback(tx, err)
|
||||
}
|
||||
query, args = builder.Update(user.SpouseTable).
|
||||
Set(user.SpouseColumn, ids[0]).
|
||||
Where(sql.EQ(user.FieldID, eid).And().IsNull(user.SpouseColumn)).Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return 0, rollback(tx, err)
|
||||
}
|
||||
affected, err := res.RowsAffected()
|
||||
if err != nil {
|
||||
return 0, rollback(tx, err)
|
||||
}
|
||||
if int(affected) < len(uu.spouse) {
|
||||
return 0, rollback(tx, &ConstraintError{msg: fmt.Sprintf("\"spouse\" (%v) already connected to a different \"User\"", eid)})
|
||||
}
|
||||
for k, _ := range nodes {
|
||||
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
||||
}
|
||||
spec.Edges.Add = append(spec.Edges.Add, edge)
|
||||
}
|
||||
if len(uu.removedFollowers) > 0 {
|
||||
eids := make([]int, len(uu.removedFollowers))
|
||||
for eid := range uu.removedFollowers {
|
||||
eid := int(eid)
|
||||
eids = append(eids, eid)
|
||||
if nodes := uu.removedFollowers; len(nodes) > 0 {
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
Rel: sqlgraph.M2M,
|
||||
Inverse: true,
|
||||
Table: user.FollowersTable,
|
||||
Columns: user.FollowersPrimaryKey,
|
||||
Bidi: false,
|
||||
Target: &sqlgraph.EdgeTarget{
|
||||
IDSpec: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
query, args := builder.Delete(user.FollowersTable).
|
||||
Where(sql.InInts(user.FollowersPrimaryKey[1], ids...)).
|
||||
Where(sql.InInts(user.FollowersPrimaryKey[0], eids...)).
|
||||
Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return 0, rollback(tx, err)
|
||||
for k, _ := range nodes {
|
||||
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
||||
}
|
||||
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
||||
}
|
||||
if len(uu.followers) > 0 {
|
||||
values := make([][]int, 0, len(ids))
|
||||
for _, id := range ids {
|
||||
for eid := range uu.followers {
|
||||
eid := int(eid)
|
||||
values = append(values, []int{id, eid})
|
||||
}
|
||||
if nodes := uu.followers; len(nodes) > 0 {
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
Rel: sqlgraph.M2M,
|
||||
Inverse: true,
|
||||
Table: user.FollowersTable,
|
||||
Columns: user.FollowersPrimaryKey,
|
||||
Bidi: false,
|
||||
Target: &sqlgraph.EdgeTarget{
|
||||
IDSpec: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
builder := builder.Insert(user.FollowersTable).
|
||||
Columns(user.FollowersPrimaryKey[1], user.FollowersPrimaryKey[0])
|
||||
for _, v := range values {
|
||||
builder.Values(v[0], v[1])
|
||||
}
|
||||
query, args := builder.Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return 0, rollback(tx, err)
|
||||
for k, _ := range nodes {
|
||||
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
||||
}
|
||||
spec.Edges.Add = append(spec.Edges.Add, edge)
|
||||
}
|
||||
if len(uu.removedFollowing) > 0 {
|
||||
eids := make([]int, len(uu.removedFollowing))
|
||||
for eid := range uu.removedFollowing {
|
||||
eid := int(eid)
|
||||
eids = append(eids, eid)
|
||||
if nodes := uu.removedFollowing; len(nodes) > 0 {
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
Rel: sqlgraph.M2M,
|
||||
Inverse: false,
|
||||
Table: user.FollowingTable,
|
||||
Columns: user.FollowingPrimaryKey,
|
||||
Bidi: false,
|
||||
Target: &sqlgraph.EdgeTarget{
|
||||
IDSpec: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
query, args := builder.Delete(user.FollowingTable).
|
||||
Where(sql.InInts(user.FollowingPrimaryKey[0], ids...)).
|
||||
Where(sql.InInts(user.FollowingPrimaryKey[1], eids...)).
|
||||
Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return 0, rollback(tx, err)
|
||||
for k, _ := range nodes {
|
||||
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
||||
}
|
||||
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
||||
}
|
||||
if len(uu.following) > 0 {
|
||||
values := make([][]int, 0, len(ids))
|
||||
for _, id := range ids {
|
||||
for eid := range uu.following {
|
||||
eid := int(eid)
|
||||
values = append(values, []int{id, eid})
|
||||
}
|
||||
if nodes := uu.following; len(nodes) > 0 {
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
Rel: sqlgraph.M2M,
|
||||
Inverse: false,
|
||||
Table: user.FollowingTable,
|
||||
Columns: user.FollowingPrimaryKey,
|
||||
Bidi: false,
|
||||
Target: &sqlgraph.EdgeTarget{
|
||||
IDSpec: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
builder := builder.Insert(user.FollowingTable).
|
||||
Columns(user.FollowingPrimaryKey[0], user.FollowingPrimaryKey[1])
|
||||
for _, v := range values {
|
||||
builder.Values(v[0], v[1])
|
||||
}
|
||||
query, args := builder.Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return 0, rollback(tx, err)
|
||||
for k, _ := range nodes {
|
||||
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
||||
}
|
||||
spec.Edges.Add = append(spec.Edges.Add, edge)
|
||||
}
|
||||
if err = tx.Commit(); err != nil {
|
||||
if n, err = sqlgraph.UpdateNodes(ctx, uu.driver, spec); err != nil {
|
||||
if cerr, ok := isSQLConstraintError(err); ok {
|
||||
err = cerr
|
||||
}
|
||||
return 0, err
|
||||
}
|
||||
return len(ids), nil
|
||||
return n, nil
|
||||
}
|
||||
|
||||
// UserUpdateOne is the builder for updating a single User entity.
|
||||
@@ -495,162 +484,161 @@ func (uuo *UserUpdateOne) ExecX(ctx context.Context) {
|
||||
}
|
||||
|
||||
func (uuo *UserUpdateOne) sqlSave(ctx context.Context) (u *User, err error) {
|
||||
var (
|
||||
builder = sql.Dialect(uuo.driver.Dialect())
|
||||
selector = builder.Select(user.Columns...).From(builder.Table(user.Table))
|
||||
)
|
||||
user.ID(uuo.id)(selector)
|
||||
rows := &sql.Rows{}
|
||||
query, args := selector.Query()
|
||||
if err = uuo.driver.Query(ctx, query, args, rows); err != nil {
|
||||
return nil, err
|
||||
spec := &sqlgraph.UpdateSpec{
|
||||
Node: &sqlgraph.NodeSpec{
|
||||
Table: user.Table,
|
||||
Columns: user.Columns,
|
||||
ID: &sqlgraph.FieldSpec{
|
||||
Value: uuo.id,
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var ids []int
|
||||
for rows.Next() {
|
||||
var id int
|
||||
u = &User{config: uuo.config}
|
||||
if err := u.FromRows(rows); err != nil {
|
||||
return nil, fmt.Errorf("ent: failed scanning row into User: %v", err)
|
||||
}
|
||||
id = int(u.ID)
|
||||
ids = append(ids, id)
|
||||
}
|
||||
switch n := len(ids); {
|
||||
case n == 0:
|
||||
return nil, &ErrNotFound{fmt.Sprintf("User with id: %v", uuo.id)}
|
||||
case n > 1:
|
||||
return nil, fmt.Errorf("ent: more than one User with the same id: %v", uuo.id)
|
||||
}
|
||||
|
||||
tx, err := uuo.driver.Tx(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var (
|
||||
res sql.Result
|
||||
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
|
||||
}
|
||||
if !updater.Empty() {
|
||||
query, args := updater.Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return nil, rollback(tx, err)
|
||||
}
|
||||
spec.Fields.Set = append(spec.Fields.Set, &sqlgraph.FieldSpec{
|
||||
Type: field.TypeString,
|
||||
Value: *value,
|
||||
Column: user.FieldName,
|
||||
})
|
||||
}
|
||||
if uuo.clearedSpouse {
|
||||
query, args := builder.Update(user.SpouseTable).
|
||||
SetNull(user.SpouseColumn).
|
||||
Where(sql.InInts(user.FieldID, ids...)).
|
||||
Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return nil, rollback(tx, err)
|
||||
}
|
||||
query, args = builder.Update(user.SpouseTable).
|
||||
SetNull(user.SpouseColumn).
|
||||
Where(sql.InInts(user.SpouseColumn, ids...)).
|
||||
Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return nil, rollback(tx, err)
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
Rel: sqlgraph.O2O,
|
||||
Inverse: false,
|
||||
Table: user.SpouseTable,
|
||||
Columns: []string{user.SpouseColumn},
|
||||
Bidi: true,
|
||||
Target: &sqlgraph.EdgeTarget{
|
||||
IDSpec: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
||||
}
|
||||
if len(uuo.spouse) > 0 {
|
||||
if n := len(ids); n > 1 {
|
||||
return nil, rollback(tx, fmt.Errorf("ent: can't link O2O edge \"spouse\" to %d vertices (> 1)", n))
|
||||
if nodes := uuo.spouse; len(nodes) > 0 {
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
Rel: sqlgraph.O2O,
|
||||
Inverse: false,
|
||||
Table: user.SpouseTable,
|
||||
Columns: []string{user.SpouseColumn},
|
||||
Bidi: true,
|
||||
Target: &sqlgraph.EdgeTarget{
|
||||
IDSpec: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
for eid := range uuo.spouse {
|
||||
eid := int(eid)
|
||||
query, args := builder.Update(user.SpouseTable).
|
||||
Set(user.SpouseColumn, eid).
|
||||
Where(sql.EQ(user.FieldID, ids[0])).Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return nil, rollback(tx, err)
|
||||
}
|
||||
query, args = builder.Update(user.SpouseTable).
|
||||
Set(user.SpouseColumn, ids[0]).
|
||||
Where(sql.EQ(user.FieldID, eid).And().IsNull(user.SpouseColumn)).Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return nil, rollback(tx, err)
|
||||
}
|
||||
affected, err := res.RowsAffected()
|
||||
if err != nil {
|
||||
return nil, rollback(tx, err)
|
||||
}
|
||||
if int(affected) < len(uuo.spouse) {
|
||||
return nil, rollback(tx, &ConstraintError{msg: fmt.Sprintf("\"spouse\" (%v) already connected to a different \"User\"", eid)})
|
||||
}
|
||||
for k, _ := range nodes {
|
||||
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
||||
}
|
||||
spec.Edges.Add = append(spec.Edges.Add, edge)
|
||||
}
|
||||
if len(uuo.removedFollowers) > 0 {
|
||||
eids := make([]int, len(uuo.removedFollowers))
|
||||
for eid := range uuo.removedFollowers {
|
||||
eid := int(eid)
|
||||
eids = append(eids, eid)
|
||||
if nodes := uuo.removedFollowers; len(nodes) > 0 {
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
Rel: sqlgraph.M2M,
|
||||
Inverse: true,
|
||||
Table: user.FollowersTable,
|
||||
Columns: user.FollowersPrimaryKey,
|
||||
Bidi: false,
|
||||
Target: &sqlgraph.EdgeTarget{
|
||||
IDSpec: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
query, args := builder.Delete(user.FollowersTable).
|
||||
Where(sql.InInts(user.FollowersPrimaryKey[1], ids...)).
|
||||
Where(sql.InInts(user.FollowersPrimaryKey[0], eids...)).
|
||||
Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return nil, rollback(tx, err)
|
||||
for k, _ := range nodes {
|
||||
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
||||
}
|
||||
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
||||
}
|
||||
if len(uuo.followers) > 0 {
|
||||
values := make([][]int, 0, len(ids))
|
||||
for _, id := range ids {
|
||||
for eid := range uuo.followers {
|
||||
eid := int(eid)
|
||||
values = append(values, []int{id, eid})
|
||||
}
|
||||
if nodes := uuo.followers; len(nodes) > 0 {
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
Rel: sqlgraph.M2M,
|
||||
Inverse: true,
|
||||
Table: user.FollowersTable,
|
||||
Columns: user.FollowersPrimaryKey,
|
||||
Bidi: false,
|
||||
Target: &sqlgraph.EdgeTarget{
|
||||
IDSpec: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
builder := builder.Insert(user.FollowersTable).
|
||||
Columns(user.FollowersPrimaryKey[1], user.FollowersPrimaryKey[0])
|
||||
for _, v := range values {
|
||||
builder.Values(v[0], v[1])
|
||||
}
|
||||
query, args := builder.Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return nil, rollback(tx, err)
|
||||
for k, _ := range nodes {
|
||||
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
||||
}
|
||||
spec.Edges.Add = append(spec.Edges.Add, edge)
|
||||
}
|
||||
if len(uuo.removedFollowing) > 0 {
|
||||
eids := make([]int, len(uuo.removedFollowing))
|
||||
for eid := range uuo.removedFollowing {
|
||||
eid := int(eid)
|
||||
eids = append(eids, eid)
|
||||
if nodes := uuo.removedFollowing; len(nodes) > 0 {
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
Rel: sqlgraph.M2M,
|
||||
Inverse: false,
|
||||
Table: user.FollowingTable,
|
||||
Columns: user.FollowingPrimaryKey,
|
||||
Bidi: false,
|
||||
Target: &sqlgraph.EdgeTarget{
|
||||
IDSpec: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
query, args := builder.Delete(user.FollowingTable).
|
||||
Where(sql.InInts(user.FollowingPrimaryKey[0], ids...)).
|
||||
Where(sql.InInts(user.FollowingPrimaryKey[1], eids...)).
|
||||
Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return nil, rollback(tx, err)
|
||||
for k, _ := range nodes {
|
||||
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
||||
}
|
||||
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
||||
}
|
||||
if len(uuo.following) > 0 {
|
||||
values := make([][]int, 0, len(ids))
|
||||
for _, id := range ids {
|
||||
for eid := range uuo.following {
|
||||
eid := int(eid)
|
||||
values = append(values, []int{id, eid})
|
||||
}
|
||||
if nodes := uuo.following; len(nodes) > 0 {
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
Rel: sqlgraph.M2M,
|
||||
Inverse: false,
|
||||
Table: user.FollowingTable,
|
||||
Columns: user.FollowingPrimaryKey,
|
||||
Bidi: false,
|
||||
Target: &sqlgraph.EdgeTarget{
|
||||
IDSpec: &sqlgraph.FieldSpec{
|
||||
Type: field.TypeUint64,
|
||||
Column: user.FieldID,
|
||||
},
|
||||
},
|
||||
}
|
||||
builder := builder.Insert(user.FollowingTable).
|
||||
Columns(user.FollowingPrimaryKey[0], user.FollowingPrimaryKey[1])
|
||||
for _, v := range values {
|
||||
builder.Values(v[0], v[1])
|
||||
}
|
||||
query, args := builder.Query()
|
||||
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
||||
return nil, rollback(tx, err)
|
||||
for k, _ := range nodes {
|
||||
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
||||
}
|
||||
spec.Edges.Add = append(spec.Edges.Add, edge)
|
||||
}
|
||||
if err = tx.Commit(); err != nil {
|
||||
u = &User{config: uuo.config}
|
||||
spec.ScanTypes = []interface{}{
|
||||
&sql.NullInt64{},
|
||||
&sql.NullString{},
|
||||
}
|
||||
spec.Assign = func(values ...interface{}) error {
|
||||
if m, n := len(values), len(spec.ScanTypes); m != n {
|
||||
return fmt.Errorf("mismatch number of scan values: %d != %d", m, n)
|
||||
}
|
||||
value, ok := values[0].(*sql.NullInt64)
|
||||
if !ok {
|
||||
return fmt.Errorf("unexpected type %T for field id", value)
|
||||
}
|
||||
u.ID = uint64(value.Int64)
|
||||
values = values[1:]
|
||||
if value, ok := values[0].(*sql.NullString); !ok {
|
||||
return fmt.Errorf("unexpected type %T for field name", values[0])
|
||||
} else if value.Valid {
|
||||
u.Name = value.String
|
||||
}
|
||||
return nil
|
||||
}
|
||||
if err = sqlgraph.UpdateNode(ctx, uuo.driver, spec); err != nil {
|
||||
if cerr, ok := isSQLConstraintError(err); ok {
|
||||
err = cerr
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return u, nil
|
||||
|
||||
Reference in New Issue
Block a user