gocron/models/model.go

87 lines
2.0 KiB
Go
Raw Normal View History

2017-03-10 09:24:06 +00:00
package models
import (
2017-04-02 02:38:49 +00:00
"fmt"
_ "github.com/go-sql-driver/mysql"
"github.com/go-xorm/core"
"github.com/go-xorm/xorm"
"gopkg.in/macaron.v1"
"strings"
2017-04-03 02:20:08 +00:00
"time"
2017-03-10 09:24:06 +00:00
)
type Status int8
type CommonMap map[string]interface{}
2017-03-14 06:31:46 +00:00
var Db *xorm.Engine
2017-03-10 09:24:06 +00:00
const (
2017-04-02 02:38:49 +00:00
Disabled Status = 0 // 禁用
Failure Status = 0 // 失败
Enabled Status = 1 // 启用
Running Status = 1 // 运行中
Finish Status = 2 // 完成
2017-03-10 09:24:06 +00:00
)
const (
2017-04-02 02:38:49 +00:00
Page = 1 // 当前页数
PageSize = 20 // 每页多少条数据
MaxPageSize = 100000 // 每次最多取多少条
2017-03-10 09:24:06 +00:00
)
2017-03-10 10:08:45 +00:00
// 创建Db
func CreateDb(config map[string]string) *xorm.Engine {
2017-04-02 02:38:49 +00:00
dsn := getDbEngineDSN(config["engine"], config)
engine, err := xorm.NewEngine(config["engine"], dsn)
if err != nil {
panic(err)
}
if config["prefix"] != "" {
// 设置表前缀
mapper := core.NewPrefixMapper(core.SnakeMapper{}, config["prefix"])
engine.SetTableMapper(mapper)
}
2017-04-03 02:20:08 +00:00
// 本地环境开启日志
2017-04-02 02:38:49 +00:00
if macaron.Env == macaron.DEV {
engine.ShowSQL(true)
engine.Logger().SetLevel(core.LOG_DEBUG)
}
2017-03-10 09:24:06 +00:00
2017-04-03 02:20:08 +00:00
go keepDbAlived(engine)
2017-04-02 02:38:49 +00:00
return engine
2017-03-10 09:24:06 +00:00
}
2017-04-01 09:04:40 +00:00
// 创建临时数据库连接
func CreateTmpDb(config map[string]string) (*xorm.Engine, error) {
dsn := getDbEngineDSN(config["engine"], config)
return xorm.NewEngine(config["engine"], dsn)
}
2017-04-01 09:04:40 +00:00
// 获取数据库引擎DSN mysql,sqlite
func getDbEngineDSN(engine string, config map[string]string) string {
2017-04-02 02:38:49 +00:00
engine = strings.ToLower(engine)
var dsn string = ""
switch engine {
case "mysql":
dsn = fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=%s",
config["user"],
config["password"],
config["host"],
config["port"],
config["database"],
config["charset"])
}
2017-04-01 09:04:40 +00:00
2017-04-02 02:38:49 +00:00
return dsn
2017-04-01 09:04:40 +00:00
}
2017-04-03 02:20:08 +00:00
// 定时ping, 防止因数据库超时设置被断开
func keepDbAlived(engine *xorm.Engine) {
t := time.Tick(180 * time.Second)
for {
<- t
engine.Ping()
}
}