AttachmentServiceImpl

package gboat2.attachment.service.impl;

import gboat2.base.core.dao.QuerySupport;
import gboat2.base.core.service.BaseService;
import gboat2.attachment.business.IAttachConfigBusiness;
import gboat2.attachment.model.AttachConfig;
import gboat2.attachment.model.Attachment;
import gboat2.attachment.model.AttachmentOperEnum;
import gboat2.attachment.model.AttachmentVO;
import gboat2.attachment.model.DownloadRecord;
import gboat2.attachment.service.IAttachmentService;
import gboat2.attachment.util.ConfigUtil;
import gboat2.attachment.util.MD5Util;

import java.io.File;
import java.io.IOException;
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.transaction.annotation.Transactional;

/**
 * 附件服务类
 * @author tanxw
 * @since jdk1.6
 * @date 2012-4-23
 *
 */
@Transactional
public class AttachmentServiceImpl extends BaseService implements IAttachmentService {

	@Autowired
	private IAttachConfigBusiness attachConfigBusiness;

	@SuppressWarnings("deprecation")
    @Override
	public String[][] getAvailableAttachTypeByCode(String attachCode) {
		return attachConfigBusiness.getAvailableAttachTypeByCode(attachCode);
	}

	@Override
	public AttachConfig getAttachConfigBy(String attachCode, String attachType) {
		return attachConfigBusiness.getAttachConfigBy(attachCode, attachType);
	}

	@Override
	public String getAttachConfigIdBy(String attachCode, String attachType) {
		return attachConfigBusiness.getAttachConfigIdBy(attachCode, attachType);
	}

	public void deleteAttachments(String[] attachIds) {
		if (null == attachIds) {
			return;
		}
		for (String s : attachIds) {
			String fileid = s.trim();
			Attachment attach = (Attachment) this.get(Attachment.class, fileid);
			if (null != attach) {
				//不删除实体文件
				//				attach.setUsable("0");
				//				this.update(attach);
				this.delete(attach);
				this.deleteRealFile(attach.getSavePath());
			}
		}
	}

	private void deleteRealFile(String path) {
		if (StringUtils.isNotBlank(path)) {
			File file = new File(ConfigUtil.getSavepathRoot() + path);
			if (file.exists()) {
				file.delete();
			}
		}
	}

	public void deleteAttachment(String attachId) {
		Attachment attach = (Attachment) this.get(Attachment.class, attachId);
		if (null != attach) {
			//不删除实体文件
			//				attach.setUsable("0");
			//				this.update(attach);
			this.delete(attach);
			this.deleteRealFile(attach.getSavePath());
		}
	}

	/**
	  * {@inheritDoc}
	  * @see gboat2.attachment.service.IAttachmentService#saveAttachment(gboat2.attachment.model.Attachment)
	  */
	public String saveAttachment(Attachment attach) {
		return baseDAO.save(attach);
	}

	@SuppressWarnings("unchecked")
    public List<Attachment> getAttachmentsByBelongId(String belongId) {
		HashMap<String, Object> params = new HashMap<String, Object>();
		params.put(QuerySupport.PARAM_TABLENAME, Attachment.class);
		params.put("_belongId", belongId);
		params.put(QuerySupport.PARAM_ORDERBY, "uploadDate desc");
		List<Attachment> list = (List<Attachment>) baseDAO.queryList(params);
		return list;
	}

	@SuppressWarnings("unchecked")
    public List<AttachmentVO> getAttachmentVOsByBelongId(String belongId) {
		HashMap<String, Object> params = new HashMap<String, Object>();
		params.put(QuerySupport.PARAM_TABLENAME, AttachmentVO.class);
		params.put("_belongId", belongId);

		params.put(QuerySupport.PARAM_ORDERBY, "attachOrder");

		List<AttachmentVO> list = (List<AttachmentVO>) baseDAO.queryList(params);

		for (Iterator<AttachmentVO> iterator = list.iterator(); iterator.hasNext();) {
			AttachmentVO attachmentVO = (AttachmentVO) iterator.next();
			if (StringUtils.isNotEmpty(attachmentVO.getConfigId())) {
				attachmentVO.setAttachName(attachmentVO.getConfigAttachType());
			}
		}

		return list;
	}

	@SuppressWarnings("unchecked")
    @Override
	public List<Attachment> getAttachmentsBy(String belongId, String[] attachNames) {
		if (attachNames == null) {
			return null;
		}
		Map<String, Object> params = new HashMap<String, Object>();
		params.put(QuerySupport.PARAM_TABLENAME, Attachment.class);
		params.put("_belongId", belongId);
		params.put("_attachName_in", convertAttachNameArrayToString(attachNames));
		List<Attachment> atts = (List<Attachment>) super.query(params);
		return atts;
	}

