springboot(十三)-分庫分表-手動配置

sharding-jdbc簡介

Sharding-JDBC直接封裝JDBC API，可以理解為增強版的JDBC驅動，舊代碼遷移成本幾乎為零：
可適用於任何基於java的ORM框架，如：JPA, Hibernate, Mybatis, Spring JDBC Template或直接使用JDBC。
可基於任何第三方的數據庫連接池，如：DBCP, C3P0, BoneCP, Druid等。
理論上可支持任意實現JDBC規范的數據庫。雖然目前僅支持MySQL，但已有支持Oracle，SQLServer，DB2等數據庫的計划。
Sharding-JDBC定位為輕量級java框架，使用客戶端直連數據庫，以jar包形式提供服務，未使用中間層，無需額外部署，無其他依賴，DBA也無需改變原有的運維方式。SQL解析使用Druid解析器，是目前性能最高的SQL解析器。
具體的介紹可以上它的文檔那里看看，簡單歸納起來就是，它是一個增強版的JDBC，對使用者透明，邏輯代碼什么的都不用動，它來完成分庫分表的操作；然后它還支持分布式事務（不完善）。看起來很不錯的樣子。
下面用個小例子來看一下分庫分表的使用。使用的是SpringBoot，mybatis，DBCP連接池。

1.新建一個springboot項目

ArtifactId為sharding-jdbc-manualConfiguration.自己配置好目錄結構。

 

 

2.pom.xml

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
  xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
  <modelVersion>4.0.0</modelVersion>

  <groupId>com.itmuch.boot</groupId>
  <artifactId>sharding-jdbc-manualConfiguration</artifactId>
  <version>0.0.1-SNAPSHOT</version>
  <packaging>jar</packaging>

  <name>sharding-jdbc-manualConfiguration</name>
  <url>http://maven.apache.org</url>

  <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>1.5.9.RELEASE</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>

    <properties>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
        <java.version>1.8</java.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>1.3.1</version>
        </dependency>

        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <scope>runtime</scope>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>

        <!--sharding-jdbc -->
        <dependency>
            <groupId>io.shardingjdbc</groupId>
            <artifactId>sharding-jdbc-core</artifactId>
            <version>2.0.3</version>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-devtools</artifactId>
        </dependency>
<!--         <dependency> -->
<!--             <groupId>com.alibaba</groupId> -->
<!--             <artifactId>druid</artifactId> -->
<!--             <version>1.1.3</version> -->
<!--         </dependency> -->
        <dependency>
            <groupId>commons-dbcp</groupId>
            <artifactId>commons-dbcp</artifactId>
        </dependency>
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
        </dependency>

        <!-- https://mvnrepository.com/artifact/com.alibaba/fastjson -->
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>fastjson</artifactId>
            <version>1.2.47</version>
        </dependency>

    </dependencies>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
        </plugins>
    </build>
</project>

簡單提一下：

這里主要是這個依賴

　　　　 <dependency>
            <groupId>io.shardingjdbc</groupId>
            <artifactId>sharding-jdbc-core</artifactId>
            <version>2.0.3</version>
        </dependency>

它是當當網開源的sharding-jdbc，當然這個不重要啦！

順便說一下：

<!--         <dependency> -->
<!--             <groupId>com.alibaba</groupId> -->
<!--             <artifactId>druid</artifactId> -->
<!--             <version>1.1.3</version> -->
<!--         </dependency> -->
        <dependency>
            <groupId>commons-dbcp</groupId>
            <artifactId>commons-dbcp</artifactId>
        </dependency>

我是用的DBCP數據庫連接池，這里的druid連接池被我注掉了，當然你也可以使用它，把DBCP注掉。它們都差不多。我想很多其他博友應該有寫。大家找找看咯！

在這里我要吐槽下：TMD，CSDN博客上面的文章叫什么東西啊，我讀過幾次，在上面down下來的代碼都是不能運行的，然后在博客園上面再找相同內容的東西，一比對，發現很明顯的配置漏洞。我也是無語了。

所以我是推薦咱博客園的文章的，大家都很優秀！！！

application.yml

我們是手動配置數據源，那這里我們可以上面都不用謝了呀！直接放個空文件得了，當然你如果不想使用8080端口，想在這里配置一下當前項目的使用端口號，你可以在這里配置下咯！！！

 數據庫

我們要准備三個數據庫user_0,user_1,user_2,每個數據庫里准備兩張表user_info_0,user_info_1.一共使用六張一樣一樣的表。

