uploadaction.java

一个不错的bbs论坛系统．对初学者很有帮助

package com.yhbbs.upload.action;

import java.io.File;
import java.util.ArrayList;
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.apache.log4j.Logger;
import org.apache.struts.action.Action;
import org.apache.struts.action.ActionForm;
import org.apache.struts.action.ActionForward;
import org.apache.struts.action.ActionMapping;
import org.apache.struts.action.ActionMessage;
import org.apache.struts.action.ActionMessages;
import org.apache.struts.upload.FormFile;

import com.yhbbs.bbs.biz.BbsPropBiz;
import com.yhbbs.upload.biz.UpLoadBiz;
import com.yhbbs.upload.form.UpLoadForm;
import com.yhbbs.upload.itface.UpFile;
import com.yhbbs.user.itface.bean.UserSession;
import com.yhbbs.utils.Constants;
import com.yhbbs.utils.DateUtils;
import com.yhbbs.utils.ReqUtils;

/**
 * <p>Title:系统上传文件列表Action</p>
 * <li>因为该Action总是在帖子发表编辑Action之后调用的，</li>
 * <li>帖子Action用到了令牌，如果该Action也用令牌的话，帖子Action检查到的令牌将不一致。</li>
 * <li>故此处不用令牌机制处理</li>
 * <li>上传文件列表模块</li>
 * <br><b>WebSite: www.yyhweb.com</b>
 * <br><b>CopyRight: yyhweb[由由华网]</b>
 * @author stephen
 * @version YHBBS-2.0
 */
