mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
@@ -111,7 +111,7 @@ func (uc *UserCreate) check() error {
|
||||
}
|
||||
|
||||
func (uc *UserCreate) sqlSave(ctx context.Context) (*User, error) {
|
||||
u, _spec := uc.createSpec()
|
||||
_node, _spec := uc.createSpec()
|
||||
if err := sqlgraph.CreateNode(ctx, uc.driver, _spec); err != nil {
|
||||
if cerr, ok := isSQLConstraintError(err); ok {
|
||||
err = cerr
|
||||
@@ -119,13 +119,13 @@ func (uc *UserCreate) sqlSave(ctx context.Context) (*User, error) {
|
||||
return nil, err
|
||||
}
|
||||
id := _spec.ID.Value.(int64)
|
||||
u.ID = int(id)
|
||||
return u, nil
|
||||
_node.ID = int(id)
|
||||
return _node, nil
|
||||
}
|
||||
|
||||
func (uc *UserCreate) createSpec() (*User, *sqlgraph.CreateSpec) {
|
||||
var (
|
||||
u = &User{config: uc.config}
|
||||
_node = &User{config: uc.config}
|
||||
_spec = &sqlgraph.CreateSpec{
|
||||
Table: user.Table,
|
||||
ID: &sqlgraph.FieldSpec{
|
||||
@@ -140,7 +140,7 @@ func (uc *UserCreate) createSpec() (*User, *sqlgraph.CreateSpec) {
|
||||
Value: value,
|
||||
Column: user.FieldAge,
|
||||
})
|
||||
u.Age = value
|
||||
_node.Age = value
|
||||
}
|
||||
if value, ok := uc.mutation.Name(); ok {
|
||||
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{
|
||||
@@ -148,7 +148,7 @@ func (uc *UserCreate) createSpec() (*User, *sqlgraph.CreateSpec) {
|
||||
Value: value,
|
||||
Column: user.FieldName,
|
||||
})
|
||||
u.Name = value
|
||||
_node.Name = value
|
||||
}
|
||||
if nodes := uc.mutation.FriendsIDs(); len(nodes) > 0 {
|
||||
edge := &sqlgraph.EdgeSpec{
|
||||
@@ -169,7 +169,7 @@ func (uc *UserCreate) createSpec() (*User, *sqlgraph.CreateSpec) {
|
||||
}
|
||||
_spec.Edges = append(_spec.Edges, edge)
|
||||
}
|
||||
return u, _spec
|
||||
return _node, _spec
|
||||
}
|
||||
|
||||
// UserCreateBulk is the builder for creating a bulk of User entities.
|
||||
|
||||
@@ -83,23 +83,23 @@ func (uq *UserQuery) QueryFriends() *UserQuery {
|
||||
|
||||
// First returns the first User entity in the query. Returns *NotFoundError when no user was found.
|
||||
func (uq *UserQuery) First(ctx context.Context) (*User, error) {
|
||||
us, err := uq.Limit(1).All(ctx)
|
||||
nodes, err := uq.Limit(1).All(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if len(us) == 0 {
|
||||
if len(nodes) == 0 {
|
||||
return nil, &NotFoundError{user.Label}
|
||||
}
|
||||
return us[0], nil
|
||||
return nodes[0], nil
|
||||
}
|
||||
|
||||
// FirstX is like First, but panics if an error occurs.
|
||||
func (uq *UserQuery) FirstX(ctx context.Context) *User {
|
||||
u, err := uq.First(ctx)
|
||||
node, err := uq.First(ctx)
|
||||
if err != nil && !IsNotFound(err) {
|
||||
panic(err)
|
||||
}
|
||||
return u
|
||||
return node
|
||||
}
|
||||
|
||||
// FirstID returns the first User id in the query. Returns *NotFoundError when no id was found.
|
||||
@@ -126,13 +126,13 @@ func (uq *UserQuery) FirstXID(ctx context.Context) int {
|
||||
|
||||
// Only returns the only User entity in the query, returns an error if not exactly one entity was returned.
|
||||
func (uq *UserQuery) Only(ctx context.Context) (*User, error) {
|
||||
us, err := uq.Limit(2).All(ctx)
|
||||
nodes, err := uq.Limit(2).All(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
switch len(us) {
|
||||
switch len(nodes) {
|
||||
case 1:
|
||||
return us[0], nil
|
||||
return nodes[0], nil
|
||||
case 0:
|
||||
return nil, &NotFoundError{user.Label}
|
||||
default:
|
||||
@@ -142,11 +142,11 @@ func (uq *UserQuery) Only(ctx context.Context) (*User, error) {
|
||||
|
||||
// OnlyX is like Only, but panics if an error occurs.
|
||||
func (uq *UserQuery) OnlyX(ctx context.Context) *User {
|
||||
u, err := uq.Only(ctx)
|
||||
node, err := uq.Only(ctx)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return u
|
||||
return node
|
||||
}
|
||||
|
||||
// OnlyID returns the only User id in the query, returns an error if not exactly one id was returned.
|
||||
@@ -185,11 +185,11 @@ func (uq *UserQuery) All(ctx context.Context) ([]*User, error) {
|
||||
|
||||
// AllX is like All, but panics if an error occurs.
|
||||
func (uq *UserQuery) AllX(ctx context.Context) []*User {
|
||||
us, err := uq.All(ctx)
|
||||
nodes, err := uq.All(ctx)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return us
|
||||
return nodes
|
||||
}
|
||||
|
||||
// IDs executes the query and returns a list of User ids.
|
||||
|
||||
@@ -344,11 +344,11 @@ func (uuo *UserUpdateOne) Save(ctx context.Context) (*User, error) {
|
||||
|
||||
// SaveX is like Save, but panics if an error occurs.
|
||||
func (uuo *UserUpdateOne) SaveX(ctx context.Context) *User {
|
||||
u, err := uuo.Save(ctx)
|
||||
node, err := uuo.Save(ctx)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return u
|
||||
return node
|
||||
}
|
||||
|
||||
// Exec executes the query on the entity.
|
||||
@@ -364,7 +364,7 @@ func (uuo *UserUpdateOne) ExecX(ctx context.Context) {
|
||||
}
|
||||
}
|
||||
|
||||
func (uuo *UserUpdateOne) sqlSave(ctx context.Context) (u *User, err error) {
|
||||
func (uuo *UserUpdateOne) sqlSave(ctx context.Context) (_node *User, err error) {
|
||||
_spec := &sqlgraph.UpdateSpec{
|
||||
Node: &sqlgraph.NodeSpec{
|
||||
Table: user.Table,
|
||||
@@ -455,9 +455,9 @@ func (uuo *UserUpdateOne) sqlSave(ctx context.Context) (u *User, err error) {
|
||||
}
|
||||
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
||||
}
|
||||
u = &User{config: uuo.config}
|
||||
_spec.Assign = u.assignValues
|
||||
_spec.ScanValues = u.scanValues()
|
||||
_node = &User{config: uuo.config}
|
||||
_spec.Assign = _node.assignValues
|
||||
_spec.ScanValues = _node.scanValues()
|
||||
if err = sqlgraph.UpdateNode(ctx, uuo.driver, _spec); err != nil {
|
||||
if _, ok := err.(*sqlgraph.NotFoundError); ok {
|
||||
err = &NotFoundError{user.Label}
|
||||
@@ -466,5 +466,5 @@ func (uuo *UserUpdateOne) sqlSave(ctx context.Context) (u *User, err error) {
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return u, nil
|
||||
return _node, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user