From 7fdeb3f39d1476f26263501f981e3c76d63c15e9 Mon Sep 17 00:00:00 2001 From: pengren Date: Thu, 25 Apr 2024 15:20:24 +0800 Subject: [PATCH] =?UTF-8?q?=E7=BC=96=E5=8F=B7=EF=BC=9AZSSAC-585=20?= =?UTF-8?q?=E6=8F=8F=E8=BF=B0=EF=BC=9A=E6=96=B0=E5=A2=9Eappcode=E9=AA=8C?= =?UTF-8?q?=E8=AF=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../java/com/sf/common/utils/http/RequestUtils.java | 9 +++++++-- .../DeploymentApplyEnvironmentController.java | 6 ++++++ .../controller/DeploymentModuleListController.java | 12 +++++++++--- .../DeploymentServicePublishController.java | 7 +++++++ .../deployment/controller/SysApkInfoController.java | 7 +++++++ 5 files changed, 36 insertions(+), 5 deletions(-) diff --git a/sf-common/src/main/java/com/sf/common/utils/http/RequestUtils.java b/sf-common/src/main/java/com/sf/common/utils/http/RequestUtils.java index e9d20df..a5eb338 100644 --- a/sf-common/src/main/java/com/sf/common/utils/http/RequestUtils.java +++ b/sf-common/src/main/java/com/sf/common/utils/http/RequestUtils.java @@ -10,8 +10,13 @@ import javax.servlet.http.HttpServletRequest; public class RequestUtils { public static String getHeader(String code){ - RequestAttributes requestAttributes1 = RequestContextHolder.getRequestAttributes(); - HttpServletRequest request = ((ServletRequestAttributes) requestAttributes1).getRequest(); + HttpServletRequest request = null; + try { + RequestAttributes requestAttributes1 = RequestContextHolder.getRequestAttributes(); + request = ((ServletRequestAttributes) requestAttributes1).getRequest(); + } catch (Exception e) { + throw new SecurityException("获取header中获取appCode信息"); + } if(StringUtils.isBlank(request.getHeader(code))){ throw new SecurityException("参数异常appCode为空"); } diff --git a/sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentApplyEnvironmentController.java b/sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentApplyEnvironmentController.java index 53b31fa..5a0e6b6 100644 --- a/sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentApplyEnvironmentController.java +++ b/sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentApplyEnvironmentController.java @@ -52,6 +52,9 @@ public class DeploymentApplyEnvironmentController extends BaseController public TableDataInfo list(DeploymentApplyEnvironment deploymentApplyEnvironment) { deploymentApplyEnvironment.setAppCode(RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode())); + if(StringUtils.isEmpty(deploymentApplyEnvironment.getAppCode())){ + throw new SecurityException("参数异常header中的appCode为空"); + } startPage(); List list = deploymentApplyEnvironmentService.selectDeploymentApplyEnvironmentList(deploymentApplyEnvironment); return getDataTable(list); @@ -89,6 +92,9 @@ public class DeploymentApplyEnvironmentController extends BaseController public AjaxResult add(@RequestBody DeploymentApplyEnvironment deploymentApplyEnvironment) { deploymentApplyEnvironment.setAppCode( RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode())); + if(StringUtils.isEmpty(deploymentApplyEnvironment.getAppCode())){ + throw new SecurityException("参数异常header中的appCode为空"); + } return toAjax(deploymentApplyEnvironmentService.insertDeploymentApplyEnvironment(deploymentApplyEnvironment)); } diff --git a/sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentModuleListController.java b/sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentModuleListController.java index 7b9957b..c012605 100644 --- a/sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentModuleListController.java +++ b/sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentModuleListController.java @@ -47,6 +47,9 @@ public class DeploymentModuleListController extends BaseController public TableDataInfo list(DeploymentModuleList deploymentModuleList) { deploymentModuleList.setAppCode(RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode())); + if(StringUtils.isEmpty(deploymentModuleList.getAppCode())){ + throw new SecurityException("参数异常header中的appCode为空"); + } startPage(); List list = deploymentModuleListService.selectDeploymentModuleListList(deploymentModuleList); return getDataTable(list); @@ -57,11 +60,11 @@ public class DeploymentModuleListController extends BaseController @GetMapping("/qurey/list") public AjaxResult queryByCore(ModuleListByCoreRequest request) { - if(StringUtils.isEmpty(request.getAppCore())){ - throw new SecurityException("参数异常appCode为空"); - } DeploymentModuleList deploymentModuleList = new DeploymentModuleList(); deploymentModuleList.setAppCode(RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode())); + if(StringUtils.isEmpty(deploymentModuleList.getAppCode())){ + throw new SecurityException("参数异常header中的appCode为空"); + } deploymentModuleList.setSysType(request.getSysType()); List list = deploymentModuleListService.selectDeploymentModuleListList(deploymentModuleList); return success(list); @@ -99,6 +102,9 @@ public class DeploymentModuleListController extends BaseController public AjaxResult add(@RequestBody DeploymentModuleList deploymentModuleList) { deploymentModuleList.setAppCode(RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode())); + if(StringUtils.isEmpty(deploymentModuleList.getAppCode())){ + throw new SecurityException("参数异常header中的appCode为空"); + } return toAjax(deploymentModuleListService.insertDeploymentModuleList(deploymentModuleList)); } diff --git a/sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentServicePublishController.java b/sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentServicePublishController.java index 6c04f89..bb482bf 100644 --- a/sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentServicePublishController.java +++ b/sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentServicePublishController.java @@ -5,6 +5,7 @@ import javax.servlet.http.HttpServletResponse; import com.sf.common.enums.RequestHeaderEnums; import com.sf.common.utils.DateUtils; +import com.sf.common.utils.StringUtils; import com.sf.common.utils.http.RequestUtils; import com.sf.system.deployment.domain.DeploymentServicePublish; import com.sf.system.deployment.service.IDeploymentServicePublishService; @@ -47,6 +48,9 @@ public class DeploymentServicePublishController extends BaseController public TableDataInfo list(DeploymentServicePublish deploymentServicePublish) { deploymentServicePublish.setAppCode(RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode())); + if(StringUtils.isEmpty(deploymentServicePublish.getAppCode())){ + throw new SecurityException("参数异常header中的appCode为空"); + } startPage(); List list = deploymentServicePublishService.selectDeploymentServicePublishList(deploymentServicePublish); return getDataTable(list); @@ -84,6 +88,9 @@ public class DeploymentServicePublishController extends BaseController public AjaxResult add(@RequestBody DeploymentServicePublish deploymentServicePublish) { deploymentServicePublish.setAppCode(RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode())); + if(StringUtils.isEmpty(deploymentServicePublish.getAppCode())){ + throw new SecurityException("参数异常header中的appCode为空"); + } return toAjax(deploymentServicePublishService.insertDeploymentServicePublish(deploymentServicePublish)); } diff --git a/sf-system/src/main/java/com/sf/system/deployment/controller/SysApkInfoController.java b/sf-system/src/main/java/com/sf/system/deployment/controller/SysApkInfoController.java index dfc5970..65808c6 100644 --- a/sf-system/src/main/java/com/sf/system/deployment/controller/SysApkInfoController.java +++ b/sf-system/src/main/java/com/sf/system/deployment/controller/SysApkInfoController.java @@ -6,6 +6,7 @@ import com.sf.common.core.domain.AjaxResult; import com.sf.common.core.page.TableDataInfo; import com.sf.common.enums.BusinessType; import com.sf.common.enums.RequestHeaderEnums; +import com.sf.common.utils.StringUtils; import com.sf.common.utils.http.RequestUtils; import com.sf.common.utils.poi.ExcelUtil; import com.sf.system.deployment.domain.SysApkInfo; @@ -38,6 +39,9 @@ public class SysApkInfoController extends BaseController public TableDataInfo list(SysApkInfo sysApkInfo) { sysApkInfo.setAppCode(RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode())); + if(StringUtils.isEmpty(sysApkInfo.getAppCode())){ + throw new SecurityException("参数异常header中的appCode为空"); + } startPage(); List list = sysApkInfoService.selectSysApkInfoList(sysApkInfo); return getDataTable(list); @@ -75,6 +79,9 @@ public class SysApkInfoController extends BaseController public AjaxResult add(@RequestBody SysApkInfo sysApkInfo) { sysApkInfo.setAppCode(RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode())); + if(StringUtils.isEmpty(sysApkInfo.getAppCode())){ + throw new SecurityException("参数异常header中的appCode为空"); + } sysApkInfo.setCreated(getUsername()); sysApkInfo.setModified(getUsername()); return toAjax(sysApkInfoService.insertSysApkInfo(sysApkInfo));