EmailParseService.java 28 KB

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