diff options
author | 2021-07-12 15:25:20 +0200 | |
---|---|---|
committer | 2021-07-28 14:22:05 +0200 | |
commit | cc41ed25adcd0d5a4714ac92eee387c41eacbd3e (patch) | |
tree | 84cd4979946b23053bcd7bc2cd29e7540ef6b242 /src/main/java/org/onap/a1pesimulator | |
parent | aaf4218122ed0e18912581a04ee677f6613b8e8f (diff) |
O1 PM Bulk support - FTP upload or file copy implementation
Issue-ID: INT-1945
Signed-off-by: Stanislav Marszalek <s.marszalek2@partner.samsung.com>
Change-Id: If08908035719798d8d7b129ddcdb6ef83f1647fe
Diffstat (limited to 'src/main/java/org/onap/a1pesimulator')
-rw-r--r-- | src/main/java/org/onap/a1pesimulator/service/fileready/FtpServerService.java | 39 |
1 files changed, 37 insertions, 2 deletions
diff --git a/src/main/java/org/onap/a1pesimulator/service/fileready/FtpServerService.java b/src/main/java/org/onap/a1pesimulator/service/fileready/FtpServerService.java index 3d751f5..d71cdf0 100644 --- a/src/main/java/org/onap/a1pesimulator/service/fileready/FtpServerService.java +++ b/src/main/java/org/onap/a1pesimulator/service/fileready/FtpServerService.java @@ -10,12 +10,14 @@ import java.io.IOException; import java.nio.file.Files; import java.util.zip.GZIPOutputStream; + import org.onap.a1pesimulator.data.fileready.FileData; import org.onap.a1pesimulator.exception.NotUploadedToFtpException; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Service; +import org.springframework.util.FileCopyUtils; import net.schmizz.sshj.SSHClient; import net.schmizz.sshj.sftp.SFTPClient; @@ -27,6 +29,14 @@ public class FtpServerService { private static final Logger log = LoggerFactory.getLogger(FtpServerService.class); + //true - file will be uploaded to FTP; false - file will be copied into xmlPmLocation + @Value("${ftp.server.upload}") + private boolean ftpServerUpload; + + // location where archived file will be copied + @Value("${xml.pm.location}") + private String xmlPmLocation; + @Value("${ftp.server.url}") private String ftpServerUrl; @@ -48,7 +58,7 @@ public class FtpServerService { public Mono<FileData> uploadFileToFtp(FileData fileData) { return Mono.just(fileData) .flatMap(this::tryToCompressFile) - .flatMap(this::tryToUploadFileToFtp) + .flatMap(this::tryToUploadOrSaveFileToFtp) .onErrorResume(throwable -> resumeError(throwable, fileData)) .doOnNext(file -> deletePMBulkFile(file.getPmBulkFile())); } @@ -73,12 +83,33 @@ public class FtpServerService { log.trace("Compressing file {}", fileData.getPmBulkFile().getName()); return Mono.just(fileData); } catch (IOException e) { - log.error("Could not compress file ", e); + log.error("Could not compress file", e); return Mono.empty(); } } /** + * Upload file to FTP or copy it to mounted location + * + * @param fileData data about file + * @return fileData for fileReadyEvent + */ + private Mono<FileData> tryToUploadOrSaveFileToFtp(FileData fileData) { + if (ftpServerUpload) { + return tryToUploadFileToFtp(fileData); + } else { + File fileOnFtp = new File(xmlPmLocation, fileData.getArchivedPmBulkFile().getName()); + try { + FileCopyUtils.copy(fileData.getArchivedPmBulkFile(), fileOnFtp); + log.info("Uploading file to the location: {}", fileOnFtp); + return Mono.just(fileData); + } catch (IOException e) { + return Mono.error(new NotUploadedToFtpException("File was not copied to FTP location", e)); + } + } + } + + /** * Upload file to FTP * * @param fileData archived file in Mono @@ -155,6 +186,10 @@ public class FtpServerService { /** * Try to clean up things after an exception + * + * @param throwable error thrown + * @param fileData data about files which needs to be deleted + * @return empty Mono object */ private Mono<FileData> resumeError(Throwable throwable, FileData fileData) { log.error("Error occurs while uploading file to FTP server", throwable); |