Feat- worker start callback

Dockerfile

@@ -1,4 +1,4 @@
-FROM composer:2.0 as composer
+FROM composer:2.0 AS composer
 
 ARG TESTING=false
 ENV TESTING=$TESTING
@@ -11,7 +11,7 @@ COPY composer.json /usr/local/src/
 RUN composer install --ignore-platform-reqs --optimize-autoloader \
     --no-plugins --no-scripts --prefer-dist
 
-FROM appwrite/base:0.9.0 as final
+FROM appwrite/base:0.9.0 AS final
 
 LABEL maintainer="team@appwrite.io"
 

docker-compose.yml

@@ -1,5 +1,3 @@
-version: '3'
-
 services:
   dns-server:
     image: dns-dev

src/DNS/Adapter.php

@@ -22,4 +22,15 @@ abstract public function start(): void;
      * @return string
      */
     abstract public function getName(): string;
+
+    /**
+     * Worker Start Callback (optional for adapters that support it)
+     *
+     * @param callable $callback
+     */
+    public function onWorkerStart(callable $callback): void
+    {
+        // Default implementation does nothing
+        // Adapters that support worker events should override this
+    }
 }

src/DNS/Adapter/Native.php

@@ -69,4 +69,19 @@ public function getName(): string
     {
         return 'native';
     }
+
+    /**
+     * Worker start callback
+     *
+     * @param callable $callback
+     * @return void
+     */
+    public function onWorkerStart(callable $callback): void
+    {
+        if (is_callable($callback)) {
+            call_user_func($callback, $this->server, 0); // Pass dummy worker ID
+        } else {
+            throw new Exception('Worker start callback must be callable.');
+        }
+    }
 }

src/DNS/Adapter/Swoole.php

@@ -10,6 +10,7 @@ class Swoole extends Adapter
     protected Server $server;
     protected string $host;
     protected int $port;
+    protected $workerStartCallback = null;
 
     public function __construct(string $host = '0.0.0.0', int $port = 53)
     {
@@ -32,6 +33,29 @@ public function onPacket(callable $callback): void
         });
     }
 
+    /**
+     * Set callback for worker start event
+     *
+     * @param callable $callback
+     */
+    public function onWorkerStart(callable $callback): void
+    {
+        $this->workerStartCallback = $callback;
+        $this->server->on('WorkerStart', function ($server, $workerId) use ($callback) {
+            call_user_func($callback, $server, $workerId);
+        });
+    }
+
+    /**
+     * Get the server instance for configuration
+     *
+     * @return Server
+     */
+    public function getServer(): Server
+    {
+        return $this->server;
+    }
+
     /**
      * Start the DNS server
      */

src/DNS/Server.php

@@ -154,6 +154,10 @@ public function start(): void
 
             Console::success('[DNS] Server is ready to accept connections');
 
+            $this->adapter->onWorkerStart(function ($server, $workerId) {
+                Console::info("[WORKER] Worker #{$workerId} - PID: " . getmypid() .  " - Started at " . date('Y-m-d H:i:s T') . ' Memory: ' . round(memory_get_usage(true) / 1024 / 1024, 2) . "MB");
+            });
+
             $this->adapter->onPacket(function (string $buffer, string $ip, int $port) {
                 $startTime = microtime(true);
                 try {