feat(coroutine): Added Concurrent and Parallel

Signed-off-by: Flc゛ <[email protected]>
go-fries · Jan 25, 2024 · dbff970 · dbff970
1 parent 6540102
commit dbff970
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/coroutine/concurrent_test.go b/coroutine/concurrent_test.go
@@ -2,6 +2,7 @@ package coroutine
 
 import (
 	"bytes"
+	"sync"
 	"testing"
 	"time"
 
@@ -12,16 +13,23 @@ func TestConcurrent(t *testing.T) {
 	var (
 		start  = time.Now()
 		buffer bytes.Buffer
+		mu     sync.Mutex
 	)
 
 	Concurrent(2, func() {
 		time.Sleep(1 * time.Second)
+		mu.Lock()
+		defer mu.Unlock()
 		buffer.WriteString("1")
 	}, func() {
 		time.Sleep(2 * time.Second)
+		mu.Lock()
+		defer mu.Unlock()
 		buffer.WriteString("2")
 	}, func() {
 		time.Sleep(3 * time.Second)
+		mu.Lock()
+		defer mu.Unlock()
 		buffer.WriteString("3")
 	})
 

diff --git a/coroutine/parallel_test.go b/coroutine/parallel_test.go
@@ -2,25 +2,34 @@ package coroutine
 
 import (
 	"bytes"
+	"sync"
 	"testing"
 	"time"
 
 	"github.com/stretchr/testify/assert"
 )
 
 func TestParallel(t *testing.T) {
-	start := time.Now()
-
-	var buffer bytes.Buffer
+	var (
+		start  = time.Now()
+		buffer bytes.Buffer
+		mu     sync.Mutex
+	)
 
 	Parallel(func() {
 		time.Sleep(1 * time.Second)
+		mu.Lock()
+		defer mu.Unlock()
 		buffer.WriteString("1")
 	}, func() {
 		time.Sleep(2 * time.Second)
+		mu.Lock()
+		defer mu.Unlock()
 		buffer.WriteString("2")
 	}, func() {
 		time.Sleep(3 * time.Second)
+		mu.Lock()
+		defer mu.Unlock()
 		buffer.WriteString("3")
 	})