	@Override
	public List<Attachment> getAttachmentsByAttachIds(String attachIds) {
		if (StringUtils.isEmpty(attachIds)) {
			return Collections.emptyList();
		}
		Map<String, Object> params = new HashMap<String, Object>();
		params.put(QuerySupport.PARAM_TABLENAME, Attachment.class);
		params.put("_attachId_in", attachIds);
		List<Attachment> atts = (List<Attachment>) super.query(params);
		return atts;
	}


	private String convertAttachNameArrayToString(String[] attachNames) {
		StringBuilder builder = new StringBuilder();
		for (String attachName : attachNames) {
			builder.append("'").append(attachName).append("',");
		}
		builder.deleteCharAt(builder.length() - 1);
		return builder.toString();
	}

	@SuppressWarnings("unchecked")
    @Override
	public List<Attachment> getAttachmentsBy(String belongId, String attachType) {
		if (StringUtils.isEmpty(belongId) || StringUtils.isEmpty(attachType)) {
			return null;
		}
		Map<String, Object> params = new HashMap<String, Object>();
		params.put(QuerySupport.PARAM_TABLENAME, Attachment.class);
		params.put("_belongId", belongId);
		params.put("_attachType", attachType);
		List<Attachment> atts = (List<Attachment>) super.query(params);
		return atts;
	}

	/* (non-Javadoc)
	 * @see gboat2.attachment.service.IUploadFileService#getAttachmentVOBy(java.lang.String, java.lang.String)
	 */
	@SuppressWarnings("unchecked")
    @Override
	public List<AttachmentVO> getAttachmentVOBy(String belongId, String attachType) {
		if (StringUtils.isEmpty(belongId) || StringUtils.isEmpty(attachType)) {
			return null;
		}
		Map<String, Object> params = new HashMap<String, Object>();
		params.put(QuerySupport.PARAM_TABLENAME, AttachmentVO.class);
		params.put("_belongId", belongId);
		params.put("_configAttachType", attachType);
		List<AttachmentVO> atts = (List<AttachmentVO>) super.query(params);
		return atts;
	}

	/* (non-Javadoc)
	 * @see gboat2.attachment.service.IUploadFileService#getAttachmentsByBelongIdAndConfigId(java.lang.String, java.lang.String)
	 */
	@SuppressWarnings("unchecked")
    @Override
	public List<Attachment> getAttachmentsByBelongIdAndConfigId(String belongId, String configId) {
		if (StringUtils.isEmpty(belongId)) {
			return null;
		}
		Map<String, Object> params = new HashMap<String, Object>();
		params.put(QuerySupport.PARAM_TABLENAME, Attachment.class);
		params.put("_belongId", belongId);
		if (StringUtils.isEmpty(configId)) {
			params.put("_configId_null", null);
		} else {
			params.put("_configId", configId);
		}
		List<Attachment> atts = (List<Attachment>) super.query(params);
		return atts;
	}

	public void updateAttachsBelongId(String[] attachIds, String belongId) {
		if (null == attachIds || attachIds.length <= 0) {
			return;
		}

		StringBuilder builder = new StringBuilder();
		for (String attachId : attachIds) {
			builder.append("'").append(attachId).append("',");
		}
		builder.deleteCharAt(builder.length() - 1);

		Map<String, Object> params = new HashMap<String, Object>();
		params.put("belongId", belongId);
		baseDAO.updateByQuery("update Attachment set belongId=:belongId where attachId in (" + builder.toString() + ")", params);
	}

	@SuppressWarnings("unchecked")
    @Override
	public DownloadRecord getDownLoad(String attachmentId) {
		Map<String, Object> params = new HashMap<String, Object>();
		params.put(QuerySupport.PARAM_TABLENAME, DownloadRecord.class);
		params.put("_fileId", attachmentId);
		List<DownloadRecord> recordList = (List<DownloadRecord>) super.query(params);
		if (null != recordList && recordList.size() > 0) {
			return recordList.get(0);
		}
		return null;
	}

	/**
	 * 删除 附件 by belongId
	 * @param belongId
	 * 		主对象的主键，同时又是附件的belongId
	 */
	@SuppressWarnings("unchecked")
    @Override
	public void deleteAttachmentByBelongId(String belongId) {
		if (StringUtils.isEmpty(belongId)) {
			return;
		}
		Map<String, Object> params = new HashMap<String, Object>();
		params.put(QuerySupport.PARAM_TABLENAME, Attachment.class);
		params.put("_belongId", belongId);
		List<Attachment> list = (List<Attachment>) baseDAO.queryList(params);
		if (list != null) {
			for (Attachment attachment : list) {
				if (attachment == null) {
					continue;
				}
				this.deleteAttachment(attachment.getAttachId());
			}
		}
	}

	@Override
	public void virtualDelete(String attachId) {
		Attachment attach = (Attachment) this.get(Attachment.class, attachId);
		if (null != attach) {
			this.delete(attach); //删除记录，不删除文件
		}

	}

