// 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 pet import ( "context" "fmt" "time" "entgo.io/ent" "entgo.io/ent/dialect/gremlin/graph/dsl" "entgo.io/ent/entc/integration/gremlin/ent/predicate" "github.com/google/uuid" ) // Mutation represents an operation that mutates the Pet nodes in the graph. type Mutation struct { op ent.Op typ string age *float64 addage *float64 name *string uuid *uuid.UUID nickname *string trained *bool optional_time *time.Time clearedFields map[string]struct{} team *string clearedteam bool owner *string clearedowner bool predicates []predicate.Pet } // NewMutation creates a new Mutation for the Pet entity. func NewMutation(op ent.Op) *Mutation { return &Mutation{ op: op, typ: "Pet", clearedFields: make(map[string]struct{}), } } // Predicates returns the list of predicates set on the mutation. func (m *Mutation) Predicates() []predicate.Pet { return m.predicates } // SetAge sets the "age" field. func (m *Mutation) SetAge(f float64) { m.age = &f m.addage = nil } // Age returns the value of the "age" field in the mutation. func (m *Mutation) Age() (r float64, exists bool) { v := m.age if v == nil { return } return *v, true } // AddAge adds f to the "age" field. func (m *Mutation) AddAge(f float64) { if m.addage != nil { *m.addage += f } else { m.addage = &f } } // AddedAge returns the value that was added to the "age" field in this mutation. func (m *Mutation) AddedAge() (r float64, exists bool) { v := m.addage if v == nil { return } return *v, true } // ResetAge resets all changes to the "age" field. func (m *Mutation) ResetAge() { m.age = nil m.addage = nil } // 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 } // SetUUID sets the "uuid" field. func (m *Mutation) SetUUID(u uuid.UUID) { m.uuid = &u } // UUID returns the value of the "uuid" field in the mutation. func (m *Mutation) UUID() (r uuid.UUID, exists bool) { v := m.uuid if v == nil { return } return *v, true } // ClearUUID clears the value of the "uuid" field. func (m *Mutation) ClearUUID() { m.uuid = nil m.clearedFields[FieldUUID] = struct{}{} } // UUIDCleared returns if the "uuid" field was cleared in this mutation. func (m *Mutation) UUIDCleared() bool { _, ok := m.clearedFields[FieldUUID] return ok } // ResetUUID resets all changes to the "uuid" field. func (m *Mutation) ResetUUID() { m.uuid = nil delete(m.clearedFields, FieldUUID) } // SetNickname sets the "nickname" field. func (m *Mutation) SetNickname(s string) { m.nickname = &s } // Nickname returns the value of the "nickname" field in the mutation. func (m *Mutation) Nickname() (r string, exists bool) { v := m.nickname if v == nil { return } return *v, true } // ClearNickname clears the value of the "nickname" field. func (m *Mutation) ClearNickname() { m.nickname = nil m.clearedFields[FieldNickname] = struct{}{} } // NicknameCleared returns if the "nickname" field was cleared in this mutation. func (m *Mutation) NicknameCleared() bool { _, ok := m.clearedFields[FieldNickname] return ok } // ResetNickname resets all changes to the "nickname" field. func (m *Mutation) ResetNickname() { m.nickname = nil delete(m.clearedFields, FieldNickname) } // SetTrained sets the "trained" field. func (m *Mutation) SetTrained(b bool) { m.trained = &b } // Trained returns the value of the "trained" field in the mutation. func (m *Mutation) Trained() (r bool, exists bool) { v := m.trained if v == nil { return } return *v, true } // ResetTrained resets all changes to the "trained" field. func (m *Mutation) ResetTrained() { m.trained = nil } // SetOptionalTime sets the "optional_time" field. func (m *Mutation) SetOptionalTime(t time.Time) { m.optional_time = &t } // OptionalTime returns the value of the "optional_time" field in the mutation. func (m *Mutation) OptionalTime() (r time.Time, exists bool) { v := m.optional_time if v == nil { return } return *v, true } // ClearOptionalTime clears the value of the "optional_time" field. func (m *Mutation) ClearOptionalTime() { m.optional_time = nil m.clearedFields[FieldOptionalTime] = struct{}{} } // OptionalTimeCleared returns if the "optional_time" field was cleared in this mutation. func (m *Mutation) OptionalTimeCleared() bool { _, ok := m.clearedFields[FieldOptionalTime] return ok } // ResetOptionalTime resets all changes to the "optional_time" field. func (m *Mutation) ResetOptionalTime() { m.optional_time = nil delete(m.clearedFields, FieldOptionalTime) } // SetTeamID sets the "team" edge to the User entity by id. func (m *Mutation) SetTeamID(id string) { m.team = &id } // ClearTeam clears the "team" edge to the User entity. func (m *Mutation) ClearTeam() { m.clearedteam = true } // TeamCleared reports if the "team" edge to the User entity was cleared. func (m *Mutation) TeamCleared() bool { return m.clearedteam } // TeamID returns the "team" edge ID in the mutation. func (m *Mutation) TeamID() (id string, exists bool) { if m.team != nil { return *m.team, true } return } // TeamIDs returns the "team" edge IDs in the mutation. // Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use // TeamID instead. It exists only for internal usage by the builders. func (m *Mutation) TeamIDs() (ids []string) { if id := m.team; id != nil { ids = append(ids, *id) } return } // ResetTeam resets all changes to the "team" edge. func (m *Mutation) ResetTeam() { m.team = nil m.clearedteam = false } // SetOwnerID sets the "owner" edge to the User entity by id. func (m *Mutation) SetOwnerID(id string) { m.owner = &id } // ClearOwner clears the "owner" edge to the User entity. func (m *Mutation) ClearOwner() { m.clearedowner = true } // OwnerCleared reports if the "owner" edge to the User entity was cleared. func (m *Mutation) OwnerCleared() bool { return m.clearedowner } // OwnerID returns the "owner" edge ID in the mutation. func (m *Mutation) OwnerID() (id string, exists bool) { if m.owner != nil { return *m.owner, true } return } // OwnerIDs returns the "owner" edge IDs in the mutation. // Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use // OwnerID instead. It exists only for internal usage by the builders. func (m *Mutation) OwnerIDs() (ids []string) { if id := m.owner; id != nil { ids = append(ids, *id) } return } // ResetOwner resets all changes to the "owner" edge. func (m *Mutation) ResetOwner() { m.owner = nil m.clearedowner = false } // Where appends a list predicates to the Mutation builder. func (m *Mutation) Where(ps ...predicate.Pet) { 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(*dsl.Traversal)) { p := make([]predicate.Pet, 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 (Pet). 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, 6) if m.age != nil { fields = append(fields, FieldAge) } if m.name != nil { fields = append(fields, FieldName) } if m.uuid != nil { fields = append(fields, FieldUUID) } if m.nickname != nil { fields = append(fields, FieldNickname) } if m.trained != nil { fields = append(fields, FieldTrained) } if m.optional_time != nil { fields = append(fields, FieldOptionalTime) } 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 FieldAge: return m.Age() case FieldName: return m.Name() case FieldUUID: return m.UUID() case FieldNickname: return m.Nickname() case FieldTrained: return m.Trained() case FieldOptionalTime: return m.OptionalTime() } 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 Pet 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 FieldAge: v, ok := value.(float64) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetAge(v) return nil 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 FieldUUID: v, ok := value.(uuid.UUID) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetUUID(v) return nil case FieldNickname: v, ok := value.(string) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetNickname(v) return nil case FieldTrained: v, ok := value.(bool) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetTrained(v) return nil case FieldOptionalTime: v, ok := value.(time.Time) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetOptionalTime(v) return nil } return fmt.Errorf("unknown Pet 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.(float64) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.AddAge(v) return nil } return fmt.Errorf("unknown Pet 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(FieldUUID) { fields = append(fields, FieldUUID) } if m.FieldCleared(FieldNickname) { fields = append(fields, FieldNickname) } if m.FieldCleared(FieldOptionalTime) { fields = append(fields, FieldOptionalTime) } 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 FieldUUID: m.ClearUUID() return nil case FieldNickname: m.ClearNickname() return nil case FieldOptionalTime: m.ClearOptionalTime() return nil } return fmt.Errorf("unknown Pet 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 FieldAge: m.ResetAge() return nil case FieldName: m.ResetName() return nil case FieldUUID: m.ResetUUID() return nil case FieldNickname: m.ResetNickname() return nil case FieldTrained: m.ResetTrained() return nil case FieldOptionalTime: m.ResetOptionalTime() return nil } return fmt.Errorf("unknown Pet 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.team != nil { edges = append(edges, EdgeTeam) } if m.owner != nil { edges = append(edges, EdgeOwner) } 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 EdgeTeam: if id := m.team; id != nil { return []ent.Value{*id} } case EdgeOwner: if id := m.owner; id != nil { return []ent.Value{*id} } } return nil } // RemovedEdges returns all edge names that were removed in this mutation. func (m *Mutation) RemovedEdges() []string { edges := make([]string, 0, 2) 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 { 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.clearedteam { edges = append(edges, EdgeTeam) } if m.clearedowner { edges = append(edges, EdgeOwner) } 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 EdgeTeam: return m.clearedteam case EdgeOwner: return m.clearedowner } 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 EdgeTeam: m.ClearTeam() return nil case EdgeOwner: m.ClearOwner() return nil } return fmt.Errorf("unknown Pet 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 EdgeTeam: m.ResetTeam() return nil case EdgeOwner: m.ResetOwner() return nil } return fmt.Errorf("unknown Pet edge %s", name) }