com.github.wzc789376152.file.config.ftp.FtpClientFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of file Show documentation
文件管理工具
There is a newer version: 2.1.1
package com.github.wzc789376152.file.config.ftp;

import org.apache.commons.net.ftp.FTP;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.pool2.PooledObject;
import org.apache.commons.pool2.PooledObjectFactory;
import org.apache.commons.pool2.impl.DefaultPooledObject;

import java.io.File;
import java.io.IOException;

public class FtpClientFactory implements PooledObjectFactory {
    public FtpClientFactory(FtpProperties properties) {
        this.properties = properties;
        if (properties == null) {
            properties = new FtpProperties();
        }
    }

    protected FtpProperties properties;

    //创建连接到池中
    @Override
    public PooledObject makeObject() {
        FTPClient ftpClient = new FTPClient();//创建客户端实例
        return new DefaultPooledObject<>(ftpClient);
    }

    //销毁连接，当连接池空闲数量达到上限时，调用此方法销毁连接
    @Override
    public void destroyObject(PooledObject pooledObject) {
        FTPClient ftpClient = pooledObject.getObject();
        try {
            ftpClient.logout();
            if (ftpClient.isConnected()) {
                ftpClient.disconnect();
            }
        } catch (IOException e) {
            throw new RuntimeException("Could not disconnect from server.", e);
        }
    }

    //链接状态检查
    @Override
    public boolean validateObject(PooledObject pooledObject) {
        FTPClient ftpClient = pooledObject.getObject();
        try {
            return ftpClient.sendNoOp();
        } catch (IOException e) {
            return false;
        }
    }

    //初始化连接
    @Override
    public void activateObject(PooledObject pooledObject) throws Exception {
        FTPClient ftpClient = pooledObject.getObject();
        ftpClient.connect(properties.getHost(), properties.getPort());
        ftpClient.login(properties.getUsername(), properties.getPassword());
        if (properties.getEncoding() != null) {
            ftpClient.setControlEncoding(properties.getEncoding());
        }
        ftpClient.setFileType(FTP.BINARY_FILE_TYPE);//设置上传文件类型为二进制，否则将无法打开文件
    }

    //钝化连接，使链接变为可用状态
    @Override
    public void passivateObject(PooledObject pooledObject) throws Exception {
        FTPClient ftpClient = pooledObject.getObject();
        try {
            ftpClient.changeWorkingDirectory("/");
            ftpClient.logout();
            if (ftpClient.isConnected()) {
                ftpClient.disconnect();
            }
        } catch (IOException e) {
            throw new RuntimeException("Could not disconnect from server.", e);
        }
    }

    //用于连接池中获取pool属性
    public FtpProperties getProperties() {
        return properties;
    }
}