package cn.iocoder.yudao.module.erp.controller.admin.sale; import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog; import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum; import cn.iocoder.yudao.framework.common.pojo.CommonResult; import cn.iocoder.yudao.framework.common.pojo.PageParam; 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.erp.controller.admin.sale.vo.customer.ErpCustomerPageReqVO; import cn.iocoder.yudao.module.erp.controller.admin.sale.vo.customer.ErpCustomerRespVO; import cn.iocoder.yudao.module.erp.controller.admin.sale.vo.customer.ErpCustomerSaveReqVO; import cn.iocoder.yudao.module.erp.dal.dataobject.sale.ErpCustomerDO; import cn.iocoder.yudao.module.erp.service.sale.ErpCustomerService; import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.Parameter; import io.swagger.v3.oas.annotations.tags.Tag; import jakarta.annotation.Resource; import jakarta.servlet.http.HttpServletResponse; import jakarta.validation.Valid; import org.springframework.security.access.prepost.PreAuthorize; import org.springframework.validation.annotation.Validated; import org.springframework.web.bind.annotation.*; import java.io.IOException; import java.util.List; import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.EXPORT; import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success; import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertList; @Tag(name = "管理后台 - ERP 客户") @RestController @RequestMapping("/erp/customer") @Validated public class ErpCustomerController { @Resource private ErpCustomerService customerService; @PostMapping("/create") @Operation(summary = "创建客户") @PreAuthorize("@ss.hasPermission('erp:customer:create')") public CommonResult createCustomer(@Valid @RequestBody ErpCustomerSaveReqVO createReqVO) { return success(customerService.createCustomer(createReqVO)); } @PutMapping("/update") @Operation(summary = "更新客户") @PreAuthorize("@ss.hasPermission('erp:customer:update')") public CommonResult updateCustomer(@Valid @RequestBody ErpCustomerSaveReqVO updateReqVO) { customerService.updateCustomer(updateReqVO); return success(true); } @DeleteMapping("/delete") @Operation(summary = "删除客户") @Parameter(name = "id", description = "编号", required = true) @PreAuthorize("@ss.hasPermission('erp:customer:delete')") public CommonResult deleteCustomer(@RequestParam("id") Long id) { customerService.deleteCustomer(id); return success(true); } @GetMapping("/get") @Operation(summary = "获得客户") @Parameter(name = "id", description = "编号", required = true, example = "1024") @PreAuthorize("@ss.hasPermission('erp:customer:query')") public CommonResult getCustomer(@RequestParam("id") Long id) { ErpCustomerDO customer = customerService.getCustomer(id); return success(BeanUtils.toBean(customer, ErpCustomerRespVO.class)); } @GetMapping("/page") @Operation(summary = "获得客户分页") @PreAuthorize("@ss.hasPermission('erp:customer:query')") public CommonResult> getCustomerPage(@Valid ErpCustomerPageReqVO pageReqVO) { PageResult pageResult = customerService.getCustomerPage(pageReqVO); return success(BeanUtils.toBean(pageResult, ErpCustomerRespVO.class)); } @GetMapping("/simple-list") @Operation(summary = "获得客户精简列表", description = "只包含被开启的客户,主要用于前端的下拉选项") public CommonResult> getCustomerSimpleList() { List list = customerService.getCustomerListByStatus(CommonStatusEnum.ENABLE.getStatus()); return success(convertList(list, customer -> new ErpCustomerRespVO().setId(customer.getId()).setName(customer.getName()))); } @GetMapping("/export-excel") @Operation(summary = "导出客户 Excel") @PreAuthorize("@ss.hasPermission('erp:customer:export')") @ApiAccessLog(operateType = EXPORT) public void exportCustomerExcel(@Valid ErpCustomerPageReqVO pageReqVO, HttpServletResponse response) throws IOException { pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE); List list = customerService.getCustomerPage(pageReqVO).getList(); // 导出 Excel ExcelUtils.write(response, "客户.xls", "数据", ErpCustomerRespVO.class, BeanUtils.toBean(list, ErpCustomerRespVO.class)); } }