<!-- springboot已集成，不需要再引入版本 -->
<dependency>
  <groupId>org.springframework.boot</groupId>
  <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>

private void validRateBasic (String phone) {
        String key = "LIMIT:RATE:" + phone;
        redisTemplate.setKeySerializer(new StringRedisSerializer());
        redisTemplate.setValueSerializer(new StringRedisSerializer());
        try {
            String num = (String) redisTemplate.opsForValue().get(key);
            if (ObjectUtil.isNull(num)) {
                redisTemplate.opsForValue().set(key, "1", 60, TimeUnit.SECONDS);
            } else if (Integer.parseInt(num) >= 20) {
                Long expire = redisTemplate.getExpire(key);
                throw new CheckedException("操作頻繁，請" + expire + "s后再試");
            } else {
                redisTemplate.opsForValue().increment(key);
            }
        } catch (Exception e) {
            if (e instanceof CheckedException) {
                throw new CheckedException(e.getMessage());
            } else {
                log.info("校驗(yàn)上傳速率失敗,error:{}",e);
                throw new CheckedException("操作失敗，請稍后再試");
            }
        }

    }

private void validRate(String phone) {
        String key = "LIMIT:RATE:" + phone;SpringBoot整合redis實(shí)現(xiàn)計(jì)數(shù)器限流
        int retryTimes = 0;
        // 失敗重試五次
        redisTemplate.setKeySerializer(new StringRedisSerializer());
        redisTemplate.setValueSerializer(new StringRedisSerializer());
        while(retryTimes < 6) {
            retryTimes++;
            try {
                // 在事務(wù)之外獲取這個(gè)鍵的值
                String num = (String) redisTemplate.opsForValue().get(key);

                // 使用SessionCallback進(jìn)行原子性操作
                SessionCallback<Object> sessionCallback = new SessionCallback<Object>() {
                    @Override
                    public Object execute(RedisOperations operations) throws DataAccessException {
                        operations.watch(key);
                        operations.multi();
                        // 在事務(wù)內(nèi)部再次檢查這個(gè)鍵的值
                        String currentNum = (String) operations.opsForValue().get(key);
                        if (num == null ? currentNum != null : !num.equals(currentNum)) {
                            // 這個(gè)鍵的值被修改了，所以取消這個(gè)事務(wù)
                            operations.discard();
                            return null;
                        }
                        if (ObjectUtil.isNull(num)) {
                            operations.opsForValue().set(key, "1", 60, TimeUnit.SECONDS);
                        } else if (Integer.parseInt(num) >= 5) {
                            Long expire = operations.getExpire(key);
                            throw new CheckedException("操作頻繁，請" + expire + "s后再試");
                        } else {
                            operations.opsForValue().increment(key);
                        }
                        // 提交事務(wù)并返回結(jié)果
                        return operations.exec();
                    }
                };

                // 執(zhí)行SessionCallback
                List<Object> results = (List<Object>) redisTemplate.execute(sessionCallback);
                if (CollectionUtils.isEmpty(results)) {
                    // 如果事務(wù)執(zhí)行失敗，重新嘗試事務(wù)
                    log.info("重試");
                    continue;
                }
                return;
            } catch (Exception e) {
                // 在重試的情況下捕獲任何異常
                if (retryTimes >= 5) {
                    throw new CheckedException("操作頻繁，請稍后再試");
                }
            }
        }
    }

private void validRateLua(String phone) {
        String key = "LIMIT:RATE:" + phone;
        int retryTimes = 0;
        // 創(chuàng)建Lua腳本，返回新的計(jì)數(shù)值
        String luaScript =
                "local num = redis.call('GET', KEYS[1]);" +
                        "if num == false then " +
                        "   redis.call('SET', KEYS[1], ARGV[1], 'EX', ARGV[2]);" +
                        "   return ARGV[1];" +
                        "elseif tonumber(num) <= tonumber(ARGV[3]) then " +
                        "   local newNum = redis.call('INCR', KEYS[1]);" +
                        "   return newNum;" +
                        "else " +
                        "   return num;" +
                        "end;";
        RedisScript<String> redisScript = new DefaultRedisScript<>(luaScript, String.class);
        redisTemplate.setKeySerializer(new StringRedisSerializer());
        redisTemplate.setValueSerializer(new StringRedisSerializer());
        while(retryTimes < 5) {
            retryTimes++;
            try {
                // 執(zhí)行Lua腳本
                String num = (String) redisTemplate.execute(redisScript, Collections.singletonList(key), "1", "60","5");
                if (num != null && Integer.parseInt(num) > 5) {
                    Long expire = redisTemplate.getExpire(key);
                    throw new CheckedException("操作頻繁，請" + expire + "s后再試");
                }

                return;
            } catch (Exception e) {
                if (e instanceof CheckedException) {
                    throw new CheckedException(e.getMessage());
                } else {
                    // 在重試的情況下捕獲任何異常
                    // 有需要的可以加入指數(shù)退避、最大重試時(shí)間等
                    if (retryTimes >= 5) {
                        log.error("上傳失敗,error:{}",e);
                        throw new CheckedException("操作頻繁，請稍后再試");
                    }
                }
            }
        }
    }