初始化

src/websocket/Handler.php

@@ -0,0 +1,71 @@
+<?php
+
+namespace think\worker\websocket;
+
+use think\Event;
+use think\Request;
+use think\worker\contract\websocket\HandlerInterface;
+use think\worker\websocket\Event as WsEvent;
+
+class Handler implements HandlerInterface
+{
+    protected $event;
+
+    public function __construct(Event $event)
+    {
+        $this->event = $event;
+    }
+
+    /**
+     * "onOpen" listener.
+     *
+     * @param Request $request
+     */
+    public function onOpen(Request $request)
+    {
+        $this->event->trigger('worker.websocket.Open', $request);
+    }
+
+    /**
+     * "onMessage" listener.
+     *
+     * @param Frame $frame
+     */
+    public function onMessage(Frame $frame)
+    {
+        $this->event->trigger('worker.websocket.Message', $frame);
+
+        $event = $this->decode($frame->data);
+        if ($event) {
+            $this->event->trigger('worker.websocket.Event', $event);
+        }
+    }
+
+    /**
+     * "onClose" listener.
+     */
+    public function onClose()
+    {
+        $this->event->trigger('worker.websocket.Close');
+    }
+
+    protected function decode($payload)
+    {
+        $data = json_decode($payload, true);
+        if (!empty($data['type'])) {
+            return new WsEvent($data['type'], $data['data'] ?? null);
+        }
+        return null;
+    }
+
+    public function encodeMessage($message)
+    {
+        if ($message instanceof WsEvent) {
+            return json_encode([
+                'type' => $message->type,
+                'data' => $message->data,
+            ]);
+        }
+        return $message;
+    }
+}