dbuser.java

来自「本系统实现了网上书店功能」· Java 代码 · 共 87 行

package com.util;

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

import javax.sql.DataSource;

import com.bean.User;

public class DBUser {
	private DataSource ds;

	public DBUser(DataSource dataSource) {
		this.ds = dataSource;
	}

	public boolean checkUser(String username, String password)throws SQLException {
               Connection con = ds.getConnection();
               String presql = "select * from user where username=? and password=?";
               PreparedStatement pstmt = con.prepareStatement(presql);
               pstmt.setString(1, username);
               pstmt.setString(2, password);
               ResultSet rs = pstmt.executeQuery();
               boolean flag = rs.next();
               con.close();
               if(flag)
	            return true;
                else 
	               return false; 
}
	public boolean validateUser(User user) throws SQLException
	{
		Connection con=ds.getConnection();
		String presql="select * from user where username=?";
		PreparedStatement pstmt = con.prepareStatement(presql);
		pstmt.setString(1, user.getUsername());
		  ResultSet rs = pstmt.executeQuery();
          boolean flag = rs.next();
          con.close();
          if(flag)
           return true;
           else 
              return false; 
	}
	
	public int save(User user) throws SQLException {
		Connection con = ds.getConnection();
		String presql = "insert into user(username,password,name,email,phone,address) values(?,?,?,?,?,?)";
		PreparedStatement pstmt = con.prepareStatement(presql);
		pstmt.setString(1,user.getUsername());
		pstmt.setString(2, user.getPassword());
		pstmt.setString(3, user.getName());
		pstmt.setString(4, user.getEmail());
		pstmt.setString(5, user.getPhone());
		pstmt.setString(6, user.getAddress());
		int flag = pstmt.executeUpdate();
		con.close();
		return flag;
	}
	public User getUserInfo(String username) throws SQLException{
		Connection con  = ds.getConnection();
		String presql="select * from user where username=?";
		PreparedStatement pstmt = con.prepareStatement(presql);
		pstmt.setString(1,username);
		ResultSet rs = pstmt.executeQuery();
		if(rs.next()){
			String username_sql = rs.getString("username");
			String password_sql=rs.getString("password");
			String name_sql=rs.getString("name");
			String address_sql=rs.getString("address");
			String phone_sql=rs.getString("phone");
			String email_sql=rs.getString("email");
			User userbean = new User();
			userbean.setUsername(username_sql);
			userbean.setPassword(password_sql);
			userbean.setAddress(address_sql);
			userbean.setName(name_sql);
			userbean.setPhone(phone_sql);
			userbean.setEmail(email_sql);
			return  userbean;
		}
		return null;
	}
}