+ UnmanageTasks

business/jxutils/tasksch/task_man.go

@@ -7,10 +7,13 @@ import (
 	"git.rosy.net.cn/jx-callback/business/jxutils"
 )
 
+const (
+	defLastHours  = 24
+	maxStoreHours = 48 // 最多存放小时数
+)
+
 var (
-	defTaskMan   TaskMan
-	defLastHours = 24
-	maxStoreTime = 48 * time.Hour // 最多存两天时间
+	defTaskMan TaskMan
 )
 
 type TaskMan struct {
@@ -39,7 +42,7 @@ func (m *TaskMan) GetTasks(taskID string, fromStatus, toStatus int, lastHours in
 }
 
 func (m *TaskMan) ManageTask(task ITask) ITask {
-	m.taskMap.StoreWithTimeout(task.GetID(), task, maxStoreTime)
+	m.taskMap.StoreWithTimeout(task.GetID(), task, maxStoreHours*time.Hour)
 	return task
 }
 
@@ -57,3 +60,15 @@ func IsTaskRunning(taskID string) bool {
 	}
 	return false
 }
+
+func UnmanageTasks(taskIDs []string) {
+	if len(taskIDs) == 0 {
+		allTasks := GetTasks("", TaskStatusBegin, TaskStatusEnd, maxStoreHours, "")
+		for _, v := range allTasks {
+			taskIDs = append(taskIDs, v.GetID())
+		}
+	}
+	for _, v := range taskIDs {
+		defTaskMan.taskMap.Delete(v)
+	}
+}