|
- package db
-
- import (
- "code.fnuoos.com/go_rely_warehouse/zyos_go_pay.git/md"
- "database/sql"
- "fmt"
- _ "github.com/go-sql-driver/mysql" //必须导入mysql驱动,否则会panic
- "os"
- "time"
- "xorm.io/xorm"
- "xorm.io/xorm/log"
- )
-
- var Db *xorm.Engine
-
- //根据DB配置文件初始化数据库
- func InitDB(c *md.DBCfg) error {
- var (
- err error
- f *os.File
- )
- //创建Orm引擎
- if Db, err = xorm.NewEngine("mysql", fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4", c.User, c.Psw, c.Host, c.Name)); err != nil {
- return err
- }
- Db.SetConnMaxLifetime(c.MaxLifetime * time.Second) //设置最长连接时间
- Db.SetMaxOpenConns(c.MaxOpenConns) //设置最大打开连接数
- Db.SetMaxIdleConns(c.MaxIdleConns) //设置连接池的空闲数大小
- if err = Db.Ping(); err != nil { //尝试ping数据库
- return err
- }
- if c.ShowLog { //根据配置文件设置日志
- Db.ShowSQL(true) //设置是否打印sql
- Db.Logger().SetLevel(0) //设置日志等级
- //修改日志文件存放路径文件名是%s.log
- path := fmt.Sprintf(c.Path, c.Name)
- f, err = os.OpenFile(path, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0777)
- if err != nil {
- os.RemoveAll(c.Path)
- if f, err = os.OpenFile(c.Path, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0777); err != nil {
- return err
- }
- }
- logger := log.NewSimpleLogger(f)
- logger.ShowSQL(true)
- Db.SetLogger(logger)
- }
- return nil
- }
-
- /********************************************* 公用方法 *********************************************/
-
- // QueryNativeString 查询原生sql
- func QueryNativeString(Db *xorm.Engine, sql string, args ...interface{}) ([]map[string]string, error) {
- results, err := Db.SQL(sql, args...).QueryString()
- return results, err
- }
-
- // UpdateComm common update
- func UpdateComm(Db *xorm.Engine, id interface{}, model interface{}) (int64, error) {
- row, err := Db.ID(id).Update(model)
- return row, err
- }
-
- // InsertComm common insert
- func InsertComm(Db *xorm.Engine, model interface{}) (int64, error) {
- row, err := Db.InsertOne(model)
- return row, err
- }
-
- // InsertCommWithSession common insert
- func InsertCommWithSession(session *xorm.Session, model interface{}) (int64, error) {
- row, err := session.InsertOne(model)
- return row, err
- }
-
- // ExecuteOriginalSql 执行原生sql
- func ExecuteOriginalSql(session *xorm.Session, sql string) (sql.Result, error) {
- result, err := session.Exec(sql)
- if err != nil {
- return nil, err
- }
- return result, nil
- }
|