EmailParseService.java 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562
  1. package com.simuwang.daq.service;
  2. import cn.hutool.core.bean.BeanUtil;
  3. import cn.hutool.core.collection.CollUtil;
  4. import cn.hutool.core.date.DateUtil;
  5. import cn.hutool.core.exceptions.ExceptionUtil;
  6. import cn.hutool.core.map.MapUtil;
  7. import cn.hutool.core.util.StrUtil;
  8. import com.simuwang.base.common.conts.DateConst;
  9. import com.simuwang.base.common.conts.EmailParseStatusConst;
  10. import com.simuwang.base.common.conts.NavParseStatusConst;
  11. import com.simuwang.base.common.conts.EmailTypeConst;
  12. import com.simuwang.base.common.util.EmailUtil;
  13. import com.simuwang.base.common.util.FileUtil;
  14. import com.simuwang.base.config.EmailRuleConfig;
  15. import com.simuwang.base.mapper.*;
  16. import com.simuwang.base.pojo.dos.*;
  17. import com.simuwang.base.pojo.dto.EmailContentInfoDTO;
  18. import com.simuwang.base.pojo.dto.EmailFundNavDTO;
  19. import com.simuwang.base.pojo.dto.MailboxInfoDTO;
  20. import jakarta.mail.Folder;
  21. import jakarta.mail.Message;
  22. import jakarta.mail.MessagingException;
  23. import jakarta.mail.Store;
  24. import jakarta.mail.internet.MimeMultipart;
  25. import jakarta.mail.search.ComparisonTerm;
  26. import jakarta.mail.search.ReceivedDateTerm;
  27. import jakarta.mail.search.SearchTerm;
  28. import org.slf4j.Logger;
  29. import org.slf4j.LoggerFactory;
  30. import org.springframework.beans.factory.annotation.Value;
  31. import org.springframework.stereotype.Service;
  32. import java.io.File;
  33. import java.math.BigDecimal;
  34. import java.util.*;
  35. import java.util.stream.Collectors;
  36. /**
  37. * @author mozuwen
  38. * @date 2024-09-04
  39. * @description 邮件解析服务
  40. */
  41. @Service
  42. public class EmailParseService {
  43. private static final Logger log = LoggerFactory.getLogger(EmailParseService.class);
  44. @Value("${email.file.path}")
  45. private String path;
  46. private final EmailTypeRuleMapper emailTypeRuleMapper;
  47. private final EmailRuleConfig emailRuleConfig;
  48. private final EmailFieldMappingMapper emailFieldMapper;
  49. private final EmailParserFactory emailParserFactory;
  50. private final EmailParseInfoMapper emailParseInfoMapper;
  51. private final EmailFileInfoMapper emailFileInfoMapper;
  52. private final EmailFundNavMapper emailFundNavMapper;
  53. private final EmailFundAssetMapper emailFundAssetMapper;
  54. private final AssetMapper assetMapper;
  55. private final NavMapper navMapper;
  56. private final FundService fundService;
  57. private final FundAliasMapper fundAliasMapper;
  58. public EmailParseService(EmailTypeRuleMapper emailTypeRuleMapper, EmailRuleConfig emailRuleConfig,
  59. EmailFieldMappingMapper emailFieldMapper, EmailParserFactory emailParserFactory,
  60. EmailParseInfoMapper emailParseInfoMapper, EmailFileInfoMapper emailFileInfoMapper,
  61. EmailFundNavMapper emailFundNavMapper, EmailFundAssetMapper emailFundAssetMapper,
  62. AssetMapper assetMapper, NavMapper navMapper, FundService fundService,
  63. FundAliasMapper fundAliasMapper) {
  64. this.emailTypeRuleMapper = emailTypeRuleMapper;
  65. this.emailRuleConfig = emailRuleConfig;
  66. this.emailFieldMapper = emailFieldMapper;
  67. this.emailParserFactory = emailParserFactory;
  68. this.emailParseInfoMapper = emailParseInfoMapper;
  69. this.emailFileInfoMapper = emailFileInfoMapper;
  70. this.emailFundNavMapper = emailFundNavMapper;
  71. this.emailFundAssetMapper = emailFundAssetMapper;
  72. this.assetMapper = assetMapper;
  73. this.navMapper = navMapper;
  74. this.fundService = fundService;
  75. this.fundAliasMapper = fundAliasMapper;
  76. }
  77. /**
  78. * 解析指定邮箱指定时间范围内的邮件
  79. *
  80. * @param mailboxInfoDTO 邮箱配置信息
  81. * @param startDate 邮件起始日期(yyyy-MM-dd HH:mm:ss)
  82. * @param endDate 邮件截止日期(yyyy-MM-dd HH:mm:ss, 为null,将解析邮件日期小于等于startDate的当天邮件)
  83. */
  84. public void parseEmail(MailboxInfoDTO mailboxInfoDTO, Date startDate, Date endDate) {
  85. // 邮件类型配置
  86. Map<Integer, List<String>> emailTypeMap = getEmailType();
  87. // 邮件字段识别映射表
  88. Map<String, List<String>> emailFieldMap = getEmailFieldMapping();
  89. Map<String, List<EmailContentInfoDTO>> emailContentMap;
  90. try {
  91. emailContentMap = realEmail(mailboxInfoDTO, emailTypeMap, startDate, endDate);
  92. } catch (Exception e) {
  93. log.info("采集邮件失败 -> 邮箱配置信息:{},堆栈信息:{}", mailboxInfoDTO, ExceptionUtil.stacktraceToString(e));
  94. return;
  95. }
  96. for (Map.Entry<String, List<EmailContentInfoDTO>> emailEntry : emailContentMap.entrySet()) {
  97. List<EmailContentInfoDTO> emailContentInfoDTOList = emailEntry.getValue();
  98. if (CollUtil.isEmpty(emailContentInfoDTOList)) {
  99. log.warn("未采集到正文或附件");
  100. continue;
  101. }
  102. log.info("开始解析邮件数据 -> 邮件主题:{},邮件日期:{}", emailContentInfoDTOList.get(0).getEmailTitle(), emailContentInfoDTOList.get(0).getEmailDate());
  103. List<EmailFundNavDTO> emailFundNavDTOList = CollUtil.newArrayList();
  104. Map<EmailContentInfoDTO, List<EmailFundNavDTO>> fileNameNavMap = MapUtil.newHashMap();
  105. for (EmailContentInfoDTO emailContentInfoDTO : emailContentInfoDTOList) {
  106. try {
  107. List<EmailFundNavDTO> fundNavDTOList = parseEmail(emailContentInfoDTO, emailFieldMap);
  108. fileNameNavMap.put(emailContentInfoDTO, fundNavDTOList);
  109. emailFundNavDTOList.addAll(fundNavDTOList);
  110. } catch (Exception e) {
  111. log.error("堆栈信息:{}", ExceptionUtil.stacktraceToString(e));
  112. }
  113. }
  114. // 保存相关信息 -> 邮件信息表,邮件文件表,邮件净值表,邮件规模表,基金净值表
  115. saveRelatedTable(mailboxInfoDTO.getAccount(), emailContentInfoDTOList, fileNameNavMap);
  116. }
  117. }
  118. public List<EmailFundNavDTO> parseEmail(EmailContentInfoDTO emailContentInfoDTO, Map<String, List<String>> emailFieldMap) {
  119. Integer emailType = emailContentInfoDTO.getEmailType();
  120. AbstractEmailParser emailParser = emailParserFactory.getInstance(emailType);
  121. return emailParser.parse(emailContentInfoDTO, emailFieldMap);
  122. }
  123. public void saveRelatedTable(String emailAddress, List<EmailContentInfoDTO> emailContentInfoDTOList, Map<EmailContentInfoDTO, List<EmailFundNavDTO>> fileNameNavMap) {
  124. String emailTitle = CollUtil.isNotEmpty(emailContentInfoDTOList) ? emailContentInfoDTOList.get(0).getEmailTitle() : null;
  125. String emailDate = CollUtil.isNotEmpty(emailContentInfoDTOList) ? emailContentInfoDTOList.get(0).getEmailDate() : null;
  126. Integer emailType = CollUtil.isNotEmpty(emailContentInfoDTOList) ? emailContentInfoDTOList.get(0).getEmailType() : null;
  127. Integer emailId = CollUtil.isNotEmpty(emailContentInfoDTOList) ? emailContentInfoDTOList.get(0).getEmailId() : null;
  128. String senderEmail = CollUtil.isNotEmpty(emailContentInfoDTOList) ? emailContentInfoDTOList.get(0).getSenderEmail() : null;
  129. Date parseDate = new Date();
  130. int emailParseStatus = EmailParseStatusConst.SUCCESS;
  131. EmailParseInfoDO emailParseInfoDO = buildEmailParseInfo(emailId, emailAddress, senderEmail, emailDate, emailTitle, emailType, emailParseStatus, parseDate);
  132. emailId = saveEmailParseInfo(emailParseInfoDO);
  133. for (Map.Entry<EmailContentInfoDTO, List<EmailFundNavDTO>> fileNameNavEntry : fileNameNavMap.entrySet()) {
  134. // 保存邮件文件表
  135. EmailContentInfoDTO emailContentInfoDTO = fileNameNavEntry.getKey();
  136. Integer fileId = saveEmailFileInfo(emailId, emailContentInfoDTO.getFielId(), emailContentInfoDTO.getFileName(), emailContentInfoDTO.getFilePath(), parseDate);
  137. List<EmailFundNavDTO> fundNavDTOList = fileNameNavEntry.getValue();
  138. if (CollUtil.isEmpty(fundNavDTOList)) {
  139. continue;
  140. }
  141. for (EmailFundNavDTO fundNavDTO : fundNavDTOList) {
  142. // 设置净值数据的解析状态
  143. setNavParseStatus(fundNavDTO, emailTitle);
  144. }
  145. // 保存净值表和规模表
  146. saveNavAndAssetNet(fileId, fundNavDTOList, parseDate);
  147. }
  148. // 更新邮件解析结果 -> 当【净值日期】和【备案编码/基金名称】能正常解读,即识别为【成功】
  149. long successNavCount = fileNameNavMap.values().stream().flatMap(List::stream).filter(Objects::nonNull).count();
  150. emailParseStatus = successNavCount >= 1 ? EmailParseStatusConst.SUCCESS : EmailParseStatusConst.FAIL;
  151. emailParseInfoMapper.updateParseStatus(emailId, emailParseStatus);
  152. }
  153. private void saveNavAndAssetNet(Integer fileId, List<EmailFundNavDTO> fundNavDTOList, Date parseDate) {
  154. if (CollUtil.isEmpty(fundNavDTOList)) {
  155. return;
  156. }
  157. // 净值数据
  158. List<EmailFundNavDO> emailFundNavDOList = fundNavDTOList.stream()
  159. .map(e -> buildEmailFundNavDo(fileId, e, parseDate)).filter(CollUtil::isNotEmpty).flatMap(List::stream).collect(Collectors.toList());
  160. if (CollUtil.isNotEmpty(emailFundNavDOList)) {
  161. // 先删除文件id下的净值数据(考虑到重新解析的需求,如果是首次解析,那么fiel_id下不存在净值数据)
  162. emailFundNavMapper.deleteByFileId(fileId);
  163. emailFundNavMapper.batchInsert(emailFundNavDOList);
  164. List<NavDO> navDOList = emailFundNavDOList.stream().filter(e -> StrUtil.isNotBlank(e.getFundId()))
  165. .map(e -> BeanUtil.copyProperties(e, NavDO.class)).collect(Collectors.toList());
  166. saveNavDo(navDOList);
  167. }
  168. // 保存规模数据
  169. List<EmailFundAssetDO> emailFundAssetDOList = fundNavDTOList.stream()
  170. .map(e -> buildEmailFundAssetDo(fileId, e, parseDate)).filter(CollUtil::isNotEmpty).flatMap(List::stream).collect(Collectors.toList());
  171. if (CollUtil.isNotEmpty(emailFundAssetDOList)) {
  172. // 先删除file_id下的规模数据(考虑到重新解析的需求,如果是首次解析,那么fiel_id下不存在规模数据)
  173. emailFundAssetMapper.deleteByFileId(fileId);
  174. emailFundAssetMapper.batchInsert(emailFundAssetDOList);
  175. List<AssetDO> assetDOList = emailFundAssetDOList.stream().filter(e -> StrUtil.isNotBlank(e.getFundId()))
  176. .map(e -> BeanUtil.copyProperties(e, AssetDO.class)).collect(Collectors.toList());
  177. saveAssetDo(assetDOList);
  178. }
  179. }
  180. public void saveNavDo(List<NavDO> navDOList) {
  181. if (CollUtil.isEmpty(navDOList)) {
  182. return;
  183. }
  184. Map<String, List<NavDO>> fundIdNavMap = navDOList.stream().collect(Collectors.groupingBy(NavDO::getFundId));
  185. for (Map.Entry<String, List<NavDO>> entry : fundIdNavMap.entrySet()) {
  186. List<NavDO> navDOS = entry.getValue();
  187. List<String> priceDateList = navDOS.stream().map(NavDO::getPriceDate).map(e -> DateUtil.format(e, DateConst.YYYY_MM_DD)).collect(Collectors.toList());
  188. List<String> dateList = navMapper.queryFundNavByDate(entry.getKey(), priceDateList);
  189. List<NavDO> updateNavDoList = navDOS.stream().filter(e -> dateList.contains(DateUtil.format(e.getPriceDate(), DateConst.YYYY_MM_DD))).collect(Collectors.toList());
  190. List<NavDO> insertNavDoList = navDOS.stream().filter(e -> !dateList.contains(DateUtil.format(e.getPriceDate(), DateConst.YYYY_MM_DD))).collect(Collectors.toList());
  191. if (CollUtil.isNotEmpty(insertNavDoList)) {
  192. navMapper.batchInsert(insertNavDoList);
  193. }
  194. if (CollUtil.isNotEmpty(updateNavDoList)) {
  195. navMapper.batchUpdate(updateNavDoList);
  196. }
  197. }
  198. }
  199. public void saveAssetDo(List<AssetDO> assetDOList) {
  200. if (CollUtil.isEmpty(assetDOList)) {
  201. return;
  202. }
  203. Map<String, List<AssetDO>> fundIdNavMap = assetDOList.stream().collect(Collectors.groupingBy(AssetDO::getFundId));
  204. for (Map.Entry<String, List<AssetDO>> entry : fundIdNavMap.entrySet()) {
  205. List<AssetDO> assetDOS = entry.getValue();
  206. List<String> priceDateList = assetDOS.stream().map(AssetDO::getPriceDate).map(e -> DateUtil.format(e, DateConst.YYYY_MM_DD)).collect(Collectors.toList());
  207. List<String> dateList = assetMapper.queryFundNavByDate(entry.getKey(), priceDateList);
  208. List<AssetDO> updateAssetDoList = assetDOS.stream().filter(e -> dateList.contains(DateUtil.format(e.getPriceDate(), DateConst.YYYY_MM_DD))).collect(Collectors.toList());
  209. List<AssetDO> insertAssetDoList = assetDOS.stream().filter(e -> !dateList.contains(DateUtil.format(e.getPriceDate(), DateConst.YYYY_MM_DD))).collect(Collectors.toList());
  210. if (CollUtil.isNotEmpty(insertAssetDoList)) {
  211. assetMapper.batchInsert(insertAssetDoList);
  212. }
  213. if (CollUtil.isNotEmpty(updateAssetDoList)) {
  214. assetMapper.batchUpdate(updateAssetDoList);
  215. }
  216. }
  217. }
  218. private List<EmailFundAssetDO> buildEmailFundAssetDo(Integer fileId, EmailFundNavDTO fundNavDTO, Date parseDate) {
  219. List<EmailFundAssetDO> fundAssetDOList = CollUtil.newArrayList();
  220. BigDecimal assetNet = StrUtil.isNotBlank(fundNavDTO.getAssetNet()) ? new BigDecimal(fundNavDTO.getAssetNet()) : null;
  221. BigDecimal assetShare = StrUtil.isNotBlank(fundNavDTO.getAssetShare()) ? new BigDecimal(fundNavDTO.getAssetShare()) : null;
  222. if (assetNet == null) {
  223. return fundAssetDOList;
  224. }
  225. Integer isStored = fundNavDTO.getParseStatus() != null
  226. && (fundNavDTO.getParseStatus().equals(NavParseStatusConst.ASSET_NET_NEGATIVE) || fundNavDTO.getParseStatus().equals(NavParseStatusConst.SUCCESS)) ? 1 : 0;
  227. Date priceDate = DateUtil.parse(fundNavDTO.getPriceDate(), DateConst.YYYY_MM_DD);
  228. if (CollUtil.isNotEmpty(fundNavDTO.getFundIdList())) {
  229. for (String fundId : fundNavDTO.getFundIdList()) {
  230. EmailFundAssetDO emailFundAssetDO = new EmailFundAssetDO();
  231. emailFundAssetDO.setFileId(fileId);
  232. emailFundAssetDO.setPriceDate(priceDate);
  233. emailFundAssetDO.setFundId(fundId);
  234. emailFundAssetDO.setFundName(fundNavDTO.getFundName());
  235. emailFundAssetDO.setRegisterNumber(fundNavDTO.getRegisterNumber());
  236. emailFundAssetDO.setAssetNet(assetNet);
  237. emailFundAssetDO.setAssetShare(assetShare);
  238. emailFundAssetDO.setIsStored(isStored);
  239. emailFundAssetDO.setExceptionStatus(fundNavDTO.getParseStatus());
  240. emailFundAssetDO.setIsvalid(1);
  241. emailFundAssetDO.setCreatorId(0);
  242. emailFundAssetDO.setCreateTime(parseDate);
  243. emailFundAssetDO.setUpdaterId(0);
  244. emailFundAssetDO.setUpdateTime(parseDate);
  245. fundAssetDOList.add(emailFundAssetDO);
  246. }
  247. } else {
  248. EmailFundAssetDO emailFundAssetDO = new EmailFundAssetDO();
  249. emailFundAssetDO.setFileId(fileId);
  250. emailFundAssetDO.setPriceDate(priceDate);
  251. emailFundAssetDO.setFundName(fundNavDTO.getFundName());
  252. emailFundAssetDO.setRegisterNumber(fundNavDTO.getRegisterNumber());
  253. emailFundAssetDO.setAssetNet(assetNet);
  254. emailFundAssetDO.setAssetShare(assetShare);
  255. emailFundAssetDO.setIsStored(isStored);
  256. emailFundAssetDO.setExceptionStatus(fundNavDTO.getParseStatus());
  257. emailFundAssetDO.setIsvalid(1);
  258. emailFundAssetDO.setCreatorId(0);
  259. emailFundAssetDO.setCreateTime(parseDate);
  260. emailFundAssetDO.setUpdaterId(0);
  261. emailFundAssetDO.setUpdateTime(parseDate);
  262. fundAssetDOList.add(emailFundAssetDO);
  263. }
  264. return fundAssetDOList;
  265. }
  266. private List<EmailFundNavDO> buildEmailFundNavDo(Integer fileId, EmailFundNavDTO fundNavDTO, Date parseDate) {
  267. List<EmailFundNavDO> fundNavDOList = CollUtil.newArrayList();
  268. Date priceDate = DateUtil.parse(fundNavDTO.getPriceDate(), DateConst.YYYY_MM_DD);
  269. BigDecimal nav = StrUtil.isNotBlank(fundNavDTO.getNav()) ? new BigDecimal(fundNavDTO.getNav()) : null;
  270. BigDecimal cumulativeNavWithdrawal = StrUtil.isNotBlank(fundNavDTO.getCumulativeNavWithdrawal()) ? new BigDecimal(fundNavDTO.getCumulativeNavWithdrawal()) : null;
  271. if (nav == null && cumulativeNavWithdrawal == null) {
  272. return CollUtil.newArrayList();
  273. }
  274. Integer isStored = fundNavDTO.getParseStatus() != null && !fundNavDTO.getParseStatus().equals(NavParseStatusConst.NAV_DEFICIENCY)
  275. && !fundNavDTO.getParseStatus().equals(NavParseStatusConst.NOT_MATCH) ? 1 : 0;
  276. if (CollUtil.isNotEmpty(fundNavDTO.getFundIdList())) {
  277. for (String fundId : fundNavDTO.getFundIdList()) {
  278. EmailFundNavDO emailFundNavDO = new EmailFundNavDO();
  279. emailFundNavDO.setFileId(fileId);
  280. emailFundNavDO.setIsStored(isStored);
  281. emailFundNavDO.setPriceDate(priceDate);
  282. emailFundNavDO.setNav(nav);
  283. emailFundNavDO.setFundId(fundId);
  284. emailFundNavDO.setCumulativeNavWithdrawal(cumulativeNavWithdrawal);
  285. emailFundNavDO.setFundName(fundNavDTO.getFundName());
  286. emailFundNavDO.setRegisterNumber(fundNavDTO.getRegisterNumber());
  287. emailFundNavDO.setExceptionStatus(fundNavDTO.getParseStatus());
  288. emailFundNavDO.setIsvalid(1);
  289. emailFundNavDO.setCreatorId(0);
  290. emailFundNavDO.setCreateTime(parseDate);
  291. emailFundNavDO.setUpdaterId(0);
  292. emailFundNavDO.setUpdateTime(parseDate);
  293. fundNavDOList.add(emailFundNavDO);
  294. }
  295. } else {
  296. EmailFundNavDO emailFundNavDO = new EmailFundNavDO();
  297. emailFundNavDO.setFileId(fileId);
  298. emailFundNavDO.setPriceDate(priceDate);
  299. emailFundNavDO.setNav(nav);
  300. emailFundNavDO.setCumulativeNavWithdrawal(cumulativeNavWithdrawal);
  301. emailFundNavDO.setFundName(fundNavDTO.getFundName());
  302. emailFundNavDO.setRegisterNumber(fundNavDTO.getRegisterNumber());
  303. emailFundNavDO.setExceptionStatus(fundNavDTO.getParseStatus());
  304. emailFundNavDO.setIsStored(isStored);
  305. emailFundNavDO.setIsvalid(1);
  306. emailFundNavDO.setCreatorId(0);
  307. emailFundNavDO.setCreateTime(parseDate);
  308. emailFundNavDO.setUpdaterId(0);
  309. emailFundNavDO.setUpdateTime(parseDate);
  310. fundNavDOList.add(emailFundNavDO);
  311. }
  312. return fundNavDOList;
  313. }
  314. private Integer saveEmailFileInfo(Integer emailId, Integer fileId, String fileName, String filePath, Date parseDate) {
  315. EmailFileInfoDO emailFileInfoDO = buildEmailFileInfoDO(emailId, fileId, fileName, filePath, parseDate);
  316. if (emailFileInfoDO.getId() != null) {
  317. emailFileInfoMapper.updateTimeById(fileId, parseDate);
  318. return emailFileInfoDO.getId();
  319. }
  320. emailFileInfoMapper.insert(emailFileInfoDO);
  321. return emailFileInfoDO.getId();
  322. }
  323. private EmailFileInfoDO buildEmailFileInfoDO(Integer emailId, Integer fileId, String fileName, String filePath, Date parseDate) {
  324. EmailFileInfoDO emailFileInfoDO = new EmailFileInfoDO();
  325. emailFileInfoDO.setId(fileId);
  326. emailFileInfoDO.setEmailId(emailId);
  327. emailFileInfoDO.setFileName(fileName);
  328. emailFileInfoDO.setFilePath(filePath);
  329. emailFileInfoDO.setIsvalid(1);
  330. emailFileInfoDO.setCreatorId(0);
  331. emailFileInfoDO.setCreateTime(parseDate);
  332. emailFileInfoDO.setUpdaterId(0);
  333. emailFileInfoDO.setUpdateTime(parseDate);
  334. return emailFileInfoDO;
  335. }
  336. private void setNavParseStatus(EmailFundNavDTO fundNavDTO, String emailTitle) {
  337. // 1.单位净值或累计净值缺失
  338. if (StrUtil.isBlank(fundNavDTO.getNav()) || StrUtil.isBlank(fundNavDTO.getCumulativeNavWithdrawal())) {
  339. fundNavDTO.setParseStatus(NavParseStatusConst.NAV_DEFICIENCY);
  340. return;
  341. }
  342. // 2.匹配基金(考虑到解析估值表时已经匹配上基金的情况)
  343. if (CollUtil.isEmpty(fundNavDTO.getFundIdList())) {
  344. List<String> fundIdList = fundService.getFundIdByNamesAndCode(fundNavDTO.getFundName(), fundNavDTO.getRegisterNumber());
  345. if (CollUtil.isEmpty(fundIdList)) {
  346. // 判断是否写入别名管理表fund_alias
  347. saveFundAlias(fundNavDTO.getFundName(), fundNavDTO.getRegisterNumber());
  348. fundNavDTO.setParseStatus(NavParseStatusConst.NOT_MATCH);
  349. return;
  350. }
  351. fundNavDTO.setFundIdList(fundIdList);
  352. }
  353. // 考虑单独规模文件时 -> 无单位净值和累计净值
  354. // 3.单位净值或累计净值不大于0
  355. if (!emailTitle.contains("规模")) {
  356. if (StrUtil.isBlank(fundNavDTO.getNav()) || StrUtil.isBlank(fundNavDTO.getCumulativeNavWithdrawal())
  357. || (fundNavDTO.getNav().compareTo("0") <= 0 || fundNavDTO.getCumulativeNavWithdrawal().compareTo("0") <= 0)) {
  358. fundNavDTO.setParseStatus(NavParseStatusConst.NAV_NEGATIVE);
  359. return;
  360. }
  361. }
  362. // 4.资产净值不大于0
  363. if (StrUtil.isNotBlank(fundNavDTO.getAssetNet()) && fundNavDTO.getAssetNet().compareTo("0") <= 0) {
  364. fundNavDTO.setParseStatus(NavParseStatusConst.ASSET_NET_NEGATIVE);
  365. return;
  366. }
  367. fundNavDTO.setParseStatus(NavParseStatusConst.SUCCESS);
  368. }
  369. private void saveFundAlias(String fundName, String registerNumber) {
  370. // 未识别到基金名称和备案编码的数据不写入别名管理
  371. if (StrUtil.isBlank(fundName) && StrUtil.isBlank(registerNumber)) {
  372. return;
  373. }
  374. List<FundAliasDO> fundAliasDOList = CollUtil.newArrayList();
  375. if (StrUtil.isNotBlank(fundName) && StrUtil.isNotBlank(registerNumber)) {
  376. fundAliasDOList = fundAliasMapper.queryFundByNameAndRegisterNumber(fundName, registerNumber);
  377. }
  378. if (StrUtil.isBlank(fundName) && StrUtil.isNotBlank(registerNumber)) {
  379. fundAliasDOList = fundAliasMapper.queryFundByRegisterNumber(registerNumber);
  380. }
  381. if (StrUtil.isNotBlank(fundName) && StrUtil.isBlank(registerNumber)) {
  382. fundAliasDOList = fundAliasMapper.queryFundByName(fundName);
  383. }
  384. if (CollUtil.isNotEmpty(fundAliasDOList)) {
  385. return;
  386. }
  387. log.info("写入别名表 -> 基金名称:{},备案编码:{}", fundName, registerNumber);
  388. fundAliasMapper.insert(fundName, registerNumber);
  389. }
  390. private Integer saveEmailParseInfo(EmailParseInfoDO emailParseInfoDO) {
  391. if (emailParseInfoDO == null) {
  392. return null;
  393. }
  394. // 重新邮件功能 -> 修改解析时间和更新时间
  395. if (emailParseInfoDO.getId() != null) {
  396. emailParseInfoMapper.updateParseTime(emailParseInfoDO.getId(), emailParseInfoDO.getParseDate());
  397. return emailParseInfoDO.getId();
  398. }
  399. emailParseInfoMapper.insert(emailParseInfoDO);
  400. return emailParseInfoDO.getId();
  401. }
  402. private EmailParseInfoDO buildEmailParseInfo(Integer emailId, String emailAddress, String senderEmail, String emailDate,
  403. String emailTitle, Integer emailType, Integer parseStatus, Date parseDate) {
  404. EmailParseInfoDO emailParseInfoDO = new EmailParseInfoDO();
  405. emailParseInfoDO.setId(emailId);
  406. emailParseInfoDO.setSenderEmail(senderEmail);
  407. emailParseInfoDO.setEmail(emailAddress);
  408. emailParseInfoDO.setEmailDate(DateUtil.parse(emailDate, DateConst.YYYY_MM_DD_HH_MM_SS));
  409. emailParseInfoDO.setParseDate(parseDate);
  410. emailParseInfoDO.setEmailTitle(emailTitle);
  411. emailParseInfoDO.setEmailType(emailType);
  412. emailParseInfoDO.setParseStatus(parseStatus);
  413. emailParseInfoDO.setIsvalid(1);
  414. emailParseInfoDO.setCreatorId(0);
  415. emailParseInfoDO.setCreateTime(parseDate);
  416. emailParseInfoDO.setUpdaterId(0);
  417. emailParseInfoDO.setUpdateTime(parseDate);
  418. return emailParseInfoDO;
  419. }
  420. public Map<String, List<String>> getEmailFieldMapping() {
  421. List<EmailFieldMappingDO> emailFieldMappingDOList = emailFieldMapper.getEmailFieldMapping();
  422. return emailFieldMappingDOList.stream()
  423. .collect(Collectors.toMap(EmailFieldMappingDO::getCode, v -> Arrays.stream(v.getName().split(",")).toList()));
  424. }
  425. public Map<Integer, List<String>> getEmailType() {
  426. Map<Integer, List<String>> emailTypeMap = MapUtil.newHashMap(3, true);
  427. EmailTypeRuleDO emailTypeRuleDO = emailTypeRuleMapper.getEmailTypeRule();
  428. String nav = emailTypeRuleDO != null && StrUtil.isNotBlank(emailTypeRuleDO.getNav()) ? emailTypeRuleDO.getNav() : emailRuleConfig.getNav();
  429. String valuation = emailTypeRuleDO != null && StrUtil.isNotBlank(emailTypeRuleDO.getValuation()) ? emailTypeRuleDO.getValuation() : emailRuleConfig.getValuation();
  430. String report = emailTypeRuleDO != null && StrUtil.isNotBlank(emailTypeRuleDO.getReport()) ? emailTypeRuleDO.getReport() : emailRuleConfig.getReport();
  431. emailTypeMap.put(EmailTypeConst.VALUATION_EMAIL_TYPE, Arrays.stream(valuation.split(",")).toList());
  432. emailTypeMap.put(EmailTypeConst.REPORT_EMAIL_TYPE, Arrays.stream(report.split(",")).toList());
  433. emailTypeMap.put(EmailTypeConst.NAV_EMAIL_TYPE, Arrays.stream(nav.split(",")).toList());
  434. return emailTypeMap;
  435. }
  436. /**
  437. * 读取邮件
  438. *
  439. * @param mailboxInfoDTO 邮箱配置信息
  440. * @param emailTypeMap 邮件类型识别规则映射表
  441. * @param startDate 邮件起始日期
  442. * @param endDate 邮件截止日期(为null,将解析邮件日期小于等于startDate的当天邮件)
  443. * @return 读取到的邮件信息
  444. * @throws Exception 异常信息
  445. */
  446. private Map<String, List<EmailContentInfoDTO>> realEmail(MailboxInfoDTO mailboxInfoDTO, Map<Integer, List<String>> emailTypeMap, Date startDate, Date endDate) throws Exception {
  447. Store store = EmailUtil.getStoreNew(mailboxInfoDTO);
  448. if (store == null) {
  449. return MapUtil.newHashMap();
  450. }
  451. // 默认读取收件箱的邮件
  452. Folder folder = store.getFolder("INBOX");
  453. folder.open(Folder.READ_ONLY);
  454. Message[] messages = getEmailMessage(folder, mailboxInfoDTO.getProtocol(), startDate);
  455. if (messages == null || messages.length == 0) {
  456. log.info("获取不到邮件 -> 邮箱信息:{},开始时间:{},结束时间:{}", mailboxInfoDTO, startDate, endDate);
  457. return MapUtil.newHashMap();
  458. }
  459. Map<String, List<EmailContentInfoDTO>> emailMessageMap = MapUtil.newHashMap();
  460. for (Message message : messages) {
  461. List<EmailContentInfoDTO> emailContentInfoDTOList = CollUtil.newArrayList();
  462. String uuidKey = UUID.randomUUID().toString().replaceAll("-", "");
  463. Integer emailType;
  464. String senderEmail;
  465. try {
  466. Date emailDate = message.getSentDate();
  467. boolean isNotParseConditionSatisfied = emailDate == null || (endDate != null && emailDate.compareTo(endDate) > 0) || (startDate != null && emailDate.compareTo(startDate) < 0);
  468. if (isNotParseConditionSatisfied) {
  469. continue;
  470. }
  471. senderEmail = message.getFrom()[0].toString();
  472. emailType = EmailUtil.getEmailTypeBySubject(message.getSubject(), emailTypeMap);
  473. String emailDateStr = DateUtil.format(emailDate, DateConst.YYYY_MM_DD_HH_MM_SS);
  474. if (emailType == null) {
  475. log.info("邮件不满足解析条件 -> 邮件主题:{},邮件日期:{}", message.getSubject(), emailDateStr);
  476. continue;
  477. }
  478. log.info("邮件采集成功 -> 邮件主题:{},邮件日期:{}", message.getSubject(), emailDateStr);
  479. Object content = message.getContent();
  480. // 1.邮件为MIME多部分消息体:可能既有邮件又有正文
  481. if (content instanceof MimeMultipart) {
  482. emailContentInfoDTOList = EmailUtil.collectMimeMultipart(message, mailboxInfoDTO.getAccount(), path);
  483. }
  484. // 2.邮件只有正文
  485. if (content instanceof String) {
  486. EmailContentInfoDTO emailContentInfoDTO = new EmailContentInfoDTO();
  487. emailContentInfoDTO.setEmailContent(content.toString());
  488. emailContentInfoDTO.setEmailDate(emailDateStr);
  489. String fileName = message.getSubject() + DateUtil.format(emailDate, DateConst.YYYYMMDDHHMMSS24);
  490. String filePath = path + mailboxInfoDTO.getAccount() + "/" + DateUtil.format(emailDate, DateConst.YYYY_MM_DD) + "/" + fileName + ".html";
  491. File saveFile = new File(filePath);
  492. saveFile.setReadable(true);
  493. if (!saveFile.exists()) {
  494. if (!saveFile.getParentFile().exists()) {
  495. saveFile.getParentFile().mkdirs();
  496. saveFile.getParentFile().setExecutable(true);
  497. }
  498. }
  499. FileUtil.writeFile(filePath, content.toString());
  500. emailContentInfoDTO.setFilePath(filePath);
  501. emailContentInfoDTOList.add(emailContentInfoDTO);
  502. }
  503. if (CollUtil.isNotEmpty(emailContentInfoDTOList)) {
  504. emailContentInfoDTOList.forEach(e -> {
  505. e.setEmailType(emailType);
  506. e.setSenderEmail(senderEmail);
  507. });
  508. emailMessageMap.put(uuidKey, emailContentInfoDTOList);
  509. }
  510. } catch (Exception e) {
  511. log.error("获取邮箱的邮件报错,堆栈信息:{}", ExceptionUtil.stacktraceToString(e));
  512. }
  513. }
  514. folder.close(false);
  515. store.close();
  516. return emailMessageMap;
  517. }
  518. private Message[] getEmailMessage(Folder folder, String protocol, Date startDate) {
  519. Message[] messages;
  520. try {
  521. if (protocol.contains("imap")) {
  522. // 获取邮件日期大于等于startDate的邮件(搜索条件只支持按天)
  523. SearchTerm startDateTerm = new ReceivedDateTerm(ComparisonTerm.GE, startDate);
  524. messages = folder.search(startDateTerm);
  525. } else {
  526. messages = folder.getMessages();
  527. }
  528. } catch (MessagingException e) {
  529. throw new RuntimeException(e);
  530. }
  531. return messages;
  532. }
  533. }