数据库错误操作优化

db/hotimedb.go

@@ -271,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)
@@ -534,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())
 		}
 	}()
@@ -562,18 +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 {
 			return that.Query(query, args...)
-		} else {
-
-			err = that.InitDb()
-			that.LastErr.SetError(err)
-			if err == nil {
-				return that.Query(query, args...)
-			}
 		}
+		that.LastErr.SetError(err)
 		return nil
 	}
 
@@ -582,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())
 		}
 	}()
@@ -603,20 +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 {
 			return that.Exec(query, args...)
-		} else {
-			e = that.InitDb()
-			that.LastErr.SetError(e)
-			if e == nil {
-				return that.Exec(query, args...)
-			}
 		}
+		that.LastErr.SetError(e)
 		return resl, that.LastErr
 	}
 
@@ -663,11 +665,13 @@ func (that *HoTimeDB) Select(table string, qu ...interface{}) []Map {
 			data := ObjToSlice(qu[intQs])
 			for i := 0; i < len(data); i++ {
 				k := data.GetString(i)
-				if strings.Contains(k, " AS ") {
+				if strings.Contains(k, " AS ") ||
+					strings.Contains(k, ".") {
 
 					query += " " + k + " "
 
 				} else {
+
 					query += " `" + k + "` "
 				}