Wizzer's Blog

upload.json 配置文件，增加扩展配置项extOption，内容可以由用户自定义，返回一个Map对象。

我要实现功能是，用户在相册里只允许上传图片文件、视频里只允许上传视频格式的文件等，

结合上传控件和JS，实现在页面级和代码级的文件类型分类限制、分类保存等功能。

Nutz issue 提交需求不被接纳所以只好自己实现了，大家如需要的话可以提issue，希望以后Nutz能内置：

https://github.com/nutzam/nutz/issues/568

upload.json：

var ioc = {
		upload : {
			type : "org.nutz.mvc.upload.UploadAdaptor",
			args : [{refer : "uploadCtx"}]
		},
		uploadCtx : {
			type : "org.nutz.mvc.upload.UploadingContext",
			args : [{refer: "filePool"}],
			fields : {
				ignoreNull : true,
				maxFileSize : 10485760,
				nameFilter : ".+(jpg|gif|png|jpeg|doc|docx|xls|xlsx|ppt|pptx|wps|pdf|txt|chm|mp3|mp4|3gp|rm|swf|flv|asf|wmv|wma|z|zip|rar|ios|jar)",
				extOption: {
					"images":"jpg,gif,png,jpeg",
					"document":"doc,docx,xls,xlsx,ppt,pptx,wps,pdf,txt,chm",
				    "music":"mp3",
				    "video":"mp4,3gp,rm,swf,flv,asf,wmv,wma",
				    "archive":"z,zip,rar,ios,jar"			            
				   }
			}
		},
		filePool : {
			type : "com.hits.common.file.FilePool",
			args : ["/temp/", 2000]
		}
};

改写Nutz源码：

org.nutz.mvc.upload.UploadingContext 类，增加代码：

    /**
     * 一个扩展配置项，用户可自定义内容
     */
    private Map<String, String> extOption;

    public Map<String, String> getExtOption() {
        return extOption;
    }

    public UploadingContext setExtOption(String extOption) {
        this.extOption = Json.fromJsonAsMap(String.class,extOption);
        return this;
    }

增加一个调用类：

FileType.java

/** 
 * 类描述： 
 * 创建人：Wizzer 
 * 联系方式：www.wizzer.cn
 * 创建时间：2013-12-25 下午1:13:30 
 * @version 
 */
public class FileType { 
	/**
	 * 根据文件名获取文件类型，默认返回other
	 * @param upload
	 * @param suffixname
	 * @return
	 */
	public static String getFileType(UploadAdaptor upload,String suffixname) {
		Map<String,String> hm=upload.getContext().getExtOption();
		String str = StringUtil.null2String(suffixname).toLowerCase();
		Set<Map.Entry<String, String>> set = hm.entrySet();
        for (Iterator<Map.Entry<String, String>> it = set.iterator(); it.hasNext();) {
            Map.Entry<String, String> entry = (Map.Entry<String, String>) it.next();
            if(entry.getValue().toLowerCase().indexOf(str)>-1)
            	return entry.getKey();
        }
		return "other";
	}
	/**
	 * 根据文件类型获取文件后缀名，默认返回nameFilter配置
	 * @param upload
	 * @param filetype
	 * @return
	 */
	public static String getSuffixname(UploadAdaptor upload,String filetype) {
		Map<String,String> hm=upload.getContext().getExtOption();
		String str = StringUtil.null2String(filetype).toLowerCase();
		if(!"".equals(str)){
			Set<Map.Entry<String, String>> set = hm.entrySet();
	        for (Iterator<Map.Entry<String, String>> it = set.iterator(); it.hasNext();) {
	            Map.Entry<String, String> entry = (Map.Entry<String, String>) it.next();
	            if(entry.getKey().toLowerCase().equals(str))
	            	return entry.getValue();
	        }
		}
        String name=upload.getContext().getNameFilter(); 
		return name.substring(name.indexOf("(")+1,name.lastIndexOf(")")).replace("|", ",");
	}
}

使用示例：

	@Inject
	protected UploadAdaptor upload;

	@At
	@Ok("->:/private/file/uploadOne.html")
	public void uploadOne(@Param("filetype") String filetype,HttpServletRequest req){
		req.setAttribute("filetype", filetype);
		req.setAttribute("allowExtensions", FileType.getSuffixname(upload, filetype));

	}

如上所述，在限制文件上传类型的同时，还可以将文件分类保存在不同类型的目录下。