package etimer

import (
	"container/list"
	"context"
	"sync"
	"time"

	"go.uber.org/atomic"
)

type ExtendTimerMgr struct {
	uid          atomic.Int64
	triggerQueue chan *ExtendTimer

	ctxWithCancel context.Context
	cancelFunc    context.CancelFunc
	wg            sync.WaitGroup
	closeFlag     bool

	lastMillSec      int64
	millSecTimerList *list.List

	minTimerList *list.List
}

func NewExtendTimerMgr(triggerQueueSize int) *ExtendTimerMgr {
	return &ExtendTimerMgr{
		uid:          *atomic.NewInt64(0),
		triggerQueue: make(chan *ExtendTimer, triggerQueueSize),
	}
}

func (u *ExtendTimerMgr) Init(ctx context.Context) {
	u.ctxWithCancel, u.cancelFunc = context.WithCancel(ctx)
}

func (u *ExtendTimerMgr) UnInit() {
	if u.closeFlag {
		return
	}

	u.closeFlag = true
	if u.cancelFunc != nil {
		u.cancelFunc()
		u.wg.Wait()
		u.cancelFunc = nil
	}
}

func (u *ExtendTimerMgr) AddOnceTimer(d time.Duration, cb FuncType, args ArgType) *ExtendTimer {
	uid := u.uid.Add(1)
	extendTimer := newExtendTimer(uid, cb, args, false)
	return extendTimer
}

func (u *ExtendTimerMgr) GetExtendTimerTriggerChannel() chan *ExtendTimer {
	return u.triggerQueue
}

func (u *ExtendTimerMgr) ExecExtendTimerTriggerLogic(timer *ExtendTimer) {
	if timer.state != TimerRunningState {
		return
	}
	timer.Call()
	timer.SetInvalid()
}