// 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 entv2 import ( "context" "entgo.io/ent/dialect/sql" "entgo.io/ent/dialect/sql/sqlgraph" "entgo.io/ent/entc/integration/migrate/entv2/pet" "entgo.io/ent/entc/integration/migrate/entv2/predicate" "entgo.io/ent/schema/field" ) // PetDelete is the builder for deleting a Pet entity. type PetDelete struct { config hooks []Hook mutation *PetMutation } // Where appends a list predicates to the PetDelete builder. func (pd *PetDelete) Where(ps ...predicate.Pet) *PetDelete { pd.mutation.Where(ps...) return pd } // Exec executes the deletion query and returns how many vertices were deleted. func (pd *PetDelete) Exec(ctx context.Context) (int, error) { return withHooks[int, PetMutation](ctx, pd.sqlExec, pd.mutation, pd.hooks) } // ExecX is like Exec, but panics if an error occurs. func (pd *PetDelete) ExecX(ctx context.Context) int { n, err := pd.Exec(ctx) if err != nil { panic(err) } return n } func (pd *PetDelete) sqlExec(ctx context.Context) (int, error) { _spec := &sqlgraph.DeleteSpec{ Node: &sqlgraph.NodeSpec{ Table: pet.Table, ID: &sqlgraph.FieldSpec{ Type: field.TypeInt, Column: pet.FieldID, }, }, } if ps := pd.mutation.predicates; len(ps) > 0 { _spec.Predicate = func(selector *sql.Selector) { for i := range ps { ps[i](selector) } } } affected, err := sqlgraph.DeleteNodes(ctx, pd.driver, _spec) if err != nil && sqlgraph.IsConstraintError(err) { err = &ConstraintError{msg: err.Error(), wrap: err} } pd.mutation.done = true return affected, err } // PetDeleteOne is the builder for deleting a single Pet entity. type PetDeleteOne struct { pd *PetDelete } // Exec executes the deletion query. func (pdo *PetDeleteOne) Exec(ctx context.Context) error { n, err := pdo.pd.Exec(ctx) switch { case err != nil: return err case n == 0: return &NotFoundError{pet.Label} default: return nil } } // ExecX is like Exec, but panics if an error occurs. func (pdo *PetDeleteOne) ExecX(ctx context.Context) { pdo.pd.ExecX(ctx) }