最近再開發中遇到需要將文件上傳到Linux服務器上,至此整理代碼筆記。
此種連接方法中有考慮到並發問題,在進行創建FTP連接的時候將每一個連接對象存放至
ThreadLocal<Ftp> 中以確保每個線程之間對FTP的打開與關閉互不影響。
package com.test.utils;
import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileFilter;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.Session;
public class Ftp {
//打印log日志
private static final Log logger = LogFactory.getLog(Ftp.class);
private static Date last_push_date = null;
private Session sshSession;
private ChannelSftp channel;
private static ThreadLocal<Ftp> sftpLocal = new ThreadLocal<Ftp>();
private Ftp(String host, int port, String username, String password) throws Exception {
JSch jsch = new JSch();
jsch.getSession(username, host, port);
//根據用戶名,密碼,端口號獲取session
sshSession = jsch.getSession(username, host, port);
sshSession.setPassword(password);
//修改服務器/etc/ssh/sshd_config 中 GSSAPIAuthentication的值yes為no,解決用戶不能遠程登錄
sshSession.setConfig("userauth.gssapi-with-mic", "no");
//為session對象設置properties,第一次訪問服務器時不用輸入yes
sshSession.setConfig("StrictHostKeyChecking", "no");
sshSession.connect();
//獲取sftp通道
channel = (ChannelSftp)sshSession.openChannel("sftp");
channel.connect();
logger.info("連接ftp成功!" + sshSession);
}
/**
* 是否已連接
*
* @return
*/
private boolean isConnected() {
return null != channel && channel.isConnected();
}
/**
* 獲取本地線程存儲的sftp客戶端
*
* @return
* @throws Exception
*/
public static Ftp getSftpUtil(String host, int port, String username, String password) throws Exception {
//獲取本地線程
Ftp sftpUtil = sftpLocal.get();
if (null == sftpUtil || !sftpUtil.isConnected()) {
//將新連接防止本地線程,實現並發處理
sftpLocal.set(new Ftp(host, port, username, password));
}
return sftpLocal.get();
}
/**
* 釋放本地線程存儲的sftp客戶端
*/
public static void release() {
if (null != sftpLocal.get()) {
sftpLocal.get().closeChannel();
logger.info("關閉連接" + sftpLocal.get().sshSession);
sftpLocal.set(null);
}
}
/**
* 關閉通道
*
* @throws Exception
*/
public void closeChannel() {
if (null != channel) {
try {
channel.disconnect();
} catch (Exception e) {
logger.error("關閉SFTP通道發生異常:", e);
}
}
if (null != sshSession) {
try {
sshSession.disconnect();
} catch (Exception e) {
logger.error("SFTP關閉 session異常:", e);
}
}
}
/**
* @param directory 上傳ftp的目錄
* @param uploadFile 本地文件目錄
*
*/
public void upload(String directory, String uploadFile) throws Exception {
try {
//執行列表展示ls 命令
channel.ls(directory);
//執行盤符切換cd 命令
channel.cd(directory);
List<File> files = getFiles(uploadFile, new ArrayList<File>());
for (int i = 0; i < files.size(); i++) {
File file = files.get(i);
InputStream input = new BufferedInputStream(new FileInputStream(file));
channel.put(input, file.getName());
try {
if (input != null) input.close();
} catch (Exception e) {
e.printStackTrace();
logger.error(file.getName() + "關閉文件時.....異常!" + e.getMessage());
}
if (file.exists()) {
boolean b = file.delete();
logger.info(file.getName() + "文件上傳完畢!刪除標識:" + b);
}
}
}catch (Exception e) {
logger.error("【子目錄創建中】:",e);
//創建子目錄
channel.mkdir(directory);
}
}
//獲取文件
public List<File> getFiles(String realpath, List<File> files) {
File realFile = new File(realpath);
if (realFile.isDirectory()) {
File[] subfiles = realFile.listFiles(new FileFilter() {
@Override
public boolean accept(File file) {
if (null == last_push_date ) {
return true;
} else {
long modifyDate = file.lastModified();
return modifyDate > last_push_date.getTime();
}
}
});
for (File file : subfiles) {
if (file.isDirectory()) {
getFiles(file.getAbsolutePath(), files);
} else {
files.add(file);
}
if (null == last_push_date) {
last_push_date = new Date(file.lastModified());
} else {
long modifyDate = file.lastModified();
if (modifyDate > last_push_date.getTime()) {
last_push_date = new Date(modifyDate);
}
}
}
}
return files;
}
}
