mybatis-flex/mybatis-flex-test/mybatis-flex-native-test/src/main/java/com/mybatisflex/test/MultiDataSourceTester.java

/**
 * Copyright (c) 2022-2023, Mybatis-Flex (fuhai999@gmail.com).
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.mybatisflex.test;

import com.mybatisflex.core.MybatisFlexBootstrap;
import com.mybatisflex.core.datasource.DataSourceKey;
import com.mybatisflex.core.row.Db;
import com.mybatisflex.core.row.Row;
import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseBuilder;
import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseType;

import javax.sql.DataSource;
import java.util.List;

public class MultiDataSourceTester {

    public static void main(String[] args) {
        DataSource dataSource = new EmbeddedDatabaseBuilder()
                .setType(EmbeddedDatabaseType.H2)
                .setName("db1")
                .addScript("schema.sql")
                .addScript("data.sql")
                .build();

        DataSource dataSource2 = new EmbeddedDatabaseBuilder()
                .setType(EmbeddedDatabaseType.H2)
                .setName("db2")
                .addScript("schema02.sql")
                .addScript("data02.sql")
                .build();

        MybatisFlexBootstrap.getInstance()
                .setDataSource(dataSource)
                .addDataSource("ds2", dataSource2)
                .start();

        //默认查询 db1
        List<Row> rows1 = Db.selectAll("tb_account");
        System.out.println(rows1);

        System.out.println("------");

        List<Row> rows =  DataSourceKey.use("ds2"
                , () -> Db.selectAll("tb_account"));


        //查询数据源 ds2
        DataSourceKey.use("ds2");
        rows = Db.selectAll("tb_account");
        System.out.println(rows);

        boolean success = Db.tx(() -> {
            Db.updateById("tb_account",Row.ofKey("id",1)
                    .set("user_name","测试的user"));
            return false;
        });
        System.out.println("tx: " + success);

        rows = Db.selectAll("tb_account");
        System.out.println(rows);

    }
}