分布式调度 redis scheduler锁的实现参考

最新推荐文章于 2024-08-19 21:42:33 发布

mumu2lili

最新推荐文章于 2024-08-19 21:42:33 发布

阅读量140

点赞数 1

分类专栏：云计算 java 文章标签： redis junit 数据库

本文链接：https://blog.csdn.net/eif88/article/details/141332096

版权

java 同时被 2 个专栏收录

7 篇文章 0 订阅

订阅专栏

云计算

5 篇文章 0 订阅

订阅专栏

scheduler竞争锁

注意参数类型和返回值

    @Autowired
    private StringRedisTemplate redisTemplate;

    @Autowired
    private XfuzzConfig xfuzzConfig;

    private volatile boolean scheduler = false;

    @Scheduled(fixedDelay = 5 * 1000, initialDelay = 10 * 1000)
    public void acquireLock() {
        String id = IpUtils.matchIp(xfuzzConfig.getEthName());
        String key = "lock:scheduler";
        int seconds = 60;
        if (!scheduler) {
            scheduler = redisTemplate.opsForValue().setIfAbsent(key, id, seconds, TimeUnit.SECONDS);
            return;
        }

        String lua = "if redis.call('get',KEYS[1]) == ARGV[1] then\n" +
                "    redis.call('expire', KEYS[1], ARGV[2])\n" +
                "    return true\n" +
                "else\n" +
                "end";
        DefaultRedisScript<Boolean> redisScript = new DefaultRedisScript<>();
        redisScript.setResultType(Boolean.class);
        redisScript.setScriptText(lua);
        List<String> keyList = new ArrayList<>();
        keyList.add(key);
        scheduler = redisTemplate.execute(redisScript, keyList, id, seconds + "");

    }

获取 ip（常见网卡匹配）

    public static String matchIp(String ifName) {
        if (ifName.contains(",")) {
            for (String reg : ifName.split(",")) {
                String ip = getIp(reg);
                if (StringUtils.isNotBlank(ip)) {
                    return ip;
                }
            }

            throw new RuntimeException("未获取到网卡" + ifName + " ip");
        } else {
            String ip = getIp(ifName);
            if (StringUtils.isNotBlank(ip)) {
                return ip;
            }

            throw new RuntimeException("未获取到网卡" + ifName + " ip");
        }
    }
          
    
    public static String getIp(String ifName) {
        ifName = ifName.trim();
        try {
            Enumeration<NetworkInterface> networkInterfaces = NetworkInterface.getNetworkInterfaces();
            while (networkInterfaces.hasMoreElements()) {
                NetworkInterface networkInterface = (NetworkInterface) networkInterfaces.nextElement();
                if (networkInterface.isLoopback() || !networkInterface.isUp()) {
                    continue;
                }
                String name = networkInterface.getName();
                boolean match = StringUtils.isNotBlank(name) && name.matches(ifName);
                name = networkInterface.getDisplayName();
                match = match || (StringUtils.isNotBlank(name) && name.matches(ifName));
                if (!match) {
                    continue;
                }

                Enumeration<InetAddress> addresses = networkInterface.getInetAddresses();
                while (addresses.hasMoreElements()) {
                    InetAddress inetAddress = addresses.nextElement();
                    if (inetAddress.isSiteLocalAddress()) {
                        return inetAddress.getHostAddress();
                    }
                }

            }

        } catch (Exception e) {
            log.error("获取网卡 {} ip失败", ifName, e);
        }

        return null;
    }