DROP TABLE IF EXISTS `user_info_0`;
CREATE TABLE `user_info_0` (
  `user_id` bigint(19) NOT NULL,
  `user_name` varchar(45) DEFAULT NULL,
  `account` varchar(45) DEFAULT NULL,
  `password` varchar(45) DEFAULT NULL,
  PRIMARY KEY (`user_id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;

-- ----------------------------
-- Table structure for user_info_1
-- ----------------------------
DROP TABLE IF EXISTS `user_info_1`;
CREATE TABLE `user_info_1` (
  `user_id` bigint(19) NOT NULL,
  `user_name` varchar(45) DEFAULT NULL,
  `account` varchar(45) DEFAULT NULL,
  `password` varchar(45) DEFAULT NULL,
  PRIMARY KEY (`user_id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;

Application.java

@SpringBootApplication
@EnableAutoConfiguration(exclude = { DataSourceAutoConfiguration.class })
public class Application {

    public static void main(String[] args) {
        SpringApplication.run(Application.class, args);
    }
}

啟動類嘛，我們都一樣！不不不。。。注意了，多了個注解，看到它要干嘛了嗎？它要把自動配置數據源的功能排除掉。

這里你要是看過我前面的文章“為什么用springboot”中的springboot自動配置的原理，這個東西，so easy啦！

 testController.java

@RestController
public class testController {
    @Resource
    UserInfoMapper userInfoMaper;

    @Resource
    DemoService demoService;

    @GetMapping("insert/{id}")
    public String insertData(@PathVariable Long id) {
        demoService.demo();
        return "success";
    }

    @GetMapping("get/{id}")
    public String getData(@PathVariable Long id) {
        UserInfo userInfoByUserId = demoService.getUserInfoByUserId(id);
        System.out.println("得到的結果為：" + JSON.toJSON(userInfoByUserId));
        return JSON.toJSON(userInfoByUserId).toString();
    }
}

對外接口，我們提供兩個吧，向數據庫插入數據傳的id沒用，我傳了玩兒的！！！

插入的內容請看下面的service.

DemoService.java

@Service
public class DemoService {

    @Resource
    UserInfoMapper userInfoMapper;

    public static Long userId = 100L;

    public void demo() {
        System.out.println("Insert--------------");

        for (int i = 1; i <= 100; i++) {
            UserInfo userInfo = new UserInfo();
            userInfo.setUserId(userId);
            System.out.println(userId);

            userInfo.setAccount("Account" + i);
            userInfo.setPassword("pass" + i);
            userInfo.setUserName("name" + i);
            userId++;
            userInfoMapper.insert(userInfo);
            System.out.println("第" + i + "條");
        }
        System.out.println("over..........");
    }

    public UserInfo getUserInfoByUserId(Long id) {
        return userInfoMapper.selectByPrimaryKey(id);
    }
}

我們就試驗向數據庫插100條數據咯，再提供下查詢方法。哦哦，Entity在下面

UserInfo.java

public class UserInfo {
    private Long userId;

    private String userName;

    private String account;

    private String password;

    public Long getUserId() {
        return userId;
    }

    public void setUserId(Long userId) {
        this.userId = userId;
    }

    public String getUserName() {
        return userName;
    }

    public void setUserName(String userName) {
        this.userName = userName == null ? null : userName.trim();
    }

    public String getAccount() {
        return account;
    }

    public void setAccount(String account) {
        this.account = account == null ? null : account.trim();
    }

    public String getPassword() {
        return password;
    }

    public void setPassword(String password) {
        this.password = password == null ? null : password.trim();
    }
}

UserInfoMapper.java

@Mapper
public interface UserInfoMapper {
    /**
     * This method was generated by MyBatis Generator. This method corresponds to
     * the database table user_info
     *
     * @mbg.generated Tue Mar 13 23:47:19 CST 2018
     */
    int insert(UserInfo record);

    /**
     * This method was generated by MyBatis Generator. This method corresponds to
     * the database table user_info
     *
     * @mbg.generated Tue Mar 13 23:47:19 CST 2018
     */
    int insertSelective(UserInfo record);

    /**
     * This method was generated by MyBatis Generator. This method corresponds to
     * the database table user_info
     *
     * @mbg.generated Tue Mar 13 23:47:19 CST 2018
     */
    UserInfo selectByPrimaryKey(Long userId);

    /**
     * This method was generated by MyBatis Generator. This method corresponds to
     * the database table user_info
     *
     * @mbg.generated Tue Mar 13 23:47:19 CST 2018
     */
    int updateByPrimaryKeySelective(UserInfo record);

    /**
     * This method was generated by MyBatis Generator. This method corresponds to
     * the database table user_info
     *
     * @mbg.generated Tue Mar 13 23:47:19 CST 2018
     */
    int updateByPrimaryKey(UserInfo record);
}

因為是用mybatis的逆向工程自動生成的，東西多余了點。懶得刪了。但是注意下，類名上面的@Mapper注解是我加的。

這里我展開說一下，我在網上看到很多人在springboot項目中使用Mybatis作為持久層框架的時候，都不這么用。都是在resource根目錄下面建一個文件夾mapper,然后把所有的XXXMapper.xml文件放到里面，XXXMapper.java還是放到com...mapper中不變。然后在啟動類上面添加一個針對這個com...mapper文件夾的@ComponentScan(“com...mapper”)注解。還要在根目錄resource下面配置mybatis-config.xml文件。更有甚者，居然有人在application.properties文件或application.yml文件中配置mybatis的東西，例如：

mybatis:

  mapper- locations: classpath:mybatis/mapper/*.xml

  type-aliases- package: com.sun.shard.bean

雖然springboot也能支持這樣使用，但是我是不建議也不喜歡這樣的風格，因為你想啊，我們為什么要用springboot?不就是想盡量少寫點配置嘛，把這些繁瑣的東西交給springboot來做就得了！要不然springboot出mybatis-spring-boot-starter這個啟動器干啥！

對了，大家想了解為啥，建議大家自己看下源碼，你加了mybatis-spring-boot-starter這個依賴，在Maven Dependencies下面找到

看明白這個類，基本就沒啥東西了。

 扯遠了，我們繼續我們的分庫分表。

UserInfoMapper.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.itmuch.boot.mapper.UserInfoMapper">
  <resultMap id="BaseResultMap" type="com.itmuch.boot.entity.UserInfo">
    <!--
      WARNING - @mbg.generated
      This element is automatically generated by MyBatis Generator, do not modify.
      This element was generated on Tue Mar 13 23:47:19 CST 2018.
    -->
    <id column="user_id" jdbcType="BIGINT" property="userId" />
    <result column="user_name" jdbcType="VARCHAR" property="userName" />
    <result column="account" jdbcType="VARCHAR" property="account" />
    <result column="password" jdbcType="VARCHAR" property="password" />
  </resultMap>
  <sql id="Base_Column_List">
    <!--
      WARNING - @mbg.generated
      This element is automatically generated by MyBatis Generator, do not modify.
      This element was generated on Tue Mar 13 23:47:19 CST 2018.
    -->
    user_id, user_name, account, password
  </sql>
  <select id="selectByPrimaryKey" parameterType="java.lang.Long" resultMap="BaseResultMap">
    <!--
      WARNING - @mbg.generated
      This element is automatically generated by MyBatis Generator, do not modify.
      This element was generated on Tue Mar 13 23:47:19 CST 2018.
    -->
    select 
    <include refid="Base_Column_List" />
    from user_info
    where user_id = #{userId,jdbcType=BIGINT}
  </select>
  <insert id="insert" parameterType="com.itmuch.boot.entity.UserInfo">
    <!--
      WARNING - @mbg.generated
      This element is automatically generated by MyBatis Generator, do not modify.
      This element was generated on Tue Mar 13 23:47:19 CST 2018.
    -->
    insert into user_info (user_id, user_name, account, 
      password)
    values (#{userId,jdbcType=BIGINT}, #{userName,jdbcType=VARCHAR}, #{account,jdbcType=VARCHAR}, 
      #{password,jdbcType=VARCHAR})
  </insert>
  <insert id="insertSelective" parameterType="com.itmuch.boot.entity.UserInfo">
    <!--
      WARNING - @mbg.generated
      This element is automatically generated by MyBatis Generator, do not modify.
      This element was generated on Tue Mar 13 23:47:19 CST 2018.
    -->
    insert into user_info
    <trim prefix="(" suffix=")" suffixOverrides=",">
      <if test="userId != null">
        user_id,
      </if>
      <if test="userName != null">
        user_name,
      </if>
      <if test="account != null">
        account,
      </if>
      <if test="password != null">
        password,
      </if>
    </trim>
    <trim prefix="values (" suffix=")" suffixOverrides=",">
      <if test="userId != null">
        #{userId,jdbcType=BIGINT},
      </if>
      <if test="userName != null">
        #{userName,jdbcType=VARCHAR},
      </if>
      <if test="account != null">
        #{account,jdbcType=VARCHAR},
      </if>
      <if test="password != null">
        #{password,jdbcType=VARCHAR},
      </if>
    </trim>
  </insert>
  <update id="updateByPrimaryKeySelective" parameterType="com.itmuch.boot.entity.UserInfo">
    <!--
      WARNING - @mbg.generated
      This element is automatically generated by MyBatis Generator, do not modify.
      This element was generated on Tue Mar 13 23:47:19 CST 2018.
    -->
    update user_info
    <set>
      <if test="userName != null">
        user_name = #{userName,jdbcType=VARCHAR},
      </if>
      <if test="account != null">
        account = #{account,jdbcType=VARCHAR},
      </if>
      <if test="password != null">
        password = #{password,jdbcType=VARCHAR},
      </if>
    </set>
    where user_id = #{userId,jdbcType=BIGINT}
  </update>
  <update id="updateByPrimaryKey" parameterType="com.itmuch.boot.entity.UserInfo">
    <!--
      WARNING - @mbg.generated
      This element is automatically generated by MyBatis Generator, do not modify.
      This element was generated on Tue Mar 13 23:47:19 CST 2018.
    -->
    update user_info
    set user_name = #{userName,jdbcType=VARCHAR},
      account = #{account,jdbcType=VARCHAR},
      password = #{password,jdbcType=VARCHAR}
    where user_id = #{userId,jdbcType=BIGINT}
  </update>
</mapper>

 

 

 到這里我們一套流程算是下來了。接下來是重點，手動添加數據源配置。

DataSourceConfig.java

@Configuration
@MapperScan(basePackages = "com.itmuch.boot.mapper")
public class DataSourceConfig {

    @Bean(name = "shardingDataSource")
    DataSource getShardingDataSource() throws SQLException {
        ShardingRuleConfiguration shardingRuleConfig;
        shardingRuleConfig = new ShardingRuleConfiguration();
        shardingRuleConfig.getTableRuleConfigs().add(getUserTableRuleConfiguration());
        shardingRuleConfig.getBindingTableGroups().add("user_info");
        shardingRuleConfig.setDefaultDatabaseShardingStrategyConfig(
                new StandardShardingStrategyConfiguration("user_id", DemoDatabaseShardingAlgorithm.class.getName()));
        shardingRuleConfig.setDefaultTableShardingStrategyConfig(
                new StandardShardingStrategyConfiguration("user_id", DemoTableShardingAlgorithm.class.getName()));
        return new ShardingDataSource(shardingRuleConfig.build(createDataSourceMap()));
    }

    @Bean
    TableRuleConfiguration getUserTableRuleConfiguration() {
        TableRuleConfiguration orderTableRuleConfig = new TableRuleConfiguration();
        orderTableRuleConfig.setLogicTable("user_info");
        orderTableRuleConfig.setActualDataNodes("user_${0..2}.user_info_${0..1}");
        orderTableRuleConfig.setKeyGeneratorColumnName("user_id");
        return orderTableRuleConfig;
    }

    private Map<String, DataSource> createDataSourceMap() {
        Map<String, DataSource> result = new HashMap<>();
        result.put("user_0", createDataSource("user_0"));
        result.put("user_1", createDataSource("user_1"));
        result.put("user_2", createDataSource("user_2"));
        return result;
    }

    private DataSource createDataSource(final String dataSourceName) {
        BasicDataSource result = new BasicDataSource();
        result.setDriverClassName(com.mysql.jdbc.Driver.class.getName());
        result.setUrl(String.format("jdbc:mysql://localhost:3306/%s", dataSourceName));

        result.setUsername("root");
        result.setPassword("root");
        return result;
    }

}

 

 

 

 這個數據源配置類大家也看到了，並不復雜，你看啊，我們准備了三個數據庫user_0,user_1,user_2；每個數據中准備了兩張相同的業務表user_info_0和user_info_1.帶着這個信息去看這個類，不用多說。

DemoDatabaseShardingAlgorithm.java

 

public class DemoDatabaseShardingAlgorithm implements PreciseShardingAlgorithm<Long> {
    @Override
    public String doSharding(Collection<String> collection, PreciseShardingValue<Long> preciseShardingValue) {
        for (String each : collection) {
            if (each.endsWith(Long.parseLong(preciseShardingValue.getValue().toString()) % 3 + "")) {
                return each;
            }
        }
        throw new IllegalArgumentException();
    }
}

 

 顧名思義，這是分庫算法，根據選擇的字段（上面DataSourceConfig.java第12行我們選的字段是user_id）對3取余。

 DemoTableShardingAlgorithm.java

public class DemoTableShardingAlgorithm implements PreciseShardingAlgorithm<Long> {
    @Override
    public String doSharding(Collection<String> collection, PreciseShardingValue<Long> preciseShardingValue) {
        for (String each : collection) {
            if (each.endsWith(Long.parseLong(preciseShardingValue.getValue().toString()) % 2 + "")) {
                return each;
            }
        }
        throw new IllegalArgumentException();
    }
}

 

 顧名思義，這是分表算法，根據選擇的字段（上面DataSourceConfig.java第14行我們選的字段也是user_id）對2取余。

 

測試

代碼就這么多，啟動application.java類，我們訪問localhost:8080/insert/1試試。

然后看數據庫：我們之前建的三個庫，共六張表，四張表都是17條記錄，兩張表里面分別有16條，一共100條。根據user_id判斷，沒有重復的，那基本哦了呀！

 

---

 

代碼地址：https://gitee.com/fengyuduke/my_open_resources/blob/master/sharding-jdbc-manualConfiguration.rar