add

2025-08-13 18:31:52 +08:00
parent 17ad88608b
commit 275713f893
77 changed files with 10658 additions and 178 deletions
--- a/app/common/library/token/driver/Redis.php
+++ b/app/common/library/token/driver/Redis.php
@@ -0,0 +1,146 @@
+<?php
+
+namespace app\common\library\token\driver;
+
+use Throwable;
+use BadFunctionCallException;
+use app\common\library\token\Driver;
+
+/**
+ * @see Driver
+ */
+class Redis extends Driver
+{
+    /**
+     * 默认配置
+     * @var array
+     */
+    protected array $options = [];
+
+    /**
+     * Token 过期后缓存继续保留的时间(s)
+     */
+    protected int $expiredHold = 60 * 60 * 24 * 2;
+
+    /**
+     * 构造函数
+     * @access public
+     * @param array $options 参数
+     * @throws Throwable
+     */
+    public function __construct(array $options = [])
+    {
+        if (!extension_loaded('redis')) {
+            throw new BadFunctionCallException('未安装redis扩展');
+        }
+        if (!empty($options)) {
+            $this->options = array_merge($this->options, $options);
+        }
+        $this->handler = new \Redis();
+        if ($this->options['persistent']) {
+            $this->handler->pconnect($this->options['host'], $this->options['port'], $this->options['timeout'], 'persistent_id_' . $this->options['select']);
+        } else {
+            $this->handler->connect($this->options['host'], $this->options['port'], $this->options['timeout']);
+        }
+
+        if ('' != $this->options['password']) {
+            $this->handler->auth($this->options['password']);
+        }
+
+        if (false !== $this->options['select']) {
+            $this->handler->select($this->options['select']);
+        }
+    }
+
+    /**
+     * @throws Throwable
+     */
+    public function set(string $token, string $type, int $userId, int $expire = null): bool
+    {
+        if (is_null($expire)) {
+            $expire = $this->options['expire'];
+        }
+        $expireTime = $expire !== 0 ? time() + $expire : 0;
+        $token      = $this->getEncryptedToken($token);
+        $tokenInfo  = [
+            'token'       => $token,
+            'type'        => $type,
+            'user_id'     => $userId,
+            'create_time' => time(),
+            'expire_time' => $expireTime,
+        ];
+        $tokenInfo  = json_encode($tokenInfo, JSON_UNESCAPED_UNICODE);
+        if ($expire) {
+            $expire += $this->expiredHold;
+            $result = $this->handler->setex($token, $expire, $tokenInfo);
+        } else {
+            $result = $this->handler->set($token, $tokenInfo);
+        }
+        $this->handler->sAdd($this->getUserKey($type, $userId), $token);
+        return $result;
+    }
+
+    /**
+     * @throws Throwable
+     */
+    public function get(string $token): array
+    {
+        $key  = $this->getEncryptedToken($token);
+        $data = $this->handler->get($key);
+        if (is_null($data) || false === $data) {
+            return [];
+        }
+        $data = json_decode($data, true);
+
+        $data['token']      = $token; // 返回未加密的token给客户端使用
+        $data['expires_in'] = $this->getExpiredIn($data['expire_time'] ?? 0); // 过期时间
+        return $data;
+    }
+
+    /**
+     * @throws Throwable
+     */
+    public function check(string $token, string $type, int $userId): bool
+    {
+        $data = $this->get($token);
+        if (!$data || ($data['expire_time'] && $data['expire_time'] <= time())) return false;
+        return $data['type'] == $type && $data['user_id'] == $userId;
+    }
+
+    /**
+     * @throws Throwable
+     */
+    public function delete(string $token): bool
+    {
+        $data = $this->get($token);
+        if ($data) {
+            $key = $this->getEncryptedToken($token);
+            $this->handler->del($key);
+            $this->handler->sRem($this->getUserKey($data['type'], $data['user_id']), $key);
+        }
+        return true;
+    }
+
+    /**
+     * @throws Throwable
+     */
+    public function clear(string $type, int $userId): bool
+    {
+        $userKey = $this->getUserKey($type, $userId);
+        $keys    = $this->handler->sMembers($userKey);
+        $this->handler->del($userKey);
+        $this->handler->del($keys);
+        return true;
+    }
+
+    /**
+     * 获取会员的key
+     * @param $type
+     * @param $userId
+     * @return string
+     */
+    protected function getUserKey($type, $userId): string
+    {
+        return $this->options['prefix'] . $type . '-' . $userId;
+    }
+}