// 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 versioned import ( "context" "errors" "fmt" "sync" "time" "entgo.io/ent" "entgo.io/ent/dialect/sql" "entgo.io/ent/entc/integration/multischema/versioned/friendship" "entgo.io/ent/entc/integration/multischema/versioned/group" "entgo.io/ent/entc/integration/multischema/versioned/pet" "entgo.io/ent/entc/integration/multischema/versioned/predicate" "entgo.io/ent/entc/integration/multischema/versioned/user" ) const ( // Operation types. OpCreate = ent.OpCreate OpDelete = ent.OpDelete OpDeleteOne = ent.OpDeleteOne OpUpdate = ent.OpUpdate OpUpdateOne = ent.OpUpdateOne // Node types. TypeFriendship = "Friendship" TypeGroup = "Group" TypePet = "Pet" TypeUser = "User" ) // FriendshipMutation represents an operation that mutates the Friendship nodes in the graph. type FriendshipMutation struct { config op Op typ string id *int weight *int addweight *int created_at *time.Time clearedFields map[string]struct{} user *int cleareduser bool friend *int clearedfriend bool done bool oldValue func(context.Context) (*Friendship, error) predicates []predicate.Friendship } var _ ent.Mutation = (*FriendshipMutation)(nil) // friendshipOption allows management of the mutation configuration using functional options. type friendshipOption func(*FriendshipMutation) // newFriendshipMutation creates new mutation for the Friendship entity. func newFriendshipMutation(c config, op Op, opts ...friendshipOption) *FriendshipMutation { m := &FriendshipMutation{ config: c, op: op, typ: TypeFriendship, clearedFields: make(map[string]struct{}), } for _, opt := range opts { opt(m) } return m } // withFriendshipID sets the ID field of the mutation. func withFriendshipID(id int) friendshipOption { return func(m *FriendshipMutation) { var ( err error once sync.Once value *Friendship ) m.oldValue = func(ctx context.Context) (*Friendship, error) { once.Do(func() { if m.done { err = errors.New("querying old values post mutation is not allowed") } else { value, err = m.Client().Friendship.Get(ctx, id) } }) return value, err } m.id = &id } } // withFriendship sets the old Friendship of the mutation. func withFriendship(node *Friendship) friendshipOption { return func(m *FriendshipMutation) { m.oldValue = func(context.Context) (*Friendship, error) { return node, nil } m.id = &node.ID } } // Client returns a new `ent.Client` from the mutation. If the mutation was // executed in a transaction (ent.Tx), a transactional client is returned. func (m FriendshipMutation) Client() *Client { client := &Client{config: m.config} client.init() return client } // Tx returns an `ent.Tx` for mutations that were executed in transactions; // it returns an error otherwise. func (m FriendshipMutation) Tx() (*Tx, error) { if _, ok := m.driver.(*txDriver); !ok { return nil, errors.New("versioned: mutation is not running in a transaction") } tx := &Tx{config: m.config} tx.init() return tx, nil } // ID returns the ID value in the mutation. Note that the ID is only available // if it was provided to the builder or after it was returned from the database. func (m *FriendshipMutation) ID() (id int, exists bool) { if m.id == nil { return } return *m.id, true } // IDs queries the database and returns the entity ids that match the mutation's predicate. // That means, if the mutation is applied within a transaction with an isolation level such // as sql.LevelSerializable, the returned ids match the ids of the rows that will be updated // or updated by the mutation. func (m *FriendshipMutation) IDs(ctx context.Context) ([]int, error) { switch { case m.op.Is(OpUpdateOne | OpDeleteOne): id, exists := m.ID() if exists { return []int{id}, nil } fallthrough case m.op.Is(OpUpdate | OpDelete): return m.Client().Friendship.Query().Where(m.predicates...).IDs(ctx) default: return nil, fmt.Errorf("IDs is not allowed on %s operations", m.op) } } // SetWeight sets the "weight" field. func (m *FriendshipMutation) SetWeight(i int) { m.weight = &i m.addweight = nil } // Weight returns the value of the "weight" field in the mutation. func (m *FriendshipMutation) Weight() (r int, exists bool) { v := m.weight if v == nil { return } return *v, true } // OldWeight returns the old "weight" field's value of the Friendship entity. // If the Friendship object wasn't provided to the builder, the object is fetched from the database. // An error is returned if the mutation operation is not UpdateOne, or the database query fails. func (m *FriendshipMutation) OldWeight(ctx context.Context) (v int, err error) { if !m.op.Is(OpUpdateOne) { return v, errors.New("OldWeight is only allowed on UpdateOne operations") } if m.id == nil || m.oldValue == nil { return v, errors.New("OldWeight requires an ID field in the mutation") } oldValue, err := m.oldValue(ctx) if err != nil { return v, fmt.Errorf("querying old value for OldWeight: %w", err) } return oldValue.Weight, nil } // AddWeight adds i to the "weight" field. func (m *FriendshipMutation) 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 *FriendshipMutation) 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 *FriendshipMutation) ResetWeight() { m.weight = nil m.addweight = nil } // SetCreatedAt sets the "created_at" field. func (m *FriendshipMutation) SetCreatedAt(t time.Time) { m.created_at = &t } // CreatedAt returns the value of the "created_at" field in the mutation. func (m *FriendshipMutation) CreatedAt() (r time.Time, exists bool) { v := m.created_at if v == nil { return } return *v, true } // OldCreatedAt returns the old "created_at" field's value of the Friendship entity. // If the Friendship object wasn't provided to the builder, the object is fetched from the database. // An error is returned if the mutation operation is not UpdateOne, or the database query fails. func (m *FriendshipMutation) OldCreatedAt(ctx context.Context) (v time.Time, err error) { if !m.op.Is(OpUpdateOne) { return v, errors.New("OldCreatedAt is only allowed on UpdateOne operations") } if m.id == nil || m.oldValue == nil { return v, errors.New("OldCreatedAt requires an ID field in the mutation") } oldValue, err := m.oldValue(ctx) if err != nil { return v, fmt.Errorf("querying old value for OldCreatedAt: %w", err) } return oldValue.CreatedAt, nil } // ResetCreatedAt resets all changes to the "created_at" field. func (m *FriendshipMutation) ResetCreatedAt() { m.created_at = nil } // SetUserID sets the "user_id" field. func (m *FriendshipMutation) SetUserID(i int) { m.user = &i } // UserID returns the value of the "user_id" field in the mutation. func (m *FriendshipMutation) UserID() (r int, exists bool) { v := m.user if v == nil { return } return *v, true } // OldUserID returns the old "user_id" field's value of the Friendship entity. // If the Friendship object wasn't provided to the builder, the object is fetched from the database. // An error is returned if the mutation operation is not UpdateOne, or the database query fails. func (m *FriendshipMutation) OldUserID(ctx context.Context) (v int, err error) { if !m.op.Is(OpUpdateOne) { return v, errors.New("OldUserID is only allowed on UpdateOne operations") } if m.id == nil || m.oldValue == nil { return v, errors.New("OldUserID requires an ID field in the mutation") } oldValue, err := m.oldValue(ctx) if err != nil { return v, fmt.Errorf("querying old value for OldUserID: %w", err) } return oldValue.UserID, nil } // ResetUserID resets all changes to the "user_id" field. func (m *FriendshipMutation) ResetUserID() { m.user = nil } // SetFriendID sets the "friend_id" field. func (m *FriendshipMutation) SetFriendID(i int) { m.friend = &i } // FriendID returns the value of the "friend_id" field in the mutation. func (m *FriendshipMutation) FriendID() (r int, exists bool) { v := m.friend if v == nil { return } return *v, true } // OldFriendID returns the old "friend_id" field's value of the Friendship entity. // If the Friendship object wasn't provided to the builder, the object is fetched from the database. // An error is returned if the mutation operation is not UpdateOne, or the database query fails. func (m *FriendshipMutation) OldFriendID(ctx context.Context) (v int, err error) { if !m.op.Is(OpUpdateOne) { return v, errors.New("OldFriendID is only allowed on UpdateOne operations") } if m.id == nil || m.oldValue == nil { return v, errors.New("OldFriendID requires an ID field in the mutation") } oldValue, err := m.oldValue(ctx) if err != nil { return v, fmt.Errorf("querying old value for OldFriendID: %w", err) } return oldValue.FriendID, nil } // ResetFriendID resets all changes to the "friend_id" field. func (m *FriendshipMutation) ResetFriendID() { m.friend = nil } // ClearUser clears the "user" edge to the User entity. func (m *FriendshipMutation) ClearUser() { m.cleareduser = true m.clearedFields[friendship.FieldUserID] = struct{}{} } // UserCleared reports if the "user" edge to the User entity was cleared. func (m *FriendshipMutation) 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 *FriendshipMutation) UserIDs() (ids []int) { if id := m.user; id != nil { ids = append(ids, *id) } return } // ResetUser resets all changes to the "user" edge. func (m *FriendshipMutation) ResetUser() { m.user = nil m.cleareduser = false } // ClearFriend clears the "friend" edge to the User entity. func (m *FriendshipMutation) ClearFriend() { m.clearedfriend = true m.clearedFields[friendship.FieldFriendID] = struct{}{} } // FriendCleared reports if the "friend" edge to the User entity was cleared. func (m *FriendshipMutation) FriendCleared() bool { return m.clearedfriend } // FriendIDs returns the "friend" edge IDs in the mutation. // Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use // FriendID instead. It exists only for internal usage by the builders. func (m *FriendshipMutation) FriendIDs() (ids []int) { if id := m.friend; id != nil { ids = append(ids, *id) } return } // ResetFriend resets all changes to the "friend" edge. func (m *FriendshipMutation) ResetFriend() { m.friend = nil m.clearedfriend = false } // Where appends a list predicates to the FriendshipMutation builder. func (m *FriendshipMutation) Where(ps ...predicate.Friendship) { m.predicates = append(m.predicates, ps...) } // WhereP appends storage-level predicates to the FriendshipMutation builder. Using this method, // users can use type-assertion to append predicates that do not depend on any generated package. func (m *FriendshipMutation) WhereP(ps ...func(*sql.Selector)) { p := make([]predicate.Friendship, len(ps)) for i := range ps { p[i] = ps[i] } m.Where(p...) } // Op returns the operation name. func (m *FriendshipMutation) Op() Op { return m.op } // SetOp allows setting the mutation operation. func (m *FriendshipMutation) SetOp(op Op) { m.op = op } // Type returns the node type of this mutation (Friendship). func (m *FriendshipMutation) 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 *FriendshipMutation) Fields() []string { fields := make([]string, 0, 4) if m.weight != nil { fields = append(fields, friendship.FieldWeight) } if m.created_at != nil { fields = append(fields, friendship.FieldCreatedAt) } if m.user != nil { fields = append(fields, friendship.FieldUserID) } if m.friend != nil { fields = append(fields, friendship.FieldFriendID) } 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 *FriendshipMutation) Field(name string) (ent.Value, bool) { switch name { case friendship.FieldWeight: return m.Weight() case friendship.FieldCreatedAt: return m.CreatedAt() case friendship.FieldUserID: return m.UserID() case friendship.FieldFriendID: return m.FriendID() } 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 *FriendshipMutation) OldField(ctx context.Context, name string) (ent.Value, error) { switch name { case friendship.FieldWeight: return m.OldWeight(ctx) case friendship.FieldCreatedAt: return m.OldCreatedAt(ctx) case friendship.FieldUserID: return m.OldUserID(ctx) case friendship.FieldFriendID: return m.OldFriendID(ctx) } return nil, fmt.Errorf("unknown Friendship 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 *FriendshipMutation) SetField(name string, value ent.Value) error { switch name { case friendship.FieldWeight: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetWeight(v) return nil case friendship.FieldCreatedAt: v, ok := value.(time.Time) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetCreatedAt(v) return nil case friendship.FieldUserID: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetUserID(v) return nil case friendship.FieldFriendID: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetFriendID(v) return nil } return fmt.Errorf("unknown Friendship field %s", name) } // AddedFields returns all numeric fields that were incremented/decremented during // this mutation. func (m *FriendshipMutation) AddedFields() []string { var fields []string if m.addweight != nil { fields = append(fields, friendship.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 *FriendshipMutation) AddedField(name string) (ent.Value, bool) { switch name { case friendship.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 *FriendshipMutation) AddField(name string, value ent.Value) error { switch name { case friendship.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 Friendship numeric field %s", name) } // ClearedFields returns all nullable fields that were cleared during this // mutation. func (m *FriendshipMutation) ClearedFields() []string { return nil } // FieldCleared returns a boolean indicating if a field with the given name was // cleared in this mutation. func (m *FriendshipMutation) 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 *FriendshipMutation) ClearField(name string) error { return fmt.Errorf("unknown Friendship 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 *FriendshipMutation) ResetField(name string) error { switch name { case friendship.FieldWeight: m.ResetWeight() return nil case friendship.FieldCreatedAt: m.ResetCreatedAt() return nil case friendship.FieldUserID: m.ResetUserID() return nil case friendship.FieldFriendID: m.ResetFriendID() return nil } return fmt.Errorf("unknown Friendship field %s", name) } // AddedEdges returns all edge names that were set/added in this mutation. func (m *FriendshipMutation) AddedEdges() []string { edges := make([]string, 0, 2) if m.user != nil { edges = append(edges, friendship.EdgeUser) } if m.friend != nil { edges = append(edges, friendship.EdgeFriend) } return edges } // AddedIDs returns all IDs (to other nodes) that were added for the given edge // name in this mutation. func (m *FriendshipMutation) AddedIDs(name string) []ent.Value { switch name { case friendship.EdgeUser: if id := m.user; id != nil { return []ent.Value{*id} } case friendship.EdgeFriend: if id := m.friend; id != nil { return []ent.Value{*id} } } return nil } // RemovedEdges returns all edge names that were removed in this mutation. func (m *FriendshipMutation) 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 *FriendshipMutation) RemovedIDs(name string) []ent.Value { return nil } // ClearedEdges returns all edge names that were cleared in this mutation. func (m *FriendshipMutation) ClearedEdges() []string { edges := make([]string, 0, 2) if m.cleareduser { edges = append(edges, friendship.EdgeUser) } if m.clearedfriend { edges = append(edges, friendship.EdgeFriend) } return edges } // EdgeCleared returns a boolean which indicates if the edge with the given name // was cleared in this mutation. func (m *FriendshipMutation) EdgeCleared(name string) bool { switch name { case friendship.EdgeUser: return m.cleareduser case friendship.EdgeFriend: return m.clearedfriend } 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 *FriendshipMutation) ClearEdge(name string) error { switch name { case friendship.EdgeUser: m.ClearUser() return nil case friendship.EdgeFriend: m.ClearFriend() return nil } return fmt.Errorf("unknown Friendship 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 *FriendshipMutation) ResetEdge(name string) error { switch name { case friendship.EdgeUser: m.ResetUser() return nil case friendship.EdgeFriend: m.ResetFriend() return nil } return fmt.Errorf("unknown Friendship edge %s", name) } // GroupMutation represents an operation that mutates the Group nodes in the graph. type GroupMutation struct { config op Op typ string id *int name *string clearedFields map[string]struct{} users map[int]struct{} removedusers map[int]struct{} clearedusers bool done bool oldValue func(context.Context) (*Group, error) predicates []predicate.Group } var _ ent.Mutation = (*GroupMutation)(nil) // groupOption allows management of the mutation configuration using functional options. type groupOption func(*GroupMutation) // newGroupMutation creates new mutation for the Group entity. func newGroupMutation(c config, op Op, opts ...groupOption) *GroupMutation { m := &GroupMutation{ config: c, op: op, typ: TypeGroup, clearedFields: make(map[string]struct{}), } for _, opt := range opts { opt(m) } return m } // withGroupID sets the ID field of the mutation. func withGroupID(id int) groupOption { return func(m *GroupMutation) { var ( err error once sync.Once value *Group ) m.oldValue = func(ctx context.Context) (*Group, error) { once.Do(func() { if m.done { err = errors.New("querying old values post mutation is not allowed") } else { value, err = m.Client().Group.Get(ctx, id) } }) return value, err } m.id = &id } } // withGroup sets the old Group of the mutation. func withGroup(node *Group) groupOption { return func(m *GroupMutation) { m.oldValue = func(context.Context) (*Group, error) { return node, nil } m.id = &node.ID } } // Client returns a new `ent.Client` from the mutation. If the mutation was // executed in a transaction (ent.Tx), a transactional client is returned. func (m GroupMutation) Client() *Client { client := &Client{config: m.config} client.init() return client } // Tx returns an `ent.Tx` for mutations that were executed in transactions; // it returns an error otherwise. func (m GroupMutation) Tx() (*Tx, error) { if _, ok := m.driver.(*txDriver); !ok { return nil, errors.New("versioned: mutation is not running in a transaction") } tx := &Tx{config: m.config} tx.init() return tx, nil } // ID returns the ID value in the mutation. Note that the ID is only available // if it was provided to the builder or after it was returned from the database. func (m *GroupMutation) ID() (id int, exists bool) { if m.id == nil { return } return *m.id, true } // IDs queries the database and returns the entity ids that match the mutation's predicate. // That means, if the mutation is applied within a transaction with an isolation level such // as sql.LevelSerializable, the returned ids match the ids of the rows that will be updated // or updated by the mutation. func (m *GroupMutation) IDs(ctx context.Context) ([]int, error) { switch { case m.op.Is(OpUpdateOne | OpDeleteOne): id, exists := m.ID() if exists { return []int{id}, nil } fallthrough case m.op.Is(OpUpdate | OpDelete): return m.Client().Group.Query().Where(m.predicates...).IDs(ctx) default: return nil, fmt.Errorf("IDs is not allowed on %s operations", m.op) } } // SetName sets the "name" field. func (m *GroupMutation) SetName(s string) { m.name = &s } // Name returns the value of the "name" field in the mutation. func (m *GroupMutation) Name() (r string, exists bool) { v := m.name if v == nil { return } return *v, true } // OldName returns the old "name" field's value of the Group entity. // If the Group object wasn't provided to the builder, the object is fetched from the database. // An error is returned if the mutation operation is not UpdateOne, or the database query fails. func (m *GroupMutation) OldName(ctx context.Context) (v string, err error) { if !m.op.Is(OpUpdateOne) { return v, errors.New("OldName is only allowed on UpdateOne operations") } if m.id == nil || m.oldValue == nil { return v, errors.New("OldName requires an ID field in the mutation") } oldValue, err := m.oldValue(ctx) if err != nil { return v, fmt.Errorf("querying old value for OldName: %w", err) } return oldValue.Name, nil } // ResetName resets all changes to the "name" field. func (m *GroupMutation) ResetName() { m.name = nil } // AddUserIDs adds the "users" edge to the User entity by ids. func (m *GroupMutation) AddUserIDs(ids ...int) { if m.users == nil { m.users = make(map[int]struct{}) } for i := range ids { m.users[ids[i]] = struct{}{} } } // ClearUsers clears the "users" edge to the User entity. func (m *GroupMutation) ClearUsers() { m.clearedusers = true } // UsersCleared reports if the "users" edge to the User entity was cleared. func (m *GroupMutation) UsersCleared() bool { return m.clearedusers } // RemoveUserIDs removes the "users" edge to the User entity by IDs. func (m *GroupMutation) RemoveUserIDs(ids ...int) { if m.removedusers == nil { m.removedusers = make(map[int]struct{}) } for i := range ids { delete(m.users, ids[i]) m.removedusers[ids[i]] = struct{}{} } } // RemovedUsers returns the removed IDs of the "users" edge to the User entity. func (m *GroupMutation) RemovedUsersIDs() (ids []int) { for id := range m.removedusers { ids = append(ids, id) } return } // UsersIDs returns the "users" edge IDs in the mutation. func (m *GroupMutation) UsersIDs() (ids []int) { for id := range m.users { ids = append(ids, id) } return } // ResetUsers resets all changes to the "users" edge. func (m *GroupMutation) ResetUsers() { m.users = nil m.clearedusers = false m.removedusers = nil } // Where appends a list predicates to the GroupMutation builder. func (m *GroupMutation) Where(ps ...predicate.Group) { m.predicates = append(m.predicates, ps...) } // WhereP appends storage-level predicates to the GroupMutation builder. Using this method, // users can use type-assertion to append predicates that do not depend on any generated package. func (m *GroupMutation) WhereP(ps ...func(*sql.Selector)) { p := make([]predicate.Group, len(ps)) for i := range ps { p[i] = ps[i] } m.Where(p...) } // Op returns the operation name. func (m *GroupMutation) Op() Op { return m.op } // SetOp allows setting the mutation operation. func (m *GroupMutation) SetOp(op Op) { m.op = op } // Type returns the node type of this mutation (Group). func (m *GroupMutation) 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 *GroupMutation) Fields() []string { fields := make([]string, 0, 1) if m.name != nil { fields = append(fields, group.FieldName) } 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 *GroupMutation) Field(name string) (ent.Value, bool) { switch name { case group.FieldName: return m.Name() } 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 *GroupMutation) OldField(ctx context.Context, name string) (ent.Value, error) { switch name { case group.FieldName: return m.OldName(ctx) } return nil, fmt.Errorf("unknown Group 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 *GroupMutation) SetField(name string, value ent.Value) error { switch name { case group.FieldName: v, ok := value.(string) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetName(v) return nil } return fmt.Errorf("unknown Group field %s", name) } // AddedFields returns all numeric fields that were incremented/decremented during // this mutation. func (m *GroupMutation) AddedFields() []string { return nil } // 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 *GroupMutation) AddedField(name string) (ent.Value, bool) { 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 *GroupMutation) AddField(name string, value ent.Value) error { switch name { } return fmt.Errorf("unknown Group numeric field %s", name) } // ClearedFields returns all nullable fields that were cleared during this // mutation. func (m *GroupMutation) ClearedFields() []string { return nil } // FieldCleared returns a boolean indicating if a field with the given name was // cleared in this mutation. func (m *GroupMutation) 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 *GroupMutation) ClearField(name string) error { return fmt.Errorf("unknown Group 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 *GroupMutation) ResetField(name string) error { switch name { case group.FieldName: m.ResetName() return nil } return fmt.Errorf("unknown Group field %s", name) } // AddedEdges returns all edge names that were set/added in this mutation. func (m *GroupMutation) AddedEdges() []string { edges := make([]string, 0, 1) if m.users != nil { edges = append(edges, group.EdgeUsers) } return edges } // AddedIDs returns all IDs (to other nodes) that were added for the given edge // name in this mutation. func (m *GroupMutation) AddedIDs(name string) []ent.Value { switch name { case group.EdgeUsers: ids := make([]ent.Value, 0, len(m.users)) for id := range m.users { ids = append(ids, id) } return ids } return nil } // RemovedEdges returns all edge names that were removed in this mutation. func (m *GroupMutation) RemovedEdges() []string { edges := make([]string, 0, 1) if m.removedusers != nil { edges = append(edges, group.EdgeUsers) } return edges } // RemovedIDs returns all IDs (to other nodes) that were removed for the edge with // the given name in this mutation. func (m *GroupMutation) RemovedIDs(name string) []ent.Value { switch name { case group.EdgeUsers: ids := make([]ent.Value, 0, len(m.removedusers)) for id := range m.removedusers { ids = append(ids, id) } return ids } return nil } // ClearedEdges returns all edge names that were cleared in this mutation. func (m *GroupMutation) ClearedEdges() []string { edges := make([]string, 0, 1) if m.clearedusers { edges = append(edges, group.EdgeUsers) } return edges } // EdgeCleared returns a boolean which indicates if the edge with the given name // was cleared in this mutation. func (m *GroupMutation) EdgeCleared(name string) bool { switch name { case group.EdgeUsers: return m.clearedusers } 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 *GroupMutation) ClearEdge(name string) error { switch name { } return fmt.Errorf("unknown Group 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 *GroupMutation) ResetEdge(name string) error { switch name { case group.EdgeUsers: m.ResetUsers() return nil } return fmt.Errorf("unknown Group edge %s", name) } // PetMutation represents an operation that mutates the Pet nodes in the graph. type PetMutation struct { config op Op typ string id *int name *string clearedFields map[string]struct{} owner *int clearedowner bool done bool oldValue func(context.Context) (*Pet, error) predicates []predicate.Pet } var _ ent.Mutation = (*PetMutation)(nil) // petOption allows management of the mutation configuration using functional options. type petOption func(*PetMutation) // newPetMutation creates new mutation for the Pet entity. func newPetMutation(c config, op Op, opts ...petOption) *PetMutation { m := &PetMutation{ config: c, op: op, typ: TypePet, clearedFields: make(map[string]struct{}), } for _, opt := range opts { opt(m) } return m } // withPetID sets the ID field of the mutation. func withPetID(id int) petOption { return func(m *PetMutation) { var ( err error once sync.Once value *Pet ) m.oldValue = func(ctx context.Context) (*Pet, error) { once.Do(func() { if m.done { err = errors.New("querying old values post mutation is not allowed") } else { value, err = m.Client().Pet.Get(ctx, id) } }) return value, err } m.id = &id } } // withPet sets the old Pet of the mutation. func withPet(node *Pet) petOption { return func(m *PetMutation) { m.oldValue = func(context.Context) (*Pet, error) { return node, nil } m.id = &node.ID } } // Client returns a new `ent.Client` from the mutation. If the mutation was // executed in a transaction (ent.Tx), a transactional client is returned. func (m PetMutation) Client() *Client { client := &Client{config: m.config} client.init() return client } // Tx returns an `ent.Tx` for mutations that were executed in transactions; // it returns an error otherwise. func (m PetMutation) Tx() (*Tx, error) { if _, ok := m.driver.(*txDriver); !ok { return nil, errors.New("versioned: mutation is not running in a transaction") } tx := &Tx{config: m.config} tx.init() return tx, nil } // ID returns the ID value in the mutation. Note that the ID is only available // if it was provided to the builder or after it was returned from the database. func (m *PetMutation) ID() (id int, exists bool) { if m.id == nil { return } return *m.id, true } // IDs queries the database and returns the entity ids that match the mutation's predicate. // That means, if the mutation is applied within a transaction with an isolation level such // as sql.LevelSerializable, the returned ids match the ids of the rows that will be updated // or updated by the mutation. func (m *PetMutation) IDs(ctx context.Context) ([]int, error) { switch { case m.op.Is(OpUpdateOne | OpDeleteOne): id, exists := m.ID() if exists { return []int{id}, nil } fallthrough case m.op.Is(OpUpdate | OpDelete): return m.Client().Pet.Query().Where(m.predicates...).IDs(ctx) default: return nil, fmt.Errorf("IDs is not allowed on %s operations", m.op) } } // SetName sets the "name" field. func (m *PetMutation) SetName(s string) { m.name = &s } // Name returns the value of the "name" field in the mutation. func (m *PetMutation) Name() (r string, exists bool) { v := m.name if v == nil { return } return *v, true } // OldName returns the old "name" field's value of the Pet entity. // If the Pet object wasn't provided to the builder, the object is fetched from the database. // An error is returned if the mutation operation is not UpdateOne, or the database query fails. func (m *PetMutation) OldName(ctx context.Context) (v string, err error) { if !m.op.Is(OpUpdateOne) { return v, errors.New("OldName is only allowed on UpdateOne operations") } if m.id == nil || m.oldValue == nil { return v, errors.New("OldName requires an ID field in the mutation") } oldValue, err := m.oldValue(ctx) if err != nil { return v, fmt.Errorf("querying old value for OldName: %w", err) } return oldValue.Name, nil } // ResetName resets all changes to the "name" field. func (m *PetMutation) ResetName() { m.name = nil } // SetOwnerID sets the "owner_id" field. func (m *PetMutation) SetOwnerID(i int) { m.owner = &i } // OwnerID returns the value of the "owner_id" field in the mutation. func (m *PetMutation) OwnerID() (r int, exists bool) { v := m.owner if v == nil { return } return *v, true } // OldOwnerID returns the old "owner_id" field's value of the Pet entity. // If the Pet object wasn't provided to the builder, the object is fetched from the database. // An error is returned if the mutation operation is not UpdateOne, or the database query fails. func (m *PetMutation) OldOwnerID(ctx context.Context) (v int, err error) { if !m.op.Is(OpUpdateOne) { return v, errors.New("OldOwnerID is only allowed on UpdateOne operations") } if m.id == nil || m.oldValue == nil { return v, errors.New("OldOwnerID requires an ID field in the mutation") } oldValue, err := m.oldValue(ctx) if err != nil { return v, fmt.Errorf("querying old value for OldOwnerID: %w", err) } return oldValue.OwnerID, nil } // ClearOwnerID clears the value of the "owner_id" field. func (m *PetMutation) ClearOwnerID() { m.owner = nil m.clearedFields[pet.FieldOwnerID] = struct{}{} } // OwnerIDCleared returns if the "owner_id" field was cleared in this mutation. func (m *PetMutation) OwnerIDCleared() bool { _, ok := m.clearedFields[pet.FieldOwnerID] return ok } // ResetOwnerID resets all changes to the "owner_id" field. func (m *PetMutation) ResetOwnerID() { m.owner = nil delete(m.clearedFields, pet.FieldOwnerID) } // ClearOwner clears the "owner" edge to the User entity. func (m *PetMutation) ClearOwner() { m.clearedowner = true m.clearedFields[pet.FieldOwnerID] = struct{}{} } // OwnerCleared reports if the "owner" edge to the User entity was cleared. func (m *PetMutation) OwnerCleared() bool { return m.OwnerIDCleared() || m.clearedowner } // 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 *PetMutation) OwnerIDs() (ids []int) { if id := m.owner; id != nil { ids = append(ids, *id) } return } // ResetOwner resets all changes to the "owner" edge. func (m *PetMutation) ResetOwner() { m.owner = nil m.clearedowner = false } // Where appends a list predicates to the PetMutation builder. func (m *PetMutation) Where(ps ...predicate.Pet) { m.predicates = append(m.predicates, ps...) } // WhereP appends storage-level predicates to the PetMutation builder. Using this method, // users can use type-assertion to append predicates that do not depend on any generated package. func (m *PetMutation) WhereP(ps ...func(*sql.Selector)) { p := make([]predicate.Pet, len(ps)) for i := range ps { p[i] = ps[i] } m.Where(p...) } // Op returns the operation name. func (m *PetMutation) Op() Op { return m.op } // SetOp allows setting the mutation operation. func (m *PetMutation) SetOp(op Op) { m.op = op } // Type returns the node type of this mutation (Pet). func (m *PetMutation) 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 *PetMutation) Fields() []string { fields := make([]string, 0, 2) if m.name != nil { fields = append(fields, pet.FieldName) } if m.owner != nil { fields = append(fields, pet.FieldOwnerID) } 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 *PetMutation) Field(name string) (ent.Value, bool) { switch name { case pet.FieldName: return m.Name() case pet.FieldOwnerID: return m.OwnerID() } 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 *PetMutation) OldField(ctx context.Context, name string) (ent.Value, error) { switch name { case pet.FieldName: return m.OldName(ctx) case pet.FieldOwnerID: return m.OldOwnerID(ctx) } 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 *PetMutation) SetField(name string, value ent.Value) error { switch name { case pet.FieldName: v, ok := value.(string) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetName(v) return nil case pet.FieldOwnerID: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetOwnerID(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 *PetMutation) 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 *PetMutation) 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 *PetMutation) AddField(name string, value ent.Value) error { switch name { } return fmt.Errorf("unknown Pet numeric field %s", name) } // ClearedFields returns all nullable fields that were cleared during this // mutation. func (m *PetMutation) ClearedFields() []string { var fields []string if m.FieldCleared(pet.FieldOwnerID) { fields = append(fields, pet.FieldOwnerID) } return fields } // FieldCleared returns a boolean indicating if a field with the given name was // cleared in this mutation. func (m *PetMutation) 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 *PetMutation) ClearField(name string) error { switch name { case pet.FieldOwnerID: m.ClearOwnerID() 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 *PetMutation) ResetField(name string) error { switch name { case pet.FieldName: m.ResetName() return nil case pet.FieldOwnerID: m.ResetOwnerID() return nil } return fmt.Errorf("unknown Pet field %s", name) } // AddedEdges returns all edge names that were set/added in this mutation. func (m *PetMutation) AddedEdges() []string { edges := make([]string, 0, 1) if m.owner != nil { edges = append(edges, pet.EdgeOwner) } return edges } // AddedIDs returns all IDs (to other nodes) that were added for the given edge // name in this mutation. func (m *PetMutation) AddedIDs(name string) []ent.Value { switch name { case pet.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 *PetMutation) RemovedEdges() []string { edges := make([]string, 0, 1) return edges } // RemovedIDs returns all IDs (to other nodes) that were removed for the edge with // the given name in this mutation. func (m *PetMutation) RemovedIDs(name string) []ent.Value { return nil } // ClearedEdges returns all edge names that were cleared in this mutation. func (m *PetMutation) ClearedEdges() []string { edges := make([]string, 0, 1) if m.clearedowner { edges = append(edges, pet.EdgeOwner) } return edges } // EdgeCleared returns a boolean which indicates if the edge with the given name // was cleared in this mutation. func (m *PetMutation) EdgeCleared(name string) bool { switch name { case pet.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 *PetMutation) ClearEdge(name string) error { switch name { case pet.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 *PetMutation) ResetEdge(name string) error { switch name { case pet.EdgeOwner: m.ResetOwner() return nil } return fmt.Errorf("unknown Pet edge %s", name) } // UserMutation represents an operation that mutates the User nodes in the graph. type UserMutation struct { config op Op typ string id *int name *string clearedFields map[string]struct{} pets map[int]struct{} removedpets map[int]struct{} clearedpets bool groups map[int]struct{} removedgroups map[int]struct{} clearedgroups bool friends map[int]struct{} removedfriends map[int]struct{} clearedfriends bool followers map[int]struct{} removedfollowers map[int]struct{} clearedfollowers bool following map[int]struct{} removedfollowing map[int]struct{} clearedfollowing bool friendships map[int]struct{} removedfriendships map[int]struct{} clearedfriendships bool done bool oldValue func(context.Context) (*User, error) predicates []predicate.User } var _ ent.Mutation = (*UserMutation)(nil) // userOption allows management of the mutation configuration using functional options. type userOption func(*UserMutation) // newUserMutation creates new mutation for the User entity. func newUserMutation(c config, op Op, opts ...userOption) *UserMutation { m := &UserMutation{ config: c, op: op, typ: TypeUser, clearedFields: make(map[string]struct{}), } for _, opt := range opts { opt(m) } return m } // withUserID sets the ID field of the mutation. func withUserID(id int) userOption { return func(m *UserMutation) { var ( err error once sync.Once value *User ) m.oldValue = func(ctx context.Context) (*User, error) { once.Do(func() { if m.done { err = errors.New("querying old values post mutation is not allowed") } else { value, err = m.Client().User.Get(ctx, id) } }) return value, err } m.id = &id } } // withUser sets the old User of the mutation. func withUser(node *User) userOption { return func(m *UserMutation) { m.oldValue = func(context.Context) (*User, error) { return node, nil } m.id = &node.ID } } // Client returns a new `ent.Client` from the mutation. If the mutation was // executed in a transaction (ent.Tx), a transactional client is returned. func (m UserMutation) Client() *Client { client := &Client{config: m.config} client.init() return client } // Tx returns an `ent.Tx` for mutations that were executed in transactions; // it returns an error otherwise. func (m UserMutation) Tx() (*Tx, error) { if _, ok := m.driver.(*txDriver); !ok { return nil, errors.New("versioned: mutation is not running in a transaction") } tx := &Tx{config: m.config} tx.init() return tx, nil } // ID returns the ID value in the mutation. Note that the ID is only available // if it was provided to the builder or after it was returned from the database. func (m *UserMutation) ID() (id int, exists bool) { if m.id == nil { return } return *m.id, true } // IDs queries the database and returns the entity ids that match the mutation's predicate. // That means, if the mutation is applied within a transaction with an isolation level such // as sql.LevelSerializable, the returned ids match the ids of the rows that will be updated // or updated by the mutation. func (m *UserMutation) IDs(ctx context.Context) ([]int, error) { switch { case m.op.Is(OpUpdateOne | OpDeleteOne): id, exists := m.ID() if exists { return []int{id}, nil } fallthrough case m.op.Is(OpUpdate | OpDelete): return m.Client().User.Query().Where(m.predicates...).IDs(ctx) default: return nil, fmt.Errorf("IDs is not allowed on %s operations", m.op) } } // SetName sets the "name" field. func (m *UserMutation) SetName(s string) { m.name = &s } // Name returns the value of the "name" field in the mutation. func (m *UserMutation) Name() (r string, exists bool) { v := m.name if v == nil { return } return *v, true } // OldName returns the old "name" field's value of the User entity. // If the User object wasn't provided to the builder, the object is fetched from the database. // An error is returned if the mutation operation is not UpdateOne, or the database query fails. func (m *UserMutation) OldName(ctx context.Context) (v string, err error) { if !m.op.Is(OpUpdateOne) { return v, errors.New("OldName is only allowed on UpdateOne operations") } if m.id == nil || m.oldValue == nil { return v, errors.New("OldName requires an ID field in the mutation") } oldValue, err := m.oldValue(ctx) if err != nil { return v, fmt.Errorf("querying old value for OldName: %w", err) } return oldValue.Name, nil } // ResetName resets all changes to the "name" field. func (m *UserMutation) ResetName() { m.name = nil } // AddPetIDs adds the "pets" edge to the Pet entity by ids. func (m *UserMutation) 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 *UserMutation) ClearPets() { m.clearedpets = true } // PetsCleared reports if the "pets" edge to the Pet entity was cleared. func (m *UserMutation) PetsCleared() bool { return m.clearedpets } // RemovePetIDs removes the "pets" edge to the Pet entity by IDs. func (m *UserMutation) 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 *UserMutation) RemovedPetsIDs() (ids []int) { for id := range m.removedpets { ids = append(ids, id) } return } // PetsIDs returns the "pets" edge IDs in the mutation. func (m *UserMutation) PetsIDs() (ids []int) { for id := range m.pets { ids = append(ids, id) } return } // ResetPets resets all changes to the "pets" edge. func (m *UserMutation) ResetPets() { m.pets = nil m.clearedpets = false m.removedpets = nil } // AddGroupIDs adds the "groups" edge to the Group entity by ids. func (m *UserMutation) AddGroupIDs(ids ...int) { if m.groups == nil { m.groups = make(map[int]struct{}) } for i := range ids { m.groups[ids[i]] = struct{}{} } } // ClearGroups clears the "groups" edge to the Group entity. func (m *UserMutation) ClearGroups() { m.clearedgroups = true } // GroupsCleared reports if the "groups" edge to the Group entity was cleared. func (m *UserMutation) GroupsCleared() bool { return m.clearedgroups } // RemoveGroupIDs removes the "groups" edge to the Group entity by IDs. func (m *UserMutation) RemoveGroupIDs(ids ...int) { if m.removedgroups == nil { m.removedgroups = make(map[int]struct{}) } for i := range ids { delete(m.groups, ids[i]) m.removedgroups[ids[i]] = struct{}{} } } // RemovedGroups returns the removed IDs of the "groups" edge to the Group entity. func (m *UserMutation) RemovedGroupsIDs() (ids []int) { for id := range m.removedgroups { ids = append(ids, id) } return } // GroupsIDs returns the "groups" edge IDs in the mutation. func (m *UserMutation) GroupsIDs() (ids []int) { for id := range m.groups { ids = append(ids, id) } return } // ResetGroups resets all changes to the "groups" edge. func (m *UserMutation) ResetGroups() { m.groups = nil m.clearedgroups = false m.removedgroups = nil } // AddFriendIDs adds the "friends" edge to the User entity by ids. func (m *UserMutation) AddFriendIDs(ids ...int) { if m.friends == nil { m.friends = make(map[int]struct{}) } for i := range ids { m.friends[ids[i]] = struct{}{} } } // ClearFriends clears the "friends" edge to the User entity. func (m *UserMutation) ClearFriends() { m.clearedfriends = true } // FriendsCleared reports if the "friends" edge to the User entity was cleared. func (m *UserMutation) FriendsCleared() bool { return m.clearedfriends } // RemoveFriendIDs removes the "friends" edge to the User entity by IDs. func (m *UserMutation) RemoveFriendIDs(ids ...int) { if m.removedfriends == nil { m.removedfriends = make(map[int]struct{}) } for i := range ids { delete(m.friends, ids[i]) m.removedfriends[ids[i]] = struct{}{} } } // RemovedFriends returns the removed IDs of the "friends" edge to the User entity. func (m *UserMutation) RemovedFriendsIDs() (ids []int) { for id := range m.removedfriends { ids = append(ids, id) } return } // FriendsIDs returns the "friends" edge IDs in the mutation. func (m *UserMutation) FriendsIDs() (ids []int) { for id := range m.friends { ids = append(ids, id) } return } // ResetFriends resets all changes to the "friends" edge. func (m *UserMutation) ResetFriends() { m.friends = nil m.clearedfriends = false m.removedfriends = nil } // AddFollowerIDs adds the "followers" edge to the User entity by ids. func (m *UserMutation) AddFollowerIDs(ids ...int) { if m.followers == nil { m.followers = make(map[int]struct{}) } for i := range ids { m.followers[ids[i]] = struct{}{} } } // ClearFollowers clears the "followers" edge to the User entity. func (m *UserMutation) ClearFollowers() { m.clearedfollowers = true } // FollowersCleared reports if the "followers" edge to the User entity was cleared. func (m *UserMutation) FollowersCleared() bool { return m.clearedfollowers } // RemoveFollowerIDs removes the "followers" edge to the User entity by IDs. func (m *UserMutation) RemoveFollowerIDs(ids ...int) { if m.removedfollowers == nil { m.removedfollowers = make(map[int]struct{}) } for i := range ids { delete(m.followers, ids[i]) m.removedfollowers[ids[i]] = struct{}{} } } // RemovedFollowers returns the removed IDs of the "followers" edge to the User entity. func (m *UserMutation) RemovedFollowersIDs() (ids []int) { for id := range m.removedfollowers { ids = append(ids, id) } return } // FollowersIDs returns the "followers" edge IDs in the mutation. func (m *UserMutation) FollowersIDs() (ids []int) { for id := range m.followers { ids = append(ids, id) } return } // ResetFollowers resets all changes to the "followers" edge. func (m *UserMutation) ResetFollowers() { m.followers = nil m.clearedfollowers = false m.removedfollowers = nil } // AddFollowingIDs adds the "following" edge to the User entity by ids. func (m *UserMutation) AddFollowingIDs(ids ...int) { if m.following == nil { m.following = make(map[int]struct{}) } for i := range ids { m.following[ids[i]] = struct{}{} } } // ClearFollowing clears the "following" edge to the User entity. func (m *UserMutation) ClearFollowing() { m.clearedfollowing = true } // FollowingCleared reports if the "following" edge to the User entity was cleared. func (m *UserMutation) FollowingCleared() bool { return m.clearedfollowing } // RemoveFollowingIDs removes the "following" edge to the User entity by IDs. func (m *UserMutation) RemoveFollowingIDs(ids ...int) { if m.removedfollowing == nil { m.removedfollowing = make(map[int]struct{}) } for i := range ids { delete(m.following, ids[i]) m.removedfollowing[ids[i]] = struct{}{} } } // RemovedFollowing returns the removed IDs of the "following" edge to the User entity. func (m *UserMutation) RemovedFollowingIDs() (ids []int) { for id := range m.removedfollowing { ids = append(ids, id) } return } // FollowingIDs returns the "following" edge IDs in the mutation. func (m *UserMutation) FollowingIDs() (ids []int) { for id := range m.following { ids = append(ids, id) } return } // ResetFollowing resets all changes to the "following" edge. func (m *UserMutation) ResetFollowing() { m.following = nil m.clearedfollowing = false m.removedfollowing = nil } // AddFriendshipIDs adds the "friendships" edge to the Friendship entity by ids. func (m *UserMutation) AddFriendshipIDs(ids ...int) { if m.friendships == nil { m.friendships = make(map[int]struct{}) } for i := range ids { m.friendships[ids[i]] = struct{}{} } } // ClearFriendships clears the "friendships" edge to the Friendship entity. func (m *UserMutation) ClearFriendships() { m.clearedfriendships = true } // FriendshipsCleared reports if the "friendships" edge to the Friendship entity was cleared. func (m *UserMutation) FriendshipsCleared() bool { return m.clearedfriendships } // RemoveFriendshipIDs removes the "friendships" edge to the Friendship entity by IDs. func (m *UserMutation) RemoveFriendshipIDs(ids ...int) { if m.removedfriendships == nil { m.removedfriendships = make(map[int]struct{}) } for i := range ids { delete(m.friendships, ids[i]) m.removedfriendships[ids[i]] = struct{}{} } } // RemovedFriendships returns the removed IDs of the "friendships" edge to the Friendship entity. func (m *UserMutation) RemovedFriendshipsIDs() (ids []int) { for id := range m.removedfriendships { ids = append(ids, id) } return } // FriendshipsIDs returns the "friendships" edge IDs in the mutation. func (m *UserMutation) FriendshipsIDs() (ids []int) { for id := range m.friendships { ids = append(ids, id) } return } // ResetFriendships resets all changes to the "friendships" edge. func (m *UserMutation) ResetFriendships() { m.friendships = nil m.clearedfriendships = false m.removedfriendships = nil } // Where appends a list predicates to the UserMutation builder. func (m *UserMutation) Where(ps ...predicate.User) { m.predicates = append(m.predicates, ps...) } // WhereP appends storage-level predicates to the UserMutation builder. Using this method, // users can use type-assertion to append predicates that do not depend on any generated package. func (m *UserMutation) 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 *UserMutation) Op() Op { return m.op } // SetOp allows setting the mutation operation. func (m *UserMutation) SetOp(op Op) { m.op = op } // Type returns the node type of this mutation (User). func (m *UserMutation) 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 *UserMutation) Fields() []string { fields := make([]string, 0, 1) if m.name != nil { fields = append(fields, user.FieldName) } 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 *UserMutation) Field(name string) (ent.Value, bool) { switch name { case user.FieldName: return m.Name() } 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 *UserMutation) OldField(ctx context.Context, name string) (ent.Value, error) { switch name { case user.FieldName: return m.OldName(ctx) } 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 *UserMutation) SetField(name string, value ent.Value) error { switch name { case user.FieldName: v, ok := value.(string) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetName(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 *UserMutation) AddedFields() []string { return nil } // 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 *UserMutation) AddedField(name string) (ent.Value, bool) { 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 *UserMutation) 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 *UserMutation) ClearedFields() []string { return nil } // FieldCleared returns a boolean indicating if a field with the given name was // cleared in this mutation. func (m *UserMutation) 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 *UserMutation) ClearField(name string) error { 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 *UserMutation) ResetField(name string) error { switch name { case user.FieldName: m.ResetName() return nil } return fmt.Errorf("unknown User field %s", name) } // AddedEdges returns all edge names that were set/added in this mutation. func (m *UserMutation) AddedEdges() []string { edges := make([]string, 0, 6) if m.pets != nil { edges = append(edges, user.EdgePets) } if m.groups != nil { edges = append(edges, user.EdgeGroups) } if m.friends != nil { edges = append(edges, user.EdgeFriends) } if m.followers != nil { edges = append(edges, user.EdgeFollowers) } if m.following != nil { edges = append(edges, user.EdgeFollowing) } if m.friendships != nil { edges = append(edges, user.EdgeFriendships) } return edges } // AddedIDs returns all IDs (to other nodes) that were added for the given edge // name in this mutation. func (m *UserMutation) AddedIDs(name string) []ent.Value { switch name { case user.EdgePets: ids := make([]ent.Value, 0, len(m.pets)) for id := range m.pets { ids = append(ids, id) } return ids case user.EdgeGroups: ids := make([]ent.Value, 0, len(m.groups)) for id := range m.groups { ids = append(ids, id) } return ids case user.EdgeFriends: ids := make([]ent.Value, 0, len(m.friends)) for id := range m.friends { ids = append(ids, id) } return ids case user.EdgeFollowers: ids := make([]ent.Value, 0, len(m.followers)) for id := range m.followers { ids = append(ids, id) } return ids case user.EdgeFollowing: ids := make([]ent.Value, 0, len(m.following)) for id := range m.following { ids = append(ids, id) } return ids case user.EdgeFriendships: ids := make([]ent.Value, 0, len(m.friendships)) for id := range m.friendships { ids = append(ids, id) } return ids } return nil } // RemovedEdges returns all edge names that were removed in this mutation. func (m *UserMutation) RemovedEdges() []string { edges := make([]string, 0, 6) if m.removedpets != nil { edges = append(edges, user.EdgePets) } if m.removedgroups != nil { edges = append(edges, user.EdgeGroups) } if m.removedfriends != nil { edges = append(edges, user.EdgeFriends) } if m.removedfollowers != nil { edges = append(edges, user.EdgeFollowers) } if m.removedfollowing != nil { edges = append(edges, user.EdgeFollowing) } if m.removedfriendships != nil { edges = append(edges, user.EdgeFriendships) } return edges } // RemovedIDs returns all IDs (to other nodes) that were removed for the edge with // the given name in this mutation. func (m *UserMutation) RemovedIDs(name string) []ent.Value { switch name { case user.EdgePets: ids := make([]ent.Value, 0, len(m.removedpets)) for id := range m.removedpets { ids = append(ids, id) } return ids case user.EdgeGroups: ids := make([]ent.Value, 0, len(m.removedgroups)) for id := range m.removedgroups { ids = append(ids, id) } return ids case user.EdgeFriends: ids := make([]ent.Value, 0, len(m.removedfriends)) for id := range m.removedfriends { ids = append(ids, id) } return ids case user.EdgeFollowers: ids := make([]ent.Value, 0, len(m.removedfollowers)) for id := range m.removedfollowers { ids = append(ids, id) } return ids case user.EdgeFollowing: ids := make([]ent.Value, 0, len(m.removedfollowing)) for id := range m.removedfollowing { ids = append(ids, id) } return ids case user.EdgeFriendships: ids := make([]ent.Value, 0, len(m.removedfriendships)) for id := range m.removedfriendships { ids = append(ids, id) } return ids } return nil } // ClearedEdges returns all edge names that were cleared in this mutation. func (m *UserMutation) ClearedEdges() []string { edges := make([]string, 0, 6) if m.clearedpets { edges = append(edges, user.EdgePets) } if m.clearedgroups { edges = append(edges, user.EdgeGroups) } if m.clearedfriends { edges = append(edges, user.EdgeFriends) } if m.clearedfollowers { edges = append(edges, user.EdgeFollowers) } if m.clearedfollowing { edges = append(edges, user.EdgeFollowing) } if m.clearedfriendships { edges = append(edges, user.EdgeFriendships) } return edges } // EdgeCleared returns a boolean which indicates if the edge with the given name // was cleared in this mutation. func (m *UserMutation) EdgeCleared(name string) bool { switch name { case user.EdgePets: return m.clearedpets case user.EdgeGroups: return m.clearedgroups case user.EdgeFriends: return m.clearedfriends case user.EdgeFollowers: return m.clearedfollowers case user.EdgeFollowing: return m.clearedfollowing case user.EdgeFriendships: return m.clearedfriendships } 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 *UserMutation) 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 *UserMutation) ResetEdge(name string) error { switch name { case user.EdgePets: m.ResetPets() return nil case user.EdgeGroups: m.ResetGroups() return nil case user.EdgeFriends: m.ResetFriends() return nil case user.EdgeFollowers: m.ResetFollowers() return nil case user.EdgeFollowing: m.ResetFollowing() return nil case user.EdgeFriendships: m.ResetFriendships() return nil } return fmt.Errorf("unknown User edge %s", name) }