feat: foreign key violatation error (go-gorm#150)

saeidee · Saeid Saeidee · jinzhu · web-flow · commit cb7f946bd69e · 2023-06-07T15:06:51.000+08:00
Co-authored-by: Saeid Saeidee &lt;s.saeidee@sensysgatso.com&gt;
Co-authored-by: Jinzhu &lt;wosmvp@gmail.com&gt;
diff --git a/error_translator.go b/error_translator.go
@@ -6,8 +6,9 @@ import (
 	"gorm.io/gorm"
 )
 
-var errCodes = map[string]int{
-	"uniqueConstraint": 2067,
+var errCodes = map[int]error{
+	2067: gorm.ErrDuplicatedKey,
+	768:  gorm.ErrForeignKeyViolated,
 }
 
 type ErrMessage struct {
@@ -30,8 +31,8 @@ func (dialector Dialector) Translate(err error) error {
 		return err
 	}
 
-	if errMsg.ExtendedCode == errCodes["uniqueConstraint"] {
-		return gorm.ErrDuplicatedKey
+	if translatedErr, found := errCodes[errMsg.ExtendedCode]; found {
+		return translatedErr
 	}
 	return err
 }
diff --git a/go.mod b/go.mod
@@ -4,7 +4,7 @@ go 1.20
 
 require (
 	github.com/mattn/go-sqlite3 v1.14.17
-	gorm.io/gorm v1.25.0
+	gorm.io/gorm v1.25.2-0.20230530020048-26663ab9bf55
 )
 
 require (
diff --git a/go.sum b/go.sum
@@ -6,3 +6,5 @@ github.com/mattn/go-sqlite3 v1.14.17 h1:mCRHCLDUBXgpKAqIKsaAaAsrAlbkeomtRFKXh2L6
 github.com/mattn/go-sqlite3 v1.14.17/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg=
 gorm.io/gorm v1.25.0 h1:+KtYtb2roDz14EQe4bla8CbQlmb9dN3VejSai3lprfU=
 gorm.io/gorm v1.25.0/go.mod h1:L4uxeKpfBml98NYqVqwAdmV1a2nBtAec/cf3fpucW/k=
+gorm.io/gorm v1.25.2-0.20230530020048-26663ab9bf55 h1:sC1Xj4TYrLqg1n3AN10w871An7wJM0gzgcm8jkIkECQ=
+gorm.io/gorm v1.25.2-0.20230530020048-26663ab9bf55/go.mod h1:L4uxeKpfBml98NYqVqwAdmV1a2nBtAec/cf3fpucW/k=

Original file line number	Diff line number	Diff line change
`@@ -6,8 +6,9 @@ import (`
`6`	`6`	`"gorm.io/gorm"`
`7`	`7`	`)`
`8`	`8`
`9`		`-var errCodes = map[string]int{`
`10`		`- "uniqueConstraint": 2067,`
	`9`	`+var errCodes = map[int]error{`
	`10`	`+ 2067: gorm.ErrDuplicatedKey,`
	`11`	`+ 768: gorm.ErrForeignKeyViolated,`
`11`	`12`	`}`
`12`	`13`
`13`	`14`	`type ErrMessage struct {`
`@@ -30,8 +31,8 @@ func (dialector Dialector) Translate(err error) error {`
`30`	`31`	`return err`
`31`	`32`	`}`
`32`	`33`
`33`		`- if errMsg.ExtendedCode == errCodes["uniqueConstraint"] {`
`34`		`- return gorm.ErrDuplicatedKey`
	`34`	`+ if translatedErr, found := errCodes[errMsg.ExtendedCode]; found {`
	`35`	`+ return translatedErr`
`35`	`36`	`}`
`36`	`37`	`return err`
`37`	`38`	`}`