第15周作业

一、题目

　　编写一个应用程序，输入用户名和密码，访问test数据库中t_login表（字段包括id、username、password），验证登录是否成功。当登录成功后，将t_user表（id、name、sex、birthday）的信息进行显示（要求使用DB.java完成登录和获取t_user表中数据的操作），最后再对t_user表进行一条记录的添加操作。

二、代码

DB.java

/*
 * 创建DB类，分别定义加载驱动器和连接方法、查询方法，增删改方法，和关闭流方法
 */
package jdbc;

import java.sql.*;

public class DB {
    Connection con; //连接对象
    Statement  st;  //执行对象
    ResultSet  rs;   //查询对象
    
    String dbuser = "root";
    String dbpass = "0000";
    String dbstr = "jdbc:mysql://localhost:3306/test1";
    
    DB(){
        //1
        try {
            Class.forName("com.mysql.jdbc.Driver");//加载驱动器
        } catch (ClassNotFoundException e) {
            System.out.println("加载驱动器错误");
            e.printStackTrace();
        }
    }
    Connection getCon(){
        //2
        try {
            con = DriverManager.getConnection(dbstr,dbuser,dbpass);//创建连接
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return con;
    }
    //查询方法
    ResultSet  getQuery(String sql){
        //3
        try {
            st = this.getCon().createStatement();
            rs = st.executeQuery(sql);
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return rs;
    }
    //增删改方法
    int  getUpdate(String sql){
        int n = 0;
        try {
            st = this.getCon().createStatement();
            n = st.executeUpdate(sql);
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return n;
    }
    //关闭流
    void close(){
        try {
             if(rs!=null)   rs.close();
             if(st!=null) st.close();
             if(con!=null) con.close();
        } catch (SQLException e) {
                e.printStackTrace();
        }
        
    }

}

UserAdd.java

/*
 * 创建用户增加类，增添数据
 */
package jdbc;


public class UserAdd {

    public void addUser(String name,String sex,String birthday) {
        DB db = new DB();//创建DB对象
        String sqlAdd = "insert into t_user  (name,sex,birthday) value('"+name+"','"+sex+"','"+birthday+"')";
        int n = db.getUpdate(sqlAdd);
        if (n > 0) {//判断是否修改了行数
            System.out.println("插入数据成功");
        } else {
            System.out.println("插入数据失败");
        }
        db.close();

    }
}

UserLogin.java

/*
 * 创建系统登录类，登陆成功后，可以查询数据和添加数据
 */
package jdbc;

import java.sql.*;
import java.util.*;

public class UserLogin {

    public static void main(String[] args) {
        Scanner reader = new Scanner(System.in);
        System.out.println("输入用户名:");
        String username = reader.next();
        System.out.println("输入密码:");
        String password = reader.next();
        //声明对象
        ResultSet rs = null;
        DB db = new DB();
        try {
            String sql = "select * from t_login where username='" + username
                    + "' and password = '" + password + "'";
            rs = db.getQuery(sql);
            boolean b = rs.next();
            if (b) {
                System.out.println("登录成功,进入系统管理");
                System.out.println("1 用户增加  2 用户查询  3系统退出");
                int n = reader.nextInt();
                while(true){
                    switch (n) {
                    case 1: {
                        UserAdd adduser = new UserAdd();//创建增加用户类对象
                        System.out.println("用户名");  String  name = reader.next();
                        System.out.println("性别");    String  sex = reader.next();
                        System.out.println("生日");    String  birthday = reader.next();
                        adduser.addUser(name, sex,birthday);
                        break;
                    }
                    case 2:{
                        UserSelect  allUser = new UserSelect();//创建用户查询对象
                        allUser.allUser();
                        break;
                    }
                    case 3:{
                        System.out.println("程序结束");System.exit(0);
                    }
                    }
                    System.out.println("1 用户增加  2 用户查询  3系统退出");
                    n = reader.nextInt();
                }
                
                
            } else {
                System.out.println("登录失败");
            }
            db.close();//关闭流
        } catch (SQLException e) {
            System.out.println("DB连接错误");
            e.printStackTrace();
        }
    }
}

UserSelect.java

/*
 * 创建查询用户表类，获取用户表信息
 */
package jdbc;


import java.sql.*;

public class UserSelect {
    ResultSet rs;
    public void allUser() {
        DB db = new DB();

        String sqlQuery = "SELECT * FROM  t_user";
        rs = db.getQuery(sqlQuery);
        try {
            System.out.println("序号        用户名         性别      生日");
            while (rs.next()) { // 将结果集的指针下下移，如果结果集结束，返回-1
                //分别获得用户表每一列信息
                int id = rs.getInt(1);
                String name = rs.getString(2);
                String sex = rs.getString(3);
                String birthday = rs.getString(4);
                System.out.println(id + "      " + name + "       "
                        + sex + "        " + birthday);
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }

    }
}

 

三、运行结果