From 16ae9b2dbe9130c90e15af8c87605f69f97c664c Mon Sep 17 00:00:00 2001 From: guzeng Date: Mon, 20 Feb 2023 09:33:09 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=9F=A5=E8=AF=A2=E6=9D=A1?= =?UTF-8?q?=E4=BB=B6=EF=BC=8C=E5=85=BC=E5=AE=B9=E4=B8=8D=E5=90=8C=E6=9C=8D?= =?UTF-8?q?=E5=8A=A1=E5=95=86=E6=95=B0=E6=8D=AE=E5=BA=93?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- chain.go | 6 +++++- conn.go | 3 +++ db.go | 14 ++++++++++---- 3 files changed, 18 insertions(+), 5 deletions(-) diff --git a/chain.go b/chain.go index 81c66d5..3a98af1 100644 --- a/chain.go +++ b/chain.go @@ -266,7 +266,11 @@ func (this *Query) BuildSelectSql() (map[string]interface{}, error) { } else { title = "*" } - sql = helper.StringJoin("/*slave*/ select ", title) + if DB_PROVIDER == "TencentDB" { + sql = helper.StringJoin("/*slave*/ select ", title) + } else { + sql = helper.StringJoin("select ", title) + } if this.alias != "" { table = helper.StringJoin(table, " as ", this.alias) diff --git a/conn.go b/conn.go index d07a68d..06b0a18 100755 --- a/conn.go +++ b/conn.go @@ -17,6 +17,9 @@ var DB *sql.DB var SLAVER_DB *sql.DB +//db类型,默认空，如TencentDB（腾讯）, +var DB_PROVIDER string + func Connect(DBHOST, DBUSER, DBPWD, DBNAME, DBPORT string, conns ...int) error { log.Println("database connectting...") diff --git a/db.go b/db.go index 145c49b..c464abb 100644 --- a/db.go +++ b/db.go @@ -316,8 +316,11 @@ func GetRow(dbName, table_name, alias string, titles string, join [][]string, wh } else { title = "*" } - sql_str = helper.StringJoin("/*slave*/ select ", title) - + if DB_PROVIDER == "TencentDB" { + sql_str = helper.StringJoin("/*slave*/ select ", title) + } else { + sql_str = helper.StringJoin("select ", title) + } if alias != "" { table = helper.StringJoin(table, " as ", alias) } @@ -463,8 +466,11 @@ func FetchRows(dbName, table_name, alias string, titles string, join [][]string, } else { title = "*" } - sql_str = helper.StringJoin("/*slave*/ select ", title) - + if DB_PROVIDER == "TencentDB" { + sql_str = helper.StringJoin("/*slave*/ select ", title) + } else { + sql_str = helper.StringJoin("select ", title) + } if alias != "" { table = helper.StringJoin(table, " as ", alias) }