cmsmultifileuploadbean.java

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

package com.redmoon.oa.netdisk;

import cn.js.fan.util.ErrMsgException;
import java.util.Vector;
import java.util.Iterator;
import com.redmoon.kit.util.FileInfo;
import com.redmoon.kit.util.FileUpload;

public class CMSMultiFileUploadBean extends FileUpload {
    Vector attachments = new Vector();
    boolean isAttachmentsSeparated = false;

    public CMSMultiFileUploadBean() {
    }

    
    public void separateAttachments() {
        Iterator ir = files.iterator();

        while (ir.hasNext()) {
            FileInfo fi = (FileInfo) ir.next();
            if (fi.fieldName.indexOf("attachment") != -1) {
                
                
                attachments.addElement(fi);
            }
        }

        ir = attachments.iterator();
        while (ir.hasNext()) {
            FileInfo fi = (FileInfo) ir.next();
            files.remove(fi);
        }

        isAttachmentsSeparated = true;
    }

    public Vector getAttachments() {
        if (!isAttachmentsSeparated)
            separateAttachments();
        return attachments;
    }

    public Vector getFiles() {
        if (!isAttachmentsSeparated)
            separateAttachments();
        return files;
    }

    
    public void writeFile(boolean isRandName) {
        Vector files = getFiles();
        int size = files.size();
        if (size == 0)
            return;
        java.util.Enumeration e = files.elements();
        while (e.hasMoreElements()) {
            FileInfo fi = (FileInfo) e.nextElement();
            if (!isRandName)
                fi.write(savePath, "");
            else 
                fi.write(savePath, getRandName());
        }
    }

    public void writeAttachment(boolean isRandName) throws ErrMsgException {
        Vector files = getAttachments();
        int size = files.size();
        if (size == 0)
            return;
        java.util.Enumeration e = files.elements();
        while (e.hasMoreElements()) {
            FileInfo fi = (FileInfo) e.nextElement();
            if (!isRandName)
                fi.write(savePath, "");
            else { 
                String newfilename = getRandName();
                fi.write(savePath,
                             newfilename + "." + fi.getExt());
            }
        }
    }
}