/*
* This is eMonocot, a global online biodiversity information resource.
*
* Copyright © 2011–2015 The Board of Trustees of the Royal Botanic Gardens, Kew and The University of Oxford
*
* eMonocot is free software: you can redistribute it and/or modify it under the terms of the
* GNU Affero General Public License as published by the Free Software Foundation, either version 3
* of the License, or (at your option) any later version.
*
* eMonocot 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 Affero General Public License for more details.
*
* The complete text of the GNU Affero General Public License is in the source repository as the file
* ‘COPYING’. It is also available from <http://www.gnu.org/licenses/>.
*/
package org.emonocot.harvest.common;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.batch.core.StepContribution;
import org.springframework.batch.core.scope.context.ChunkContext;
import org.springframework.batch.core.step.tasklet.Tasklet;
import org.springframework.batch.repeat.RepeatStatus;
import org.springframework.core.io.FileSystemResource;
import org.springframework.core.io.Resource;
public class CopyFileTasklet implements Tasklet {
Logger logger = LoggerFactory.getLogger(CopyFileTasklet.class);
private Resource from;
private FileSystemResource to;
private boolean append = false;
public void setFrom(Resource from) {
this.from = from;
}
public void setTo(FileSystemResource to) {
this.to = to;
}
public void setAppend(boolean append) {
this.append = append;
}
@Override
public RepeatStatus execute(StepContribution contribution,
ChunkContext chunkContext) throws Exception {
InputStream inputStream = null;
OutputStream outputStream = null;
try {
inputStream = from.getInputStream();
outputStream = new FileOutputStream(to.getFile(),append);
byte[] buffer = new byte[4096];
int bytes_read;
while ((bytes_read = inputStream.read(buffer)) != -1) {
outputStream.write(buffer, 0, bytes_read);
}
} finally {
if (inputStream != null)
try {
inputStream.close();
} catch (IOException e) {
}
if (outputStream != null)
try {
outputStream.close();
} catch (IOException e) {
;
}
}
return RepeatStatus.FINISHED;
}
}