public class UpLoadAction extends Action{
	public ActionForward execute(ActionMapping mapping,ActionForm form,
			HttpServletRequest request,HttpServletResponse response){

		Logger bbslog = Logger.getLogger(UpLoadAction.class);
		ActionMessages errors = new ActionMessages();
		HttpSession session = request.getSession();
		UserSession bbsuser = (UserSession) session.getAttribute(Constants.bbsuser);
		int userId = 0;
		// 只有登录用户和已经审核用户才可以上传文件
		if(bbsuser==null || bbsuser.getUsertype()<2){
			errors.add("article.post.upload.nouser",new ActionMessage("article.post.upload.nouser"));
			saveErrors(request,errors);
			return mapping.findForward("Failure");
		}
		userId = bbsuser.getUserId();
		List attlist = null; 	  // 用来获得该文章的所有附件列表
		int allowsize = 6;  	 // 允许上传文件数
		int leaveNum = 0;  		// 还可以上传文件数
		
    	int upMaxSize = BbsPropBiz.getParameter().getUploadSize();  // 剩余空间（单位K）
    	int filesize = UpLoadBiz.getUserFileSize(userId)/1000;
    	int fleave = upMaxSize-filesize>0?upMaxSize-filesize:0;
    	request.setAttribute("fleave",fleave);
    	
        // 取得上传文件的两个列表 
		UpLoadForm upform = (UpLoadForm )form;
		boolean isUploadOk = false;
        boolean isDataOk = false;
        List uplist = null;   // 上传文件对应系统上传文件列表
        List formlist = null; // 上传文件列表
        int artId = 0;
        int fId = 0;
        String fileaction="";
        String action = "";
        String delFilePath = "";
        uplist = upform.getUplist();
        formlist = upform.getFormlist();
        action = upform.getAction();
        artId = upform.getArtId();
        // 帖子附件列表和剩余数
		attlist = getArtFiles(artId,userId);
    	leaveNum = allowsize-attlist.size();
    	
        fId = ReqUtils.getInt(request,"fId");
        delFilePath = ReqUtils.getString(request,"filePath");
        fileaction = ReqUtils.getString(request,"faction");
        String ymd = "";
        String filePath = "";
        String uploadDir = "";
        // 预设上传附件,设置令牌
        if(action!=null && (action.equals("insert")||action.equals("update"))){
        	request.setAttribute("allattmts", attlist);
    		request.setAttribute("curNum",leaveNum);
    		request.setAttribute("action","bbsupfile");//bbsupfile只是为了和其他action不冲突
    		request.setAttribute("artId",artId);
        	return mapping.findForward("Display");
        }
        // 删除已经上传的附件
        if(fileaction.equals("delete")){
        	uploadDir = request.getRealPath("")+"/webapp/upload/";
        	UpLoadBiz.delFile(uploadDir+delFilePath,fId);
        	artId = ReqUtils.getInt(request,"aId");
        	attlist = getArtFiles(artId,userId);
        	leaveNum = allowsize-attlist.size();
        	request.setAttribute("allattmts", attlist);
    		request.setAttribute("curNum",leaveNum);
    		request.setAttribute("action",action);
    		request.setAttribute("artId",artId);
    		return mapping.findForward("Display");
        }
       
        if(uplist!=null && uplist.size()>0){
	        //  创建路径
        	uploadDir = request.getRealPath("")+"/webapp/upload";
	        int year = DateUtils.getYear();
	        int month = DateUtils.getMonth();
	        int day = DateUtils.getDay();
	        ymd = year + "/" + month + "/" + day;
	        filePath = uploadDir + "/" + ymd;
	        File file = new File(filePath);
	        if(!file.exists())
	            file.mkdirs();
	        else
	        	file = null;
        }
        // 系统属性对上传文件限制
        int maxid = 0;
        long maxSize = 1024000l;
        String allow = "gif,jpg,png,bmp,swf,rar,zip,doc,txt";
        String ftime = DateUtils.getCurFormatDate(Constants.dateFS);
        UpFile curFile = null;
        // 循环处理上传的文件
        for(int i=0;i<uplist.size();i++){
        	curFile = (UpFile) uplist.get(i);	   // 取得一个上传文件
        	if(curFile!=null){
        		if(curFile.getFilesize()>maxSize || curFile.getFilesize()>fleave*1024){ // 如果该文件大小超过规定或剩余空间大小
        			if(!isUploadOk){
        				isUploadOk = true;
        			}
        			bbslog.warn("The upload file's size is bigger than maxsize!");
        			((FormFile)formlist.get(i)).destroy();
        		}else{
        			if(curFile.getFiletype().length()<=0 || curFile.getFiletype().indexOf(",")!=-1 || allow.indexOf(curFile.getFiletype())==-1){
        				if(!isUploadOk){
        					isUploadOk = true;
        				}
        				bbslog.warn("The Upload file type not right!");
        				((FormFile)formlist.get(i)).destroy();
        			}else{						  // 设置该上传文件对应系统文件的属性.
        				maxid = (UpLoadBiz.getMaxId()+1);
        				curFile.setArtid(artId);
        				curFile.setAuthor(userId);
        				curFile.setFilename(userId+"_"+maxid+"."+curFile.getFiletype());
        				curFile.setPath(ymd+"/"+curFile.getFilename());
        				curFile.setPtime(ftime);
        				// 将上传文件写到系统目录下
        				if(UpLoadBiz.writeFile((FormFile)formlist.get(i),filePath+"/"+curFile.getFilename(),curFile,1)){
        					if(!isDataOk)
        						isDataOk = true;
        				}
        			}
        		}
        	}
        }
        
        if(isDataOk){
			 uplist.clear();
			 uplist = null;
        }
		 if(isUploadOk){
            errors.add("upload.file.fail", new ActionMessage("upload.file.fail"));
            saveErrors(request, errors);
        }
		// list和num每次操作结束后都会发生变化，所以在跳转前要重新计算
		attlist = getArtFiles(artId,userId);
	    leaveNum = allowsize-attlist.size();
    	request.setAttribute("allattmts", attlist);
		request.setAttribute("curNum",leaveNum);
		request.setAttribute("action",action);
		request.setAttribute("artId",artId);
		return mapping.findForward("Display");
						
	}
	/**
	 * 根据帖子Id取得该帖子所有附件；同时根据用户Id判断该上传文件是否为该用户上传
	 * @param artId 帖子Id
	 * @param userId 用户Id
	 * @return 上传文件列表
	 */
	public List getArtFiles(int artId,int userId){
		List attlist = null;
		attlist = UpLoadBiz.getArtFile(artId,userId);
		if(attlist==null){
			attlist = new ArrayList();
		}
		return attlist;
	}
}