You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
399 lines
7.8 KiB
399 lines
7.8 KiB
package dbquery
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"git.tetele.net/tgo/helper"
|
|
)
|
|
|
|
var stmt *sql.Stmt
|
|
var err error
|
|
|
|
type Query struct {
|
|
dbname string
|
|
table string
|
|
alias string
|
|
title string
|
|
where []string
|
|
where_or []string
|
|
join [][]string //[["tablea as a","a.id=b.id","left"]]
|
|
data []string
|
|
value []interface{}
|
|
orderby string
|
|
page int
|
|
page_size int
|
|
stmt *sql.Stmt
|
|
}
|
|
|
|
func (this *Query) Db(dbname string) *Query {
|
|
this.dbname = dbname
|
|
return this
|
|
}
|
|
|
|
func (this *Query) Table(tablename string) *Query {
|
|
this.table = tablename
|
|
return this
|
|
}
|
|
func (this *Query) Alias(tablename string) *Query {
|
|
this.alias = tablename
|
|
return this
|
|
}
|
|
|
|
func (this *Query) Title(title string) *Query {
|
|
this.title = title
|
|
return this
|
|
}
|
|
func (this *Query) Page(page int) *Query {
|
|
this.page = page
|
|
return this
|
|
}
|
|
func (this *Query) PageSize(page_num int) *Query {
|
|
this.page_size = page_num
|
|
return this
|
|
}
|
|
|
|
func (this *Query) Orderby(orderby string) *Query {
|
|
this.orderby = orderby
|
|
return this
|
|
}
|
|
func (this *Query) Where(where string) *Query {
|
|
this.where = append(this.where, where)
|
|
return this
|
|
}
|
|
func (this *Query) Wheres(wheres []string) *Query {
|
|
this.where = append(this.where, wheres...)
|
|
return this
|
|
}
|
|
func (this *Query) WhereOr(where string) *Query {
|
|
this.where_or = append(this.where_or, where)
|
|
return this
|
|
}
|
|
func (this *Query) Value(value interface{}) *Query {
|
|
this.value = append(this.value, value)
|
|
return this
|
|
}
|
|
func (this *Query) Values(values []interface{}) *Query {
|
|
this.value = append(this.value, values...)
|
|
return this
|
|
}
|
|
func (this *Query) Join(join []string) *Query {
|
|
this.join = append(this.join, join)
|
|
return this
|
|
}
|
|
func (this *Query) Data(data string) *Query {
|
|
this.data = append(this.data, data)
|
|
return this
|
|
}
|
|
func (this *Query) Datas(datas []string) *Query {
|
|
this.data = append(this.data, datas...)
|
|
return this
|
|
}
|
|
|
|
// func (this *Query) Insert(where string) *Query {
|
|
// this.insert = append(this.insert, where)
|
|
// return this
|
|
// }
|
|
|
|
// 拼查询sql
|
|
func (this *Query) QueryStmt() error {
|
|
|
|
if this.dbname == "" && this.table == "" {
|
|
return errors.New("参数错误，没有数据表")
|
|
}
|
|
// if len(this.where)+len(this.where_or) < len(this.value) {
|
|
// return errors.New("参数错误，条件值错误")
|
|
// }
|
|
|
|
table := getTableName(this.dbname, this.table)
|
|
|
|
// var err error
|
|
|
|
var sql, title string
|
|
|
|
if this.title != "" {
|
|
title = this.title
|
|
} else {
|
|
title = "*"
|
|
}
|
|
sql = helper.StringJoin("select ", title)
|
|
|
|
if this.alias != "" {
|
|
table = helper.StringJoin(table, " as ", this.alias)
|
|
}
|
|
|
|
sql = helper.StringJoin(sql, " from ", table)
|
|
|
|
if len(this.join) > 0 {
|
|
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]), " on ", joinitem[1])
|
|
} else { //默认左连接
|
|
sql = helper.StringJoin(sql, " left join ", getTableName(this.dbname, joinitem[0]), " on ", joinitem[1])
|
|
}
|
|
}
|
|
}
|
|
if len(this.where) > 0 || len(this.where_or) > 0 {
|
|
sql = helper.StringJoin(sql, " where ")
|
|
}
|
|
if len(this.where) > 0 {
|
|
sql = helper.StringJoin(sql, " (", strings.Join(this.where, " and "), " ) ")
|
|
}
|
|
if len(this.where_or) > 0 {
|
|
if len(this.where) > 0 {
|
|
sql = helper.StringJoin(sql, " or ", strings.Join(this.where_or, " or "))
|
|
} else {
|
|
sql = helper.StringJoin(sql, strings.Join(this.where_or, " or "))
|
|
}
|
|
}
|
|
|
|
if this.orderby != "" {
|
|
sql = helper.StringJoin(sql, " order by ", this.orderby)
|
|
}
|
|
|
|
if this.page > 0 || this.page_size > 0 {
|
|
|
|
if this.page < 0 {
|
|
this.page = 0
|
|
}
|
|
if this.page_size < 1 {
|
|
this.page_size = 10
|
|
}
|
|
from := strconv.Itoa(this.page * this.page_size)
|
|
offset := strconv.Itoa(this.page_size)
|
|
if from != "" && offset != "" {
|
|
sql = helper.StringJoin(sql, " limit ", from, " , ", offset)
|
|
}
|
|
}
|
|
// log.Println(sql)
|
|
condition_len := 0 //所有条件数
|
|
for _, ch2 := range sql {
|
|
if string(ch2) == "?" {
|
|
condition_len++
|
|
}
|
|
}
|
|
if condition_len != len(this.value) {
|
|
return errors.New("参数错误，条件值错误")
|
|
}
|
|
|
|
stmt, err = DB.Prepare(sql)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
this.stmt = stmt
|
|
|
|
return nil
|
|
}
|
|
|
|
// 拼更新sql
|
|
func (this *Query) UpdateStmt() error {
|
|
|
|
if this.dbname == "" && this.table == "" {
|
|
return errors.New("参数错误，没有数据表")
|
|
}
|
|
if len(this.where) < 1 {
|
|
return errors.New("参数错误，缺少条件")
|
|
}
|
|
|
|
dbName := getTableName(this.dbname, this.table)
|
|
|
|
// var stmt *sql.Stmt
|
|
// var err error
|
|
|
|
var sql string
|
|
|
|
sql = helper.StringJoin("update ", dbName, " set ", strings.Join(this.data, " , "))
|
|
|
|
sql = helper.StringJoin(sql, " where ", strings.Join(this.where, " and "))
|
|
|
|
condition_len := 0 //所有条件数
|
|
|
|
for _, ch2 := range sql {
|
|
if string(ch2) == "?" {
|
|
condition_len++
|
|
}
|
|
}
|
|
if condition_len != len(this.value) {
|
|
return errors.New("参数错误，条件值错误")
|
|
}
|
|
|
|
stmt, err = DB.Prepare(sql)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
this.stmt = stmt
|
|
|
|
return nil
|
|
}
|
|
|
|
// 拼插入sql
|
|
func (this *Query) CreateStmt() error {
|
|
|
|
if this.dbname == "" && this.table == "" {
|
|
return errors.New("参数错误，没有数据表")
|
|
}
|
|
|
|
dbName := getTableName(this.dbname, this.table)
|
|
|
|
// var stmt *sql.Stmt
|
|
// var err error
|
|
|
|
var sql string
|
|
|
|
sql = helper.StringJoin("insert into ", dbName, " set ", strings.Join(this.data, " , "))
|
|
|
|
condition_len := 0 //所有条件数
|
|
|
|
for _, ch2 := range sql {
|
|
if string(ch2) == "?" {
|
|
condition_len++
|
|
}
|
|
}
|
|
if condition_len != len(this.value) {
|
|
return errors.New("参数错误，条件值错误")
|
|
}
|
|
|
|
stmt, err = DB.Prepare(sql)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
this.stmt = stmt
|
|
|
|
return nil
|
|
}
|
|
|
|
// 拼删除sql
|
|
func (this *Query) DeleteStmt() error {
|
|
|
|
if this.dbname == "" && this.table == "" {
|
|
return errors.New("参数错误，没有数据表")
|
|
}
|
|
if len(this.where) < 1 {
|
|
return errors.New("参数错误，缺少条件")
|
|
}
|
|
// if len(this.where) != len(this.value) {
|
|
// return errors.New("参数错误，条件值错误")
|
|
// }
|
|
|
|
dbName := getTableName(this.dbname, this.table)
|
|
|
|
// var stmt *sql.Stmt
|
|
// var err error
|
|
|
|
var sql string
|
|
|
|
sql = helper.StringJoin("delete from ", dbName, " where ", strings.Join(this.where, " and "))
|
|
|
|
if this.page_size > 0 {
|
|
sql = helper.StringJoin(sql, " limit ", strconv.Itoa(this.page_size))
|
|
}
|
|
|
|
condition_len := 0 //所有条件数
|
|
|
|
for _, ch2 := range sql {
|
|
if string(ch2) == "?" {
|
|
condition_len++
|
|
}
|
|
}
|
|
if condition_len != len(this.value) {
|
|
return errors.New("参数错误，条件值错误")
|
|
}
|
|
|
|
stmt, err = DB.Prepare(sql)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
this.stmt = stmt
|
|
|
|
return nil
|
|
}
|
|
|
|
/**
|
|
* 执行查询列表
|
|
* return list error
|
|
*/
|
|
func (this *Query) Select() ([]map[string]string, error) {
|
|
|
|
err := this.QueryStmt()
|
|
if err != nil {
|
|
return []map[string]string{}, err
|
|
}
|
|
|
|
if this.stmt == nil {
|
|
return []map[string]string{}, errors.New("缺少必要参数")
|
|
}
|
|
|
|
return StmtForQueryList(this.stmt, this.value)
|
|
}
|
|
|
|
/**
|
|
* 执行查询一条数据
|
|
* return row error
|
|
*/
|
|
func (this *Query) Find() (map[string]string, error) {
|
|
|
|
err := this.QueryStmt()
|
|
if err != nil {
|
|
return map[string]string{}, err
|
|
}
|
|
|
|
if this.stmt == nil {
|
|
return nil, errors.New("缺少必要参数")
|
|
}
|
|
return StmtForQueryRow(this.stmt, this.value)
|
|
}
|
|
|
|
/**
|
|
* 执行更新
|
|
* return is_updated error
|
|
*/
|
|
func (this *Query) Update() (int64, error) {
|
|
|
|
err := this.UpdateStmt()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return StmtForUpdateExec(this.stmt, this.value)
|
|
}
|
|
|
|
/**
|
|
* 执行删除
|
|
* return is_delete error
|
|
*/
|
|
func (this *Query) Delete() (int64, error) {
|
|
|
|
err := this.DeleteStmt()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return StmtForUpdateExec(this.stmt, this.value)
|
|
}
|
|
|
|
/**
|
|
* 执行写入
|
|
* return is_insert error
|
|
*/
|
|
func (this *Query) Create() (int64, error) {
|
|
|
|
err := this.CreateStmt()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return StmtForInsertExec(this.stmt, this.value)
|
|
}
|