feat: add query approval total function

Author: jianggang

src/main/java/com/featureprobe/api/controller/ApprovalRecordController.java

@@ -1,9 +1,13 @@
 package com.featureprobe.api.controller;
 
 import com.featureprobe.api.base.doc.DefaultApiResponses;
+import com.featureprobe.api.base.doc.GetApiResponse;
+import com.featureprobe.api.base.enums.ApprovalStatusEnum;
 import com.featureprobe.api.dto.ApprovalRecordQueryRequest;
 import com.featureprobe.api.dto.ApprovalRecordResponse;
+import com.featureprobe.api.dto.ApprovalTotalResponse;
 import com.featureprobe.api.service.ApprovalRecordService;
+import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import lombok.AllArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -25,8 +29,17 @@ public class ApprovalRecordController {
     private ApprovalRecordService approvalRecordService;
 
     @GetMapping
+    @GetApiResponse
+    @Operation(summary = "Get approval list", description = "Get approval list.")
     public Page<ApprovalRecordResponse> list(@Validated ApprovalRecordQueryRequest queryRequest) {
         return approvalRecordService.list(queryRequest);
     }
 
+    @GetMapping("/total")
+    @GetApiResponse
+    @Operation(summary = "Get approval total by status", description = "Get approval list.")
+    public ApprovalTotalResponse total(ApprovalStatusEnum status) {
+        return new ApprovalTotalResponse(approvalRecordService.total(status));
+    }
+
 }

src/main/java/com/featureprobe/api/dto/ApprovalTotalResponse.java

@@ -0,0 +1,14 @@
+package com.featureprobe.api.dto;
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+public class ApprovalTotalResponse {
+
+    private long total;
+
+}

src/main/java/com/featureprobe/api/repository/ApprovalRecordRepository.java

@@ -1,5 +1,6 @@
 package com.featureprobe.api.repository;
 
+import com.featureprobe.api.base.enums.ApprovalStatusEnum;
 import com.featureprobe.api.entity.ApprovalRecord;
 import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.data.jpa.repository.JpaSpecificationExecutor;
@@ -14,4 +15,5 @@ public interface ApprovalRecordRepository extends JpaRepository<ApprovalRecord,
     Optional<ApprovalRecord> findByProjectKeyAndEnvironmentKeyAndToggleKey(String projectKey, String environmentKey,
                                                                            String toggleKey);
 
+    long countByStatusAndReviewersIsContaining(ApprovalStatusEnum status, String account);
 }

src/main/java/com/featureprobe/api/service/ApprovalRecordService.java

@@ -1,6 +1,7 @@
 package com.featureprobe.api.service;
 
 import com.featureprobe.api.auth.TokenHelper;
+import com.featureprobe.api.base.enums.ApprovalStatusEnum;
 import com.featureprobe.api.base.enums.ApprovalTypeEnum;
 import com.featureprobe.api.base.enums.ResourceType;
 import com.featureprobe.api.base.enums.SketchStatusEnum;
@@ -75,6 +76,11 @@ public Page<ApprovalRecordResponse> list(ApprovalRecordQueryRequest queryRequest
         return new PageImpl<>(sortedRes, pageable, res.getTotalElements());
     }
 
+    public long total(ApprovalStatusEnum status) {
+        return approvalRecordRepository.countByStatusAndReviewersIsContaining(status,
+                "\"" + TokenHelper.getAccount() + "\"");
+    }
+
     private Specification<ApprovalRecord> buildListSpec(ApprovalRecordQueryRequest queryRequest) {
         return (root, query, cb) -> {
             Predicate p1 = cb.equal(root.get("submitBy"), TokenHelper.getAccount());

src/test/groovy/com/featureprobe/api/service/ApprovalRecordServiceSpec.groovy

@@ -1,5 +1,6 @@
 package com.featureprobe.api.service
 
+import com.featureprobe.api.auth.TokenHelper
 import com.featureprobe.api.base.enums.ApprovalStatusEnum
 import com.featureprobe.api.base.enums.ApprovalTypeEnum
 import com.featureprobe.api.dto.ApprovalRecordQueryRequest
@@ -16,6 +17,10 @@ import com.featureprobe.api.repository.ToggleRepository
 import org.hibernate.internal.SessionImpl
 import org.springframework.data.domain.PageImpl
 import org.springframework.data.domain.Pageable
+import org.springframework.security.core.context.SecurityContextHolder
+import org.springframework.security.core.context.SecurityContextImpl
+import org.springframework.security.oauth2.jwt.Jwt
+import org.springframework.security.oauth2.server.resource.authentication.JwtAuthenticationToken
 import spock.lang.Specification
 
 import javax.persistence.EntityManager
@@ -56,5 +61,20 @@ class ApprovalRecordServiceSpec extends Specification {
         1 == list.content.size()
     }
 
+    def "Query approval record total by status" () {
+        given:
+        setAuthContext("Admin", "ADMIN")
+        when:
+        def total = approvalRecordService.total(ApprovalStatusEnum.PENDING)
+        then:
+        1 * approvalRecordRepository.countByStatusAndReviewersIsContaining(ApprovalStatusEnum.PENDING, "\"Admin\"") >> 1
+        1 == total
+    }
+
+    private setAuthContext(String account, String role) {
+        SecurityContextHolder.setContext(new SecurityContextImpl(
+                new JwtAuthenticationToken(new Jwt.Builder("21212").header("a","a")
+                        .claim("role", role).claim("account", account).build())))
+    }
 }