add several pool methods

white-poto · Feb 10, 2017 · 6397ddc · 6397ddc
1 parent 4d934e0
commit 6397ddc
Show file tree

Hide file tree

Showing 3 changed files with 67 additions and 37 deletions.
diff --git a/src/AbstractPool.php b/src/AbstractPool.php
@@ -113,4 +113,71 @@ public function aliveCount()
 
         return $count;
     }
+
+    /**
+     * get process by name
+     *
+     * @param string $name process name
+     * @return Process|null
+     */
+    public function getProcessByName($name)
+    {
+        foreach ($this->processes as $process) {
+            if ($process->name() == $name) {
+                return $process;
+            }
+        }
+
+        return null;
+    }
+
+    /**
+     * remove process by name
+     *
+     * @param string $name process name
+     * @throws \RuntimeException
+     */
+    public function removeProcessByName($name)
+    {
+        foreach ($this->processes as $key => $process) {
+            if ($process->name() == $name) {
+                if ($process->isRunning()) {
+                    throw new \RuntimeException("can not remove a running process");
+                }
+                unset($this->processes[$key]);
+            }
+        }
+    }
+
+    /**
+     * remove exited process
+     */
+    public function removeExitedProcess()
+    {
+        foreach ($this->processes as $key => $process) {
+            if ($process->isStopped()) {
+                unset($this->processes[$key]);
+            }
+        }
+    }
+
+    /**
+     * return process count
+     *
+     * @return int
+     */
+    public function count()
+    {
+        return count($this->processes);
+    }
+
+    /**
+     * get all processes
+     *
+     * @return Process[]
+     */
+    public function getProcesses()
+    {
+        return $this->processes;
+    }
 }
diff --git a/src/ParallelPool.php b/src/ParallelPool.php
@@ -103,24 +103,4 @@ public function start()
             }
         }
     }
-
-    /**
-     * return process count
-     *
-     * @return int
-     */
-    public function count()
-    {
-        return count($this->processes);
-    }
-
-    /**
-     * get all processes
-     *
-     * @return Process[]
-     */
-    public function getProcesses()
-    {
-        return $this->processes;
-    }
 }
diff --git a/src/Pool.php b/src/Pool.php
@@ -35,21 +35,4 @@ public function execute(Process $process, $name = null)
 
         return array_push($this->processes, $process);
     }
-
-    /**
-     * get process by name
-     *
-     * @param string $name process name
-     * @return Process|null
-     */
-    public function getProcessByName($name)
-    {
-        foreach ($this->processes as $process) {
-            if ($process->name() == $name) {
-                return $process;
-            }
-        }
-
-        return null;
-    }
 }