go__ent_demo/todo/ent/todo_delete.go

111 lines
2.4 KiB
Go

// Code generated by entc, DO NOT EDIT.
package ent
import (
"context"
"fmt"
"todo/ent/predicate"
"todo/ent/todo"
"entgo.io/ent/dialect/sql"
"entgo.io/ent/dialect/sql/sqlgraph"
"entgo.io/ent/schema/field"
)
// TodoDelete is the builder for deleting a Todo entity.
type TodoDelete struct {
config
hooks []Hook
mutation *TodoMutation
}
// Where appends a list predicates to the TodoDelete builder.
func (td *TodoDelete) Where(ps ...predicate.Todo) *TodoDelete {
td.mutation.Where(ps...)
return td
}
// Exec executes the deletion query and returns how many vertices were deleted.
func (td *TodoDelete) Exec(ctx context.Context) (int, error) {
var (
err error
affected int
)
if len(td.hooks) == 0 {
affected, err = td.sqlExec(ctx)
} else {
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) {
mutation, ok := m.(*TodoMutation)
if !ok {
return nil, fmt.Errorf("unexpected mutation type %T", m)
}
td.mutation = mutation
affected, err = td.sqlExec(ctx)
mutation.done = true
return affected, err
})
for i := len(td.hooks) - 1; i >= 0; i-- {
if td.hooks[i] == nil {
return 0, fmt.Errorf("ent: uninitialized hook (forgotten import ent/runtime?)")
}
mut = td.hooks[i](mut)
}
if _, err := mut.Mutate(ctx, td.mutation); err != nil {
return 0, err
}
}
return affected, err
}
// ExecX is like Exec, but panics if an error occurs.
func (td *TodoDelete) ExecX(ctx context.Context) int {
n, err := td.Exec(ctx)
if err != nil {
panic(err)
}
return n
}
func (td *TodoDelete) sqlExec(ctx context.Context) (int, error) {
_spec := &sqlgraph.DeleteSpec{
Node: &sqlgraph.NodeSpec{
Table: todo.Table,
ID: &sqlgraph.FieldSpec{
Type: field.TypeInt,
Column: todo.FieldID,
},
},
}
if ps := td.mutation.predicates; len(ps) > 0 {
_spec.Predicate = func(selector *sql.Selector) {
for i := range ps {
ps[i](selector)
}
}
}
return sqlgraph.DeleteNodes(ctx, td.driver, _spec)
}
// TodoDeleteOne is the builder for deleting a single Todo entity.
type TodoDeleteOne struct {
td *TodoDelete
}
// Exec executes the deletion query.
func (tdo *TodoDeleteOne) Exec(ctx context.Context) error {
n, err := tdo.td.Exec(ctx)
switch {
case err != nil:
return err
case n == 0:
return &NotFoundError{todo.Label}
default:
return nil
}
}
// ExecX is like Exec, but panics if an error occurs.
func (tdo *TodoDeleteOne) ExecX(ctx context.Context) {
tdo.td.ExecX(ctx)
}