使用MySQL實現一個分布式鎖

發布時間: 2019-12-31 13:29:11 來源: 互聯網 欄目: Mysql 點擊:

在分布式系統中,分布鎖是一個最基礎的工具類。這篇文章主要介紹了用MySQL實現一個分布式鎖,本文通過實例代碼相結合給大家介紹的非常詳細,具有一定的參考借鑒價值,需要的朋友可以參考下

介紹

在分布式系統中,分布鎖是一個最基礎的工具類。例如,部署了2個有付款功能的微服務中,用戶有可能對一個訂單發起2次付款操作,而這2次請求可能被發到2個服務中,所以必須得用分布式鎖防止重復提交,獲取到鎖的服務正常進行付款操作,獲取不到鎖的服務提示重復操作。

我司封裝了大量的基礎工具類,當我們想使用分布式鎖的時候只要做3件事情

1.在數據庫中建globallocktable表
2.引入相應的jar包
3.在代碼中寫上@Autowired GlobalLockComponent globalLockComponent即可使用這個組件

看完這篇文章你也可以用springboot-starter的方式實現一個同樣的功能。但我們不是用這個方式來實現的,另開一篇文章分析我們是怎么實現的。

這篇文章先分析一下MySQL分布式的實現

建表

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實現的話可以先刪除后添加??梢钥吹絼h除的時候使用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實現一個分布式鎖,希望對大家有所幫助!

本文標題: 使用MySQL實現一個分布式鎖
本文地址: http://www.1921352.live/shujuku/mysql/296439.html

如果認為本文對您有所幫助請贊助本站

支付寶掃一掃贊助微信掃一掃贊助

  • 支付寶掃一掃贊助
  • 微信掃一掃贊助
  • 支付寶先領紅包再贊助
    聲明:凡注明"本站原創"的所有文字圖片等資料,版權均屬編程客棧所有,歡迎轉載,但務請注明出處。
    mysql中null(IFNULL,COALESCE和NULLIF)相關知識點總結windows7下mysql8.0.18部署安裝教程圖解
    Top 网上挖矿机赚钱