- 用SyncMapWithTimeout管理tasksch任务

business/jxutils/jxutils_sync_map.go

@@ -0,0 +1,29 @@
+package jxutils
+
+import (
+	"sync"
+	"time"
+
+	"git.rosy.net.cn/baseapi/utils"
+)
+
+type SyncMapWithTimeout struct {
+	sync.Map
+	timers sync.Map
+}
+
+func (m *SyncMapWithTimeout) StoreWithTimeout(key, value interface{}, timeout time.Duration) {
+	m.Map.Store(key, value)
+	m.timers.Store(key, utils.AfterFuncWithRecover(timeout, func() {
+		m.Delete(key)
+	}))
+}
+
+func (m *SyncMapWithTimeout) Delete(key interface{}) {
+	m.Map.Delete(key)
+	if value, ok := m.timers.Load(key); ok {
+		timer := value.(*time.Timer)
+		timer.Stop()
+	}
+	m.timers.Delete(key)
+}