1
0

EmailParseService.java 29 KB

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