Fixed error mapper for constraint errors (#63)

* Fixed error mapper for constraint errors Fixed error mapper for constraint errors and added error 1451. * Fixed error mapper regardless of whether SQLSTATE is present or not * Combined the two constraint errors Co-authored-by: Surya Asriadie <[email protected]> --------- Co-authored-by: Surya Asriadie <[email protected]>
go-rel · Jan 23, 2024 · 685eddb · 685eddb
1 parent 57be534
commit 685eddb
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/mysql.go b/mysql.go
@@ -122,20 +122,28 @@ func errorMapper(err error) error {
 		msg          = err.Error()
 		errCodeSep   = ':'
 		errCodeIndex = strings.IndexRune(msg, errCodeSep)
+		errStateSep   = '('
+		errStateIndex = -1
 	)
 
 	if errCodeIndex < 0 {
 		errCodeIndex = 0
 	}
 
-	switch msg[:errCodeIndex] {
+	errStateIndex = strings.IndexRune(msg[:errCodeIndex], errStateSep)
+	errStateIndex--
+	if errStateIndex < 0 {
+		errStateIndex = errCodeIndex
+	}
+
+	switch (msg[:errCodeIndex])[:errStateIndex] {
 	case "Error 1062":
 		return rel.ConstraintError{
 			Key:  sql.ExtractString(msg, "key '", "'"),
 			Type: rel.UniqueConstraint,
 			Err:  err,
 		}
-	case "Error 1452":
+	case "Error 1451", "Error 1452":
 		return rel.ConstraintError{
 			Key:  sql.ExtractString(msg, "CONSTRAINT `", "`"),
 			Type: rel.ForeignKeyConstraint,