java封装数据库操作

编程小白狼

发布于 2025-04-13 14:48:46

15100

代码可运行

文章被收录于专栏：编程小白狼编程小白狼

运行总次数：0

代码可运行

在 Java 中封装数据库操作是一个常见的需求，可以通过以下方法实现高效且可维护的代码结构。以下是一个 基于 JDBC 的封装示例，并附上最佳实践建议：

1. 封装数据库操作的核心思想

隐藏实现细节：将 Connection、Statement、ResultSet 等对象的创建和释放封装在内部。
统一异常处理：统一处理 SQLException，避免重复代码。
使用接口与实现分离：定义 DAO（Data Access Object）接口，分离业务逻辑与数据库操作。

2. 基础封装示例

步骤 1：创建数据库连接工具类

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

public class DBUtil {
    private static final String URL = "jdbc:mysql://localhost:3306/your_database";
    private static final String USER = "root";
    private static final String PASSWORD = "password";

    static {
        try {
            Class.forName("com.mysql.cj.jdbc.Driver");
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }
    }

    public static Connection getConnection() throws SQLException {
        return DriverManager.getConnection(URL, USER, PASSWORD);
    }

    public static void close(Connection conn) {
        if (conn != null) {
            try {
                conn.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }
}

步骤 2：定义 DAO 接口

public interface UserDao {
    void addUser(User user);
    User getUserById(int id);
    void updateUser(User user);
    void deleteUser(int id);
}

步骤 3：实现 DAO 类（使用 PreparedStatement 防止 SQL 注）

import java.sql.*;

public class UserDaoImpl implements UserDao {
    @Override
    public void addUser(User user) {
        String sql = "INSERT INTO users (name, email) VALUES (?, ?)";
        try (Connection conn = DBUtil.getConnection();
             PreparedStatement pstmt = conn.prepareStatement(sql)) {
            
            pstmt.setString(1, user.getName());
            pstmt.setString(2, user.getEmail());
            pstmt.executeUpdate();
            
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    @Override
    public User getUserById(int id) {
        String sql = "SELECT * FROM users WHERE id = ?";
        User user = null;
        try (Connection conn = DBUtil.getConnection();
             PreparedStatement pstmt = conn.prepareStatement(sql)) {
            
            pstmt.setInt(1, id);
            ResultSet rs = pstmt.executeQuery();
            if (rs.next()) {
                user = new User();
                user.setId(rs.getInt("id"));
                user.setName(rs.getString("name"));
                user.setEmail(rs.getString("email"));
            }
            
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return user;
    }

    // 其他方法类似，省略...
}

3. 高级封装（使用模板模式）

通过 JdbcTemplate 简化重复代码（类似 Spring JDBC 的思路）：

import java.sql.*;

public class JdbcTemplate {
    public static int update(String sql, Object... params) {
        try (Connection conn = DBUtil.getConnection();
             PreparedStatement pstmt = conn.prepareStatement(sql)) {
            
            for (int i = 0; i < params.length; i++) {
                pstmt.setObject(i + 1, params[i]);
            }
            return pstmt.executeUpdate();
            
        } catch (SQLException e) {
            e.printStackTrace();
            return -1;
        }
    }

    public static <T> T query(String sql, ResultSetHandler<T> handler, Object... params) {
        try (Connection conn = DBUtil.getConnection();
             PreparedStatement pstmt = conn.prepareStatement(sql)) {
            
            for (int i = 0; i < params.length; i++) {
                pstmt.setObject(i + 1, params[i]);
            }
            ResultSet rs = pstmt.executeQuery();
            return handler.handle(rs);
            
        } catch (SQLException e) {
            e.printStackTrace();
            return null;
        }
    }
}

// 定义结果集处理器接口
public interface ResultSetHandler<T> {
    T handle(ResultSet rs) throws SQLException;
}

// 使用示例
User user = JdbcTemplate.query(
    "SELECT * FROM users WHERE id = ?",
    rs -> {
        if (rs.next()) {
            User u = new User();
            u.setId(rs.getInt("id"));
            u.setName(rs.getString("name"));
            return u;
        }
        return null;
    },
    1
);

4. 使用 ORM 框架（推荐）

Hibernate：基于 JPA 规范的全自动 ORM。
MyBatis：半自动化的 SQL 映射框架，灵活性更高。

MyBatis 示例

定义 Mapper 接口：

public interface UserMapper {
    @Insert("INSERT INTO users (name, email) VALUES (#{name}, #{email})")
    void insertUser(User user);
    
    @Select("SELECT * FROM users WHERE id = #{id}")
    User getUserById(int id);
}

通过 SqlSession 调用：

try (SqlSession session = sqlSessionFactory.openSession()) {
    UserMapper mapper = session.getMapper(UserMapper.class);
    User user = mapper.getUserById(1);
}