package scheduler

import (
	"gitee.com/h79/goutils/common/algorithm"
	"sync"
)

type Tasks struct {
	rm       sync.Mutex
	groups   []*groupScheduled
	groupMod uint32
}

var inst *Tasks

func Instance() *Tasks {
	return inst
}

func SetInstance(tasks *Tasks) {
	inst = tasks
}

func NewTasks(groupNum uint32) *Tasks {
	if groupNum == 0 {
		groupNum = 20
	}
	if groupNum > 2000 {
		groupNum = 2000
	}
	var (
		i      uint32
		groups = make([]*groupScheduled, groupNum)
	)
	for i = 0; i < groupNum; i++ {
		groups[i] = newGroupScheduled()
	}
	return &Tasks{
		groupMod: groupNum,
		groups:   groups,
	}
}

func (job *Tasks) groupIndex(key string) uint32 {
	id := algorithm.HashCode(key)
	return id % job.groupMod
}

func (job *Tasks) AddTask(task Task) {
	job.rm.Lock()
	groups := job.groups[job.groupIndex(task.GetKey())]
	job.rm.Unlock()

	groups.AddTask(task)
}

func (job *Tasks) AddTriggerTask(task Task, trigger Trigger) error {
	job.rm.Lock()
	groups := job.groups[job.groupIndex(task.GetKey())]
	job.rm.Unlock()

	return groups.AddTriggerTask(task, trigger)
}

func (job *Tasks) StopTask(jobId string) {
	job.rm.Lock()
	groups := job.groups[job.groupIndex(jobId)]
	job.rm.Unlock()

	groups.StopTask(jobId)
}

func (job *Tasks) RemoveTask(jobId string) {
	job.rm.Lock()
	groups := job.groups[job.groupIndex(jobId)]
	job.rm.Unlock()

	groups.RemoveTask(jobId)
}

func (job *Tasks) HasTask(jobId string) bool {
	job.rm.Lock()
	groups := job.groups[job.groupIndex(jobId)]
	job.rm.Unlock()
	return groups.HasTask(jobId)
}

func (job *Tasks) Run() {
	job.rm.Lock()
	for i := range job.groups {
		job.groups[i].Run()
	}
	job.rm.Unlock()
}