Merge pull request #35 from DrOctavius/main

db client error codes
kyaxcorp · Mar 29, 2023 · 1484e0b · 1484e0b
2 parents fd518b9 + dc01585
commit 1484e0b
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/core/clients/db/dberr/codes.go b/core/clients/db/dberr/codes.go
@@ -0,0 +1,22 @@
+package dberr
+
+import (
+	"errors"
+	"github.com/go-sql-driver/mysql"
+	"github.com/jackc/pgx/v5/pgconn"
+)
+
+func IsDuplicateKey(_err error) bool {
+	if pgError := _err.(*pgconn.PgError); errors.Is(_err, pgError) {
+		switch pgError.Code {
+		case "23505":
+			return true
+		}
+	} else if mysqlErr := _err.(*mysql.MySQLError); errors.Is(_err, mysqlErr) {
+		switch mysqlErr.Number {
+		case 1062:
+			return true
+		}
+	}
+	return false
+}