pref: 预览草稿时，不会增加访问量 (#834)

halo-dev · Aug 6, 2020 · 3f2fefa · 3f2fefa
1 parent 782079d
commit 3f2fefa
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/main/java/run/halo/app/service/impl/BasePostServiceImpl.java b/src/main/java/run/halo/app/service/impl/BasePostServiceImpl.java
@@ -215,9 +215,16 @@ public void increaseVisit(long visits, Integer postId) {
         Assert.isTrue(visits > 0, "Visits to increase must not be less than 1");
         Assert.notNull(postId, "Post id must not be null");
 
-        long affectedRows = basePostRepository.updateVisit(visits, postId);
+        boolean finishedIncrease;
+        if (basePostRepository.getByIdAndStatus(postId, PostStatus.DRAFT).isPresent())
+        {
+            finishedIncrease = true;
+            log.info("Post with id: [{}] is a draft and visits will not be updated", postId);
+        } else {
+            finishedIncrease = basePostRepository.updateVisit(visits, postId) == 1;
+        }
 
-        if (affectedRows != 1) {
+        if (!finishedIncrease) {
             log.error("Post with id: [{}] may not be found", postId);
             throw new BadRequestException("Failed to increase visits " + visits + " for post with id " + postId);
         }