	/* (non-Javadoc)
	 * @see gboat2.attachment.service.IUploadFileService#saveAttachment(byte[], java.lang.String, java.lang.String, java.lang.String)
	 */
	@Override
	public Attachment generateAttachment(byte[] fileContent, String belongId, String attachName, String attachType, String configId,
			AttachmentOperEnum attOperEnum) throws IOException {
		String relativeDir = ConfigUtil.generateRelativeSaveDir();
		String targetDir = ConfigUtil.getSavepathRoot() + relativeDir; //目标文件夹
		String targetFileName = ConfigUtil.genearteNewFilename(attachName);
		String targetFullFileName = targetDir + targetFileName;
		//保存到po字段中的savepath值
		String savepath = relativeDir + targetFileName;
		String downloadurl = ConfigUtil.getDownloadContextPath() + savepath;

		// 判断当前文件夹存在不存在
		File path = new File(targetDir);
		if (!path.exists() && !path.mkdirs()) {
			String msg = "文件夹创建失败！：" + targetDir;
			//logger.error(msg);
			throw new IOException(msg);
		}

		File file = null;
		try {
			file = new File(targetFullFileName);
			FileUtils.writeByteArrayToFile(file, fileContent);
		} catch (IOException e1) {
			String msg = "服务器保存文件失败";
			//logger.error(msg, e1);
			throw new IOException(msg, e1);
		}

		String md5 = "";
		try {
			md5 = MD5Util.getFileMD5String(file);
		} catch (IOException e) {
			String msg = "生产MD5校验串失败";
			throw new IOException(msg, e);
		}
		/*
		 * 1.如果操作为null 空值，默认为save保存操作；
		 * 2.如果操作为save， 则为 save保存操作；
		 * 3.如果操作为updae，则为update 更新操作；
		 * 4.如果操作为saveorupdate，则为更新或新增保存操作；
		 * 5.如果操作为其他，则默认为保存操作。
		 */
		Attachment att = null;
		List<Attachment> atts = null;
		if (null == attOperEnum) {
			att = new Attachment();
		} else if (attOperEnum.equals(AttachmentOperEnum.SAVE)) {
			att = new Attachment();
		} else if (attOperEnum.equals(AttachmentOperEnum.UPDATE)) {
			atts = (List<Attachment>) this.getAttachmentsByBelongIdAndConfigId(belongId, configId);
			if (atts != null && atts.size() > 0) {
				att = atts.get(0);
			} else {
				att = new Attachment();
			}
		} else if (attOperEnum.equals(AttachmentOperEnum.SAVEORUPDATE)) {
			atts = (List<Attachment>) this.getAttachmentsByBelongIdAndConfigId(belongId, configId);
			if (atts != null && atts.size() > 0) {
				att = atts.get(0);
			} else {
				att = new Attachment();
			}
		} else {
			att = new Attachment();
		}

		att.setBelongId(belongId);
		att.setAttachName(attachType);
		att.setAttachType(attachType);
		att.setConfigId(configId);
		att.setDownloadUrl(downloadurl);
		att.setOriginalName(attachName);
		att.setSavePath(savepath);
		att.setValidateCode(md5);
		att.setUploadDate(new Date());
		att.setSize(file.length());

		if (att != null && StringUtils.isNotEmpty(att.getAttachId())) {
			//(更新附件)可以多次推送附件
			this.update(att);
		} else {
			this.saveAttachment(att);
		}
		return att;
	}

	@SuppressWarnings("unchecked")
	public Boolean isDownLoadFile(String organId, String belongId, String attachType) {
		Map<String, Object> params = new HashMap<String, Object>();

		params.put(QuerySupport.PARAM_TABLENAME, DownloadRecord.class);
		params.put("_organId", organId);
		params.put("_belongId", belongId);
		params.put("_attachType_like", attachType);

		List<Attachment> list = (List<Attachment>) baseDAO.queryList(params);

		return list.size() > 0 ? Boolean.TRUE : Boolean.FALSE;
	}

	@SuppressWarnings("unchecked")
	public DownloadRecord getLastDownloadRecord(String organId, String belongId, String attachType) {
		Map<String, Object> params = new HashMap<String, Object>();

		params.put(QuerySupport.PARAM_TABLENAME, DownloadRecord.class);
		params.put("_organId", organId);
		params.put("_belongId", belongId);
		params.put("_attachType_like", attachType);

		params.put(QuerySupport.PARAM_ORDERBY, "downloadDate desc");

		List<DownloadRecord> list = (List<DownloadRecord>) baseDAO.queryList(params);

		if (list.size() > 0)
			return list.get(0);

		return null;
	}

	@Override
	public List<AttachConfig> getAvailableAttachTypesByCode(String attachCode) {
		return attachConfigBusiness.getAvailableAttachTypesByCode(attachCode);
	}

	@Override
	public Attachment getAttachment(String attachId) {
		return (Attachment) this.get(Attachment.class, attachId);
	}
}