微社区项目开发笔记(后端篇)

　　废话不说，先来张包结构的图片：

     　　

　　　很明显，采用了经典MVC三层架构。除了架构所需的包以外，还添加了一个工具包tool，里面有MD5加密和生成验证码的类，具体代码如下：

　　　　

package com.jiy.JcMiniSNS.tool;

import java.security.MessageDigest;

/**
 * MD5工具类
 * @author 来自网络
 *
 */
public class MD5Tool {
    
    //十六进制下数字到字符的映射数组    
    private final static String[] hexDigits = {"0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "a", "b", "c", "d", "e", "f"};   
      
    /**  对字符串进行MD5加密     */    
    private static String encodeByMD5(String originString){    
        if (originString != null){    
            try{    
                //创建具有指定算法名称的信息摘要    
                MessageDigest md = MessageDigest.getInstance("MD5");    
                //使用指定的字节数组对摘要进行最后更新，然后完成摘要计算    
                byte[] results = md.digest(originString.getBytes());    
                //将得到的字节数组变成字符串返回    
                String resultString = byteArrayToHexString(results);    
                //return resultString.toUpperCase();    
                return resultString;    
            } catch(Exception ex){    
                ex.printStackTrace();    
            }    
        }    
        return null;    
    }   
      
    /** * 把inputString加密 */  
    public static String generatePassword(String inputString) {  
        return encodeByMD5(inputString);  
    }  
  
    /** 
     * 验证输入的密码是否正确 
     * @param password 加密后的密码 
     * @param inputString 输入的字符串 
     * @return 验证结果，TRUE:正确 FALSE:错误 
     */  
    public static boolean validatePassword(String password, String inputString) {  
        if (password.equals(encodeByMD5(inputString))) {  
            return true;  
        } else {  
            return false;  
        }  
    }  
      
    /**   
     * 转换字节数组为十六进制字符串  
     * @param     字节数组  
     * @return    十六进制字符串  
     */    
    private static String byteArrayToHexString(byte[] b){    
        StringBuffer resultSb = new StringBuffer();    
        for (int i = 0; i < b.length; i++){    
            resultSb.append(byteToHexString(b[i]));    
        }    
        return resultSb.toString();    
    }    
        
    /** 将一个字节转化成十六进制形式的字符串     */    
    private static String byteToHexString(byte b){    
        int n = b;    
        if (n < 0)    
            n = 256 + n;    
        int d1 = n / 16;    
        int d2 = n % 16;    
        return hexDigits[d1] + hexDigits[d2];    
    }    
}

package com.jiy.JcMiniSNS.tool;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.Random;

import javax.imageio.ImageIO;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * 验证码生成类
 * @author jiy
 *
 */
public class VerificationCode {
    
    private static final int WIDTH=60;
    private static final int HEIGHT=30;
    
    private BufferedImage img;
    private Graphics g;
    
    public VerificationCode(HttpServletRequest request,HttpServletResponse response) throws IOException
    {
        init();
        drawCode(request);
        drawLine();
        ImageIO.write(img, "gif", response.getOutputStream());
    }
    
    private void init()
    {
        img = new BufferedImage(WIDTH, HEIGHT, BufferedImage.TYPE_INT_RGB);
        g  = img.getGraphics();
        g.setColor(Color.WHITE);
        g.fillRect(0, 0, WIDTH, HEIGHT);
        g.setColor(Color.RED);
        g.drawRect(0, 0, WIDTH-1, HEIGHT-1);
    }
    
    /**
     * 画干扰线条
     */
    private void drawLine()
    {
        g.setColor(Color.BLUE);
        Random r = new Random();
        for(int i=0; i<3; i++){
            g.drawLine(r.nextInt(WIDTH), r.nextInt(HEIGHT), r.nextInt(WIDTH), r.nextInt(HEIGHT));
        }
    }
    
