快捷導(dǎo)航

Spring?Boot?實(shí)現(xiàn)Redis分布式鎖原理

更新時(shí)間：2022年08月04日 09:37:42 作者：劍圣無(wú)痕

這篇文章主要介紹了Spring?Boot實(shí)現(xiàn)Redis分布式鎖原理，文章圍繞主題展開(kāi)詳細(xì)的內(nèi)容介紹，具有一定的參考價(jià)值，需要的朋友可以參考一下

分布式鎖實(shí)現(xiàn)

引入jar包

<dependency>
 <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-data-redis</artifactId>
        <exclusions>
   <exclusion>
 <groupId>io.lettuce</groupId>
 <artifactId>lettuce-core</artifactId>
</exclusion>
</exclusions>
</dependency>
 <dependency>
    <groupId>redis.clients</groupId>
    <artifactId>jedis</artifactId>
 </dependency>

說(shuō)明：本文采用jedis來(lái)實(shí)現(xiàn)分布式鎖。

封裝工具類(lèi)

@Component
public class RedisLockUtil
{
    private static final Logger logger = LoggerFactory.getLogger(RedisLockUtil.class);
    private static final Long RELEASE_SUCCESS = 1L;
    private static final String LOCK_SUCCESS = "OK";
    private static final String RELEASE_LOCK_SCRIPT = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
    @Resource
    private RedisTemplate<String, Object> redisTemplate;
    /**
     * 加鎖方法僅針對(duì)單實(shí)例 Redis,哨兵、集群模式無(wú)法使用
     *
     * @param lockKey 加鎖鍵
     * @param clientId 加鎖客戶(hù)端唯一標(biāo)識(shí)(采用UUID)
     * @param seconds 鎖過(guò)期時(shí)間
     * @return true標(biāo)識(shí)加鎖成功、false代表加鎖失敗
     */
    public Boolean tryLock(String lockKey, String clientId, long seconds)
    {
        try
        {
            return redisTemplate
                    .execute((RedisCallback<Boolean>) redisConnection -> {
                        Jedis jedis = (Jedis) redisConnection.getNativeConnection();
                        SetParams params =new SetParams();
                        params.nx();
                        params.px(seconds);
                        String result = jedis.set(lockKey, clientId, params);
                        if (LOCK_SUCCESS.equals(result))
                        {
                            return Boolean.TRUE;
                        }
                        return Boolean.FALSE;
                    });
        }
        catch (Exception e)
        {
            logger.error("tryLock error",e);
        }

        return false;
    }
    /**
     *釋放鎖，保持原子性操作，采用了lua腳本
     *
     * @param lockKey
     * @param clientId
     * @return
     */
    public Boolean unLock(String lockKey, String clientId)
    {
        try
        {
            return  redisTemplate
                    .execute((RedisCallback<Boolean>) redisConnection -> {
                        Jedis jedis = (Jedis) redisConnection.getNativeConnection();
                        Object result = jedis.eval(RELEASE_LOCK_SCRIPT,
                                Collections.singletonList(lockKey),
                                Collections.singletonList(clientId));
                        if (RELEASE_SUCCESS.equals(result))
                        {
                            return Boolean.TRUE;
                        }
                        return Boolean.FALSE;
                    });
        }
        catch (Exception e)
        {
            logger.error("unlock error",e);
        }
        return Boolean.FALSE;
    }
}

說(shuō)明：加鎖的原理是基于Redis的NX、PX命令，而解鎖采用的是lua腳本實(shí)現(xiàn)。

模擬秒殺扣減庫(kù)存

public int lockStock()
    {
        String lockKey="lock:stock";
        String clientId = UUID.randomUUID().toString();
        long seconds =1000l;

        try
        {
            //加鎖
            boolean flag=redisLockUtil.tryLock(lockKey, clientId, seconds);
            //加鎖成功
            if(flag)
            {
               logger.info("加鎖成功 clientId：{}",clientId);
               int stockNum= Integer.valueOf((String)redisUtil.get("seckill:goods:stock"));
               if(stockNum>0)
               {
                  stockNum--;
                  redisUtil.set("seckill:goods:stock",String.valueOf(stockNum));
                  logger.info("秒殺成功,剩余庫(kù)存:{}",stockNum);
               }
               else
               {
                  logger.error("秒殺失敗,剩余庫(kù)存:{}", stockNum);
               }
               //獲取庫(kù)存數(shù)量
               return stockNum;
            }
            else
            {
                logger.error("加鎖失敗：clientId：{}",clientId);
            }
        }
        catch (Exception e)
        {
           logger.error("decry stock eror",e);
        }
        finally
        {
           redisLockUtil.unLock(lockKey, clientId);
        }
        return 0;
    }

