添加错误判断函数

db_operations/db_operations.go

@@ -22,7 +22,6 @@ type EventDBOperations interface {
 
 type BaseDBOperations interface {
 	// 会重置数据库连接的方法
-	NewSession() DBOperations
 	Table(name string, args ...any) DBOperations
 
 	// 执行SQL语句，使用Raw之后，为了触发SQL执行，需要调用Row或者Rows

db_operations/dberr.go

@@ -0,0 +1,19 @@
+package db_operations
+
+import (
+	"errors"
+	"strings"
+)
+
+var (
+	ErrDBRecordHasExist = errors.New("记录已存在")
+	ErrDBRecordNotExist = errors.New("记录不存在")
+)
+
+func IsErrorDBRecordHasExist(err error) bool {
+	return strings.Contains(err.Error(), "记录已存在")
+}
+
+func IsErrorDBRecordNotExist(err error) bool {
+	return strings.Contains(err.Error(), "记录不存在")
+}

db_operations/dberr/dberr.go

@@ -1,9 +0,0 @@
-package dberr
-
-import "errors"
-
-var (
-	ErrDBOperationsNotExist = errors.New("数据库操作不存在")
-	ErrDBRecordHasExist     = errors.New("记录已存在")
-	ErrDBRecordNotExist     = errors.New("记录不存在")
-)

db_operations/operations.go

@@ -2,7 +2,6 @@ package db_operations
 
 import (
 	"database/sql"
-	"git.sxidc.com/service-supports/ds-sdk/db_operations/dberr"
 	"git.sxidc.com/service-supports/fslog"
 	"github.com/mitchellh/mapstructure"
 	"gorm.io/gorm"
@@ -182,7 +181,7 @@ func (op *Operations) Create(tableRow *TableRow) error {
 	err := op.processDB.Create(tableRow.ToMap()).Error
 	if err != nil {
 		if strings.Contains(err.Error(), "SQLSTATE 23505") {
-			return dberr.ErrDBRecordHasExist
+			return ErrDBRecordHasExist
 		}
 
 		return err
@@ -200,7 +199,7 @@ func (op *Operations) CreateBatch(tableRows []TableRow) error {
 	err := op.processDB.Create(tableRowMaps).Error
 	if err != nil {
 		if strings.Contains(err.Error(), "SQLSTATE 23505") {
-			return dberr.ErrDBRecordHasExist
+			return ErrDBRecordHasExist
 		}
 
 		return err
@@ -217,7 +216,7 @@ func (op *Operations) Updates(newTableRow *TableRow) error {
 	err := op.processDB.Updates(newTableRow.ToMap()).Error
 	if err != nil {
 		if strings.Contains(err.Error(), "SQLSTATE 23505") {
-			return dberr.ErrDBRecordHasExist
+			return ErrDBRecordHasExist
 		}
 
 		return err
@@ -267,7 +266,7 @@ func (op *Operations) Row() (*TableRow, error) {
 	}
 
 	if valueMap == nil || len(valueMap) == 0 {
-		return nil, dberr.ErrDBRecordNotExist
+		return nil, ErrDBRecordNotExist
 	}
 
 	return NewTableRowFromMap(valueMap), nil