gocron/modules/app/app.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) {
			err := os.Mkdir(value, 0755)
			if err != nil {
				logger.Fatal(fmt.Sprintf("创建目录失败:%s", err.Error()))
			}
		}
	}
}