mirror of https://github.com/ouqiang/gocron
122 lines
2.5 KiB
Go
122 lines
2.5 KiB
Go
package app
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/ouqiang/gocron/modules/logger"
|
|
"github.com/ouqiang/gocron/modules/setting"
|
|
"github.com/ouqiang/gocron/modules/utils"
|
|
"io/ioutil"
|
|
"strconv"
|
|
"strings"
|
|
"fmt"
|
|
)
|
|
|
|
var (
|
|
AppDir string // 应用根目录
|
|
ConfDir string // 配置目录
|
|
LogDir string // 日志目录
|
|
DataDir string // 存放session等
|
|
AppConfig string // 应用配置文件
|
|
Installed bool // 应用是否安装过
|
|
Setting *setting.Setting // 应用配置
|
|
VersionId int // 版本号
|
|
VersionFile string // 版本号文件
|
|
)
|
|
|
|
func InitEnv(versionString string) {
|
|
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"
|
|
VersionFile = ConfDir + "/.version"
|
|
createDirIfNotExists(ConfDir, LogDir, DataDir)
|
|
Installed = IsInstalled()
|
|
VersionId = ToNumberVersion(versionString)
|
|
}
|
|
|
|
// 判断应用是否已安装
|
|
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("创建安装锁文件conf/install.lock失败")
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// 更新应用版本号文件
|
|
func UpdateVersionFile() {
|
|
err := ioutil.WriteFile(VersionFile,
|
|
[]byte(strconv.Itoa(VersionId)),
|
|
0644,
|
|
)
|
|
|
|
if err != nil {
|
|
logger.Fatal(err)
|
|
}
|
|
}
|
|
|
|
// 获取应用当前版本号, 从版本号文件中读取
|
|
func GetCurrentVersionId() int {
|
|
if !utils.FileExist(VersionFile) {
|
|
return 0
|
|
}
|
|
|
|
bytes, err := ioutil.ReadFile(VersionFile)
|
|
if err != nil {
|
|
logger.Fatal(err)
|
|
}
|
|
|
|
versionId, err := strconv.Atoi(strings.TrimSpace(string(bytes)))
|
|
if err != nil {
|
|
logger.Fatal(err)
|
|
}
|
|
|
|
return versionId
|
|
}
|
|
|
|
// 把字符串版本号a.b.c转换为整数版本号abc
|
|
func ToNumberVersion(versionString string) int {
|
|
v := strings.Replace(versionString, ".", "", -1)
|
|
if len(v) < 3 {
|
|
v += "0"
|
|
}
|
|
|
|
versionId, err := strconv.Atoi(v)
|
|
if err != nil {
|
|
logger.Fatal(err)
|
|
}
|
|
|
|
return versionId
|
|
}
|
|
|
|
// 检测目录是否存在
|
|
func createDirIfNotExists(path ...string) {
|
|
for _, value := range path {
|
|
if utils.FileExist(value) {
|
|
continue
|
|
}
|
|
err := os.Mkdir(value, 0755)
|
|
if err != nil {
|
|
logger.Fatal(fmt.Sprintf("创建目录失败:%s", err.Error()))
|
|
}
|
|
}
|
|
}
|