測(cè)試代碼

@RequestMapping("/redisLockTest")
    public void redisLockTest()
    {
        // 初始化秒殺庫(kù)存數(shù)量
        redisUtil.set("seckill:goods:stock", "10");

        List<Future> futureList = new ArrayList<>();

        //多線程異步執(zhí)行
        ExecutorService executors = Executors.newScheduledThreadPool(10);
        //
        for (int i = 0; i < 30; i++)
        {
            futureList.add(executors.submit(this::lockStock));

            try
            {
               Thread.sleep(100);
            }
            catch (InterruptedException e) 
            {
               logger.error("redisLockTest error",e);
            }
        }

        // 等待結(jié)果，防止主線程退出
        futureList.forEach(t -> {
            try 
            {
                int stockNum =(int) t.get();
                logger.info("庫(kù)存剩余數(shù)量:{}",stockNum);
            }
            catch (Exception e)
            {
               logger.error("get stock num error",e);
            }
        });
    }

執(zhí)行結(jié)果如下：

方案優(yōu)化

上述分布式鎖實(shí)現(xiàn)庫(kù)存扣減是否存在相關(guān)問(wèn)題呢？

問(wèn)題1：扣減庫(kù)存邏輯無(wú)法保證原子性，

具體的代碼如下：

int stockNum= Integer.valueOf((String)redisUtil.get("seckill:goods:stock"));
if(stockNum>0)
 {
    stockNum--;
    redisUtil.set("seckill:goods:stock",String.valueOf(stockNum));
 }

這是典型的RMW模型，前面章節(jié)已經(jīng)介紹了具體的實(shí)現(xiàn)方案，可以采用lua腳本和Redis的incry原子性命令實(shí)現(xiàn)，這里采用lua腳本來(lái)實(shí)現(xiàn)原子性的庫(kù)存扣減。

具體實(shí)現(xiàn)如下：

  public long surplusStock(String key ,int num)
   {
       StringBuilder lua_surplusStock = new StringBuilder();
       lua_surplusStock.append("   local key = KEYS[1];");
       lua_surplusStock.append("   local subNum = tonumber(ARGV[1]);");
       lua_surplusStock.append("   local surplusStock=tonumber(redis.call('get',key));");
       lua_surplusStock.append("    if (surplusStock- subNum>= -1) then");
       lua_surplusStock.append("        return redis.call('incrby', KEYS[1], 0-subNum);");
       lua_surplusStock.append("    else ");
       lua_surplusStock.append("    return -1;");
       lua_surplusStock.append("    end");
       
       List<String> keys = new ArrayList<>();
       keys.add(key);
       // 腳本里的ARGV參數(shù)
       List<String> args = new ArrayList<>();
       args.add(Integer.toString(num));

       long result = redisTemplate.execute(new RedisCallback<Long>() {
           @Override
           public Long doInRedis(RedisConnection connection) throws DataAccessException {
               Object nativeConnection = connection.getNativeConnection();
               // 單機(jī)模式
               if (nativeConnection instanceof Jedis) 
               {
                   return (Long) ((Jedis) nativeConnection).eval(lua_surplusStock.toString(), keys, args);
               }
               return -1l;
           }
       });
       return result;
   }

問(wèn)題2:如果加鎖失敗，則會(huì)直接訪問(wèn)，無(wú)法重入鎖

因?yàn)閱螜C(jī)版本的鎖是無(wú)法重入鎖，所以加鎖失敗就直接返回，此問(wèn)題的解決方案，可以采用Redisson來(lái)實(shí)現(xiàn)，關(guān)于Redisson實(shí)現(xiàn)分布式鎖，將在后續(xù)的文章中進(jìn)行詳細(xì)的講解。

總結(jié)

本文主要講解了Spring Boot集成Redis實(shí)現(xiàn)單機(jī)版本分布式鎖，雖然單機(jī)版分布式鎖存在鎖的續(xù)期、鎖的重入問(wèn)題，但是我們還是需要掌握其原理和實(shí)現(xiàn)方法

到此這篇關(guān)于Spring Boot 實(shí)現(xiàn)Redis分布式鎖原理的文章就介紹到這了,更多相關(guān)Spring Boot Redis分布式鎖內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: