FTPUtils.java 上传下载
1 package com.demo.cronmodule.utils; 2 3 import com.alibaba.fastjson.JSON; 4 import com.demo.basebusiness.http.vo.ResultVo; 5 import com.demo.qs.node.protocol.FtpProtocolUtil; 6 import com.demo.qs.node.protocol.StatusResult; 7 import org.apache.commons.logging.Log; 8 import org.apache.commons.logging.LogFactory; 9 import org.apache.commons.net.ftp.FTP; 10 import org.apache.commons.net.ftp.FTPClient; 11 import org.apache.commons.net.ftp.FTPFile; 12 import org.apache.commons.net.ftp.FTPReply; 13 import org.springframework.stereotype.Service; 14 15 import javax.servlet.http.HttpServletResponse; 16 import javax.servlet.http.HttpServletResponseWrapper; 17 import java.io.*; 18 import java.net.URLEncoder; 19 import java.rmi.Remote; 20 import java.util.ArrayList; 21 import java.util.List; 22 import java.util.zip.ZipEntry; 23 import java.util.zip.ZipOutputStream; 24 25 /** 26 * @ClassName FtpApche 27 * @Description 28 * @Author username 29 * @Date 2020/4/30 14:20 30 */ 31 public class FtpApcheUtil { 32 33 private static Log logger = LogFactory.getLog(FtpApcheUtil.class); 34 private static String encoding = System.getProperty("file.encoding"); 35 36 public String ip="192.168.50.88"; 37 public int port=2121; 38 public String username="admin"; 39 public String password="admin"; 40 public String remotePath="/"; 41 42 private static FTPClient ftpClient = new FTPClient(); 43 44 public static FtpApcheUtil getInstance(String ip, int port, String username, String password){ 45 return new FtpApcheUtil(ip, port, username, password); 46 } 47 private FtpApcheUtil (String ip, int port, String username, String password){ 48 /* @param ip FTP服务器hostname 49 * @param port FTP服务器端口 50 * @param username FTP登录账号 51 * @param password FTP登录密码 52 */ 53 init(ip, port, username, password); 54 } 55 56 public void init(String ip, int port, String username, String password) { 57 this.ip = ip; 58 this.port = port; 59 this.username = username; 60 this.password = password; 61 } 62 public boolean connect(){ 63 try { 64 65 int reply; 66 ftpClient.connect(this.ip, this.port);// 连接FTP服务器 67 //ftpClient.connect(url); //如果采用默认端口，可以使用ftp.connect(url)的方式直接连接FTP服务器 68 // 登录 69 ftpClient.login(this.username, this.password); 70 ftpClient.setControlEncoding(encoding); 71 // 检验是否连接成功 72 reply = ftpClient.getReplyCode(); 73 if (!FTPReply.isPositiveCompletion(reply)) { 74 logger.info("FTP server refused connection"); 75 ftpClient.disconnect(); 76 return false; 77 } 78 return true; 79 } catch (IOException e) { 80 logger.error("ftpClient.connect error",e); 81 return false; 82 } 83 }; 84 /** 85 * Description: 向FTP服务器上传文件 86 * 87 * @param path FTP服务器保存目录,如果是根目录则为“/” 88 * @param filename 上传到FTP服务器上的文件名 89 * @param input 本地文件输入流 90 * @return 成功返回true，否则返回false 91 * @Version1.0 92 */ 93 public boolean uploadFile( String path, String filename, InputStream input) { 94 boolean result = false; 95 try { 96 97 boolean connectSucceess=connect(); 98 if(!connectSucceess){ 99 return result; 100 } 101 // 转移工作目录至指定目录下 102 boolean change = ftpClient.changeWorkingDirectory(path); 103 ftpClient.setFileType(FTP.BINARY_FILE_TYPE); 104 if (change) { 105 result = ftpClient.storeFile(new String(filename.getBytes(encoding), "iso-8859-1"), input); 106 if (result) { 107 logger.info("上传成功"); 108 } 109 } 110 input.close(); 111 ftpClient.logout(); 112 } catch (IOException e) { 113 logger.error("FtpApcheUtil.uploadFile catch IOException::",e); 114 } finally { 115 if (ftpClient.isConnected()) { 116 try { 117 ftpClient.disconnect(); 118 } catch (IOException e) { 119 logger.error("FtpApcheUtil.uploadFile finally IOException:",e); 120 } 121 } 122 } 123 return result; 124 } 125 126 /** 127 * 将本地文件上传到FTP服务器上 128 */ 129 public void doUpLoadFromDisk(String sourcePath,String sourceFileName,String targetPath,String targetFileName) { 130 try { 131 String sourceFile=sourcePath+sourceFileName; 132 FileInputStream in = new FileInputStream(new File(sourceFile)); 133 boolean flag = uploadFile(targetPath, targetFileName, in); 134 if(!flag){ 135 logger.info("upload file failed"); 136 } 137 } catch (Exception e) { 138 logger.error("doUpLoadFromDisk Exception:",e); 139 } 140 } 141 142 143 /** 144 * Description: 从FTP服务器下载文件 145 * 146 * @param remotePath FTP服务器上的相对路径 147 * @param fileName 要下载的文件名 148 * @param localPath 下载后保存到本地的路径 149 * @return 150 * @Version1.0 151 */ 152 public boolean downFile(String remotePath, String fileName, 153 String localPath) { 154 boolean result = false; 155 try { 156 /* 157 * 为了上传和下载中文文件，有些地方建议使用以下两句代替 158 * new String(remotePath.getBytes(encoding),"iso-8859-1")转码。 159 * 经过测试，通不过。 160 */ 161 // FTPClientConfig conf = new FTPClientConfig(FTPClientConfig.SYST_NT); 162 // conf.setServerLanguageCode("zh"); 163 boolean connectSucceess=connect(); 164 if(!connectSucceess){ 165 return result; 166 } 167 // 转移到FTP服务器目录至指定的目录下 168 ftpClient.changeWorkingDirectory(new String(remotePath.getBytes(encoding), "iso-8859-1")); 169 // 设置文件传输类型为二进制 170 ftpClient.setFileType(FTPClient.BINARY_FILE_TYPE); 171 // 获取文件列表 172 FTPFile[] fs = ftpClient.listFiles(); 173 for (FTPFile ff : fs) { 174 if (ff.getName().equals(fileName)) { 175 File localFile = new File(localPath + "/" + ff.getName()); 176 OutputStream is = new FileOutputStream(localFile); 177 ftpClient.retrieveFile(ff.getName(), is); 178 is.close(); 179 } 180 } 181 ftpClient.logout(); 182 result = true; 183 } catch (IOException e) { 184 logger.error("downFile file failed catch:",e); 185 } finally { 186 if (ftpClient.isConnected()) { 187 try { 188 ftpClient.disconnect(); 189 } catch (IOException ioe) { 190 logger.error("downFile file failed finally:",ioe); 191 } 192 } 193 } 194 return result; 195 } 196 197 /** 198 * 将FTP服务器上文件下载到本地 199 */ 200 public void doDownFile(String localPath,String remotePath ,String fileName) { 201 try { 202 String rename ="test_ftp2.au3"; 203 localPath="C:\\Users\\username"; //下载至本地文件夹 204 init(ip, port, username, password); 205 boolean flag = downFile(remotePath, fileName, localPath); 206 System.out.println(flag); 207 } catch (Exception e) { 208 logger.error("testDownFile:",e); 209 } 210 } 211 /** 212 * 实现文件打包压缩下载 213 * 214 * @param zipname 压缩包的文件名 215 * @param ftpRemotePath ftp文件路径 216 * @param ftpFileList 要下载的文件名集合 217 * @param namelist 压缩后的文件名集合 218 */ 219 public ResultVo<Object> ftpDownloadZipFile(ResultVo<Object> resultVo,String zipname, String ftpRemotePath, String localPath, List<String> ftpFileList, List<String> namelist) { 220 boolean connectSucceess = connect(); 221 if (!connectSucceess) { 222 resultVo.setCode(ResultVo.CODE_FAIL); 223 resultVo.setMsg("连接FTP服务器失败"); 224 return resultVo; 225 } 226 FTPClient ftp = this.ftpClient; 227 try { 228 ftp.enterLocalPassiveMode(); 229 // 转移到FTP服务器目录至指定的目录下 230 ftpClient.changeWorkingDirectory(new String(ftpRemotePath.getBytes(encoding), "iso-8859-1")); 231 // 设置文件传输类型为二进制 232 ftpClient.setFileType(FTPClient.BINARY_FILE_TYPE); 233 234 byte[] buf = new byte[1024]; 235 Set<String> failFileSet = new HashSet<>(); 236 Set<String> successFileSet = new HashSet<>(); 237 238 File zipFile =null; 239 ZipOutputStream zipOut = null; 240 for (int i = 0; i < ftpFileList.size(); i++) { 241 242 InputStream bis = ftp.retrieveFileStream(ftpFileList.get(i)); 243 if (bis == null) { 244 failFileSet.add(ftpFileList.get(i)); 245 } else { 246 if(zipFile==null) { 247 zipFile = new File(localPath + "/" + zipname); 248 zipOut = new ZipOutputStream(new FileOutputStream(zipFile)); 249 } 250 ZipEntry entry = new ZipEntry(namelist.get(i));//重命名 251 zipOut.putNextEntry(entry); 252 int readLen = -1; 253 while ((readLen = bis.read(buf, 0, 1024)) != -1) { 254 zipOut.write(buf, 0, readLen); 255 } 256 zipOut.closeEntry(); 257 bis.close(); 258 ftp.completePendingCommand(); 259 //调用ftp.retrieveFileStream这个接口后，一定要手动close掉返回的InputStream，然后再调用completePendingCommand方法 260 // ，若不是按照这个顺序，则会导致后面对FTPClient的操作都失败 261 } 262 } 263 if(successFileSet.size()==0){ 264 resultVo.setCode(ResultVo.CODE_FAIL); 265 resultVo.setMsg("下载失败，文件不存在："+ String.join(",", failFileSet)); 266 } 267 else if (failFileSet.size() > 0) { 268 resultVo.setCode(ResultVo.CODE_SUCCESS); 269 resultVo.setMsg("部分下载失败，以下文件不存在："+ String.join(",", failFileSet)); 270 } 271 if(zipOut!=null) 272 zipOut.close(); 273 ftp.logout(); 274 resultVo.setCode(ResultVo.CODE_SUCCESS); 275 resultVo.setMsg(ResultVo.MSG_SUCCESS); 276 //删除压缩包 zipFile.delete(); 277 } catch (Exception e) { 278 resultVo.setCode(ResultVo.CODE_FAIL); 279 resultVo.setMsg("FTP下载失败"); 280 logger.error("FTP下载失败: ", e); 281 } 282 return resultVo; 283 } 284 285 //普通springboot下载 286 public ResultVo<Object> downloadFile(ResultVo<Object> resultVo, HttpServletResponse response, String zipname, String serverPath) { 287 FileInputStream fis = null; 288 BufferedInputStream bis = null; 289 try { 290 //下载 291 byte[] buffer = new byte[1024]; 292 293 // 清除buffer缓存 294 response.reset(); 295 response.setHeader("content-type", "application/octet-stream"); 296 response.setContentType("application/octet-stream"); 297 // response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(zipname, "UTF-8")); 298 299 response.setHeader("Content-Disposition", "attachment; filename=" 300 + URLEncoder.encode(zipname, encoding));// + ".zip" 301 response.setHeader("Pragma", "no-cache"); 302 response.setHeader("Cache-Control", "no-cache"); 303 response.setDateHeader("Expires", 0); 304 File file = new File(serverPath + zipname); 305 fis = new FileInputStream(file); 306 bis = new BufferedInputStream(fis); 307 OutputStream os = response.getOutputStream(); 308 int i = 0; 309 while ((i = bis.read(buffer)) != -1) { 310 os.write(buffer, 0, i); 311 os.flush(); 312 } 313 resultVo.setCode(ResultVo.CODE_SUCCESS); 314 resultVo.setMsg(ResultVo.MSG_SUCCESS); 315 //删除压缩包 316 //zipFile.delete(); 317 } catch (Exception e) { 318 resultVo.setCode(ResultVo.CODE_FAIL); 319 resultVo.setMsg("服务器文件下载失败"); 320 logger.error("文件打包下载有误: ", e); 321 } finally { 322 if (bis != null) { 323 try { 324 bis.close(); 325 } catch (IOException e) { 326 } 327 } 328 if (fis != null) { 329 try { 330 fis.close(); 331 } catch (IOException e) { 332 } 333 } 334 335 return resultVo; 336 } 337 } 338 339 public static void main(String[] args) { 340 String ip="192.168.50.88"; 341 int port=2121; 342 String username="admin"; 343 String password="admin"; 344 String remotePath="/"; 345 List<String> ftpFileList=new ArrayList(){{ 346 add("test_ftp1.au3"); 347 add("test.au3"); 348 }}; 349 String zipName="test1.zip"; 350 FtpApcheUtil fa = FtpApcheUtil.getInstance(ip, port, username, password); 351 String localPath="C:\\Users\\username"; 352 fa.zipDownloadFile(zipName,remotePath,localPath,ftpFileList,ftpFileList); 353 /* FtpProtocolUtil p = FtpProtocolUtil.getInstance(); 354 FtpApcheUtil ftpApcheUtil = FtpApcheUtil.getInstance(); 355 ftpApcheUtil.testUpLoadFromDisk();*/ 356 /* boolean connect = p.connect(ip, port, username, password); 357 List<StatusResult> results= p.upload("C:\\Users\\username\\", "test_ftp1.au3", remotePath);*/ 358 /* String localName ="test_ftp2.au3"; 359 String rename ="test_ftp1.au3"; 360 String localPath="C:\\Users\\username"; 361 StatusResult results = p.download(localPath,localName,remotePath,rename,true); 362 logger.info(JSON.toJSONString(results));*/ 363 } 364 365 366 }