[DEV] ftp download transfer stream us byte array

atms-api/src/main/java/pwc/taxtech/atms/common/ftp/FTPClientPool.java

@@ -6,11 +6,16 @@ import org.apache.commons.pool2.impl.GenericObjectPool;
 import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Component;
+import pwc.taxtech.atms.exception.ServiceException;
 
 
 import javax.annotation.PostConstruct;
+import java.io.BufferedOutputStream;
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
+import java.io.OutputStream;
 
 @Component
 public class FTPClientPool {
@@ -39,11 +44,16 @@ public class FTPClientPool {
         clientConfig.setPassword(ftpPwd);
         ftpClientConfig = clientConfig;
         pool = new GenericObjectPool<>(new FtpClientFactory(clientConfig), config);
+
+        Runtime.getRuntime().addShutdownHook(new Thread(() -> {
+            pool.close();
+        }));
         ftpRootPath = pool.borrowObject().printWorkingDirectory();
     }
 
+    // this method should only be used in test
     public FTPClient getClient() throws Exception {
-        return pool.borrowObject();
+       return pool.borrowObject();
     }
 
     /**
@@ -56,7 +66,7 @@ public class FTPClientPool {
      */
     public void upload(String filePath, String fileName, InputStream inputStream) throws Exception {
         String upPath;
-        FTPClient client = getClient();
+        FTPClient client = pool.borrowObject();
         try {
             if (StringUtils.isBlank(filePath)) {
                 upPath = ftpRootPath;
@@ -83,12 +93,18 @@ public class FTPClientPool {
      * @throws Exception Exception
      */
     public InputStream download(String filePath) throws Exception {
-        FTPClient client = getClient();
-        try {
-            client.changeWorkingDirectory(ftpRootPath);
-            return StringUtils.isBlank(filePath) ? null : client.retrieveFileStream(filePath);
+        FTPClient client = pool.borrowObject();
+        if (StringUtils.isBlank(filePath)) throw new ServiceException("file path should not empty");
+        client.changeWorkingDirectory(ftpRootPath);
+        try (InputStream in = client.retrieveFileStream(filePath); OutputStream out = new ByteArrayOutputStream();){
+            int len = 0;
+            byte[] bytes = new byte[1024];
+            while ((len = in.read(bytes)) != -1) {
+                out.write(bytes, 0, len);
+            }
+            return new ByteArrayInputStream(((ByteArrayOutputStream) out).toByteArray());
         } finally {
-            //pool.returnObject(client);
+            pool.returnObject(client);
         }
     }
 
@@ -131,7 +147,7 @@ public class FTPClientPool {
         if (StringUtils.isBlank(filePath)) {
             return;
         }
-        FTPClient client = getClient();
+        FTPClient client = pool.borrowObject();
         try {
             if (!isExist(filePath, client)) {
                 return;

atms-api/src/main/java/pwc/taxtech/atms/common/ftp/FtpClientFactory.java

@@ -21,40 +21,36 @@ public class FtpClientFactory implements PooledObjectFactory<FTPClient> {
         this.config = config;
     }
 
-
     @Override
     public PooledObject<FTPClient> makeObject() throws Exception {
         FTPClient ftpClient = new FTPClient();
         if (null != config.getClientTimeout()) {
             ftpClient.setConnectTimeout(config.getClientTimeout());
         }
-        try {
-            ftpClient.connect(config.getHost(), config.getPort());
-            int reply = ftpClient.getReplyCode();
-            if (!FTPReply.isPositiveCompletion(reply)) {
-                ftpClient.disconnect();
-                logger.warn("FTPServer refused connection");
-                return null;
-            }
-            boolean result = ftpClient.login(StringUtils.defaultString(config.getUsername()),
-                    StringUtils.defaultString(config.getPassword()));
-            if (!result) {
-                throw new ServiceException("ftpClient登陆失败! userName:" + config.getUsername() + " ; password:" + config.getPassword());
-            }
-            ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
-            if (null != config.getTransferFileType()) {
-                ftpClient.setFileType(config.getTransferFileType());
-            }
-            ftpClient.setBufferSize(1024);
-            ftpClient.setControlEncoding(StringUtils.defaultString(config.getEncoding(), "UTF-8"));
-            ftpClient.enterLocalPassiveMode();
+        ftpClient.connect(config.getHost(), config.getPort());
+        int reply = ftpClient.getReplyCode();
+        if (!FTPReply.isPositiveCompletion(reply)) {
+            ftpClient.disconnect();
+            logger.warn("FTPServer refused connection");
+            throw new ServiceException("FTPServer refused connection");
+        }
+        boolean result = ftpClient.login(StringUtils.defaultString(config.getUsername()),
+                StringUtils.defaultString(config.getPassword()));
+        if (!result) {
+            throw new ServiceException("ftpClient登陆失败! userName:" + config.getUsername() + " ; password:" + config.getPassword());
+        }
+        ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
+        if (null != config.getTransferFileType()) {
+            ftpClient.setFileType(config.getTransferFileType());
+        }
+        ftpClient.setBufferSize(1024);
+        ftpClient.setControlEncoding(StringUtils.defaultString(config.getEncoding(), "UTF-8"));
+        ftpClient.enterLocalPassiveMode();
 //            if (StringUtils.equals(config.getPassiveMode(), "true")) {
 //                ftpClient.enterLocalPassiveMode();
 //            }
-        } catch (Exception e) {
-            logger.error("create ftp client error", e);
-        }
         return new DefaultPooledObject<>(ftpClient);
+
     }
 
     @Override
@@ -64,11 +60,10 @@ public class FtpClientFactory implements PooledObjectFactory<FTPClient> {
             if (ftpClient != null && ftpClient.isConnected()) {
                 ftpClient.logout();
             }
-        } catch (IOException io) {
-            io.printStackTrace();
         } finally {
             try {
-                ftpClient.disconnect();
+                if (ftpClient != null && ftpClient.isConnected())
+                    ftpClient.disconnect();
             } catch (IOException io) {
                 io.printStackTrace();
             }