初始化

src/conduit/driver/socket/Server.php

@@ -0,0 +1,102 @@
+<?php
+
+namespace think\worker\conduit\driver\socket;
+
+use think\worker\Worker;
+use Workerman\Connection\TcpConnection;
+use Workerman\Protocols\Frame;
+
+class Server
+{
+    protected $data = [];
+
+    /** @var array<string,TcpConnection[]> */
+    protected $subscribers = [];
+
+    public function onMessage(TcpConnection $connection, $buffer)
+    {
+        if (empty($buffer)) {
+            return;
+        }
+        /** @var Command $command */
+        $command = unserialize($buffer);
+
+        $result = Result::create($command->id);
+
+        switch ($command->name) {
+            case 'get':
+                $result->data = $this->data[$command->key] ?? null;
+                break;
+            case 'set':
+                $this->data[$command->key] = $command->data;
+                break;
+            case 'inc':
+                if (!isset($this->data[$command->key]) || !is_integer($this->data[$command->key])) {
+                    $this->data[$command->key] = 0;
+                }
+                $result->data = $this->data[$command->key] += $command->data ?? 1;
+                break;
+            case 'sAdd':
+                if (!isset($this->data[$command->key]) || !is_array($this->data[$command->key])) {
+                    $this->data[$command->key] = [];
+                }
+                $this->data[$command->key] = array_merge($this->data[$command->key], $command->data);
+                break;
+            case 'sRem':
+                if (!isset($this->data[$command->key]) || !is_array($this->data[$command->key])) {
+                    $this->data[$command->key] = [];
+                }
+                $this->data[$command->key] = array_diff($this->data[$command->key], [$command->data]);
+                break;
+            case 'sMembers':
+                if (!isset($this->data[$command->key]) || !is_array($this->data[$command->key])) {
+                    $this->data[$command->key] = [];
+                }
+                $result->data = $this->data[$command->key];
+                break;
+            case 'subscribe':
+                if (!isset($this->subscribers[$command->key])) {
+                    $this->subscribers[$command->key] = [];
+                }
+                $this->subscribers[$command->key][] = $connection;
+                break;
+            case 'publish':
+                if (!empty($this->subscribers[$command->key])) {
+                    foreach ($this->subscribers[$command->key] as $conn) {
+                        $conn->send(serialize(Event::create($command->key, $command->data)));
+                    }
+                }
+                break;
+        }
+
+        if (isset($result->id)) {
+            $connection->send(serialize($result));
+        }
+    }
+
+    public function onClose(TcpConnection $connection)
+    {
+        if (!empty($this->subscribers)) {
+            foreach ($this->subscribers as $key => $connections) {
+                $this->subscribers[$key] = array_udiff($connections, [$connection], function ($a, $b) {
+                    return $a <=> $b;
+                });
+            }
+        }
+    }
+
+    public static function run($domain)
+    {
+        //启动服务端
+        $server = new self();
+
+        $worker = new Worker($domain);
+
+        $worker->name       = 'conduit';
+        $worker->protocol   = Frame::class;
+        $worker->reloadable = false;
+
+        $worker->onMessage = [$server, 'onMessage'];
+        $worker->onClose   = [$server, 'onClose'];
+    }
+}