2024年最全玩转Mysql系列 - 第26篇：聊聊mysql如何实现分布式锁？，你们觉得作为一名程序员最大的悲哀是什么

线程、数据库、算法、JVM、分布式、微服务、框架、Spring相关知识

一线互联网P7面试集锦+各种大厂面试集锦

学习笔记以及面试真题解析

本文已被CODING开源项目：【一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码】收录

需要这份系统化的资料的朋友，可以点击这里获取

create table t_lock(

lock_key varchar(32) PRIMARY KEY NOT NULL COMMENT ‘锁唯一标志’,

request_id varchar(64) NOT NULL DEFAULT ‘’ COMMENT ‘用来标识请求对象的’,

lock_count INT NOT NULL DEFAULT 0 COMMENT ‘当前上锁次数’,

timeout BIGINT NOT NULL DEFAULT 0 COMMENT ‘锁超时时间’,

version INT NOT NULL DEFAULT 0 COMMENT ‘版本号，每次更新+1’

)COMMENT ‘锁信息表’;

分布式锁工具类：

package com.itsoku.sql;

import lombok.Builder;

import lombok.Getter;

import lombok.Setter;

import lombok.extern.slf4j.Slf4j;

import org.junit.Test;

import java.sql.*;

import java.util.Objects;

import java.util.UUID;

import java.util.concurrent.TimeUnit;

/**

* 工作10年的前阿里P7分享Java、算法、数据库方面的技术干货！坚信用技术改变命运，让家人过上更体面的生活！

* 喜欢的请关注公众号：路人甲Java

*/

@Slf4j

