skyouc
2025-03-21 beff9e424cbc5a86fb02ae5b089bd6ae9d04d78e
rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/CompanysController.java
@@ -4,6 +4,7 @@
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.vincent.rsf.framework.common.Cools;
import com.vincent.rsf.framework.common.R;
import com.vincent.rsf.framework.exception.CoolException;
import com.vincent.rsf.server.common.utils.ExcelUtil;
import com.vincent.rsf.server.common.annotation.OperationLog;
import com.vincent.rsf.server.common.domain.BaseParam;
@@ -57,10 +58,17 @@
    @OperationLog("Create 字典数据集")
    @PostMapping("/companys/save")
    public R save(@RequestBody Companys companys) {
        if (Objects.isNull(companys.getName())) {
            throw new CoolException("企业名称不能为空!!");
        }
        if (Objects.isNull(companys.getBreifCode())) {
            throw new CoolException("企业助记码不能为空!!");
        }
        if (Objects.isNull(companys.getType())) {
            throw new CoolException("企业类型不能为空!!");
        }
        companys.setCreateBy(getLoginUserId());
        companys.setCreateTime(new Date());
        companys.setUpdateBy(getLoginUserId());
        companys.setUpdateTime(new Date());
        if (!companysService.save(companys)) {
            return R.error("Save Fail");
        }
@@ -106,7 +114,13 @@
    @PreAuthorize("hasAuthority('manager:companys:list')")
    @PostMapping("/companys/export")
    public void export(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception {
        ExcelUtil.build(ExcelUtil.create(companysService.list(), Companys.class), response);
        List<Companys> companies = new ArrayList<>();
        if (!Objects.isNull(map.get("ids"))) {
            companies = companysService.list(new LambdaQueryWrapper<Companys>().in(Companys::getId, map.get("ids")));
        } else {
            companies = companysService.list();
        }
        ExcelUtil.build(ExcelUtil.create(companies, Companys.class), response);
    }
}