服务器之家:专注于服务器技术及软件下载分享
分类导航

PHP教程|ASP.NET教程|Java教程|ASP教程|编程技术|正则表达式|C/C++|IOS|C#|Swift|Android|VB|R语言|JavaScript|易语言|vb.net|

服务器之家 - 编程语言 - Java教程 - java基于mongodb实现分布式锁的示例代码

java基于mongodb实现分布式锁的示例代码

2021-11-02 13:29jqpeng的技术记事本 Java教程

本文主要介绍了java基于mongodb实现分布式锁,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下

原理

通过线程安全findAndModify 实现锁

实现

定义锁存储对象:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
/**
 * mongodb 分布式锁
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
@Document(collection = "distributed-lock-doc")
public class LockDocument {
    @Id
    private String id;
    private long expireAt;
    private String token;
}

定义Lock API:

?
1
2
3
4
5
6
7
8
public interface LockService {
 
    String acquire(String key, long expiration);
 
    boolean release(String key, String token);
 
    boolean refresh(String key, String token, long expiration);
}

获取锁:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
@Override
  public String acquire(String key, long expiration) {
      Query query = Query.query(Criteria.where("_id").is(key));
      String token = this.generateToken();
      Update update = new Update()
          .setOnInsert("_id", key)
          .setOnInsert("expireAt", System.currentTimeMillis() + expiration)
          .setOnInsert("token", token);
 
      FindAndModifyOptions options = new FindAndModifyOptions().upsert(true)
                                                               .returnNew(true);
      LockDocument doc = mongoTemplate.findAndModify(query, update, options,
                                                     LockDocument.class);
      boolean locked = doc.getToken() != null && doc.getToken().equals(token);
 
      // 如果已过期
      if (!locked && doc.getExpireAt() < System.currentTimeMillis()) {
          DeleteResult deleted = this.mongoTemplate.remove(
              Query.query(Criteria.where("_id").is(key)
                                  .and("token").is(doc.getToken())
                                  .and("expireAt").is(doc.getExpireAt())),
              LockDocument.class);
          if (deleted.getDeletedCount() >= 1) {
              // 成功释放锁, 再次尝试获取锁
              return this.acquire(key, expiration);
          }
      }
 
      log.debug("Tried to acquire lock for key {} with token {} . Locked: {}",
                key, token, locked);
      return locked ? token : null;
  }

原理:

  • 先尝试upsert锁对象,如果成功且token一致,说明拿到锁
  • 否则加锁失败
  • 如果未拿到锁,但是锁已过期,尝试删除锁
    • 如果删除成功,再次尝试拿锁
    • 如果失败,说明锁可能已经续期了

释放和续期锁:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
@Override
public boolean release(String key, String token) {
    Query query = Query.query(Criteria.where("_id").is(key)
                                      .and("token").is(token));
    DeleteResult deleted = mongoTemplate.remove(query, LockDocument.class);
    boolean released = deleted.getDeletedCount() == 1;
    if (released) {
        log.debug("Remove query successfully affected 1 record for key {} with token {}",
                  key, token);
    } else if (deleted.getDeletedCount() > 0) {
        log.error("Unexpected result from release for key {} with token {}, released {}",
                  key, token, deleted);
    } else {
        log.error("Remove query did not affect any records for key {} with token {}",
                  key, token);
    }
 
    return released;
}
 
@Override
public boolean refresh(String key, String token,
                       long expiration) {
    Query query = Query.query(Criteria.where("_id").is(key)
                                      .and("token").is(token));
    Update update = Update.update("expireAt",
                                  System.currentTimeMillis() + expiration);
    UpdateResult updated =
        mongoTemplate.updateFirst(query, update, LockDocument.class);
 
    final boolean refreshed = updated.getModifiedCount() == 1;
    if (refreshed) {
        log.debug("Refresh query successfully affected 1 record for key {} " +
                  "with token {}", key, token);
    } else if (updated.getModifiedCount() > 0) {
        log.error("Unexpected result from refresh for key {} with token {}, " +
                  "released {}", key, token, updated);
    } else {
        log.warn("Refresh query did not affect any records for key {} with token {}. " +
                 "This is possible when refresh interval fires for the final time " +
                 "after the lock has been released",
                 key, token);
    }
 
    return refreshed;
}

使用  

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
private  LockService lockService;
 
private void tryAcquireLockAndSchedule() {
        while (!this.stopSchedule) {
            // 尝试拿锁
            this.token = this.lockService.acquire(SCHEDULER_LOCK, 20000);
            if (this.token != null) {
    // 拿到锁
            } else {
                // 等待LOCK_EXPIRATION, 再次尝试
                Thread.sleep(LOCK_EXPIRATION);
            }
        }
    }
  • 先尝试拿锁,如果获取到token,说明拿锁成功
  • 否则可以sleep一段时间后再拿锁

完整代码,可到github查看 https://github.com/jadepeng/docker-pipeline/blob/main/pipeline-master/src/main/java/com/github/jadepeng/pipeline/service/impl/MongoLockService.java

到此这篇关于java基于mongodb实现分布式锁的示例代码的文章就介绍到这了,更多相关java mongodb实现分布式锁内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!

原文链接:https://www.cnblogs.com/xiaoqi/p/mongodb-lock.html

延伸 · 阅读

精彩推荐
  • Java教程Java实现抢红包功能

    Java实现抢红包功能

    这篇文章主要为大家详细介绍了Java实现抢红包功能,采用多线程模拟多人同时抢红包,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙...

    littleschemer13532021-05-16
  • Java教程20个非常实用的Java程序代码片段

    20个非常实用的Java程序代码片段

    这篇文章主要为大家分享了20个非常实用的Java程序片段,对java开发项目有所帮助,感兴趣的小伙伴们可以参考一下 ...

    lijiao5352020-04-06
  • Java教程xml与Java对象的转换详解

    xml与Java对象的转换详解

    这篇文章主要介绍了xml与Java对象的转换详解的相关资料,需要的朋友可以参考下...

    Java教程网2942020-09-17
  • Java教程小米推送Java代码

    小米推送Java代码

    今天小编就为大家分享一篇关于小米推送Java代码,小编觉得内容挺不错的,现在分享给大家,具有很好的参考价值,需要的朋友一起跟随小编来看看吧...

    富贵稳中求8032021-07-12
  • Java教程Java BufferWriter写文件写不进去或缺失数据的解决

    Java BufferWriter写文件写不进去或缺失数据的解决

    这篇文章主要介绍了Java BufferWriter写文件写不进去或缺失数据的解决方案,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望...

    spcoder14552021-10-18
  • Java教程Java使用SAX解析xml的示例

    Java使用SAX解析xml的示例

    这篇文章主要介绍了Java使用SAX解析xml的示例,帮助大家更好的理解和学习使用Java,感兴趣的朋友可以了解下...

    大行者10067412021-08-30
  • Java教程升级IDEA后Lombok不能使用的解决方法

    升级IDEA后Lombok不能使用的解决方法

    最近看到提示IDEA提示升级,寻思已经有好久没有升过级了。升级完毕重启之后,突然发现好多错误,本文就来介绍一下如何解决,感兴趣的可以了解一下...

    程序猿DD9332021-10-08
  • Java教程Java8中Stream使用的一个注意事项

    Java8中Stream使用的一个注意事项

    最近在工作中发现了对于集合操作转换的神器,java8新特性 stream,但在使用中遇到了一个非常重要的注意点,所以这篇文章主要给大家介绍了关于Java8中S...

    阿杜7482021-02-04