    /**
     * 画验证码并存入session
     * @param request
     */
    private void drawCode(HttpServletRequest request)
    {
        Random r = new Random();
        String checkCode = r.nextInt(9000)+1000+"";//1000-9999
        g.setColor(Color.red);
        g.setFont(new Font("微软雅黑",2,16));
        g.drawString(checkCode, 10, 20);
        //在Session中添加属性"checkCode"=验证码
        request.getSession().setAttribute("checkCode", checkCode);
        //System.out.println(checkCode);
    }
}

　　　在程序开发的过程中，我发现了以前经常用的BaseDao的不合理的地方：为了更好地封装数据库操作，大家通常会在BaseDao里面写一个叫做getQuery(String sql,Object...paras)的方法，并且把常用的Connection、PreparedStatement、ResultSet作为BaseDao的成员，然后让dao层的代码继承BaseDao。这样的做法明显提高了dao层代码的开发效率，但我发现在实际运用的时候出现了问题：

/**
     * 执行查询操作
     * @param sql sql语句
     * @param paras 参数数组
     * @return 查询结果的ResultSet对象
     */
    public ResultSet getQuery(String sql,Object...paras)
    {
        try {
            pstmt = conn.prepareStatement(sql);
            if(paras != null && paras.length != 0)
            {                
                for(int i = 0;i < paras.length;i++)
                {
                    pstmt.setObject(i+1, paras[i]);
                }
            }
            res = pstmt.executeQuery();
        } catch (SQLException e) {
            throw new RuntimeException("执行查询失败！"+e.getMessage());
        }
        return res;
    }

　　　　如果在动态消息管理的实现类里面有一个找动态的方法叫做findDynamics()调用了getQuery(String sql,Object...paras)方法，同时在同一个类中也有一个叫做findTells()用于寻找相关动态的评论的方法调用了getQuery(String sql,Object...paras)这个方法，在findDynamics()遍历所查询到的ResultSet用于初始化相关对象的时候调用了findTells()方法，findTells()方法在调用getQuery(String sql,Object...paras)的过程中执行到了 res = pstmt.executeQuery(); 这行代码，由于是在同一个类中，所以他们的res对象是共享的，于是在findDynamics()遍历res对象第一遍的时候，res已经被重新赋值了。这样就导致无论怎么样只能得到一条dynamics的记录。

　　　　所以，getQuery(String sql,Object...paras)方法中最好不要使用成员变量中的结果集。改进后完整的BaseDao类如下：

package com.jiy.JcMiniSNS.dao;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;


/**
 * JDBC连接基本类
 * @author jiy
 *
 */
public class BaseDao {
    
    
//    private static final String DRIVER = "com.microsoft.sqlserver.jdbc.SQLServerDriver";
//    private static final String URL = "jdbc:sqlserver://localhost:1433;DatabaseName=db_miniSNS";
    
    private static final String DRIVER = "com.mysql.jdbc.Driver";
    private static final String URL = "jdbc:mysql://localhost:3306/db_minisns";
    private static final String USER = "root";
    private static final String PWD = "root";
    
    public Connection conn = null;
    public PreparedStatement pstmt = null;
    public ResultSet res = null;
    
    static{
        try {
            Class.forName(DRIVER);
        } catch (ClassNotFoundException e) {
            throw new RuntimeException("数据库驱动加载失败！");
        }
    }
    
    /**
     * 获取jdbc连接
     */
    public void getConnection()
    {
        try {
            conn = DriverManager.getConnection(URL,USER,PWD);
            
        } catch (SQLException e) {
            throw new RuntimeException("获取数据库连接失败！" + e.getMessage());
        }
    }
    
    /**
     * 执行查询操作
     * @param sql sql语句
     * @param paras 参数数组
     * @return 查询结果的ResultSet对象
     */
    public ResultSet getQuery(String sql,Object...paras)
    {
        ResultSet queryRes = null;
        try {
            pstmt = conn.prepareStatement(sql);
            if(paras != null && paras.length != 0)
            {                
                for(int i = 0;i < paras.length;i++)
                {
                    pstmt.setObject(i+1, paras[i]);
                }
            }
            queryRes = pstmt.executeQuery();
        } catch (SQLException e) {
            throw new RuntimeException("执行查询失败！"+e.getMessage());
        }
        return queryRes;
    }
    
    
    /**
     * 执行更新操作
     * @param sql sql语句
     * @param paras 参数列表
     * @return 受影响的行数
     */
    public int getUpdate(String sql,Object...paras)
    {
        int result = 0;
        try {
            pstmt = conn.prepareStatement(sql);
            if(paras != null && paras.length != 0)
            {                
                for(int i = 0;i < paras.length;i++)
                {
                    pstmt.setObject(i+1, paras[i]);
                }
            }
            result = pstmt.executeUpdate();
        } catch (SQLException e) {
            throw new RuntimeException("执行更新失败！" +e.getMessage());
        }
        return result;
    }
    
    /**
     * 关闭资源
     * 关闭ResultSet、PreparedStatement、Connection对象
     */
    public void closeAll()
    {
        if(res != null)
        {
            try {
                res.close();
            } catch (SQLException e) {
                throw new RuntimeException("关闭ResultSet失败！");
            }
        }
        if(pstmt != null)
        {
            try {
                pstmt.close();
            } catch (SQLException e) {
                throw new RuntimeException("关闭PreparedStatement失败！");
            }
        }
        if(conn != null)
        {
            try {
                conn.close();
            } catch (SQLException e) {
                throw new RuntimeException("关闭Connection失败！");
            }
        }
    }
}

　　　　

 　　　　在访问网站首页的时候，需要加载数据库的数据怎么办呢？除了使用ajax，还可以在jsp里面加入如下代码：

　　　　

<%
    try{
        if(request.getAttribute("dynamicList") == null)
            request.getRequestDispatcher("initDynamicForIndexServlet").forward(request, response);
    }catch(Exception e){
        
    }
%>

　　　　

　　到此，整个项目的笔记就结束了，由于刚开始写博客所以写的不是很好，请见谅！本人QQ:2632790902,欢迎指教不足之处！