mirror of
https://gitee.com/johng/gf.git
synced 2024-12-10 22:43:22 +08:00
526 lines
17 KiB
Go
526 lines
17 KiB
Go
// Copyright 2017 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
|
||
//
|
||
// This Source Code Form is subject to the terms of the MIT License.
|
||
// If a copy of the MIT was not distributed with this file,
|
||
// You can obtain one at https://gitee.com/johng/gf.
|
||
//
|
||
|
||
package gdb
|
||
|
||
import (
|
||
"fmt"
|
||
"errors"
|
||
"strings"
|
||
"reflect"
|
||
"database/sql"
|
||
"gitee.com/johng/gf/g/util/gstr"
|
||
"gitee.com/johng/gf/g/util/gconv"
|
||
"gitee.com/johng/gf/g/container/gring"
|
||
"gitee.com/johng/gf/g/os/gtime"
|
||
"time"
|
||
"gitee.com/johng/gf/g/os/glog"
|
||
)
|
||
|
||
const (
|
||
gDEFAULT_DEBUG_SQL_LENGTH = 1000 // 默认调试模式下记录的SQL条数
|
||
)
|
||
|
||
// 是否开启调试服务
|
||
func (db *Db) SetDebug(debug bool) {
|
||
db.debug.Set(debug)
|
||
if debug && db.sqls == nil {
|
||
db.sqls = gring.New(gDEFAULT_DEBUG_SQL_LENGTH)
|
||
}
|
||
}
|
||
|
||
// 获取已经执行的SQL列表(仅在debug=true时有效)
|
||
func (db *Db) GetQueriedSqls() []*Sql {
|
||
if db.sqls == nil {
|
||
return nil
|
||
}
|
||
sqls := make([]*Sql, 0)
|
||
db.sqls.Prev()
|
||
db.sqls.RLockIteratorPrev(func(value interface{}) bool {
|
||
if value == nil {
|
||
return false
|
||
}
|
||
sqls = append(sqls, value.(*Sql))
|
||
return true
|
||
})
|
||
return sqls
|
||
}
|
||
|
||
// 打印已经执行的SQL列表(仅在debug=true时有效)
|
||
func (db *Db) PrintQueriedSqls() {
|
||
sqls := db.GetQueriedSqls()
|
||
for k, v := range sqls {
|
||
fmt.Println(len(sqls) - k, ":")
|
||
fmt.Println(" Sql :", v.Sql)
|
||
fmt.Println(" Args :", v.Args)
|
||
fmt.Println(" Error:", v.Error)
|
||
fmt.Println(" Start:", gtime.NewFromTimeStamp(v.Start).Format("Y-m-d H:i:s.u"))
|
||
fmt.Println(" End :", gtime.NewFromTimeStamp(v.End).Format("Y-m-d H:i:s.u"))
|
||
fmt.Println(" Cost :", v.End - v.Start, "ms")
|
||
fmt.Println(" Func :", v.Func)
|
||
}
|
||
}
|
||
|
||
// 打印SQL对象(仅在debug=true时有效)
|
||
func (db *Db) printSql(v *Sql) {
|
||
s := fmt.Sprintf("%s, %v, %s, %s, %d ms, %s", v.Sql, v.Args,
|
||
gtime.NewFromTimeStamp(v.Start).Format("Y-m-d H:i:s.u"),
|
||
gtime.NewFromTimeStamp(v.End).Format("Y-m-d H:i:s.u"),
|
||
v.End - v.Start, v.Func,
|
||
)
|
||
if v.Error != nil {
|
||
s += "\nError: " + v.Error.Error()
|
||
glog.Backtrace(true, 1).Error(s)
|
||
} else {
|
||
glog.Debug(s)
|
||
}
|
||
}
|
||
|
||
// 关闭链接
|
||
func (db *Db) Close() error {
|
||
if db.master != nil {
|
||
if err := db.master.Close(); err == nil {
|
||
db.master = nil
|
||
} else {
|
||
return err
|
||
}
|
||
}
|
||
if db.slave != nil {
|
||
if err := db.slave.Close(); err == nil {
|
||
db.slave = nil
|
||
} else {
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 数据库sql查询操作,主要执行查询
|
||
func (db *Db) Query(query string, args ...interface{}) (*sql.Rows, error) {
|
||
var err error
|
||
var rows *sql.Rows
|
||
p := db.link.handleSqlBeforeExec(&query)
|
||
if db.debug.Val() {
|
||
militime1 := gtime.Millisecond()
|
||
rows, err = db.slave.Query(*p, args ...)
|
||
militime2 := gtime.Millisecond()
|
||
s := &Sql{
|
||
Sql : *p,
|
||
Args : args,
|
||
Error : err,
|
||
Start : militime1,
|
||
End : militime2,
|
||
Func : "DB:Query",
|
||
}
|
||
db.sqls.Put(s)
|
||
db.printSql(s)
|
||
} else {
|
||
rows, err = db.slave.Query(*p, args ...)
|
||
}
|
||
if err == nil {
|
||
return rows, nil
|
||
} else {
|
||
err = db.formatError(err, p, args...)
|
||
}
|
||
return nil, err
|
||
}
|
||
|
||
// 执行一条sql,并返回执行情况,主要用于非查询操作
|
||
func (db *Db) Exec(query string, args ...interface{}) (sql.Result, error) {
|
||
var err error
|
||
var result sql.Result
|
||
p := db.link.handleSqlBeforeExec(&query)
|
||
if db.debug.Val() {
|
||
militime1 := gtime.Millisecond()
|
||
result, err = db.master.Exec(*p, args ...)
|
||
militime2 := gtime.Millisecond()
|
||
s := &Sql{
|
||
Sql : *p,
|
||
Args : args,
|
||
Error : err,
|
||
Start : militime1,
|
||
End : militime2,
|
||
Func : "DB:Exec",
|
||
}
|
||
db.sqls.Put(s)
|
||
db.printSql(s)
|
||
} else {
|
||
result, err = db.master.Exec(*p, args ...)
|
||
}
|
||
return result, db.formatError(err, p, args...)
|
||
}
|
||
|
||
// 格式化错误信息
|
||
func (db *Db) formatError(err error, query *string, args ...interface{}) error {
|
||
if err != nil {
|
||
errstr := fmt.Sprintf("DB ERROR: %s\n", err.Error())
|
||
errstr += fmt.Sprintf("DB QUERY: %s\n", *query)
|
||
if len(args) > 0 {
|
||
errstr += fmt.Sprintf("DB PARAM: %v\n", args)
|
||
}
|
||
err = errors.New(errstr)
|
||
}
|
||
return err
|
||
}
|
||
|
||
|
||
// 数据库查询,获取查询结果集,以列表结构返回
|
||
func (db *Db) GetAll(query string, args ...interface{}) (Result, error) {
|
||
// 执行sql
|
||
rows, err := db.Query(query, args ...)
|
||
if err != nil || rows == nil {
|
||
return nil, err
|
||
}
|
||
// 列名称列表
|
||
columns, err := rows.Columns()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
// 返回结构组装
|
||
values := make([]sql.RawBytes, len(columns))
|
||
scanArgs := make([]interface{}, len(values))
|
||
records := make(Result, 0)
|
||
for i := range values {
|
||
scanArgs[i] = &values[i]
|
||
}
|
||
for rows.Next() {
|
||
err = rows.Scan(scanArgs...)
|
||
if err != nil {
|
||
return records, err
|
||
}
|
||
row := make(Record)
|
||
// 注意col字段是一个[]byte类型(slice类型本身是一个指针),多个记录循环时该变量指向的是同一个内存地址
|
||
for i, col := range values {
|
||
k := columns[i]
|
||
v := make([]byte, len(col))
|
||
copy(v, col)
|
||
row[k] = v
|
||
}
|
||
records = append(records, row)
|
||
}
|
||
return records, nil
|
||
}
|
||
|
||
// 数据库查询,获取查询结果记录,以关联数组结构返回
|
||
func (db *Db) GetOne(query string, args ...interface{}) (Record, error) {
|
||
list, err := db.GetAll(query, args ...)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if len(list) > 0 {
|
||
return list[0], nil
|
||
}
|
||
return nil, nil
|
||
}
|
||
|
||
// 数据库查询,获取查询结果记录,自动映射数据到给定的struct对象中
|
||
func (db *Db) GetStruct(obj interface{}, query string, args ...interface{}) error {
|
||
one, err := db.GetOne(query, args...)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return one.ToStruct(obj)
|
||
}
|
||
|
||
|
||
// 数据库查询,获取查询字段值
|
||
func (db *Db) GetValue(query string, args ...interface{}) (Value, error) {
|
||
one, err := db.GetOne(query, args ...)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
for _, v := range one {
|
||
return v, nil
|
||
}
|
||
return nil, nil
|
||
}
|
||
|
||
// 数据库查询,获取查询数量
|
||
func (db *Db) GetCount(query string, args ...interface{}) (int, error) {
|
||
val, err := db.GetValue(query, args ...)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
return gconv.Int(val), nil
|
||
}
|
||
|
||
// 数据表查询,其中tables可以是多个联表查询语句,这种查询方式较复杂,建议使用链式操作
|
||
func (db *Db) Select(tables, fields string, condition interface{}, groupBy, orderBy string, first, limit int, args ... interface{}) (Result, error) {
|
||
s := fmt.Sprintf("SELECT %s FROM %s ", fields, tables)
|
||
if condition != nil {
|
||
s += fmt.Sprintf("WHERE %s ", db.formatCondition(condition))
|
||
}
|
||
if len(groupBy) > 0 {
|
||
s += fmt.Sprintf("GROUP BY %s ", groupBy)
|
||
}
|
||
if len(orderBy) > 0 {
|
||
s += fmt.Sprintf("ORDER BY %s ", orderBy)
|
||
}
|
||
if limit > 0 {
|
||
s += fmt.Sprintf("LIMIT %d,%d ", first, limit)
|
||
}
|
||
return db.GetAll(s, args ... )
|
||
}
|
||
|
||
// sql预处理,执行完成后调用返回值sql.Stmt.Exec完成sql操作
|
||
// 记得调用sql.Stmt.Close关闭操作对象
|
||
func (db *Db) Prepare(query string) (*sql.Stmt, error) {
|
||
return db.master.Prepare(query)
|
||
}
|
||
|
||
// ping一下,判断或保持数据库链接(master)
|
||
func (db *Db) PingMaster() error {
|
||
err := db.master.Ping()
|
||
return err
|
||
}
|
||
|
||
// ping一下,判断或保持数据库链接(slave)
|
||
func (db *Db) PingSlave() error {
|
||
err := db.slave.Ping()
|
||
return err
|
||
}
|
||
|
||
// 设置数据库连接池中空闲链接的大小
|
||
func (db *Db) SetMaxIdleConns(n int) {
|
||
db.master.SetMaxIdleConns(n)
|
||
// 比较的是指向的变量地址
|
||
if db.master != db.slave {
|
||
db.slave.SetMaxIdleConns(n)
|
||
}
|
||
}
|
||
|
||
// 设置数据库连接池最大打开的链接数量
|
||
func (db *Db) SetMaxOpenConns(n int) {
|
||
db.master.SetMaxOpenConns(n)
|
||
// 比较的是指向的变量地址
|
||
if db.master != db.slave {
|
||
db.slave.SetMaxOpenConns(n)
|
||
}
|
||
}
|
||
|
||
// 设置数据库连接可重复利用的时间,超过该时间则被关闭废弃
|
||
// 如果 d <= 0 表示该链接会一直重复利用
|
||
func (db *Db) SetConnMaxLifetime(d time.Duration) {
|
||
db.master.SetConnMaxLifetime(d)
|
||
// 比较的是指向的变量地址
|
||
if db.master != db.slave {
|
||
db.slave.SetConnMaxLifetime(d)
|
||
}
|
||
}
|
||
|
||
// 事务操作,开启,会返回一个底层的事务操作对象链接如需要嵌套事务,那么可以使用该对象,否则请忽略
|
||
func (db *Db) Begin() (*Tx, error) {
|
||
if tx, err := db.master.Begin(); err == nil {
|
||
return &Tx {
|
||
db : db,
|
||
tx : tx,
|
||
}, nil
|
||
} else {
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
// 根据insert选项获得操作名称
|
||
func (db *Db) getInsertOperationByOption(option uint8) string {
|
||
oper := "INSERT"
|
||
switch option {
|
||
case OPTION_REPLACE:
|
||
oper = "REPLACE"
|
||
case OPTION_SAVE:
|
||
case OPTION_IGNORE:
|
||
oper = "INSERT IGNORE"
|
||
}
|
||
return oper
|
||
}
|
||
|
||
// insert、replace, save, ignore操作
|
||
// 0: insert: 仅仅执行写入操作,如果存在冲突的主键或者唯一索引,那么报错返回
|
||
// 1: replace: 如果数据存在(主键或者唯一索引),那么删除后重新写入一条
|
||
// 2: save: 如果数据存在(主键或者唯一索引),那么更新,否则写入一条新数据
|
||
// 3: ignore: 如果数据存在(主键或者唯一索引),那么什么也不做
|
||
func (db *Db) insert(table string, data Map, option uint8) (sql.Result, error) {
|
||
var fields []string
|
||
var values []string
|
||
var params []interface{}
|
||
for k, v := range data {
|
||
fields = append(fields, db.charl + k + db.charr)
|
||
values = append(values, "?")
|
||
params = append(params, gconv.String(v))
|
||
}
|
||
operation := db.getInsertOperationByOption(option)
|
||
updatestr := ""
|
||
if option == OPTION_SAVE {
|
||
var updates []string
|
||
for k, _ := range data {
|
||
updates = append(updates,
|
||
fmt.Sprintf("%s%s%s=VALUES(%s%s%s)",
|
||
db.charl, k, db.charr,
|
||
db.charl, k, db.charr,
|
||
),
|
||
)
|
||
}
|
||
updatestr = fmt.Sprintf("ON DUPLICATE KEY UPDATE %s", strings.Join(updates, ","))
|
||
}
|
||
return db.Exec(
|
||
fmt.Sprintf("%s INTO %s%s%s(%s) VALUES(%s) %s",
|
||
operation, db.charl, table, db.charr, strings.Join(fields, ","),
|
||
strings.Join(values, ","),
|
||
updatestr),
|
||
params...
|
||
)
|
||
}
|
||
|
||
// CURD操作:单条数据写入, 仅仅执行写入操作,如果存在冲突的主键或者唯一索引,那么报错返回
|
||
func (db *Db) Insert(table string, data Map) (sql.Result, error) {
|
||
return db.insert(table, data, OPTION_INSERT)
|
||
}
|
||
|
||
// CURD操作:单条数据写入, 如果数据存在(主键或者唯一索引),那么删除后重新写入一条
|
||
func (db *Db) Replace(table string, data Map) (sql.Result, error) {
|
||
return db.insert(table, data, OPTION_REPLACE)
|
||
}
|
||
|
||
// CURD操作:单条数据写入, 如果数据存在(主键或者唯一索引),那么更新,否则写入一条新数据
|
||
func (db *Db) Save(table string, data Map) (sql.Result, error) {
|
||
return db.insert(table, data, OPTION_SAVE)
|
||
}
|
||
|
||
// 批量写入数据
|
||
func (db *Db) batchInsert(table string, list List, batch int, option uint8) (sql.Result, error) {
|
||
var keys []string
|
||
var values []string
|
||
var bvalues []string
|
||
var params []interface{}
|
||
var result sql.Result
|
||
var size = len(list)
|
||
// 判断长度
|
||
if size < 1 {
|
||
return result, errors.New("empty data list")
|
||
}
|
||
// 首先获取字段名称及记录长度
|
||
for k, _ := range list[0] {
|
||
keys = append(keys, k)
|
||
values = append(values, "?")
|
||
}
|
||
keyStr := db.charl + strings.Join(keys, db.charl + "," + db.charr) + db.charr
|
||
valueHolderStr := "(" + strings.Join(values, ",") + ")"
|
||
// 操作判断
|
||
operation := db.getInsertOperationByOption(option)
|
||
updatestr := ""
|
||
if option == OPTION_SAVE {
|
||
var updates []string
|
||
for _, k := range keys {
|
||
updates = append(updates,
|
||
fmt.Sprintf("%s%s%s=VALUES(%s%s%s)",
|
||
db.charl, k, db.charr,
|
||
db.charl, k, db.charr,
|
||
),
|
||
)
|
||
}
|
||
updatestr = fmt.Sprintf(" ON DUPLICATE KEY UPDATE %s", strings.Join(updates, ","))
|
||
}
|
||
// 构造批量写入数据格式(注意map的遍历是无序的)
|
||
for i := 0; i < size; i++ {
|
||
for _, k := range keys {
|
||
params = append(params, gconv.String(list[i][k]))
|
||
}
|
||
bvalues = append(bvalues, valueHolderStr)
|
||
if len(bvalues) == batch {
|
||
r, err := db.Exec(fmt.Sprintf("%s INTO %s%s%s(%s) VALUES%s %s",
|
||
operation, db.charl, table, db.charr, keyStr, strings.Join(bvalues, ","),
|
||
updatestr),
|
||
params...)
|
||
if err != nil {
|
||
return result, err
|
||
}
|
||
result = r
|
||
params = params[:0]
|
||
bvalues = bvalues[:0]
|
||
}
|
||
}
|
||
// 处理最后不构成指定批量的数据
|
||
if len(bvalues) > 0 {
|
||
r, err := db.Exec(fmt.Sprintf("%s INTO %s%s%s(%s) VALUES%s %s",
|
||
operation, db.charl, table, db.charr, keyStr, strings.Join(bvalues, ","),
|
||
updatestr),
|
||
params...)
|
||
if err != nil {
|
||
return result, err
|
||
}
|
||
result = r
|
||
}
|
||
return result, nil
|
||
}
|
||
|
||
// CURD操作:批量数据指定批次量写入
|
||
func (db *Db) BatchInsert(table string, list List, batch int) (sql.Result, error) {
|
||
return db.batchInsert(table, list, batch, OPTION_INSERT)
|
||
}
|
||
|
||
// CURD操作:批量数据指定批次量写入, 如果数据存在(主键或者唯一索引),那么删除后重新写入一条
|
||
func (db *Db) BatchReplace(table string, list List, batch int) (sql.Result, error) {
|
||
return db.batchInsert(table, list, batch, OPTION_REPLACE)
|
||
}
|
||
|
||
// CURD操作:批量数据指定批次量写入, 如果数据存在(主键或者唯一索引),那么更新,否则写入一条新数据
|
||
func (db *Db) BatchSave(table string, list List, batch int) (sql.Result, error) {
|
||
return db.batchInsert(table, list, batch, OPTION_SAVE)
|
||
}
|
||
|
||
// CURD操作:数据更新,统一采用sql预处理
|
||
// data参数支持字符串或者关联数组类型,内部会自行做判断处理
|
||
func (db *Db) Update(table string, data interface{}, condition interface{}, args ...interface{}) (sql.Result, error) {
|
||
var params []interface{}
|
||
var updates string
|
||
refValue := reflect.ValueOf(data)
|
||
if refValue.Kind() == reflect.Map {
|
||
var fields []string
|
||
keys := refValue.MapKeys()
|
||
for _, k := range keys {
|
||
fields = append(fields, fmt.Sprintf("%s%s%s=?", db.charl, k, db.charr))
|
||
params = append(params, gconv.String(refValue.MapIndex(k).Interface()))
|
||
}
|
||
updates = strings.Join(fields, ",")
|
||
} else {
|
||
updates = gconv.String(data)
|
||
}
|
||
for _, v := range args {
|
||
params = append(params, gconv.String(v))
|
||
}
|
||
return db.Exec(fmt.Sprintf("UPDATE %s%s%s SET %s WHERE %s", db.charl, table, db.charr, updates, db.formatCondition(condition)), params...)
|
||
}
|
||
|
||
// CURD操作:删除数据
|
||
func (db *Db) Delete(table string, condition interface{}, args ...interface{}) (sql.Result, error) {
|
||
return db.Exec(fmt.Sprintf("DELETE FROM %s%s%s WHERE %s", db.charl, table, db.charr, db.formatCondition(condition)), args...)
|
||
}
|
||
|
||
// 格式化SQL查询条件
|
||
func (db *Db) formatCondition(condition interface{}) (where string) {
|
||
if reflect.ValueOf(condition).Kind() == reflect.Map {
|
||
ks := reflect.ValueOf(condition).MapKeys()
|
||
vs := reflect.ValueOf(condition)
|
||
for _, k := range ks {
|
||
key := gconv.String(k.Interface())
|
||
value := gconv.String(vs.MapIndex(k).Interface())
|
||
isNum := gstr.IsNumeric(value)
|
||
if len(where) > 0 {
|
||
where += " AND "
|
||
}
|
||
if isNum || value == "?" {
|
||
where += key + "=" + value
|
||
} else {
|
||
where += key + "='" + value + "'"
|
||
}
|
||
}
|
||
} else {
|
||
where += gconv.String(condition)
|
||
}
|
||
return
|
||
}
|
||
|