springboot sftp连接池实现

以下是使用Spring Boot和commons-pool2实现SFTP连接池的示例代码：

1.添加依赖

<dependency>
    <groupId>com.jcraft</groupId>
    <artifactId>jsch</artifactId>
    <version>0.1.55</version>
</dependency>
<dependency>
    <groupId>org.apache.commons</groupId>
    <artifactId>commons-pool2</artifactId>
    <version>2.6.2</version>
</dependency>

2.配置连接池

@Configuration
public class SftpPoolConfig {

    @Value("${sftp.host}")
    private String host;

    @Value("${sftp.port}")
    private int port;

    @Value("${sftp.username}")
    private String username;

    @Value("${sftp.password}")
    private String password;

    @Value("${sftp.timeout}")
    private int timeout;

    @Value("${sftp.pool.maxTotal}")
    private int maxTotal;

    @Value("${sftp.pool.maxIdle}")
    private int maxIdle;

    @Value("${sftp.pool.minIdle}")
    private int minIdle;

    @Bean
    public GenericObjectPool<ChannelSftp> sftpPool() {
        GenericObjectPoolConfig<ChannelSftp> poolConfig = new GenericObjectPoolConfig<>();
        poolConfig.setMaxTotal(maxTotal);
        poolConfig.setMaxIdle(maxIdle);
        poolConfig.setMinIdle(minIdle);
        poolConfig.setTestOnBorrow(true);
        poolConfig.setTestOnReturn(true);
        poolConfig.setTestWhileIdle(true);
        poolConfig.setTimeBetweenEvictionRunsMillis(60000);
        poolConfig.setMinEvictableIdleTimeMillis(1800000);

        return new GenericObjectPool<>(new SftpPoolFactory(host, port, username, password, timeout), poolConfig);
    }
}

3.实现连接池工厂

public class SftpPoolFactory extends BasePooledObjectFactory<ChannelSftp> {

    private String host;
    private int port;
    private String username;
    private String password;
    private int timeout;

    public SftpPoolFactory(String host, int port, String username, String password, int timeout) {
        this.host = host;
        this.port = port;
        this.username = username;
        this.password = password;
        this.timeout = timeout;
    }

    @Override
    public ChannelSftp create() throws Exception {
        JSch jsch = new JSch();
        Session session = jsch.getSession(username, host, port);
        session.setPassword(password);
        Properties config = new Properties();
        config.put("StrictHostKeyChecking", "no");
        session.setConfig(config);
        session.setTimeout(timeout);
        session.connect();        Channel channel = session.openChannel("sftp");
        channel.connect();
        return (ChannelSftp) channel;
    }

    @Override
    public PooledObject<ChannelSftp> wrap(ChannelSftp sftp) {
        return new DefaultPooledObject<>(sftp);
    }

    @Override
    public void destroyObject(PooledObject<ChannelSftp> p) throws Exception {
        ChannelSftp sftp = p.getObject();
        if (sftp.isConnected()) {
            sftp.disconnect();
        }
    }

    @Override
    public boolean validateObject(PooledObject<ChannelSftp> p) {
        ChannelSftp sftp = p.getObject();
        return sftp.isConnected();
    }
}

4.使用连接池

@Service
public class SftpService {

    @Autowired
    private GenericObjectPool<ChannelSftp> sftpPool;

    public void upload(String remotePath, String fileName, InputStream inputStream) throws Exception {
        ChannelSftp sftp = sftpPool.borrowObject();
        try {
            sftp.cd(remotePath);
        } catch (SftpException e) {
            sftp.mkdir(remotePath);
            sftp.cd(remotePath);
        }
        sftp.put(inputStream, fileName);
        sftpPool.returnObject(sftp);
    }

    public InputStream download(String remotePath, String fileName) throws Exception {
        ChannelSftp sftp = sftpPool.borrowObject();
        sftp.cd(remotePath);
        InputStream inputStream = sftp.get(fileName);
        sftpPool.returnObject(sftp);
        return inputStream;
    }
}

相关问题：