package dao;
/**
* @author 作者: guya
* description:
*/
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;
 
import bean.Hero;
 
public class HeroDAO {
 
    public int getTotal() {
    	Connection c=null;
    	Statement s=null;
    	ResultSet rs=null;
        int total = 0;
        try {
        	c=DbUtil.getConnection();
            s= c.createStatement();
            String sql = "select count(*) from hero";
            rs = s.executeQuery(sql);
            while (rs.next()) {
                total = rs.getInt(1);
            }
            
        } catch (SQLException e) {       
            e.printStackTrace();
        }finally {
        	DbUtil.close(c, s, rs);
		}
        return total;
    }
    public void add(Hero hero) {
    	Connection c=null;
    	PreparedStatement ps=null;
    	ResultSet rs=null;
        try {
            c=DbUtil.getConnection();
            String sql = "insert into hero values(null,?,?,?)";
            ps = c.prepareStatement(sql,Statement.RETURN_GENERATED_KEYS);
            ps.setString(1, hero.name);
            ps.setFloat(2, hero.hp);
            ps.setInt(3, hero.damage);
            ps.execute();
            rs = ps.getGeneratedKeys();
            if (rs.next()) {
                int id = rs.getInt(1);
                hero.id = id;
            }
        } catch (SQLException e) { 
            e.printStackTrace();
        }finally {
        	 DbUtil.close(c, ps, rs);
		}
    }
    public void update(Hero hero) {
    	Connection c=null;
    	PreparedStatement ps=null;
    	ResultSet rs=null;
        try {
            c=DbUtil.getConnection();
            String sql = "update hero set name= ?, hp = ? , damage = ? where id = ?";
            ps = c.prepareStatement(sql);
            ps.setString(1, hero.name);
            ps.setFloat(2, hero.hp);
            ps.setInt(3, hero.damage);
            ps.setInt(4, hero.id);
            ps.execute();
        } catch (SQLException e) {         
            e.printStackTrace();
        }finally {
        	DbUtil.close(c, ps);
		}
    }
    public void delete(int id) {
    	Connection c=null;
    	Statement s=null;
    	ResultSet rs=null;
        try {
            c=DbUtil.getConnection();
            s = c.createStatement();
            String sql = "delete from hero where id = " + id;
            s.execute(sql);
        }catch (SQLException e) {   
            e.printStackTrace();
        }finally {
        	DbUtil.close(c, s);
		}
    }
    public Hero get(int id) {
    	Connection c=null;
    	Statement s=null;
    	ResultSet rs=null;
        Hero hero = null;
        try {
            c=DbUtil.getConnection();
            s = c.createStatement();
            String sql = "select * from hero where id = " + id;
            rs = s.executeQuery(sql);
            if (rs.next()) {
                hero = new Hero();
                hero.setName(rs.getString(2));
                hero.setHp(rs.getFloat("hp"));
                hero.setDamage(rs.getInt(4));
                hero.setId(id);
            }
        } catch (SQLException e) {     
            e.printStackTrace();
        }finally {
        	DbUtil.close(c, s, rs);
		}
        return hero;
    }
    public List<Hero> list() {
        return list(0, Short.MAX_VALUE);
    }
    public List<Hero> list(int start, int count) {
    	Connection c=null;
    	PreparedStatement ps=null;
    	ResultSet rs=null;
        List<Hero> heros = new ArrayList<Hero>();
        try {
            c=DbUtil.getConnection();
            String sql = "select * from hero order by id desc limit ?,? ";
            ps = c.prepareStatement(sql);
            ps.setInt(1, start);
            ps.setInt(2, count);
            rs = ps.executeQuery();
            while (rs.next()) {
                Hero hero = new Hero();
                hero = new Hero();
                hero.setName(rs.getString(2));
                hero.setHp(rs.getFloat("hp"));
                hero.setDamage(rs.getInt(4));
                hero.setId(rs.getInt(1));
                heros.add(hero);
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }finally {
        	DbUtil.close(c, ps, rs);
		}
        return heros;
    }
 
}