1 /**
2 * 建立FTP链接，FTP服务器地址、端口、登陆用户信息都在配置里配置即可。
3 * @throws IOException
4 */
5 public boolean connectFtp(String ftpAddress, String ftpPort, String frpUserName, String frpPassword) throws IOException{
6 log.info("*****连接FTP服务器...*****");
7 try{
8 ftpClient.connect(ftpAddress, Integer.valueOf(ftpPort).intValue());
9 ftpClient.setControlEncoding("GB2312");
10 int reply = ftpClient.getReplyCode();
11 if(FTPReply.isPositiveCompletion(reply)){
12 if(ftpClient.login(frpUserName,frpPassword)){
13 log.info("*****连接FTP服务器成功！*****");
14 return true;
15 }
16 }else{
17 log.error("*****连接失败!响应代码为【"+ reply+"】*****");
18 }
19 disconnect();
20 }catch (Exception e) {
21 log.error("*****连接失败：" + e.getMessage());
22 }
23 return false;
24 }
25
26 /**
27 * 设置FTP客户端 被动模式、数据模式为二进制、字符编码GBK
28 */
29 public void setConnectType(){
30 try {
31 ftpClient.enterLocalPassiveMode();
32 ftpClient.setDefaultTimeout(1000 * 120);//120秒
33 ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
34 ftpClient.setControlEncoding("GB2312");
35 } catch (IOException e) {
36 e.printStackTrace();
37 }
38 }
39
40
41 /**
42 * 断开与远程服务器的连接
43 * @throws IOException
44 */
45 public void disconnect() {
46 if(ftpClient.isConnected()){
47 try {
48 ftpClient.disconnect();
49 } catch (IOException e) {
50 e.printStackTrace();
51 }
52 }
53 }
54
55
56 /**
57 * 过滤不符合的文件并批量下载
58 * @param remoteFileReg 文件前缀的正则表达式
59 * @param localPath 本地路径 .property 文件配置
60 * @param remote_down_path ftp文件路径
61 * @return List 下载到本地的文件路径 集合
62 * @throws IOException
63 */
64 @SuppressWarnings("unchecked")
65 public List downloads(String remoteFileReg,String localPath,String remote_down_path) throws IOException{
66 List<String> fileNames = new ArrayList<String>();
67 log.info("*****转移到服务器目录：" + remote_down_path);
68 setConnectType();
69 boolean changeFlag = ftpClient.changeWorkingDirectory(remote_down_path);
70 FTPFile[] files = ftpClient.listFiles();
71 //String[] names = ftpClient.listNames();
72 log.info("*****改变目录是否成功：" + changeFlag);
73 log.info("*****服务器上report目录下所有校验报告的文件数为：【" +files.length + "】" );
74 if(files.length == 0){
75 log.info("*****未在服务器上找到文件！*****");
76 return null;
77 }else{//目录下有文件
78 //把 bak文件的前缀找出来 ，区分读取和未读取的xls 和 xlsx ,只下载 未读取的文件
79 List<String> bakList = new ArrayList<String>();
80 List<String> list = new ArrayList<String>();
81
82 for (int i = 0; i < files.length; i++) {
83 FTPFile ftpFile = files[i];
84 String fileName = ftpFile.getName();
85
86 if(!fileName.endsWith(".bak") && ftpFile == null){
87 log.info("******* "+ fileName + "文件无数据!");
88 continue;
89 }
90 //匹配指定的文件前缀 和后缀 为 .bak 格式的文件
91 //bak 文件是文件读取完毕后生成的标记文件
92 Pattern bak = Pattern.compile("^"+remoteFileReg+"\\.bak");
93 Matcher m = bak.matcher(fileName);
94 if (m.find()) {
95 //取.bak文件的 前缀
96 //System.out.println(fileName);
97 //System.out.println(fileName.split("\\.")[0]);
98 bakList.add(fileName.split("\\.")[0]);
99 continue;
100 }
101
102 //匹配指定的文件前缀 和后缀 为 .xls .xlsx 格式的文件
103 //TODO 以后遇到其他的格式文件 需要把后缀抽出来作为参数传入
104 Pattern xls = Pattern.compile("^"+remoteFileReg+"\\.xls$"+"|"+"^"+remoteFileReg+"\\.xlsx$"+"|"+"^"+remoteFileReg+"\\.csv$");
105 Matcher mm = xls.matcher(fileName);
106 if(mm.find()){
107 list.add(fileName);
108 continue;
109 }
110 }
111
112 Iterator<String> it = list.iterator();
113 while (it.hasNext()) {
114 String xls = it.next();
115 for (int i = 0; i < bakList.size(); i++) {
116 String bak = bakList.get(i);
117 //bak文件存在 , 去掉此文件
118 if (xls.indexOf(bak) !=-1) {
119 it.remove();
120 bakList.remove(i);
121 }
122 }
123 }
124
125
126 for (String fFile : list) {
127 //下载未读取的文件
128 File downFile = new File(localPath + fFile);
129 //System.out.println(localPath);
130 File downPath = new File(localPath);
131 if(!downPath.exists()){
132 downPath.mkdirs();
133 }
134 String fileDir = remote_down_path + fFile;
135 OutputStream os = new FileOutputStream(downFile);
136 ftpClient.retrieveFile(new String(fileDir.getBytes("GB2312"),"ISO-8859-1"), os);
137 log.info("*****文件已下载到：" + downFile.getAbsolutePath() + "******");
138 fileNames.add(downFile.getAbsolutePath());
139 os.close();
140 }
141 log.info("**** 此次共下载了【"+list.size()+"】个文件! *****");
142 }
143 return fileNames;
144 }
145
146 /**
147 * 上传标志文件
148 * @param remoteFile
149 * @param localFile
150 * @return
151 */
152 public boolean upload(String localFileName,String remoteFileName){
153
154 boolean b = false;
155 try {
156 File file = new File(localFileName);
157 FileInputStream input = new FileInputStream(file);
158 b = ftpClient.changeWorkingDirectory(remoteFileName);
159 log.info("*****改变目录是否成功：" + b);
160 String remoteFile = remoteFileName + file.getName();
161 b = ftpClient.storeFile(new String(remoteFile.getBytes("GB2312"),"ISO-8859-1"), input);
162 if(b){
163 log.info(" ****** 标志文件"+localFileName+"上传成功!");
164 }
165 input.close();
166 } catch (FileNotFoundException e) {
167 e.printStackTrace();
168 } catch (IOException e) {
169 e.printStackTrace();
170 }
171 return b;
172 }
173
