package cn.iocoder.yudao.module.mes.controller.admin.em.quota; import cn.iocoder.yudao.framework.common.pojo.CommonResult; import cn.iocoder.yudao.framework.common.pojo.PageResult; import cn.iocoder.yudao.module.mes.controller.admin.em.quota.vo.MesEmQuotaPageReqVO; import cn.iocoder.yudao.module.mes.controller.admin.em.quota.vo.MesEmQuotaRespVO; import cn.iocoder.yudao.module.mes.controller.admin.em.quota.vo.MesEmQuotaSaveReqVO; import cn.iocoder.yudao.module.mes.dal.dataobject.em.quota.MesEmQuotaDO; import cn.iocoder.yudao.module.mes.service.em.quota.MesEmQuotaService; 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.validation.Valid; import org.springframework.security.access.prepost.PreAuthorize; import org.springframework.validation.annotation.Validated; import org.springframework.web.bind.annotation.*; import java.util.Collections; import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success; @Tag(name = "管理后台 - MES 能耗考核指标") @RestController @RequestMapping("/mes/em/quota") @Validated public class MesEmQuotaController { @Resource private MesEmQuotaService quotaService; @PostMapping("/create") @Operation(summary = "创建能耗考核指标") @PreAuthorize("@ss.hasPermission('mes:em-quota:create')") public CommonResult createQuota(@Valid @RequestBody MesEmQuotaSaveReqVO createReqVO) { return success(quotaService.createQuota(createReqVO)); } @PutMapping("/update") @Operation(summary = "更新能耗考核指标") @PreAuthorize("@ss.hasPermission('mes:em-quota:update')") public CommonResult updateQuota(@Valid @RequestBody MesEmQuotaSaveReqVO updateReqVO) { quotaService.updateQuota(updateReqVO); return success(true); } @DeleteMapping("/delete") @Operation(summary = "删除能耗考核指标") @Parameter(name = "id", description = "编号", required = true) @PreAuthorize("@ss.hasPermission('mes:em-quota:delete')") public CommonResult deleteQuota(@RequestParam("id") Long id) { quotaService.deleteQuota(id); return success(true); } @GetMapping("/get") @Operation(summary = "获得能耗考核指标") @Parameter(name = "id", description = "编号", required = true) public CommonResult getQuota(@RequestParam("id") Long id) { MesEmQuotaDO quota = quotaService.getQuota(id); if (quota == null) { return success(null); } return success(quotaService.getQuotaRespVOList(Collections.singletonList(quota)).get(0)); } @GetMapping("/page") @Operation(summary = "获得能耗考核指标分页") public CommonResult> getQuotaPage(@Valid MesEmQuotaPageReqVO pageReqVO) { PageResult pageResult = quotaService.getQuotaPage(pageReqVO); return success(new PageResult<>(quotaService.getQuotaRespVOList(pageResult.getList()), pageResult.getTotal())); } }