hunter/hunter.go

package hunter

import (
	"log"
	"strconv"

	astack "github.com/474420502/focus/stack/arraystack"

	pqueue "github.com/474420502/focus/priority_queue"
	"github.com/474420502/requests"
)

// IGobalBefore 全局任务执行之前
type IGobalBefore interface {
	GobalBefore(cxt *TaskContext)
}

// IGobalAfter 全局任务执行之后
type IGobalAfter interface {
	GobalAfter(cxt *TaskContext)
}

// Hunter 任务相关 必须有序
type Hunter struct {
	share map[string]interface{}

	session *requests.Session

	tasks       []ITask
	createQueue func() *pqueue.PriorityQueue

	executes *astack.Stack // *TaskContext
}

// NewHunter 默认最大优先, tasks为预先需要addtask
func NewHunter(tasks ...ITask) *Hunter {
	hunter := NewPriorityMaxHunter()
	for _, task := range tasks {
		hunter.AddTask(task)
	}
	return hunter
}

// NewHunterFromTasks 默认最大优先
// func NewHunterFromTasks(tasks ...ITask) *Hunter {
// 	hunter := NewPriorityMaxHunter()
// 	for _, task := range tasks {
// 		hunter.AddTask(task)
// 	}
// 	return hunter
// }

// NewPriorityHunter 自定义优先处理队列
func NewPriorityHunter(queueCreator func() *pqueue.PriorityQueue) *Hunter {
	hunter := &Hunter{}
	hunter.createQueue = queueCreator
	hunter.executes = astack.New()

	hunter.share = make(map[string]interface{})
	return hunter
}

// NewPriorityMaxHunter 最大优先
func NewPriorityMaxHunter() *Hunter {
	return NewPriorityHunter(CreatePriorityMaxQueue)
}

// NewPriorityMinHunter 最小优先
func NewPriorityMinHunter() *Hunter {
	return NewPriorityHunter(CreatePriorityMinQueue)
}

// Session Get session *requests.Session
func (hunter *Hunter) Session() *requests.Session {
	return hunter.session
}

// SetSession Set session *requests.Session
func (hunter *Hunter) SetSession(session *requests.Session) {
	hunter.session = session
}

// GetShare 获取share的数据, 存储用的
func (hunter *Hunter) GetShare(key string) interface{} {
	if v, ok := hunter.share[key]; ok {
		return v
	}
	return nil
}

// SetShare 设置share的数据, 存储用的
func (hunter *Hunter) SetShare(key string, value interface{}) {
	hunter.share[key] = value
}

// Execute 执行任务
func (hunter *Hunter) Execute() {

	if hunter.executes.Size() != 0 {
		hunter.recursionTasks()
	}

	for _, task := range hunter.tasks {
		hunter.execute(task)
	}

}

// Execute 执行任务
func (hunter *Hunter) execute(task ITask) {
	cxt := NewContext()

	btask := &BaseTask{}
	btask.SetTask(task)
	btask.SetParent(nil)
	btask.SetChildren(hunter.createQueue())

	cxt.parent = btask
	cxt.parent.Children().Push(btask)

	cxt.hunter = hunter

	// hunter.recursionTasks(cxt)

	hunter.executes.Push(cxt)
	hunter.recursionTasks()
}

func (hunter *Hunter) recursionTasks() {
	// 这层再加一个 Children 提取
	for icxt, ok := hunter.executes.Peek(); ok; icxt, ok = hunter.executes.Peek() {

		cxt := icxt.(*TaskContext)
		if children := cxt.parent.Children(); children != nil && children.Size() > 0 {
			if itask, ok := children.Pop(); ok {
				sautoid := strconv.Itoa(cxt.autoid)

				tasknode := itask.(ITaskNode)
				tasknode.SetID(sautoid)

				cxt.current = tasknode
				cxt.current.SetPath(cxt.parent.Path()) //

				task := tasknode.Task()

				if before, ok := task.(IBefore); ok {
					before.Before(cxt)
				}

				tasknode.Task().Execute(cxt)

				if after, ok := task.(IAfter); ok {
					after.After(cxt)
				}

				cxt.autoid++

				ncxt := NewContext()
				ncxt.autoid = 0
				ncxt.parent = cxt.current
				ncxt.parent.SetPath(ncxt.parent.Path() + "/" + ncxt.parent.TaskID()) //补正ncxt的路径
				ncxt.hunter = cxt.hunter

				// hunter.recursionTasks(ncxt)
				hunter.executes.Push(ncxt)
			}
		} else {
			hunter.executes.Pop()
		}
	}

}

// Stop 停止任务
func (hunter *Hunter) close(itask ITask) {
	if iclose, ok := itask.(IClose); ok {
		if err := iclose.Close(); err != nil {
			log.Println(err)
		}
	}

}

// Stop 停止任务
func (hunter *Hunter) Stop() {
	for _, task := range hunter.tasks {
		hunter.close(task)
	}
}

// AddTask 执行任务
func (hunter *Hunter) AddTask(task ITask) {
	hunter.tasks = append(hunter.tasks, task)
}

// SavePoint 持续
func (hunter *Hunter) savePoint() {
	// f, err := os.OpenFile("./gob.log", os.O_CREATE|os.O_TRUNC|os.O_RDWR, os.ModePerm)
	// if err != nil {
	// 	panic(err)
	// }
	// enc := gob.NewEncoder(f)
	// err = enc.Encode(hunter.executes.Values())
	// if err != nil {
	// 	panic(err)
	// }
	// for _, iv := range hunter.executes.Values() {
	// 	tc := iv.(*TaskContext)

	// }
}

// Execute 执行
// func (hunter *Hunter) Execute() {
// 	if itask, ok := hunter.task.Children().Top(); ok {
// 		task := itask.(ITask)
// 	}
// }