xiaoguangbin/AnalysisSystemForRadionuclide
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

下载邮件增加锁避免异步线程导致邮件内容下载串文件

Browse Source 
关闭文件输出流
This commit is contained in:
qiaoqinzheng 2024-04-19 16:36:46 +08:00
parent a8c32e43bd
commit 99f74539d2
1 changed files with 65 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

117
jeecg-boot-base-core/src/main/java/org/jeecg/common/email/EmailServiceManager.java
View File

@ -66,6 +66,8 @@ public class EmailServiceManager {
private RedisUtil redisUtil; private RedisUtil redisUtil;
private Object downloadEmlLocal = new Object();
@NotNull @NotNull
public static EmailServiceManager getInstance(){ public static EmailServiceManager getInstance(){
return new EmailServiceManager(); return new EmailServiceManager();
@ -525,44 +527,47 @@ public class EmailServiceManager {
* 当计数大于10000后从0开始，服务重启后也从0开始 * 当计数大于10000后从0开始，服务重启后也从0开始
*/ */
public File downloadEmailToEmlDir(@NotNull Message message,Integer emailCounter,Integer batchesCounter) throws MessagingException { public File downloadEmailToEmlDir(@NotNull Message message,Integer emailCounter,Integer batchesCounter) throws MessagingException {
String subject = ""; synchronized (downloadEmlLocal) {
File emlFile = null; String subject = "";
String status = EmailLogManager.STATUS_SUCCESS; File emlFile = null;
Date receivedDate = null; String status = EmailLogManager.STATUS_SUCCESS;
try { Date receivedDate = null;
//获取发件人 FileOutputStream outputStream = null;
final String address = ((InternetAddress) message.getFrom()[0]).getAddress(); try {
final String from = address.substring(0,address.indexOf(StringConstant.AT)); //获取发件人
//获取主题 final String address = ((InternetAddress) message.getFrom()[0]).getAddress();
subject = MimeUtility.decodeText(message.getSubject()); final String from = address.substring(0,address.indexOf(StringConstant.AT));
if(subject.indexOf(StringConstant.SLASH) != -1){ //获取主题
subject = StringUtils.replace(subject,StringConstant.SLASH,""); subject = MimeUtility.decodeText(message.getSubject());
} if(subject.indexOf(StringConstant.SLASH) != -1){
if(subject.indexOf(StringConstant.COLON) != -1){ subject = StringUtils.replace(subject,StringConstant.SLASH,"");
subject = StringUtils.replace(subject,StringConstant.COLON,""); }
} if(subject.indexOf(StringConstant.COLON) != -1){
receivedDate = message.getReceivedDate(); subject = StringUtils.replace(subject,StringConstant.COLON,"");
StringBuilder fileName = new StringBuilder(); }
fileName.append(from); receivedDate = message.getReceivedDate();
fileName.append(StringConstant.UNDER_LINE); StringBuilder fileName = new StringBuilder();
fileName.append(subject); fileName.append(from);
fileName.append(StringConstant.UNDER_LINE); fileName.append(StringConstant.UNDER_LINE);
fileName.append(DateUtils.formatDate(new Date(),"YYMMdd")); fileName.append(subject);
fileName.append(StringConstant.UNDER_LINE); fileName.append(StringConstant.UNDER_LINE);
fileName.append(DateUtils.formatDate(new Date(),"HHmmssSSS")); fileName.append(DateUtils.formatDate(new Date(),"YYMMdd"));
fileName.append(StringConstant.UNDER_LINE); fileName.append(StringConstant.UNDER_LINE);
fileName.append("receive"); fileName.append(DateUtils.formatDate(new Date(),"HHmmssSSS"));
fileName.append(StringConstant.UNDER_LINE); fileName.append(StringConstant.UNDER_LINE);
fileName.append(DateUtils.formatDate(receivedDate,"YYMMdd")); fileName.append("receive");
fileName.append(StringConstant.UNDER_LINE); fileName.append(StringConstant.UNDER_LINE);
fileName.append(DateUtils.formatDate(receivedDate,"HHmmssSSS")); fileName.append(DateUtils.formatDate(receivedDate,"YYMMdd"));
fileName.append(StringConstant.UNDER_LINE); fileName.append(StringConstant.UNDER_LINE);
fileName.append(emailCounter); fileName.append(DateUtils.formatDate(receivedDate,"HHmmssSSS"));
fileName.append(SAVE_EML_SUFFIX); fileName.append(StringConstant.UNDER_LINE);
final String rootPath = spectrumPathProperties.getRootPath(); fileName.append(emailCounter);
final String emlPath = spectrumPathProperties.getEmlPath(); fileName.append(SAVE_EML_SUFFIX);
emlFile = new File(rootPath+emlPath+File.separator+fileName); final String rootPath = spectrumPathProperties.getRootPath();
message.writeTo(new FileOutputStream(emlFile)); final String emlPath = spectrumPathProperties.getEmlPath();
emlFile = new File(rootPath+emlPath+File.separator+fileName);
outputStream = new FileOutputStream(emlFile);
message.writeTo(outputStream);
// int bufferSize = 1024 * 1024; // 1M // int bufferSize = 1024 * 1024; // 1M
// InputStream inputStream = message.getInputStream(); // InputStream inputStream = message.getInputStream();
@ -580,20 +585,28 @@ public class EmailServiceManager {
// // 关闭流 // // 关闭流
// bufferedInputStream.close(); // bufferedInputStream.close();
// bufferedOutputStream.close(); // bufferedOutputStream.close();
} catch (MessagingException | IOException e) { } catch (MessagingException | IOException e) {
// 下载邮件失败 抛出自定义邮件下载异常 // 下载邮件失败 抛出自定义邮件下载异常
status = EmailLogManager.STATUS_ERROR; status = EmailLogManager.STATUS_ERROR;
String errorMsg = StrUtil.format("The email download failed, the subject of the email is {}, the reason is {}.", subject, e.getMessage()); String errorMsg = StrUtil.format("The email download failed, the subject of the email is {}, the reason is {}.", subject, e.getMessage());
log.error(errorMsg); log.error(errorMsg);
throw new DownloadEmailException(errorMsg); throw new DownloadEmailException(errorMsg);
}catch (Exception e) { }catch (Exception e) {
log.error("",e); log.error("",e);
}finally { }finally {
EmailLogEvent event = new EmailLogEvent(batchesCounter,Thread.currentThread().getId(),EmailLogManager.GS_TYPE_GET,status,EmailLogManager.GETIDEML,subject,DateUtils.formatDate(receivedDate,"yyyy-MM-dd HH:mm:ss:SSS"), EmailLogEvent event = new EmailLogEvent(batchesCounter,Thread.currentThread().getId(),EmailLogManager.GS_TYPE_GET,status,EmailLogManager.GETIDEML,subject,DateUtils.formatDate(receivedDate,"yyyy-MM-dd HH:mm:ss:SSS"),
(Objects.isNull(emlFile)?" ":emlFile.getAbsolutePath())); (Objects.isNull(emlFile)?" ":emlFile.getAbsolutePath()));
EmailLogManager.getInstance().offer(Thread.currentThread().getId(),event); EmailLogManager.getInstance().offer(Thread.currentThread().getId(),event);
try {
if (Objects.nonNull(outputStream)) {
outputStream.close();
}
} catch (IOException e) {
throw new RuntimeException(e);
}
}
return emlFile;
} }
return emlFile;
} }
/** /**
@ -656,7 +669,7 @@ public class EmailServiceManager {
String key = RedisConstant.EMAIL_MSG_ID+StringConstant.COLON+messageId; String key = RedisConstant.EMAIL_MSG_ID+StringConstant.COLON+messageId;
int numberKey = redisUtil.get(key) != null? (int) redisUtil.get(key):0; int numberKey = redisUtil.get(key) != null? (int) redisUtil.get(key):0;
// exist = redisUtil.hasKey(key); // exist = redisUtil.hasKey(key);
if(numberKey > taskProperties.getForceDeletedNumber()){ if(numberKey >= taskProperties.getForceDeletedNumber()){
exist = true; exist = true;
log.info("Check: Remove Email:{},receiveTime:{}",message.getSubject(), DateUtils.formatDate(message.getReceivedDate(),"yyyy-MM-dd HH:mm:ss")); log.info("Check: Remove Email:{},receiveTime:{}",message.getSubject(), DateUtils.formatDate(message.getReceivedDate(),"yyyy-MM-dd HH:mm:ss"));
message.setFlag(Flags.Flag.DELETED,true); message.setFlag(Flags.Flag.DELETED,true);