From 179c4d64313c9b7572778da4aaaf6c6584fe457d Mon Sep 17 00:00:00 2001 From: mitao <2763622819@qq.com> Date: 星期二, 20 五月 2025 23:48:08 +0800 Subject: [PATCH] 修改文件上传类型限制 --- springcloud_k8s_panzhihuazhihuishequ/service_equipment/src/main/java/com/panzhihua/service_equipment/api/UnionInspectionApi.java | 35 ++++++++++++++++++++++++++++------- 1 files changed, 28 insertions(+), 7 deletions(-) diff --git a/springcloud_k8s_panzhihuazhihuishequ/service_equipment/src/main/java/com/panzhihua/service_equipment/api/UnionInspectionApi.java b/springcloud_k8s_panzhihuazhihuishequ/service_equipment/src/main/java/com/panzhihua/service_equipment/api/UnionInspectionApi.java index 068f6d0..12eee1f 100644 --- a/springcloud_k8s_panzhihuazhihuishequ/service_equipment/src/main/java/com/panzhihua/service_equipment/api/UnionInspectionApi.java +++ b/springcloud_k8s_panzhihuazhihuishequ/service_equipment/src/main/java/com/panzhihua/service_equipment/api/UnionInspectionApi.java @@ -2,11 +2,11 @@ import com.baomidou.mybatisplus.core.metadata.IPage; import com.panzhihua.common.model.dtos.equipment.UnionInspectionDto; -import com.panzhihua.common.model.dtos.equipment.UnionReportDto; +import com.panzhihua.common.model.dtos.equipment.UnionUserDto; import com.panzhihua.common.model.vos.R; +import com.panzhihua.service_equipment.annotation.CurrentUser; import com.panzhihua.service_equipment.model.dos.UnionReport; import com.panzhihua.service_equipment.service.UnionInspectionService; -import com.panzhihua.service_equipment.service.UnionReportService; import lombok.extern.slf4j.Slf4j; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestBody; @@ -25,7 +25,6 @@ private UnionInspectionService unionInspectionService; - /** * 新增户外劳工站巡查 * @@ -33,8 +32,19 @@ * @return 新增结果 */ @PostMapping("/add") - public R add(@RequestBody UnionInspectionDto unionInspectionDto) { - return unionInspectionService.add(unionInspectionDto); + public R add(@RequestBody UnionInspectionDto unionInspectionDto, @CurrentUser UnionUserDto unionUserDto) { + if (unionInspectionDto.getIsApplets().equals(1)) { + log.info("小程序进入无需验证"); + return unionInspectionService.add(unionInspectionDto); + } else { + log.info("h5或者后台进入"); + if (unionUserDto != null) { + unionInspectionDto.setCreateUnionUserId(unionUserDto.getId()); + return unionInspectionService.add(unionInspectionDto); + } else { + return R.fail("请重新登录"); + } + } } @@ -45,7 +55,18 @@ * @return 动态结果 */ @PostMapping("/query") - public R<IPage<UnionReport>> query(@RequestBody UnionInspectionDto unionInspectionDto) { - return unionInspectionService.query(unionInspectionDto); + public R<IPage<UnionReport>> query(@RequestBody UnionInspectionDto unionInspectionDto, @CurrentUser UnionUserDto unionUserDto) { + if (unionInspectionDto.getIsApplets().equals(1)) { + log.info("小程序进入无需验证"); + return unionInspectionService.query(unionInspectionDto); + } else { + log.info("h5或者后台进入"); + if (unionUserDto != null) { + unionInspectionDto.setCreateUnionUserId(unionUserDto.getId()); + return unionInspectionService.query(unionInspectionDto); + } else { + return R.fail("请重新登录"); + } + } } } -- Gitblit v1.7.1