51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package dbpool
|
||
|
||
import (
|
||
"database/sql"
|
||
"log"
|
||
|
||
_ "github.com/go-sql-driver/mysql"
|
||
)
|
||
|
||
var db *sql.DB
|
||
|
||
func init() {
|
||
db, _ = sql.Open("mysql", "root:Lc753951!!@tcp(fanyu.online:53306)/douy?charset=utf8&parseTime=True&loc=Local")
|
||
db.SetMaxOpenConns(20)
|
||
db.SetMaxIdleConns(10)
|
||
db.SetConnMaxLifetime(100)
|
||
err := db.Ping()
|
||
if err != nil {
|
||
log.Println("database init failed, err: ", err)
|
||
return
|
||
}
|
||
log.Println("mysql conn pool has initiated.")
|
||
}
|
||
|
||
// QueryRows 是一个通用的查询函数，它执行SQL查询并返回单个结果
|
||
func QueryRows(query string, args ...any) *sql.Row {
|
||
return db.QueryRow(query, args...)
|
||
}
|
||
|
||
// Query 是一个通用的查询函数，它执行SQL查询并返回结果集
|
||
func Query(query string, args ...any) (*sql.Rows, error) {
|
||
|
||
rows, err := db.Query(query, args...)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return rows, nil
|
||
}
|
||
|
||
// ExecuteSQL 是一个通用的插入函数，它执行SQL插入并返回结果集
|
||
func ExecuteSQL(query string, args ...any) (sql.Result, error) {
|
||
|
||
result, err := db.Exec(query, args...)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return result, nil
|
||
}
|