diff --git a/ssjygl-xsx-service/src/main/java/com/cowr/service/ssjygl/sms/noticecontact/SmsNoticeContactService.java b/ssjygl-xsx-service/src/main/java/com/cowr/service/ssjygl/sms/noticecontact/SmsNoticeContactService.java index 91e2945..d643bff 100644 --- a/ssjygl-xsx-service/src/main/java/com/cowr/service/ssjygl/sms/noticecontact/SmsNoticeContactService.java +++ b/ssjygl-xsx-service/src/main/java/com/cowr/service/ssjygl/sms/noticecontact/SmsNoticeContactService.java @@ -26,6 +26,11 @@ public class SmsNoticeContactService extends BaseService { private static Log log = Log.getLog(SmsNoticeContactService.class); public static final SmsNoticeContactService me = new SmsNoticeContactService(); + private static List chktype = new ArrayList() {{ + add(1); // 每日销售汇总 + add(2); // 累计销售汇总 + }}; + public Result save(SmsNoticeContact model, Sysuser sysuser){ try { model.setId(StrKit.getRandomUUID()); @@ -33,7 +38,16 @@ public class SmsNoticeContactService extends BaseService { List list = JSONArray.parseArray(type, Integer.class); for(Integer i : list){ - if(i != 1){ + boolean check = false; + + for (Integer integer : chktype) { + if (integer.equals(i)) { + check = true; + break; + } + } + + if(!check){ return Result.failed("短信类型错误"); } } @@ -67,7 +81,16 @@ public class SmsNoticeContactService extends BaseService { List list = JSONArray.parseArray(type, Integer.class); for(Integer i : list){ - if(i != 1){ + boolean check = false; + + for (Integer integer : chktype) { + if (integer.equals(i)) { + check = true; + break; + } + } + + if(!check){ return Result.failed("短信类型错误"); } }