gocron/modules/app/app.go

package app

import (
	"os"
	"runtime"

	"github.com/ouqiang/cron-scheduler/modules/utils"
	"github.com/ouqiang/cron-scheduler/modules/crontask"
	"github.com/ouqiang/cron-scheduler/models"
	"github.com/ouqiang/cron-scheduler/modules/ansible"
	"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 + "目录无权限操作")
		}
	}
}