Automatically run global loop in a shutdown function

reactphp · Apr 25, 2017 · 4d5e850 · 4d5e850
1 parent c52f792
commit 4d5e850
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/GlobalLoop.php b/src/GlobalLoop.php
@@ -12,6 +12,7 @@ final class GlobalLoop
     private static $factory;
 
     private static $didRun = false;
+    private static $disableRunOnShutdown = false;
 
     public static function setFactory(callable $factory = null)
     {
@@ -24,6 +25,11 @@ public static function setFactory(callable $factory = null)
         self::$factory = $factory;
     }
 
+    public function disableRunOnShutdown()
+    {
+        self::$disableRunOnShutdown = true;
+    }
+
     /**
      * @return LoopInterface
      */
@@ -33,6 +39,14 @@ public static function get()
             return self::$loop;
         }
 
+        register_shutdown_function(function () {
+            if (self::$disableRunOnShutdown || self::$didRun || !self::$loop) {
+                return;
+            }
+
+            self::$loop->run();
+        });
+
         self::$loop = self::create();
 
         self::$loop->futureTick(function () {