聊聊Java三种常见的分布式锁

CREATE TABLE `t_database_lock` (
  `id` bigint NOT NULL COMMENT '主键',
  `method_name` varchar(50) COLLATE utf8_bin NOT NULL DEFAULT '方法名字',
  `create_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
  `update_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '最近更新时间'
  PRIMARY KEY(`id`),
  UNIQUE INDEX `uk_t_database_lock_mname`(`method_name`) USING BTREE COMMENT '方法名唯一索引',
  INDEX `idx_t_accounting_balance_account_utime`(`update_time`) USING BTREE COMMENT '更新时间索引',
  INDEX `idx_t_accounting_balance_account_ctime`(`create_time`) USING BTREE COMMENT '创建时间索引'
) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin

public void execute(String methodName) {
    try {
        if (lock(methodName)) { // 加锁
            // 你的业务逻辑
            run();
        }
    } finally {
        unlock(); // 释放锁
    }
}

private boolean lock(String methodName) {
    try {
        // 插入一条数据
        int count = "INSERT INTO method_lock (method_name, desc) VALUES ('methodName', '需要加锁的methodName');";
        if (count == 1) { // 1 表示插入成功
            return true;
        }
        return false;
    } catch (Exception e) {
        return false;
    }
    return false;
}

private boolean lock(String methodName) {
    try {
        // 插入一条数据
        int count = "INSERT INTO method_lock (method_name, desc) VALUES ('methodName', '需要加锁的methodName');";
        if (count == 1) { // 1 表示插入成功
            return true;
        }
        return false;
    } catch (Exception e) {
        return false;
    }
    return false;
}

if (jedis.setnx(key) == 1) { // 加锁成功
    expire(key, 100);
    try {
        doSomething(); //业务处理
    } catch (Exception e) {
    } finally {
        jedis.del(key); // 释放锁
    }
}

public boolean lock(String key, Long expireTime) {
    Jedis jedis = new Jedis();
    //系统时间 + 设置的过期时间
    long expires = System.currentTimeMillis() + expireTime;
    String expiresStr = String.valueOf(expires);
    // 如果当前锁不存在，返回加锁成功
    if (jedis.setnx(key, expiresStr) == 1) {
        return true;
    }
    // 如果锁已经存在，获取锁的过期时间
    String currentValueStr = jedis.get(key);
    // 锁过期
    if (currentValueStr != null && Long.parseLong(currentValueStr) < System.currentTimeMillis()) {
        // 锁已过期，获取上一个锁的过期时间，并设置现在锁的过期时间
        String oldValueStr = jedis.getSet(key, expiresStr);
        if (oldValueStr != null && oldValueStr.equals(currentValueStr)) {
            // 只有上一个锁的过期时间和当前过期时间相同才可加锁
            return true;
        }
    }
    return false;
}

public void execute() {
    Jedis jedis = new Jedis();
    String key = "key";
    String uuid = "uuid";
    if ("1".equals(jedis.set(key, uuid, "NX", "EX", 100))) { //加锁
        try {
            doSomething(); //业务处理
        } finally {
            //判断是不是当前线程加的锁,是才释放
            if (uuid.equals(jedis.get(key))) {
                jedis.del(key); //释放锁
            }
        }
    }
}