snapshot of https://github.com/ent/ent.git (master, 2026-09-15; blobs > 131072B omitted)

entc/integration/ent/document_delete.go

@@ -0,0 +1,138 @@
+// 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"
+
+	"entgo.io/ent"
+	"entgo.io/ent/dialect/sql"
+	"entgo.io/ent/dialect/sql/sqlgraph"
+	"entgo.io/ent/entc/integration/ent/document"
+	"entgo.io/ent/entc/integration/ent/predicate"
+	"entgo.io/ent/schema/field"
+)
+
+// DocumentDelete is the builder for deleting a Document entity.
+type DocumentDelete struct {
+	config
+	hooks    []Hook
+	mutation *DocumentMutation
+}
+
+// Where appends a list predicates to the DocumentDelete builder.
+func (_d *DocumentDelete) Where(ps ...predicate.Document) *DocumentDelete {
+	_d.mutation.Where(ps...)
+	return _d
+}
+
+// Exec executes the deletion query and returns how many vertices were deleted.
+func (_d *DocumentDelete) Exec(ctx context.Context) (int, error) {
+	return withHooks(ctx, _d.sqlExec, _d.mutation, _d.hooks)
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (_d *DocumentDelete) ExecX(ctx context.Context) int {
+	n, err := _d.Exec(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return n
+}
+
+func (_d *DocumentDelete) sqlExec(ctx context.Context) (int, error) {
+	_spec := sqlgraph.NewDeleteSpec(document.Table, sqlgraph.NewFieldSpec(document.FieldID, field.TypeInt))
+	if ps := _d.mutation.Predicates(); len(ps) > 0 {
+		_spec.Predicate = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	// Collect blob keys before deleting rows so we can remove blobs from storage afterward.
+	_blobKeys, _blobErr := ent.NewBlobs(_d.mutation.blobOpeners.Document).Delete(ctx, &sqlgraph.BlobSpec{
+		Driver:    _d.driver,
+		Predicate: _spec.Predicate,
+		Table:     document.Table,
+		Columns: map[string]string{
+			document.FieldContent:     "content_key",
+			document.FieldThumbnail:   "thumbnail_key",
+			document.FieldAttachment:  "attachment_key",
+			document.FieldMetadata:    "metadata_key",
+			document.FieldPayload:     "payload_key",
+			document.FieldDescription: "description_key",
+			document.FieldArchive:     "archive_key",
+		},
+		CheckRefs: map[string]bool{
+			document.FieldContent:    true,
+			document.FieldThumbnail:  true,
+			document.FieldAttachment: true,
+		},
+	})
+	if _blobErr != nil {
+		return 0, _blobErr
+	}
+	affected, err := sqlgraph.DeleteNodes(ctx, _d.driver, _spec)
+	if err != nil && sqlgraph.IsConstraintError(err) {
+		err = &ConstraintError{msg: err.Error(), wrap: err}
+	}
+	if err == nil {
+		// Resolve which blobs are unreferenced now that the rows are gone, while the
+		// transaction is still open, and remove them from storage once it commits.
+		_blobCleanup, _blobErr := _blobKeys.Cleanup(ctx)
+		if _blobErr != nil {
+			return 0, _blobErr
+		}
+		if txd, ok := _d.driver.(*txDriver); ok {
+			txd.mu.Lock()
+			txd.onCommit = append(txd.onCommit, func(next Committer) Committer {
+				return CommitFunc(func(ctx context.Context, tx *Tx) error {
+					if err := next.Commit(ctx, tx); err != nil {
+						return err
+					}
+					return _blobCleanup(ctx)
+				})
+			})
+			txd.mu.Unlock()
+		} else {
+			err = _blobCleanup(ctx)
+		}
+	}
+	_d.mutation.done = true
+	return affected, err
+}
+
+// DocumentDeleteOne is the builder for deleting a single Document entity.
+type DocumentDeleteOne struct {
+	_d *DocumentDelete
+}
+
+// Where appends a list predicates to the DocumentDelete builder.
+func (_d *DocumentDeleteOne) Where(ps ...predicate.Document) *DocumentDeleteOne {
+	_d._d.mutation.Where(ps...)
+	return _d
+}
+
+// Exec executes the deletion query.
+func (_d *DocumentDeleteOne) Exec(ctx context.Context) error {
+	n, err := _d._d.Exec(ctx)
+	switch {
+	case err != nil:
+		return err
+	case n == 0:
+		return &NotFoundError{document.Label}
+	default:
+		return nil
+	}
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (_d *DocumentDeleteOne) ExecX(ctx context.Context) {
+	if err := _d.Exec(ctx); err != nil {
+		panic(err)
+	}
+}