Spring Boot+SQL/JPA实战悲观锁和乐观锁
原创业务还原
首先环境是:Spring Boot 2.1.0 + data-jpa + mysql + lombok
数据库设计
对于一个有评论功能的博客系统来说,通常会有两个表:1.文章表 2.评论表。其中文章表除了保存一些文章信息等,还有个字段保存评论数量。我们设计一个最精简的表结构来还原该业务场景。
article 文章表
字段 |
类型 |
备注 |
---|---|---|
id |
INT |
自增主键id |
title |
VARCHAR |
文章标题 |
comment_count |
INT |
文章的评论数量 |
comment 评论表
字段 |
类型 |
备注 |
---|---|---|
id |
INT |
自增主键id |
article_id |
INT |
评论的文章id |
content |
VARCHAR |
评论内容 |
当一个用户评论的时候,1. 根据文章id获取到文章 2. 插入一条评论记录 3. 该文章的评论数增加并保存
代码实现
首先在maven中引入对应的依赖
<parent>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-parent</artifactId>
<version>2.1.0.RELEASE</version>
<relativePath/> <!-- lookup parent from repository -->
</parent>
<dependencies>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-jpa</artifactId>
</dependency>
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
</dependency>
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<optional>true</optional>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-test</artifactId>
<scope>test</scope>
</dependency>
</dependencies>
然后编写对应 数据库 的实体类
@Data
@Entity
public class Article {
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Long id;
private String title;
private Long commentCount;
}
@Data
@Entity
public class Comment {
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Long id;
private Long articleId;
private String content;
}
接着创建这两个实体类对应的Repository,由于spring-jpa-data的
CrudRepository
已经帮我们实现了最常见的CRUD操作,所以我们的Repository只需要继承
CrudRepository
接口其他啥都不用做。
public interface ArticleRepository extends CrudRepository<Article, Long> {
}
public interface CommentRepository extends CrudRepository<Comment, Long> {
}
接着我们就简单的实现一下Controller接口和Service实现类。
@Slf4j
@RestController
public class CommentController {
@Autowired
private CommentService commentService;
@PostMapping("comment")
public String comment(Long articleId, String content) {
try {
commentService.postComment(articleId, content);
} catch (Exception e) {
log.error("{}", e);
return "error: " + e.getMessage();
return "success";
}
@Slf4j
@Service
public class CommentService {
@Autowired
private ArticleRepository articleRepository;
@Autowired
private CommentRepository commentRepository;
public void postComment(Long articleId, String content) {
Optional<Article> articleOptional = articleRepository.findById(articleId);
if (!articleOptional.isPresent()) {
throw new RuntimeException("没有对应的文章");
Article article = articleOptional.get();
Comment comment = new Comment();
comment.setArticleId(articleId);
comment.setContent(content);
commentRepository.save(comment);
article.setCommentCount(article.getCommentCount() + 1);
articleRepository.save(article);
}
并发问题分析
从刚才的代码实现里可以看出这个简单的评论功能的流程,当用户发起评论的请求时,从数据库找出对应的文章的实体类
Article
,然后根据文章信息生成对应的评论实体类
Comment
,并且插入到数据库中,接着增加该文章的评论数量,再把修改后的文章更新到数据库中,整个流程如下流程图。
在这个流程中有个问题,当有多个用户同时并发评论时,他们同时进入步骤1中拿到Article,然后插入对应的Comment,最后在步骤3中更新评论数量保存到数据库。只是由于他们是同时在步骤1拿到的Article,所以他们的Article.commentCount的值相同,那么在步骤3中保存的Article.commentCount+1也相同,那么原来应该+3的评论数量,只加了1。
我们用测试用例代码试一下
@RunWith(SpringRunner.class)
@SpringBootTest(classes = LockAndTransactionApplication.class, webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class CommentControllerTests {
@Autowired
private TestRestTemplate testRestTemplate;
@Test
public void concurrentComment() {
String url = "http://localhost:9090/comment";
for (int i = 0; i < 100; i++) {
int finalI = i;
new Thread(() -> {
MultiValueMap<String, String> params = new LinkedMultiValueMap<>();
params.add("articleId", "1");
params.add("content", "测试内容" + finalI);
String result = testRestTemplate.postForObject(url, params, String.class);
}).start();
}
这里我们开了100个线程,同时发送评论请求,对应的文章id为1。
在发送请求前,数据库数据为
select * from article
select count(*) comment_count from comment
发送请求后,数据库数据为
select * from article
select count(*) comment_count from comment
明显的看到在article表里的comment_count的值不是100,这个值不一定是我图里的14,但是必然是不大于100的,而comment表的数量肯定等于100。
这就展示了在文章开头里提到的并发问题,这种问题其实十分的常见,只要有类似上面这样评论功能的流程的系统,都要小心避免出现这种问题。
下面就用实例展示展示如何通过悲观锁和乐观锁防止出现并发数据问题,同时给出SQL方案和JPA自带方案,SQL方案可以通用“任何系统”,甚至不限语言,而JPA方案十分快捷,如果你恰好用的也是JPA,那就可以简单的使用上乐观锁或悲观锁。最后也会根据业务比较一下乐观锁和悲观锁的一些区别
悲观锁解决并发问题
悲观锁顾名思义就是悲观的认为自己操作的数据都会被其他线程操作,所以就必须自己独占这个数据,可以理解为”独占锁“。在java中
synchronized
和
ReentrantLock
等锁就是悲观锁,数据库中表锁、行锁、读写锁等也是悲观锁。
利用SQL解决并发问题
行锁就是操作数据的时候把这一行数据锁住,其他线程想要读写必须等待,但同一个表的其他数据还是能被其他线程操作的。只要在需要查询的sql后面加上
for update
,就能锁住查询的行,
特别要注意查询条件必须要是索引列,如果不是索引就会变成表锁,把整个表都锁住。
现在在原有的代码的基础上修改一下,先在
ArticleRepository
增加一个手动写sql查询方法。
public interface ArticleRepository extends CrudRepository<Article, Long> {
@Query(value = "select * from article a where a.id = :id for update", nativeQuery = true)
Optional<Article> findArticleForUpdate(Long id);
}
然后把
CommentService
中使用的查询方法由原来的
findById
改为我们自定义的方法
public class CommentService {
public void postComment(Long articleId, String content) {
// Optional<Article> articleOptional = articleRepository.findById(articleId);
Optional<Article> articleOptional = articleRepository.findArticleForUpdate(articleId);
}
这样我们查出来的
Article
,
在我们没有将其提交事务之前,其他线程是不能获取修改的
,保证了同时只有一个线程能操作对应数据。
现在再用测试用例测一下,
article.comment_count
的值必定是100。
利用JPA自带行锁解决并发问题
对于刚才提到的在sql后面增加
for update
,JPA有提供一个更优雅的方式,就是
@Lock
注解,这个注解的参数可以传入想要的锁级别。
现在在
ArticleRepository
中增加JPA的锁方法,其中
LockModeType.PESSIMISTIC_WRITE
参数就是行锁。
public interface ArticleRepository extends CrudRepository<Article, Long> {
@Lock(value = LockModeType.PESSIMISTIC_WRITE)
@Query("select a from Article a where a.id = :id")
Optional<Article> findArticleWithPessimisticLock(Long id);
}
同样的只要在
CommentService
里把查询方法改为
findArticleWithPessimisticLock()
,再测试用例测一下,肯定不会有并发问题。而且这时看一下控制台打印信息,发现实际上查询的sql还是加了
for update
,只不过是JPA帮我们加了而已。
乐观锁解决并发问题
乐观锁顾名思义就是特别乐观,认为自己拿到的资源不会被其他线程操作所以不上锁,只是在插入数据库的时候再判断一下数据有没有被修改。所以悲观锁是限制其他线程,而乐观锁是限制自己,虽然他的名字有锁,但是实际上不算上锁,只是在最后操作的时候再判断具体怎么操作。
乐观锁通常为版本号机制或者CAS算法
利用SQL实现版本号解决并发问题
版本号机制就是在数据库中加一个字段当作版本号,比如我们加个字段version。那么这时候拿到
Article
的时候就会带一个版本号,比如拿到的版本是1,然后你对这个
Article
一通操作,操作完之后要插入到数据库了。发现哎呀,怎么数据库里的
Article
版本是2,和我手里的版本不一样啊,说明我手里的
Article
不是最新的了,那么就不能放到数据库了。这样就避免了并发时数据冲突的问题。
所以我们现在给article表加一个字段version
article 文章表
字段 |
类型 |
备注 |
---|---|---|
version |
INT DEFAULT 0 |
版本号 |
然后对应的实体类也增加version字段
@Data
@Entity
public class Article {
private Long version;
}
接着在
ArticleRepository
增加更新的方法,注意这里是更新方法,和悲观锁时增加查询方法不同。
public interface ArticleRepository extends CrudRepository<Article, Long> {
@Modifying
@Query(value = "update article set comment_count = :commentCount, version = version + 1 where id = :id and version = :version", nativeQuery = true)
int updateArticleWithVersion(Long id, Long commentCount, Long version);
}
可以看到update的where有一个判断version的条件,并且会set version = version + 1。这就保证了只有当数据库里的版本号和要更新的实体类的版本号相同的时候才会更新数据。
接着在
CommentService
里稍微修改一下代码。
// CommentService
public void postComment(Long articleId, String content) {
Optional<Article> articleOptional = articleRepository.findById(articleId);
int count = articleRepository.updateArticleWithVersion(article.getId(), article.getCommentCount() + 1, article.getVersion());
if (count == 0) {
throw new RuntimeException("服务器繁忙,更新数据失败");
// articleRepository.save(article);
}
首先对于
Article
的查询方法只需要普通的
findById()
方法就行不用上任何锁。
然后更新
Article
的时候改用新加的
updateArticleWithVersion()
方法。可以看到这个方法有个返回值,这个返回值代表更新了的数据库行数,如果值为0的时候表示没有符合条件可以更新的行。
这之后就可以由我们自己决定怎么处理了, 这里是直接回滚,spring就会帮我们回滚之前的数据操作,把这次的所有操作都取消以保证数据的一致性 。
现在再用测试用例测一下
select * from article
select count(*) comment_count from comment
现在看到
Article
里的comment_count和
Comment
的数量都不是100了,
但是这两个的值必定是一样的了
。因为刚才我们处理的时候假如
Article
表的数据发生了冲突,那么就不会更新到数据库里,这时抛出异常使其事务回滚,这样就能保证没有更新
Article
的时候
Comment
也不会插入,就解决了数据不统一的问题。
这种直接回滚的处理方式用户体验比较差,通常来说如果判断
Article
更新条数为0时,会尝试重新从数据库里查询信息并重新修改,再次尝试更新数据,如果不行就再查询,直到能够更新为止。当然也不会是无线的循环这样的操作,会设置一个上线,比如循环3次查询修改更新都不行,这时候才会抛出异常。
利用JPA实现版本现解决并发问题
JPA对悲观锁有实现方式,乐观锁自然也是有的,现在就用JPA自带的方法实现乐观锁。
首先在
Article
实体类的version字段上加上
@Version
注解,我们进注解看一下源码的注释,可以看到有部分写到:
The following types are supported for version properties: int, Integer, short, Short, long, Long, java.sql.Timestamp.
注释里面说版本号的类型支持int, short, long三种基本数据类型和他们的包装类以及Timestamp,我们现在用的是Long类型。
@Data
@Entity
public class Article {
@Version
private Long version;
}
接着只需要在
CommentService
里的评论流程修改回我们最开头的“会触发并发问题”的业务代码就行了。说明JPA的这种乐观锁实现方式是非侵入式的。
// CommentService
public void postComment(Long articleId, String content) {
Optional<Article> articleOptional = articleRepository.findById(articleId);