From 6801490e3b5b32b97c2ef92df9107900af13a25c Mon Sep 17 00:00:00 2001
From: 疯狂的狮子li <15040126243@163.com>
Date: 星期四, 02 九月 2021 11:29:50 +0800
Subject: [PATCH] update 分页合理化

---
 ruoyi-framework/src/main/java/com/ruoyi/framework/config/RedisConfig.java |   30 ++++++++++++++++++++++++++++++
 1 files changed, 30 insertions(+), 0 deletions(-)

diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/RedisConfig.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/RedisConfig.java
index 307b9a6..d66d86b 100644
--- a/ruoyi-framework/src/main/java/com/ruoyi/framework/config/RedisConfig.java
+++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/config/RedisConfig.java
@@ -1,5 +1,6 @@
 package com.ruoyi.framework.config;
 
+import cn.hutool.core.util.StrUtil;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.framework.config.properties.RedissonProperties;
 import org.redisson.Redisson;
@@ -16,6 +17,7 @@
 import org.springframework.cache.annotation.EnableCaching;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.data.redis.core.script.DefaultRedisScript;
 
 import java.io.IOException;
 import java.util.HashMap;
@@ -89,4 +91,32 @@
 		return new RedissonSpringCacheManager(redissonClient, config, JsonJacksonCodec.INSTANCE);
 	}
 
+	@Bean
+	public DefaultRedisScript<Long> limitScript() {
+		DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
+		redisScript.setScriptText(limitScriptText());
+		redisScript.setResultType(Long.class);
+		return redisScript;
+	}
+
+	/**
+	 * 闄愭祦鑴氭湰
+	 */
+	private String limitScriptText() {
+		return StrUtil.builder()
+			.append("local key = KEYS[1]\n")
+			.append("local count = tonumber(ARGV[1])\n")
+			.append("local time = tonumber(ARGV[2])\n")
+			.append("local current = redis.call('get', key);\n")
+			.append("if current and tonumber(current) > count then\n")
+			.append("    return current;\n")
+			.append("end\n")
+			.append("current = redis.call('incr', key)\n")
+			.append("if tonumber(current) == 1 then\n")
+			.append("    redis.call('expire', key, time)\n")
+			.append("end\n")
+			.append("return current;")
+			.toString();
+	}
+
 }

--
Gitblit v1.9.3