public class UploadFileServlet extends HttpServlet {

// \attached\

private static String baseFileDir = File.separator + "attached"

+ File.separator;// 上传文件存储目录

// /attached/

private static String baseURLDir = "/attached/";// 上传文件目录URL

// 可以上传的文件类型

private static String fileExt = "jpg,jpeg,bmp,gif,png";

private static Long maxSize = 0l;

// 0:不建目录 1:按天存入目录 2:按月存入目录 3:按扩展名存目录 建议使用按天存

private static String dirType = "1";

/**

* 文件上传初始化工作

*/

public void init() throws ServletException {

/* 获取文件上传存储的相对路径 */

// 如果配置文件中的baseDir不为空，就以配置文件为主，否则使用上文定义的地址

if (!StringUtils.isBlank(this.getInitParameter("baseDir"))) {

baseFileDir = this.getInitParameter("baseDir");

}

// 拿到baseFileDir的绝对路径

// E:\workspace\sts\.metadata\.plugins\org.eclipse.wst.server.core\tmp0\wtpwebapps\HTMLEditor\attached

String realBaseDir = this.getServletConfig().getServletContext()

.getRealPath(baseFileDir);

File baseFile = new File(realBaseDir);

// 如果工程中不存在attached文件夹，则创建

if (!baseFile.exists()) {

baseFile.mkdir();

}

/* 获取文件类型参数 */

fileExt = this.getInitParameter("fileExt");

if (StringUtils.isBlank(fileExt))

fileExt = "jpg,jpeg,gif,bmp,png";

/* 获取文件大小参数 */

String maxSize_str = this.getInitParameter("maxSize");

if (StringUtils.isNotBlank(maxSize_str))

maxSize = new Long(maxSize_str);

/* 获取文件目录类型参数 */

dirType = this.getInitParameter("dirType");

// 如果web.xml中未配置dirType，缺省值为1

if (StringUtils.isBlank(dirType))

dirType = "1";

// 如果web.xml中配置了非法的dirType，修改其为1

if (",0,1,2,3,".indexOf("," + dirType + ",") < 0)

dirType = "1";

}

/**

* 上传文件数据处理过程

*/

@SuppressWarnings({ "unchecked" })

public void doPost(HttpServletRequest request, HttpServletResponse response)

throws ServletException, IOException {

response.setContentType("text/html; charset=UTF-8");

response.setHeader("Cache-Control", "no-cache");

String err = "";

String newFileName = "";

DiskFileUpload upload = new DiskFileUpload();

try {

List items = upload.parseRequest(request);

Map fields = new HashMap();

Iterator iter = items.iterator();

while (iter.hasNext()) {

FileItem item = (FileItem) iter.next();

/**

* isFormField方法用来判断FileItem对象里面封装的数据是一个普通文本表单字段，

* 还是一个文件表单字段。如果是普通文本表单字段，返回一个true否则返回一个false。

* 因此可以用该方法判断是否是普通表单域还是文件上传表单域。

*/

// item.getFieldName()值为filedata

if (item.isFormField()) {

fields.put(item.getFieldName(), item.getString());

} else {

fields.put(item.getFieldName(), item);

}

}

/* 获取表单的上传文件 */

FileItem uploadFile = (FileItem) fields.get("filedata");

//获取上传文件名称 如：Image 001.jpg

String fileNameLong = uploadFile.getName();

/* 获取文件扩展名 */

/* 索引加1的效果是只取xxx.jpg的jpg */

String extensionName = fileNameLong.substring(fileNameLong

.lastIndexOf(".") + 1);

/* 检查文件类型 */

// 返回指定字符在此字符串中第一次出现处的索引。如果没有找到，则返回-1

if (("," + fileExt.toLowerCase() + ",").indexOf(","

+ extensionName.toLowerCase() + ",") < 0) {

printInfo(response, "不允许上传此类型的文件", "");

return;

}

/* 文件是否为空 */

if (uploadFile.getSize() == 0) {

printInfo(response, "上传文件不能为空", "");

return;

}

/* 检查文件大小 */

if (maxSize > 0 && uploadFile.getSize() > maxSize) {

printInfo(response, "上传文件的大小超出限制", "");

return;

}

// 0:不建目录, 1:按天存入目录, 2:按月存入目录, 3:按扩展名存目录.建议使用按天存.

String fileFolder = "";

if (dirType.equalsIgnoreCase("1"))

fileFolder = new SimpleDateFormat("yyyyMMdd")

.format(new Date());

if (dirType.equalsIgnoreCase("2"))

fileFolder = new SimpleDateFormat("yyyyMM").format(new Date());

if (dirType.equalsIgnoreCase("3"))

fileFolder = extensionName.toLowerCase();

/* 文件存储的相对路径 */

String saveDirPath = baseFileDir + fileFolder + "/";

/* 文件存储在容器中的绝对路径 */

String saveFilePath = this.getServletConfig().getServletContext()

.getRealPath("")

+ baseFileDir + fileFolder + "/";

/* 构建文件目录以及目录文件 */

File fileDir = new File(saveFilePath);

if (!fileDir.exists()) {

fileDir.mkdirs();

}

/* 重命名文件 */

String filename = UUID.randomUUID().toString();

File savefile = new File(saveFilePath + filename + "."

+ extensionName);

/* 存储上传文件 */

uploadFile.write(savefile);

// http://localhost:80/HTMLEditor

String projectBasePath = request.getScheme() + "://"

+ request.getServerName() + ":" + request.getServerPort()

+ request.getContextPath();

// 上传文件URL回调

// newFileName =projectBasePath + baseURLDir + fileFolder + "/" +

// filename + "." + extensionName;

// 加上！表示为立即上传模式，不加！上传成功之后需要点击确定按钮才会在编辑器中显示

newFileName = "!" + projectBasePath + baseURLDir + fileFolder + "/"

+ filename + "." + extensionName;

System.out.println(newFileName);

} catch (Exception ex) {

newFileName = "";

err = "错误: " + ex.getMessage();

}

printInfo(response, err, newFileName);

}

/**

* 使用I/O流输出 json格式的数据

*

*@param response

*@param err

*@param newFileName

*@throws IOException

*/

public void printInfo(HttpServletResponse response, String err,

String newFileName) throws IOException {

PrintWriter out = response.getWriter();

// String filename = newFileName.substring(newFileName.lastIndexOf("/")

// + 1);

out.println("{\"err\":\"" + err + "\",\"msg\":\"" + newFileName + "\"}");

out.flush();

out.close();

}

}