qq_bot/sqlite3/db.go

37 lines
469 B
Go

package sqlite3
import (
"github.com/jmoiron/sqlx"
_ "github.com/mattn/go-sqlite3"
)
var sqlDb *sqlx.DB
func InitDB() {
if sqlDb == nil {
sqlDb, _ = sqlx.Open("sqlite3", "./data.db")
}
}
func TryCreateTable(query string) error {
InitDB()
_, err := sqlDb.Exec(query)
if err != nil {
return err
}
return nil
}
func GetSqlDB() *sqlx.DB {
return sqlDb
}
func GetSqlTran() (*sqlx.Tx, error) {
return sqlDb.Beginx()
}
func CloseDB() {
sqlDb.Close()
}