72 lines
1.3 KiB
Go
72 lines
1.3 KiB
Go
package lib
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
mu sync.Mutex
|
|
instance *CooldownManager
|
|
)
|
|
|
|
type CooldownManager struct {
|
|
cooldowns map[string]time.Time
|
|
mu sync.Mutex
|
|
}
|
|
|
|
func NewCooldownManager() *CooldownManager {
|
|
return &CooldownManager{
|
|
cooldowns: make(map[string]time.Time),
|
|
}
|
|
}
|
|
|
|
func GetInstance() *CooldownManager {
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
|
|
if instance == nil {
|
|
instance = &CooldownManager{
|
|
cooldowns: make(map[string]time.Time),
|
|
}
|
|
}
|
|
|
|
return instance
|
|
}
|
|
|
|
func (m *CooldownManager) StartCooldown(userID string, key string, duration time.Duration) {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
|
|
m.cooldowns[userID+":"+key] = time.Now().Add(duration)
|
|
}
|
|
|
|
func (m *CooldownManager) IsOnCooldown(userID string, key string) (bool, time.Duration) {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
|
|
cooldownEnd, exists := m.cooldowns[userID+":"+key]
|
|
if !exists {
|
|
return false, 0
|
|
}
|
|
|
|
if time.Now().After(cooldownEnd) {
|
|
delete(m.cooldowns, userID+":"+key)
|
|
return false, 0
|
|
}
|
|
|
|
return true, time.Until(cooldownEnd)
|
|
}
|
|
|
|
func CancelCooldown(userID string, key string) {
|
|
manager := GetInstance()
|
|
|
|
// Handle non-existent keys gracefully
|
|
if _, exists := manager.cooldowns[userID+":"+key]; !exists {
|
|
return
|
|
}
|
|
|
|
manager.mu.Lock()
|
|
defer manager.mu.Unlock()
|
|
delete(manager.cooldowns, userID+":"+key)
|
|
}
|