package scheduler

type (
	baseTaskWrapper struct {
		handle  TimerHandler
		exec    TimerFunc
		owner   *goCornTimer
		running bool
	}

	// runLimitsTask 限制次数类函数包装，
	runLimitsWrapper struct {
		*baseTaskWrapper
		count    int
		maxCount int
	}

	// abortTask 在函数抛出错误时候中断定时器执行
	abortWrapper struct {
		*baseTaskWrapper
	}
)

func (wp *runLimitsWrapper) runTask() {
	defer func() {
		if err := recover(); err != nil {
			wp.cancel()
		}
	}()

	// 非可执行状态
	if !wp.running {
		return
	}

	// 执行指令
	err := wp.exec()
	if err != nil {
		wp.cancel()
	}

	// 计算执行次数，达到上限则执行，也停止业务
	wp.count++
	if wp.count >= wp.maxCount {
		wp.cancel()
		return
	}
}

func (wp *runLimitsWrapper) cancel() {
	// 设置状态不可执行
	wp.running = false
	// 通知调度器删除
	_ = wp.owner.CancelTimer(wp.handle)
}

func (wp *abortWrapper) runTask() {
	defer func() {
		if err := recover(); err != nil {
			wp.cancel()
		}
	}()

	// 非可执行状态
	if !wp.running {
		return
	}

	// 执行指令
	err := wp.exec()
	if err != nil {
		wp.cancel()
	}
}

func (wp *abortWrapper) cancel() {
	// 设置状态不可执行
	wp.running = false
	// 通知调度器删除
	_ = wp.owner.CancelTimer(wp.handle)
}