Add KCP support

src/Node/NodeConnection.php

@@ -4,7 +4,7 @@ declare(strict_types=1);
 
 namespace LayLink\Node;
 
-use Workerman\Connection\TcpConnection;
+use LayLink\Transport\FrameServerConnection;
 
 final class NodeConnection
 {
@@ -15,7 +15,7 @@ final class NodeConnection
         public readonly string $nodeId,
         public readonly string $nodeType,
         public readonly string $nodeZone,
-        public readonly TcpConnection $connection,
+        public readonly FrameServerConnection $connection,
     ) {
         $this->lastHeartbeat = time();
     }

src/Node/NodeRegistry.php

@@ -4,23 +4,30 @@ declare(strict_types=1);
 
 namespace LayLink\Node;
 
-use Workerman\Connection\TcpConnection;
+use LayLink\Transport\FrameServerConnection;
 
 final class NodeRegistry
 {
-    /** @var array<string, NodeConnection> */
+    /** @var array<string, array<int, NodeConnection>> */
     private array $nodes = [];
 
     public function register(NodeConnection $node): void
     {
-        $this->nodes[$node->nodeId] = $node;
+        $this->nodes[$node->nodeId][$node->connection->id()] = $node;
     }
 
-    public function unregisterByConnection(TcpConnection $connection): ?NodeConnection
+    public function unregisterByConnection(FrameServerConnection $connection): ?NodeConnection
     {
-        foreach ($this->nodes as $nodeId => $node) {
-            if ($node->connection === $connection) {
-                unset($this->nodes[$nodeId]);
+        foreach ($this->nodes as $nodeId => $connections) {
+            foreach ($connections as $connectionId => $node) {
+                if ($node->connection !== $connection) {
+                    continue;
+                }
+
+                unset($this->nodes[$nodeId][$connectionId]);
+                if ($this->nodes[$nodeId] === []) {
+                    unset($this->nodes[$nodeId]);
+                }
                 return $node;
             }
         }
@@ -30,12 +37,26 @@ final class NodeRegistry
 
     public function get(string $nodeId): ?NodeConnection
     {
-        return $this->nodes[$nodeId] ?? null;
+        $connections = $this->nodes[$nodeId] ?? [];
+        return $connections === [] ? null : array_values($connections)[0];
+    }
+
+    public function getByConnection(FrameServerConnection $connection): ?NodeConnection
+    {
+        foreach ($this->nodes as $connections) {
+            foreach ($connections as $node) {
+                if ($node->connection === $connection) {
+                    return $node;
+                }
+            }
+        }
+
+        return null;
     }
 
     public function isOnline(string $nodeId): bool
     {
-        return isset($this->nodes[$nodeId]);
+        return ($this->nodes[$nodeId] ?? []) !== [];
     }
 
     /**
@@ -43,6 +64,13 @@ final class NodeRegistry
      */
     public function all(): array
     {
-        return array_values($this->nodes);
+        $nodes = [];
+        foreach ($this->nodes as $connections) {
+            foreach ($connections as $node) {
+                $nodes[] = $node;
+            }
+        }
+
+        return $nodes;
     }
 }