jyqyfw/jyinfomatch/src/qyfw/handler.go

package qyfw

import (
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"qfw/util"
	"strings"
	"time"
)

var (
	AllJobs = map[string]*Job{}
)

//加载全部脚本
func LoadAllLuaScript() {
	filepath.Walk("./luascript", func(path string, info os.FileInfo, err error) error {
		if info.IsDir() || !strings.HasSuffix(info.Name(), ".lua") {
			return nil
		}
		bs, err := ioutil.ReadFile(path)
		if err != nil {
			log.Println(err)
			return err
		}
		sp := NewLuaScript(info.Name(), string(bs))
		if sp != nil {
			AllJobs[sp.Name] = sp
			log.Println("成功加载脚本", info.Name())
		}
		return nil
	})
	go runJob()
}

//开始定时任务
func runJob() {
	util.Try(func() {
		LoadDatasByEs(&SysConfig)
		util.WriteSysConfig(SysConfig)
	}, func(e interface{}) {})
	time.AfterFunc(time.Duration(util.IntAll(SysConfig["durationMinutes"]))*time.Minute, runJob)
}

//初始化脚本
func NewLuaScript(name, luafile string) *Job {
	defer util.Catch()
	job := &Job{}
	job.Results = &[]map[string]interface{}{}
	job.Name = name
	job.ScriptFile = luafile
	job.EachListPool = make(chan bool, PoolSize)
	//只是单纯的验证一下，lua脚本是否有问题
	script := &Script{}
	if !script.LoadScript(name, luafile, true) {
		return nil
	}
	job.Appid = script.GetVar("appid")
	if job.Appid == "" {
		log.Println("error：从脚本", name, "中获取到Appid为空！")
		return nil
	}
	return job
}