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 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 != "" { 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{"=", ">", "<", "!=", "<=", ">="} }