From 2caf9a867c33792eec8c4e8372d0ff0bb40466e7 Mon Sep 17 00:00:00 2001
From: chaz6chez <chaz6chez1993@outlook.com>
Date: Fri, 8 Nov 2024 10:17:47 +0800
Subject: [PATCH] feat Utils/Pool

---
 src/Utils/Pool/Pool.php | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/Utils/Pool/Pool.php b/src/Utils/Pool/Pool.php
index 354f2a2..862a5b2 100644
--- a/src/Utils/Pool/Pool.php
+++ b/src/Utils/Pool/Pool.php
@@ -14,6 +14,7 @@
 use function Workbunny\WebmanCoroutine\wait_for;
 
 use Workerman\Worker;
+use function Workbunny\WebmanCoroutine\wakeup;
 
 class Pool
 {
@@ -203,11 +204,13 @@ public static function idle(string $name): Pool|null
     public static function getIdle(string $name, int $timeout = -1): Pool
     {
         $pool = null;
+        // 使用pool.idle.$name事件等待空闲对象
         wait_for(function () use (&$pool, $name) {
             $pool = self::idle($name);
 
             return $pool !== null;
-        }, $timeout);
+        }, timeout: $timeout, event: "pool.idle.$name");
+        // 获取到的对象加锁
         $pool->setIdle(false);
 
         return $pool;
@@ -366,6 +369,14 @@ public function isIdle(): bool
     public function setIdle(bool $idle): void
     {
         $this->_idle = $idle;
+        if ($idle) {
+            $id = spl_object_hash($this);
+            $name = $this->getName();
+            // 唤醒pool.wait.$name.$id
+            wakeup("pool.wait.$name.$id");
+            // 唤醒pool.idle.$name
+            wakeup("pool.idle.$name");
+        }
     }
 
     /**
@@ -397,9 +408,12 @@ public function setForce(bool $force): void
      */
     public function wait(?\Closure $closure = null): void
     {
+        $id = spl_object_hash($this);
+        $name = $this->getName();
+        // 永久等待pool.wait.$name.$id唤醒事件
         wait_for(function () {
             return $this->isIdle();
-        });
+        }, timeout: -1, event: "pool.wait.$name.$id");
         if ($closure) {
             $this->setIdle(false);
             try {