tuqin
2 天以前 531b50cab6e42e6146a419bc71c5b494fff774b2
yudao-module-hrm/src/main/java/cn/iocoder/yudao/module/hrm/controller/admin/attendance/HrmAttendanceRuleController.java
@@ -7,6 +7,9 @@
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmAttendanceRuleImportExcelVO;
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmAttendanceRuleImportReqVO;
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmAttendanceRuleImportRespVO;
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmAttendanceRulePageReqVO;
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmAttendanceRuleRespVO;
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmAttendanceRuleSaveReqVO;
@@ -25,6 +28,8 @@
import org.springframework.web.bind.annotation.*;
import java.io.IOException;
import java.math.BigDecimal;
import java.time.LocalTime;
import java.util.*;
import java.util.stream.Collectors;
@@ -129,4 +134,27 @@
        ExcelUtils.write(response, "考勤规则.xls", "数据", HrmAttendanceRuleRespVO.class, data);
    }
    @GetMapping("/get-import-template")
    @Operation(summary = "获得导入考勤规则模板")
    @PreAuthorize("@ss.hasPermission('hrm:attendance-rule:import')")
    public void importTemplate(HttpServletResponse response) throws IOException {
        // 手动创建导出 demo
        List<HrmAttendanceRuleImportExcelVO> list = Collections.singletonList(
                HrmAttendanceRuleImportExcelVO.builder().name("标准考勤规则").deptName("研发部")
                        .workStartTime(LocalTime.of(9, 0)).workEndTime(LocalTime.of(18, 0))
                        .lateAllowMinutes(10).earlyAllowMinutes(10).overtimeThreshold(BigDecimal.ONE)
                        .status(0).remark("").build()
        );
        // 输出
        ExcelUtils.write(response, "考勤规则导入模板.xls", "考勤规则列表", HrmAttendanceRuleImportExcelVO.class, list);
    }
    @PostMapping("/import")
    @Operation(summary = "导入考勤规则")
    @PreAuthorize("@ss.hasPermission('hrm:attendance-rule:import')")
    public CommonResult<HrmAttendanceRuleImportRespVO> importExcel(@Valid HrmAttendanceRuleImportReqVO importReqVO)
            throws Exception {
        List<HrmAttendanceRuleImportExcelVO> list = ExcelUtils.read(importReqVO.getFile(), HrmAttendanceRuleImportExcelVO.class);
        return success(attendanceRuleService.importAttendanceRuleList(list, importReqVO));
    }
}