mirror of https://github.com/ouqiang/gocron
113 lines
2.7 KiB
Go
113 lines
2.7 KiB
Go
package app
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/ouqiang/gocron/models"
|
|
"github.com/ouqiang/gocron/service"
|
|
"github.com/ouqiang/gocron/modules/setting"
|
|
"github.com/ouqiang/gocron/modules/logger"
|
|
"runtime"
|
|
"github.com/ouqiang/gocron/modules/notify"
|
|
)
|
|
|
|
var (
|
|
AppDir string // 应用根目录
|
|
ConfDir string // 配置目录
|
|
LogDir string // 日志目录
|
|
DataDir string // 存放session等
|
|
AppConfig string // 应用配置文件
|
|
Installed bool // 应用是否安装过
|
|
)
|
|
|
|
func InitEnv() {
|
|
runtime.GOMAXPROCS(runtime.NumCPU())
|
|
logger.InitLogger()
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
logger.Fatal(err)
|
|
}
|
|
AppDir = wd
|
|
ConfDir = AppDir + "/conf"
|
|
LogDir = AppDir + "/log"
|
|
DataDir = AppDir + "/data"
|
|
AppConfig = ConfDir + "/app.ini"
|
|
checkDirExists(ConfDir, LogDir, DataDir)
|
|
Installed = IsInstalled()
|
|
if Installed {
|
|
InitDb()
|
|
InitResource()
|
|
settingModel := new(models.Setting)
|
|
notify.SlackUrl, _ = settingModel.SlackUrl()
|
|
}
|
|
}
|
|
|
|
// 判断应用是否安装过
|
|
func IsInstalled() bool {
|
|
_, err := os.Stat(ConfDir + "/install.lock")
|
|
if os.IsNotExist(err) {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// 创建安装锁文件
|
|
func CreateInstallLock() error {
|
|
_, err := os.Create(ConfDir + "/install.lock")
|
|
if err != nil {
|
|
logger.Error("创建安装锁文件失败")
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// 初始化资源
|
|
func InitResource() {
|
|
// 初始化定时任务
|
|
serviceTask := new(service.Task)
|
|
serviceTask.Initialize()
|
|
}
|
|
|
|
// 初始化DB
|
|
func InitDb() {
|
|
dbConfig := getDbConfig(AppConfig)
|
|
models.Db = models.CreateDb(dbConfig)
|
|
}
|
|
|
|
// 检测目录是否存在
|
|
func checkDirExists(path ...string) {
|
|
for _, value := range path {
|
|
_, err := os.Stat(value)
|
|
if os.IsNotExist(err) {
|
|
logger.Fatal(value + "目录不存在")
|
|
}
|
|
if os.IsPermission(err) {
|
|
logger.Fatal(value + "目录无权限操作")
|
|
}
|
|
}
|
|
}
|
|
|
|
// 获取数据库配置
|
|
func getDbConfig(configFile string) map[string]string {
|
|
config, err := setting.Read(configFile)
|
|
if err != nil {
|
|
logger.Fatal(err)
|
|
}
|
|
section := config.Section("db")
|
|
if err != nil {
|
|
logger.Fatal(err)
|
|
}
|
|
var db map[string]string = make(map[string]string)
|
|
db["user"] = section.Key("user").String()
|
|
db["password"] = section.Key("password").String()
|
|
db["host"] = section.Key("host").String()
|
|
db["port"] = section.Key("port").String()
|
|
db["database"] = section.Key("database").String()
|
|
db["charset"] = section.Key("charset").String()
|
|
db["prefix"] = section.Key("prefix").String()
|
|
db["engine"] = section.Key("engine").String()
|
|
|
|
return db
|
|
}
|