refine concurrent flow

amyangfei · May 8, 2019 · da20216 · da20216
1 parent 3510fb2
commit da20216
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/pkg/hello/hello.go b/pkg/hello/hello.go
@@ -79,18 +79,15 @@ func Boundary() {
 		})
 	}()
 
-	go func() {
-		time.Sleep(time.Second * 300)
-		fmt.Println("after long sleep 4")
-	}()
-
 	go func() {
 		fmt.Println("before sleep 1")
 		time.Sleep(time.Second * 300)
 		fmt.Println("after long sleep 1")
 	}()
 
+	wg.Add(1)
 	go func(cctx context.Context) {
+		defer wg.Done()
 		fmt.Println("before sleep 2")
 		select {
 		case <-cctx.Done():
@@ -101,12 +98,10 @@ func Boundary() {
 		fmt.Println("after long sleep 2")
 	}(ctx)
 
-	wg.Add(1)
 	go func() {
-		defer wg.Done()
 		time.Sleep(time.Millisecond * 100)
+		cancel()
 	}()
 
 	wg.Wait()
-	cancel()
 }