7 Commits

8 changed files with 94 additions and 17 deletions
Split View
  1. +54
    -10
      chain.go
  2. +0
    -0
      conn.go
  3. +2
    -0
      db.go
  4. +0
    -0
      db_test.go
  5. +0
    -0
      prepare.go
  6. +7
    -2
      sqlserver.go
  7. +3
    -3
      transaction.go
  8. +28
    -2
      transaction_chain.go

+ 54
- 10
chain.go View File

@ -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,12 +293,23 @@ func (this *Query) BuildSelectSql() (map[string]interface{}, error) {
} else {
title = "*"
}
if DB_PROVIDER == "TencentDB" {
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 {
sql = helper.StringJoin("select ", title)
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)
}
@ -279,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 {
@ -315,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
@ -358,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
}
@ -798,7 +842,7 @@ func (this *Query) Update() (int64, error) {
return StmtForUpdateExec(this.stmt, this.value)
}
//批量更新
// 批量更新
func (this *Query) UpdateAll() (int64, error) {
err := this.UpdateAllStmt()


+ 0
- 0
conn.go View File


+ 2
- 0
db.go View File

@ -406,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
}
@ -436,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


+ 0
- 0
db_test.go View File


+ 0
- 0
prepare.go View File


+ 7
- 2
sqlserver.go View File

@ -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)


+ 3
- 3
transaction.go View File

@ -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("参数错误,没有要写入的数据")
@ -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("参数错误,没有删除条件")


+ 28
- 2
transaction_chain.go View File

@ -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", //字段名


Loading…
Cancel
Save