// 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/edgefield/ent/predicate" ) // Mutation represents an operation that mutates the User nodes in the graph. type Mutation struct { op ent.Op typ string clearedFields map[string]struct{} pets map[int]struct{} removedpets map[int]struct{} clearedpets bool parent *int clearedparent bool children map[int]struct{} removedchildren map[int]struct{} clearedchildren bool spouse *int clearedspouse bool card *int clearedcard bool metadata *int clearedmetadata bool info map[int]struct{} removedinfo map[int]struct{} clearedinfo bool rentals map[int]struct{} removedrentals map[int]struct{} clearedrentals 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 } // SetParentID sets the "parent_id" field. func (m *Mutation) SetParentID(i int) { m.parent = &i } // ParentID returns the value of the "parent_id" field in the mutation. func (m *Mutation) ParentID() (r int, exists bool) { v := m.parent if v == nil { return } return *v, true } // ClearParentID clears the value of the "parent_id" field. func (m *Mutation) ClearParentID() { m.parent = nil m.clearedFields[FieldParentID] = struct{}{} } // ParentIDCleared returns if the "parent_id" field was cleared in this mutation. func (m *Mutation) ParentIDCleared() bool { _, ok := m.clearedFields[FieldParentID] return ok } // ResetParentID resets all changes to the "parent_id" field. func (m *Mutation) ResetParentID() { m.parent = nil delete(m.clearedFields, FieldParentID) } // SetSpouseID sets the "spouse_id" field. func (m *Mutation) SetSpouseID(i int) { m.spouse = &i } // SpouseID returns the value of the "spouse_id" field in the mutation. func (m *Mutation) SpouseID() (r int, exists bool) { v := m.spouse if v == nil { return } return *v, true } // ClearSpouseID clears the value of the "spouse_id" field. func (m *Mutation) ClearSpouseID() { m.spouse = nil m.clearedFields[FieldSpouseID] = struct{}{} } // SpouseIDCleared returns if the "spouse_id" field was cleared in this mutation. func (m *Mutation) SpouseIDCleared() bool { _, ok := m.clearedFields[FieldSpouseID] return ok } // ResetSpouseID resets all changes to the "spouse_id" field. func (m *Mutation) ResetSpouseID() { m.spouse = nil delete(m.clearedFields, FieldSpouseID) } // AddPetIDs adds the "pets" edge to the Pet entity by ids. func (m *Mutation) AddPetIDs(ids ...int) { if m.pets == nil { m.pets = make(map[int]struct{}) } for i := range ids { m.pets[ids[i]] = struct{}{} } } // ClearPets clears the "pets" edge to the Pet entity. func (m *Mutation) ClearPets() { m.clearedpets = true } // PetsCleared reports if the "pets" edge to the Pet entity was cleared. func (m *Mutation) PetsCleared() bool { return m.clearedpets } // RemovePetIDs removes the "pets" edge to the Pet entity by IDs. func (m *Mutation) RemovePetIDs(ids ...int) { if m.removedpets == nil { m.removedpets = make(map[int]struct{}) } for i := range ids { delete(m.pets, ids[i]) m.removedpets[ids[i]] = struct{}{} } } // RemovedPets returns the removed IDs of the "pets" edge to the Pet entity. func (m *Mutation) RemovedPetsIDs() (ids []int) { for id := range m.removedpets { ids = append(ids, id) } return } // PetsIDs returns the "pets" edge IDs in the mutation. func (m *Mutation) PetsIDs() (ids []int) { for id := range m.pets { ids = append(ids, id) } return } // ResetPets resets all changes to the "pets" edge. func (m *Mutation) ResetPets() { m.pets = nil m.clearedpets = false m.removedpets = nil } // ClearParent clears the "parent" edge to the User entity. func (m *Mutation) ClearParent() { m.clearedparent = true m.clearedFields[FieldParentID] = struct{}{} } // ParentCleared reports if the "parent" edge to the User entity was cleared. func (m *Mutation) ParentCleared() bool { return m.ParentIDCleared() || m.clearedparent } // ParentIDs returns the "parent" edge IDs in the mutation. // Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use // ParentID instead. It exists only for internal usage by the builders. func (m *Mutation) ParentIDs() (ids []int) { if id := m.parent; id != nil { ids = append(ids, *id) } return } // ResetParent resets all changes to the "parent" edge. func (m *Mutation) ResetParent() { m.parent = nil m.clearedparent = false } // AddChildIDs adds the "children" edge to the User entity by ids. func (m *Mutation) AddChildIDs(ids ...int) { if m.children == nil { m.children = make(map[int]struct{}) } for i := range ids { m.children[ids[i]] = struct{}{} } } // ClearChildren clears the "children" edge to the User entity. func (m *Mutation) ClearChildren() { m.clearedchildren = true } // ChildrenCleared reports if the "children" edge to the User entity was cleared. func (m *Mutation) ChildrenCleared() bool { return m.clearedchildren } // RemoveChildIDs removes the "children" edge to the User entity by IDs. func (m *Mutation) RemoveChildIDs(ids ...int) { if m.removedchildren == nil { m.removedchildren = make(map[int]struct{}) } for i := range ids { delete(m.children, ids[i]) m.removedchildren[ids[i]] = struct{}{} } } // RemovedChildren returns the removed IDs of the "children" edge to the User entity. func (m *Mutation) RemovedChildrenIDs() (ids []int) { for id := range m.removedchildren { ids = append(ids, id) } return } // ChildrenIDs returns the "children" edge IDs in the mutation. func (m *Mutation) ChildrenIDs() (ids []int) { for id := range m.children { ids = append(ids, id) } return } // ResetChildren resets all changes to the "children" edge. func (m *Mutation) ResetChildren() { m.children = nil m.clearedchildren = false m.removedchildren = nil } // ClearSpouse clears the "spouse" edge to the User entity. func (m *Mutation) ClearSpouse() { m.clearedspouse = true m.clearedFields[FieldSpouseID] = struct{}{} } // SpouseCleared reports if the "spouse" edge to the User entity was cleared. func (m *Mutation) SpouseCleared() bool { return m.SpouseIDCleared() || m.clearedspouse } // SpouseIDs returns the "spouse" edge IDs in the mutation. // Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use // SpouseID instead. It exists only for internal usage by the builders. func (m *Mutation) SpouseIDs() (ids []int) { if id := m.spouse; id != nil { ids = append(ids, *id) } return } // ResetSpouse resets all changes to the "spouse" edge. func (m *Mutation) ResetSpouse() { m.spouse = nil m.clearedspouse = false } // SetCardID sets the "card" edge to the Card entity by id. func (m *Mutation) SetCardID(id int) { m.card = &id } // ClearCard clears the "card" edge to the Card entity. func (m *Mutation) ClearCard() { m.clearedcard = true } // CardCleared reports if the "card" edge to the Card entity was cleared. func (m *Mutation) CardCleared() bool { return m.clearedcard } // CardID returns the "card" edge ID in the mutation. func (m *Mutation) CardID() (id int, exists bool) { if m.card != nil { return *m.card, true } return } // CardIDs returns the "card" edge IDs in the mutation. // Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use // CardID instead. It exists only for internal usage by the builders. func (m *Mutation) CardIDs() (ids []int) { if id := m.card; id != nil { ids = append(ids, *id) } return } // ResetCard resets all changes to the "card" edge. func (m *Mutation) ResetCard() { m.card = nil m.clearedcard = false } // SetMetadataID sets the "metadata" edge to the Metadata entity by id. func (m *Mutation) SetMetadataID(id int) { m.metadata = &id } // ClearMetadata clears the "metadata" edge to the Metadata entity. func (m *Mutation) ClearMetadata() { m.clearedmetadata = true } // MetadataCleared reports if the "metadata" edge to the Metadata entity was cleared. func (m *Mutation) MetadataCleared() bool { return m.clearedmetadata } // MetadataID returns the "metadata" edge ID in the mutation. func (m *Mutation) MetadataID() (id int, exists bool) { if m.metadata != nil { return *m.metadata, true } return } // MetadataIDs returns the "metadata" edge IDs in the mutation. // Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use // MetadataID instead. It exists only for internal usage by the builders. func (m *Mutation) MetadataIDs() (ids []int) { if id := m.metadata; id != nil { ids = append(ids, *id) } return } // ResetMetadata resets all changes to the "metadata" edge. func (m *Mutation) ResetMetadata() { m.metadata = nil m.clearedmetadata = false } // AddInfoIDs adds the "info" edge to the Info entity by ids. func (m *Mutation) AddInfoIDs(ids ...int) { if m.info == nil { m.info = make(map[int]struct{}) } for i := range ids { m.info[ids[i]] = struct{}{} } } // ClearInfo clears the "info" edge to the Info entity. func (m *Mutation) ClearInfo() { m.clearedinfo = true } // InfoCleared reports if the "info" edge to the Info entity was cleared. func (m *Mutation) InfoCleared() bool { return m.clearedinfo } // RemoveInfoIDs removes the "info" edge to the Info entity by IDs. func (m *Mutation) RemoveInfoIDs(ids ...int) { if m.removedinfo == nil { m.removedinfo = make(map[int]struct{}) } for i := range ids { delete(m.info, ids[i]) m.removedinfo[ids[i]] = struct{}{} } } // RemovedInfo returns the removed IDs of the "info" edge to the Info entity. func (m *Mutation) RemovedInfoIDs() (ids []int) { for id := range m.removedinfo { ids = append(ids, id) } return } // InfoIDs returns the "info" edge IDs in the mutation. func (m *Mutation) InfoIDs() (ids []int) { for id := range m.info { ids = append(ids, id) } return } // ResetInfo resets all changes to the "info" edge. func (m *Mutation) ResetInfo() { m.info = nil m.clearedinfo = false m.removedinfo = nil } // AddRentalIDs adds the "rentals" edge to the Rental entity by ids. func (m *Mutation) AddRentalIDs(ids ...int) { if m.rentals == nil { m.rentals = make(map[int]struct{}) } for i := range ids { m.rentals[ids[i]] = struct{}{} } } // ClearRentals clears the "rentals" edge to the Rental entity. func (m *Mutation) ClearRentals() { m.clearedrentals = true } // RentalsCleared reports if the "rentals" edge to the Rental entity was cleared. func (m *Mutation) RentalsCleared() bool { return m.clearedrentals } // RemoveRentalIDs removes the "rentals" edge to the Rental entity by IDs. func (m *Mutation) RemoveRentalIDs(ids ...int) { if m.removedrentals == nil { m.removedrentals = make(map[int]struct{}) } for i := range ids { delete(m.rentals, ids[i]) m.removedrentals[ids[i]] = struct{}{} } } // RemovedRentals returns the removed IDs of the "rentals" edge to the Rental entity. func (m *Mutation) RemovedRentalsIDs() (ids []int) { for id := range m.removedrentals { ids = append(ids, id) } return } // RentalsIDs returns the "rentals" edge IDs in the mutation. func (m *Mutation) RentalsIDs() (ids []int) { for id := range m.rentals { ids = append(ids, id) } return } // ResetRentals resets all changes to the "rentals" edge. func (m *Mutation) ResetRentals() { m.rentals = nil m.clearedrentals = false m.removedrentals = 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.parent != nil { fields = append(fields, FieldParentID) } if m.spouse != nil { fields = append(fields, FieldSpouseID) } 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 FieldParentID: return m.ParentID() case FieldSpouseID: return m.SpouseID() } 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 FieldParentID: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetParentID(v) return nil case FieldSpouseID: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetSpouseID(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 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 { } 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 { } 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(FieldParentID) { fields = append(fields, FieldParentID) } if m.FieldCleared(FieldSpouseID) { fields = append(fields, FieldSpouseID) } 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 FieldParentID: m.ClearParentID() return nil case FieldSpouseID: m.ClearSpouseID() 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 FieldParentID: m.ResetParentID() return nil case FieldSpouseID: m.ResetSpouseID() 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, 8) if m.pets != nil { edges = append(edges, EdgePets) } if m.parent != nil { edges = append(edges, EdgeParent) } if m.children != nil { edges = append(edges, EdgeChildren) } if m.spouse != nil { edges = append(edges, EdgeSpouse) } if m.card != nil { edges = append(edges, EdgeCard) } if m.metadata != nil { edges = append(edges, EdgeMetadata) } if m.info != nil { edges = append(edges, EdgeInfo) } if m.rentals != nil { edges = append(edges, EdgeRentals) } 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 EdgePets: ids := make([]ent.Value, 0, len(m.pets)) for id := range m.pets { ids = append(ids, id) } return ids case EdgeParent: if id := m.parent; id != nil { return []ent.Value{*id} } case EdgeChildren: ids := make([]ent.Value, 0, len(m.children)) for id := range m.children { ids = append(ids, id) } return ids case EdgeSpouse: if id := m.spouse; id != nil { return []ent.Value{*id} } case EdgeCard: if id := m.card; id != nil { return []ent.Value{*id} } case EdgeMetadata: if id := m.metadata; id != nil { return []ent.Value{*id} } case EdgeInfo: ids := make([]ent.Value, 0, len(m.info)) for id := range m.info { ids = append(ids, id) } return ids case EdgeRentals: ids := make([]ent.Value, 0, len(m.rentals)) for id := range m.rentals { 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, 8) if m.removedpets != nil { edges = append(edges, EdgePets) } if m.removedchildren != nil { edges = append(edges, EdgeChildren) } if m.removedinfo != nil { edges = append(edges, EdgeInfo) } if m.removedrentals != nil { edges = append(edges, EdgeRentals) } 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 EdgePets: ids := make([]ent.Value, 0, len(m.removedpets)) for id := range m.removedpets { ids = append(ids, id) } return ids case EdgeChildren: ids := make([]ent.Value, 0, len(m.removedchildren)) for id := range m.removedchildren { ids = append(ids, id) } return ids case EdgeInfo: ids := make([]ent.Value, 0, len(m.removedinfo)) for id := range m.removedinfo { ids = append(ids, id) } return ids case EdgeRentals: ids := make([]ent.Value, 0, len(m.removedrentals)) for id := range m.removedrentals { 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, 8) if m.clearedpets { edges = append(edges, EdgePets) } if m.clearedparent { edges = append(edges, EdgeParent) } if m.clearedchildren { edges = append(edges, EdgeChildren) } if m.clearedspouse { edges = append(edges, EdgeSpouse) } if m.clearedcard { edges = append(edges, EdgeCard) } if m.clearedmetadata { edges = append(edges, EdgeMetadata) } if m.clearedinfo { edges = append(edges, EdgeInfo) } if m.clearedrentals { edges = append(edges, EdgeRentals) } 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 EdgePets: return m.clearedpets case EdgeParent: return m.clearedparent case EdgeChildren: return m.clearedchildren case EdgeSpouse: return m.clearedspouse case EdgeCard: return m.clearedcard case EdgeMetadata: return m.clearedmetadata case EdgeInfo: return m.clearedinfo case EdgeRentals: return m.clearedrentals } 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 { case EdgeParent: m.ClearParent() return nil case EdgeSpouse: m.ClearSpouse() return nil case EdgeCard: m.ClearCard() return nil case EdgeMetadata: m.ClearMetadata() return nil } 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 EdgePets: m.ResetPets() return nil case EdgeParent: m.ResetParent() return nil case EdgeChildren: m.ResetChildren() return nil case EdgeSpouse: m.ResetSpouse() return nil case EdgeCard: m.ResetCard() return nil case EdgeMetadata: m.ResetMetadata() return nil case EdgeInfo: m.ResetInfo() return nil case EdgeRentals: m.ResetRentals() return nil } return fmt.Errorf("unknown User edge %s", name) }