/**
 * 
 */
package com.permission.dao.impl;

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

import com.permission.dao.IUserAndRoleDAO;
import com.permission.sys.BaseDAO;
import com.permission.sys.PageRoll;
import com.permission.utils.common.PermissionException;
import com.permission.utils.db.DBUtil;
import com.permission.vo.UserAndRole;

/**
 * @author Jack
 * 
 */
public class UserAndRoleDAOImpl extends BaseDAO implements IUserAndRoleDAO {

	@Override
	public void add(UserAndRole userAndRole) throws PermissionException {
		Connection conn = null;
		PreparedStatement ps = null;
		String sql = "insert into UserAndRole(UserID,RoleID) values (?,?)";
		try {
			conn = DBUtil.getConnection();
			ps = conn.prepareStatement(sql);
			ps.setInt(1, userAndRole.getUserID());
			ps.setInt(2, userAndRole.getRoleID());
			ps.executeUpdate();
		} catch (SQLException e) {
			e.printStackTrace();
			throw new PermissionException("添加用户与角色关系失败");
		} finally {
			DBUtil.close(null, ps, conn);
		}
	}

	@Override
	public void delete(Integer id) throws PermissionException {
		String sql = "delete from UserAndRole where id=" + id;
		try {
			executeUpdate(sql);
		} catch (SQLException e) {
			e.printStackTrace();
			throw new PermissionException("删除用户与角色关系失败");
		}
	}

	@Override
	public UserAndRole findByID(Integer id) throws PermissionException {
		Connection conn = null;
		PreparedStatement ps = null;
		ResultSet rs = null;
		UserAndRole userAndRole = null;
		String sql = "select id,UserID,RoleID from UserAndRole where id=?";
		try {
			conn = DBUtil.getConnection();
			ps = conn.prepareStatement(sql);
			ps.setInt(1, id);
			rs = ps.executeQuery();
			if (rs.next()) {
				userAndRole = new UserAndRole();
				userAndRole.setId(rs.getInt(1));
				userAndRole.setUserID(rs.getInt(2));
				userAndRole.setRoleID(rs.getInt(3));
			}
		} catch (SQLException e) {
			e.printStackTrace();
			throw new PermissionException("查询用户与角色关系失败");
		} finally {
			DBUtil.close(rs, ps, conn);
		}
		return userAndRole;
	}

	@Override
	public List<UserAndRole> list(PageRoll pageRoll) throws PermissionException {
		Connection conn = null;
		PreparedStatement ps = null;
		ResultSet rs = null;
		List<UserAndRole> list = new ArrayList<UserAndRole>();
		String sql = "select id,UserID,RoleID from UserAndRole limit ?,?";
		try {
			conn = DBUtil.getConnection();
			ps = conn.prepareStatement(sql);
			ps.setInt(1, (pageRoll.getCurrPage() - 1) * pageRoll.getPageSize());
			ps.setInt(2, pageRoll.getPageSize());
			rs = ps.executeQuery();
			while (rs.next()) {
				UserAndRole userAndRole = new UserAndRole();
				userAndRole.setId(rs.getInt(1));
				userAndRole.setUserID(rs.getInt(2));
				userAndRole.setRoleID(rs.getInt(3));
				list.add(userAndRole);
			}
			sql = "select count(*) from UserAndRole";
			ps = conn.prepareStatement(sql);
			rs = ps.executeQuery();
			rs.next();
			pageRoll.setCountSize(rs.getInt(1));

		} catch (SQLException e) {
			e.printStackTrace();
			throw new PermissionException("查询用户与角色关系列表失败");
		} finally {
			DBUtil.close(rs, ps, conn);
		}
		return list;
	}

	@Override
	public void update(UserAndRole userAndRole) throws PermissionException {
		Connection conn = null;
		PreparedStatement ps = null;
		String sql = "update UserAndRole set UserID=?,RoleID=? where id=?";
		try {
			conn = DBUtil.getConnection();
			ps = conn.prepareStatement(sql);
			ps.setInt(1, userAndRole.getUserID());
			ps.setInt(2, userAndRole.getRoleID());
			ps.setInt(3, userAndRole.getId());
			ps.executeUpdate();
		} catch (SQLException e) {
			e.printStackTrace();
			throw new PermissionException("修改用户与角色关系失败");
		} finally {
			DBUtil.close(null, ps, conn);
		}
	}

	@Override
	public List<UserAndRole> findByUserID(Integer userID)
			throws PermissionException {
		Connection conn = null;
		PreparedStatement ps = null;
		ResultSet rs = null;
		List<UserAndRole> list = new ArrayList<UserAndRole>();
		String sql = "select id,UserID,RoleID from UserAndRole where UserID=?";
		try {
			conn = DBUtil.getConnection();
			ps = conn.prepareStatement(sql);
			ps.setInt(1, userID);
			rs = ps.executeQuery();
			while (rs.next()) {
				UserAndRole userAndRole = new UserAndRole();
				userAndRole.setId(rs.getInt(1));
				userAndRole.setUserID(rs.getInt(2));
				userAndRole.setRoleID(rs.getInt(3));
				list.add(userAndRole);
			}
		} catch (SQLException e) {
			e.printStackTrace();
			throw new PermissionException("根据用户ID查询用户与角色关系失败");
		} finally {
			DBUtil.close(rs, ps, conn);
		}
		return list;
	}

	@Override
	public List<UserAndRole> findByRoleID(Integer roleID)
			throws PermissionException {
		Connection conn = null;
		PreparedStatement ps = null;
		ResultSet rs = null;
		List<UserAndRole> list = new ArrayList<UserAndRole>();
		String sql = "select id,UserID,RoleID from UserAndRole where RoleID=?";
		try {
			conn = DBUtil.getConnection();
			ps = conn.prepareStatement(sql);
			ps.setInt(1, roleID);
			rs = ps.executeQuery();
			while (rs.next()) {
				UserAndRole userAndRole = new UserAndRole();
				userAndRole.setId(rs.getInt(1));
				userAndRole.setUserID(rs.getInt(2));
				userAndRole.setRoleID(rs.getInt(3));
				list.add(userAndRole);
			}
		} catch (SQLException e) {
			e.printStackTrace();
			throw new PermissionException("根据角色ID查询用户与角色关系失败");
		} finally {
			DBUtil.close(rs, ps, conn);
		}
		return list;
	}

	@Override
	public void delete(Integer userID, Integer roleID)
			throws PermissionException {
		StringBuffer sql = new StringBuffer();
		sql.append("delete from UserAndRole where userID=");
		sql.append(userID);
		sql.append(" and roleID=");
		sql.append(roleID);
		try {
			executeUpdate(sql.toString());
		} catch (SQLException e) {
			e.printStackTrace();
			throw new PermissionException("删除用户与角色关系失败");
		}
	}
	

}