基於Mybatis-plus實現多租戶架構的全過程

2022-02-09 19:01:12

多租戶(Multi-Tenant)是SaaS中的一個重要概念，它是一種軟體架構技術，在多個租戶的環境下，共用同一套系統範例，並且租戶之間的資料具有隔離性，也就是說一個租戶不能去存取其他租戶的資料。基於不同的隔離級別，通常具有下面三種實現方案：

1、每個租戶使用獨立DataBase，隔離級別高，效能好，但成本大

2、租戶之間共用DataBase，使用獨立的Schema

3、租戶之間共用Schema，在表上新增租戶欄位，共用資料程度最高，隔離級別最低。

Mybatis-plus在第3層隔離級別上，提供了基於分頁外掛的多租戶的解決方案，我們對此來進行介紹。在正式開始前，首先做好準備工作建立兩張表，在基礎欄位後都新增租戶欄位tenant_id：

CREATE TABLE `user` (
  `id` bigint(20) NOT NULL,
  `name` varchar(20) DEFAULT NULL,
  `phone` varchar(11) DEFAULT NULL,
  `address` varchar(64) DEFAULT NULL,
  `tenant_id` bigint(20) DEFAULT NULL,
  PRIMARY KEY (`id`)
)
CREATE TABLE `dept` (
  `id` bigint(20) NOT NULL,
  `dept_name` varchar(64) DEFAULT NULL,
  `comment` varchar(128) DEFAULT NULL,
  `tenant_id` bigint(20) DEFAULT NULL,
  PRIMARY KEY (`id`)
)

在專案中匯入需要的依賴：

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>3.3.2</version>
</dependency>
<dependency>
    <groupId>com.github.jsqlparser</groupId>
    <artifactId>jsqlparser</artifactId>
    <version>3.1</version>
</dependency>

Mybatis-plus 設定類：

@EnableTransactionManagement(proxyTargetClass = true)
@Configuration
public class MybatisPlusConfig {
    @Bean
    public PaginationInterceptor paginationInterceptor() {
        PaginationInterceptor paginationInterceptor = new PaginationInterceptor();

        List<ISqlParser> sqlParserList=new ArrayList<>();
        TenantSqlParser tenantSqlParser=new TenantSqlParser();
        tenantSqlParser.setTenantHandler(new TenantHandler() {
            @Override
            public Expression getTenantId(boolean select) {               
                String tenantId = "3";
                return new StringValue(tenantId);
            }

            @Override
            public String getTenantIdColumn() {
                return "tenant_id";
            }

            @Override
            public boolean doTableFilter(String tableName) {
                return false;
            }
        });

        sqlParserList.add(tenantSqlParser);
        paginationInterceptor.setSqlParserList(sqlParserList);
        return paginationInterceptor;
    }
}

這裡主要實現的功能：

建立SQL解析器集合
建立租戶SQL解析器
設定租戶處理器，具體處理租戶邏輯

這裡暫時把租戶的id固定寫成3，來進行測試。測試執行全表語句：

public List<User> getUserList() {
    return userMapper.selectList(new LambdaQueryWrapper<User>().isNotNull(User::getId));
}

使用外掛解析執行的SQL語句，可以看到自動在查詢條件後加上了租戶過濾條件：

那麼在實際的專案中，怎麼將租戶資訊傳給租戶處理器呢，根據情況我們可以從快取或者請求頭中獲取，以從Request請求頭獲取為例：

