ars.module.system.service.AbstractAttachmentService Maven / Gradle / Ivy

Go to download
package ars.module.system.service;

import java.util.Map;
import java.util.HashMap;
import java.util.Map.Entry;

import ars.util.Nfile;
import ars.util.Beans;
import ars.util.Strings;
import ars.server.Servers;
import ars.file.Operator;
import ars.file.NameGenerator;
import ars.file.DirectoryGenerator;
import ars.file.RandomNameGenerator;
import ars.file.DateDirectoryGenerator;
import ars.file.disk.DiskOperator;
import ars.invoke.request.Requester;
import ars.module.system.model.Attachment;
import ars.module.system.service.AttachmentService;
import ars.database.service.StandardGeneralService;

/**
 * 附件业务操作抽象实现
 * 
 * @author yongqiangwu
 * 
 * @param 
 *            数据模型
 */
public abstract class AbstractAttachmentService extends StandardGeneralService
		implements AttachmentService {
	private Map operators = new HashMap(); // 文件操作接口对象映射
	private NameGenerator nameGenerator = new RandomNameGenerator(); // 文件名称生成器
	private DirectoryGenerator directoryGenerator = new DateDirectoryGenerator(); // 文件目录生成器

	public AbstractAttachmentService() {
		this.operators.put("*", new DiskOperator());
	}

	public Map getOperators() {
		return operators;
	}

	public void setOperators(Map operators) {
		this.operators = operators;
	}

	/**
	 * 查找文件操作接口对象
	 * 
	 * @param path
	 *            文件名称或路径
	 * @return 文件操作接口对象
	 */
	protected Operator lookupOperator(String path) {
		if (this.operators != null && !this.operators.isEmpty()) {
			path = path.toLowerCase();
			for (Entry entry : this.operators.entrySet()) {
				if (Strings.matches(path, entry.getKey().toLowerCase())) {
					return entry.getValue();
				}
			}
		}
		throw new RuntimeException("No matching file operator found:" + path);
	}

	@Override
	public void deleteObject(Requester requester, T object) {
		super.deleteObject(requester, object);
		String path = object.getPath();
		try {
			this.lookupOperator(path).delete(path);
		} catch (Exception e) {
			throw new RuntimeException(e);
		}
	}

	@Override
	public Attachment upload(Requester requester, Nfile file, Boolean previewable, Boolean downloadable,
			Map parameters) throws Exception {
		String name = file.getName();
		StringBuilder buffer = new StringBuilder();
		if (this.directoryGenerator != null) {
			buffer.append(this.directoryGenerator.generate(name));
		}
		if (this.nameGenerator != null) {
			buffer.append('/').append(this.nameGenerator.generate(name));
		}
		String path = buffer.toString();
		this.lookupOperator(file.getName()).write(file, path);
		T attachment = Beans.getInstance(this.getModel());
		attachment.setName(file.getName());
		attachment.setPath(path);
		attachment.setSize(file.getSize());
		attachment.setCreator(requester.getUser());
		attachment.setPreviewable(previewable == null ? true : previewable);
		attachment.setDownloadable(downloadable == null ? true : downloadable);
		Integer id = (Integer) this.saveObject(requester, attachment);
		attachment.setId(id);
		return attachment;
	}

	@Override
	public Nfile download(final Requester requester, Map parameters) throws Exception {
		final T attachment = this.object(requester, parameters);
		if (attachment == null) {
			return null;
		} else if (attachment.getDownloadable() != Boolean.TRUE) {
			throw new RuntimeException("Attachment is not downloadable:" + attachment);
		}
		final String path = attachment.getPath();
		Nfile file = this.lookupOperator(path).read(path);
		if (file != null) {
			Servers.execute(new Runnable() {

				@Override
				public void run() {
					synchronized (path.intern()) {
						T entity = getRepository().get(attachment.getId());
						if (entity != null) {
							entity.setDownloads(entity.getDownloads() + 1);
							updateObject(requester, entity);
						}
					}
				}

			});
		}
		return file;
	}

	@Override
	public Nfile preview(final Requester requester, Map parameters) throws Exception {
		final T attachment = this.object(requester, parameters);
		if (attachment == null) {
			return null;
		} else if (attachment.getPreviewable() != Boolean.TRUE) {
			throw new RuntimeException("Attachment is not previewable:" + attachment);
		}
		return this.lookupOperator(attachment.getPath()).preview(attachment.getPath());
	}

}