利用PreparedStatement处理登录、注册、转账等功能

最新推荐文章于 2023-03-11 14:49:27 发布

荒--

最新推荐文章于 2023-03-11 14:49:27 发布

阅读量1.3k

点赞数 2

分类专栏： JDBC

本文链接：https://blog.csdn.net/wangqing84411433/article/details/69096462

版权

JDBC 专栏收录该内容

11 篇文章 0 订阅

订阅专栏

package day03;

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

import day01.DBUtil;
import day01.DBUtil2;

/**
 * 与用户相关的业务逻辑
 */
public class UserService {
	public static void main(String[] args) {
		/**
		 * 程序启动后:选择1，2，3，4等操作
		 * 1.注册新用户  用户ID从1开始
		 * 2.更改用户信息
		 * 3.删除用户信息
		 * 4.查询用户信息
		 */
		System.out.println("请输入选项:");
		System.out.print("1:注册	");
		System.out.print("2:登录	");
		System.out.print("3:修改	");
		System.out.print("4:删除	");
		System.out.print("5:查询	");
		System.out.println("6:转账");
		Scanner scan = new Scanner(System.in);
		int option = Integer.parseInt(scan.nextLine().trim());
		switch(option){
			case 1:
				//注册
				regUser(scan);
				break;
			case 2:
				//登录
				login(scan);
				break;
			case 3:
				//修改
				modify(scan);
				break;
			case 4:
				//删除某条记录
				delete(scan);
				break;
			case 5:
				//查询某条记录
				find(scan);
				break;
			case 6:
				giveMoney(scan);
				break;
			default:
				System.out.println("输入有错");
		}
	}
	
	//转账
	public static void giveMoney(Scanner scan){
		/**
		 * 1.获取用户输入的信息
		 * 2.必要的验证,查看转出账户余额是否够
		 * --事务开始的地方
		 * 3.更新转出账户的余额
		 * 4.更新转入账户的余额
		 * --事务结束的地方
		 */
		System.out.println("现在是转账操作");
		System.out.println("请输入您的帐号:");
		String fromUser = scan.nextLine().trim();
		System.out.println("请输入收款人的帐号:");
		String toUser = scan.nextLine().trim();
		System.out.println("请输入转出金额:");
		String money = scan.nextLine().trim();
		//查看转出账户余额是否够
		String countSql = "SELECT money FROM user_wq "
					+ "WHERE name=?";
		try {
			Connection conn = DBUtil2.getConnection();
			//事物中有多个业务要处理时,关闭自动提交
			conn.setAutoCommit(false);
			
			PreparedStatement ps = conn.prepareStatement(countSql);
			ps.setString(1, fromUser);
			ResultSet rs = ps.executeQuery();
			//判断是否查询出数据
			if(rs.next()){
				int count = rs.getInt("money");
				//判断余额是否足够
				if(count >= Integer.parseInt(money)){
					//执行转账操作
					String fromSql = "UPDATE user_wq SET money=money-? "
								+ "WHERE name=?";
					ps = conn.prepareStatement(fromSql);
					ps.setInt(1, Integer.parseInt(money));
					ps.setString(2, fromUser);
					//修改当前用户的余额
					int flag = ps.executeUpdate();//此处并不会自动提交
					if(flag > 0){
						//修改收款人余额
						String toSql = "UPDATE user_wq SET money=money+? "
									+ "WHERE name=?";
						ps = conn.prepareStatement(toSql);
						ps.setInt(1, Integer.parseInt(money));
						ps.setString(2, toUser);
						if(ps.executeUpdate() > 0){
							System.out.println("转账成功");
							/**
							 * 两次更新账户操作均成功,我们才认为这次
							 * 转账操作完毕,提交事务
							 */
							conn.commit();//提交事务
						}else{
							System.out.println("转账失败:没有收款人:"+toUser);
							/**
							 * 若第二次更新失败,那么整次操作就算失败
							 * 应该回滚事务
							 */
							conn.rollback();//回滚事务
						}
					}
				}else{
					System.out.println("余额不足");
				}
			}else{
				System.out.println("没有该用户:"+fromUser);
			}
		} catch (SQLException e) {
			e.printStackTrace();
		} finally {
			DBUtil2.closeConnection();
		}
	}
	
