298 lines
8.5 KiB
Go
298 lines
8.5 KiB
Go
// Copyright 2019-present Facebook Inc. All rights reserved.
|
|
// This source code is licensed under the Apache 2.0 license found
|
|
// in the LICENSE file in the root directory of this source tree.
|
|
|
|
// Code generated by ent, DO NOT EDIT.
|
|
|
|
package ent
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"sync"
|
|
|
|
"entgo.io/ent"
|
|
"entgo.io/ent/examples/edgeindex/ent/city"
|
|
"entgo.io/ent/examples/edgeindex/ent/street"
|
|
)
|
|
|
|
const (
|
|
// Operation types.
|
|
OpCreate = ent.OpCreate
|
|
OpDelete = ent.OpDelete
|
|
OpDeleteOne = ent.OpDeleteOne
|
|
OpUpdate = ent.OpUpdate
|
|
OpUpdateOne = ent.OpUpdateOne
|
|
|
|
// Node types.
|
|
TypeCity = "City"
|
|
TypeStreet = "Street"
|
|
)
|
|
|
|
// CityMutation represents an operation that mutates the City nodes in the graph.
|
|
type CityMutation struct {
|
|
city.Mutation
|
|
config
|
|
id *int
|
|
done bool
|
|
oldValue func(context.Context) (*City, error)
|
|
}
|
|
|
|
var _ ent.Mutation = (*CityMutation)(nil)
|
|
|
|
// cityOption allows management of the mutation configuration using functional options.
|
|
type cityOption func(*CityMutation)
|
|
|
|
// newCityMutation creates new mutation for the City entity.
|
|
func newCityMutation(c config, op Op, opts ...cityOption) *CityMutation {
|
|
m := &CityMutation{
|
|
Mutation: *city.NewMutation(op),
|
|
config: c,
|
|
}
|
|
for _, opt := range opts {
|
|
opt(m)
|
|
}
|
|
return m
|
|
}
|
|
|
|
// 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 *CityMutation) ID() (id int, exists bool) {
|
|
if m.id == nil {
|
|
return
|
|
}
|
|
return *m.id, true
|
|
}
|
|
|
|
// withCityID sets the ID field of the mutation.
|
|
func withCityID(id int) cityOption {
|
|
return func(m *CityMutation) {
|
|
var (
|
|
err error
|
|
once sync.Once
|
|
value *City
|
|
)
|
|
m.oldValue = func(ctx context.Context) (*City, error) {
|
|
once.Do(func() {
|
|
if m.done {
|
|
err = errors.New("querying old values post mutation is not allowed")
|
|
} else {
|
|
value, err = m.Client().City.Get(ctx, id)
|
|
}
|
|
})
|
|
return value, err
|
|
}
|
|
m.id = &id
|
|
}
|
|
}
|
|
|
|
// withCity sets the old City of the mutation.
|
|
func withCity(node *City) cityOption {
|
|
return func(m *CityMutation) {
|
|
m.oldValue = func(context.Context) (*City, 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 CityMutation) 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 CityMutation) Tx() (*Tx, error) {
|
|
if _, ok := m.driver.(*txDriver); !ok {
|
|
return nil, errors.New("ent: mutation is not running in a transaction")
|
|
}
|
|
tx := &Tx{config: m.config}
|
|
tx.init()
|
|
return tx, nil
|
|
}
|
|
|
|
// 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 *CityMutation) 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().City.Query().Where(m.Predicates()...).IDs(ctx)
|
|
default:
|
|
return nil, fmt.Errorf("IDs is not allowed on %s operations", m.Op())
|
|
}
|
|
}
|
|
|
|
// OldName returns the old "name" field's value of the City entity.
|
|
// If the City 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 *CityMutation) 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 _, exists := m.ID(); !exists || 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
|
|
}
|
|
|
|
// 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 *CityMutation) OldField(ctx context.Context, name string) (ent.Value, error) {
|
|
switch name {
|
|
case city.FieldName:
|
|
return m.OldName(ctx)
|
|
}
|
|
return nil, fmt.Errorf("unknown City field %s", name)
|
|
}
|
|
|
|
// StreetMutation represents an operation that mutates the Street nodes in the graph.
|
|
type StreetMutation struct {
|
|
street.Mutation
|
|
config
|
|
id *int
|
|
done bool
|
|
oldValue func(context.Context) (*Street, error)
|
|
}
|
|
|
|
var _ ent.Mutation = (*StreetMutation)(nil)
|
|
|
|
// streetOption allows management of the mutation configuration using functional options.
|
|
type streetOption func(*StreetMutation)
|
|
|
|
// newStreetMutation creates new mutation for the Street entity.
|
|
func newStreetMutation(c config, op Op, opts ...streetOption) *StreetMutation {
|
|
m := &StreetMutation{
|
|
Mutation: *street.NewMutation(op),
|
|
config: c,
|
|
}
|
|
for _, opt := range opts {
|
|
opt(m)
|
|
}
|
|
return m
|
|
}
|
|
|
|
// 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 *StreetMutation) ID() (id int, exists bool) {
|
|
if m.id == nil {
|
|
return
|
|
}
|
|
return *m.id, true
|
|
}
|
|
|
|
// withStreetID sets the ID field of the mutation.
|
|
func withStreetID(id int) streetOption {
|
|
return func(m *StreetMutation) {
|
|
var (
|
|
err error
|
|
once sync.Once
|
|
value *Street
|
|
)
|
|
m.oldValue = func(ctx context.Context) (*Street, error) {
|
|
once.Do(func() {
|
|
if m.done {
|
|
err = errors.New("querying old values post mutation is not allowed")
|
|
} else {
|
|
value, err = m.Client().Street.Get(ctx, id)
|
|
}
|
|
})
|
|
return value, err
|
|
}
|
|
m.id = &id
|
|
}
|
|
}
|
|
|
|
// withStreet sets the old Street of the mutation.
|
|
func withStreet(node *Street) streetOption {
|
|
return func(m *StreetMutation) {
|
|
m.oldValue = func(context.Context) (*Street, 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 StreetMutation) 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 StreetMutation) Tx() (*Tx, error) {
|
|
if _, ok := m.driver.(*txDriver); !ok {
|
|
return nil, errors.New("ent: mutation is not running in a transaction")
|
|
}
|
|
tx := &Tx{config: m.config}
|
|
tx.init()
|
|
return tx, nil
|
|
}
|
|
|
|
// 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 *StreetMutation) 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().Street.Query().Where(m.Predicates()...).IDs(ctx)
|
|
default:
|
|
return nil, fmt.Errorf("IDs is not allowed on %s operations", m.Op())
|
|
}
|
|
}
|
|
|
|
// OldName returns the old "name" field's value of the Street entity.
|
|
// If the Street 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 *StreetMutation) 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 _, exists := m.ID(); !exists || 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
|
|
}
|
|
|
|
// 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 *StreetMutation) OldField(ctx context.Context, name string) (ent.Value, error) {
|
|
switch name {
|
|
case street.FieldName:
|
|
return m.OldName(ctx)
|
|
}
|
|
return nil, fmt.Errorf("unknown Street field %s", name)
|
|
}
|