package com.walker.file;

import com.walker.file.ftp.FtpConfig;
import com.walker.file.ftp.FtpConnector;
import com.walker.infrastructure.utils.WaitConsoleInput;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

public class TestFtpConnector {

    private static FtpConfig ftpConfig = null;

    /*@Test
    public void testFileSystem() throws Exception{
        DefaultFileSystemEngine fileSystemEngine = new DefaultFileSystemEngine();
        fileSystemEngine.setFileRoot("d:/tmp/");

        FileInputStream inputStream = new FileInputStream(new File("d:/tmp/test.txt"));
        fileSystemEngine.uploadFile(inputStream, "test_fs.txt");
    }*/

//    @BeforeAll
//    public static void SetUp() {
//        ftpConfig = new FtpConfig();
//        ftpConfig.setIp("116.198.40.76");
//        ftpConfig.setPort("22");
//        ftpConfig.setUserName("mysftp");
//        ftpConfig.setPassword("Bjjmy_2020");
//    }

    /*@Test
    public void testMultipleThreadUpload() throws Exception{
        Map<String, String> data = new HashMap<>();
        data.put("d:/tmp/test_user_2200.csv", "test_user_2200.csv");
        data.put("d:/tmp/年度总结及新年计划-2022.pptx", "年度总结及新年计划-2022.pptx");
        data.put("d:/tmp/邮务类收入_增幅业务数据手动录入模板.xlsx", "邮务类收入_增幅业务数据手动录入模板.xlsx");

        for(Map.Entry<String, String> entry : data.entrySet()){
            FileInputStream inputStream = new FileInputStream(new File(entry.getKey()));
            this.runUpload(entry.getValue(), inputStream);
        }

        WaitConsoleInput.waitInTimes(TimeUnit.SECONDS, 6);
    }*/

    private void runUpload(final String fileName, final InputStream inputStream){
        new Thread(){
            public void run(){
                FtpConnector ftpConnector = new FtpConnector(ftpConfig);
                try {
                    ftpConnector.connect();
                    ftpConnector.upload("/train", fileName, inputStream);
                    System.out.println("文件上传完毕:" + fileName + ", " + System.currentTimeMillis());
                    ftpConnector.logout();
                } catch (Exception e) {
                    throw new RuntimeException(e);
                }
            }
        }.start();
    }

/*    @Test
    public void testUploadFile(){

        FtpConnector ftpConnector = new FtpConnector(ftpConfig);
        try {
            ftpConnector.connect();
//            TimeUnit.SECONDS.sleep(3);

            FileInputStream inputStream = new FileInputStream(new File("d:/tmp/test.txt"));
            ftpConnector.upload("/train", "ftpConnector.txt", inputStream);
            ftpConnector.logout();

        } catch (Exception e) {
            System.out.println("ftp连接异常:" + e.getMessage());
            e.printStackTrace();
        }
    }*/
}