|
@@ -1,34 +1,41 @@
|
|
|
package cn.iocoder.yudao.module.pms.controller.admin.maintenance;
|
|
|
|
|
|
+import cn.hutool.core.collection.CollUtil;
|
|
|
+import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
|
|
|
+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.collection.MapUtils;
|
|
|
+import cn.iocoder.yudao.framework.common.util.number.NumberUtils;
|
|
|
+import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
|
|
|
+import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
|
|
|
import cn.iocoder.yudao.module.pms.controller.admin.maintenance.vo.IotMaintenancePlanPageReqVO;
|
|
|
import cn.iocoder.yudao.module.pms.controller.admin.maintenance.vo.IotMaintenancePlanRespVO;
|
|
|
import cn.iocoder.yudao.module.pms.controller.admin.maintenance.vo.IotMaintenancePlanSaveReqVO;
|
|
|
+import cn.iocoder.yudao.module.pms.controller.admin.maintenance.vo.IotMaintenanceSaveVO;
|
|
|
import cn.iocoder.yudao.module.pms.dal.dataobject.maintenance.IotMaintenancePlanDO;
|
|
|
import cn.iocoder.yudao.module.pms.service.maintenance.IotMaintenancePlanService;
|
|
|
-import org.springframework.web.bind.annotation.*;
|
|
|
-import javax.annotation.Resource;
|
|
|
-import org.springframework.validation.annotation.Validated;
|
|
|
-import org.springframework.security.access.prepost.PreAuthorize;
|
|
|
-import io.swagger.v3.oas.annotations.tags.Tag;
|
|
|
-import io.swagger.v3.oas.annotations.Parameter;
|
|
|
+import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
|
|
|
+import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
|
|
|
import io.swagger.v3.oas.annotations.Operation;
|
|
|
+import io.swagger.v3.oas.annotations.Parameter;
|
|
|
+import io.swagger.v3.oas.annotations.tags.Tag;
|
|
|
+import org.springframework.security.access.prepost.PreAuthorize;
|
|
|
+import org.springframework.validation.annotation.Validated;
|
|
|
+import org.springframework.web.bind.annotation.*;
|
|
|
|
|
|
-import javax.validation.constraints.*;
|
|
|
-import javax.validation.*;
|
|
|
-import javax.servlet.http.*;
|
|
|
-import java.util.*;
|
|
|
+import javax.annotation.Resource;
|
|
|
+import javax.servlet.http.HttpServletResponse;
|
|
|
+import javax.validation.Valid;
|
|
|
import java.io.IOException;
|
|
|
+import java.util.Collections;
|
|
|
+import java.util.List;
|
|
|
+import java.util.Map;
|
|
|
+import java.util.stream.Stream;
|
|
|
|
|
|
-import cn.iocoder.yudao.framework.common.pojo.PageParam;
|
|
|
-import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
|
|
-import cn.iocoder.yudao.framework.common.pojo.CommonResult;
|
|
|
-import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
|
|
|
+import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.EXPORT;
|
|
|
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
|
|
|
-
|
|
|
-import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
|
|
|
-
|
|
|
-import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
|
|
|
-import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.*;
|
|
|
+import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertListByFlatMap;
|
|
|
|
|
|
|
|
|
@Tag(name = "管理后台 - 保养计划")
|
|
@@ -39,11 +46,13 @@ public class IotMaintenancePlanController {
|
|
|
|
|
|
@Resource
|
|
|
private IotMaintenancePlanService iotMaintenancePlanService;
|
|
|
+ @Resource
|
|
|
+ private AdminUserApi adminUserApi;
|
|
|
|
|
|
@PostMapping("/create")
|
|
|
@Operation(summary = "创建保养计划")
|
|
|
@PreAuthorize("@ss.hasPermission('rq:iot-maintenance-plan:create')")
|
|
|
- public CommonResult<Long> createIotMaintenancePlan(@Valid @RequestBody IotMaintenancePlanSaveReqVO createReqVO) {
|
|
|
+ public CommonResult<Long> createIotMaintenancePlan(@Valid @RequestBody IotMaintenanceSaveVO createReqVO) {
|
|
|
return success(iotMaintenancePlanService.createIotMaintenancePlan(createReqVO));
|
|
|
}
|
|
|
|
|
@@ -78,7 +87,7 @@ public class IotMaintenancePlanController {
|
|
|
@PreAuthorize("@ss.hasPermission('rq:iot-maintenance-plan:query')")
|
|
|
public CommonResult<PageResult<IotMaintenancePlanRespVO>> getIotMaintenancePlanPage(@Valid IotMaintenancePlanPageReqVO pageReqVO) {
|
|
|
PageResult<IotMaintenancePlanDO> pageResult = iotMaintenancePlanService.getIotMaintenancePlanPage(pageReqVO);
|
|
|
- return success(BeanUtils.toBean(pageResult, IotMaintenancePlanRespVO.class));
|
|
|
+ return success(new PageResult<>(buildMaintenancePlanList(pageResult.getList()), pageResult.getTotal()));
|
|
|
}
|
|
|
|
|
|
@GetMapping("/export-excel")
|
|
@@ -94,4 +103,19 @@ public class IotMaintenancePlanController {
|
|
|
BeanUtils.toBean(list, IotMaintenancePlanRespVO.class));
|
|
|
}
|
|
|
|
|
|
+ private List<IotMaintenancePlanRespVO> buildMaintenancePlanList(List<IotMaintenancePlanDO> plans) {
|
|
|
+ if (CollUtil.isEmpty(plans)) {
|
|
|
+ return Collections.emptyList();
|
|
|
+ }
|
|
|
+ // 查询保养计划 负责人
|
|
|
+ Map<Long, AdminUserRespDTO> userMap = adminUserApi.getUserMap(convertListByFlatMap(plans,
|
|
|
+ plan -> Stream.of(NumberUtils.parseLong(plan.getResponsiblePerson()))));
|
|
|
+ // 2. 转换成 VO
|
|
|
+ return BeanUtils.toBean(plans, IotMaintenancePlanRespVO.class, planVO -> {
|
|
|
+ // 2.2 设置创建人、负责人名称
|
|
|
+ MapUtils.findAndThen(userMap, NumberUtils.parseLong(planVO.getResponsiblePerson()),
|
|
|
+ user -> planVO.setResponsiblePersonName(user.getNickname()));
|
|
|
+ });
|
|
|
+ }
|
|
|
+
|
|
|
}
|