@Override
public Expression getTenantId(boolean select) {
    ServletRequestAttributes attributes=(ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
    HttpServletRequest request = attributes.getRequest();
    String tenantId = request.getHeader("tenantId");
    return new StringValue(tenantId);
}

前端在發起http請求時，在Header中加入tenantId欄位，後端在處理器中獲取後，設定為當前這次請求的租戶過濾條件。

如果是基於請求頭攜帶租戶資訊的情況，那麼在使用中可能會遇到一個坑，如果當使用多執行緒的時候，新開啟的非同步執行緒並不會自動攜帶當前執行緒的Request請求。

@Override
public List<User> getUserListByFuture() {
    Callable getUser=()-> userMapper.selectList(new LambdaQueryWrapper<User>().isNotNull(User::getId));
    FutureTask<List<User>> future=new FutureTask<>(getUser);
    new Thread(future).start();
    try {
        return future.get();
    } catch (Exception e) {
        e.printStackTrace();
    }
    return null;
}

執行上面的方法，可以看出是獲取不到當前的Request請求的，因此無法獲得租戶id，會導致後續報錯空指標異常：

修改的話也非常簡單，開啟RequestAttributes的子執行緒共用，修改上面的程式碼：

@Override
public List<User> getUserListByFuture() {
    ServletRequestAttributes sra = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
    Callable getUser=()-> {
        RequestContextHolder.setRequestAttributes(sra, true);
        return userMapper.selectList(new LambdaQueryWrapper<User>().isNotNull(User::getId));
    };
    FutureTask<List<User>> future=new FutureTask<>(getUser);
    new Thread(future).start();
    try {
        return future.get();
    } catch (Exception e) {
        e.printStackTrace();
    }
    return null;
}

這樣修改後，在非同步執行緒中也能正常的獲取租戶資訊了。

那麼，有的小夥伴可能要問了，在業務中並不是所有的查詢都需要過濾租戶條件啊，針對這種情況，有兩種方式來進行處理。

1、如果整張表的所有SQL操作都不需要針對租戶進行操作，那麼就對錶進行過濾，修改doTableFilter方法，新增表的名稱：

@Override
public boolean doTableFilter(String tableName) {
    List<String> IGNORE_TENANT_TABLES= Arrays.asList("dept");
    return IGNORE_TENANT_TABLES.stream().anyMatch(e->e.equalsIgnoreCase(tableName));
}

這樣，在dept表的所有查詢都不進行過濾：

2、如果有一些特定的SQL語句不想被執行租戶過濾，可以通過@SqlParser註解的形式開啟，注意註解只能加在Mapper介面的方法上：

@SqlParser(filter = true)
@Select("select * from user where name =#{name}")
User selectUserByName(@Param(value="name") String name);

或在分頁攔截器中指定需要過濾的方法：

@Bean
public PaginationInterceptor paginationInterceptor() {
    PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
    paginationInterceptor.setSqlParserFilter(metaObject->{
        MappedStatement ms = SqlParserHelper.getMappedStatement(metaObject);
        // 對應Mapper、dao中的方法
        if("com.cn.tenant.dao.UserMapper.selectUserByPhone".equals(ms.getId())){
            return true;
        }
        return false;
    });
    ...
}

上面這兩種方式實現的功能相同，但是如果需要過濾的SQL語句很多，那麼第二種方式設定起來會比較麻煩，因此建議通過註解的方式進行過濾。

除此之外，還有一個比較容易踩的坑就是在複製Bean時，不要複製租戶id欄位，否則會導致SQL語句報錯：

public void createSnapshot(Long userId){
    User user = userMapper.selectOne(new LambdaQueryWrapper<User>().eq(User::getId, userId));
    UserSnapshot userSnapshot=new UserSnapshot();
    BeanUtil.copyProperties(user,userSnapshot);
    userSnapshotMapper.insert(userSnapshot);
}

檢視報錯可以看出，本身Bean的租戶欄位不為空的情況下，SQL又自動新增一次租戶查詢條件，因此導致了報錯：

我們可以修改複製Bean語句，手動忽略租戶id欄位，這裡使用的是hutool的BeanUtil工具類，可以新增忽略欄位。

BeanUtil.copyProperties(user,userSnapshot,"tenantId");

在忽略了租戶id的拷貝後，查詢可以正常執行。

最後，再來看一下對聯表查詢的支援，首先看一下包含子查詢的SQL：

@Select("select * from user where id in (select id from user_snapshot)")
List<User> selectSnapshot();

檢視執行結果，可以看見，在子查詢的內部也自動新增的租戶查詢條件：

再來看一下使用Join進行聯表查詢：

@Select("select u.* from user u left join user_snapshot us on u.id=us.id")
List<User> selectSnapshot();

同樣，會在左右兩張表上都新增租戶的過濾條件：

再看一下不使用Join的普通聯表查詢：

@Select("select u.* from user u ,user_snapshot us,dept d where u.id=us.id and d.id is not null")
List<User> selectSnapshot();

檢視執行結果，可以看見在這種情況下，只在FROM關鍵字後面的第一張表上新增了租戶的過濾條件，因此如果使用這種查詢方式，需要額外注意，使用者需要手動在SQL語句中新增租戶過濾。

總結

到此這篇關於基於Mybatis-plus實現多租戶架構的文章就介紹到這了,更多相關Mybatis-plus實現多租戶架構內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

基於Mybatis-plus實現多租戶架構的全過程

熱門文章