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

scheduler竞争锁

注意参数类型和返回值

    @Autowiredprivate StringRedisTemplate redisTemplate;@Autowiredprivate 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;}

redis 任务管理

利用 zset

    public boolean isRunning(String id) {Double expire = redisTemplate.opsForZSet().score(VM_VM_IDS, id);return expire != null && expire.longValue() >= System.currentTimeMillis();}public void addTask(String id) {long expireAt = System.currentTimeMillis() + 60 * 1000;redisTemplate.opsForZSet().add(VM_VM_IDS, id, expireAt);}public void removeTask(String id) {redisTemplate.opsForZSet().remove(VM_VM_IDS, id);}public void removeExpiredTask() {long now = System.currentTimeMillis();redisTemplate.opsForZSet().removeRangeByScore(VM_VM_IDS, 0, now);}