SpringBoot 基于TDDL实现MySQL分库分表

MySQL分库分表是大型互联网应用中常用的数据库扩展方案，可以提高数据库的扩展能力和性能。TDDL是一款阿里巴巴开源的数据库中间件，支持MySQL的分库分表，本文将介绍如何使用TDDL实现MySQL分库分表。

引入依赖

在Spring Boot项目中，我们可以使用Maven或Gradle来管理依赖。下面是使用Maven的示例，需要添加TDDL和MySQL驱动的依赖：

<dependency>
    <groupId>com.taobao.tddl</groupId>
    <artifactId>tddl-client</artifactId>
    <version>5.0.0</version>
</dependency>
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <version>8.0.27</version>
</dependency>

配置TDDL

在Spring Boot项目中，可以通过application.properties或application.yml文件配置TDDL。下面是一个简单的配置示例：
properties

tddl.datasource.type=com.alibaba.druid.pool.DruidDataSource
tddl.datasource.ds_0.url=jdbc:mysql://localhost:3306/db_0?useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTC
tddl.datasource.ds_0.username=root
tddl.datasource.ds_0.password=123456
tddl.config.url.rule=db${0..2}:jdbc:mysql://db-{0..2}:3306/db_{0..2}?useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTC

yml

tddl:
    datasource:
        type:com.alibaba.druid.pool.DruidDataSource
        ds_0:
            url:jdbc:mysql://localhost:3306/db_0?useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTC
            username:root
            password:123456
    config:
        url:
            rule:db${0..2}:jdbc:mysql://db-{0..2}:3306/db_{0..2}?useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTC

在上述配置中，我们配置了一个MySQL数据库连接池，以及分片规则。分片规则定义了3个分片库，分别为db_0、db_1和db_2，每个分片库的连接地址为jdbc:mysql://db-{0…2}:3306/db_{0…2}，其中{0…2}表示分片库的编号，从0开始。

配置数据源和事务管理器

在Spring Boot中，我们可以使用@Bean注解将TDDL数据源和事务管理器注入到应用程序中。下面是一个简单的配置示例：

@Configuration
public class DataSourceConfig {

    @Bean
    public DataSource dataSource() throws SQLException {
        TDataSource dataSource = new TDataSource();
        dataSource.setAppName("myapp");
        dataSource.setDynamicRule(true);
        dataSource.init();
        return dataSource;
    }

    @Bean
    public PlatformTransactionManager transactionManager() throws SQLException {
        return new DataSourceTransactionManager(dataSource());
    }
}

在上述配置中，我们创建了一个TDataSource对象，并设置了应用程序名称和动态规则。动态规则表示TDDL可以根据实际情况动态调整分片规则，提高了系统的可用性和灵活性。然后将TDataSource对象注入到Spring容器中，并将它作为数据源配置给事务管理器。

创建数据表和实体类

在分库分表的方案中，我们需要将数据表按照一定的规则分散到不同的分片库和分片表中。下面是一个示例的订单数据表
创建脚本，我们将其按照用户ID进行分片，并且在每个分片库中创建2张分片表：

CREATE TABLE `t_order_0` (
  `id` bigint(20) NOT NULL AUTO_INCREMENT,
  `user_id` bigint(20) NOT NULL,
  `status` int(11) NOT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;

CREATE TABLE `t_order_1` (
  `id` bigint(20) NOT NULL AUTO_INCREMENT,
  `user_id` bigint(20) NOT NULL,
  `status` int(11) NOT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;

然后，我们创建一个实体类Order，用于映射订单数据表：

@Entity
@Table(name = "t_order")
public class Order {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    @Column(name = "user_id")
    private Long userId;

    private Integer status;

    // 省略getter和setter方法
}

编写DAO层代码

在Spring Boot中，我们可以使用Spring Data JPA或MyBatis等框架来访问数据库。下面是使用Spring Data JPA的示例：

public interface OrderRepository extends JpaRepository<Order, Long> {

    @Query(value = "/*+TDDL({'type':'direct','column':'user_id','force':true})*/SELECT * FROM t_order WHERE user_id = ?1", nativeQuery = true)
    List<Order> findByUserId(Long userId);
}

在上述代码中，我们定义了一个OrderRepository接口，并使用Spring Data JPA的特性来访问数据库。使用@Query注解可以自定义SQL查询语句，我们使用TDDL的注释来指定分片规则，以便TDDL能够正确地路由查询请求。

测试分库分表

最后，我们可以编写一个简单的测试用例来测试分库分表是否生效：

@SpringBootTest
public class OrderRepositoryTests {

    @Autowired
    private OrderRepository orderRepository;

    @Test
    void testInsert() {
        Order order = new Order();
        order.setUserId(100L);
        order.setStatus(0);
        orderRepository.save(order);
    }

    @Test
    void testSelect() {
        List<Order> orders = orderRepository.findByUserId(100L);
        System.out.println(orders);
    }
}

在上述测试用例中，我们向数据库中插入一条数据，并使用分片键查询数据。由于我们的分片规则是按照用户ID进行分片，因此查询结果应该只包含符合条件的订单记录。

总结

本文介绍了如何使用TDDL实现MySQL分库分表，并通过一个简单的示例演示了如何配置TDDL、创建数据表和实体类、编写DAO层代码以及测试分库分表。TDDL是一款功能强大的数据库中间件，支持丰富的分片规则和路由策略，可以提高数据库的扩展能力和性能。如果您的应用程序需要进行数据库水平扩展，TDDL可能是一个不错的选择。不过，在实际应用中，您需要根据具体的业务需求和实际情况选择适合的数据库中间件，并进行合理的配置和优化。

除了TDDL，还有许多其他的数据库中间件和分布式数据库系统可以实现分库分表，例如MyCAT、Sharding-JDBC、OceanBase等。每种方案都有其优缺点和适用场景，您可以根据具体情况进行选择。无论使用哪种方案，分库分表的实现都需要对数据库的分片规则、路由策略、事务管理等方面进行深入理解和优化，这是一个具有挑战性的工作。