Support to reload .env when using hyperf/watcher. (#6936)

Co-authored-by: 李铭昕 <715557344@qq.com>

Author: albertcht

src/ConfigProvider.php

@@ -13,6 +13,7 @@
 namespace Hyperf\Watcher;
 
 use Hyperf\Watcher\Command\WatchCommand;
+use Hyperf\Watcher\Listener\ReloadDotenvAndConfigListener;
 
 class ConfigProvider
 {
@@ -24,6 +25,9 @@ public function __invoke(): array
             'commands' => [
                 WatchCommand::class,
             ],
+            'listeners' => [
+                ReloadDotenvAndConfigListener::class,
+            ],
             'publish' => [
                 [
                     'id' => 'config',

src/Event/BeforeServerRestart.php

@@ -0,0 +1,20 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of Hyperf.
+ *
+ * @link     https://www.hyperf.io
+ * @document https://hyperf.wiki
+ * @contact  group@hyperf.io
+ * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
+ */
+
+namespace Hyperf\Watcher\Event;
+
+class BeforeServerRestart
+{
+    public function __construct(public string $pid)
+    {
+    }
+}

src/Listener/ReloadDotenvAndConfigListener.php

@@ -0,0 +1,69 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of Hyperf.
+ *
+ * @link     https://www.hyperf.io
+ * @document https://hyperf.wiki
+ * @contact  group@hyperf.io
+ * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
+ */
+
+namespace Hyperf\Watcher\Listener;
+
+use Hyperf\Contract\ConfigInterface;
+use Hyperf\Event\Contract\ListenerInterface;
+use Hyperf\Framework\Event\BeforeWorkerStart;
+use Hyperf\Support\DotenvManager;
+use Hyperf\Watcher\Event\BeforeServerRestart;
+use Psr\Container\ContainerInterface;
+use Swoole\Atomic;
+
+class ReloadDotenvAndConfigListener implements ListenerInterface
+{
+    protected static Atomic $restartCounter;
+
+    public function __construct(protected ContainerInterface $container)
+    {
+        static::$restartCounter = new Atomic(0);
+    }
+
+    public function listen(): array
+    {
+        return [
+            BeforeWorkerStart::class,
+            BeforeServerRestart::class,
+        ];
+    }
+
+    public function process(object $event): void
+    {
+        if ($event instanceof BeforeWorkerStart
+            && $event->workerId === 0
+            && static::$restartCounter->get() === 0
+        ) {
+            static::$restartCounter->add();
+            return;
+        }
+
+        static::$restartCounter->add();
+
+        $this->reloadDotenv();
+        $this->reloadConfig();
+    }
+
+    protected function reloadConfig(): void
+    {
+        if (! method_exists($this->container, 'unbind')) {
+            return;
+        }
+
+        $this->container->unbind(ConfigInterface::class);
+    }
+
+    protected function reloadDotenv(): void
+    {
+        DotenvManager::reload([BASE_PATH]);
+    }
+}

src/Watcher.php

@@ -20,8 +20,10 @@
 use Hyperf\Support\Filesystem\FileNotFoundException;
 use Hyperf\Support\Filesystem\Filesystem;
 use Hyperf\Watcher\Driver\DriverInterface;
+use Hyperf\Watcher\Event\BeforeServerRestart;
 use PhpParser\PrettyPrinter\Standard;
 use Psr\Container\ContainerInterface;
+use Psr\EventDispatcher\EventDispatcherInterface;
 use Symfony\Component\Console\Output\OutputInterface;
 use Throwable;
 
@@ -109,6 +111,8 @@ public function restart($isStart = true)
             $pid = $this->filesystem->get($file);
             try {
                 $this->output->writeln('Stop server...');
+                $this->container->get(EventDispatcherInterface::class)
+                    ->dispatch(new BeforeServerRestart($pid));
                 if (posix_kill((int) $pid, 0)) {
                     posix_kill((int) $pid, SIGTERM);
                 }