	//注册
	public static void regUser(Scanner scan){
		/**
		 * 1.获取用户输入的相关信息
		 * 2.获取连接
		 * 3.获取Satatement
		 * 4.先获取ID的最大值,对该值+1,作为当前记录的主键值
		 * 5.插入记录
		 * 6.关闭连接
		 */
		try{
			System.out.println("现在是注册操作:");
			System.out.println("请输入用户名:");
			String user = scan.nextLine().trim();
			System.out.println("请输入密码:");
			String pwd = scan.nextLine().trim();
			System.out.println("请输入账户金额:");
			String money = scan.nextLine().trim();
			System.out.println("请输入邮箱:");
			String email = scan.nextLine().trim();
			Connection conn = DBUtil2.getConnection();
			Statement state = conn.createStatement();
			//此处要使用别名,否则后面获取时要使用rs.getInt("MAX(id)");
			String idSql = "SELECT MAX(id) id FROM user_wq";
			ResultSet rs = state.executeQuery(idSql);
			int id = -1;
			if(rs.next()){
				id = rs.getInt("id");
			}
			//统计出最大值后对ID加1
			id++;
			rs.close();//关闭节约资源
			String sql = "INSERT INTO user_wq VALUES (?,?,?,?,?)";
			PreparedStatement ps = conn.prepareStatement(sql);
			ps.setInt(1, id);
			ps.setString(2, user);
			ps.setString(3, pwd);
			ps.setString(4, money);
			ps.setString(5, email);
			int flag = ps.executeUpdate();
			if(flag >0){
				System.out.println("注册成功!欢迎你"+user);
			}else{
				System.out.println("呵呵");
			}
		}catch(Exception e){
			e.printStackTrace();
		}finally{
			DBUtil.closeConnection();
		}
	}

	//登录(有注入攻击 a' or '1'='1 )
	public static void login(Scanner scan){
		/**
		 * 1.要求输入用户名和密码
		 * 2.根据用户输入作为条件去表中查询
		 * 3.若查询出数据,则说明输入正确
		 */
		System.out.println("现在是登录操作");
		System.out.println("请输入用户名:");
		String user = scan.nextLine().trim().toLowerCase();
		System.out.println("请输入密码:");
		String pwd = scan.nextLine().trim();
		try{
			Connection conn = DBUtil2.getConnection();
			//注意输入端额大小写和数据库里面的大小写
			String sql = "SELECT * FROM user_wq WHERE "
						+"LOWER(name)="+"LOWER(?)"+" AND "
					    +"password=?";
			PreparedStatement ps = conn.prepareStatement(sql);
			ps.setString(1, user);
			ps.setString(2, pwd);
			ResultSet rs = ps.executeQuery();
			//根据用户的输入看能否查到数据
			if(rs.next()){
				System.out.println("登陆成功");
			}else{
				System.out.println("用户名或密码错误");
			}
		}catch(Exception e){
			e.printStackTrace();
		}finally{
			DBUtil2.closeConnection();
		}
	}

	//修改(以修改用户名为例)
	public static void modify(Scanner scan){
		try {
			System.out.println("现在是修改操作");
			System.out.println("请输入修改数据的ID:");
			int id = Integer.parseInt(scan.nextLine().trim());
			System.out.println("请输入修改后的姓名:");
			String name = scan.nextLine().trim();
			Connection conn = DBUtil2.getConnection();
			String sql = "UPDATE user_wq SET name=? WHERE id=?";
			PreparedStatement ps = conn.prepareStatement(sql);
			ps.setString(1, name);
			ps.setInt(2, id);
			int flag = ps.executeUpdate();
			if(flag > 0){
				System.out.println("修改成功");
			}else{
				System.out.println("修改失败");
			}
		} catch (SQLException e) {
			e.printStackTrace();
		} finally {
			DBUtil2.closeConnection();
		}
	}
	
	//删除某条记录(以用户名为例)
	public static void delete(Scanner scan){
		try {
			System.out.println("现在是删除操作");
			System.out.println("请输入删除的姓名:");
			String name = scan.nextLine().trim();
			Connection conn = DBUtil2.getConnection();
			String sql = "DELETE FROM user_wq WHERE name=?";
			PreparedStatement ps = conn.prepareStatement(sql);
			ps.setString(1, name);
			int flag = ps.executeUpdate();
			if(flag > 0){
				System.out.println("删除成功");
			}else{
				System.out.println("删除失败");
			}
		} catch (SQLException e) {
			e.printStackTrace();
		} finally {
			DBUtil2.closeConnection();
		}
	}
	