public class LockUtils {

//将requestid保存在该变量中

static ThreadLocal requestIdTL = new ThreadLocal<>();

/**

* 获取当前线程requestid

* @return

*/

public static String getRequestId() {

String requestId = requestIdTL.get();

if (requestId == null || “”.equals(requestId)) {

requestId = UUID.randomUUID().toString();

requestIdTL.set(requestId);

}

log.info(“requestId:{}”, requestId);

return requestId;

}

/**

* 获取锁

* @param lock_key        锁key

* @param locktimeout(毫秒) 持有锁的有效时间，防止死锁

* @param gettimeout(毫秒)  获取锁的超时时间，这个时间内获取不到将重试

* @return

*/

public static boolean lock(String lock_key, long locktimeout, int gettimeout) throws Exception {

log.info(“start”);

boolean lockResult = false;

String request_id = getRequestId();

long starttime = System.currentTimeMillis();

while (true) {

LockModel lockModel = LockUtils.get(lock_key);

if (Objects.isNull(lockModel)) {

//插入一条记录,重新尝试获取锁

LockUtils.insert(LockModel.builder().lock_key(lock_key).request_id(“”).lock_count(0).timeout(0L).version(0).build());

} else {

String reqid = lockModel.getRequest_id();

//如果reqid为空字符，表示锁未被占用

if (“”.equals(reqid)) {

lockModel.setRequest_id(request_id);

lockModel.setLock_count(1);

lockModel.setTimeout(System.currentTimeMillis() + locktimeout);

if (LockUtils.update(lockModel) == 1) {

lockResult = true;

break;

}

} else if (request_id.equals(reqid)) {

//如果request_id和表中request_id一样表示锁被当前线程持有者，此时需要加重入锁

lockModel.setTimeout(System.currentTimeMillis() + locktimeout);

lockModel.setLock_count(lockModel.getLock_count() + 1);

if (LockUtils.update(lockModel) == 1) {

lockResult = true;

break;

}

} else {

//锁不是自己的，并且已经超时了，则重置锁，继续重试

if (lockModel.getTimeout() < System.currentTimeMillis()) {

LockUtils.resetLock(lockModel);

} else {

//如果未超时，休眠100毫秒，继续重试

if (starttime + gettimeout > System.currentTimeMillis()) {

TimeUnit.MILLISECONDS.sleep(100);

} else {

break;

}

}

}

}

}

log.info(“end”);

return lockResult;

}

/**

* 释放锁

* @param lock_key

* @throws Exception

*/

public static void unlock(String lock_key) throws Exception {

//获取当前线程requestId

String requestId = getRequestId();

LockModel lockModel = LockUtils.get(lock_key);

//当前线程requestId和库中request_id一致 && lock_count>0，表示可以释放锁

if (Objects.nonNull(lockModel) && requestId.equals(lockModel.getRequest_id()) && lockModel.getLock_count() > 0) {

if (lockModel.getLock_count() == 1) {

//重置锁

resetLock(lockModel);

} else {

lockModel.setLock_count(lockModel.getLock_count() - 1);

LockUtils.update(lockModel);

}

}

}

/**

* 重置锁

* @param lockModel

* @return

* @throws Exception

*/

public static int resetLock(LockModel lockModel) throws Exception {

lockModel.setRequest_id(“”);

lockModel.setLock_count(0);

lockModel.setTimeout(0L);

return LockUtils.update(lockModel);

}

/**

* 更新lockModel信息，内部采用乐观锁来更新

* @param lockModel

* @return

* @throws Exception

*/

public static int update(LockModel lockModel) throws Exception {

return exec(conn -> {

String sql = “UPDATE t_lock SET request_id = ?,lock_count = ?,timeout = ?,version = version + 1 WHERE lock_key = ? AND  version = ?”;

PreparedStatement ps = conn.prepareStatement(sql);

int colIndex = 1;

ps.setString(colIndex++, lockModel.getRequest_id());

ps.setInt(colIndex++, lockModel.getLock_count());

ps.setLong(colIndex++, lockModel.getTimeout());

ps.setString(colIndex++, lockModel.getLock_key());

ps.setInt(colIndex++, lockModel.getVersion());

return ps.executeUpdate();

});

}

public static LockModel get(String lock_key) throws Exception {

return exec(conn -> {

String sql = “select * from t_lock t WHERE t.lock_key=?”;

PreparedStatement ps = conn.prepareStatement(sql);

int colIndex = 1;

ps.setString(colIndex++, lock_key);

ResultSet rs = ps.executeQuery();

if (rs.next()) {

return LockModel.builder().

lock_key(lock_key).

request_id(rs.getString(“request_id”)).

lock_count(rs.getInt(“lock_count”)).

timeout(rs.getLong(“timeout”)).

version(rs.getInt(“version”)).build();

}

return null;

});

}

public static int insert(LockModel lockModel) throws Exception {

return exec(conn -> {

String sql = “insert into t_lock (lock_key, request_id, lock_count, timeout, version) VALUES (?,?,?,?,?)”;

PreparedStatement ps = conn.prepareStatement(sql);

int colIndex = 1;

ps.setString(colIndex++, lockModel.getLock_key());

ps.setString(colIndex++, lockModel.getRequest_id());

ps.setInt(colIndex++, lockModel.getLock_count());

ps.setLong(colIndex++, lockModel.getTimeout());

ps.setInt(colIndex++, lockModel.getVersion());

return ps.executeUpdate();

});

}

public static  T exec(SqlExec sqlExec) throws Exception {

Connection conn = getConn();

try {

return sqlExec.exec(conn);

} finally {

closeConn(conn);

}

}

@FunctionalInterface

public interface SqlExec {

T exec(Connection conn) throws Exception;

}

@Getter

@Setter

@Builder

public static class LockModel {

private String lock_key;

private String request_id;

private Integer lock_count;

private Long timeout;

private Integer version;

}

private static final String url = “jdbc:mysql://localhost:3306/javacode2018?useSSL=false”;        //数据库地址

private static final String username = “root”;        //数据库用户名

private static final String password = “root123”;        //数据库密码

private static final String driver = “com.mysql.jdbc.Driver”;        //mysql驱动

/**

* 连接数据库

* @return

*/

public static Connection getConn() {

Connection conn = null;

try {

Class.forName(driver);  //加载数据库驱动

try {

conn = DriverManager.getConnection(url, username, password);  //连接数据库

} catch (SQLException e) {

e.printStackTrace();

}

} catch (ClassNotFoundException e) {

本次面试答案，以及收集到的大厂必问面试题分享：

本文已被CODING开源项目：【一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码】收录

需要这份系统化的资料的朋友，可以点击这里获取

lass.forName(driver);  //加载数据库驱动

try {

conn = DriverManager.getConnection(url, username, password);  //连接数据库

} catch (SQLException e) {

e.printStackTrace();

}

} catch (ClassNotFoundException e) {

本次面试答案，以及收集到的大厂必问面试题分享：

[外链图片转存中…(img-SNM8Y7cr-1715116738779)]

本文已被CODING开源项目：【一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码】收录

需要这份系统化的资料的朋友，可以点击这里获取