分布式锁-Go服务器开发

分布式锁是一种用于在分布式系统中控制对共享资源的访问的机制。它可以确保在多个节点（如微服务、容器或虚拟机）之间有效地协调对某一资源的访问，避免因并发操作导致的数据不一致性。

使用场景

防止数据竞争：在高并发环境下，多个进程可能同时尝试修改相同的数据。
任务调度：确保某些任务只由一个实例执行，例如定时任务或后台作业。
临界区保护：当多个服务需要访问共享数据库或文件时，使用分布式锁可以保护这些操作。

常见实现方式

基于数据库的分布式锁
- 利用数据库表来存储锁信息，通过插入和删除记录来获取和释放锁。
- 这种方法简单，但会带来性能瓶颈。

   -- 锁定操作示例
   INSERT INTO locks (resource, locked_at) VALUES ('my_resource', NOW()) 
   ON DUPLICATE KEY UPDATE locked_at = NOW();

   -- 解锁操作示例
   DELETE FROM locks WHERE resource = 'my_resource';

基于 Redis 的分布式锁

Redis 提供了轻量级、高性能的分布式锁方案，可以通过 SETNX 命令实现。
使用 SET 命令配合过期时间来避免死锁问题。

import redis
import time


class DistributedLock:
    def __init__(self, redis_client, lock_name):
        self.redis_client = redis_client
        self.lock_name = lock_name


    def acquire(self, timeout=10):
        end_time = time.time() + timeout
        while time.time() < end_time:
            if self.redis_client.setnx(self.lock_name, "locked"):
                # 设置过期时间，防止死锁
                self.redis_client.expire(self.lock_name, 10)
                return True
            time.sleep(0.01)  # 等待后重试
        return False


    def release(self):
        self.redis_client.delete(self.lock_name)

# 示例使用
r = redis.Redis()
lock = DistributedLock(r, "my_lock")


if lock.acquire():
    try:
        # 执行临界区代码块...
        print("Locked section")
    finally:
        lock.release()
else:
    print("Could not acquire lock.")