865 lines
21 KiB
Go
865 lines
21 KiB
Go
// 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)
|
|
}
|