mirror of https://github.com/ouqiang/gocron
97 lines
2.0 KiB
Go
97 lines
2.0 KiB
Go
package app
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/ouqiang/gocron/modules/logger"
|
|
"runtime"
|
|
"github.com/ouqiang/gocron/modules/utils"
|
|
"gopkg.in/ini.v1"
|
|
"io/ioutil"
|
|
"strconv"
|
|
)
|
|
|
|
var (
|
|
AppDir string // 应用根目录
|
|
ConfDir string // 配置目录
|
|
LogDir string // 日志目录
|
|
DataDir string // 存放session等
|
|
AppConfig string // 应用配置文件
|
|
Installed bool // 应用是否安装过
|
|
Setting *ini.Section // 应用配置
|
|
PidFile string
|
|
)
|
|
|
|
|
|
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"
|
|
PidFile = LogDir + "/gocron.pid"
|
|
checkDirExists(ConfDir, LogDir, DataDir)
|
|
Installed = IsInstalled()
|
|
}
|
|
|
|
// 判断应用是否安装过
|
|
func IsInstalled() bool {
|
|
_, err := os.Stat(ConfDir + "/install.lock")
|
|
if os.IsNotExist(err) {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
func WritePid() {
|
|
pid := os.Getpid()
|
|
pidStr := strconv.Itoa(pid)
|
|
err := ioutil.WriteFile(PidFile, []byte(pidStr), 0644)
|
|
if err != nil {
|
|
logger.Fatalf("写入pid文件失败", err)
|
|
}
|
|
}
|
|
|
|
func GetPid() int {
|
|
bytes, err := ioutil.ReadFile(PidFile)
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
pidStr := string(bytes)
|
|
pid, err := strconv.Atoi(pidStr)
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
|
|
return pid
|
|
}
|
|
|
|
func RemovePid() {
|
|
os.Remove(PidFile)
|
|
}
|
|
|
|
// 创建安装锁文件
|
|
func CreateInstallLock() error {
|
|
_, err := os.Create(ConfDir + "/install.lock")
|
|
if err != nil {
|
|
logger.Error("创建安装锁文件conf/install.lock失败")
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// 检测目录是否存在
|
|
func checkDirExists(path ...string) {
|
|
for _, value := range path {
|
|
if !utils.FileExist(value) {
|
|
logger.Fatal(value + "目录不存在或无权限访问")
|
|
}
|
|
}
|
|
} |