golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-04-30 14:40:57 +03:00
Code Issues Packages Projects Releases Wiki Activity

entc/gen: catch constraint failures in delete operations (#2664)

Browse Source
This commit is contained in:
Ariel Mashraki
2022-06-19 14:34:32 +03:00
committed by GitHub
parent b89b13bbdb
commit abefaa7f24
105 changed files with 545 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
dialect/sql/sqlgraph/errors.go
View File

@@ -9,7 +9,7 @@ import (
"strings"
)
// IsConstraintError returns true if the error resulted from a DB constraint violation
// IsConstraintError returns true if the error resulted from a database constraint violation.
func IsConstraintError(err error) bool {
var e *ConstraintError
return errors.As(err, &e) || IsUniqueConstraintError(err) || IsForeignKeyConstraintError(err)
@@ -18,12 +18,11 @@ func IsConstraintError(err error) bool {
// IsUniqueConstraintError reports if the error resulted from a DB uniqueness constraint violation.
// e.g. duplicate value in unique index.
func IsUniqueConstraintError(err error) bool {
uniquenessErrors := []string{
for _, s := range []string{
"Error 1062", // MySQL
"violates unique constraint", // Postgres
"UNIQUE constraint failed", // SQLite
}
for _, s := range uniquenessErrors {
} {
if strings.Contains(err.Error(), s) {
return true
}
@@ -31,15 +30,15 @@ func IsUniqueConstraintError(err error) bool {
return false
}
// IsForeignKeyConstraintError reports if the error resulted from a DB FK constraint violation.
// IsForeignKeyConstraintError reports if the error resulted from a database foreign-key constraint violation.
// e.g. parent row does not exist.
func IsForeignKeyConstraintError(err error) bool {
fkErrors := []string{
"Error 1452", // MySQL
for _, s := range []string{
"Error 1451", // MySQL (Cannot delete or update a parent row).
"Error 1452", // MySQL (Cannot add or update a child row).
"violates foreign key constraint", // Postgres
"FOREIGN KEY constraint failed", // SQLite
}
for _, s := range fkErrors {
} {
if strings.Contains(err.Error(), s) {
return true
}