SpringBoot中Zookeeper分布式锁的原理和用法详解

<dependency>
    <groupId>org.springframework.integration</groupId>
    <artifactId>spring-integration-zookeeper</artifactId>
    <version>5.5.0</version>
</dependency>

@Configuration
public class ZookeeperLockConfiguration {
    @Bean
    public ZookeeperLockRegistry zookeeperLockRegistry(CuratorFramework curatorFramework) {
        return new ZookeeperLockRegistry(curatorFramework, "/locks");
    }
    @Bean
    public CuratorFramework curatorFramework() throws Exception {
        return CuratorFrameworkFactory.newClient("127.0.0.1:2181", new RetryUntilElapsed(1000, 4));
    }
}

@Autowired
private ZookeeperLockRegistry zookeeperLockRegistry;
public void doSomething() {
    Lock lock = zookeeperLockRegistry.obtain("my-lock");
    if (lock.tryLock()) {
        try {
            // TODO: 执行业务逻辑
        } finally {
            lock.unlock();
        }
    } else {
        // TODO: 获取锁失败的处理逻辑
    }
}

@SpringBootApplication
public class ZookeeperLockDemoApplication {
    public static void main(String[] args) {
        SpringApplication.run(ZookeeperLockDemoApplication.class, args);
    }
    @Bean
    public ZookeeperLockRegistry zookeeperLockRegistry(CuratorFramework curatorFramework) {
        return new ZookeeperLockRegistry(curatorFramework, "/locks");
    }
    @Bean
    public CuratorFramework curatorFramework() throws Exception {
        return CuratorFrameworkFactory.newClient("127.0.0.1:2181", new RetryUntilElapsed(1000, 4));
    }
}

@RestController
public class CounterController {
    private AtomicInteger counter = new AtomicInteger(0);
    @Autowired
    private ZookeeperLockRegistry zookeeperLockRegistry;
    @GetMapping("/counter")
    public int getCounter() {
        return counter.get();
    }
    @PostMapping("/counter")
    public int increaseCounter() {
        Lock lock = zookeeperLockRegistry.obtain("/counter-lock");
        try {
            if (lock.tryLock(10, TimeUnit.SECONDS)) {
                try {
                    counter.incrementAndGet();
                } finally {
                    lock.unlock();
                }
            } else {
                throw new RuntimeException("Failed to acquire lock for counter!");
            }
        } catch (InterruptedException e) {
            throw new RuntimeException("Failed to acquire lock for counter!", e);
        }
        return counter.get();
    }
}