优化项目立项流程

amos-boot-system-ugp/amos-boot-module-ugp-biz/src/main/java/com/yeejoin/amos/boot/module/ugp/biz/service/impl/ProblemInitiationServiceImpl.java

@@ -57,6 +57,9 @@ public class ProblemInitiationServiceImpl {
     @Autowired
     QualityProblemServiceImpl qualityProblemService;
 
+    @Autowired
+    InstallNoticeMsgServiceImpl installNoticeMsgService;
+
 
     @Value("${params.work.flow.problemDefinitionKey}")
     private String problemDefinitionKey;
@@ -200,24 +203,46 @@ public class ProblemInitiationServiceImpl {
                 }
             }
             if (bool) {
-                LambdaQueryWrapper<OrgUsr> wrapperQueryWrapper = new LambdaQueryWrapper<>();
-                wrapperQueryWrapper.eq(OrgUsr::getAmosOrgId,agencyUserModel.getSequenceNbr());
 
                 SmsRecordModel smsRecordModel = sendSmsMassage(smsCode, agencyUserModel.getMobile(), smsParams);
-                RectifyMsg rectifyMsg = new RectifyMsg();
                 if (ValidationUtil.isEmpty(smsRecordModel)) {
                     continue;
                 }
-                rectifyMsg.setContent(smsRecordModel.getSmsContent());
-                rectifyMsg.setProblemId(sequenceNbr);
-                rectifyMsg.setNoticeUnitId(Long.valueOf(noticeUnitId));
-                rectifyMsg.setMsgReceiver(orgUsrServiceImpl.getOne(wrapperQueryWrapper).getSequenceNbr());
-                rectifyMsg.setSendTime(smsRecordModel.getSendTime());
-                rectifyMsgServiceImpl.save(rectifyMsg);
+                if(SMSEnum.项目安装告知申请.getCode().equals(smsCode) || SMSEnum.项目立项驳回短信.getCode().equals(smsCode)){
+                    saveProjectSmsLog(smsRecordModel,sequenceNbr,noticeUnitId,agencyUserModel);
+                }else {
+                    saveProblemSmsLog(smsRecordModel,sequenceNbr,noticeUnitId,agencyUserModel);
+                }
+
             }
         }
     }
 
+    public void saveProjectSmsLog(SmsRecordModel smsRecordModel,Long sequenceNbr,String noticeUnitId,AgencyUserModel agencyUserModel){
+        LambdaQueryWrapper<OrgUsr> wrapperQueryWrapper = new LambdaQueryWrapper<>();
+        wrapperQueryWrapper.eq(OrgUsr::getAmosOrgId,agencyUserModel.getSequenceNbr());
+        InstallNoticeMsg installNoticeMsg = new InstallNoticeMsg();
+        installNoticeMsg.setContent(smsRecordModel.getSmsContent());
+        installNoticeMsg.setInstallNoticeId(smsRecordModel.getSequenceNbr());
+        installNoticeMsg.setTargetUnitId(Long.valueOf(smsRecordModel.getAgencyCode()));
+        installNoticeMsg.setTargetPersonId(Long.valueOf(agencyUserModel.getUserId()));
+        installNoticeMsg.setSendTime(smsRecordModel.getSendTime());
+        installNoticeMsgService.save(installNoticeMsg);
+    }
+
+    public void saveProblemSmsLog(SmsRecordModel smsRecordModel,Long sequenceNbr,String noticeUnitId,AgencyUserModel agencyUserModel){
+        LambdaQueryWrapper<OrgUsr> wrapperQueryWrapper = new LambdaQueryWrapper<>();
+        wrapperQueryWrapper.eq(OrgUsr::getAmosOrgId,agencyUserModel.getSequenceNbr());
+        RectifyMsg rectifyMsg = new RectifyMsg();
+        rectifyMsg.setContent(smsRecordModel.getSmsContent());
+        rectifyMsg.setProblemId(sequenceNbr);
+        rectifyMsg.setNoticeUnitId(Long.valueOf(noticeUnitId));
+        rectifyMsg.setMsgReceiver(orgUsrServiceImpl.getOne(wrapperQueryWrapper).getSequenceNbr());
+        rectifyMsg.setSendTime(smsRecordModel.getSendTime());
+        rectifyMsgServiceImpl.save(rectifyMsg);
+    }
+
+
     public SmsRecordModel sendSmsMassage(String smsCode, String mobile, HashMap<String, String> smsParams){
         SmsRecordModel smsRecordModel = new SmsRecordModel();