// Code generated by ent, DO NOT EDIT. package ent import ( "api/ent/group" "api/ent/predicate" "api/ent/user" "context" "errors" "fmt" "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) { return withHooks[int, GroupMutation](ctx, gu.sqlSave, gu.mutation, gu.hooks) } // 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) } } func (gu *GroupUpdate) sqlSave(ctx context.Context) (n int, err error) { _spec := sqlgraph.NewUpdateSpec(group.Table, group.Columns, sqlgraph.NewFieldSpec(group.FieldID, field.TypeInt)) 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.SetField(group.FieldName, field.TypeString, value) } if gu.mutation.UsersCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2M, Inverse: false, Table: group.UsersTable, Columns: []string{group.UsersColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt), }, } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := gu.mutation.RemovedUsersIDs(); len(nodes) > 0 && !gu.mutation.UsersCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2M, Inverse: false, Table: group.UsersTable, Columns: []string{group.UsersColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt), }, } 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.O2M, Inverse: false, Table: group.UsersTable, Columns: []string{group.UsersColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt), }, } 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{msg: err.Error(), wrap: err} } return 0, err } gu.mutation.done = true 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...) } // Where appends a list predicates to the GroupUpdate builder. func (guo *GroupUpdateOne) Where(ps ...predicate.Group) *GroupUpdateOne { guo.mutation.Where(ps...) return guo } // 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) { return withHooks[*Group, GroupMutation](ctx, guo.sqlSave, guo.mutation, guo.hooks) } // 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) } } func (guo *GroupUpdateOne) sqlSave(ctx context.Context) (_node *Group, err error) { _spec := sqlgraph.NewUpdateSpec(group.Table, group.Columns, sqlgraph.NewFieldSpec(group.FieldID, field.TypeInt)) 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.SetField(group.FieldName, field.TypeString, value) } if guo.mutation.UsersCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2M, Inverse: false, Table: group.UsersTable, Columns: []string{group.UsersColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt), }, } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := guo.mutation.RemovedUsersIDs(); len(nodes) > 0 && !guo.mutation.UsersCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2M, Inverse: false, Table: group.UsersTable, Columns: []string{group.UsersColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt), }, } 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.O2M, Inverse: false, Table: group.UsersTable, Columns: []string{group.UsersColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt), }, } 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{msg: err.Error(), wrap: err} } return nil, err } guo.mutation.done = true return _node, nil }