	//查询某条记录(以用户名为例)
	public static void find(Scanner scan){
		try {
			System.out.println("现在是查询操作");
			System.out.println("请输入用户的姓名:");
			String str = scan.nextLine().trim();
			Connection conn = DBUtil2.getConnection();
			String sql = "SELECT * FROM user_wq WHERE name=?";
			PreparedStatement ps = conn.prepareStatement(sql);
			ps.setString(1, str);
			ResultSet rs = ps.executeQuery();
			if(rs.next()){
				int id = rs.getInt("id");
				String name = rs.getString("name");
				String password = rs.getString("password");
				int money = rs.getInt("money");
				String email = rs.getString("email");
				System.out.println("查询结果为:");
				System.out.println("id="+id+"\nname="+name+"\npassword="+password+"\nmoney="+money+"\nemail="+email);
			}
		} catch (SQLException e) {
			e.printStackTrace();
		} finally {
			DBUtil2.closeConnection();
		}
	}
}

所用到的工具类为：

package day01;

import java.io.IOException;
import java.io.InputStream;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Collection;
import java.util.Properties;

import org.apache.commons.dbcp.BasicDataSource;

/**
 * 使用连接池技术管理连接
 */
public class DBUtil2 {
	// 数据库连接池
	private static BasicDataSource ds;
	// 为不同线程管理连接
	private static ThreadLocal<Connection> tl;
	static {
		Properties p = new Properties();
		InputStream is = DBUtil2.class.getClassLoader().getResourceAsStream("wq/config.properties");
		try {
			p.load(is);
			is.close();
			String driver = p.getProperty("driver").trim();
			String url = p.getProperty("url").trim();
			String user = p.getProperty("user").trim();
			String password = p.getProperty("password").trim();
			int initsize = Integer.parseInt((p.getProperty("initsize")));
			int maxactive = Integer.parseInt(p.getProperty("maxactive"));
			long maxwait = Integer.parseInt(p.getProperty("maxwait"));
			// 初始化连接池
			ds = new BasicDataSource();
			// 设置驱动
			ds.setDriverClassName(driver);
			// 设置url
			ds.setUrl(url);
			// 设置数据库用户名
			ds.setUsername(user);
			// 设置数据库密码
			ds.setPassword(password);
			// 设置初始化连接数量
			ds.setInitialSize(initsize);
			// 设置连接池允许的最大连接数量
			ds.setMaxActive(maxactive);
			// 设置最大等待时间
			ds.setMaxWait(maxwait);
			//初始化线程本地
			tl = new ThreadLocal<Connection>();
		} catch (IOException e) {
			e.printStackTrace();
			throw new RuntimeException("加载配置文件失败", e);
		}
	}
	
	/**
	 * 通过连接池获取一个空闲连接
	 * @throws SQLException 
	 */
	public static Connection getConnection() throws SQLException{
		Connection conn = ds.getConnection();
		tl.set(conn);
		return conn;
	}
	/**
	 * 关闭数据库连接
	 */
	public static void closeConnection(){
		try {
			Connection conn = tl.get();
			if(conn != null){
				//回复连接为自动提交事务
				conn.setAutoCommit(true);
				/**
				 * 通过连接池获取的此处的Connection的
				 * close()方法实际上并没有将连接关闭,
				 * 而是将该连接归还给连接池
				 */
				conn.close();
				tl.remove();//将该连接从线程本地删除
			}
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}
}

测试结果如下：

请输入选项:
1:注册	2:登录	3:修改	4:删除	5:查询	6:转账
6
现在是转账操作
请输入您的帐号:
jack
请输入收款人的帐号:
rose
请输入转出金额:
1000
转账成功

注意：事务中执行多条sql语句时，将连接的自动提交功能设置为false，当事务完成之后再手动提交事务，最后再将连接的提交事务方式修改为true(在工具类中修改)

荒--

关注

2
点赞
踩
8

收藏

觉得还不错? 一键收藏
打赏
1
评论
利用PreparedStatement处理登录、注册、转账等功能

package day02;import java.sql.Connection;import java.sql.ResultSet;import java.sql.Statement;import java.util.Scanner;import day01.DBUtil;import day01.DBUtil2;/** * 与用户相关的业务逻辑 */public cl
复制链接

扫一扫