package classmate;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.Statement;
import java.sql.ResultSet;
import java.sql.SQLException;

public class DBUser {

	DataSource dataSource;
	
	public DBUser(DataSource dataSource) {
		
			this.dataSource = dataSource;
	}
	
	
	public boolean checkUser(String name,String psw) throws Exception{
        Connection connect = null;
		String strSql;
		ResultSet rs;
		boolean result=false;
        strSql = "select * from classuser where username='"
					+ name + "' and password='" + psw + "'";
		try {
			connect = dataSource.getConnection();
			Statement stmt = connect.createStatement();
			rs = stmt.executeQuery(strSql);
			if ( rs.next()) {
				result=true;
			}
		} 
		catch(SQLException ex) { 
			ex.printStackTrace();
		}
		finally{
			if(connect!=null)
				connect.close();
		}
		return result;
	
	}
	
	public  int Insert(String username,String password) throws Exception{
        Connection connect = null;
      	String strSql;
   		int result = 0;
     	strSql = "insert into classuser values('" 
        		+ username 	+"','"
				+ password 	+"')";
		try {
			connect = dataSource.getConnection();
			Statement stmt = connect.createStatement();
			result = stmt.executeUpdate(strSql);
		} 
		catch(SQLException ex) { 
			System.err.println(ex.getMessage());
		}
		finally{
			if(connect!=null)
				connect.close();
		}
		return  result;
	}	
	
}