Mybatis怎么實現一個增刪改查程序

Mybatis怎么實現一個增刪改查程序

MyBatis 是一個優秀的持久層框架，它支持定制化 SQL、存儲過程以及高級映射。MyBatis 避免了幾乎所有的 JDBC 代碼和手動設置參數以及獲取結果集。MyBatis 可以使用簡單的 XML 或注解來配置和映射原生信息，將接口和 Java 的 POJOs(Plain Old Java Objects,普通的 Java對象)映射成數據庫中的記錄。

本文將介紹如何使用 MyBatis 實現一個簡單的增刪改查（CRUD）程序。

1. 環境準備

在開始之前，確保你已經安裝了以下工具和庫：

• JDK 1.8 或更高版本
• Maven 3.x
• MySQL 數據庫
• MyBatis 3.x

2. 創建 Maven 項目

首先，創建一個 Maven 項目。你可以使用以下命令：

mvn archetype:generate -DgroupId=com.example -DartifactId=mybatis-demo -DarchetypeArtifactId=maven-archetype-quickstart -DinteractiveMode=false

這將創建一個名為 mybatis-demo 的 Maven 項目。

3. 添加依賴

在 pom.xml 文件中添加 MyBatis 和 MySQL 驅動的依賴：

<dependencies>
    <!-- MyBatis -->
    <dependency>
        <groupId>org.mybatis</groupId>
        <artifactId>mybatis</artifactId>
        <version>3.5.7</version>
    </dependency>

    <!-- MySQL Connector -->
    <dependency>
        <groupId>mysql</groupId>
        <artifactId>mysql-connector-java</artifactId>
        <version>8.0.26</version>
    </dependency>

    <!-- JUnit for testing -->
    <dependency>
        <groupId>junit</groupId>
        <artifactId>junit</artifactId>
        <version>4.13.2</version>
        <scope>test</scope>
    </dependency>
</dependencies>

4. 創建數據庫表

在 MySQL 中創建一個簡單的表 user：

CREATE TABLE user (
    id INT PRIMARY KEY AUTO_INCREMENT,
    name VARCHAR(50) NOT NULL,
    email VARCHAR(50) NOT NULL
);

5. 創建實體類

在 src/main/java/com/example 目錄下創建一個 User 類：

package com.example;

public class User {
    private int id;
    private String name;
    private String email;

    // Getters and Setters
    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public String getEmail() {
        return email;
    }

    public void setEmail(String email) {
        this.email = email;
    }

    @Override
    public String toString() {
        return "User{" +
                "id=" + id +
                ", name='" + name + '\'' +
                ", email='" + email + '\'' +
                '}';
    }
}

6. 創建 MyBatis 配置文件

在 src/main/resources 目錄下創建一個 mybatis-config.xml 文件：

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE configuration
  PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
  "http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
    <environments default="development">
        <environment id="development">
            <transactionManager type="JDBC"/>
            <dataSource type="POOLED">
                <property name="driver" value="com.mysql.cj.jdbc.Driver"/>
                <property name="url" value="jdbc:mysql://localhost:3306/mybatis_demo"/>
                <property name="username" value="root"/>
                <property name="password" value="password"/>
            </dataSource>
        </environment>
    </environments>
    <mappers>
        <mapper resource="com/example/UserMapper.xml"/>
    </mappers>
</configuration>

7. 創建 Mapper 接口和 XML 文件

在 src/main/java/com/example 目錄下創建一個 UserMapper 接口：

package com.example;

import java.util.List;

public interface UserMapper {
    void insertUser(User user);
    User getUserById(int id);
    List<User> getAllUsers();
    void updateUser(User user);
    void deleteUser(int id);
}

在 src/main/resources/com/example 目錄下創建一個 UserMapper.xml 文件：

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper
  PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
  "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.example.UserMapper">
    <insert id="insertUser" parameterType="com.example.User">
        INSERT INTO user (name, email) VALUES (#{name}, #{email});
    </insert>

    <select id="getUserById" parameterType="int" resultType="com.example.User">
        SELECT * FROM user WHERE id = #{id};
    </select>

    <select id="getAllUsers" resultType="com.example.User">
        SELECT * FROM user;
    </select>

    <update id="updateUser" parameterType="com.example.User">
        UPDATE user SET name = #{name}, email = #{email} WHERE id = #{id};
    </update>

    <delete id="deleteUser" parameterType="int">
        DELETE FROM user WHERE id = #{id};
    </delete>
</mapper>

8. 創建 MyBatis 工具類

在 src/main/java/com/example 目錄下創建一個 MyBatisUtil 類：

package com.example;

import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;

import java.io.IOException;
import java.io.InputStream;

public class MyBatisUtil {
    private static SqlSessionFactory sqlSessionFactory;

    static {
        try {
            String resource = "mybatis-config.xml";
            InputStream inputStream = Resources.getResourceAsStream(resource);
            sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static SqlSession getSqlSession() {
        return sqlSessionFactory.openSession();
    }
}

9. 編寫測試類

在 src/test/java/com/example 目錄下創建一個 UserMapperTest 類：

package com.example;

import org.apache.ibatis.session.SqlSession;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import java.util.List;

import static org.junit.Assert.*;

public class UserMapperTest {
    private SqlSession sqlSession;
    private UserMapper userMapper;

    @Before
    public void setUp() {
        sqlSession = MyBatisUtil.getSqlSession();
        userMapper = sqlSession.getMapper(UserMapper.class);
    }

    @After
    public void tearDown() {
        sqlSession.close();
    }

    @Test
    public void testInsertUser() {
        User user = new User();
        user.setName("John Doe");
        user.setEmail("john.doe@example.com");
        userMapper.insertUser(user);
        sqlSession.commit();
        assertTrue(user.getId() > 0);
    }

    @Test
    public void testGetUserById() {
        User user = userMapper.getUserById(1);
        assertNotNull(user);
        assertEquals("John Doe", user.getName());
    }

    @Test
    public void testGetAllUsers() {
        List<User> users = userMapper.getAllUsers();
        assertFalse(users.isEmpty());
    }

    @Test
    public void testUpdateUser() {
        User user = userMapper.getUserById(1);
        user.setEmail("john.doe.updated@example.com");
        userMapper.updateUser(user);
        sqlSession.commit();
        User updatedUser = userMapper.getUserById(1);
        assertEquals("john.doe.updated@example.com", updatedUser.getEmail());
    }

    @Test
    public void testDeleteUser() {
        userMapper.deleteUser(1);
        sqlSession.commit();
        User user = userMapper.getUserById(1);
        assertNull(user);
    }
}

10. 運行測試

在項目根目錄下運行以下命令來執行測試：

mvn test

如果一切順利，你應該會看到所有測試都通過了。

11. 總結

通過以上步驟，我們成功地使用 MyBatis 實現了一個簡單的增刪改查程序。MyBatis 提供了強大的 SQL 映射功能，使得我們可以輕松地將 Java 對象與數據庫表進行映射，并通過簡單的 XML 或注解來配置 SQL 語句。

希望這篇文章對你理解和使用 MyBatis 有所幫助！