com.quincy.core.sftp.impl.ChannelSftpSourceImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quincy-sftp Show documentation

The newest version!

package com.quincy.core.sftp.impl;

import org.apache.commons.pool2.ObjectPool;
import org.apache.commons.pool2.SwallowedExceptionListener;
import org.apache.commons.pool2.impl.AbandonedConfig;
import org.apache.commons.pool2.impl.GenericObjectPool;
import org.apache.commons.pool2.impl.GenericObjectPoolConfig;

import com.jcraft.jsch.ChannelSftp;
import com.quincy.core.sftp.ChannelSftpSource;
import com.quincy.core.sftp.PoolableChannelSftp;
import com.quincy.core.sftp.PoolableChannelSftpFactory;

import lombok.extern.slf4j.Slf4j;

@Slf4j
public class ChannelSftpSourceImpl implements ChannelSftpSource {
	private ObjectPool pool;

	public ChannelSftpSourceImpl(PoolableChannelSftpFactory f, GenericObjectPoolConfig pc, AbandonedConfig ac) {
		GenericObjectPool pool = new GenericObjectPool(f, pc, ac);
		pool.setSwallowedExceptionListener(new SwallowedExceptionListener() {
			@Override
			public void onSwallowException(Exception e) {
				log.error("\r\nSFTP_CONNECTION_POOL_ERR:\r\n", e);
			}
		});
		f.setPool(pool);
		this.pool = pool;
	}

	@Override
	public ChannelSftp get() throws Exception {
		PoolableChannelSftp channel = null;
		while(true) {
			channel = pool.borrowObject();
			if(channel.isClosed()||!channel.isConnected()) {
				channel.reallyDisconnect();
				pool.invalidateObject(channel);
			} else
				return channel;
		}
	}

	@Override
	public void close() {
		this.pool.close();
	}
}