增加自定义设置方法

hs-web · Aug 30, 2017 · 72e62cb · 72e62cb
1 parent 1333cac
commit 72e62cb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/hsweb-expands-request/src/main/java/org/hswebframework/expands/request/ftp/FtpRequest.java b/hsweb-expands-request/src/main/java/org/hswebframework/expands/request/ftp/FtpRequest.java
@@ -1,12 +1,14 @@
 package org.hswebframework.expands.request.ftp;
 
+import org.apache.commons.net.ftp.FTPClient;
 import org.apache.commons.net.ftp.FTPFile;
 
 import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.util.List;
+import java.util.function.Consumer;
 
 /**
  * Created by zhouhao on 16-6-24.
@@ -41,4 +43,5 @@ public interface FtpRequest {
 
     void download(String name, OutputStream outputStream) throws IOException;
 
+    FtpRequest setting(Consumer<FTPClient> clientConsumer);
 }
diff --git a/...request/src/main/java/org/hswebframework/expands/request/ftp/simple/SimpleFtpRequest.java b/...request/src/main/java/org/hswebframework/expands/request/ftp/simple/SimpleFtpRequest.java
@@ -11,6 +11,7 @@
 import java.io.*;
 import java.util.Arrays;
 import java.util.List;
+import java.util.function.Consumer;
 
 public class SimpleFtpRequest implements FtpRequest {
     private Logger logger = LoggerFactory.getLogger(FtpRequest.class);
@@ -37,6 +38,11 @@ public SimpleFtpRequest(String addr, int port, String username, String password)
         }
     }
 
+    public FtpRequest setting(Consumer<FTPClient> clientConsumer) {
+        clientConsumer.accept(ftp);
+        return this;
+    }
+
     @Override
     public FtpRequest encode(String encode) {
         ftp.setControlEncoding(encode);