package scheduler

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

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

var (
	inst      *Tasks
	tasksOnce sync.Once
)

func Instance() *Tasks {
	return inst
}

func NewCapacityTasks(groupNum uint32, capacity int) *Tasks {
	return newTasks(groupNum, capacity)
}

func NewTasks(groupNum uint32) *Tasks {
	return newTasks(groupNum, 5000)
}

func newTasks(groupNum uint32, capacity int) *Tasks {
	if groupNum == 0 {
		groupNum = 20
	}
	if groupNum > 2000 {
		groupNum = 2000
	}
	var i uint32
	var groups = make([]*groupScheduled, groupNum)
	for i = 0; i < groupNum; i++ {
		groups[i] = newGroupScheduled(capacity)
	}
	var tasks = &Tasks{
		groupMod: groupNum,
		groups:   groups,
	}
	tasksOnce.Do(func() {
		inst = tasks
	})
	return tasks
}

func (job *Tasks) groupScheduled(group uint32) *groupScheduled {
	job.rm.Lock()
	defer job.rm.Unlock()
	return job.groups[group]
}

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

func (job *Tasks) AddTask(task Task) {
	groups := job.groupScheduled(job.groupIndex(task.GetKey()))
	groups.AddTask(task)
}

func (job *Tasks) AddTriggerTask(task Task, trigger Trigger) error {
	groups := job.groupScheduled(job.groupIndex(task.GetKey()))
	return groups.AddTriggerTask(task, trigger)
}

func (job *Tasks) StopTask(jobId string) {
	groups := job.groupScheduled(job.groupIndex(jobId))
	groups.StopTask(jobId)
}

func (job *Tasks) RemoveTask(jobId string) {
	groups := job.groupScheduled(job.groupIndex(jobId))
	groups.RemoveTask(jobId)
}

func (job *Tasks) HasTask(jobId string) bool {
	groups := job.groupScheduled(job.groupIndex(jobId))
	return groups.HasTask(jobId)
}

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