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.
153 lines
3.4 KiB
153 lines
3.4 KiB
package dbquery
|
|
|
|
import (
|
|
"database/sql"
|
|
"log"
|
|
|
|
"errors"
|
|
"strings"
|
|
"time"
|
|
|
|
"git.tetele.net/tgo/helper"
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
)
|
|
|
|
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...")
|
|
var dbConnErr error
|
|
if DBHOST != "" && DBUSER != "" && DBPWD != "" && DBPORT != "" { //&& DBNAME != ""
|
|
|
|
for i := 0; i < 10; i++ {
|
|
DB, dbConnErr = sql.Open("mysql", DBUSER+":"+DBPWD+"@tcp("+DBHOST+":"+DBPORT+")/"+DBNAME+"?charset=utf8mb4")
|
|
if dbConnErr != nil {
|
|
log.Println("ERROR", "can not connect to Database, ", dbConnErr)
|
|
time.Sleep(time.Second * 5)
|
|
} else {
|
|
if len(conns) > 0 {
|
|
DB.SetMaxOpenConns(conns[0]) //用于设置最大打开的连接数，默认值为0表示不限制
|
|
} else {
|
|
DB.SetMaxOpenConns(200) //默认值为0表示不限制
|
|
}
|
|
if len(conns) > 1 {
|
|
DB.SetMaxIdleConns(conns[1]) //用于设置闲置的连接数
|
|
} else {
|
|
DB.SetMaxIdleConns(50)
|
|
}
|
|
|
|
DB.Ping()
|
|
|
|
log.Println("database connected")
|
|
DB.SetConnMaxLifetime(time.Minute * 2)
|
|
break
|
|
}
|
|
}
|
|
} else {
|
|
return errors.New("db connection params errors")
|
|
}
|
|
return dbConnErr
|
|
}
|
|
|
|
func CloseConn() error {
|
|
return DB.Close()
|
|
}
|
|
|
|
func ConnectSlaver(DBHOST, DBUSER_SLAVER, DBPWD_SLAVER, DBNAME, DBPORT string, conns ...int) error {
|
|
|
|
log.Println("database connectting with slaver...")
|
|
var dbConnErr error
|
|
if DBHOST != "" && DBUSER_SLAVER != "" && DBPWD_SLAVER != "" && DBPORT != "" { //&& DBNAME != ""
|
|
|
|
for i := 0; i < 10; i++ {
|
|
SLAVER_DB, dbConnErr = sql.Open("mysql", DBUSER_SLAVER+":"+DBPWD_SLAVER+"@tcp("+DBHOST+":"+DBPORT+")/"+DBNAME+"?charset=utf8mb4")
|
|
if dbConnErr != nil {
|
|
log.Println("ERROR", "can not connect to Database, ", dbConnErr)
|
|
time.Sleep(time.Second * 5)
|
|
} else {
|
|
if len(conns) > 0 {
|
|
SLAVER_DB.SetMaxOpenConns(conns[0]) //用于设置最大打开的连接数，默认值为0表示不限制
|
|
} else {
|
|
SLAVER_DB.SetMaxOpenConns(200) //默认值为0表示不限制
|
|
}
|
|
if len(conns) > 1 {
|
|
SLAVER_DB.SetMaxIdleConns(conns[1]) //用于设置闲置的连接数
|
|
} else {
|
|
SLAVER_DB.SetMaxIdleConns(50)
|
|
}
|
|
|
|
SLAVER_DB.Ping()
|
|
|
|
log.Println("database connected")
|
|
SLAVER_DB.SetConnMaxLifetime(time.Minute * 2)
|
|
break
|
|
}
|
|
}
|
|
} else {
|
|
return errors.New("db connection params errors")
|
|
}
|
|
return dbConnErr
|
|
}
|
|
|
|
func CloseSlaverConn() error {
|
|
return SLAVER_DB.Close()
|
|
}
|
|
|
|
/**
|
|
* 检测表名
|
|
*/
|
|
func getTableName(dbName, table string, dbtype ...string) string {
|
|
|
|
var db_type string = "mysql"
|
|
|
|
if len(dbtype) > 0 {
|
|
if dbtype[0] != "" {
|
|
db_type = dbtype[0]
|
|
}
|
|
}
|
|
|
|
var ret string
|
|
|
|
switch db_type {
|
|
|
|
case "mysql":
|
|
|
|
if strings.Contains(table, ".") {
|
|
ret = table
|
|
}
|
|
if dbName != "" {
|
|
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
|
|
}
|
|
|
|
case "mssql":
|
|
ret = helper.StringJoin(dbName, ".", table)
|
|
|
|
}
|
|
return ret
|
|
}
|
|
|
|
func GetDbTableName(dbName, table string) string {
|
|
return getTableName(dbName, table)
|
|
}
|
|
|
|
func judg() []string {
|
|
return []string{"=", ">", "<", "!=", "<=", ">="}
|
|
}
|