// 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 relationship import ( "context" "errors" "fmt" "entgo.io/ent" "entgo.io/ent/dialect/sql" "entgo.io/ent/entc/integration/edgeschema/ent/predicate" ) // Mutation represents an operation that mutates the Relationship nodes in the graph. type Mutation struct { op ent.Op typ string weight *int addweight *int clearedFields map[string]struct{} user *int cleareduser bool relative *int clearedrelative bool info *int clearedinfo bool predicates []predicate.Relationship } // NewMutation creates a new Mutation for the Relationship entity. func NewMutation(op ent.Op) *Mutation { return &Mutation{ op: op, typ: "Relationship", clearedFields: make(map[string]struct{}), } } // Predicates returns the list of predicates set on the mutation. func (m *Mutation) Predicates() []predicate.Relationship { return m.predicates } // SetWeight sets the "weight" field. func (m *Mutation) SetWeight(i int) { m.weight = &i m.addweight = nil } // Weight returns the value of the "weight" field in the mutation. func (m *Mutation) Weight() (r int, exists bool) { v := m.weight if v == nil { return } return *v, true } // AddWeight adds i to the "weight" field. func (m *Mutation) AddWeight(i int) { if m.addweight != nil { *m.addweight += i } else { m.addweight = &i } } // AddedWeight returns the value that was added to the "weight" field in this mutation. func (m *Mutation) AddedWeight() (r int, exists bool) { v := m.addweight if v == nil { return } return *v, true } // ResetWeight resets all changes to the "weight" field. func (m *Mutation) ResetWeight() { m.weight = nil m.addweight = nil } // SetUserID sets the "user_id" field. func (m *Mutation) SetUserID(i int) { m.user = &i } // UserID returns the value of the "user_id" field in the mutation. func (m *Mutation) UserID() (r int, exists bool) { v := m.user if v == nil { return } return *v, true } // ResetUserID resets all changes to the "user_id" field. func (m *Mutation) ResetUserID() { m.user = nil } // SetRelativeID sets the "relative_id" field. func (m *Mutation) SetRelativeID(i int) { m.relative = &i } // RelativeID returns the value of the "relative_id" field in the mutation. func (m *Mutation) RelativeID() (r int, exists bool) { v := m.relative if v == nil { return } return *v, true } // ResetRelativeID resets all changes to the "relative_id" field. func (m *Mutation) ResetRelativeID() { m.relative = nil } // SetInfoID sets the "info_id" field. func (m *Mutation) SetInfoID(i int) { m.info = &i } // InfoID returns the value of the "info_id" field in the mutation. func (m *Mutation) InfoID() (r int, exists bool) { v := m.info if v == nil { return } return *v, true } // ClearInfoID clears the value of the "info_id" field. func (m *Mutation) ClearInfoID() { m.info = nil m.clearedFields[FieldInfoID] = struct{}{} } // InfoIDCleared returns if the "info_id" field was cleared in this mutation. func (m *Mutation) InfoIDCleared() bool { _, ok := m.clearedFields[FieldInfoID] return ok } // ResetInfoID resets all changes to the "info_id" field. func (m *Mutation) ResetInfoID() { m.info = nil delete(m.clearedFields, FieldInfoID) } // ClearUser clears the "user" edge to the User entity. func (m *Mutation) ClearUser() { m.cleareduser = true m.clearedFields[FieldUserID] = struct{}{} } // UserCleared reports if the "user" edge to the User entity was cleared. func (m *Mutation) UserCleared() bool { return m.cleareduser } // UserIDs returns the "user" edge IDs in the mutation. // Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use // UserID instead. It exists only for internal usage by the builders. func (m *Mutation) UserIDs() (ids []int) { if id := m.user; id != nil { ids = append(ids, *id) } return } // ResetUser resets all changes to the "user" edge. func (m *Mutation) ResetUser() { m.user = nil m.cleareduser = false } // ClearRelative clears the "relative" edge to the User entity. func (m *Mutation) ClearRelative() { m.clearedrelative = true m.clearedFields[FieldRelativeID] = struct{}{} } // RelativeCleared reports if the "relative" edge to the User entity was cleared. func (m *Mutation) RelativeCleared() bool { return m.clearedrelative } // RelativeIDs returns the "relative" edge IDs in the mutation. // Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use // RelativeID instead. It exists only for internal usage by the builders. func (m *Mutation) RelativeIDs() (ids []int) { if id := m.relative; id != nil { ids = append(ids, *id) } return } // ResetRelative resets all changes to the "relative" edge. func (m *Mutation) ResetRelative() { m.relative = nil m.clearedrelative = false } // ClearInfo clears the "info" edge to the RelationshipInfo entity. func (m *Mutation) ClearInfo() { m.clearedinfo = true m.clearedFields[FieldInfoID] = struct{}{} } // InfoCleared reports if the "info" edge to the RelationshipInfo entity was cleared. func (m *Mutation) InfoCleared() bool { return m.InfoIDCleared() || m.clearedinfo } // InfoIDs returns the "info" edge IDs in the mutation. // Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use // InfoID instead. It exists only for internal usage by the builders. func (m *Mutation) InfoIDs() (ids []int) { if id := m.info; id != nil { ids = append(ids, *id) } return } // ResetInfo resets all changes to the "info" edge. func (m *Mutation) ResetInfo() { m.info = nil m.clearedinfo = false } // Where appends a list predicates to the Mutation builder. func (m *Mutation) Where(ps ...predicate.Relationship) { 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.Relationship, 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 (Relationship). 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, 4) if m.weight != nil { fields = append(fields, FieldWeight) } if m.user != nil { fields = append(fields, FieldUserID) } if m.relative != nil { fields = append(fields, FieldRelativeID) } if m.info != nil { fields = append(fields, FieldInfoID) } 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 FieldWeight: return m.Weight() case FieldUserID: return m.UserID() case FieldRelativeID: return m.RelativeID() case FieldInfoID: return m.InfoID() } 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, errors.New("edge schema Relationship does not support getting old values") } // 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 FieldWeight: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetWeight(v) return nil case FieldUserID: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetUserID(v) return nil case FieldRelativeID: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetRelativeID(v) return nil case FieldInfoID: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetInfoID(v) return nil } return fmt.Errorf("unknown Relationship 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.addweight != nil { fields = append(fields, FieldWeight) } 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 FieldWeight: return m.AddedWeight() } 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 FieldWeight: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.AddWeight(v) return nil } return fmt.Errorf("unknown Relationship 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(FieldInfoID) { fields = append(fields, FieldInfoID) } 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 FieldInfoID: m.ClearInfoID() return nil } return fmt.Errorf("unknown Relationship 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 FieldWeight: m.ResetWeight() return nil case FieldUserID: m.ResetUserID() return nil case FieldRelativeID: m.ResetRelativeID() return nil case FieldInfoID: m.ResetInfoID() return nil } return fmt.Errorf("unknown Relationship field %s", name) } // AddedEdges returns all edge names that were set/added in this mutation. func (m *Mutation) AddedEdges() []string { edges := make([]string, 0, 3) if m.user != nil { edges = append(edges, EdgeUser) } if m.relative != nil { edges = append(edges, EdgeRelative) } if m.info != nil { edges = append(edges, EdgeInfo) } 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 EdgeUser: if id := m.user; id != nil { return []ent.Value{*id} } case EdgeRelative: if id := m.relative; id != nil { return []ent.Value{*id} } case EdgeInfo: if id := m.info; 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, 3) 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, 3) if m.cleareduser { edges = append(edges, EdgeUser) } if m.clearedrelative { edges = append(edges, EdgeRelative) } if m.clearedinfo { edges = append(edges, EdgeInfo) } 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 EdgeUser: return m.cleareduser case EdgeRelative: return m.clearedrelative case EdgeInfo: return m.clearedinfo } 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 EdgeUser: m.ClearUser() return nil case EdgeRelative: m.ClearRelative() return nil case EdgeInfo: m.ClearInfo() return nil } return fmt.Errorf("unknown Relationship 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 EdgeUser: m.ResetUser() return nil case EdgeRelative: m.ResetRelative() return nil case EdgeInfo: m.ResetInfo() return nil } return fmt.Errorf("unknown Relationship edge %s", name) }