watermarkmgr.java~37~

云网论坛CWBBS 源码,内容丰富,学习,参考,教学的好资料,具体见内说明,

package com.redmoon.forum.ui;

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


public class WaterMarkMgr {
    FileUpload fileUpload = null;
    static Logger logger;
    public WaterMarkMgr() {
    }

    public FileUpload doUpload(ServletContext application,
                               HttpServletRequest request) throws
            ErrMsgException {
        fileUpload = new FileUpload();
        String[] extnames = {"gif"};
        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 void modify(ServletContext application,
                       HttpServletRequest request) throws ErrMsgException {
        doUpload(application, request);
        String filename = getFilename(fileUpload);
        if (!filename.equals("")) {
            delimg(Global.getRealPath() + "images/watermark.gif");
            //fileUpload.writeFile(true);
        } else {
            throw new ErrMsgException("文件名不能为空！");
        }
    }

    public void delimg(String src) throws ErrMsgException {
        try {
            File file = new File(src);
            file.delete();
        } catch (Exception e) {
            logger.error("ThemeMgr modify fail!");
        }
    }

    public String getFilename(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) {
                vpath = "images/";
                String filepath = Global.getRealPath() + vpath;
                fu.setSavePath(filepath);
                boolean re = fi.write(filepath, "watermark.gif");
                System.out.print("============" + re);
                return vpath + fi.getDiskName();
            }
        }
        return "";
    }
}