initial 2.0.0-a4 commit

2026-07-12 03:05:34 +08:00 · 2020-11-03 21:02:24 +08:00
parent da584e0542
commit 29fa9d8662
48 changed files with 794 additions and 1470 deletions
--- a/src/ZM/Utils/CoMessage.php
+++ b/src/ZM/Utils/CoMessage.php
@@ -0,0 +1,80 @@
+<?php
+
+
+namespace ZM\Utils;
+
+
+use Co;
+use Exception;
+use ZM\Store\LightCacheInside;
+use ZM\Store\Lock\SpinLock;
+use ZM\Store\ZMAtomic;
+
+class CoMessage
+{
+    /**
+     * @param array $hang
+     * @param array $compare
+     * @param int $timeout
+     * @return bool
+     * @throws Exception
+     */
+    public static function yieldByWS(array $hang, array $compare, $timeout = 600) {
+        $cid = Co::getuid();
+        $api_id = ZMAtomic::get("wait_msg_id")->add(1);
+        $hang["compare"] = $compare;
+        $hang["coroutine"] = $cid;
+        $hang["worker_id"] = server()->worker_id;
+        $hang["result"] = null;
+        SpinLock::lock("wait_api");
+        $wait = LightCacheInside::get("wait_api", "wait_api");
+        $wait[$api_id] = $hang;
+        LightCacheInside::set("wait_api", "wait_api", $wait);
+        SpinLock::unlock("wait_api");
+        $id = swoole_timer_after($timeout * 1000, function () use ($api_id) {
+            $r = LightCacheInside::get("wait_api", "wait_api")[$api_id] ?? null;
+            if (is_array($r)) {
+                Co::resume($r["coroutine"]);
+            }
+        });
+        Co::suspend();
+        SpinLock::lock("wait_api");
+        $sess = LightCacheInside::get("wait_api", "wait_api");
+        $result = $sess[$api_id]["result"];
+        unset($sess[$api_id]);
+        LightCacheInside::set("wait_api", "wait_api", $sess);
+        SpinLock::unlock("wait_api");
+        if (isset($id)) swoole_timer_clear($id);
+        if ($result === null) return false;
+        return $result;
+    }
+
+    public static function resumeByWS() {
+        $dat = ctx()->getData();
+        $last = null;
+        SpinLock::lock("wait_api");
+        $all = LightCacheInside::get("wait_api", "wait_api") ?? [];
+        foreach ($all as $k => $v) {
+            foreach ($v["compare"] as $vs) {
+                if ($v[$vs] != ($dat[$vs] ?? null)) {
+                    continue 2;
+                }
+            }
+            $last = $k;
+        }
+        if($last !== null) {
+            $all[$last]["result"] = $dat;
+            LightCacheInside::set("wait_api", "wait_api", $all);
+            SpinLock::unlock("wait_api");
+            if ($all[$last]["worker_id"] != server()->worker_id) {
+                ZMUtil::sendActionToWorker($all[$k]["worker_id"], "resume_ws_message", $all[$last]);
+            } else {
+                Co::resume($all[$last]["coroutine"]);
+            }
+            return true;
+        } else {
+            SpinLock::unlock("wait_api");
+            return false;
+        }
+    }
+}