ehlxr/budd
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add redis lock test

Browse Source
This commit is contained in:
ehlxr
2021-08-30 10:14:08 +08:00
parent 9d6a001b85
commit f601c051d4
7 changed files with 233 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
budd-common/src/main/java/io/github/ehlxr/common/Result.java
View File

@@ -15,7 +15,7 @@ import java.util.Objects;
/**
* 统一输出结果集
*
* @author lixiangrong
* @author ehlxr
* @since 2020/3/18.
*/
@JsonInclude(JsonInclude.Include.NON_NULL)

108
budd-common/src/main/java/io/github/ehlxr/enums/CodeEnum.java
View File

@@ -27,7 +27,7 @@ package io.github.ehlxr.enums;
/**
* 自定义状态码
*
* @author lixiangrong
* @author ehlxr
* @since 2020/3/18.
*/
public enum CodeEnum {
@@ -36,71 +36,6 @@ public enum CodeEnum {
*/
SUCCESSFUL(200, "success"),
LOGIN_TYPE_ERROR(201, "登录类型错误"),
/**
* 用户不存在
*/
USER_NOT_EXIST(202, "用户不存在"),
/**
* 医生状态修改失败
*/
DOCTOR_STATUS_FAILURE(204,"医生状态修改失败"),
/**
* 医生状态不正常
*/
DOCTOR_STATUS_NOT_NORMAL(205,"医生状态不正常"),
DOCTOR_ID_EMPTY(206,"医生不允许为空"),
DEVICE_LOGIN_NULL(209, "设备尚未登录"),
/**
* 成功
*/
UC_CONNECT_ERROR(211, "用户中心异常，请联系管理员"),
UC_NOT_PHRID(212,"用户中心异常，未返回用户ID"),
/**
* 诊断记录不存在
*/
RAINQUIRY_NOT_EXIST(301, "诊断记录不存在"),
/**
* 成功
*/
LOGIN_TYPE_NULL(700, "登录方式不能为空"),
/**
* 成功
*/
MOBILE_NULL(701, "手机号码不能为空"),
/**
* 成功
*/
VCODE_ERROR(702, "验证码错误"),
MOBILE_HAVE(703, "此手机号已经注册过"),
/**
* 成功
*/
VCODE_NULL_ERROR(704, "验证码不能为空"),
/**
* 成功
*/
GENDER_RELATION_ERROR(705, "性别和关系错误"),
ID_CARD_NO_ERROR(706, "身份证号不能为空"),
HEAD_IMG_ERROR(707, "人脸识别照片不能为空"),
DEVICE_CODE_ERROR(708, "设备号不能为空"),
/**
* 未知异常
*/
@@ -114,43 +49,8 @@ public enum CodeEnum {
/**
* 业务异常
*/
SERVICE_EXCEPTION(602, "service exception"),
SERVICE_EXCEPTION(602, "service exception");
/**
* 上善（三疗）的token没有获取到
*/
NO_SANLIAO_TOKEN(801, "上善（三疗）的Token没有"),
/**
* No report generated
*/
NO_REPORT_GENERATED(603, "no report generated"),
NO_MATCH_DOCTOR(604,"没有空闲医生"),
NO_MATCH_DATA(604,"没有查询到数据"),
/**
* 未选择全部检查项
*/
SELECT_EXAM_WRAN(2001, "未选择全部检查项"),
/**
* 创建订单失败
*/
CREATE_ORDER_FAIL(2002,"创建订单失败"),
PENDING_ORDER_ERROR(2003,"抢单失败"),
INVALID_ORDER_ERROR(2004,"无效订单"),
INVALID_SHOP_ROLE_ERROR(2005,"当前用户无商铺管理员角色"),
DUPLICATE_VOUCHING_ORDER_ERROR(2006,"您已审核过此订单"),
GET_ORDER_ERROR(2007,"获取订单信息失败"),
GET_SHOP_ROLE_ERROR(2008,"获取商铺角色失败");
private final int code;
private final String message;
@@ -164,10 +64,6 @@ public enum CodeEnum {
return code;
}
public String getCodeString() {
return code + "";
}
public String getMessage() {
return message;
}

142
budd-common/src/main/java/io/github/ehlxr/util/RedisUtil.java
View File

@@ -24,9 +24,13 @@
package io.github.ehlxr.util;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.ListOperations;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.data.redis.core.script.RedisScript;
import org.springframework.scripting.support.StaticScriptSource;
import org.springframework.stereotype.Component;
@@ -40,13 +44,14 @@ import java.util.concurrent.TimeUnit;
* @author ehlxr
* @since 2021-08-29 18:32.
*/
@SuppressWarnings("unused")
@Component
public class RedisUtil {
private static final Logger log = LoggerFactory.getLogger(RedisUtil.class);
/**
* 解锁 lua 脚本
*/
public static final String UNLOCK = "if (redis.call('hexists', KEYS[1], ARGV[1]) == 0) then " +
private static final String UNLOCK = "if (redis.call('hexists', KEYS[1], ARGV[1]) == 0) then " +
"return nil; " +
"end; " +
"local counter = redis.call('hincrby', KEYS[1], ARGV[1], -1); " +
@@ -85,6 +90,139 @@ public class RedisUtil {
"return 0;";
private static RedisTemplate<String, String> redisTemplate;
/**
* 释放分布式锁时使用的 lua 脚本，保证原子性
* <p>
* if (redis.call('get', KEYS[1]) == ARGV[1])
* then
* return redis.call('del', KEYS[1])
* else
* return 0
* end
*/
private static final String RELEASE_LOCK_LUA = "if (redis.call('get', KEYS[1]) == ARGV[1]) then return redis.call('del', KEYS[1]) else return 0 end";
/**
* 滑动窗口限流使用的 lua 脚本，保证原子性
* <p>
* local key = KEYS[1];
* local index = tonumber(ARGV[1]);
* local time_window = tonumber(ARGV[2]);
* local now_time = tonumber(ARGV[3]);
* local far_time = redis.call('lindex', key, index);
* if (not far_time)
* then
* redis.call('lpush', key, now_time);
* redis.call('pexpire', key, time_window+1000);
* return 1;
* end
* if (now_time - far_time > time_window)
* then
* redis.call('rpop', key);
* redis.call('lpush', key, now_time);
* redis.call('pexpire', key, time_window+1000);
* return 1;
* else
* return 0;
* end
*/
private static final String SLIDE_WINDOW_LUA = "local key = KEYS[1];\n" + "local index = tonumber(ARGV[1]);\n" + "local time_window = tonumber(ARGV[2]);\n" + "local now_time = tonumber(ARGV[3]);\n" + "local far_time = redis.call('lindex', key, index);\n" + "if (not far_time)\n" + "then\n" + " redis.call('lpush', key, now_time);\n" + " redis.call('pexpire', key, time_window+1000);\n" + " return 1;\n" + "end\n" + "\n" + "if (now_time - far_time > time_window)\n" + "then\n" + " redis.call('rpop', key);\n" + " redis.call('lpush', key, now_time);\n" + " redis.call('pexpire', key, time_window+1000);\n" + " return 1;\n" + "else\n" + " return 0;\n" + "end";
/**
* 获取分布式锁
*
* @param key key
* @param value value，需要保证全局唯一，用来删除分布式锁时判断身份使用
* @param expireTime 锁过期时间，毫秒，防止业务崩溃未删除锁，导致死锁
* @return 是否获取成功锁
*/
public static boolean getDistributedLock(String key, String value, long expireTime) {
boolean result = false;
try {
Boolean set = redisTemplate.opsForValue().setIfAbsent(key, value, expireTime, TimeUnit.MILLISECONDS);
result = set == null || set;
log.info("getLock redis key: {}, value: {}, expireTime: {}, result: {}", key, value, expireTime, result);
} catch (Exception e) {
log.error("getLock redis key: {}, value: {}, expireTime: {}", key, value, expireTime, e);
}
return result;
}
/**
* 释放分布式锁
*
* @param key key
* @param value value，需要和获取锁时传入的一致
* @return 是否释放成功锁
*/
public static boolean releaseDistributedLock(String key, String value) {
Long execute = null;
try {
RedisScript<Long> redisScript = new DefaultRedisScript<>(RELEASE_LOCK_LUA, Long.class);
execute = redisTemplate.execute(redisScript, Collections.singletonList(key), value);
log.debug("releaseLock redis key: {}, value: {}, result: {}", key, value, execute);
} catch (Exception e) {
log.error("releaseLock redis key: {}, value: {}", key, value, e);
}
return Long.valueOf(1L).equals(execute);
}
/**
* 分布式限流队列，在时间窗口内（包含该时间点），判断是否达到限流的阀值
* 本接口实现的方法通过加锁避免并发问题，性能不高。只是为了说明限流逻辑如何实现
*
* @param key key
* @param count 限流阀值
* @param timeWindow 限流时间窗口
* @return 是否允许通过（通过即不进行限流）
*/
public static synchronized boolean slideWindow(String key, int count, long timeWindow) {
try {
long nowTime = System.currentTimeMillis();
ListOperations<String, String> list = redisTemplate.opsForList();
String farTime = list.index(key, count - 1);
if (farTime == null) {
list.leftPush(key, String.valueOf(nowTime));
redisTemplate.expire(key, timeWindow + 1000L, TimeUnit.MILLISECONDS);
return true;
}
if (nowTime - Long.parseLong(farTime) > timeWindow) {
list.rightPop(key);
list.leftPush(key, String.valueOf(nowTime));
redisTemplate.expire(key, timeWindow + 1000L, TimeUnit.MILLISECONDS);
return true;
}
return false;
} catch (Exception e) {
log.error("", e);
return false;
}
}
/**
* 分布式限流队列，在时间窗口内（包含该时间点），判断是否达到限流的阀值
* 本接口实现的方法通过 Lua 脚本避免并发问题，性能较高。
*
* @param key key
* @param count 限流阀值
* @param timeWindow 限流时间窗口
* @return 是否允许通过（通过即不进行限流）
*/
public static boolean slideWindowLua(String key, int count, long timeWindow) {
if (count <= 0 || timeWindow <= 0) {
return false;
}
Long execute = null;
try {
RedisScript<Long> redisScript = new DefaultRedisScript<>(SLIDE_WINDOW_LUA, Long.class);
execute = redisTemplate.execute(redisScript, Collections.singletonList(key), String.valueOf(count - 1), String.valueOf(timeWindow), String.valueOf(System.currentTimeMillis()));
log.debug("slideWindowLua redis key: {}, count: {}, timeWindow: {}, result: {}", key, count, timeWindow, execute);
} catch (Exception e) {
log.error("slideWindowLua redis key: {}, count: {}, timeWindow: {}", key, count, timeWindow, e);
}
return Long.valueOf(1L).equals(execute);
}
/**
* 加锁 不可重入
*