net.seedboxer.common.ftp.FtpUploaderCommons Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of ftp-common Show documentation
Show all versions of ftp-common Show documentation
SeedBoxer FTP common library
The newest version!
/*******************************************************************************
* FtpUploaderCommons.java
*
* Copyright (c) 2012 Team SeedBoxer.
*
* This file is part of SeedBoxer FTPCommon.
*
* SeedBoxer FTPCommon is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* SeedBoxer FTPCommon is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with SeedBoxer FTPCommon. If not, see filesInServer = listFiles();
if (fileToUpload.isDirectory()) {
uploadDirectory(fileToUpload, filesInServer, listener);
} else {
uploadFile(fileToUpload, filesInServer, listener);
}
} catch (FtpException e) {
throw e;
} catch (IOException e) {
throw new FtpTransferException(e);
}
}
private void uploadDirectory(File directoryToUpload, Map filesInServer,
FtpUploaderListener listener) throws IOException {
// Create the directory if it was not created
boolean exist = filesInServer.containsKey(directoryToUpload.getName());
if (!exist) {
try {
LOGGER.debug(""+ftpClient.makeDirectory(directoryToUpload.getName()));
LOGGER.debug("Directory created! {}", directoryToUpload.getName());
} catch (Exception e){
LOGGER.debug("Could not create directory, may be it already exist");
}
}
// Change to the directory
LOGGER.debug("Moving to directory {}", directoryToUpload.getName());
ftpClient.changeWorkingDirectory(directoryToUpload.getName());
// Check the files inside the directory
Map filesInServerDirectory;
if (!exist){
filesInServerDirectory = Collections.emptyMap();
}else{
filesInServerDirectory = listFiles();
}
// Upload all directories first
for (File childFile : directoryToUpload.listFiles(directoryFileFilter)){
uploadDirectory(childFile, filesInServerDirectory, listener);
}
// Upload all files
for (File childFile : directoryToUpload.listFiles(normalFileFilter)){
uploadFile(childFile, filesInServerDirectory, listener);
}
// Back to the original directory
LOGGER.debug("Moving to directory up");
ftpClient.changeToParentDirectory();
}
private void uploadFile(File fileToUpload, Map filesListInServer,
FtpUploaderListener listener) throws IOException {
String fileName = fileToUpload.getName();
Long size = filesListInServer.get(fileName);
if (size != null && size != 0L) {
// Tell listener that already exist and transfer (part) of the file
listener.bytesTransferred(size);
if (size == fileToUpload.length()) {
LOGGER.debug("File already exists {}", fileName);
} else {
LOGGER.trace("Resuming file {} from {} MB", fileName, (size / (1024*1024)));
// Set the offset
ftpClient.setRestartOffset(size);
// Create stream and skip first SIZE bytes
InputStream ins = new FileInputStream(fileToUpload);
ins.skip(size);
// Upload file
storeFile(fileName, ins, fileToUpload.length()-size, listener);
LOGGER.debug("File {} successfully uploaded", fileName);
}
} else {
storeFile(fileName, new FileInputStream(fileToUpload), fileToUpload.length(), listener);
LOGGER.debug("File {} successfully uploaded", fileName);
}
}
private void storeFile(String fileName, InputStream ins, Long size,
final FtpUploaderListener listener) throws IOException {
cleanAbortedStatus();
OutputStream outs = ftpClient.storeFileStream(fileName);
CopyStreamAdapter adapter = new CopyStreamAdapter() {
@Override
public void bytesTransferred(long totalBytesTransferred, int bytesTransferred, long streamSize) {
if (listener != null) {
listener.bytesTransferred(bytesTransferred);
}
if (aborted) {
throw new AbortedTransferException();
}
}
};
try {
Util.copyStream(ins, outs, ftpClient.getBufferSize(), size, adapter);
} finally {
outs.close();
ins.close();
}
try {
if (!ftpClient.completePendingCommand()) {
throw new FtpTransferException();
}
} catch (MalformedServerReplyException e) {
if (!e.getMessage().toLowerCase().contains("ok") &&
!e.getMessage().toLowerCase().contains("complete")) {
throw e;
}
}
}
private void cleanAbortedStatus() {
aborted = false;
}
/**
* List files inside the current folder.
*
* @return List with files names and size
* @throws IOException
*/
private Map listFiles() throws FtpException {
int attempts = 0;
Map files = new LinkedHashMap();
while (true){
try {
FTPListParseEngine engine = null;
if (type.startsWith("UNIX")) {
engine = ftpClient.initiateListParsing(FTPClientConfig.SYST_UNIX, null);
} else {
engine = ftpClient.initiateListParsing();
}
FTPFile[] list = engine.getFiles();
if (list != null){
for (FTPFile ftpFile : list){
files.put(ftpFile.getName(), ftpFile.getSize());
}
}
return files;
} catch (Exception e) {
attempts++;
if (attempts > 3) {
throw new FtpListFilesException(e);
} else {
LOGGER.trace("First attempt to get list of files FAILED! attempt={}", attempts);
}
}
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy