package dao;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.util.ArrayList;
import java.util.List;

import util.DBUtil;
import entity.User;

public class UserDAO {

	// 1.添加用户数据
	public int save(User user) throws Exception {
		int id = -1;
		Connection conn = DBUtil.getConnection();
		PreparedStatement prep = conn.prepareStatement(
				"insert into f_user values(null,?,?,?,?,?,?,?)",
				java.sql.Statement.RETURN_GENERATED_KEYS);
		prep.setString(1, user.getUsername());
		prep.setString(2, user.getName());
		prep.setString(3, user.getPwd());
		prep.setInt(4, user.getAge());
		prep.setString(5, user.getGender());
		prep.setString(6, user.getPhone());
		prep.setString(7, user.getAsk());
		prep.executeUpdate();
		ResultSet rs = prep.getGeneratedKeys();
		if (rs.next()) {
			// rs.getInt("id"); ---error!
			id = rs.getInt(1);
		}
		DBUtil.close(conn);
		return id;
	}

	// 2.查找用户(username)
	public User findByUsername(String username) throws Exception {
		User user = null;
		Connection conn = DBUtil.getConnection();
		PreparedStatement prep = conn
				.prepareStatement("select*from f_user where username=?");
		prep.setString(1, username);
		ResultSet rs = prep.executeQuery();
		if (rs.next()) {
			user = new User();
			user.setId(rs.getInt("id"));
			user.setUsername(username);
			user.setName(rs.getString("name"));
			user.setPwd(rs.getString("pwd"));
			user.setAge(rs.getInt("age"));
			user.setGender(rs.getString("gender"));
			user.setPhone(rs.getString("phone"));
			user.setAsk(rs.getString("ask"));
		}
		DBUtil.close(conn);
		return user;
	}

	// 3.查找显示用户列表
	public List<User> findAll() throws Exception {
		List<User> users = new ArrayList<User>();
		Connection conn = DBUtil.getConnection();
		PreparedStatement prep = conn.prepareStatement("select*from f_user");
		ResultSet rs = prep.executeQuery();
		while (rs.next()) {
			User user = new User();
			user.setId(rs.getInt("id"));
			user.setUsername(rs.getString("username"));
			user.setName(rs.getString("name"));
			user.setPwd(rs.getString("pwd"));
			user.setAge(rs.getInt("age"));
			user.setGender(rs.getString("gender"));
			user.setPhone(rs.getString("phone"));
			user.setAsk(rs.getString("ask"));
			users.add(user);
		}
		DBUtil.close(conn);
		return users;
	}

	// 4.查找用户信息(ID)
	public User findById(int id) throws Exception {
		User user = null;
		Connection conn = DBUtil.getConnection();
		PreparedStatement prep = conn
				.prepareStatement("select*from f_user where id=?");
		prep.setInt(1, id);
		ResultSet rs = prep.executeQuery();
		if (rs.next()) {
			user = new User();
			user.setId(id);
			user.setUsername(rs.getString("username"));
			user.setName(rs.getString("name"));
			user.setPwd(rs.getString("pwd"));
			user.setAge(rs.getInt("age"));
			user.setGender(rs.getString("gender"));
			user.setPhone(rs.getString("phone"));
			user.setAsk(rs.getString("ask"));
		}
		DBUtil.close(conn);
		return user;
	}

}