优化waitgroup

entry.go

@@ -45,11 +45,11 @@ func init() {
 
 // AddToTaskList add the task to the execution list
 func AddToTaskList(ts ...Tasker) {
-	wg.Add(1)
 	for _, t := range ts {
 		if t == nil {
 			continue
 		}
+		wg.Add(1)
 		editC <- t
 	}
 }
@@ -75,10 +75,13 @@ func (tl *taskList) stop(id string) {
 // ChangeInterval changes the interval between the tasks specified by the ID,
 // Apply only to polling tasks.
 func ChangeInterval(id string, interval time.Duration) error {
-	wg.Wait()
 	tsk := tasks.get(id)
 	if tsk == nil {
-		return fmt.Errorf("Task does not exist")
+		wg.Wait()
+		tsk = tasks.get(id)
+		if tsk == nil {
+			return fmt.Errorf("Task does not exist")
+		}
 	}
 	var task *Task
 	var ok bool

task_test.go

@@ -49,15 +49,15 @@ func init() {
 
 func TestChangeInterval(t *testing.T) {
 	var p []int
-	tk := NewTask(time.Second*1, func() {
+	tk := NewTask(time.Second*2, func() {
 		p = append(p, 1)
 	})
 	AddToTaskList(tk)
-	ChangeInterval(tk.ID(), time.Second*2)
+	ChangeInterval(tk.ID(), time.Second*1)
 	select {
 	case <-time.After(time.Second*2 + time.Millisecond*100):
-		if len(p) != 1 {
-			t.Errorf("TestTask() fail , need len : %d , actually len : %d", 1, len(p))
+		if len(p) != 2 {
+			t.Errorf("TestTask() fail , need len : %d , actually len : %d", 2, len(p))
 		}
 		return
 	}