From 07e9c8a562a89ce710e587e7b67fb9f036f3b279 Mon Sep 17 00:00:00 2001 From: guzeng Date: Tue, 1 Mar 2022 17:58:45 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9GetRow=E6=9F=A5=E8=AF=A2?= =?UTF-8?q?=E6=96=B9=E6=B3=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- db.go | 20 ++++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) diff --git a/db.go b/db.go index a8fce76..6b454b1 100644 --- a/db.go +++ b/db.go @@ -356,9 +356,19 @@ func GetRow(dbName, table_name, alias string, titles string, join [][]string, wh var err error var queryNum int = 0 + sql_str = helper.StringJoin(sql_str, " limit 1") + + var db *sql.DB + + if SLAVER_DB != nil { + db = SLAVER_DB + } else { + db = DB + } + for queryNum < 2 { //如发生错误，继续查询2次，防止数据库连接断开问题 - rows, err = SLAVER_DB.Query(sql_str, valueList...) + rows, err = db.Query(sql_str, valueList...) if err == nil { break @@ -495,12 +505,18 @@ func FetchRows(dbName, table_name, alias string, titles string, join [][]string, return 0, list, errors.New("参数错误，条件值错误") } + var db *sql.DB + if SLAVER_DB != nil { + db = SLAVER_DB + } else { + db = DB + } var rows *sql.Rows var err error var queryNum int = 0 for queryNum < 2 { //如发生错误，继续查询2次，防止数据库连接断开问题 - rows, err = SLAVER_DB.Query(sql_str, valueList...) + rows, err = db.Query(sql_str, valueList...) if err == nil { break