github.com/angenalZZZ/gofunc@v0.0.0-20210507121333-48ff1be3917b/js/job.go

package js

import (
	"fmt"
	"os"
	"strings"
	"sync/atomic"
	"time"

	"github.com/angenalZZZ/gofunc/f"
	"github.com/dop251/goja"
)

// JobJs use cron job worker in javascript.
type JobJs struct {
	/*
		CRON Expression Format
		----------     ----------   --------------    --------------------------
		e.g. "@daily", "@hourly", "@every 1h30m",
			"30 * * * *"                     every hour on the half hour
			"30 3-6,20-23 * * *"             in the range 3-6am, 8-11pm
			"CRON_TZ=Asia/Tokyo 30 04 * * *" at 04:30 Tokyo time every day
		----------     ----------   --------------    --------------------------
		Field name   | Mandatory? | Allowed values  | Allowed special characters
		----------   | ---------- | --------------  | --------------------------
		Minutes      | Yes        | 0-59            | * / , -
		Hours        | Yes        | 0-23            | * / , -
		Day of month | Yes        | 1-31            | * / , - ?
		Month        | Yes        | 1-12 or JAN-DEC | * / , -
		Day of week  | Yes        | 0-6 or SUN-SAT  | * / , - ?
		----------   | ---------- | --------------  | --------------------------
		特定字符的含义如下：
		* 表示匹配该域的任意值，假如在minute域使用*, 即表示每分钟都会触发事件
		/ 表示起始时间开始触发，然后每隔固定时间触发一次，例如在minute域使用5/20,则意味着在5分的时候开始触发一次，而25，45等分别触发一次
		, 表示列出枚举值。例如：在minute域使用5,20，则意味着在5和20分每分钟触发一次
		- 表示范围，例如在minute域使用5-20，表示从5分到20分每分钟触发一次
		? 字符仅被用于“日”和“周”两个子表达式，表示不指定值，当两个子表达式其中之一被指定了值以后，为了避免冲突，需要将另一个子表达式的值设为?
	*/
	Spec string
	// the javascript content
	Script string
	// the javascript filepath
	File string
	// the javascript file modify time
	FileModTime time.Time
	// the last run time
	LastRunTime time.Time
	// the last run id
	LastRunID uint64

	// the job name
	Name string
	// the job parent list
	Parent []*JobJs
	// the job parent var'name
	ParentName string

	// R create a javascript runtime
	R func(*GoRuntimeParam) *GoRuntime
	// P create a javascript runtime input parameter
	P *GoRuntimeParam

	// Func func runner
	Func func(goja.FunctionCall) goja.Value
	// Self func this object
	Self goja.Value
}

// RunLogTimeFormat the time layout.
var RunLogTimeFormat = "15:04:05.000"

// Run implementation cron.Job interface.
func (j *JobJs) Run() {
	if j.R == nil {
		return
	}

	r := j.R(j.P)
	if r == nil {
		return
	}
	defer func() { r.Clear() }()

	id := atomic.AddUint64(&j.LastRunID, 1)
	j.LastRunTime = time.Now()
	s := fmt.Sprintf("%s [job] %q #%d started running. ", j.LastRunTime.Format(RunLogTimeFormat), j.Name, id)

	var (
		res goja.Value
		err error
	)

	if j.Func != nil {
		var jobs []*JobJs
		jobs, err = NewJobs(r, j.Script, j.ParentName, j.Name)
		if err == nil {
			if len(jobs) == 0 {
				j.RemoveFromParent()
				return
			}
			j1 := jobs[0] // a named job is found
			fmt.Println(s)
			res = j1.Func(goja.FunctionCall{This: j1.Self})
		}
	} else if j.Script != "" {
		fmt.Println(s)
		res, err = r.RunString(j.Script)
	} else {
		return
	}

	ts := time.Now()
	fmt.Printf("%s [job] %q #%d takes %s ", ts.Format(RunLogTimeFormat), j.Name, id, ts.Sub(j.LastRunTime))
	if err != nil {
		fmt.Printf("error: %v", err)
	} else if res != nil {
		if v := res.Export(); v != nil {
			fmt.Printf("return: %+v", v)
		} else {
			fmt.Print("ok.")
		}
	} else {
		fmt.Print("ok.")
	}
	fmt.Println()
	fmt.Println()

	return
}

// Init the javascript job if file is not empty.
func (j *JobJs) Init() error {
	if j.File == "" {
		return nil
	}
	if !j.FileIsMod() {
		return os.ErrNotExist
	}
	if err := j.FileMod(); err != nil {
		return err
	}
	return nil
}

// FileIsMod check javascript file is modify.
func (j *JobJs) FileIsMod() bool {
	if j.File == "" {
		return false
	}
	info, err := os.Stat(j.File)
	if os.IsNotExist(err) {
		return false
	}
	if t := info.ModTime(); t.Unix() != j.FileModTime.Unix() {
		j.FileModTime = t
		return true
	}
	return false
}

// FileMod read updated javascript file content.
func (j *JobJs) FileMod() error {
	if j.File == "" {
		return os.ErrNotExist
	}
	script, err := f.ReadFile(j.File)
	if err != nil {
		return err
	}
	j.Script = strings.TrimSpace(string(script))
	return err
}

// RemoveFromParent remove self from parent.
func (j *JobJs) RemoveFromParent() {
	p := j.Parent
	if p == nil {
		return
	}
	for i, l := 0, len(p); i < l; i++ {
		if j.Name != p[i].Name {
			continue
		}
		// reset self object
		j.Script, j.Func, j.R, j.Parent = "", nil, nil, nil
		// update parent underlying array
		p = append(p[:i], p[i+1:]...)
		// maintain the correct index
		return
	}
}