Redis+Lua腳本實現(xiàn)計數(shù)器接口防刷功能(升級版)
更新時間：2022年02月09日 16:34:03 作者：當年的春天
這篇文章主要介紹了Redis+Lua腳本實現(xiàn)計數(shù)器接口防刷功能,使用腳本使得set命令和expire命令一同達到Redis被執(zhí)行且不會被干擾，在很大程度上保證了原子操作，對Redis實現(xiàn)計數(shù)器接口防刷功能感興趣的朋友一起看看吧
【前言】
Cash Loan（一）:Redis實現(xiàn)計數(shù)器防刷 中介紹了項目中應用redis來做計數(shù)器的實現(xiàn)過程，最近自己看了些關于Redis實現(xiàn)分布式鎖的代碼后，發(fā)現(xiàn)在Redis分布式鎖中出現(xiàn)一個問題在這版計數(shù)器中同樣會出現(xiàn)，于是融入了Lua腳本進行升級改造有了Redis+Lua版本。
【實現(xiàn)過程】
一、問題分析
如果set命令設置上，但是在設置失效時間時由于網絡抖動等原因導致沒有設置成功，這時就會出現(xiàn)死計數(shù)器（類似死鎖）；
二、解決方案
Redis+Lua是一個很好的解決方案，使用腳本使得set命令和expire命令一同達到Redis被執(zhí)行且不會被干擾，在很大程度上保證了原子操作；
為什么說是很大程度上保證原子操作而不是完全保證？因為在Redis內部執(zhí)行的時候出問題也有可能出現(xiàn)問題不過概率非常?。患词贯槍π「怕适录灿邢鄳慕鉀Q方案，比如解決死鎖一個思路值得參考：防止死鎖會將鎖的值存成一個時間戳，即使發(fā)生沒有將失效時間設置上在判斷是否上鎖時可以加上看看其中值距現(xiàn)在是否超過一個設定的時間，如果超過則將其刪除重新設置鎖。
三、代碼改造
1、Redis+Lua鎖的實現(xiàn)
package han.zhang.utils;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DigestUtils;
import org.springframework.data.redis.core.script.RedisScript;
import java.util.Collections;
import java.util.UUID;
public class RedisLock {
private static final LogUtils logger = LogUtils.getLogger(RedisLock.class);
private final StringRedisTemplate stringRedisTemplate;
private final String lockKey;
private final String lockValue;
private boolean locked = false;
/**
* 使用腳本在redis服務器執(zhí)行這個邏輯可以在一定程度上保證此操作的原子性
* （即不會發(fā)生客戶端在執(zhí)行setNX和expire命令之間，發(fā)生崩潰或失去與服務器的連接導致expire沒有得到執(zhí)行，發(fā)生永久死鎖）
* <p>
* 除非腳本在redis服務器執(zhí)行時redis服務器發(fā)生崩潰，不過此種情況鎖也會失效
*/
private static final RedisScript<Boolean> SETNX_AND_EXPIRE_SCRIPT;
static {
StringBuilder sb = new StringBuilder();
sb.append("if (redis.call('setnx', KEYS[1], ARGV[1]) == 1) then\n");
sb.append("\tredis.call('expire', KEYS[1], tonumber(ARGV[2]))\n");
sb.append("\treturn true\n");
sb.append("else\n");
sb.append("\treturn false\n");
sb.append("end");
SETNX_AND_EXPIRE_SCRIPT = new RedisScriptImpl<>(sb.toString(), Boolean.class);
}
private static final RedisScript<Boolean> DEL_IF_GET_EQUALS;
sb.append("if (redis.call('get', KEYS[1]) == ARGV[1]) then\n");
sb.append("\tredis.call('del', KEYS[1])\n");
DEL_IF_GET_EQUALS = new RedisScriptImpl<>(sb.toString(), Boolean.class);
public RedisLock(StringRedisTemplate stringRedisTemplate, String lockKey) {
this.stringRedisTemplate = stringRedisTemplate;
this.lockKey = lockKey;
this.lockValue = UUID.randomUUID().toString() + "." + System.currentTimeMillis();
private boolean doTryLock(int lockSeconds) {
if (locked) {
throw new IllegalStateException("already locked!");
}
locked = stringRedisTemplate.execute(SETNX_AND_EXPIRE_SCRIPT, Collections.singletonList(lockKey), lockValue,
String.valueOf(lockSeconds));
return locked;
* 嘗試獲得鎖，成功返回true，如果失敗立即返回false
*
* @param lockSeconds 加鎖的時間(秒)，超過這個時間后鎖會自動釋放
public boolean tryLock(int lockSeconds) {
try {
return doTryLock(lockSeconds);
} catch (Exception e) {
logger.error("tryLock Error", e);
return false;
* 輪詢的方式去獲得鎖，成功返回true，超過輪詢次數(shù)或異常返回false
* @param lockSeconds 加鎖的時間(秒)，超過這個時間后鎖會自動釋放
* @param tryIntervalMillis 輪詢的時間間隔(毫秒)
* @param maxTryCount 最大的輪詢次數(shù)
public boolean tryLock(final int lockSeconds, final long tryIntervalMillis, final int maxTryCount) {
int tryCount = 0;
while (true) {
if (++tryCount >= maxTryCount) {
// 獲取鎖超時
return false;
}
try {
if (doTryLock(lockSeconds)) {
return true;
}
} catch (Exception e) {
logger.error("tryLock Error", e);
Thread.sleep(tryIntervalMillis);
} catch (InterruptedException e) {
logger.error("tryLock interrupted", e);
* 解鎖操作
public void unlock() {
if (!locked) {
throw new IllegalStateException("not locked yet!");
locked = false;
// 忽略結果
stringRedisTemplate.execute(DEL_IF_GET_EQUALS, Collections.singletonList(lockKey), lockValue);
private static class RedisScriptImpl<T> implements RedisScript<T> {
private final String script;
private final String sha1;
private final Class<T> resultType;
public RedisScriptImpl(String script, Class<T> resultType) {
this.script = script;
this.sha1 = DigestUtils.sha1DigestAsHex(script);
this.resultType = resultType;
@Override
public String getSha1() {
return sha1;
public Class<T> getResultType() {
return resultType;
public String getScriptAsString() {
return script;
}2、借鑒鎖實現(xiàn)Redis+Lua計數(shù)器
（1）工具類
package han.zhang.utils;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DigestUtils;
import org.springframework.data.redis.core.script.RedisScript;
import java.util.Collections;
public class CountUtil {
private static final LogUtils logger = LogUtils.getLogger(CountUtil.class);
private final StringRedisTemplate stringRedisTemplate;
/**
* 使用腳本在redis服務器執(zhí)行這個邏輯可以在一定程度上保證此操作的原子性
* （即不會發(fā)生客戶端在執(zhí)行setNX和expire命令之間，發(fā)生崩潰或失去與服務器的連接導致expire沒有得到執(zhí)行，發(fā)生永久死計數(shù)器）
* <p>
* 除非腳本在redis服務器執(zhí)行時redis服務器發(fā)生崩潰，不過此種情況計數(shù)器也會失效
*/
private static final RedisScript<Boolean> SET_AND_EXPIRE_SCRIPT;
static {
StringBuilder sb = new StringBuilder();
sb.append("local visitTimes = redis.call('incr', KEYS[1])\n");
sb.append("if (visitTimes == 1) then\n");
sb.append("\tredis.call('expire', KEYS[1], tonumber(ARGV[1]))\n");
sb.append("\treturn false\n");
sb.append("elseif(visitTimes > tonumber(ARGV[2])) then\n");
sb.append("\treturn true\n");
sb.append("else\n");
sb.append("end");
SET_AND_EXPIRE_SCRIPT = new RedisScriptImpl<>(sb.toString(), Boolean.class);
}
public CountUtil(StringRedisTemplate stringRedisTemplate) {
this.stringRedisTemplate = stringRedisTemplate;
public boolean isOverMaxVisitTimes(String key, int seconds, int maxTimes) throws Exception {
try {
return stringRedisTemplate.execute(SET_AND_EXPIRE_SCRIPT, Collections.singletonList(key), String.valueOf(seconds), String.valueOf(maxTimes));
} catch (Exception e) {
logger.error("RedisBusiness>>>isOverMaxVisitTimes; get visit times Exception; key:" + key + "result:" + e.getMessage());
throw new Exception("already Over MaxVisitTimes");
}
private static class RedisScriptImpl<T> implements RedisScript<T> {
private final String script;
private final String sha1;
private final Class<T> resultType;
public RedisScriptImpl(String script, Class<T> resultType) {
this.script = script;
this.sha1 = DigestUtils.sha1DigestAsHex(script);
this.resultType = resultType;
@Override
public String getSha1() {
return sha1;
public Class<T> getResultType() {
return resultType;
public String getScriptAsString() {
return script;
}（2）調用測試代碼
public void run(String... strings) {
CountUtil countUtil = new CountUtil(SpringUtils.getStringRedisTemplate());
try {
for (int i = 0; i < 10; i++) {
boolean overMax = countUtil.isOverMaxVisitTimes("zhanghantest", 600, 2);
if (overMax) {
System.out.println("超過i:" + i + ":" + overMax);
} else {
System.out.println("沒超過i:" + i + ":" + overMax);
}
}
} catch (Exception e) {
logger.error("Exception {}", e.getMessage());
}
}（3）測試結果
【總結】
1、用心去不斷的改造自己的程序；
2、用代碼改變世界。
到此這篇關于Redis+Lua實現(xiàn)計數(shù)器接口防刷(升級版)的文章就介紹到這了,更多相關Redis計數(shù)器內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
這篇文章主要介紹了高并發(fā)場景分析之redis+lua防重校驗,本文通過示例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2023-07-07
跳表是zset有序集合的底層實現(xiàn)之一,本文主要介紹了Redis 跳表(Skip List)原理實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2025-04-04
Redis本身是一個cs模式的tcp server, client可以通過一個socket連續(xù)發(fā)起多個請求命令。這篇文章帶領大家學習redis為什么要提供pipeline功能,需要的朋友可以參考下2021-06-06
Redis是一個開源的使用ANSI C語言編寫、支持網絡、可基于內存亦可持久化的日志型、Key-Value數(shù)據(jù)庫，并提供多種語言的API。本文就教大家如何安裝Redis，需要的朋友可以參考下2015-10-10
本文介紹了Redis集群的搭建和讀寫操作,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2025-02-02
命令操作、系統(tǒng)配置、關鍵機制、硬件配置等會影響 Redis 的性能，還要提前準備好應對異常的方案，本文主要介紹了Redis的異步機制，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2022-05-05
Redis基本數(shù)據(jù)類型Zset有序集合常用操作
這篇文章主要為大家介紹了redis基本數(shù)據(jù)類型Zset有序集合常用操作，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-05-05
redis?zrange?與?zrangebyscore的區(qū)別解析
這篇文章主要介紹了redis?zrange與zrangebyscore的區(qū)別,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2023-06-06
這篇文章主要介紹了odoo中使用redis實現(xiàn)緩存的步驟,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2023-04-04
這篇文章主要介紹了redis啟動流程介紹,本文更分5步,分別是準備運行環(huán)境、解析命令行參數(shù)、initServer()初始化服務、loadDataFromDisk()從rdb或aof文件加載數(shù)據(jù)、aeMain()開始事件循環(huán)，接收客戶端請求,需要的朋友可以參考下2015-01-01