|
@@ -11,7 +11,10 @@ import cn.iocoder.yudao.module.pms.controller.admin.inspect.plan.vo.IotInspectPl
|
|
|
import cn.iocoder.yudao.module.pms.controller.admin.inspect.plan.vo.IotInspectPlanSaveReqVO;
|
|
|
import cn.iocoder.yudao.module.pms.dal.dataobject.inspect.IotInspectPlanDO;
|
|
|
import cn.iocoder.yudao.module.pms.service.inspect.IotInspectPlanService;
|
|
|
+import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
|
|
|
+import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
|
|
|
import cn.iocoder.yudao.module.system.controller.admin.user.vo.user.UserUpdateStatusReqVO;
|
|
|
+import com.xingyuv.captcha.util.StringUtils;
|
|
|
import io.swagger.v3.oas.annotations.Operation;
|
|
|
import io.swagger.v3.oas.annotations.Parameter;
|
|
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
|
@@ -23,8 +26,9 @@ import javax.annotation.Resource;
|
|
|
import javax.servlet.http.HttpServletResponse;
|
|
|
import javax.validation.Valid;
|
|
|
import java.io.IOException;
|
|
|
-import java.util.Arrays;
|
|
|
import java.util.List;
|
|
|
+import java.util.Objects;
|
|
|
+import java.util.stream.Collectors;
|
|
|
|
|
|
import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.EXPORT;
|
|
|
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
|
|
@@ -38,6 +42,8 @@ public class IotInspectPlanController {
|
|
|
|
|
|
@Resource
|
|
|
private IotInspectPlanService iotInspectPlanService;
|
|
|
+ @Resource
|
|
|
+ private AdminUserApi adminUserApi;
|
|
|
|
|
|
@PostMapping("/create")
|
|
|
@Operation(summary = "创建巡检计划")
|
|
@@ -79,7 +85,18 @@ public class IotInspectPlanController {
|
|
|
@PreAuthorize("@ss.hasPermission('rq:iot-inspect-plan:query')")
|
|
|
public CommonResult<PageResult<IotInspectPlanRespVO>> getIotInspectPlanPage(@Valid IotInspectPlanPageReqVO pageReqVO) {
|
|
|
PageResult<IotInspectPlanDO> pageResult = iotInspectPlanService.getIotInspectPlanPage(pageReqVO);
|
|
|
- return success(BeanUtils.toBean(pageResult, IotInspectPlanRespVO.class));
|
|
|
+ List<IotInspectPlanRespVO> collect = pageResult.getList().stream().map(e -> {
|
|
|
+ IotInspectPlanRespVO iotInspectPlanRespVO = new IotInspectPlanRespVO();
|
|
|
+ BeanUtils.copyProperties(e, iotInspectPlanRespVO);
|
|
|
+ if (StringUtils.isNotBlank(e.getCreator())) {
|
|
|
+ AdminUserRespDTO user = adminUserApi.getUser(Long.valueOf(e.getCreator()));
|
|
|
+ if (Objects.nonNull(user)) {
|
|
|
+ iotInspectPlanRespVO.setCreateName(user.getNickname());
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return iotInspectPlanRespVO;
|
|
|
+ }).collect(Collectors.toList());
|
|
|
+ return success(new PageResult<>(collect, pageResult.getTotal()));
|
|
|
}
|
|
|
|
|
|
@GetMapping("/export-excel")
|