调整 worker 调度层

Author: FishGoddess

HISTORY.md

@@ -1,5 +1,11 @@
 ## ✒ 历史版本的特性介绍 (Features in old versions)
 
+### v0.2.3-alpha
+
+> 此版本发布于 2025-06-28
+
+* 调整 worker 调度层
+
 ### v0.2.2-alpha
 
 > 此版本发布于 2025-06-26

_icons/coverage.svg

@@ -60,7 +60,7 @@ func WithSyncMutex() Option {
 	}
 }
 
-// WithRoundRobinScheduler sets the new workers function using round robin strategy.
+// WithRoundRobinScheduler sets the new scheduler function using round robin strategy.
 func WithRoundRobinScheduler() Option {
 	newSchedulerFunc := func(workers []*worker) scheduler {
 		return newRoundRobinScheduler(workers)
@@ -71,7 +71,7 @@ func WithRoundRobinScheduler() Option {
 	}
 }
 
-// WithRandomScheduler sets the new workers function using random strategy.
+// WithRandomScheduler sets the new scheduler function using random strategy.
 func WithRandomScheduler() Option {
 	newSchedulerFunc := func(workers []*worker) scheduler {
 		return newRandomScheduler(workers)

option.go

@@ -78,7 +78,7 @@ func TestWithRoundRobinScheduler(t *testing.T) {
 	conf := newDefaultConfig(workerNum)
 	WithRoundRobinScheduler()(conf)
 
-	workers := make([]*worker, workerNum)
+	workers := make([]*worker, 0, workerNum)
 	got := conf.newSchedulerFunc(workers)
 
 	scheduler, ok := got.(*roundRobinScheduler)
@@ -97,7 +97,7 @@ func TestWithRandomScheduler(t *testing.T) {
 	conf := newDefaultConfig(workerNum)
 	WithRandomScheduler()(conf)
 
-	workers := make([]*worker, workerNum)
+	workers := make([]*worker, 0, workerNum)
 	got := conf.newSchedulerFunc(workers)
 
 	scheduler, ok := got.(*randomScheduler)

option_test.go

@@ -4,19 +4,32 @@
 
 package goes
 
-import "testing"
+import (
+	"fmt"
+	"testing"
+)
 
 // go test -v -cover -run=^TestRandomScheduler$
 func TestRandomScheduler(t *testing.T) {
 	workerNum := 16
 	workers := make([]*worker, 0, workerNum)
-	for i := 0; i < workerNum; i++ {
-		worker := new(worker)
-		workers = append(workers, worker)
+	for range workerNum {
+		workers = append(workers, new(worker))
 	}
 
 	scheduler := newRandomScheduler(workers)
+	if fmt.Sprintf("%p", scheduler.workers) != fmt.Sprintf("%p", workers) {
+		t.Fatalf("scheduler.workers %p != workers %p", scheduler.workers, workers)
+	}
+
+	if len(scheduler.workers) != len(workers) {
+		t.Fatalf("len(scheduler.workers) %d != len(workers) %d", len(scheduler.workers), len(workers))
+	}
+
 	scheduler.Set(workers)
+	if fmt.Sprintf("%p", scheduler.workers) != fmt.Sprintf("%p", workers) {
+		t.Fatalf("scheduler.workers %p != workers %p", scheduler.workers, workers)
+	}
 
 	if len(scheduler.workers) != len(workers) {
 		t.Fatalf("len(scheduler.workers) %d != len(workers) %d", len(scheduler.workers), len(workers))

random_test.go

@@ -4,19 +4,32 @@
 
 package goes
 
-import "testing"
+import (
+	"fmt"
+	"testing"
+)
 
 // go test -v -cover -run=^TestRoundRobinScheduler$
 func TestRoundRobinScheduler(t *testing.T) {
 	workerNum := 16
 	workers := make([]*worker, 0, workerNum)
-	for i := 0; i < workerNum; i++ {
-		worker := new(worker)
-		workers = append(workers, worker)
+	for range workerNum {
+		workers = append(workers, new(worker))
 	}
 
 	scheduler := newRoundRobinScheduler(workers)
+	if fmt.Sprintf("%p", scheduler.workers) != fmt.Sprintf("%p", workers) {
+		t.Fatalf("scheduler.workers %p != workers %p", scheduler.workers, workers)
+	}
+
+	if len(scheduler.workers) != len(workers) {
+		t.Fatalf("len(scheduler.workers) %d != len(workers) %d", len(scheduler.workers), len(workers))
+	}
+
 	scheduler.Set(workers)
+	if fmt.Sprintf("%p", scheduler.workers) != fmt.Sprintf("%p", workers) {
+		t.Fatalf("scheduler.workers %p != workers %p", scheduler.workers, workers)
+	}
 
 	if len(scheduler.workers) != len(workers) {
 		t.Fatalf("len(scheduler.workers) %d != len(workers) %d", len(scheduler.workers), len(workers))