// Code generated by entc, DO NOT EDIT. package ent import ( "context" "errors" "fmt" "code.osinet.fr/fgm/go__ent_demo/ent/group" "code.osinet.fr/fgm/go__ent_demo/ent/predicate" "code.osinet.fr/fgm/go__ent_demo/ent/user" "entgo.io/ent/dialect/sql" "entgo.io/ent/dialect/sql/sqlgraph" "entgo.io/ent/schema/field" ) // GroupUpdate is the builder for updating Group entities. type GroupUpdate struct { config hooks []Hook mutation *GroupMutation } // Where appends a list predicates to the GroupUpdate builder. func (gu *GroupUpdate) Where(ps ...predicate.Group) *GroupUpdate { gu.mutation.Where(ps...) return gu } // SetName sets the "name" field. func (gu *GroupUpdate) SetName(s string) *GroupUpdate { gu.mutation.SetName(s) return gu } // AddUserIDs adds the "users" edge to the User entity by IDs. func (gu *GroupUpdate) AddUserIDs(ids ...int) *GroupUpdate { gu.mutation.AddUserIDs(ids...) return gu } // AddUsers adds the "users" edges to the User entity. func (gu *GroupUpdate) AddUsers(u ...*User) *GroupUpdate { ids := make([]int, len(u)) for i := range u { ids[i] = u[i].ID } return gu.AddUserIDs(ids...) } // Mutation returns the GroupMutation object of the builder. func (gu *GroupUpdate) Mutation() *GroupMutation { return gu.mutation } // ClearUsers clears all "users" edges to the User entity. func (gu *GroupUpdate) ClearUsers() *GroupUpdate { gu.mutation.ClearUsers() return gu } // RemoveUserIDs removes the "users" edge to User entities by IDs. func (gu *GroupUpdate) RemoveUserIDs(ids ...int) *GroupUpdate { gu.mutation.RemoveUserIDs(ids...) return gu } // RemoveUsers removes "users" edges to User entities. func (gu *GroupUpdate) RemoveUsers(u ...*User) *GroupUpdate { ids := make([]int, len(u)) for i := range u { ids[i] = u[i].ID } return gu.RemoveUserIDs(ids...) } // Save executes the query and returns the number of nodes affected by the update operation. func (gu *GroupUpdate) Save(ctx context.Context) (int, error) { var ( err error affected int ) if len(gu.hooks) == 0 { if err = gu.check(); err != nil { return 0, err } affected, err = gu.sqlSave(ctx) } else { var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { mutation, ok := m.(*GroupMutation) if !ok { return nil, fmt.Errorf("unexpected mutation type %T", m) } if err = gu.check(); err != nil { return 0, err } gu.mutation = mutation affected, err = gu.sqlSave(ctx) mutation.done = true return affected, err }) for i := len(gu.hooks) - 1; i >= 0; i-- { if gu.hooks[i] == nil { return 0, fmt.Errorf("ent: uninitialized hook (forgotten import ent/runtime?)") } mut = gu.hooks[i](mut) } if _, err := mut.Mutate(ctx, gu.mutation); err != nil { return 0, err } } return affected, err } // SaveX is like Save, but panics if an error occurs. func (gu *GroupUpdate) SaveX(ctx context.Context) int { affected, err := gu.Save(ctx) if err != nil { panic(err) } return affected } // Exec executes the query. func (gu *GroupUpdate) Exec(ctx context.Context) error { _, err := gu.Save(ctx) return err } // ExecX is like Exec, but panics if an error occurs. func (gu *GroupUpdate) ExecX(ctx context.Context) { if err := gu.Exec(ctx); err != nil { panic(err) } } // check runs all checks and user-defined validators on the builder. func (gu *GroupUpdate) check() error { if v, ok := gu.mutation.Name(); ok { if err := group.NameValidator(v); err != nil { return &ValidationError{Name: "name", err: fmt.Errorf(`ent: validator failed for field "Group.name": %w`, err)} } } return nil } func (gu *GroupUpdate) sqlSave(ctx context.Context) (n int, err error) { _spec := &sqlgraph.UpdateSpec{ Node: &sqlgraph.NodeSpec{ Table: group.Table, Columns: group.Columns, ID: &sqlgraph.FieldSpec{ Type: field.TypeInt, Column: group.FieldID, }, }, } if ps := gu.mutation.predicates; len(ps) > 0 { _spec.Predicate = func(selector *sql.Selector) { for i := range ps { ps[i](selector) } } } if value, ok := gu.mutation.Name(); ok { _spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{ Type: field.TypeString, Value: value, Column: group.FieldName, }) } if gu.mutation.UsersCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2M, Inverse: false, Table: group.UsersTable, Columns: group.UsersPrimaryKey, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: &sqlgraph.FieldSpec{ Type: field.TypeInt, Column: user.FieldID, }, }, } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := gu.mutation.RemovedUsersIDs(); len(nodes) > 0 && !gu.mutation.UsersCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2M, Inverse: false, Table: group.UsersTable, Columns: group.UsersPrimaryKey, 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.Clear = append(_spec.Edges.Clear, edge) } if nodes := gu.mutation.UsersIDs(); len(nodes) > 0 { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2M, Inverse: false, Table: group.UsersTable, Columns: group.UsersPrimaryKey, 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.Add = append(_spec.Edges.Add, edge) } if n, err = sqlgraph.UpdateNodes(ctx, gu.driver, _spec); err != nil { if _, ok := err.(*sqlgraph.NotFoundError); ok { err = &NotFoundError{group.Label} } else if sqlgraph.IsConstraintError(err) { err = &ConstraintError{err.Error(), err} } return 0, err } return n, nil } // GroupUpdateOne is the builder for updating a single Group entity. type GroupUpdateOne struct { config fields []string hooks []Hook mutation *GroupMutation } // SetName sets the "name" field. func (guo *GroupUpdateOne) SetName(s string) *GroupUpdateOne { guo.mutation.SetName(s) return guo } // AddUserIDs adds the "users" edge to the User entity by IDs. func (guo *GroupUpdateOne) AddUserIDs(ids ...int) *GroupUpdateOne { guo.mutation.AddUserIDs(ids...) return guo } // AddUsers adds the "users" edges to the User entity. func (guo *GroupUpdateOne) AddUsers(u ...*User) *GroupUpdateOne { ids := make([]int, len(u)) for i := range u { ids[i] = u[i].ID } return guo.AddUserIDs(ids...) } // Mutation returns the GroupMutation object of the builder. func (guo *GroupUpdateOne) Mutation() *GroupMutation { return guo.mutation } // ClearUsers clears all "users" edges to the User entity. func (guo *GroupUpdateOne) ClearUsers() *GroupUpdateOne { guo.mutation.ClearUsers() return guo } // RemoveUserIDs removes the "users" edge to User entities by IDs. func (guo *GroupUpdateOne) RemoveUserIDs(ids ...int) *GroupUpdateOne { guo.mutation.RemoveUserIDs(ids...) return guo } // RemoveUsers removes "users" edges to User entities. func (guo *GroupUpdateOne) RemoveUsers(u ...*User) *GroupUpdateOne { ids := make([]int, len(u)) for i := range u { ids[i] = u[i].ID } return guo.RemoveUserIDs(ids...) } // Select allows selecting one or more fields (columns) of the returned entity. // The default is selecting all fields defined in the entity schema. func (guo *GroupUpdateOne) Select(field string, fields ...string) *GroupUpdateOne { guo.fields = append([]string{field}, fields...) return guo } // Save executes the query and returns the updated Group entity. func (guo *GroupUpdateOne) Save(ctx context.Context) (*Group, error) { var ( err error node *Group ) if len(guo.hooks) == 0 { if err = guo.check(); err != nil { return nil, err } node, err = guo.sqlSave(ctx) } else { var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { mutation, ok := m.(*GroupMutation) if !ok { return nil, fmt.Errorf("unexpected mutation type %T", m) } if err = guo.check(); err != nil { return nil, err } guo.mutation = mutation node, err = guo.sqlSave(ctx) mutation.done = true return node, err }) for i := len(guo.hooks) - 1; i >= 0; i-- { if guo.hooks[i] == nil { return nil, fmt.Errorf("ent: uninitialized hook (forgotten import ent/runtime?)") } mut = guo.hooks[i](mut) } if _, err := mut.Mutate(ctx, guo.mutation); err != nil { return nil, err } } return node, err } // SaveX is like Save, but panics if an error occurs. func (guo *GroupUpdateOne) SaveX(ctx context.Context) *Group { node, err := guo.Save(ctx) if err != nil { panic(err) } return node } // Exec executes the query on the entity. func (guo *GroupUpdateOne) Exec(ctx context.Context) error { _, err := guo.Save(ctx) return err } // ExecX is like Exec, but panics if an error occurs. func (guo *GroupUpdateOne) ExecX(ctx context.Context) { if err := guo.Exec(ctx); err != nil { panic(err) } } // check runs all checks and user-defined validators on the builder. func (guo *GroupUpdateOne) check() error { if v, ok := guo.mutation.Name(); ok { if err := group.NameValidator(v); err != nil { return &ValidationError{Name: "name", err: fmt.Errorf(`ent: validator failed for field "Group.name": %w`, err)} } } return nil } func (guo *GroupUpdateOne) sqlSave(ctx context.Context) (_node *Group, err error) { _spec := &sqlgraph.UpdateSpec{ Node: &sqlgraph.NodeSpec{ Table: group.Table, Columns: group.Columns, ID: &sqlgraph.FieldSpec{ Type: field.TypeInt, Column: group.FieldID, }, }, } id, ok := guo.mutation.ID() if !ok { return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "Group.id" for update`)} } _spec.Node.ID.Value = id if fields := guo.fields; len(fields) > 0 { _spec.Node.Columns = make([]string, 0, len(fields)) _spec.Node.Columns = append(_spec.Node.Columns, group.FieldID) for _, f := range fields { if !group.ValidColumn(f) { return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)} } if f != group.FieldID { _spec.Node.Columns = append(_spec.Node.Columns, f) } } } if ps := guo.mutation.predicates; len(ps) > 0 { _spec.Predicate = func(selector *sql.Selector) { for i := range ps { ps[i](selector) } } } if value, ok := guo.mutation.Name(); ok { _spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{ Type: field.TypeString, Value: value, Column: group.FieldName, }) } if guo.mutation.UsersCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2M, Inverse: false, Table: group.UsersTable, Columns: group.UsersPrimaryKey, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: &sqlgraph.FieldSpec{ Type: field.TypeInt, Column: user.FieldID, }, }, } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := guo.mutation.RemovedUsersIDs(); len(nodes) > 0 && !guo.mutation.UsersCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2M, Inverse: false, Table: group.UsersTable, Columns: group.UsersPrimaryKey, 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.Clear = append(_spec.Edges.Clear, edge) } if nodes := guo.mutation.UsersIDs(); len(nodes) > 0 { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2M, Inverse: false, Table: group.UsersTable, Columns: group.UsersPrimaryKey, 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.Add = append(_spec.Edges.Add, edge) } _node = &Group{config: guo.config} _spec.Assign = _node.assignValues _spec.ScanValues = _node.scanValues if err = sqlgraph.UpdateNode(ctx, guo.driver, _spec); err != nil { if _, ok := err.(*sqlgraph.NotFoundError); ok { err = &NotFoundError{group.Label} } else if sqlgraph.IsConstraintError(err) { err = &ConstraintError{err.Error(), err} } return nil, err } return _node, nil }