first commit

2022-08-19 19:48:37 +08:00
commit afdd648b65
3275 changed files with 631084 additions and 0 deletions
--- a/vendor/webman/redis-queue/src/RedisConnection.php
+++ b/vendor/webman/redis-queue/src/RedisConnection.php
@@ -0,0 +1,99 @@
+<?php
+/**
+ * This file is part of webman.
+ *
+ * Licensed under The MIT License
+ * For full copyright and license information, please see the MIT-LICENSE.txt
+ * Redistributions of files must retain the above copyright notice.
+ *
+ * @author    walkor<walkor@workerman.net>
+ * @copyright walkor<walkor@workerman.net>
+ * @link      http://www.workerman.net/
+ * @license   http://www.opensource.org/licenses/mit-license.php MIT License
+ */
+namespace Webman\RedisQueue;
+
+use Workerman\Timer;
+use Workerman\Worker;
+
+class RedisConnection extends \Redis
+{
+    /**
+     * @var array
+     */
+    protected $config = [];
+
+    /**
+     * @param array $config
+     * @return void
+     */
+    public function connectWithConfig(array $config = [])
+    {
+        static $timer;
+        if ($config) {
+            $this->config = $config;
+        }
+        if (false === $this->connect($this->config['host'], $this->config['port'], $this->config['timeout'] ?? 2)) {
+            throw new \RuntimeException("Redis connect {$this->config['host']}:{$this->config['port']} fail.");
+        }
+        if (!empty($this->config['auth'])) {
+            $this->auth($this->config['auth']);
+        }
+        if (!empty($this->config['db'])) {
+            $this->select($this->config['db']);
+        }
+        if (!empty($this->config['prefix'])) {
+            $this->setOption(\Redis::OPT_PREFIX, $this->config['prefix']);
+        }
+        if (Worker::getAllWorkers() && !$timer) {
+            $timer = Timer::add($this->config['ping'] ?? 55, function ()  {
+                $this->execCommand('ping');
+            });
+        }
+    }
+
+    /**
+     * @param $command
+     * @param ...$args
+     * @return mixed
+     * @throws \Throwable
+     */
+    protected function execCommand($command, ...$args)
+    {
+        try {
+            return $this->{$command}(...$args);
+        } catch (\Throwable $e) {
+            $msg = strtolower($e->getMessage());
+            if ($msg === 'connection lost' || strpos($msg, 'went away')) {
+                $this->connectWithConfig();
+                return $this->{$command}(...$args);
+            }
+            throw $e;
+        }
+    }
+
+    /**
+     * @param $queue
+     * @param $data
+     * @param $delay
+     * @return bool
+     */
+    public function send($queue, $data, $delay = 0)
+    {
+        $queue_waiting = '{redis-queue}-waiting';
+        $queue_delay = '{redis-queue}-delayed';
+        $now = time();
+        $package_str = json_encode([
+            'id'       => time().rand(),
+            'time'     => $now,
+            'delay'    => $delay,
+            'attempts' => 0,
+            'queue'    => $queue,
+            'data'     => $data
+        ]);
+        if ($delay) {
+            return (bool)$this->execCommand('zAdd' ,$queue_delay, $now + $delay, $package_str);
+        }
+        return (bool)$this->execCommand('lPush', $queue_waiting.$queue, $package_str);
+    }
+}