From de0ae7016545c094bf7c7f604b009feeff2f0cca Mon Sep 17 00:00:00 2001
From: johnniang <johnniang@riseup.net>
Date: Tue, 19 Mar 2019 20:32:32 +0800
Subject: [PATCH] Complete count api

---
 .../ryanc/halo/model/dto/CountOutputDTO.java  | 33 +++++++++++++
 .../controller/admin/api/AdminController.java | 49 +++++++++++++++++++
 2 files changed, 82 insertions(+)
 create mode 100644 src/main/java/cc/ryanc/halo/model/dto/CountOutputDTO.java
 create mode 100644 src/main/java/cc/ryanc/halo/web/controller/admin/api/AdminController.java

diff --git a/src/main/java/cc/ryanc/halo/model/dto/CountOutputDTO.java b/src/main/java/cc/ryanc/halo/model/dto/CountOutputDTO.java
new file mode 100644
index 000000000..2de979b74
--- /dev/null
+++ b/src/main/java/cc/ryanc/halo/model/dto/CountOutputDTO.java
@@ -0,0 +1,33 @@
+package cc.ryanc.halo.model.dto;
+
+import lombok.Data;
+
+/**
+ * Count output DTO.
+ *
+ * @author johnniang
+ * @date 3/19/19
+ */
+@Data
+public class CountOutputDTO {
+
+    /**
+     * Post count.
+     */
+    private Long postCount;
+
+    /**
+     * Comment count.
+     */
+    private Long commentCount;
+
+    /**
+     * Attachment count.
+     */
+    private Long attachmentCount;
+
+    /**
+     * Establish days
+     */
+    private Long establishDays;
+}
diff --git a/src/main/java/cc/ryanc/halo/web/controller/admin/api/AdminController.java b/src/main/java/cc/ryanc/halo/web/controller/admin/api/AdminController.java
new file mode 100644
index 000000000..43bb72df8
--- /dev/null
+++ b/src/main/java/cc/ryanc/halo/web/controller/admin/api/AdminController.java
@@ -0,0 +1,49 @@
+package cc.ryanc.halo.web.controller.admin.api;
+
+import cc.ryanc.halo.model.dto.CountOutputDTO;
+import cc.ryanc.halo.model.enums.BlogProperties;
+import cc.ryanc.halo.model.enums.PostStatus;
+import cc.ryanc.halo.model.enums.PostType;
+import cc.ryanc.halo.service.AttachmentService;
+import cc.ryanc.halo.service.CommentService;
+import cc.ryanc.halo.service.OptionService;
+import cc.ryanc.halo.service.PostService;
+import io.swagger.annotations.ApiOperation;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+/**
+ * @author johnniang
+ * @date 3/19/19
+ */
+@RestController
+@RequestMapping("/admin/api")
+public class AdminController {
+
+    private final PostService postService;
+
+    private final AttachmentService attachmentService;
+
+    private final CommentService commentService;
+
+    private final OptionService optionService;
+
+    public AdminController(PostService postService, AttachmentService attachmentService, CommentService commentService, OptionService optionService) {
+        this.postService = postService;
+        this.attachmentService = attachmentService;
+        this.commentService = commentService;
+        this.optionService = optionService;
+    }
+
+    @GetMapping("counts")
+    @ApiOperation("Gets count info")
+    public CountOutputDTO getCount() {
+        CountOutputDTO countOutputDTO = new CountOutputDTO();
+        countOutputDTO.setPostCount(postService.countByStatus(PostStatus.PUBLISHED, PostType.POST));
+        countOutputDTO.setAttachmentCount(attachmentService.count());
+        countOutputDTO.setCommentCount(commentService.count());
+        countOutputDTO.setEstablishDays(Long.valueOf(optionService.getByProperty(BlogProperties.WIDGET_DAYCOUNT).orElse("0")));
+        return countOutputDTO;
+    }
+}