userdao.java

来自「教你一天学会struts」· Java 代码 · 共 81 行

package com.soft136.strutsoneday.dao;

import java.sql.*;
import javax.sql.rowset.CachedRowSet;
import com.sun.rowset.CachedRowSetImpl;

public class UserDao {

	public CachedRowSet login(String username, String password)
			throws SQLException, ClassNotFoundException {
		Connection conn = null;
		PreparedStatement ps = null;
		ResultSet rs = null;
		try {
			Class.forName("com.mysql.jdbc.Driver");
			conn = DriverManager.getConnection(
					"jdbc:mysql://localhost:3306/soft136", "root", "123456");
			ps = conn
					.prepareStatement("SELECT ID,USERNAME,PWD,REALNAME,PHONE,ADDRESS,GENDER,AGE,BUSINESSTYPE FROM USER WHERE USERNAME=? AND PWD=?");
			ps.setString(1, username);
			ps.setString(2, password);
			rs = ps.executeQuery();
			CachedRowSet crs = new CachedRowSetImpl();
			crs.populate(rs);
			return crs;
		} finally {
			if (ps != null) {
				ps.close();
			}
			if (conn != null) {
				conn.close();
			}
		}
	}

	public void add(String username, String password) throws Exception {
		Connection conn = null;
		PreparedStatement ps = null;
		try {
			Class.forName("com.mysql.jdbc.Driver");
			conn = DriverManager.getConnection(
					"jdbc:mysql://localhost:3306/soft136", "root", "123456");
			ps = conn
					.prepareStatement("INSERT INTO USER (USERNAME,PWD) VALUES(?,?)");
			ps.setString(1, username);
			ps.setString(2, password);
			ps.executeUpdate();
		} finally {
			if (ps != null) {
				ps.close();
			}
			if (conn != null) {
				conn.close();
			}
		}
	}

	public void updateUserBusinessType(int userId, String businessType)
			throws SQLException, ClassNotFoundException {
		Connection conn = null;
		PreparedStatement ps = null;
		try {
			Class.forName("com.mysql.jdbc.Driver");
			conn = DriverManager.getConnection(
					"jdbc:mysql://localhost:3306/soft136", "root", "123456");
			ps = conn
					.prepareStatement("UPDATE USER SET BUSINESSTYPE=? WHERE ID=?");
			ps.setString(1, businessType);
			ps.setInt(2, userId);
			ps.executeUpdate();
		} finally {
			if (ps != null) {
				ps.close();
			}
			if (conn != null) {
				conn.close();
			}
		}
	}
}