// Copyright 2019-present Facebook Inc. All rights reserved. // This source code is licensed under the Apache 2.0 license found // in the LICENSE file in the root directory of this source tree. // Code generated by ent, DO NOT EDIT. package user import ( "context" "fmt" "entgo.io/ent" "entgo.io/ent/dialect/sql" "entgo.io/ent/entc/integration/privacy/ent/predicate" ) // Mutation represents an operation that mutates the User nodes in the graph. type Mutation struct { op ent.Op typ string name *string age *uint addage *int clearedFields map[string]struct{} teams map[int]struct{} removedteams map[int]struct{} clearedteams bool tasks map[int]struct{} removedtasks map[int]struct{} clearedtasks bool predicates []predicate.User } // NewMutation creates a new Mutation for the User entity. func NewMutation(op ent.Op) *Mutation { return &Mutation{ op: op, typ: "User", clearedFields: make(map[string]struct{}), } } // Predicates returns the list of predicates set on the mutation. func (m *Mutation) Predicates() []predicate.User { return m.predicates } // SetName sets the "name" field. func (m *Mutation) SetName(s string) { m.name = &s } // Name returns the value of the "name" field in the mutation. func (m *Mutation) Name() (r string, exists bool) { v := m.name if v == nil { return } return *v, true } // ResetName resets all changes to the "name" field. func (m *Mutation) ResetName() { m.name = nil } // SetAge sets the "age" field. func (m *Mutation) SetAge(u uint) { m.age = &u m.addage = nil } // Age returns the value of the "age" field in the mutation. func (m *Mutation) Age() (r uint, exists bool) { v := m.age if v == nil { return } return *v, true } // AddAge adds u to the "age" field. func (m *Mutation) AddAge(u int) { if m.addage != nil { *m.addage += u } else { m.addage = &u } } // AddedAge returns the value that was added to the "age" field in this mutation. func (m *Mutation) AddedAge() (r int, exists bool) { v := m.addage if v == nil { return } return *v, true } // ClearAge clears the value of the "age" field. func (m *Mutation) ClearAge() { m.age = nil m.addage = nil m.clearedFields[FieldAge] = struct{}{} } // AgeCleared returns if the "age" field was cleared in this mutation. func (m *Mutation) AgeCleared() bool { _, ok := m.clearedFields[FieldAge] return ok } // ResetAge resets all changes to the "age" field. func (m *Mutation) ResetAge() { m.age = nil m.addage = nil delete(m.clearedFields, FieldAge) } // AddTeamIDs adds the "teams" edge to the Team entity by ids. func (m *Mutation) AddTeamIDs(ids ...int) { if m.teams == nil { m.teams = make(map[int]struct{}) } for i := range ids { m.teams[ids[i]] = struct{}{} } } // ClearTeams clears the "teams" edge to the Team entity. func (m *Mutation) ClearTeams() { m.clearedteams = true } // TeamsCleared reports if the "teams" edge to the Team entity was cleared. func (m *Mutation) TeamsCleared() bool { return m.clearedteams } // RemoveTeamIDs removes the "teams" edge to the Team entity by IDs. func (m *Mutation) RemoveTeamIDs(ids ...int) { if m.removedteams == nil { m.removedteams = make(map[int]struct{}) } for i := range ids { delete(m.teams, ids[i]) m.removedteams[ids[i]] = struct{}{} } } // RemovedTeams returns the removed IDs of the "teams" edge to the Team entity. func (m *Mutation) RemovedTeamsIDs() (ids []int) { for id := range m.removedteams { ids = append(ids, id) } return } // TeamsIDs returns the "teams" edge IDs in the mutation. func (m *Mutation) TeamsIDs() (ids []int) { for id := range m.teams { ids = append(ids, id) } return } // ResetTeams resets all changes to the "teams" edge. func (m *Mutation) ResetTeams() { m.teams = nil m.clearedteams = false m.removedteams = nil } // AddTaskIDs adds the "tasks" edge to the Task entity by ids. func (m *Mutation) AddTaskIDs(ids ...int) { if m.tasks == nil { m.tasks = make(map[int]struct{}) } for i := range ids { m.tasks[ids[i]] = struct{}{} } } // ClearTasks clears the "tasks" edge to the Task entity. func (m *Mutation) ClearTasks() { m.clearedtasks = true } // TasksCleared reports if the "tasks" edge to the Task entity was cleared. func (m *Mutation) TasksCleared() bool { return m.clearedtasks } // RemoveTaskIDs removes the "tasks" edge to the Task entity by IDs. func (m *Mutation) RemoveTaskIDs(ids ...int) { if m.removedtasks == nil { m.removedtasks = make(map[int]struct{}) } for i := range ids { delete(m.tasks, ids[i]) m.removedtasks[ids[i]] = struct{}{} } } // RemovedTasks returns the removed IDs of the "tasks" edge to the Task entity. func (m *Mutation) RemovedTasksIDs() (ids []int) { for id := range m.removedtasks { ids = append(ids, id) } return } // TasksIDs returns the "tasks" edge IDs in the mutation. func (m *Mutation) TasksIDs() (ids []int) { for id := range m.tasks { ids = append(ids, id) } return } // ResetTasks resets all changes to the "tasks" edge. func (m *Mutation) ResetTasks() { m.tasks = nil m.clearedtasks = false m.removedtasks = nil } // Where appends a list predicates to the Mutation builder. func (m *Mutation) Where(ps ...predicate.User) { m.predicates = append(m.predicates, ps...) } // WhereP appends storage-level predicates to the Mutation builder. Using this method, // users can use type-assertion to append predicates that do not depend on any generated package. func (m *Mutation) WhereP(ps ...func(*sql.Selector)) { p := make([]predicate.User, len(ps)) for i := range ps { p[i] = ps[i] } m.Where(p...) } // Op returns the operation name. func (m *Mutation) Op() ent.Op { return m.op } // SetOp allows setting the mutation operation. func (m *Mutation) SetOp(op ent.Op) { m.op = op } // Type returns the node type of this mutation (User). func (m *Mutation) Type() string { return m.typ } // Fields returns all fields that were changed during this mutation. Note that in // order to get all numeric fields that were incremented/decremented, call // AddedFields(). func (m *Mutation) Fields() []string { fields := make([]string, 0, 2) if m.name != nil { fields = append(fields, FieldName) } if m.age != nil { fields = append(fields, FieldAge) } return fields } // Field returns the value of a field with the given name. The second boolean // return value indicates that this field was not set, or was not defined in the // schema. func (m *Mutation) Field(name string) (ent.Value, bool) { switch name { case FieldName: return m.Name() case FieldAge: return m.Age() } return nil, false } // OldField returns the old value of the field from the database. An error is // returned if the mutation operation is not UpdateOne, or the query to the // database failed. func (m *Mutation) OldField(ctx context.Context, name string) (ent.Value, error) { return nil, fmt.Errorf("unknown User field %s", name) } // SetField sets the value of a field with the given name. It returns an error if // the field is not defined in the schema, or if the type mismatched the field // type. func (m *Mutation) SetField(name string, value ent.Value) error { switch name { case FieldName: v, ok := value.(string) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetName(v) return nil case FieldAge: v, ok := value.(uint) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetAge(v) return nil } return fmt.Errorf("unknown User field %s", name) } // AddedFields returns all numeric fields that were incremented/decremented during // this mutation. func (m *Mutation) AddedFields() []string { var fields []string if m.addage != nil { fields = append(fields, FieldAge) } return fields } // AddedField returns the numeric value that was incremented/decremented on a field // with the given name. The second boolean return value indicates that this field // was not set, or was not defined in the schema. func (m *Mutation) AddedField(name string) (ent.Value, bool) { switch name { case FieldAge: return m.AddedAge() } return nil, false } // AddField adds the value to the field with the given name. It returns an error if // the field is not defined in the schema, or if the type mismatched the field // type. func (m *Mutation) AddField(name string, value ent.Value) error { switch name { case FieldAge: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.AddAge(v) return nil } return fmt.Errorf("unknown User numeric field %s", name) } // ClearedFields returns all nullable fields that were cleared during this // mutation. func (m *Mutation) ClearedFields() []string { var fields []string if m.FieldCleared(FieldAge) { fields = append(fields, FieldAge) } return fields } // FieldCleared returns a boolean indicating if a field with the given name was // cleared in this mutation. func (m *Mutation) FieldCleared(name string) bool { _, ok := m.clearedFields[name] return ok } // ClearField clears the value of the field with the given name. It returns an // error if the field is not defined in the schema. func (m *Mutation) ClearField(name string) error { switch name { case FieldAge: m.ClearAge() return nil } return fmt.Errorf("unknown User nullable field %s", name) } // ResetField resets all changes in the mutation for the field with the given name. // It returns an error if the field is not defined in the schema. func (m *Mutation) ResetField(name string) error { switch name { case FieldName: m.ResetName() return nil case FieldAge: m.ResetAge() return nil } return fmt.Errorf("unknown User field %s", name) } // AddedEdges returns all edge names that were set/added in this mutation. func (m *Mutation) AddedEdges() []string { edges := make([]string, 0, 2) if m.teams != nil { edges = append(edges, EdgeTeams) } if m.tasks != nil { edges = append(edges, EdgeTasks) } return edges } // AddedIDs returns all IDs (to other nodes) that were added for the given edge // name in this mutation. func (m *Mutation) AddedIDs(name string) []ent.Value { switch name { case EdgeTeams: ids := make([]ent.Value, 0, len(m.teams)) for id := range m.teams { ids = append(ids, id) } return ids case EdgeTasks: ids := make([]ent.Value, 0, len(m.tasks)) for id := range m.tasks { ids = append(ids, id) } return ids } return nil } // RemovedEdges returns all edge names that were removed in this mutation. func (m *Mutation) RemovedEdges() []string { edges := make([]string, 0, 2) if m.removedteams != nil { edges = append(edges, EdgeTeams) } if m.removedtasks != nil { edges = append(edges, EdgeTasks) } return edges } // RemovedIDs returns all IDs (to other nodes) that were removed for the edge with // the given name in this mutation. func (m *Mutation) RemovedIDs(name string) []ent.Value { switch name { case EdgeTeams: ids := make([]ent.Value, 0, len(m.removedteams)) for id := range m.removedteams { ids = append(ids, id) } return ids case EdgeTasks: ids := make([]ent.Value, 0, len(m.removedtasks)) for id := range m.removedtasks { ids = append(ids, id) } return ids } return nil } // ClearedEdges returns all edge names that were cleared in this mutation. func (m *Mutation) ClearedEdges() []string { edges := make([]string, 0, 2) if m.clearedteams { edges = append(edges, EdgeTeams) } if m.clearedtasks { edges = append(edges, EdgeTasks) } return edges } // EdgeCleared returns a boolean which indicates if the edge with the given name // was cleared in this mutation. func (m *Mutation) EdgeCleared(name string) bool { switch name { case EdgeTeams: return m.clearedteams case EdgeTasks: return m.clearedtasks } return false } // ClearEdge clears the value of the edge with the given name. It returns an error // if that edge is not defined in the schema. func (m *Mutation) ClearEdge(name string) error { switch name { } return fmt.Errorf("unknown User unique edge %s", name) } // ResetEdge resets all changes to the edge with the given name in this mutation. // It returns an error if the edge is not defined in the schema. func (m *Mutation) ResetEdge(name string) error { switch name { case EdgeTeams: m.ResetTeams() return nil case EdgeTasks: m.ResetTasks() return nil } return fmt.Errorf("unknown User edge %s", name) }