场景:多节点部署相同的web应用,其定时任务会触发两次,可在业务处理的地方加redis锁进行处理
reids工具类
package ...;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.dao.DataAccessException;
import org.springframework.data.redis.connection.RedisConnection;
import org.springframework.data.redis.core.HashOperations;
import org.springframework.data.redis.core.ListOperations;
import org.springframework.data.redis.core.RedisCallback;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ValueOperations;
import org.springframework.data.redis.serializer.SerializationException;
import org.springframework.stereotype.Component;
import com.chongdong.common.util.RedisKeyPrefix;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisCluster;
import redis.clients.jedis.JedisCommands;
/**
* Redis client for all operations to Redis server.
* TODO:: Improve storage effeciency for Double, Float, Integer, Long and Short??
* @author laiss
*
*/
@Component("redisClient")
public class RedisClient {
private Logger log = LoggerFactory.getLogger(RedisClient.class);
@Autowired
protected RedisTemplate<String, Object> redisTemplate;
public static final String UNLOCK_LUA;
static {
StringBuilder sb = new StringBuilder();
sb.append("if redis.call(\"get\",KEYS[1]) == ARGV[1] ");
sb.append("then ");
sb.append(" return redis.call(\"del\",KEYS[1]) ");
sb.append("else ");
sb.append(" return 0 ");
sb.append("end ");
UNLOCK_LUA = sb.toString();
}
/**
* redis分布式加锁
* @param redisKey redisKey
* @param expire 过期时间
* @param requestId 唯一标识(一般为UUID)
* @return
*/
public boolean setLock(final String redisKey , final long expire, final String requestId) {
try {
RedisCallback<String> callback = new RedisCallback() {
@Override
public Object doInRedis(RedisConnection connection) throws DataAccessException {
JedisCommands commands = (JedisCommands) connection.getNativeConnection();
return commands.set(redisKey, requestId, "NX", "PX", expire);
}
};
String result = redisTemplate.execute(callback);
return !StringUtils.isEmpty(result);
} catch (Exception e) {
log.error("set redis occured an exception", e);
}
return false;
}
/**
* redis分布式解锁
* @param keyPrefix rediskey
* @param key 业务key
* @param requestId 唯一标识(一般为UUID)
* @return
*/
public boolean releaseLock(String redisKey , String requestId) {
// 释放锁的时候,有可能因为持锁之后方法执行时间大于锁的有效期,此时有可能已经被另外一个线程持有锁,所以不能直接删除
try {
final List<String> keys = new ArrayList<String>();
keys.add(redisKey);
final List<String> args = new ArrayList<String>();
args.add(requestId);
// 使用lua脚本删除redis中匹配value的key,可以避免由于方法执行时间过长而redis锁自动过期失效的时候误删其他线程的锁
// spring自带的执行脚本方法中,集群模式直接抛出不支持执行脚本的异常,所以只能拿到原redis的connection来执行脚本
RedisCallback<Long> callback = new RedisCallback(){
@Override
public Object doInRedis(RedisConnection connection) throws DataAccessException {
Object nativeConnection = connection.getNativeConnection();
// 集群模式和单机模式虽然执行脚本的方法一样,但是没有共同的接口,所以只能分开执行
// 集群模式
if (nativeConnection instanceof JedisCluster) {
return (Long) ((JedisCluster) nativeConnection).eval(UNLOCK_LUA, keys, args);
}
// 单机模式
else if (nativeConnection instanceof Jedis) {
return (Long) ((Jedis) nativeConnection).eval(UNLOCK_LUA, keys, args);
}
return 0L;
}
};
Long result = redisTemplate.execute(callback);
return result != null && result > 0;
} catch (Exception e) {
log.error("release lock occured an exception", e);
} finally {
// 清除掉ThreadLocal中的数据,避免内存溢出
//lockFlag.remove();
}
return false;
}
}
业务层代码
static final String KEY= "KEY";
// 过期时间10min
static final long expire = 10*60*1000;
public void doIt() {
// 请求ID
String requestId = UUID.randomUUID().toString();
/** 对此次请求加锁,如果加锁失败,说明已经有锁,标明上一个定时任务正在执行,结束此次请求 **/
if(!redisClient.setLock(KEY, expire, requestId)) {
log.info("redis锁机制触发....,程序终止");
return;
}
log.info("开始业务处理");
业务逻辑
log.info("业务处理成功");
/** redis释放锁 */
redisClient.releaseLock(AUTO_APPROVEL, requestId);
log.info("释放Redis锁成功");
}