browmgr.java

cwbbs 云网论坛源码

package com.redmoon.forum.ui;

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.util.FileInfo;
import cn.js.fan.util.StrUtil;
import cn.js.fan.cache.jcs.RMCache;
import com.cloudwebsoft.framework.util.LogUtil;


public class BrowMgr {
    public FileUpload fileUpload = null;
    public Logger logger;

    public static final String BROWPATH = "forum/images/brow/";

    public static final String cacheKey = "forum.ui.brow";

    public BrowMgr() {
    }

    public static String[] getBrows() {
        String[] files = null;
        try {
            files = (String[])RMCache.getInstance().get(cacheKey);
        }
        catch (Exception e) {
            LogUtil.getLog(BrowMgr.class).error(e.getMessage());
        }
        if (files==null) {
            FileViewer fileViewer = new FileViewer(Global.realPath +
                    BROWPATH);
            fileViewer.init();
            files = new String[fileViewer.files.size()];
            int i = 0;
            while (fileViewer.nextFile()) {
                String fileName = fileViewer.getFileName();
                int p = fileName.lastIndexOf(".");
                files[i] = fileName.substring(0, p);
                i++;
            }
            try {
                RMCache.getInstance().put(cacheKey, files);
            }
            catch (Exception e) {
                LogUtil.getLog(BrowMgr.class).error(e.getMessage());
            }
        }
        return files;
    }

    public void uploadImg(ServletContext application,
                            HttpServletRequest request) throws
            ErrMsgException {
        doUpload(application, request);
        String upFile = writeBrow(fileUpload);
        if (upFile.equals("")) {
            throw new ErrMsgException("File can't be null！");
        }
    }

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

    public String writeBrow(FileUpload fu) {
        if (fu.getRet() == fu.RET_SUCCESS) {
            Vector v = fu.getFiles();
            FileInfo fi = null;
            if (v.size() > 0) {
                fi = (FileInfo) v.get(0);
            }
            if (fi != null) {
                String group = fu.getFieldValue("group");
                String filepath = Global.getRealPath() + BROWPATH;
                fu.setSavePath(filepath);
                fi.write(filepath, getNextBrowName(group) + "." + fi.getExt());
                refresh();
                return BROWPATH + fi.getDiskName();
            }
        }
        return "";
    }

    public void refresh() {
        try {
            RMCache.getInstance().remove(cacheKey);
        } catch (Exception e) {
            logger.error("refresh:" + e.getMessage());
        }
    }

    public int getNextBrowName(String group) {
        int max = 0;
        String filepath = Global.getRealPath() + BROWPATH;
        String currentName = "";
        FileViewer fv = new FileViewer(filepath);
        fv.init();
        while (fv.nextFile()) {
            currentName = fv.getFileName();
            int p = 0;
            int m = currentName.indexOf(".");
            String f = currentName.substring(0, m);
            if (StrUtil.isNumeric(f)) {
                p = Integer.parseInt(f);
                if (p > max) {
                    max = p;
                }
            }
        }
        return max+1;
    }
}