// Code generated by ent, DO NOT EDIT. package db import ( "context" "errors" "fmt" "sync" "time" "entgo.io/ent" "entgo.io/ent/dialect/sql" "github.com/openmeterio/openmeter/pkg/framework/entutils/testutils/ent1/db/example1" "github.com/openmeterio/openmeter/pkg/framework/entutils/testutils/ent1/db/predicate" ) const ( // Operation types. OpCreate = ent.OpCreate OpDelete = ent.OpDelete OpDeleteOne = ent.OpDeleteOne OpUpdate = ent.OpUpdate OpUpdateOne = ent.OpUpdateOne // Node types. TypeExample1 = "Example1" ) // Example1Mutation represents an operation that mutates the Example1 nodes in the graph. type Example1Mutation struct { config op Op typ string id *string created_at *time.Time updated_at *time.Time deleted_at *time.Time example_value_1 *string clearedFields map[string]struct{} done bool oldValue func(context.Context) (*Example1, error) predicates []predicate.Example1 } var _ ent.Mutation = (*Example1Mutation)(nil) // example1Option allows management of the mutation configuration using functional options. type example1Option func(*Example1Mutation) // newExample1Mutation creates new mutation for the Example1 entity. func newExample1Mutation(c config, op Op, opts ...example1Option) *Example1Mutation { m := &Example1Mutation{ config: c, op: op, typ: TypeExample1, clearedFields: make(map[string]struct{}), } for _, opt := range opts { opt(m) } return m } // withExample1ID sets the ID field of the mutation. func withExample1ID(id string) example1Option { return func(m *Example1Mutation) { var ( err error once sync.Once value *Example1 ) m.oldValue = func(ctx context.Context) (*Example1, error) { once.Do(func() { if m.done { err = errors.New("querying old values post mutation is not allowed") } else { value, err = m.Client().Example1.Get(ctx, id) } }) return value, err } m.id = &id } } // withExample1 sets the old Example1 of the mutation. func withExample1(node *Example1) example1Option { return func(m *Example1Mutation) { m.oldValue = func(context.Context) (*Example1, 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 Example1Mutation) 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 Example1Mutation) Tx() (*Tx, error) { if _, ok := m.driver.(*txDriver); !ok { return nil, errors.New("db: mutation is not running in a transaction") } tx := &Tx{config: m.config} tx.init() return tx, nil } // SetID sets the value of the id field. Note that this // operation is only accepted on creation of Example1 entities. func (m *Example1Mutation) SetID(id string) { m.id = &id } // 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 *Example1Mutation) ID() (id string, 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 *Example1Mutation) IDs(ctx context.Context) ([]string, error) { switch { case m.op.Is(OpUpdateOne | OpDeleteOne): id, exists := m.ID() if exists { return []string{id}, nil } fallthrough case m.op.Is(OpUpdate | OpDelete): return m.Client().Example1.Query().Where(m.predicates...).IDs(ctx) default: return nil, fmt.Errorf("IDs is not allowed on %s operations", m.op) } } // SetCreatedAt sets the "created_at" field. func (m *Example1Mutation) SetCreatedAt(t time.Time) { m.created_at = &t } // CreatedAt returns the value of the "created_at" field in the mutation. func (m *Example1Mutation) 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 Example1 entity. // If the Example1 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 *Example1Mutation) 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 *Example1Mutation) ResetCreatedAt() { m.created_at = nil } // SetUpdatedAt sets the "updated_at" field. func (m *Example1Mutation) SetUpdatedAt(t time.Time) { m.updated_at = &t } // UpdatedAt returns the value of the "updated_at" field in the mutation. func (m *Example1Mutation) UpdatedAt() (r time.Time, exists bool) { v := m.updated_at if v == nil { return } return *v, true } // OldUpdatedAt returns the old "updated_at" field's value of the Example1 entity. // If the Example1 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 *Example1Mutation) OldUpdatedAt(ctx context.Context) (v time.Time, err error) { if !m.op.Is(OpUpdateOne) { return v, errors.New("OldUpdatedAt is only allowed on UpdateOne operations") } if m.id == nil || m.oldValue == nil { return v, errors.New("OldUpdatedAt requires an ID field in the mutation") } oldValue, err := m.oldValue(ctx) if err != nil { return v, fmt.Errorf("querying old value for OldUpdatedAt: %w", err) } return oldValue.UpdatedAt, nil } // ResetUpdatedAt resets all changes to the "updated_at" field. func (m *Example1Mutation) ResetUpdatedAt() { m.updated_at = nil } // SetDeletedAt sets the "deleted_at" field. func (m *Example1Mutation) SetDeletedAt(t time.Time) { m.deleted_at = &t } // DeletedAt returns the value of the "deleted_at" field in the mutation. func (m *Example1Mutation) DeletedAt() (r time.Time, exists bool) { v := m.deleted_at if v == nil { return } return *v, true } // OldDeletedAt returns the old "deleted_at" field's value of the Example1 entity. // If the Example1 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 *Example1Mutation) OldDeletedAt(ctx context.Context) (v *time.Time, err error) { if !m.op.Is(OpUpdateOne) { return v, errors.New("OldDeletedAt is only allowed on UpdateOne operations") } if m.id == nil || m.oldValue == nil { return v, errors.New("OldDeletedAt requires an ID field in the mutation") } oldValue, err := m.oldValue(ctx) if err != nil { return v, fmt.Errorf("querying old value for OldDeletedAt: %w", err) } return oldValue.DeletedAt, nil } // ClearDeletedAt clears the value of the "deleted_at" field. func (m *Example1Mutation) ClearDeletedAt() { m.deleted_at = nil m.clearedFields[example1.FieldDeletedAt] = struct{}{} } // DeletedAtCleared returns if the "deleted_at" field was cleared in this mutation. func (m *Example1Mutation) DeletedAtCleared() bool { _, ok := m.clearedFields[example1.FieldDeletedAt] return ok } // ResetDeletedAt resets all changes to the "deleted_at" field. func (m *Example1Mutation) ResetDeletedAt() { m.deleted_at = nil delete(m.clearedFields, example1.FieldDeletedAt) } // SetExampleValue1 sets the "example_value_1" field. func (m *Example1Mutation) SetExampleValue1(s string) { m.example_value_1 = &s } // ExampleValue1 returns the value of the "example_value_1" field in the mutation. func (m *Example1Mutation) ExampleValue1() (r string, exists bool) { v := m.example_value_1 if v == nil { return } return *v, true } // OldExampleValue1 returns the old "example_value_1" field's value of the Example1 entity. // If the Example1 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 *Example1Mutation) OldExampleValue1(ctx context.Context) (v string, err error) { if !m.op.Is(OpUpdateOne) { return v, errors.New("OldExampleValue1 is only allowed on UpdateOne operations") } if m.id == nil || m.oldValue == nil { return v, errors.New("OldExampleValue1 requires an ID field in the mutation") } oldValue, err := m.oldValue(ctx) if err != nil { return v, fmt.Errorf("querying old value for OldExampleValue1: %w", err) } return oldValue.ExampleValue1, nil } // ResetExampleValue1 resets all changes to the "example_value_1" field. func (m *Example1Mutation) ResetExampleValue1() { m.example_value_1 = nil } // Where appends a list predicates to the Example1Mutation builder. func (m *Example1Mutation) Where(ps ...predicate.Example1) { m.predicates = append(m.predicates, ps...) } // WhereP appends storage-level predicates to the Example1Mutation builder. Using this method, // users can use type-assertion to append predicates that do not depend on any generated package. func (m *Example1Mutation) WhereP(ps ...func(*sql.Selector)) { p := make([]predicate.Example1, len(ps)) for i := range ps { p[i] = ps[i] } m.Where(p...) } // Op returns the operation name. func (m *Example1Mutation) Op() Op { return m.op } // SetOp allows setting the mutation operation. func (m *Example1Mutation) SetOp(op Op) { m.op = op } // Type returns the node type of this mutation (Example1). func (m *Example1Mutation) 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 *Example1Mutation) Fields() []string { fields := make([]string, 0, 4) if m.created_at != nil { fields = append(fields, example1.FieldCreatedAt) } if m.updated_at != nil { fields = append(fields, example1.FieldUpdatedAt) } if m.deleted_at != nil { fields = append(fields, example1.FieldDeletedAt) } if m.example_value_1 != nil { fields = append(fields, example1.FieldExampleValue1) } 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 *Example1Mutation) Field(name string) (ent.Value, bool) { switch name { case example1.FieldCreatedAt: return m.CreatedAt() case example1.FieldUpdatedAt: return m.UpdatedAt() case example1.FieldDeletedAt: return m.DeletedAt() case example1.FieldExampleValue1: return m.ExampleValue1() } 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 *Example1Mutation) OldField(ctx context.Context, name string) (ent.Value, error) { switch name { case example1.FieldCreatedAt: return m.OldCreatedAt(ctx) case example1.FieldUpdatedAt: return m.OldUpdatedAt(ctx) case example1.FieldDeletedAt: return m.OldDeletedAt(ctx) case example1.FieldExampleValue1: return m.OldExampleValue1(ctx) } return nil, fmt.Errorf("unknown Example1 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 *Example1Mutation) SetField(name string, value ent.Value) error { switch name { case example1.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 example1.FieldUpdatedAt: v, ok := value.(time.Time) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetUpdatedAt(v) return nil case example1.FieldDeletedAt: v, ok := value.(time.Time) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetDeletedAt(v) return nil case example1.FieldExampleValue1: v, ok := value.(string) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetExampleValue1(v) return nil } return fmt.Errorf("unknown Example1 field %s", name) } // AddedFields returns all numeric fields that were incremented/decremented during // this mutation. func (m *Example1Mutation) 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 *Example1Mutation) 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 *Example1Mutation) AddField(name string, value ent.Value) error { switch name { } return fmt.Errorf("unknown Example1 numeric field %s", name) } // ClearedFields returns all nullable fields that were cleared during this // mutation. func (m *Example1Mutation) ClearedFields() []string { var fields []string if m.FieldCleared(example1.FieldDeletedAt) { fields = append(fields, example1.FieldDeletedAt) } return fields } // FieldCleared returns a boolean indicating if a field with the given name was // cleared in this mutation. func (m *Example1Mutation) 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 *Example1Mutation) ClearField(name string) error { switch name { case example1.FieldDeletedAt: m.ClearDeletedAt() return nil } return fmt.Errorf("unknown Example1 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 *Example1Mutation) ResetField(name string) error { switch name { case example1.FieldCreatedAt: m.ResetCreatedAt() return nil case example1.FieldUpdatedAt: m.ResetUpdatedAt() return nil case example1.FieldDeletedAt: m.ResetDeletedAt() return nil case example1.FieldExampleValue1: m.ResetExampleValue1() return nil } return fmt.Errorf("unknown Example1 field %s", name) } // AddedEdges returns all edge names that were set/added in this mutation. func (m *Example1Mutation) AddedEdges() []string { edges := make([]string, 0, 0) return edges } // AddedIDs returns all IDs (to other nodes) that were added for the given edge // name in this mutation. func (m *Example1Mutation) AddedIDs(name string) []ent.Value { return nil } // RemovedEdges returns all edge names that were removed in this mutation. func (m *Example1Mutation) RemovedEdges() []string { edges := make([]string, 0, 0) return edges } // RemovedIDs returns all IDs (to other nodes) that were removed for the edge with // the given name in this mutation. func (m *Example1Mutation) RemovedIDs(name string) []ent.Value { return nil } // ClearedEdges returns all edge names that were cleared in this mutation. func (m *Example1Mutation) ClearedEdges() []string { edges := make([]string, 0, 0) return edges } // EdgeCleared returns a boolean which indicates if the edge with the given name // was cleared in this mutation. func (m *Example1Mutation) EdgeCleared(name string) bool { 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 *Example1Mutation) ClearEdge(name string) error { return fmt.Errorf("unknown Example1 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 *Example1Mutation) ResetEdge(name string) error { return fmt.Errorf("unknown Example1 edge %s", name) }