面向对象 (13)数据库访问

题目1：编写一个应用程序，输入用户名和密码，访问test数据库中t_login表（字段包括id、username、password），验证登录是否成功。

一、源程序

Test.java

/**访问数据库 查询
 * 1个类1个方法
 */
package cn.edu.ccut.w1210;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Scanner;

public class Test {
    public static void main(String[] args) {
        Scanner reader = new Scanner(System.in);
        System.out.print("请输入用户名和密码
");
        String username= reader.nextLine();
        String password= reader.nextLine();
        try {
            Class.forName("com.mysql.jdbc.Driver");//加载JDBC驱动程序
            Connection con = DriverManager.getConnection("jdbc:mysql://localhost:3306/work","root","");//创建数据库连
            //1.拆串
            Statement st = con.createStatement();//声明SQL语句对象
            ResultSet rs = st.executeQuery("select * from t_login where username = '"+username+"' and password = '"+password+"'");//访问work数据库中t_login表
            //2.预处理
            //PreparedStatement ps = con.prepareStatement("select * from t_login where username = ? and password = ?");
            //ps.setString(1, username);
            //ps.setString(2, password);
            //ResultSet rs = ps.executeQuery();//用ResultSet对象存放查询暂存结果
            if(rs.next()){//next()指针遍历结果集，未拨动则未查询到
                System.out.print("登录成功
");
            }else{
                System.out.print("登录失败
");
            }
            if(rs!=null){
                rs.close();
            }
            st.close();
            //ps.close();
            con.close();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

二、成功界面截图

题目2：在上一题基础上，当登录成功后，将t_user表（id、name、sex、birthday）的信息进行显示（要求使用DB.java完成登录和获取t_user表中数据的操作），最后再对t_user表进行一条记录的添加操作。

一、源程序

Main.java

/**t_login查询成功后，获取t_user表信息，并添加一条记录
 * 1个类1个方法
 */
package cn.edu.ccut.w1211t2;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        DB db = DB.getInstance();//创建DB对象
        Scanner reader = new Scanner(System.in);
        System.out.print("请输入用户名和密码:
");
        String username= reader.nextLine();
        String password= reader.nextLine();//获取用户名和密码
        ResultSet rs =null;
        rs = db.DBQuery("select * from t_login where username = ? and password = ?", new Object[]{username,password});//调用DBQuery进行查询
        try {
            if(rs.next()){//next()指针遍历结果集，未拨动则未查询到
                System.out.print("登录成功!t_user表内容为:
");
                rs = db.DBQuery("select * from t_user", new Object[]{});//调用DBQuery进行查询
                char sex = ' ';
                while(rs.next()){
                    if(rs.getInt(3)==0){
                        sex = '男';
                    }else if(rs.getInt(3)==1){
                        sex = '女';
                    }
                    System.out.println("name:"+rs.getString(2)+"   sex:"+sex+"   birthday:"+rs.getString(4));//进行显示                    
                }
                System.out.print("请输入对t_user表进行一条记录的添加信息:
");
                String name;
                String birthday;//暂存添加信息
                name=reader.nextLine();
                sex=reader.nextLine().charAt(0);
                birthday=reader.nextLine();
                int isex = -1;//初值-1
                if(sex=='男'){
                    isex=0;
                }else if(sex=='女'){
                    isex=1;
                }
                if(db.DBModify("insert into t_user (name,sex,birthday)values(?,?,?)", new Object[]{name,isex,birthday})>0){//调用DBModify添加且判断
                    System.out.print("添加成功
");
                }
            }else{
                System.out.print("登录失败
");
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        db.close();//调用方法关闭数据库连接
        reader.close();
    }
}

DB.java

//自定义数据库增删改查类
package cn.edu.ccut.w1211t2;

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

public class DB {
    private static DB db;
    private Connection con; 
    private PreparedStatement ps;
    private ResultSet rs;
    public static DB getInstance(){//单态构造方法
        if(db == null){
            db = new DB();
        }
        return db;
    }
    static{//静态代码块加载JBDC驱动程序
        try {
            Class.forName("com.mysql.jdbc.Driver");
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }
    }
    private DB(){//构造方法创建数据库连接
        try {
            con = DriverManager.getConnection("jdbc:mysql://localhost:3306/work","root","");
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
    public ResultSet DBQuery(String sql,Object[] field){//自定义预处理查询方法
        try {
            ps = con.prepareStatement(sql);
            if(field.length>0){
                for(int i = 0;i<field.length;i++){
                    ps.setObject(i+1, field[i]);//数组传值从0记，预处理传从1
                }
            }
            rs = ps.executeQuery();//用ResultSet对象存放查询暂存结果
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return rs;
    }
    public int DBModify(String sql,Object[] field){//修改方法
        int n = 0;
        try {
            ps = con.prepareStatement(sql);
            if(field.length>0){
                for(int i = 0;i<field.length;i++){
                    ps.setObject(i+1, field[i]);//数组传值从0记，预处理传从1
                }
            }
            n = ps.executeUpdate();
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return n;
    }
    public void close(){//关闭数据库连接方法
        try {
            if(rs!=null){
                rs.close();
            }
            ps.close();
            con.close();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}

二、成功界面截图