FileIoLocal.java

package com.boco.nbd.wios.downloadfile.fileio;

import com.boco.nbd.wios.downloadfile.mapper.def.CptDataInfoMapper;
import com.boco.nbd.wios.downloadfile.model.CptDataInfo;
import com.boco.nbd.wios.downloadfile.service.DataStoreService;
import com.ihidea.core.support.exception.ServiceException;
import com.ihidea.core.util.DateUtilsEx;
import org.apache.catalina.connector.ClientAbortException;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Date;

/**
 * 存储本地文件IO
 * @author TYOTANN
 */
@Component
public class FileIoLocal implements IFileIo {

	private final Log logger = LogFactory.getLog(FileIoLocal.class);

	@Autowired
	private DataStoreService dataStoreService;

	@Resource
	private CptDataInfoMapper dataInfoDao;

	/**
	 * 保存到存储路径
	 */
	@Override
	public void save(FileIoEntity entity) {

		saveFile(entity.getDataInfo().getId(), entity.getContent(),
				getPath(entity.getDataInfo().getStoreName(), entity.getDataInfo().getCreateTime()));

	}

	/**
	 * 保存到备份存储路径
	 */
	@Override
    public void saveBak(FileIoEntity entity) {

		saveFile(entity.getDataInfo().getId(), entity.getContent(),
				getBakPath(entity.getDataInfo().getStoreName(), entity.getDataInfo().getCreateTime()));

	}

	/**
	 * 持久化
	 */
	private void saveFile(String id, byte[] content, String storePath) {

		File file = new File(storePath + File.separator + id);

		if (file.exists()) {
			throw new ServiceException("文件:" + file.getPath() + "已经存在!");
		}

		// 文件夹不存在的话则创建
		if (!file.getParentFile().exists()) {
			file.getParentFile().mkdirs();
		}

		try {
			FileUtils.writeByteArrayToFile(file, content);
		} catch (IOException e) {
			throw new ServiceException(e);
		}
	}

	@Override
	public boolean remove(FileIoEntity entity) {

		String path = getPath(entity.getDataInfo().getStoreName(), entity.getDataInfo().getCreateTime());

		File file = new File(path + File.separator + entity.getDataInfo().getId());

		// file.exists()==true?file.delete():true;
		return file.delete();
	}

	@Override
	public byte[] get(String id) {

		CptDataInfo dataInfo = dataInfoDao.selectByPrimaryKey(id);

		String path = getPath(dataInfo.getStoreName(), dataInfo.getCreateTime());

		try {

			File downloadFile = new File(path + File.separator + id);

			if (downloadFile.exists()) {
				return FileUtils.readFileToByteArray(downloadFile);
			} else {
				return null;
			}

		} catch (IOException e) {
			throw new ServiceException(e);
		}
	}

	@Override
	public void execute(FileIoEntity fileIoEntity, IFileInputStream fileInputStreamImpl) throws Exception {

		String filePath = getPath(fileIoEntity.getDataInfo().getStoreName(), fileIoEntity.getDataInfo().getCreateTime()) + File.separator
				+ fileIoEntity.getDataInfo().getId();
		File downloadFile = new File(filePath);

		if (downloadFile.exists()) {

			FileInputStream fis = null;

			try {
				fis = FileUtils.openInputStream(downloadFile);
				fileInputStreamImpl.execute(fileIoEntity, fis);
			} catch (ClientAbortException e) {
			} catch (IOException e) {
				throw new ServiceException(e);
			} finally {
				if (fis == null) {
					IOUtils.closeQuietly(fis);
				}
			}
		} else {
			logger.info("文件不存在:" + filePath);
			fileInputStreamImpl.execute(null, null);
		}

	}

	private String getPath(String storeName, Date createDate) {

		String storePath = dataStoreService.getInfoByName(storeName).getPath();

		return storePath + DateUtilsEx.formatToString(createDate, DateUtilsEx.DATE_FORMAT_DAY).replace(".", File.separator);
	}

	private String getBakPath(String storeName, Date createDate) {

		String storePath = dataStoreService.getInfoByName(storeName).getBakPath();

		return storePath + DateUtilsEx.formatToString(createDate, DateUtilsEx.DATE_FORMAT_DAY).replace(".", File.separator);
	}

	@Override
	public void updateContent(String id, byte[] content) {

		CptDataInfo dataInfo = dataInfoDao.selectByPrimaryKey(id);

		String path = getPath(dataInfo.getStoreName(), dataInfo.getCreateTime());

		File file = new File(path + File.separator + id);

		file.deleteOnExit();
		try {
			FileUtils.writeByteArrayToFile(file, content);
		} catch (IOException e) {
			throw new ServiceException(e);
		}
	}
}