From 5fb9d2df42a9d965e4dcd7fd14e0fdfe4feac64f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B8=87=E4=BD=B3?= Date: Mon, 13 Sep 2021 11:14:21 +0800 Subject: [PATCH] =?UTF-8?q?=E5=B9=B3=E5=8F=B0=E5=8F=82=E6=95=B0=E6=A0=A1?= =?UTF-8?q?=E9=AA=8C=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../common/constant/NotificationSystemConstant.java | 10 ++++++++++ .../reader/controller/NotificationController.java | 2 +- .../writer/controller/NotificationController.java | 2 +- 3 files changed, 12 insertions(+), 2 deletions(-) diff --git a/common/src/main/java/cn/org/gitlink/notification/common/constant/NotificationSystemConstant.java b/common/src/main/java/cn/org/gitlink/notification/common/constant/NotificationSystemConstant.java index ed2a44e..0a6c890 100644 --- a/common/src/main/java/cn/org/gitlink/notification/common/constant/NotificationSystemConstant.java +++ b/common/src/main/java/cn/org/gitlink/notification/common/constant/NotificationSystemConstant.java @@ -1,5 +1,8 @@ package cn.org.gitlink.notification.common.constant; +import java.util.HashMap; +import java.util.Map; + /** * @description: * @author: wanjia @@ -8,4 +11,11 @@ package cn.org.gitlink.notification.common.constant; public class NotificationSystemConstant { //平台编码 public static final String PLATFORM_CODE_GITLINK = "gitlink"; //gitlink平台 + public static final String PLATFORM_CODE_HEHUI = "hehui"; //hehui平台 + public static final Map PLATFORM_CODE_MAP = new HashMap() { + { + put("gitlink", PLATFORM_CODE_GITLINK); + put("hehui", PLATFORM_CODE_HEHUI); + } + }; } diff --git a/reader/src/main/java/cn/org/gitlink/notification/reader/controller/NotificationController.java b/reader/src/main/java/cn/org/gitlink/notification/reader/controller/NotificationController.java index e56839b..6abd003 100644 --- a/reader/src/main/java/cn/org/gitlink/notification/reader/controller/NotificationController.java +++ b/reader/src/main/java/cn/org/gitlink/notification/reader/controller/NotificationController.java @@ -181,7 +181,7 @@ public class NotificationController { private ResponseData validatePlatformCodeAndReceiver(String platform, Integer receiver) { //验证 {platform} 参数合法性，以判断请求来源 - if (!platform.trim().toLowerCase().equals(NotificationSystemConstant.PLATFORM_CODE_GITLINK)) { + if (!NotificationSystemConstant.PLATFORM_CODE_MAP.containsKey(platform)) { logger.debug("\t 输入参数 {platform} 的值 {" + platform + "} 无效"); return DataPacketUtil.jsonFailResult("{platform} 参数非法"); } diff --git a/writer/src/main/java/cn/org/gitlink/notification/writer/controller/NotificationController.java b/writer/src/main/java/cn/org/gitlink/notification/writer/controller/NotificationController.java index 2958914..911aa14 100644 --- a/writer/src/main/java/cn/org/gitlink/notification/writer/controller/NotificationController.java +++ b/writer/src/main/java/cn/org/gitlink/notification/writer/controller/NotificationController.java @@ -127,7 +127,7 @@ public class NotificationController { private ResponseData validatePlatformCode(String platform) { //验证 {platform} 参数合法性，以判断请求来源 - if (!platform.trim().toLowerCase().equals(NotificationSystemConstant.PLATFORM_CODE_GITLINK)) { + if (!NotificationSystemConstant.PLATFORM_CODE_MAP.containsKey(platform)) { logger.debug("\t 输入参数 {platform} 的值 {" + platform + "} 无效"); return DataPacketUtil.jsonFailResult("{platform} 参数非法"); }