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} 参数非法"); }