fileupmgr.java

来自「一个用java编写的功能强大的OA系统」· Java 代码 · 共 82 行

package com.redmoon.oa.address;

import javax.servlet.http.HttpServletRequest;
import cn.js.fan.util.ErrMsgException;
import javax.servlet.ServletContext;
import com.redmoon.kit.util.FileUpload;
import java.io.IOException;
import cn.js.fan.web.Global;
import org.apache.log4j.Logger;
import java.util.*;
import com.redmoon.kit.*;
import com.redmoon.kit.util.FileInfo;
import java.io.File;


public class FileUpMgr {
    FileUpload fileUpload = null;
    public Logger logger;
    public FileUpMgr() {
    }
    public String uploadExcel(ServletContext application,
                         HttpServletRequest request) throws
           ErrMsgException {
       String excelFile = "";
       String upFile = "";
       doUpload(application,request);
       upFile = writeExcel(fileUpload);
       if (!upFile.equals(""))
           excelFile = Global.getRealPath() + upFile;
       else
           throw new ErrMsgException("文件不能为空！");
       return  excelFile;
   }

   public FileUpload doUpload(ServletContext application,
                               HttpServletRequest request) throws
            ErrMsgException {
        fileUpload = new FileUpload();
        fileUpload.setMaxFileSize(Global.FileSize); 
        String[] extnames = {"xls"};
        fileUpload.setValidExtname(extnames); 
        int ret = 0;
        try {
            ret = fileUpload.doUpload(application, request);
            if (ret != fileUpload.RET_SUCCESS) {
                throw new ErrMsgException(fileUpload.getErrMessage());
            }
        } catch (IOException e) {
            logger.error("doUpload:" + e.getMessage());
        }
        return fileUpload;
    }

    public String writeExcel(FileUpload fu) {
        if (fu.getRet() == fu.RET_SUCCESS) {
            Vector v = fu.getFiles();
            FileInfo fi = null;
            if (v.size() > 0)
                fi = (FileInfo) v.get(0);
            String vpath = "";
            if (fi != null) {
                
                Calendar cal = Calendar.getInstance();
                String year = "" + (cal.get(cal.YEAR));
                String month = "" + (cal.get(cal.MONTH) + 1);
                vpath = "upfile/" +
                        fi.getExt() + "/" + year + "/" + month + "/";
                String filepath = Global.getRealPath() + vpath;
                fu.setSavePath(filepath);
                
                fu.writeFile(true);

                
                
                
                return vpath + fi.getDiskName();
            }
        }
        return "";
    }
}