使用redis分布式锁解决请求重复的问题

场景:多节点部署相同的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锁成功");
	}

 

  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值