Files
ent/entc/integration/migrate/entv2/conversion_create.go
2021-07-23 15:00:47 +03:00

389 lines
10 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 entc, DO NOT EDIT.
package entv2
import (
"context"
"fmt"
"entgo.io/ent/dialect/sql/sqlgraph"
"entgo.io/ent/entc/integration/migrate/entv2/conversion"
"entgo.io/ent/schema/field"
)
// ConversionCreate is the builder for creating a Conversion entity.
type ConversionCreate struct {
config
mutation *ConversionMutation
hooks []Hook
}
// SetName sets the "name" field.
func (cc *ConversionCreate) SetName(s string) *ConversionCreate {
cc.mutation.SetName(s)
return cc
}
// SetNillableName sets the "name" field if the given value is not nil.
func (cc *ConversionCreate) SetNillableName(s *string) *ConversionCreate {
if s != nil {
cc.SetName(*s)
}
return cc
}
// SetInt8ToString sets the "int8_to_string" field.
func (cc *ConversionCreate) SetInt8ToString(s string) *ConversionCreate {
cc.mutation.SetInt8ToString(s)
return cc
}
// SetNillableInt8ToString sets the "int8_to_string" field if the given value is not nil.
func (cc *ConversionCreate) SetNillableInt8ToString(s *string) *ConversionCreate {
if s != nil {
cc.SetInt8ToString(*s)
}
return cc
}
// SetUint8ToString sets the "uint8_to_string" field.
func (cc *ConversionCreate) SetUint8ToString(s string) *ConversionCreate {
cc.mutation.SetUint8ToString(s)
return cc
}
// SetNillableUint8ToString sets the "uint8_to_string" field if the given value is not nil.
func (cc *ConversionCreate) SetNillableUint8ToString(s *string) *ConversionCreate {
if s != nil {
cc.SetUint8ToString(*s)
}
return cc
}
// SetInt16ToString sets the "int16_to_string" field.
func (cc *ConversionCreate) SetInt16ToString(s string) *ConversionCreate {
cc.mutation.SetInt16ToString(s)
return cc
}
// SetNillableInt16ToString sets the "int16_to_string" field if the given value is not nil.
func (cc *ConversionCreate) SetNillableInt16ToString(s *string) *ConversionCreate {
if s != nil {
cc.SetInt16ToString(*s)
}
return cc
}
// SetUint16ToString sets the "uint16_to_string" field.
func (cc *ConversionCreate) SetUint16ToString(s string) *ConversionCreate {
cc.mutation.SetUint16ToString(s)
return cc
}
// SetNillableUint16ToString sets the "uint16_to_string" field if the given value is not nil.
func (cc *ConversionCreate) SetNillableUint16ToString(s *string) *ConversionCreate {
if s != nil {
cc.SetUint16ToString(*s)
}
return cc
}
// SetInt32ToString sets the "int32_to_string" field.
func (cc *ConversionCreate) SetInt32ToString(s string) *ConversionCreate {
cc.mutation.SetInt32ToString(s)
return cc
}
// SetNillableInt32ToString sets the "int32_to_string" field if the given value is not nil.
func (cc *ConversionCreate) SetNillableInt32ToString(s *string) *ConversionCreate {
if s != nil {
cc.SetInt32ToString(*s)
}
return cc
}
// SetUint32ToString sets the "uint32_to_string" field.
func (cc *ConversionCreate) SetUint32ToString(s string) *ConversionCreate {
cc.mutation.SetUint32ToString(s)
return cc
}
// SetNillableUint32ToString sets the "uint32_to_string" field if the given value is not nil.
func (cc *ConversionCreate) SetNillableUint32ToString(s *string) *ConversionCreate {
if s != nil {
cc.SetUint32ToString(*s)
}
return cc
}
// SetInt64ToString sets the "int64_to_string" field.
func (cc *ConversionCreate) SetInt64ToString(s string) *ConversionCreate {
cc.mutation.SetInt64ToString(s)
return cc
}
// SetNillableInt64ToString sets the "int64_to_string" field if the given value is not nil.
func (cc *ConversionCreate) SetNillableInt64ToString(s *string) *ConversionCreate {
if s != nil {
cc.SetInt64ToString(*s)
}
return cc
}
// SetUint64ToString sets the "uint64_to_string" field.
func (cc *ConversionCreate) SetUint64ToString(s string) *ConversionCreate {
cc.mutation.SetUint64ToString(s)
return cc
}
// SetNillableUint64ToString sets the "uint64_to_string" field if the given value is not nil.
func (cc *ConversionCreate) SetNillableUint64ToString(s *string) *ConversionCreate {
if s != nil {
cc.SetUint64ToString(*s)
}
return cc
}
// Mutation returns the ConversionMutation object of the builder.
func (cc *ConversionCreate) Mutation() *ConversionMutation {
return cc.mutation
}
// Save creates the Conversion in the database.
func (cc *ConversionCreate) Save(ctx context.Context) (*Conversion, error) {
var (
err error
node *Conversion
)
if len(cc.hooks) == 0 {
if err = cc.check(); err != nil {
return nil, err
}
node, err = cc.sqlSave(ctx)
} else {
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) {
mutation, ok := m.(*ConversionMutation)
if !ok {
return nil, fmt.Errorf("unexpected mutation type %T", m)
}
if err = cc.check(); err != nil {
return nil, err
}
cc.mutation = mutation
if node, err = cc.sqlSave(ctx); err != nil {
return nil, err
}
mutation.id = &node.ID
mutation.done = true
return node, err
})
for i := len(cc.hooks) - 1; i >= 0; i-- {
if cc.hooks[i] == nil {
return nil, fmt.Errorf("entv2: uninitialized hook (forgotten import entv2/runtime?)")
}
mut = cc.hooks[i](mut)
}
if _, err := mut.Mutate(ctx, cc.mutation); err != nil {
return nil, err
}
}
return node, err
}
// SaveX calls Save and panics if Save returns an error.
func (cc *ConversionCreate) SaveX(ctx context.Context) *Conversion {
v, err := cc.Save(ctx)
if err != nil {
panic(err)
}
return v
}
// Exec executes the query.
func (cc *ConversionCreate) Exec(ctx context.Context) error {
_, err := cc.Save(ctx)
return err
}
// ExecX is like Exec, but panics if an error occurs.
func (cc *ConversionCreate) ExecX(ctx context.Context) {
if err := cc.Exec(ctx); err != nil {
panic(err)
}
}
// check runs all checks and user-defined validators on the builder.
func (cc *ConversionCreate) check() error {
return nil
}
func (cc *ConversionCreate) sqlSave(ctx context.Context) (*Conversion, error) {
_node, _spec := cc.createSpec()
if err := sqlgraph.CreateNode(ctx, cc.driver, _spec); err != nil {
if sqlgraph.IsConstraintError(err) {
err = &ConstraintError{err.Error(), err}
}
return nil, err
}
id := _spec.ID.Value.(int64)
_node.ID = int(id)
return _node, nil
}
func (cc *ConversionCreate) createSpec() (*Conversion, *sqlgraph.CreateSpec) {
var (
_node = &Conversion{config: cc.config}
_spec = &sqlgraph.CreateSpec{
Table: conversion.Table,
ID: &sqlgraph.FieldSpec{
Type: field.TypeInt,
Column: conversion.FieldID,
},
}
)
if value, ok := cc.mutation.Name(); ok {
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeString,
Value: value,
Column: conversion.FieldName,
})
_node.Name = value
}
if value, ok := cc.mutation.Int8ToString(); ok {
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeString,
Value: value,
Column: conversion.FieldInt8ToString,
})
_node.Int8ToString = value
}
if value, ok := cc.mutation.Uint8ToString(); ok {
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeString,
Value: value,
Column: conversion.FieldUint8ToString,
})
_node.Uint8ToString = value
}
if value, ok := cc.mutation.Int16ToString(); ok {
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeString,
Value: value,
Column: conversion.FieldInt16ToString,
})
_node.Int16ToString = value
}
if value, ok := cc.mutation.Uint16ToString(); ok {
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeString,
Value: value,
Column: conversion.FieldUint16ToString,
})
_node.Uint16ToString = value
}
if value, ok := cc.mutation.Int32ToString(); ok {
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeString,
Value: value,
Column: conversion.FieldInt32ToString,
})
_node.Int32ToString = value
}
if value, ok := cc.mutation.Uint32ToString(); ok {
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeString,
Value: value,
Column: conversion.FieldUint32ToString,
})
_node.Uint32ToString = value
}
if value, ok := cc.mutation.Int64ToString(); ok {
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeString,
Value: value,
Column: conversion.FieldInt64ToString,
})
_node.Int64ToString = value
}
if value, ok := cc.mutation.Uint64ToString(); ok {
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{
Type: field.TypeString,
Value: value,
Column: conversion.FieldUint64ToString,
})
_node.Uint64ToString = value
}
return _node, _spec
}
// ConversionCreateBulk is the builder for creating many Conversion entities in bulk.
type ConversionCreateBulk struct {
config
builders []*ConversionCreate
}
// Save creates the Conversion entities in the database.
func (ccb *ConversionCreateBulk) Save(ctx context.Context) ([]*Conversion, error) {
specs := make([]*sqlgraph.CreateSpec, len(ccb.builders))
nodes := make([]*Conversion, len(ccb.builders))
mutators := make([]Mutator, len(ccb.builders))
for i := range ccb.builders {
func(i int, root context.Context) {
builder := ccb.builders[i]
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) {
mutation, ok := m.(*ConversionMutation)
if !ok {
return nil, fmt.Errorf("unexpected mutation type %T", m)
}
if err := builder.check(); err != nil {
return nil, err
}
builder.mutation = mutation
nodes[i], specs[i] = builder.createSpec()
var err error
if i < len(mutators)-1 {
_, err = mutators[i+1].Mutate(root, ccb.builders[i+1].mutation)
} else {
// Invoke the actual operation on the latest mutation in the chain.
if err = sqlgraph.BatchCreate(ctx, ccb.driver, &sqlgraph.BatchCreateSpec{Nodes: specs}); err != nil {
if sqlgraph.IsConstraintError(err) {
err = &ConstraintError{err.Error(), err}
}
}
}
if err != nil {
return nil, err
}
mutation.id = &nodes[i].ID
mutation.done = true
id := specs[i].ID.Value.(int64)
nodes[i].ID = int(id)
return nodes[i], nil
})
for i := len(builder.hooks) - 1; i >= 0; i-- {
mut = builder.hooks[i](mut)
}
mutators[i] = mut
}(i, ctx)
}
if len(mutators) > 0 {
if _, err := mutators[0].Mutate(ctx, ccb.builders[0].mutation); err != nil {
return nil, err
}
}
return nodes, nil
}
// SaveX is like Save, but panics if an error occurs.
func (ccb *ConversionCreateBulk) SaveX(ctx context.Context) []*Conversion {
v, err := ccb.Save(ctx)
if err != nil {
panic(err)
}
return v
}