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