whyclxw
4 天以前 bfa320956f20988fe671b0c4d25aa82fe766a98d
src/main/java/com/whyc/service/FtpService.java
@@ -1,82 +1,86 @@
package com.whyc.service;
import com.enterprisedt.net.ftp.FTPException;
import com.whyc.constant.YamlProperties;
import com.whyc.dto.FileDirPath;
import com.whyc.dto.FtpHelper;
import com.whyc.dto.ZipUtils;
import com.whyc.util.ActionUtil;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.text.ParseException;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.concurrent.CountDownLatch;
import java.util.stream.Collectors;
@ConditionalOnProperty(prefix = "configFile",name = "type",havingValue = "2")
@Service
@EnableScheduling   // 2.开启定时任务
public class FtpService {
    //定时上传指定目录下文件
    @Scheduled(cron = "59 59 23 ? * FRI")
    //@Scheduled(cron = "0/10 * * * * ?")
    private void sendFtpFile() throws IOException, FTPException, InterruptedException, ParseException {
        //先连接ftp服务器,获取备份目录,保留3次.这里是从2023年4月开始进行保留
        FtpHelper ftpRemote = new FtpHelper(YamlProperties.ftpIp, YamlProperties.ftpPort, YamlProperties.ftpUserName, YamlProperties.ftpPassword);
        String[] dirList = ftpRemote.getDirList();
        List<String> dirList2 = Arrays.asList(dirList);
        dirList2 = dirList2.stream().filter(item->Integer.parseInt(item.substring(0,4))>=2023 && Integer.parseInt(item.substring(5,7))>=4).collect(Collectors.toList());
        for (int i = 0; i < dirList2.size()-3; i++) {
            ftpRemote.deleteDir(dirList2.get(i));
        }
        ftpRemote.disconnect();
        String fileDirName = FileDirPath.getFileDirName();
        String rootDoc=fileDirName+File.separator+"doc_file";
        String timeStr= ActionUtil.sdfwithFTP.format(new Date());
        try {
            File docFile = new File(rootDoc);
            File[] list = docFile.listFiles();
            CountDownLatch latch = new CountDownLatch(list.length+1);
            if(list!=null&&list.length>0){
                for (File file:list) {
                    new Thread(){
                        @Override
                        public void run() {
                            //压缩传输
                            File fileZip = new File(file.getAbsolutePath() + ".zip");
                            try {
                                ZipUtils.toZip(file.getAbsolutePath(),new FileOutputStream(fileZip),true);
                                String pathName=timeStr+"/doc_file/"+file.getName()+".zip";
                                FtpHelper ftp = new FtpHelper(YamlProperties.ftpIp, YamlProperties.ftpPort, YamlProperties.ftpUserName, YamlProperties.ftpPassword);
                                ftp.uploadFile(fileZip, pathName);
                                fileZip.delete();
                                ftp.disconnect();
                                latch.countDown();
                            } catch (Exception e) {
                                e.printStackTrace();
                                latch.countDown();
                            }
                        }
                    }.start();
                }
            }
            latch.countDown();
            latch.await();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
//package com.whyc.service;
//
//import com.enterprisedt.net.ftp.FTPException;
//import com.whyc.constant.YamlProperties;
//import com.whyc.dto.FileDirPath;
//import com.whyc.dto.FtpHelper;
//import com.whyc.dto.ZipUtils;
//import com.whyc.util.ActionUtil;
//import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
//import org.springframework.scheduling.annotation.EnableScheduling;
//import org.springframework.scheduling.annotation.Scheduled;
//import org.springframework.stereotype.Service;
//
//import java.io.File;
//import java.io.FileOutputStream;
//import java.io.IOException;
//import java.text.ParseException;
//import java.util.Arrays;
//import java.util.Date;
//import java.util.List;
//
//@ConditionalOnProperty(prefix = "configFile",name = "type",havingValue = "2")
//@Service
//@EnableScheduling   // 2.开启定时任务
//public class FtpService {
//    //定时上传指定目录下文件,每周五凌晨开启备份
//    //@Scheduled(cron = "59 59 23 ? * FRI")
//    @Scheduled(cron = "0 30 8 * * ?")
//    public void sendFtpFile() throws IOException, FTPException, ParseException {
//        //先连接ftp服务器
//        FtpHelper ftpRemote = new FtpHelper(YamlProperties.ftpIp, YamlProperties.ftpPort, YamlProperties.ftpUserName, YamlProperties.ftpPassword);
//        String fileDirName = FileDirPath.getFileDirName();
//        String rootDoc=fileDirName+File.separator+"doc_file";
//        String timeStr= ActionUtil.sdfwithFTP.format(new Date());
//        ftpRemote.mkdir(timeStr);
//        ftpRemote.disconnect();
//        //备份过程是否出现过问题,标识符
//        boolean isError=false;
//        try {
//
//            File docFile = new File(rootDoc);
//            File[] fileList = docFile.listFiles();
//            //File[] fileList2 = new File[fileList.length];
//            //BeanUtils.copyProperties(fileList2,fileList);
//            for (File file:fileList) {
//                FtpHelper ftpTemp = new FtpHelper(YamlProperties.ftpIp, YamlProperties.ftpPort, YamlProperties.ftpUserName, YamlProperties.ftpPassword);
//                //压缩传输
//                File fileZip = new File(file.getAbsolutePath() + ".zip");
//                ZipUtils.toZip(file.getAbsolutePath(),new FileOutputStream(fileZip),true);
//                //System.out.println(fileZip.toString()+"压缩完毕");
//                //System.out.println("进行FTP传输中...");
//                try {
//                    //ZipUtils.toZip(file.getAbsolutePath(),new FileOutputStream(fileZip),true);
//                    String pathName=timeStr+"/"+file.getName()+".zip";
//                    //System.out.println("传输目标路径确认为:"+pathName);
//                    ftpTemp.uploadFile2(fileZip, pathName);
//                    //System.out.println(fileZip+"传输完毕");
//                    fileZip.delete();
//                    //System.out.println(fileZip+"传输中转压缩包删除完毕");
//                } catch (Exception e) {
//                    isError = true;
//                    e.printStackTrace();
//                    fileZip.delete();
//                }finally {
//                    ftpTemp.disconnect();
//                }
//            }
//            //System.out.println("传输完毕");
//
//            //获取备份目录,保留4次.
//            //如果过程出现错误,则不删除
//            if (!isError) {
//                FtpHelper ftpTemp = new FtpHelper(YamlProperties.ftpIp, YamlProperties.ftpPort, YamlProperties.ftpUserName, YamlProperties.ftpPassword);
//                String[] dirList = ftpRemote.getDirList();
//                List<String> dirList2 = Arrays.asList(dirList);
//
//                for (int i = 0; i < dirList2.size()-4; i++) {
//                    ftpTemp.deleteDir(dirList2.get(i));
//                }
//                ftpTemp.disconnect();
//            }
//        } catch (Exception e) {
//            e.printStackTrace();
//        }
//    }
//}