数据库错误操作优化

db/hotimedb.go

@@ -13,7 +13,6 @@ import (
 	"reflect"
 	"sort"
 	"strings"
-	"time"
 )
 
 type HoTimeDB struct {
@@ -272,7 +271,11 @@ func (that *HoTimeDB) GetType() string {
 
 // Action 事务，如果action返回true则执行成功；false则回滚
 func (that *HoTimeDB) Action(action func(db HoTimeDB) (isSuccess bool)) (isSuccess bool) {
-	db := HoTimeDB{DB: that.DB, HoTimeCache: that.HoTimeCache, Prefix: that.Prefix}
+	db := HoTimeDB{that.DB, that.ContextBase, that.DBName,
+		that.HoTimeCache, that.Log, that.Type,
+		that.Prefix, that.LastQuery, that.LastData,
+		that.ConnectFunc, that.LastErr, that.limit, that.Tx,
+		that.SlaveDB, that.Mode}
 	tx, err := db.Begin()
 	if err != nil {
 		that.LastErr.SetError(err)
@@ -315,7 +318,7 @@ func (that *HoTimeDB) InitDb(err ...*Error) *Error {
 		e := that.SlaveDB.Ping()
 		that.LastErr.SetError(e)
 	}
-	that.DB.SetConnMaxLifetime(time.Second)
+	//that.DB.SetConnMaxLifetime(time.Second)
 	return that.LastErr
 }
 
@@ -535,7 +538,7 @@ func (that *HoTimeDB) md5(query string, args ...interface{}) string {
 
 func (that *HoTimeDB) Query(query string, args ...interface{}) []Map {
 	defer func() {
-		if that.Mode == 2 {
+		if that.Mode != 0 {
 			that.Log.Info("SQL:"+that.LastQuery, " DATA:", that.LastData, " ERROR:", that.LastErr.GetError())
 		}
 	}()
@@ -563,16 +566,18 @@ func (that *HoTimeDB) Query(query string, args ...interface{}) []Map {
 		resl, err = db.Query(query, args...)
 	}
 
+	if err != nil && that.LastErr.GetError() != nil &&
+		that.LastErr.GetError().Error() == err.Error() {
+		return nil
+	}
+
 	that.LastErr.SetError(err)
 	if err != nil {
-		if err = db.Ping(); err != nil {
-			that.LastErr.SetError(err)
-			_ = that.InitDb()
-			if that.LastErr.GetError() != nil {
-				return nil
-			}
+
+		if err = db.Ping(); err == nil {
 			return that.Query(query, args...)
 		}
+		that.LastErr.SetError(err)
 		return nil
 	}
 
@@ -581,7 +586,7 @@ func (that *HoTimeDB) Query(query string, args ...interface{}) []Map {
 
 func (that *HoTimeDB) Exec(query string, args ...interface{}) (sql.Result, *Error) {
 	defer func() {
-		if that.Mode == 2 {
+		if that.Mode != 0 {
 			that.Log.Info("SQL: "+that.LastQuery, " DATA: ", that.LastData, " ERROR: ", that.LastErr.GetError())
 		}
 	}()
@@ -602,19 +607,18 @@ func (that *HoTimeDB) Exec(query string, args ...interface{}) (sql.Result, *Erro
 		resl, e = that.DB.Exec(query, args...)
 	}
 
+	if e != nil && that.LastErr.GetError() != nil &&
+		that.LastErr.GetError().Error() == e.Error() {
+		return resl, that.LastErr
+	}
 	that.LastErr.SetError(e)
-
 	//判断是否连接断开了
 	if e != nil {
 
-		if e = that.DB.Ping(); e != nil {
-			that.LastErr.SetError(e)
-			_ = that.InitDb()
-			if that.LastErr.GetError() != nil {
-				return resl, that.LastErr
-			}
+		if e = that.DB.Ping(); e == nil {
 			return that.Exec(query, args...)
 		}
+		that.LastErr.SetError(e)
 		return resl, that.LastErr
 	}