教育训练处考试制证系统后端
“djh”
2025-03-06 6061c45849de0f3ac6d05fdfa2bac4b09c21179b
exam-system/src/main/java/com/gkhy/exam/pay/controller/CoalCategoryController.java
@@ -1,9 +1,13 @@
package com.gkhy.exam.pay.controller;
import com.gkhy.exam.pay.dto.rep.CategoryCount;
import com.gkhy.exam.pay.dto.rep.NonCategoryCount;
import com.gkhy.exam.pay.dto.req.CountReq;
import com.gkhy.exam.pay.entity.CoalCategory;
import com.gkhy.exam.pay.entity.CoalTicket;
import com.gkhy.exam.pay.entity.NonCoalCategory;
import com.gkhy.exam.pay.service.CoalCategoryService;
import com.ruoyi.common.constant.HttpStatus;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.page.TableDataInfo;
@@ -15,6 +19,7 @@
import org.springframework.web.bind.annotation.*;
import java.util.List;
import java.util.stream.Collectors;
@RestController
@Api(tags = "煤矿工种类别管理")
@@ -98,4 +103,27 @@
    public AjaxResult ticketList(){
        return success(coalCategoryService.ticketList());
    }
    /**
     * 工种缴费统计
     */
    @GetMapping("/count")
    @ApiOperation(value = "工种缴费统计")
    public TableDataInfo count(CountReq countReq){
        List<CategoryCount> categoryCounts = coalCategoryService.countCategory(countReq);
        int pageNum = countReq.getPageNum(); // 当前页码
        int pageSize = countReq.getPageSize(); // 每页大小
        List<CategoryCount> pagedList = categoryCounts.stream()
                .skip((pageNum - 1) * pageSize) // 跳过前面的数据
                .limit(pageSize) // 限制当前页的数据量
                .collect(Collectors.toList());
        // 封装分页结果
        TableDataInfo dataTable = new TableDataInfo();
        dataTable.setCode(HttpStatus.SUCCESS);
        dataTable.setMsg("查询成功");
        dataTable.setRows(pagedList); // 当前页数据
        dataTable.setTotal(categoryCounts.size()); // 总数据量
        return dataTable;
    }
}