522 lines
11 KiB
PHP
522 lines
11 KiB
PHP
<?php
|
||
/**
|
||
* +----------------------------------------------------------------------
|
||
* | CRMEB [ CRMEB赋能开发者，助力企业发展 ]
|
||
* +----------------------------------------------------------------------
|
||
* | Copyright (c) 2016~2022 https://www.crmeb.com All rights reserved.
|
||
* +----------------------------------------------------------------------
|
||
* | Licensed CRMEB并不是自由软件，未经许可不能去掉CRMEB相关版权
|
||
* +----------------------------------------------------------------------
|
||
* | Author: CRMEB Team <admin@crmeb.com>
|
||
* +----------------------------------------------------------------------
|
||
*/
|
||
|
||
namespace crmeb\traits\dao;
|
||
|
||
|
||
use crmeb\services\CacheService;
|
||
use crmeb\utils\Tag;
|
||
use think\cache\TagSet;
|
||
use think\Container;
|
||
use think\facade\Log;
|
||
use think\Model;
|
||
|
||
/**
|
||
* Trait CacheDaoTrait
|
||
* @package crmeb\traits\dao
|
||
* @method Model getModel()
|
||
* @method Model getPk()
|
||
*/
|
||
trait CacheDaoTrait
|
||
{
|
||
|
||
/**
|
||
* 获取redis
|
||
* @return \Redis
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/10/29
|
||
*/
|
||
private function getRedisConnect()
|
||
{
|
||
return CacheService::redisHandler()->handler();
|
||
}
|
||
|
||
/**
|
||
* 获取缓存
|
||
* @return TagSet|\think\facade\Cache
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/10
|
||
*/
|
||
private function getCacheHander()
|
||
{
|
||
return CacheService::redisHandler();
|
||
}
|
||
|
||
/**
|
||
* 对外开放方法
|
||
* @return TagSet|\think\facade\Cache
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/10
|
||
*/
|
||
public function cacheHander()
|
||
{
|
||
return $this->getCacheHander();
|
||
}
|
||
|
||
/**
|
||
* 缓存标签
|
||
* @param null $tag
|
||
* @return Tag
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/10
|
||
*/
|
||
public function cacheTag($tag = null)
|
||
{
|
||
$key = $this->cacheKey() . 'tag';
|
||
$tag = $tag ? $key . ':' . $tag : $key;
|
||
$redis = CacheService::redisHandler($tag);
|
||
|
||
return new Tag($redis, $tag);
|
||
}
|
||
|
||
/**
|
||
* 总缓存数据量
|
||
* @return mixed
|
||
*/
|
||
public function cacheCount()
|
||
{
|
||
$cacheKey = $this->cacheKey();
|
||
$rds = $this->getRedisConnect();
|
||
|
||
return $rds->hLen($cacheKey . 'map');
|
||
}
|
||
|
||
/**
|
||
* 读取缓存全部数据
|
||
* @return mixed
|
||
*/
|
||
public function cacheList(string $key = '')
|
||
{
|
||
$cacheKey = $this->cacheKey() . $key;
|
||
$rds = $this->getRedisConnect();
|
||
$map = $rds->hGetAll($cacheKey . 'map');
|
||
//key排序
|
||
ksort($map);
|
||
|
||
$list = array_values($map) ?: [];
|
||
|
||
foreach ($list as $key => $item) {
|
||
$list[$key] = $this->unserialize($item);
|
||
}
|
||
|
||
return $list;
|
||
}
|
||
|
||
/**
|
||
* 读取缓存分页数据
|
||
* @param int $page
|
||
* @param int $limit
|
||
* @param string $key
|
||
* @return array
|
||
*/
|
||
public function cachePageData(int $page = 1, int $limit = 10, string $key = '')
|
||
{
|
||
$cacheKey = $this->cacheKey() . $key;
|
||
$rds = $this->getRedisConnect();
|
||
|
||
$page = max($page, 1);
|
||
$limit = max($limit, 1);
|
||
|
||
//先读排序
|
||
$pageList = $rds->zRangeByScore($cacheKey . 'page', ($page - 1) * $limit, ($page * $limit) - 1);
|
||
|
||
//再读数据
|
||
$list = $rds->hMGet($cacheKey . 'map', $pageList) ?: [];
|
||
|
||
if (is_array($list)) {
|
||
$newList = [];
|
||
foreach ($list as $value) {
|
||
$newList[] = $this->unserialize($value);
|
||
}
|
||
$list = $newList;
|
||
}
|
||
|
||
$count = $rds->hLen($cacheKey . 'map');
|
||
|
||
return compact('list', 'count');
|
||
}
|
||
|
||
/**
|
||
* 单个查询数据
|
||
* @param int|string $id
|
||
* @return false|string|array
|
||
*/
|
||
public function cacheInfoById($id)
|
||
{
|
||
$cacheKey = $this->cacheKey();
|
||
$rds = $this->getRedisConnect();
|
||
|
||
$value = $rds->hGet($cacheKey . 'map', $id);
|
||
|
||
return $value === null ? null : $this->unserialize($value);
|
||
}
|
||
|
||
/**
|
||
* 批量查询数据
|
||
* @param $ids
|
||
* @return mixed
|
||
*/
|
||
public function cacheInfoByIds(array $ids)
|
||
{
|
||
$cacheKey = $this->cacheKey();
|
||
$rds = $this->getRedisConnect();
|
||
|
||
$arr = $rds->hMGet($cacheKey . 'map', $ids);
|
||
if (is_array($arr)) {
|
||
$newList = [];
|
||
foreach ($arr as $key => $value) {
|
||
$arr[$key] = $this->unserialize($value);
|
||
}
|
||
$arr = $newList;
|
||
}
|
||
|
||
return $arr;
|
||
}
|
||
|
||
/**
|
||
* 更新单个缓存
|
||
* @param $info
|
||
* @return false|mixed
|
||
*/
|
||
public function cacheUpdate(array $info, $key = null)
|
||
{
|
||
$pk = $this->getPk();
|
||
if ((empty($info) || !isset($info[$pk])) && !$key) {
|
||
return false;
|
||
}
|
||
$cacheKey = $this->cacheKey();
|
||
$rds = $this->getRedisConnect();
|
||
$key = $key ?: $info[$pk];
|
||
return $rds->hSet($cacheKey . 'map', $key, $this->serialize($info));
|
||
}
|
||
|
||
/**
|
||
* 序列化数据
|
||
* @param $value
|
||
* @return string
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/10
|
||
*/
|
||
private function serialize($value)
|
||
{
|
||
try {
|
||
return serialize($value);
|
||
} catch (\Throwable $e) {
|
||
Log::error('序列化发生错误:' . $e->getMessage());
|
||
return $value;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 反序列化数据
|
||
* @param $value
|
||
* @return mixed
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/10
|
||
*/
|
||
private function unserialize($value)
|
||
{
|
||
try {
|
||
return unserialize($value);
|
||
} catch (\Throwable $e) {
|
||
Log::error('反序列化发生错误:' . $e->getMessage());
|
||
return $value;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @param int $id
|
||
* @param $field
|
||
* @param null $value
|
||
* @return false|mixed
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/1
|
||
*/
|
||
public function cacheSaveValue(int $id, $field, $value = null)
|
||
{
|
||
$pk = $this->getPk();
|
||
$info = $this->cacheInfoById($id);
|
||
if (!$info) {
|
||
$newInfo = $this->get($id);
|
||
$info = $newInfo ? $newInfo->toArray() : [];
|
||
}
|
||
if (is_array($field)) {
|
||
foreach ($field as $k => $v) {
|
||
$info[$k] = $v;
|
||
}
|
||
} else {
|
||
$info[$field] = $value;
|
||
}
|
||
|
||
$info[$pk] = $id;
|
||
return $this->cacheUpdate($info);
|
||
}
|
||
|
||
/**
|
||
* 不存在则写入，存在则返回
|
||
* @param $key
|
||
* @param callable|null $fn
|
||
* @param null $default
|
||
* @return array|false|mixed|string|null
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/1
|
||
*/
|
||
public function cacheRemember($key, callable $fn = null, $default = null)
|
||
{
|
||
|
||
//不开启数据缓存直接返回
|
||
if (!app()->config->get('cache.is_data')) {
|
||
|
||
if ($fn instanceof \Closure) {
|
||
return Container::getInstance()->invokeFunction($fn);
|
||
} else {
|
||
return $default;
|
||
}
|
||
|
||
}
|
||
|
||
$info = $this->cacheInfoById($key);
|
||
|
||
if ((null === $info || false === $info) && is_callable($fn)) {
|
||
|
||
//读取数据库缓存
|
||
$newInfo = $fn();
|
||
|
||
if (null !== $newInfo) {
|
||
//缓存数据存在则更新
|
||
$this->cacheUpdate($newInfo, $key);
|
||
}
|
||
|
||
$info = $newInfo;
|
||
}
|
||
|
||
return null !== $info ? $info : $default;
|
||
}
|
||
|
||
/**
|
||
* 批量更新缓存
|
||
* @param $list
|
||
* @return false|mixed
|
||
*/
|
||
public function cacheUpdateList(array $list, string $key = '')
|
||
{
|
||
if (empty($list)) {
|
||
return false;
|
||
}
|
||
$cacheKey = $this->cacheKey() . $key;
|
||
$pk = $this->getPk();
|
||
$rds = $this->getRedisConnect();
|
||
$map = [];
|
||
foreach ($list as $item) {
|
||
if (empty($item) || !isset($item[$pk])) {
|
||
continue;
|
||
}
|
||
$map[$item[$pk]] = $this->serialize($item);
|
||
}
|
||
return $rds->hMSet($cacheKey . 'map', $map);
|
||
}
|
||
|
||
/**
|
||
* 删除单条缓存
|
||
* @param $id
|
||
*/
|
||
public function cacheDelById(int $id)
|
||
{
|
||
$cacheKey = $this->cacheKey();
|
||
$rds = $this->getRedisConnect();
|
||
$rds->hDel($cacheKey . 'map', $id);
|
||
$rds->zRem($cacheKey . 'page', $id);
|
||
}
|
||
|
||
/**
|
||
* 批量删除缓存
|
||
* @param $ids
|
||
*/
|
||
public function cacheDelByIds(array $ids)
|
||
{
|
||
$cacheKey = $this->cacheKey();
|
||
$rds = $this->getRedisConnect();
|
||
foreach ($ids as $id) {
|
||
$rds->hDel($cacheKey . 'map', $id);
|
||
$rds->zRem($cacheKey . 'page', $id);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 创建缓存
|
||
* @param array $list
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/10/29
|
||
*/
|
||
public function cacheCreate(array $list, string $key = '')
|
||
{
|
||
$pk = $this->getPk();
|
||
$cacheKey = $this->cacheKey() . $key;
|
||
|
||
$rds = $this->getRedisConnect();
|
||
|
||
//启动事务
|
||
$rds->multi();
|
||
|
||
//删除旧数据
|
||
$rds->del($cacheKey . 'map');
|
||
$rds->del($cacheKey . 'page');
|
||
//组合数据
|
||
$map = [];
|
||
foreach ($list as $i => $item) {
|
||
$map[$item[$pk]] = $item;
|
||
//存zset 排序
|
||
$rds->zAdd($cacheKey . 'page', $i, $item[$pk]);
|
||
}
|
||
foreach ($map as $k => &$item) {
|
||
$item = $this->serialize($item);
|
||
}
|
||
//存hmset 数据
|
||
$rds->hMSet($cacheKey . 'map', $map);
|
||
|
||
//执行事务
|
||
$rds->exec();
|
||
}
|
||
|
||
protected function cacheKey()
|
||
{
|
||
return 'mc:' . $this->getModel()->getName() . ':';
|
||
}
|
||
|
||
/**
|
||
*
|
||
* @param $key
|
||
* @return string
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/10
|
||
*/
|
||
public function getCacheKey($key)
|
||
{
|
||
return $this->cacheKey() . $key;
|
||
}
|
||
|
||
/**
|
||
* 更新缓存
|
||
* @param string $key
|
||
* @param $value
|
||
* @return bool
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/1
|
||
*/
|
||
public function cacheStrUpdate(string $key, $value, int $expire = null)
|
||
{
|
||
return $this->getCacheHander()->set($this->cacheKey() . 'str:' . $key, $value, $expire);
|
||
}
|
||
|
||
/**
|
||
* 获取缓存
|
||
* @param string $key
|
||
* @return false|mixed|string
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/1
|
||
*/
|
||
public function cacheStrGet(string $key)
|
||
{
|
||
return $this->getCacheHander()->get($this->cacheKey() . 'str:' . $key);
|
||
}
|
||
|
||
/**
|
||
* 获取表缓存是否有数据
|
||
* @return false|mixed|string
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/1
|
||
*/
|
||
public function cacheStrTable()
|
||
{
|
||
return $this->getRedisConnect()->get($this->cacheKey());
|
||
}
|
||
|
||
/**
|
||
* 设置表缓存是否有数据
|
||
* @param int $value
|
||
* @return bool
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/1
|
||
*/
|
||
public function cacheStrSetTable(int $value = 1)
|
||
{
|
||
return $this->getRedisConnect()->set($this->cacheKey(), $value);
|
||
}
|
||
|
||
/**
|
||
* 删除缓存
|
||
* @param string $key
|
||
* @return int
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/1
|
||
*/
|
||
public function cacheStrDel(string $key)
|
||
{
|
||
return $this->getCacheHander()->delete($this->cacheKey() . 'str:' . $key);
|
||
}
|
||
|
||
/**
|
||
* 获取缓存,没有则写入缓存并返回
|
||
* @param string $key
|
||
* @param callable|null $fn
|
||
* @param null $default
|
||
* @return false|mixed|string|null
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2022/11/1
|
||
*/
|
||
public function cacheStrRemember(string $key, callable $fn = null, int $expire = null, $default = null)
|
||
{
|
||
|
||
//不开启数据缓存直接返回
|
||
if (!app()->config->get('cache.is_data')) {
|
||
|
||
if ($fn instanceof \Closure) {
|
||
return Container::getInstance()->invokeFunction($fn);
|
||
} else {
|
||
return $default;
|
||
}
|
||
}
|
||
|
||
$value = $this->cacheStrGet($key);
|
||
|
||
if ((null === $value || false === $value) && is_callable($fn)) {
|
||
|
||
$newValue = $fn();
|
||
|
||
if (null !== $newValue) {
|
||
$this->cacheStrUpdate($key, $value, $expire);
|
||
}
|
||
|
||
$value = $newValue;
|
||
}
|
||
|
||
return null !== $value ? $value : $default;
|
||
}
|
||
}
|