initial commit

2026-07-21 15:45:36 +08:00 · 2020-03-02 16:14:20 +08:00
parent 517082d159
commit 769b87af6a
132 changed files with 5424 additions and 2866 deletions
--- a/src/ZM/Event/Swoole/MessageEvent.php
+++ b/src/ZM/Event/Swoole/MessageEvent.php
@@ -0,0 +1,94 @@
+<?php
+
+
+namespace ZM\Event\Swoole;
+
+
+use Closure;
+use Framework\Console;
+use Framework\ZMBuf;
+use Swoole\WebSocket\Frame;
+use Swoole\WebSocket\Server;
+use ZM\Annotation\Swoole\SwooleEventAfter;
+use ZM\Annotation\Swoole\SwooleEventAt;
+use ZM\Connection\ConnectionManager;
+use Exception;
+use ZM\Event\EventHandler;
+use ZM\ModBase;
+use ZM\ModHandleType;
+use ZM\Utils\ZMUtil;
+
+class MessageEvent implements SwooleEvent
+{
+    /**
+     * @var Server
+     */
+    public $server;
+    /**
+     * @var Frame
+     */
+    public $frame;
+
+    public function __construct(Server $server, Frame $frame) {
+        $this->server = $server;
+        $this->frame = $frame;
+    }
+
+    /**
+     * @inheritDoc
+     */
+    public function onActivate() {
+        ZMUtil::checkWait();
+        try {
+            if (ConnectionManager::get($this->frame->fd)->getType() == "qq") {
+                $data = json_decode($this->frame->data, true);
+                if (isset($data["post_type"]))
+                    EventHandler::callCQEvent($data, $this, 0);
+                else
+                    EventHandler::callCQResponse($data);
+            }
+            foreach (ZMBuf::$events[SwooleEventAt::class] ?? [] as $v) {
+                if (strtolower($v->type) == "message" && $this->parseSwooleRule($v)) {
+                    $conn = ConnectionManager::get($this->frame->fd);
+                    $c = $v->class;
+                    /** @var ModBase $class */
+                    $class = new $c(["server" => $this->server, "frame" => $this->frame, "connection" => $conn], ModHandleType::SWOOLE_MESSAGE);
+                    call_user_func_array([$class, $v->method], [$conn]);
+                    if ($class->block_continue) break;
+                }
+            }
+        } catch (Exception $e) {
+            Console::error("出现错误: " . $e->getMessage());
+        }
+        return $this;
+    }
+
+    /**
+     * @inheritDoc
+     */
+    public function onAfter() {
+        foreach (ZMBuf::$events[SwooleEventAfter::class] ?? [] as $v) {
+            if (strtolower($v->type) == "message" && $this->parseSwooleRule($v) === true) {
+                $conn = ConnectionManager::get($this->frame->fd);
+                $c = $v->class;
+                /** @var ModBase $class */
+                $class = new $c(["server" => $this->server, "frame" => $this->frame, "connection" => $conn], ModHandleType::SWOOLE_MESSAGE);
+                call_user_func_array([$class, $v->method], []);
+                if ($class->block_continue) break;
+            }
+        }
+        return $this;
+    }
+
+    private function parseSwooleRule($v) {
+        switch (explode(":", $v->rule)[0]) {
+            case "connectType": //websocket连接类型
+                if ($v->callback instanceof Closure) return call_user_func($v->callback, ConnectionManager::get($this->frame->fd));
+                break;
+            case "dataEqual": //handle websocket message事件时才能用
+                if ($v->callback instanceof Closure) return call_user_func($v->callback, $this->frame->data);
+                break;
+        }
+        return true;
+    }
+}