共計 4070 個字符,預計需要花費 11 分鐘才能閱讀完成。
自動寫代碼機器人,免費開通
這篇文章將為大家詳細講解有關怎么在 MySQL 數據庫中實現一個分布式鎖,文章內容質量較高,因此丸趣 TV 小編分享給大家做個參考,希望大家閱讀完這篇文章后對相關知識有一定的了解。
建表
CREATE TABLE `globallocktable` ( `id` int(11) NOT NULL AUTO_INCREMENT,
`lockKey` varchar(60) NOT NULL COMMENT 鎖名稱 ,
`createTime` datetime NOT NULL COMMENT 創建時間 ,
PRIMARY KEY (`id`),
UNIQUE KEY `lockKey` (`lockKey`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8 ROW_FORMAT=DYNAMIC COMMENT= 全局鎖
讓別人使用的組件
@Component
public class GlobalLockComponent {
@Resource
GlobalLockTableDAO globalLockDAO;
/**
* 嘗試獲得鎖,成功為 true,失敗為 false
*/
public boolean tryLock(String key) { return GlobalLockUtil.tryLock(this.globalLockDAO, key);
}
/**
* 如果已經有其他程序占用該鎖,并且超過 timeoutMs(毫秒)時間,就強制清除這個鎖占用
* 即根據 key 先刪除記錄,再添加記錄
*/
public boolean tryLockWithClear(String key, Long timeoutMs) { return GlobalLockUtil.tryLockWithClear(this.globalLockDAO, key, timeoutMs);
}
/**
* 釋放鎖,根據 key 刪除記錄
*/
public void releasLock(String key) { GlobalLockUtil.releasLock(this.globalLockDAO, key);
}
}
鎖對象定義如下
public class GlobalLockTable {
private Integer id;
private String lockKey;
private Date createTime;
// 省略 get 和 set 方法
GlobalLockTableDAO 定義如下
public interface GlobalLockTableDAO { int deleteByPrimaryKey(Integer id);
int deleteByLockKey(String lockKey);
GlobalLockTable selectByLockKey(String key);
int insertSelectiveWithTest(GlobalLockTable record);
}
具體加鎖和解鎖邏輯
public class GlobalLockUtil { private static Logger logger = LoggerFactory.getLogger(GlobalLockUtil.class);
private static GlobalLockTable tryLockInternal(GlobalLockTableDAO lockDAO, String key) { GlobalLockTable insert = new GlobalLockTable();
insert.setCreateTime(new Date());
insert.setLockKey(key);
// 注意的地方 1
int count = lockDAO.insertSelectiveWithTest(insert);
if (count == 0) { GlobalLockTable ready = lockDAO.selectByLockKey(key);
logger.warn(can not lock the key: {}, {}, {} , insert.getLockKey(), ready.getCreateTime(),
ready.getId());
return ready;
}
logger.info(yes got the lock by key: {} , insert.getId(), insert.getLockKey());
return null;
}
/** 超時清除鎖占用,并重新加鎖 **/
public static boolean tryLockWithClear(GlobalLockTableDAO lockDAO, String key, Long timeoutMs) { GlobalLockTable lock = tryLockInternal(lockDAO, key);
if (lock == null) return true;
if (System.currentTimeMillis() - lock.getCreateTime().getTime() = timeoutMs) { logger.warn( sorry, can not get the key. : {}, {}, {} , key, lock.getId(), lock.getCreateTime());
return false;
}
logger.warn(the key already timeout wthin : {}, {}, will clear , key, timeoutMs);
// 注意的地方 2
int count = lockDAO.deleteByPrimaryKey(lock.getId());
if (count == 0) { logger.warn( sorry, the key already preemptived by others: {}, {} , lock.getId(), lock.getLockKey());
return false;
}
lock = tryLockInternal(lockDAO, key);
return lock != null ? false : true;
}
/** 加鎖 **/
public static boolean tryLock(GlobalLockTableDAO lockDAO, String key) { return tryLockInternal(lockDAO, key) == null ? true : false;
}
/** 解鎖 **/
public static void releasLock(GlobalLockTableDAO lockDAO, String key) { lockDAO.deleteByLockKey(key);
}
}
這個工具類有 2 個特別有意思的地方,先看注意的地方 2(上面代碼中標識了)
1. 為了避免鎖長時間不釋放,用 Redis 實現的話可以設置鎖超時時間,超時自動釋放(后面會寫用 Redis 實現分布式鎖)用 MySQL 實現的話可以先刪除后添加。可以看到刪除的時候使用 id 刪的,不是用 name 刪的。為啥呢?先自己想一下
因為如果是通過 name 刪的話,有可能別人刪了這個鎖后,又通過 name 加了鎖,還沒到超時時間,結果你卻根據 name 刪除了。通過 id 刪的話,當返回的 id= 0 時,說明別人已經重新加鎖了,你需要重新獲取。
2.GlobalLockTable 對象 dao 層的其他方法都見名知意,來看一個這個方法。即代碼中的注意點 1
可以看到每次嘗試加鎖的時候,并不是先 select,而是直接 insertSelectiveWithTest,這樣就少了一個查詢時間,提高了效率
insertSelectiveWithTest 的作用是當 lockKey 存在時不進行插入操作,返回 0。當 lockKey 不存在時進行插入操作,返回 1
insert id= insertSelectiveWithTest useGeneratedKeys= true keyProperty= id parameterType= com.javashitang.middleware.lock.mysql.pojo.GlobalLockTable
insert into `globallocktable` (`id`,
`lockKey`, `createTime` )
select #{id,jdbcType=INTEGER}, #{lockKey,jdbcType=VARCHAR}, #{createTime,jdbcType=TIMESTAMP}
from dual where not exists
(select 1 from globallocktable where lockKey = #{lockKey,jdbcType=VARCHAR})
/insert
使用
當我們想使用時,就只寫業務邏輯就行了,非常方便
if (!globalLockComponent.tryLock(name)) {
// 沒有獲取到鎖返回
return;
try { // 這里寫業務邏輯} catch (Exception e) {
} finally { globalLockComponent.releasLock(name)
關于怎么在 MySQL 數據庫中實現一個分布式鎖就分享到這里了,希望以上內容可以對大家有一定的幫助,可以學到更多知識。如果覺得文章不錯,可以把它分享出去讓更多的人看到。
向 AI 問一下細節