新Java运动：测试驱动开发3---用户注册4

原创

最老程序员 2022-12-15 11:19:17 博主文章分类：java ©著作权

文章标签 Java TDD 测试驱动开发测试用例用户注册 文章分类 运维

©著作权归作者所有：来自51CTO博客作者最老程序员的原创作品，请联系作者获取转载授权，否则将追究法律责任

完成数据库操作的基础架构之后，就是我们真正进行JDBC数据操作的时候了。所涉及的数据库表ER图如下所示：

新Java运动：测试驱动开发3---用户注册4_测试驱动开发

如上图所示，我们第一步是向t_user表中添加记录。由于用户注册需要操作多张表，因此需要用到事务，先写出一个简单的基于JDBC的事务框架，代码如下所示：

@Override
  public long registerUser(Map<String, Object> userInfo) {
    Connection conn = null;
    long userId = 0;
    try {
      conn = JdbcDs.getConnection();
      conn.setAutoCommit(false);
      userId = addUser(conn, userInfo);
      if (userId <= 0) {
        throw new SQLException("Fail to add user in t_user");
      } 
      conn.commit();
    } catch (SQLException e) {
      // TODO Auto-generated catch block
      e.printStackTrace();
      try {
        conn.rollback();
      } catch (SQLException e1) {
        // TODO Auto-generated catch block
        e1.printStackTrace();
      }
      userId = -1;
    } finally {
      try {
        conn.close();
      } catch (SQLException e) {
        // TODO Auto-generated catch block
        e.printStackTrace();
      }
    }
    return userId;
  }

其次是具体的用户添加操作，具体代码如下所示：

private long addUser(Connection conn, Map<String, Object> userInfo) {
    long userId = -1;
    PreparedStatement stmt = null;
    ResultSet rst = null;
    String sql = "insert into t_user(user_name, user_group_id, user_level_id) values(?, 2, 1)";
    
    try {
      stmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
      stmt.setString(1, (String)userInfo.get("userName"));
      int affectedNum = stmt.executeUpdate();
      if (1 == affectedNum) {
        rst = stmt.getGeneratedKeys();
        if (rst.next()) {
          userId = rst.getLong(1);
        }
      } else {
        userId = -1;
      }
    } catch (SQLException e) {
      // TODO Auto-generated catch block
      e.printStackTrace();
      userId = -1;
    } finally {
      try {
        if (rst != null) {
          rst.close();
        }
        if (stmt != null) {
          stmt.close();
        }
      } catch (Exception ex) {
        
      }
    }
    return userId;
  }