publishdo.java

一个bbs论坛系统

package com.lovo.servlet;

import java.io.File;
import java.io.IOException;
import java.sql.SQLException;
import java.util.Date;
import java.util.List;

import javax.servlet.ServletException;
import javax.servlet.http.*;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import com.lovo.factory.BOFactory;
import com.lovo.po.BlockPO;
import com.lovo.po.FilePO;
import com.lovo.po.LevelPO;
import com.lovo.po.PublishPO;
import com.lovo.po.ReplyPO;
import com.lovo.po.UserPO;

public class PublishDO extends HttpServlet {
	public void doGet(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		HttpSession session = request.getSession();
		request.setCharacterEncoding("utf-8");
		response.setContentType("text/html;charset=utf-8");
		response.setCharacterEncoding("utf-8");
		
		boolean isMultipart = ServletFileUpload.isMultipartContent(request);
		if (!isMultipart){
			return;
		}
		
//		 Create a factory for disk-based file items
		DiskFileItemFactory factory = new DiskFileItemFactory();
		//缓存大小
		factory.setSizeThreshold(1024);
		//临时目录
		factory.setRepository(new File("d:/temp"));
		
//		 Create a new file upload handler
		ServletFileUpload upload = new ServletFileUpload(factory);
		PublishPO publishPO = new PublishPO();
		UserPO publishMan = (UserPO)session.getAttribute("user");
		publishPO.setPublishMan(publishMan);
		publishPO.setId((int)(new Date().getTime()));
		int blockId = 0;
//		 Parse the request
		try {
			List<FileItem> items = upload.parseRequest(request);
			for(FileItem item:items){

				//普通表单域
				if(item.isFormField()){
					if(item.getFieldName().equals("title")) {
						publishPO.setTitle(item.getString());
					}
					if(item.getFieldName().equals("passage")) {
						publishPO.setContent(item.getString());
					}
					if(item.getFieldName().equals("blockId")) {
						blockId = Integer.parseInt(item.getString());
						BlockPO block = BOFactory.getBlockBOInstance().queryBlockById(blockId);
						publishPO.setBlock(block);
					}
					if(item.getFieldName().equals("score")) {
						if(item.getFieldName().matches("\\d+")) {
							int score = Integer.parseInt(item.getString());
							publishPO.setBounty(score);
						}
						
					}
				}else{
					FilePO file = new FilePO();
					//获取上传的文件信息，并存储文件
					String uploadFile = item.getName();
					if(uploadFile.length()==0) break;
					//文件名
					String fileName = uploadFile.substring(uploadFile.lastIndexOf("\\")+1);
					//文件存储路径
					String saveFile = "d:/"+new Date().getTime()+fileName;
					item.write(new File(saveFile));
					file.setName(new Date().getTime()+fileName);
					file.setUrl(saveFile);
					file.setPublish(publishPO);
					BOFactory.getFileBOInstance().insert(file);
				}
			}
			
		} catch (Exception e) {
			e.printStackTrace();
		}
		
		//调用业务逻辑
		
		boolean isPub = BOFactory.getPublishBOInstance().addPub(publishPO);
		if(isPub) {
			try {
				int weathForPub = BOFactory.getConfigBOInstance().queryVlaueByPurpose("发帖获得的金钱");
				int scoreForPub = BOFactory.getConfigBOInstance().queryVlaueByPurpose("每帖获得积分");
				int levelRiseBasePubNum = BOFactory.getConfigBOInstance().queryVlaueByPurpose("升级分数");
				
				publishMan.setWealth(publishMan.getWealth() + weathForPub);
				publishMan.setScore(publishMan.getScore() + scoreForPub);
				
				List<PublishPO> PublishList = BOFactory.getPublishBOInstance().queryPubByUserid(publishMan.getId());
				double levelNumDouble = PublishList.size() / levelRiseBasePubNum;
				int levelNum = (int)Math.sqrt(levelNumDouble);
				if(levelNum == 0) {
					levelNum = 1;
				}
				LevelPO level = BOFactory.getLevelBOInstance().queryLevelById(levelNum);
				publishMan.setLevel(level);
				boolean isModify = BOFactory.getUserBOInstance().modifyInfo(publishMan);
				if(!isModify) {
					session.setAttribute("errorInfo", "帖子发表后未能成功修改用户");
					response.sendRedirect("../web/error.jsp");
					return;
				}
			} catch (SQLException e) {
				session.setAttribute("errorInfo", "读取配置信息时出错");
				e.printStackTrace();
				response.sendRedirect("../web/error.jsp");
				return;
			}
		}
		response.sendRedirect("QueryBlockPublishDO?blockId="+blockId);
	}

	public void doPost(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		doGet(request, response);
	}

}