tgo
/
dbquery
6
0
Fork 0
Code Issues 0 Pull Requests 0 Projects 0 Releases 48 Wiki Activity
Browse Source

兼容高斯

master v1.0.8
loshiqi 3 days ago
parent
bd754d1507
commit
16cff9c0ca
6 changed files with 288 additions and 80 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +28
    -9
      chain.go
  2. +3
    -1
      conn.go
  3. +70
    -29
      db.go
  4. +49
    -7
      prepare.go
  5. +82
    -27
      transaction.go
  6. +56
    -7
      transaction_chain.go

+ 28
- 9
chain.go View File

@ -69,6 +69,9 @@ func (this *Query) Conn(conn *sql.DB) *Query {
} }
func (this *Query) Db(dbname string) *Query { func (this *Query) Db(dbname string) *Query {
this.dbname = dbname this.dbname = dbname
if DB_PROVIDER == "PgsqlDb" {
this.dbname = ""
}
return this return this
} }
@ -226,6 +229,9 @@ func (this *Query) GetTableInfo(table string) (map[string]interface{}, error) {
if this.conn == nil { if this.conn == nil {
this.conn = DB this.conn = DB
} }
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
}
stmtSql, err := this.conn.Prepare(sql) stmtSql, err := this.conn.Prepare(sql)
if err != nil { if err != nil {
return nil, err return nil, err
@ -414,7 +420,11 @@ func (this *Query) BuildSelectSql() (map[string]interface{}, error) {
from := strconv.Itoa((this.page - 1) * this.page_size) from := strconv.Itoa((this.page - 1) * this.page_size)
offset := strconv.Itoa(this.page_size) offset := strconv.Itoa(this.page_size)
if from != "" && offset != "" { if from != "" && offset != "" {
sql = helper.StringJoin(sql, " limit ", from, " , ", offset)
if DB_PROVIDER == "PgsqlDb" {
sql = helper.StringJoin(sql, " limit ", offset, " OFFSET ", from)
} else {
sql = helper.StringJoin(sql, " limit ", from, " , ", offset)
}
} }
} }
if this.debug { if this.debug {
@ -429,6 +439,9 @@ func (this *Query) BuildSelectSql() (map[string]interface{}, error) {
if condition_len != len(this.value) { if condition_len != len(this.value) {
return nil, errors.New("参数错误，条件值错误") return nil, errors.New("参数错误，条件值错误")
} }
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
}
return map[string]interface{}{ return map[string]interface{}{
"sql": sql, "sql": sql,
"value": this.value, "value": this.value,
@ -501,9 +514,7 @@ func (this *Query) UpdateStmt() error {
this.conn = DB this.conn = DB
} }
if DB_PROVIDER == "PgsqlDb" { if DB_PROVIDER == "PgsqlDb" {
log.Println("PgsqlDb sql", sql)
sql = sqlx.Rebind(sqlx.DOLLAR, sql) sql = sqlx.Rebind(sqlx.DOLLAR, sql)
log.Println("PgsqlDb sql", sql)
} }
stmt, err = this.conn.Prepare(sql) stmt, err = this.conn.Prepare(sql)
@ -608,6 +619,9 @@ func (this *Query) UpdateAllStmt() error {
if len(valSql) > 1 { if len(valSql) > 1 {
setText = " value " setText = " value "
} }
if DB_PROVIDER == "PgsqlDb" {
setText = " values "
}
sql = helper.StringJoin("insert into ", dbName, " (", strings.Join(this.data, " , "), ")", setText, strings.Join(valSql, ","), " ON DUPLICATE KEY UPDATE ", strings.Join(updSql, " , ")) sql = helper.StringJoin("insert into ", dbName, " (", strings.Join(this.data, " , "), ")", setText, strings.Join(valSql, ","), " ON DUPLICATE KEY UPDATE ", strings.Join(updSql, " , "))
if this.debug { if this.debug {
@ -626,7 +640,9 @@ func (this *Query) UpdateAllStmt() error {
if this.conn == nil { if this.conn == nil {
this.conn = DB this.conn = DB
} }
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
}
stmt, err = this.conn.Prepare(sql) stmt, err = this.conn.Prepare(sql)
if err != nil { if err != nil {
@ -697,6 +713,9 @@ func (this *Query) CreateAllStmt() error {
if len(valSql) > 1 { if len(valSql) > 1 {
setText = " value " setText = " value "
} }
if DB_PROVIDER == "PgsqlDb" {
setText = " values "
}
sql = helper.StringJoin("insert into ", dbName, " (", strings.Join(this.data, " , "), ")", setText, strings.Join(valSql, ",")) sql = helper.StringJoin("insert into ", dbName, " (", strings.Join(this.data, " , "), ")", setText, strings.Join(valSql, ","))
if this.debug { if this.debug {
@ -745,10 +764,10 @@ func (this *Query) CreateStmt() error {
value_data := []string{} value_data := []string{}
for _, rv := range this.data { for _, rv := range this.data {
dv := strings.Split(rv, "=") dv := strings.Split(rv, "=")
if len(dv) != 2 {
if len(dv) < 2 {
return errors.New("参数错误，条件值错误,=号不存在") return errors.New("参数错误，条件值错误,=号不存在")
} }
if strings.Contains(rv, "=?") {
if strings.Contains(rv, "?") {
insert_data = append(insert_data, dv[0]) insert_data = append(insert_data, dv[0])
value_data = append(value_data, "?") value_data = append(value_data, "?")
} else { } else {
@ -780,9 +799,7 @@ func (this *Query) CreateStmt() error {
this.conn = DB this.conn = DB
} }
if DB_PROVIDER == "PgsqlDb" { if DB_PROVIDER == "PgsqlDb" {
log.Println("PgsqlDb sql", sql)
sql = sqlx.Rebind(sqlx.DOLLAR, sql) sql = sqlx.Rebind(sqlx.DOLLAR, sql)
log.Println("PgsqlDb sql", sql)
} }
stmt, err = this.conn.Prepare(sql) stmt, err = this.conn.Prepare(sql)
@ -832,7 +849,9 @@ func (this *Query) DeleteStmt() error {
if this.conn == nil { if this.conn == nil {
this.conn = DB this.conn = DB
} }
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
}
stmt, err = this.conn.Prepare(sql) stmt, err = this.conn.Prepare(sql)
if err != nil { if err != nil {


+ 3
- 1
conn.go View File

@ -108,7 +108,9 @@ func CloseSlaverConn() error {
func getTableName(dbName, table string, dbtype ...string) string { func getTableName(dbName, table string, dbtype ...string) string {
var db_type string = "mysql" var db_type string = "mysql"
if DB_PROVIDER == "PgsqlDb" {
dbName = ""
}
if len(dbtype) > 0 { if len(dbtype) > 0 {
if dbtype[0] != "" { if dbtype[0] != "" {
db_type = dbtype[0] db_type = dbtype[0]


+ 70
- 29
db.go View File

@ -45,16 +45,33 @@ func Insert(dbName, table string, data map[string]string) (int64, error) {
valueList[i] = value valueList[i] = value
i++ i++
} }
result, err := DB.Exec("insert into "+dbName+" ("+strings.Join(keyList, ",")+") value("+strings.Join(keyStr, ",")+")", valueList...)
if err != nil {
log.Println("ERROR|插入", dbName, "数据失败,", err)
return insertId, err
var Sql string
Sql = "insert into " + dbName + " (" + strings.Join(keyList, ",") + ") values (" + strings.Join(keyStr, ",") + ")"
if DB_PROVIDER == "PgsqlDb" {
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
Sql = helper.StringJoin(Sql, " RETURNING id")
stmt, err = DB.Prepare(Sql)
if err != nil {
return 0, errors.New("创建失败：" + err.Error())
}
row := stmt.QueryRow(valueList...)
var id int64
err = row.Scan(&id) // 扫描 RETURNING 返回的 ID
if err != nil {
return 0, errors.New("创建失败：" + err.Error())
}
return id, nil
} else { } else {
insertId, _ = result.LastInsertId()
time.Sleep(time.Second * 2)
return insertId, nil
result, err := DB.Exec(Sql, valueList...)
if err != nil {
log.Println("ERROR|插入", dbName, "数据失败,", err)
return insertId, err
} else {
insertId, _ = result.LastInsertId()
time.Sleep(time.Second * 2)
return insertId, nil
}
} }
} }
@ -68,7 +85,6 @@ func Update(dbName, table string, data map[string]string, where map[string]strin
if dbName == "" && table == "" { if dbName == "" && table == "" {
return rowsAffected, errors.New("没有数据表") return rowsAffected, errors.New("没有数据表")
} }
if strings.Contains(table, "select ") { if strings.Contains(table, "select ") {
dbName = table dbName = table
} else { } else {
@ -110,7 +126,12 @@ func Update(dbName, table string, data map[string]string, where map[string]strin
log.Println("ERROR|修改数据表", dbName, "时条件中有空数据,条件：", where, "数据：", data) log.Println("ERROR|修改数据表", dbName, "时条件中有空数据,条件：", where, "数据：", data)
return rowsAffected, errors.New("条件中有空数据") return rowsAffected, errors.New("条件中有空数据")
} }
result, err := DB.Exec("update "+dbName+" set "+strings.Join(keyList, " , ")+" where "+strings.Join(whereStr, " and "), valueList...)
var Sql string
Sql = "update " + dbName + " set " + strings.Join(keyList, " , ") + " where " + strings.Join(whereStr, " and ")
if DB_PROVIDER == "PgsqlDb" {
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
}
result, err := DB.Exec(Sql, valueList...)
if err != nil { if err != nil {
log.Println("ERROR|修改", dbName, "数据失败,", err) log.Println("ERROR|修改", dbName, "数据失败,", err)
@ -132,7 +153,6 @@ func Delete(dbName, table string, data map[string]string, del_count ...string) (
if dbName == "" && table == "" { if dbName == "" && table == "" {
return count, errors.New("没有数据表") return count, errors.New("没有数据表")
} }
if strings.Contains(table, "select ") { if strings.Contains(table, "select ") {
dbName = table dbName = table
} else { } else {
@ -168,7 +188,12 @@ func Delete(dbName, table string, data map[string]string, del_count ...string) (
limitStr = " limit " + del_count[0] limitStr = " limit " + del_count[0]
} }
result, err := DB.Exec("delete from "+dbName+" where "+strings.Join(keyList, " and ")+limitStr, valueList...)
var Sql string
Sql = "delete from " + dbName + " where " + strings.Join(keyList, " and ") + limitStr
if DB_PROVIDER == "PgsqlDb" {
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
}
result, err := DB.Exec(Sql, valueList...)
if err != nil { if err != nil {
log.Println("ERROR|删除", dbName, "数据失败,", err) log.Println("ERROR|删除", dbName, "数据失败,", err)
@ -192,7 +217,6 @@ func GetData(dbName, table string, title string, where map[string]string, limit
if dbName == "" && table == "" { if dbName == "" && table == "" {
return count, info, errors.New("没有数据表") return count, info, errors.New("没有数据表")
} }
dbName = getTableName(dbName, table) dbName = getTableName(dbName, table)
if len(title) < 1 { if len(title) < 1 {
@ -211,7 +235,11 @@ func GetData(dbName, table string, title string, where map[string]string, limit
if _, ok := limit["from"]; ok { if _, ok := limit["from"]; ok {
from = limit["from"] from = limit["from"]
} }
limitStr += " limit " + from + ",1"
if DB_PROVIDER == "PgsqlDb" {
limitStr += " limit 1 OFFSET " + from
} else {
limitStr += " limit " + from + ",1"
}
} else { } else {
limitStr = " limit 1" limitStr = " limit 1"
@ -242,8 +270,12 @@ func GetData(dbName, table string, title string, where map[string]string, limit
var err error var err error
var queryNum int = 0 var queryNum int = 0
for queryNum < 3 { //如发生错误，继续查询3次，防止数据库连接断开问题 for queryNum < 3 { //如发生错误，继续查询3次，防止数据库连接断开问题
rows, err = DB.Query("SELECT "+title+" FROM "+dbName+" where "+strings.Join(keyList, " and ")+" "+limitStr, valueList...)
var Sql string
Sql = "SELECT " + title + " FROM " + dbName + " where " + strings.Join(keyList, " and ") + " " + limitStr
if DB_PROVIDER == "PgsqlDb" {
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
}
rows, err = DB.Query(Sql, valueList...)
if err == nil { if err == nil {
break break
@ -303,7 +335,6 @@ func GetRow(dbName, table_name, alias string, titles string, with, join [][]stri
if dbName == "" && table_name == "" { if dbName == "" && table_name == "" {
return count, info, errors.New("没有数据表") return count, info, errors.New("没有数据表")
} }
table := "" table := ""
withSql := "" withSql := ""
if len(with) > 0 { if len(with) > 0 {
@ -432,9 +463,7 @@ func GetRow(dbName, table_name, alias string, titles string, with, join [][]stri
for queryNum < 2 { //如发生错误，继续查询2次，防止数据库连接断开问题 for queryNum < 2 { //如发生错误，继续查询2次，防止数据库连接断开问题
if DB_PROVIDER == "PgsqlDb" { if DB_PROVIDER == "PgsqlDb" {
log.Println("PgsqlDb sql_str", sql_str)
sql_str = sqlx.Rebind(sqlx.DOLLAR, sql_str) sql_str = sqlx.Rebind(sqlx.DOLLAR, sql_str)
log.Println("PgsqlDb sql_str", sql_str)
} }
rows, err = db.Query(sql_str, valueList...) rows, err = db.Query(sql_str, valueList...)
@ -642,9 +671,7 @@ func FetchRows(dbName, table_name, alias string, titles string, with, join [][]s
var queryNum int = 0 var queryNum int = 0
for queryNum < 2 { //如发生错误，继续查询2次，防止数据库连接断开问题 for queryNum < 2 { //如发生错误，继续查询2次，防止数据库连接断开问题
if DB_PROVIDER == "PgsqlDb" { if DB_PROVIDER == "PgsqlDb" {
log.Println("PgsqlDb sql_str", sql_str)
sql_str = sqlx.Rebind(sqlx.DOLLAR, sql_str) sql_str = sqlx.Rebind(sqlx.DOLLAR, sql_str)
log.Println("PgsqlDb sql_str", sql_str)
} }
rows, err = db.Query(sql_str, valueList...) rows, err = db.Query(sql_str, valueList...)
@ -723,7 +750,6 @@ func GetList(dbName, table string, title string, where map[string]string, limit
if dbName == "" && table == "" { if dbName == "" && table == "" {
return list, errors.New("没有数据表") return list, errors.New("没有数据表")
} }
if strings.Contains(table, "select ") { if strings.Contains(table, "select ") {
dbName = table dbName = table
} else { } else {
@ -751,7 +777,12 @@ func GetList(dbName, table string, title string, where map[string]string, limit
from = limit["from"] from = limit["from"]
} }
if offset != "0" && from != "" { if offset != "0" && from != "" {
limitStr += " limit " + from + "," + offset
if DB_PROVIDER == "PgsqlDb" {
limitStr += " limit " + offset + " OFFSET " + from
} else {
limitStr += " limit " + from + "," + offset
}
} }
} }
@ -785,8 +816,12 @@ func GetList(dbName, table string, title string, where map[string]string, limit
} }
for queryNum < 5 { //如发生错误，继续查询5次，防止数据库连接断开问题 for queryNum < 5 { //如发生错误，继续查询5次，防止数据库连接断开问题
rows, err = DB.Query("select "+title+" from "+dbName+" where "+strings.Join(whereStr, " and ")+" "+limitStr, valueList...)
var Sql string
Sql = "select " + title + " from " + dbName + " where " + strings.Join(whereStr, " and ") + " " + limitStr
if DB_PROVIDER == "PgsqlDb" {
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
}
rows, err = DB.Query(Sql, valueList...)
if err == nil { if err == nil {
break break
@ -850,7 +885,6 @@ func GetTotal(dbName, table string, args ...string) (total int) {
if dbName == "" && table == "" { if dbName == "" && table == "" {
return return
} }
if strings.Contains(table, "select ") { if strings.Contains(table, "select ") {
dbName = table dbName = table
} else { } else {
@ -868,7 +902,6 @@ func GetTotal(dbName, table string, args ...string) (total int) {
var queryNum int = 0 var queryNum int = 0
for queryNum < 5 { //如发生错误，继续查询5次，防止数据库连接断开问题 for queryNum < 5 { //如发生错误，继续查询5次，防止数据库连接断开问题
rows, err = DB.Query("select count(" + title + ") as count from " + dbName + " limit 1") rows, err = DB.Query("select count(" + title + ") as count from " + dbName + " limit 1")
if err == nil { if err == nil {
@ -952,7 +985,12 @@ func GetCount(dbName, table string, where map[string]string, args ...string) (to
for queryNum < 5 { //如发生错误，继续查询5次，防止数据库连接断开问题 for queryNum < 5 { //如发生错误，继续查询5次，防止数据库连接断开问题
rows, err = DB.Query("select count("+title+") as count from "+dbName+" where "+strings.Join(whereStr, " and ")+" limit 1", valueList...)
var Sql string
Sql = "select count(" + title + ") as count from " + dbName + " where " + strings.Join(whereStr, " and ") + " limit 1"
if DB_PROVIDER == "PgsqlDb" {
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
}
rows, err = DB.Query(Sql, valueList...)
if err == nil { if err == nil {
break break
@ -1011,6 +1049,9 @@ func DoQuery(args ...interface{}) ([]map[string]string, error) {
for queryNum < 3 { //如发生错误，继续查询5次，防止数据库连接断开问题 for queryNum < 3 { //如发生错误，继续查询5次，防止数据库连接断开问题
if len(args) > 1 { if len(args) > 1 {
if DB_PROVIDER == "PgsqlDb" {
queryStr = sqlx.Rebind(sqlx.DOLLAR, queryStr)
}
rows, err = DB.Query(queryStr, args[1:]...) //strings.Join(args[1:], ",") rows, err = DB.Query(queryStr, args[1:]...) //strings.Join(args[1:], ",")
if err != nil { if err != nil {
log.Println("ERROR|DoQuery error:", err) log.Println("ERROR|DoQuery error:", err)


+ 49
- 7
prepare.go View File

@ -3,6 +3,7 @@ package dbquery
import ( import (
"database/sql" "database/sql"
"errors" "errors"
"github.com/jmoiron/sqlx"
"log" "log"
"strings" "strings"
@ -41,8 +42,11 @@ func StmtForRead(dbName, table string, title string, where []string, limit map[s
offset = limit["offset"] offset = limit["offset"]
} }
if from != "" && offset != "" { if from != "" && offset != "" {
limitStr += " limit " + from + "," + offset
if DB_PROVIDER == "PgsqlDb" {
limitStr += " limit " + offset + " OFFSET " + from
} else {
limitStr += " limit " + from + "," + offset
}
} }
} }
@ -52,7 +56,12 @@ func StmtForRead(dbName, table string, title string, where []string, limit map[s
if len(where) > 0 { if len(where) > 0 {
// log.Println("SELECT " + title + " FROM " + dbName + " where " + strings.Join(where, " and ") + limitStr) // log.Println("SELECT " + title + " FROM " + dbName + " where " + strings.Join(where, " and ") + limitStr)
stmt, err = DB.Prepare("SELECT " + title + " FROM " + dbName + " where " + strings.Join(where, " and ") + limitStr)
var Sql string
Sql = "SELECT " + title + " FROM " + dbName + " where " + strings.Join(where, " and ") + limitStr
if DB_PROVIDER == "PgsqlDb" {
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
}
stmt, err = DB.Prepare(Sql)
} else { } else {
// log.Println("SELECT " + title + " FROM " + dbName + limitStr) // log.Println("SELECT " + title + " FROM " + dbName + limitStr)
stmt, err = DB.Prepare("SELECT " + title + " FROM " + dbName + limitStr) stmt, err = DB.Prepare("SELECT " + title + " FROM " + dbName + limitStr)
@ -186,8 +195,12 @@ func StmtForUpdate(dbName, table string, data []string, where []string) (*sql.St
var stmt *sql.Stmt var stmt *sql.Stmt
var err error var err error
stmt, err = DB.Prepare("update " + dbName + " set " + strings.Join(data, " , ") + " where " + strings.Join(where, " and "))
var Sql string
Sql = "update " + dbName + " set " + strings.Join(data, " , ") + " where " + strings.Join(where, " and ")
if DB_PROVIDER == "PgsqlDb" {
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
}
stmt, err = DB.Prepare(Sql)
return stmt, err return stmt, err
} }
@ -224,7 +237,34 @@ func StmtForInsert(dbName, table string, data []string) (*sql.Stmt, error) {
var stmt *sql.Stmt var stmt *sql.Stmt
var err error var err error
stmt, err = DB.Prepare("insert into " + dbName + " set " + strings.Join(data, " , "))
var sql string
if DB_PROVIDER == "PgsqlDb" {
insert_data := []string{}
value_data := []string{}
for _, rv := range data {
dv := strings.Split(rv, "=")
if len(dv) < 2 {
return nil, errors.New("参数错误，条件值错误,=号不存在")
}
if strings.Contains(rv, "?") {
insert_data = append(insert_data, dv[0])
value_data = append(value_data, "?")
} else {
insert_data = append(insert_data, dv[0])
value_data = append(value_data, dv[1])
}
}
sql = helper.StringJoin("insert into ", dbName, " ("+strings.Join(insert_data, " , ")+")", " VALUES ", "("+strings.Join(value_data, " , ")+")", " RETURNING id")
} else {
sql = helper.StringJoin("insert into ", dbName, " set ", strings.Join(data, " , "))
}
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
}
//stmt, err = DB.Prepare("insert into " + dbName + " set " + strings.Join(data, " , "))
stmt, err = DB.Prepare(sql)
return stmt, err return stmt, err
} }
@ -362,7 +402,9 @@ func StmtForQuery(querysql string) (*sql.Stmt, error) {
var stmt *sql.Stmt var stmt *sql.Stmt
var err error var err error
if DB_PROVIDER == "PgsqlDb" {
querysql = sqlx.Rebind(sqlx.DOLLAR, querysql)
}
stmt, err = DB.Prepare(querysql) stmt, err = DB.Prepare(querysql)
return stmt, err return stmt, err


+ 82
- 27
transaction.go View File

@ -6,6 +6,8 @@ package dbquery
import ( import (
"database/sql" "database/sql"
"errors" "errors"
"git.tetele.net/tgo/helper"
"github.com/jmoiron/sqlx"
"log" "log"
"strings" "strings"
"time" "time"
@ -43,16 +45,33 @@ func TxInsert(tx *sql.Tx, dbname, table string, data map[string]string) (int64,
valueList[i] = value valueList[i] = value
i++ i++
} }
result, err := tx.Exec("insert into "+dbName+" ("+strings.Join(keyList, ",")+") value("+strings.Join(keyStr, ",")+")", valueList...)
if err != nil {
log.Println("ERROR", "insert into ", dbName, "error:", err)
return insertId, err
if DB_PROVIDER == "PgsqlDb" {
var Sql string
Sql = "insert into " + dbName + " (" + strings.Join(keyList, ",") + ") value(" + strings.Join(keyStr, ",") + ")"
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
Sql = helper.StringJoin(Sql, " RETURNING id")
stmt, err = tx.Prepare(Sql)
if err != nil {
return 0, errors.New("创建失败：" + err.Error())
}
row := stmt.QueryRow(valueList...)
var id int64
err = row.Scan(&id) // 扫描 RETURNING 返回的 ID
if err != nil {
return 0, errors.New("创建失败：" + err.Error())
}
return id, nil
} else { } else {
insertId, _ = result.LastInsertId()
time.Sleep(time.Second * 2)
return insertId, nil
result, err := tx.Exec("insert into "+dbName+" ("+strings.Join(keyList, ",")+") value("+strings.Join(keyStr, ",")+")", valueList...)
if err != nil {
log.Println("ERROR", "insert into ", dbName, "error:", err)
return insertId, err
} else {
insertId, _ = result.LastInsertId()
time.Sleep(time.Second * 2)
return insertId, nil
}
} }
} }
@ -84,28 +103,47 @@ func TxPreInsert(tx *sql.Tx, dbname, table string, data map[string]interface{})
var field []string = make([]string, len(data)) var field []string = make([]string, len(data))
var valuelist []interface{} = make([]interface{}, len(data)) var valuelist []interface{} = make([]interface{}, len(data))
insert_data := []string{}
value_data := []string{}
var i int = 0 var i int = 0
for key, item := range data { for key, item := range data {
field[i] = key + "=?" field[i] = key + "=?"
valuelist[i] = item valuelist[i] = item
i++ i++
}
insert_data = append(insert_data, key)
value_data = append(value_data, "?")
}
if DB_PROVIDER == "PgsqlDb" {
Sql := helper.StringJoin("insert into ", dbName, " ("+strings.Join(insert_data, " , ")+")", " VALUES ", "("+strings.Join(value_data, " , ")+")", " RETURNING id")
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
stmt, err = tx.Prepare(Sql)
if err != nil {
return 0, errors.New("创建失败：" + err.Error())
}
row := stmt.QueryRow(valuelist...)
var id int64
err = row.Scan(&id) // 扫描 RETURNING 返回的 ID
if err != nil {
return 0, errors.New("创建失败：" + err.Error())
}
return id, nil
} else {
sql := "insert into " + dbName + " set " + strings.Join(field, " , ")
stmt, err = tx.Prepare(sql)
sql := "insert into " + dbName + " set " + strings.Join(field, " , ")
stmt, err = tx.Prepare(sql)
if err != nil {
log.Println("insert prepare error:", sql, err)
return 0, errors.New("insert prepare error：" + err.Error())
}
result, err := stmt.Exec(valuelist...)
if err != nil {
log.Println("insert exec error：", sql, valuelist, err)
return 0, errors.New("insert exec error：" + err.Error())
if err != nil {
log.Println("insert prepare error:", sql, err)
return 0, errors.New("insert prepare error：" + err.Error())
}
result, err := stmt.Exec(valuelist...)
if err != nil {
log.Println("insert exec error：", sql, valuelist, err)
return 0, errors.New("insert exec error：" + err.Error())
}
insertId, _ := result.LastInsertId()
return insertId, nil
} }
insertId, _ := result.LastInsertId()
return insertId, nil
} }
@ -160,7 +198,12 @@ func TxUpdate(tx *sql.Tx, dbname, table string, data map[string]string, where ma
log.Println("ERROR", "update", dbName, "error, params empty") log.Println("ERROR", "update", dbName, "error, params empty")
return rowsAffected, errors.New("params empty") return rowsAffected, errors.New("params empty")
} }
result, err := tx.Exec("update "+dbName+" set "+strings.Join(keyList, " , ")+" where "+strings.Join(whereStr, " and "), valueList...)
var Sql string
Sql = "update " + dbName + " set " + strings.Join(keyList, " , ") + " where " + strings.Join(whereStr, " and ")
if DB_PROVIDER == "PgsqlDb" {
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
}
result, err := tx.Exec(Sql, valueList...)
if err != nil { if err != nil {
log.Println("ERROR", "update", dbName, "error:", err) log.Println("ERROR", "update", dbName, "error:", err)
@ -198,7 +241,9 @@ func TxPreUpdate(tx *sql.Tx, dbname, table string, data []string, where []string
var stmt *sql.Stmt var stmt *sql.Stmt
sql := "update " + dbName + " set " + strings.Join(data, " , ") + " where " + strings.Join(where, " and ") sql := "update " + dbName + " set " + strings.Join(data, " , ") + " where " + strings.Join(where, " and ")
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
}
stmt, err = tx.Prepare(sql) stmt, err = tx.Prepare(sql)
if err != nil { if err != nil {
@ -260,7 +305,12 @@ func TxDelete(tx *sql.Tx, dbname, table string, where map[string]string, del_cou
limitStr = " limit " + del_count[0] limitStr = " limit " + del_count[0]
} }
result, err := tx.Exec("delete from "+dbName+" where "+strings.Join(keyList, " and ")+limitStr, valueList...)
var Sql string
Sql = "delete from " + dbName + " where " + strings.Join(keyList, " and ") + limitStr
if DB_PROVIDER == "PgsqlDb" {
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
}
result, err := tx.Exec(Sql, valueList...)
if err != nil { if err != nil {
log.Println("ERROR", "delete from", dbName, "error:", err) log.Println("ERROR", "delete from", dbName, "error:", err)
@ -296,7 +346,12 @@ func TxForRead(tx *sql.Tx, dbName, table string, title string, where []string) (
if len(where) > 0 { if len(where) > 0 {
// log.Println("SELECT " + title + " FROM " + dbName + " where " + strings.Join(where, " and ") + " FOR UPDATE") // log.Println("SELECT " + title + " FROM " + dbName + " where " + strings.Join(where, " and ") + " FOR UPDATE")
stmt, err = tx.Prepare("SELECT " + title + " FROM " + dbName + " where " + strings.Join(where, " and ") + " FOR UPDATE")
var Sql string
Sql = "SELECT " + title + " FROM " + dbName + " where " + strings.Join(where, " and ") + " FOR UPDATE"
if DB_PROVIDER == "PgsqlDb" {
Sql = sqlx.Rebind(sqlx.DOLLAR, Sql)
}
stmt, err = tx.Prepare(Sql)
} else { } else {
// log.Println("SELECT " + title + " FROM " + dbName + " FOR UPDATE") // log.Println("SELECT " + title + " FROM " + dbName + " FOR UPDATE")
stmt, err = tx.Prepare("SELECT " + title + " FROM " + dbName + " FOR UPDATE") stmt, err = tx.Prepare("SELECT " + title + " FROM " + dbName + " FOR UPDATE")


+ 56
- 7
transaction_chain.go View File

@ -6,6 +6,7 @@ package dbquery
import ( import (
"database/sql" "database/sql"
"errors" "errors"
"github.com/jmoiron/sqlx"
"log" "log"
"strconv" "strconv"
"strings" "strings"
@ -331,7 +332,12 @@ func (this *TxQuery) BuildSelectSql() (map[string]interface{}, error) {
from := strconv.Itoa((this.page - 1) * this.page_size) from := strconv.Itoa((this.page - 1) * this.page_size)
offset := strconv.Itoa(this.page_size) offset := strconv.Itoa(this.page_size)
if from != "" && offset != "" { if from != "" && offset != "" {
sql = helper.StringJoin(sql, " limit ", from, " , ", offset)
if DB_PROVIDER == "PgsqlDb" {
sql = helper.StringJoin(sql, " limit ", offset, " OFFSET ", from)
} else {
sql = helper.StringJoin(sql, " limit ", from, " , ", offset)
}
} }
} }
@ -348,6 +354,9 @@ func (this *TxQuery) BuildSelectSql() (map[string]interface{}, error) {
if condition_len != len(this.value) { if condition_len != len(this.value) {
return nil, errors.New("参数错误，条件值错误") return nil, errors.New("参数错误，条件值错误")
} }
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
}
return map[string]interface{}{ return map[string]interface{}{
"sql": sql, "sql": sql,
"value": this.value, "value": this.value,
@ -365,7 +374,9 @@ func (this *TxQuery) GetTableInfo(table string) (map[string]interface{}, error)
"IS_NULLABLE", //是否为空 "IS_NULLABLE", //是否为空
} }
sql := "select `" + strings.Join(field, "`,`") + "` from information_schema.COLUMNS where table_name = ? and table_schema = ?" sql := "select `" + strings.Join(field, "`,`") + "` from information_schema.COLUMNS where table_name = ? and table_schema = ?"
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
}
stmtSql, err := this.tx.Prepare(sql) stmtSql, err := this.tx.Prepare(sql)
if err != nil { if err != nil {
return nil, err return nil, err
@ -471,7 +482,9 @@ func (this *TxQuery) UpdateStmt() error {
if condition_len != len(this.value) { if condition_len != len(this.value) {
return errors.New("参数错误，条件值错误") return errors.New("参数错误，条件值错误")
} }
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
}
stmt, err = this.tx.Prepare(sql) stmt, err = this.tx.Prepare(sql)
if err != nil { if err != nil {
@ -574,6 +587,9 @@ func (this *TxQuery) UpdateAllStmt() error {
if len(valSql) > 1 { if len(valSql) > 1 {
setText = " value " setText = " value "
} }
if DB_PROVIDER == "PgsqlDb" {
setText = " values "
}
sql = helper.StringJoin("insert into ", dbName, " (", strings.Join(this.data, " , "), ")", setText, strings.Join(valSql, ","), " ON DUPLICATE KEY UPDATE ", strings.Join(updSql, " , ")) sql = helper.StringJoin("insert into ", dbName, " (", strings.Join(this.data, " , "), ")", setText, strings.Join(valSql, ","), " ON DUPLICATE KEY UPDATE ", strings.Join(updSql, " , "))
if this.debug { if this.debug {
@ -589,6 +605,9 @@ func (this *TxQuery) UpdateAllStmt() error {
if conditionLen != len(this.value) { if conditionLen != len(this.value) {
return errors.New("参数错误，条件值数量不匹配") return errors.New("参数错误，条件值数量不匹配")
} }
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
}
stmt, err = this.tx.Prepare(sql) stmt, err = this.tx.Prepare(sql)
if err != nil { if err != nil {
@ -610,8 +629,28 @@ func (this *TxQuery) CreateStmt() error {
dbName := getTableName(this.dbname, this.table) dbName := getTableName(this.dbname, this.table)
var sql string var sql string
if DB_PROVIDER == "PgsqlDb" {
insert_data := []string{}
value_data := []string{}
for _, rv := range this.data {
dv := strings.Split(rv, "=")
if len(dv) < 2 {
return errors.New("参数错误，条件值错误,=号不存在")
}
if strings.Contains(rv, "?") {
insert_data = append(insert_data, dv[0])
value_data = append(value_data, "?")
} else {
insert_data = append(insert_data, dv[0])
value_data = append(value_data, dv[1])
}
sql = helper.StringJoin("insert into ", dbName, " set ", strings.Join(this.data, " , "))
}
sql = helper.StringJoin("insert into ", dbName, " ("+strings.Join(insert_data, " , ")+")", " VALUES ", "("+strings.Join(value_data, " , ")+")", " RETURNING id")
} else {
sql = helper.StringJoin("insert into ", dbName, " set ", strings.Join(this.data, " , "))
}
//sql = helper.StringJoin("insert into ", dbName, " set ", strings.Join(this.data, " , "))
if this.debug { if this.debug {
log.Println("insert sql:", sql, this.value) log.Println("insert sql:", sql, this.value)
@ -627,7 +666,9 @@ func (this *TxQuery) CreateStmt() error {
if condition_len != len(this.value) { if condition_len != len(this.value) {
return errors.New("参数错误，条件值错误") return errors.New("参数错误，条件值错误")
} }
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
}
stmt, err = this.tx.Prepare(sql) stmt, err = this.tx.Prepare(sql)
if err != nil { if err != nil {
@ -698,6 +739,9 @@ func (this *TxQuery) CreateAllStmt() error {
if len(valSql) > 1 { if len(valSql) > 1 {
setText = " value " setText = " value "
} }
if DB_PROVIDER == "PgsqlDb" {
setText = " values "
}
sql = helper.StringJoin("insert into ", dbName, " (", strings.Join(this.data, " , "), ")", setText, strings.Join(valSql, ",")) sql = helper.StringJoin("insert into ", dbName, " (", strings.Join(this.data, " , "), ")", setText, strings.Join(valSql, ","))
if len(this.value) == 0 { if len(this.value) == 0 {
return errors.New("参数错误，条件值错误") return errors.New("参数错误，条件值错误")
@ -716,7 +760,10 @@ func (this *TxQuery) CreateAllStmt() error {
if conditionLen != len(this.value) { if conditionLen != len(this.value) {
return errors.New("参数错误，条件值数量不匹配") return errors.New("参数错误，条件值数量不匹配")
} }
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
sql = helper.StringJoin(sql, " RETURNING id")
}
stmt, err = this.tx.Prepare(sql) stmt, err = this.tx.Prepare(sql)
if err != nil { if err != nil {
@ -762,7 +809,9 @@ func (this *TxQuery) DeleteStmt() error {
if condition_len != len(this.value) { if condition_len != len(this.value) {
return errors.New("参数错误，条件值错误") return errors.New("参数错误，条件值错误")
} }
if DB_PROVIDER == "PgsqlDb" {
sql = sqlx.Rebind(sqlx.DOLLAR, sql)
}
stmt, err = this.tx.Prepare(sql) stmt, err = this.tx.Prepare(sql)
if err != nil { if err != nil {


Write Preview
Loading…
Cancel
Save