transaction bug修复

chain.go

@@ -145,6 +145,27 @@ func (this *Query) Join(join []string) *Query {
 	this.join = append(this.join, join)
 	return this
 }
+
+/**
+ * 左连接
+ * 2023/08/10
+ * gz
+ */
+func (this *Query) LeftJoin(table_name string, condition string) *Query {
+	this.join = append(this.join, []string{table_name, condition, "left"})
+	return this
+}
+
+/**
+ * 右连接
+ * 2023/08/10
+ * gz
+ */
+func (this *Query) RightJoin(table_name string, condition string) *Query {
+	this.join = append(this.join, []string{table_name, condition, "right"})
+	return this
+}
+
 func (this *Query) Data(data string) *Query {
 	this.data = append(this.data, data)
 	return this
@@ -176,10 +197,11 @@ func (this *Query) Clean() *Query {
 	this.save_data = this.save_data[0:0]
 	this.upd_field = this.upd_field[0:0]
 	this.having = ""
+	this.alias = ""
 	return this
 }
 
-//获取表格信息
+// 获取表格信息
 func (this *Query) GetTableInfo(table string) (map[string]interface{}, error) {
 	field := []string{
 		"COLUMN_NAME",    //字段名
@@ -245,7 +267,12 @@ func (this *Query) GetTableInfo(table string) (map[string]interface{}, error) {
 	}, nil
 }
 
-//构造子查询
+// 返回表名
+func (this *Query) GetTableName(table string) string {
+	return getTableName(this.dbname, table)
+}
+
+// 构造子查询
 func (this *Query) BuildSelectSql() (map[string]interface{}, error) {
 	if this.dbname == "" && this.table == "" {
 		return nil, errors.New("参数错误，没有数据表")
@@ -266,7 +293,22 @@ func (this *Query) BuildSelectSql() (map[string]interface{}, error) {
 	} else {
 		title = "*"
 	}
-	sql = helper.StringJoin("/*slave*/ select ", title)
+
+	if this.dbtype == "mssql" {
+		if this.page_size > 0 {
+			sql = helper.StringJoin("select top ", helper.ToStr(this.page_size), " ")
+		} else {
+			sql = "select "
+		}
+	} else {
+		if DB_PROVIDER == "TencentDB" {
+			sql = "/*slave*/ select "
+		} else {
+			sql = "select "
+		}
+	}
+
+	sql = helper.StringJoin(sql, title)
 
 	if this.alias != "" {
 		table = helper.StringJoin(table, " as ", this.alias)
@@ -275,15 +317,17 @@ func (this *Query) BuildSelectSql() (map[string]interface{}, error) {
 	sql = helper.StringJoin(sql, " from ", table)
 
 	if len(this.join) > 0 {
+		join_type := "left"
 		for _, joinitem := range this.join {
 			if len(joinitem) < 2 {
 				continue
 			}
 			if len(joinitem) == 3 {
-				sql = helper.StringJoin(sql, " ", joinitem[2], " join ", getTableName(this.dbname, joinitem[0], this.dbtype), " on ", joinitem[1])
+				join_type = joinitem[2]
 			} else { //默认左连接
-				sql = helper.StringJoin(sql, " left join ", getTableName(this.dbname, joinitem[0], this.dbtype), " on ", joinitem[1])
+				join_type = "left"
 			}
+			sql = helper.StringJoin(sql, " ", join_type, " join ", getTableName(this.dbname, joinitem[0], this.dbtype), " on ", joinitem[1])
 		}
 	}
 	if len(this.where) > 0 || len(this.where_or) > 0 {
@@ -311,7 +355,7 @@ func (this *Query) BuildSelectSql() (map[string]interface{}, error) {
 		sql = helper.StringJoin(sql, " order by ", this.orderby)
 	}
 
-	if this.page > 0 || this.page_size > 0 {
+	if this.dbtype == "mysql" && (this.page > 0 || this.page_size > 0) {
 
 		if this.page < 1 {
 			this.page = 1
@@ -354,7 +398,11 @@ func (this *Query) QueryStmt() error {
 
 	if SLAVER_DB != nil {
 		this.conn = SLAVER_DB
-	} else {
+	}
+	// else {
+	// 	this.conn = DB
+	// }
+	if this.conn == nil {
 		this.conn = DB
 	}
 
@@ -794,7 +842,7 @@ func (this *Query) Update() (int64, error) {
 	return StmtForUpdateExec(this.stmt, this.value)
 }
 
-//批量更新
+// 批量更新
 func (this *Query) UpdateAll() (int64, error) {
 
 	err := this.UpdateAllStmt()

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...")
@@ -120,7 +123,16 @@ func getTableName(dbName, table string, dbtype ...string) string {
 			ret = table
 		}
 		if dbName != "" {
-			ret = helper.StringJoin(dbName, ".", table)
+			if strings.Contains(table, ",") {
+				arr := strings.Split(table, ",")
+				arrStrs := make([]string, 0, len(arr))
+				for _, v := range arr {
+					arrStrs = append(arrStrs, helper.StringJoin(dbName, ".", v))
+				}
+				ret = strings.Join(arrStrs, ",")
+			} else {
+				ret = helper.StringJoin(dbName, ".", table)
+			}
 		} else {
 			ret = table
 		}

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)
 	}
@@ -329,7 +332,9 @@ func GetRow(dbName, table_name, alias string, titles string, join [][]string, wh
 			if len(joinitem) < 2 {
 				continue
 			}
-			if len(joinitem) == 3 {
+			if len(joinitem) == 4 {
+				sql_str = helper.StringJoin(sql_str, " ", joinitem[2], " join ", joinitem[0], " on ", joinitem[1])
+			} else if len(joinitem) == 3 {
 				sql_str = helper.StringJoin(sql_str, " ", joinitem[2], " join ", getTableName(dbName, joinitem[0]), " on ", joinitem[1])
 			} else { //默认左连接
 				sql_str = helper.StringJoin(sql_str, " left join ", getTableName(dbName, joinitem[0]), " on ", joinitem[1])
@@ -401,6 +406,7 @@ func GetRow(dbName, table_name, alias string, titles string, join [][]string, wh
 	}
 
 	if err != nil {
+		log.Println("DB error:", err)
 		rows.Close()
 		return count, info, err
 	}
@@ -431,6 +437,7 @@ func GetRow(dbName, table_name, alias string, titles string, join [][]string, wh
 	}
 	rows.Close()
 	if rowerr != nil {
+		log.Println("DB row error:", rowerr)
 		return count, info, rowerr
 	}
 	return count, info, nil
@@ -463,8 +470,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)
 	}
@@ -476,7 +486,9 @@ func FetchRows(dbName, table_name, alias string, titles string, join [][]string,
 			if len(joinitem) < 2 {
 				continue
 			}
-			if len(joinitem) == 3 {
+			if len(joinitem) == 4 {
+				sql_str = helper.StringJoin(sql_str, " ", joinitem[2], " join ", joinitem[0], " on ", joinitem[1])
+			} else if len(joinitem) == 3 {
 				sql_str = helper.StringJoin(sql_str, " ", joinitem[2], " join ", getTableName(dbName, joinitem[0]), " on ", joinitem[1])
 			} else { //默认左连接
 				sql_str = helper.StringJoin(sql_str, " left join ", getTableName(dbName, joinitem[0]), " on ", joinitem[1])

sqlserver.go

@@ -15,7 +15,7 @@ import (
 
 var MSDB_CONN *sql.DB
 
-func MSConnect(DBHOST, DBUSER, DBPWD, DBNAME, DBPORT string, conns ...int) error {
+func MSConnect(DBHOST, DBUSER, DBPWD, DBNAME, DBPORT, encrypt string, conns ...int) error {
 
 	log.Println("msdb connectting...")
 
@@ -29,7 +29,12 @@ func MSConnect(DBHOST, DBUSER, DBPWD, DBNAME, DBPORT string, conns ...int) error
 
 			db_port, _ := strconv.Atoi(DBPORT)
 
-			connString := fmt.Sprintf("server=%s;port=%d;database=%s;user id=%s;password=%s", DBHOST, db_port, DBNAME, DBUSER, DBPWD)
+			params := "server=%s;port=%d;database=%s;user id=%s;password=%s"
+			if encrypt != "" {
+				params = params + ";encrypt=" + encrypt
+			}
+
+			connString := fmt.Sprintf(params, DBHOST, db_port, DBNAME, DBUSER, DBPWD)
 
 			log.Println(connString)
 

transaction.go

@@ -25,7 +25,7 @@ func TxInsert(tx *sql.Tx, dbname, table string, data map[string]string) (int64,
 	if strings.Contains(table, "select ") {
 		dbName = table
 	} else {
-		dbName = getTableName(dbName, table)
+		dbName = getTableName(dbname, table)
 	}
 	if len(data) < 1 {
 		return 0, errors.New("参数错误，没有要写入的数据")
@@ -71,7 +71,7 @@ func TxPreInsert(tx *sql.Tx, dbname, table string, data map[string]interface{})
 	if strings.Contains(table, "select ") {
 		dbName = table
 	} else {
-		dbName = getTableName(dbName, table)
+		dbName = getTableName(dbname, table)
 	}
 
 	if len(data) < 1 {
@@ -122,7 +122,7 @@ func TxUpdate(tx *sql.Tx, dbname, table string, data map[string]string, where ma
 	if strings.Contains(table, "select ") {
 		dbName = table
 	} else {
-		dbName = getTableName(dbName, table)
+		dbName = getTableName(dbname, table)
 	}
 	if len(data) < 1 {
 		return rowsAffected, errors.New("参数错误，没有要写入的数据")
@@ -186,7 +186,7 @@ func TxPreUpdate(tx *sql.Tx, dbname, table string, data []string, where []string
 	if strings.Contains(table, "select ") {
 		dbName = table
 	} else {
-		dbName = getTableName(dbName, table)
+		dbName = getTableName(dbname, table)
 	}
 
 	if len(where) < 1 {
@@ -228,7 +228,7 @@ func TxDelete(tx *sql.Tx, dbname, table string, where map[string]string, del_cou
 	if strings.Contains(table, "select ") {
 		dbName = table
 	} else {
-		dbName = getTableName(dbName, table)
+		dbName = getTableName(dbname, table)
 	}
 	if len(where) < 1 {
 		return count, errors.New("参数错误，没有删除条件")

transaction_chain.go

@@ -147,6 +147,26 @@ func (this *TxQuery) Join(join []string) *TxQuery {
 	this.join = append(this.join, join)
 	return this
 }
+
+/**
+ * 左连接
+ * 2023/08/10
+ * gz
+ */
+func (this *TxQuery) LeftJoin(table_name string, condition string) *TxQuery {
+	this.join = append(this.join, []string{table_name, condition, "left"})
+	return this
+}
+
+/**
+ * 右连接
+ * 2023/08/10
+ * gz
+ */
+func (this *TxQuery) RightJoin(table_name string, condition string) *TxQuery {
+	this.join = append(this.join, []string{table_name, condition, "right"})
+	return this
+}
 func (this *TxQuery) Data(data string) *TxQuery {
 	this.data = append(this.data, data)
 	return this
@@ -177,10 +197,16 @@ func (this *TxQuery) Clean() *TxQuery {
 	this.save_data = this.save_data[0:0]
 	this.upd_field = this.upd_field[0:0]
 	this.having = ""
+	this.alias = ""
 	return this
 }
 
-//构造子查询
+// 返回表名
+func (this *TxQuery) GetTableName(table string) string {
+	return getTableName(this.dbname, table)
+}
+
+// 构造子查询
 func (this *TxQuery) BuildSelectSql() (map[string]interface{}, error) {
 	if this.dbname == "" && this.table == "" {
 		return nil, errors.New("参数错误，没有数据表")
@@ -278,7 +304,7 @@ func (this *TxQuery) BuildSelectSql() (map[string]interface{}, error) {
 	}, nil
 }
 
-//获取表格信息
+// 获取表格信息
 func (this *TxQuery) GetTableInfo(table string) (map[string]interface{}, error) {
 	field := []string{
 		"COLUMN_NAME",    //字段名