gocron/modules/app/app.go

package app

import (
    "os"
    "runtime"

    "github.com/ouqiang/cron-scheduler/models"
    "github.com/ouqiang/cron-scheduler/modules/ansible"
    "github.com/ouqiang/cron-scheduler/modules/crontask"
    "github.com/ouqiang/cron-scheduler/modules/utils"
    "github.com/ouqiang/cron-scheduler/service"
)

var (
    AppDir       string // 应用根目录
    ConfDir      string // 配置目录
    LogDir       string // 日志目录
    DataDir      string // 数据目录，存放session文件等
    AppConfig    string // 应用配置文件
    AnsibleHosts string // ansible hosts文件
    Installed    bool   // 应用是否安装过
)

func InitEnv() {
    CheckEnv()
    wd, err := os.Getwd()
    if err != nil {
        panic(err)
    }
    AppDir = wd
    ConfDir = AppDir + "/conf"
    LogDir = AppDir + "/log"
    DataDir = AppDir + "/data"
    AppConfig = ConfDir + "/app.ini"
    AnsibleHosts = ConfDir + "/ansible_hosts.ini"
    checkDirExists(ConfDir, LogDir, DataDir)
    // ansible配置文件目录
    os.Setenv("ANSIBLE_CONFIG", ConfDir)
    Installed = IsInstalled()
    if Installed {
        InitDb()
        InitResource()
    }
}

// 判断应用是否安装过
func IsInstalled() bool {
    _, err := os.Stat(ConfDir + "/install.lock")
    if os.IsNotExist(err) {
        return false
    }

    return true
}

// 检测环境
func CheckEnv() {
    // ansible不支持安装在windows上, windows只能作为被控机
    if runtime.GOOS == "windows" {
        panic("不支持在windows上运行")
    }
    _, err := utils.ExecShell("ansible", "--version")
    if err != nil {
        panic(err)
    }
    _, err = utils.ExecShell("ansible-playbook", "--version")
    if err != nil {
        panic(err)
    }
}

// 创建安装锁文件
func CreateInstallLock() error {
    _, err := os.Create(ConfDir + "/install.lock")
    if err != nil {
        utils.RecordLog("创建安装锁文件失败")
    }

    return err
}

// 初始化资源
func InitResource() {
    // 初始化ansible Hosts
    ansible.DefaultHosts = ansible.NewHosts(AnsibleHosts)
    // 初始化定时任务
    crontask.DefaultCronTask = crontask.NewCronTask()
    serviceTask := new(service.Task)
    serviceTask.Initialize()
}

// 初始化DB
func InitDb() {
    models.Db = models.CreateDb(AppConfig)
}

// 检测目录是否存在
func checkDirExists(path ...string) {
    for _, value := range path {
        _, err := os.Stat(value)
        if os.IsNotExist(err) {
            panic(value + "目录不存在")
        }
        if os.IsPermission(err) {
            panic(value + "目录无权限操作")
        }
    }
}