1 files changed, 29 insertions, 0 deletions
diff --git a/pkg/worker/scheduler.go b/pkg/worker/scheduler.go
new file mode 100644
index 0000000..b410b33
--- /dev/null
+++ b/pkg/worker/scheduler.go
@@ -0,0 +1,29 @@
+package worker
+
+import (
+	"fmt"
+	"sync/atomic"
+)
+
+type Scheduler struct {
+	pool  chan any
+	count atomic.Int64
+}
+
+func NewScheduler(count uint) *Scheduler {
+	return &Scheduler{
+		pool: make(chan any, count),
+	}
+}
+
+func (self *Scheduler) Take() {
+	self.pool <- nil
+	self.count.Add(1)
+	fmt.Printf("<- %d\n", self.count.Load())
+}
+
+func (self *Scheduler) Return() {
+	<-self.pool
+	self.count.Add(-1)
+	fmt.Printf("-> %d\n", self.count.Load())
+}