官方發(fā)布！Mybatis-Plus分庫分表神器來了

<dependency>
??<groupId>com.baomidougroupId>
??<artifactId>mybatis-mate-starterartifactId>
??<version>1.0.8version>
dependency>

<dependency>
??<groupId>com.baomidougroupId>
??<artifactId>mybatis-mate-annotationartifactId>
??<version>1.0.8version>
dependency>

@FieldDict(type?=?"user_sex", target =?"sexText")
private?Integer sex;

private?String?sexText;

@Component
public?class?DataDict?implements?IDataDict {

????/**
?????* 從數(shù)據(jù)庫或緩存中獲取
?????*/
????private?Map<String,?String> SEX_MAP =?new?ConcurrentHashMap<String,?String>() {{
????????put("0",?"女");
????????put("1",?"男");
????}};

????@Override
????public?String?getNameByCode(FieldDict fieldDict,?String?code) {
????????System.err.println("字段類型："?+ fieldDict.type() +?"，編碼："?+ code);
????????return?SEX_MAP.get(code);
????}
}

@FieldEncrypt(algorithm = Algorithm.PBEWithMD5AndDES)
private?String?password;

@FieldSensitive(type?=?"testStrategy")
private?String?username;

@FieldSensitive(type?= SensitiveType.mobile)
private?String?mobile;

@Configuration
public?class?SensitiveStrategyConfig?{

????/**
?????* 注入脫敏策略
?????*/
????@Bean
????public?ISensitiveStrategy?sensitiveStrategy()?{
????????// 自定義 testStrategy 類型脫敏處理
????????return?new?SensitiveStrategy().addStrategy("testStrategy", t -> t +?"***test***");
????}
}

/**
?* 演示文章敏感詞過濾
?*/
@RestController
public?class?ArticleController?{
????@Autowired
????private?SensitiveWordsMapper sensitiveWordsMapper;

????// 測試訪問下面地址觀察請求地址、界面返回?cái)?shù)據(jù)及控制臺(tái)（ 普通參數(shù) ）
????// 無敏感詞 http://localhost:8080/info?content=tom&see=1&age=18
????// 英文敏感詞 http://localhost:8080/info?content=my%20content%20is%20tomcat&see=1&age=18
????// 漢字敏感詞 http://localhost:8080/info?content=%E7%8E%8B%E5%AE%89%E7%9F%B3%E5%94%90%E5%AE%8B%E5%85%AB%E5%A4%A7%E5%AE%B6&see=1
????// 多個(gè)敏感詞 http://localhost:8080/info?content=%E7%8E%8B%E5%AE%89%E7%9F%B3%E6%9C%89%E4%B8%80%E5%8F%AA%E7%8C%ABtomcat%E6%B1%A4%E5%A7%86%E5%87%AF%E7%89%B9&see=1&size=6
????// 插入一個(gè)字變成非敏感詞 http://localhost:8080/info?content=%E7%8E%8B%E7%8C%AB%E5%AE%89%E7%9F%B3%E6%9C%89%E4%B8%80%E5%8F%AA%E7%8C%ABtomcat%E6%B1%A4%E5%A7%86%E5%87%AF%E7%89%B9&see=1&size=6
????@GetMapping("/info")
????public?String?info(Article article)?throws?Exception?{
????????return?ParamsConfig.toJson(article);
????}


????// 添加一個(gè)敏感詞然后再去觀察是否生效 http://localhost:8080/add
????// 觀察【貓】這個(gè)詞被過濾了 http://localhost:8080/info?content=%E7%8E%8B%E5%AE%89%E7%9F%B3%E6%9C%89%E4%B8%80%E5%8F%AA%E7%8C%ABtomcat%E6%B1%A4%E5%A7%86%E5%87%AF%E7%89%B9&see=1&size=6
????// 嵌套敏感詞處理 http://localhost:8080/info?content=%E7%8E%8B%E7%8C%AB%E5%AE%89%E7%9F%B3%E6%9C%89%E4%B8%80%E5%8F%AA%E7%8C%ABtomcat%E6%B1%A4%E5%A7%86%E5%87%AF%E7%89%B9&see=1&size=6
????// 多層嵌套敏感詞 http://localhost:8080/info?content=%E7%8E%8B%E7%8E%8B%E7%8C%AB%E5%AE%89%E7%9F%B3%E5%AE%89%E7%9F%B3%E6%9C%89%E4%B8%80%E5%8F%AA%E7%8C%ABtomcat%E6%B1%A4%E5%A7%86%E5%87%AF%E7%89%B9&see=1&size=6
????@GetMapping("/add")
????public?String?add()?throws?Exception?{
????????Long id =?3L;
????????if?(null?== sensitiveWordsMapper.selectById(id)) {
????????????System.err.println("插入一個(gè)敏感詞："?+ sensitiveWordsMapper.insert(new?SensitiveWords(id,?"貓")));
????????????// 插入一個(gè)敏感詞，刷新算法引擎敏感詞
????????????SensitiveWordsProcessor.reloadSensitiveWords();
????????}
????????return?"ok";
????}

????// 測試訪問下面地址觀察控制臺(tái)（ 請求json參數(shù) ）
????// idea 執(zhí)行 resources 目錄 TestJson.http 文件測試
????@PostMapping("/json")
????public?String?json(@RequestBody Article article)?throws?Exception?{
????????return?ParamsConfig.toJson(article);
????}
}

@Component
public?class?PostgresDdl?implements?IDdl?{

????/**
?????* 執(zhí)行 SQL 腳本方式
?????*/
????@Override
????public?List?getSqlFiles()?{
????????return?Arrays.asList(
????????????????// 內(nèi)置包方式
????????????????"db/tag-schema.sql",
????????????????// 文件絕對(duì)路徑方式
????????????????"D:\\db\\tag-data.sql"
????????);
????}
}

ddlScript.run(new StringReader("DELETE?FROM?user;\n" +
????????????????"INSERT?INTO?user?(id, username,?password, sex, email)?VALUES\n" +
????????????????"(20,?'Duo',?'123456',?0,?'[email protected]');"));

@Component
public?class?MysqlDdl?implements?IDdl?{

????@Override
????public?void?sharding(Consumer consumer)?{
????????// 多數(shù)據(jù)源指定，主庫初始化從庫自動(dòng)同步
????????String?group?=?"mysql";
????????ShardingGroupProperty sgp = ShardingKey.getDbGroupProperty(group);
????????if?(null?!= sgp) {
????????????// 主庫
????????????sgp.getMasterKeys().forEach(key -> {
????????????????ShardingKey.change(group?+ key);
????????????????consumer.accept(this);
????????????});
????????????// 從庫
????????????sgp.getSlaveKeys().forEach(key -> {
????????????????ShardingKey.change(group?+ key);
????????????????consumer.accept(this);
????????????});
????????}
????}

????/**
?????* 執(zhí)行 SQL 腳本方式
?????*/
????@Override
????public?List?getSqlFiles()?{
????????return?Arrays.asList("db/user-mysql.sql");
????}
}

@Mapper
@Sharding("mysql")
public interface UserMapper extends BaseMapper {

????@Sharding("postgres")
????Long selectByUsername(String username);
}

@Component
public?class?MyShardingStrategy?extends?RandomShardingStrategy?{

????/**
?????* 決定切換數(shù)據(jù)源 key {@link?ShardingDatasource}
?????*
?????*?@param?group 動(dòng)態(tài)數(shù)據(jù)庫組
?????*?@param?invocation {@link?Invocation}
?????*?@param?sqlCommandType {@link?SqlCommandType}
?????*/
????@Override
????public?void?determineDatasourceKey(String group, Invocation invocation, SqlCommandType sqlCommandType)?{
????????// 數(shù)據(jù)源組 group 自定義選擇即可， keys 為數(shù)據(jù)源組內(nèi)主從多節(jié)點(diǎn)，可隨機(jī)選擇或者自己控制
????????this.changeDatabaseKey(group, sqlCommandType, keys -> chooseKey(keys, invocation));
????}
}

mybatis-mate:
??sharding:
????health: true # 健康檢測
????primary: mysql # 默認(rèn)選擇數(shù)據(jù)源
????datasource:
??????mysql: # 數(shù)據(jù)庫組
????????-?key: node1
??????????...
????????-?key: node2
??????????cluster: slave # 從庫讀寫分離時(shí)候負(fù)責(zé) sql 查詢操作，主庫 master 默認(rèn)可以不寫
??????????...
??????postgres:
????????-?key: node1 # 數(shù)據(jù)節(jié)點(diǎn)
??????????...

/**
?* 

?* 性能分析攔截器，用于輸出每條 SQL 語句及其執(zhí)行時(shí)間
?* 

?*/
@Slf4j
@Component
@Intercepts({@Signature(type?= StatementHandler.class, method =?"query", args = {Statement.class, ResultHandler.class}),
????????@Signature(type?= StatementHandler.class, method =?"update", args = {Statement.class}),
????????@Signature(type?= StatementHandler.class, method =?"batch", args = {Statement.class})})
public?class?PerformanceInterceptor?implements?Interceptor {
????/**
?????* SQL 執(zhí)行最大時(shí)長，超過自動(dòng)停止運(yùn)行，有助于發(fā)現(xiàn)問題。
?????*/
????private?long maxTime =?0;
????/**
?????* SQL 是否格式化
?????*/
????private?boolean?format =?false;
????/**
?????* 是否寫入日志文件

?????* true 寫入日志文件，不阻斷程序執(zhí)行！

?????* 超過設(shè)定的最大執(zhí)行時(shí)長異常提示！
?????*/
????private?boolean?writeInLog =?false;

????@Override
????public?Object?intercept(Invocation invocation) throws Throwable {
????????Statement statement;
????????Object?firstArg = invocation.getArgs()[0];
????????if?(Proxy.isProxyClass(firstArg.getClass())) {
????????????statement = (Statement) SystemMetaObject.forObject(firstArg).getValue("h.statement");
????????}?else?{
????????????statement = (Statement) firstArg;
????????}
????????MetaObject stmtMetaObj = SystemMetaObject.forObject(statement);
????????try?{
????????????statement = (Statement) stmtMetaObj.getValue("stmt.statement");
????????}?catch?(Exception e) {
????????????// do nothing
????????}
????????if?(stmtMetaObj.hasGetter("delegate")) {//Hikari
????????????try?{
????????????????statement = (Statement) stmtMetaObj.getValue("delegate");
????????????}?catch?(Exception e) {

????????????}
????????}

????????String?originalSql =?null;
????????if?(originalSql ==?null) {
????????????originalSql = statement.toString();
????????}
????????originalSql = originalSql.replaceAll("[\\s]+",?" ");
????????int index = indexOfSqlStart(originalSql);
????????if?(index >?0) {
????????????originalSql = originalSql.substring(index);
????????}

????????// 計(jì)算執(zhí)行 SQL 耗時(shí)
????????long start = SystemClock.now();
????????Object?result = invocation.proceed();
????????long timing = SystemClock.now() - start;

????????// 格式化 SQL 打印執(zhí)行結(jié)果
????????Object?target = PluginUtils.realTarget(invocation.getTarget());
????????MetaObject metaObject = SystemMetaObject.forObject(target);
????????MappedStatement ms = (MappedStatement) metaObject.getValue("delegate.mappedStatement");
????????StringBuilder formatSql =?new?StringBuilder();
????????formatSql.append(" Time：").append(timing);
????????formatSql.append(" ms - ID：").append(ms.getId());
????????formatSql.append("\n Execute SQL：").append(sqlFormat(originalSql, format)).append("\n");
????????if?(this.isWriteInLog()) {
????????????if?(this.getMaxTime() >=?1?&& timing >?this.getMaxTime()) {
????????????????log.error(formatSql.toString());
????????????}?else?{
????????????????log.debug(formatSql.toString());
????????????}
????????}?else?{
????????????System.err.println(formatSql);
????????????if?(this.getMaxTime() >=?1?&& timing >?this.getMaxTime()) {
????????????????throw?new?RuntimeException(" The SQL execution time is too large, please optimize ! ");
????????????}
????????}
????????return?result;
????}

????@Override
????public?Object?plugin(Object?target) {
????????if?(target?instanceof?StatementHandler) {
????????????return?Plugin.wrap(target,?this);
????????}
????????return?target;
????}

????@Override
????public?void?setProperties(Properties prop) {
????????String?maxTime = prop.getProperty("maxTime");
????????String?format = prop.getProperty("format");
????????if?(StringUtils.isNotEmpty(maxTime)) {
????????????this.maxTime = Long.parseLong(maxTime);
????????}
????????if?(StringUtils.isNotEmpty(format)) {
????????????this.format =?Boolean.valueOf(format);
????????}
????}

????public?long getMaxTime() {
????????return?maxTime;
????}

????public?PerformanceInterceptor setMaxTime(long maxTime) {
????????this.maxTime = maxTime;
????????return?this;
????}

????public?boolean?isFormat() {
????????return?format;
????}

????public?PerformanceInterceptor setFormat(boolean?format) {
????????this.format = format;
????????return?this;
????}

????public?boolean?isWriteInLog() {
????????return?writeInLog;
????}

????public?PerformanceInterceptor setWriteInLog(boolean?writeInLog) {
????????this.writeInLog = writeInLog;
????????return?this;
????}

????public?Method getMethodRegular(Class clazz,?String?methodName) {
????????if?(Object.class.equals(clazz)) {
????????????return?null;
????????}
????????for?(Method method : clazz.getDeclaredMethods()) {
????????????if?(method.getName().equals(methodName)) {
????????????????return?method;
????????????}
????????}
????????return?getMethodRegular(clazz.getSuperclass(), methodName);
????}

????/**
?????* 獲取sql語句開頭部分
?????*
?????* @param sql
?????* @return
?????*/
????private?int indexOfSqlStart(String?sql) {
????????String?upperCaseSql = sql.toUpperCase();
????????Set?set?=?new?HashSet<>();
????????set.add(upperCaseSql.indexOf("SELECT "));
????????set.add(upperCaseSql.indexOf("UPDATE "));
????????set.add(upperCaseSql.indexOf("INSERT "));
????????set.add(upperCaseSql.indexOf("DELETE "));
????????set.remove(-1);
????????if?(CollectionUtils.isEmpty(set)) {
????????????return?-1;
????????}
????????List list =?new?ArrayList<>(set);
????????Collections.sort(list, Integer::compareTo);
????????return?list.get(0);
????}

????private?final?static?SqlFormatter sqlFormatter =?new?SqlFormatter();

????/**
?????* 格式sql
?????*
?????* @param boundSql
?????* @param format
?????* @return
?????*/
????public?static?String?sqlFormat(String?boundSql,?boolean?format) {
????????if?(format) {
????????????try?{
????????????????return?sqlFormatter.format(boundSql);
????????????}?catch?(Exception ignored) {
????????????}
????????}
????????return?boundSql;
????}
}

@RestController
@AllArgsConstructor
public class TestController {
????private?BuyService?buyService;

????// 數(shù)據(jù)庫 test 表 t_order 在事務(wù)一致情況無法插入數(shù)據(jù)，能夠插入說明多數(shù)據(jù)源事務(wù)無效
????// 測試訪問 http://localhost:8080/test
????// 制造事務(wù)回滾 http://localhost:8080/test?error=true 也可通過修改表結(jié)構(gòu)制造錯(cuò)誤
????// 注釋 ShardingConfig 注入 dataSourceProvider 可測試事務(wù)無效情況
????@GetMapping("/test")
????public?String?test(Boolean error) {
????????return?buyService.buy(null != error && error);
????}
}

// 測試 test 類型數(shù)據(jù)權(quán)限范圍，混合分頁模式
@DataScope(type =?"test", value = {
????????// 關(guān)聯(lián)表 user 別名 u 指定部門字段權(quán)限
????????@DataColumn(alias =?"u", name =?"department_id"),
????????// 關(guān)聯(lián)表 user 別名 u 指定手機(jī)號(hào)字段（自己判斷處理）
????????@DataColumn(alias =?"u", name =?"mobile")
})
@Select("select u.* from user u")
List selectTestList(IPage page, Long id,?@Param("name") String username);

@Bean
public?IDataScopeProvider dataScopeProvider() {
????return?new?AbstractDataScopeProvider() {
????????@Override
????????protected?void?setWhere(PlainSelect plainSelect,?Object[] args, DataScopeProperty dataScopeProperty) {
????????????// args 中包含 mapper 方法的請求參數(shù)，需要使用可以自行獲取
????????????/*
????????????????// 測試數(shù)據(jù)權(quán)限，最終執(zhí)行 SQL 語句
????????????????SELECT u.* FROM user u WHERE (u.department_id IN ('1', '2', '3', '5'))
????????????????AND u.mobile LIKE '%1533%'
?????????????*/
????????????if?("test".equals(dataScopeProperty.getType())) {
????????????????// 業(yè)務(wù) test 類型
????????????????List dataColumns = dataScopeProperty.getColumns();
????????????????for?(DataColumnProperty dataColumn : dataColumns) {
????????????????????if?("department_id".equals(dataColumn.getName())) {
????????????????????????// 追加部門字段 IN 條件，也可以是 SQL 語句
????????????????????????Set<String> deptIds =?new?HashSet<>();
????????????????????????deptIds.add("1");
????????????????????????deptIds.add("2");
????????????????????????deptIds.add("3");
????????????????????????deptIds.add("5");
????????????????????????ItemsList itemsList =?new?ExpressionList(deptIds.stream().map(StringValue::new).collect(Collectors.toList()));
????????????????????????InExpression inExpression =?new?InExpression(new?Column(dataColumn.getAliasDotName()), itemsList);
????????????????????????if?(null?== plainSelect.getWhere()) {
????????????????????????????// 不存在 where 條件
????????????????????????????plainSelect.setWhere(new?Parenthesis(inExpression));
????????????????????????}?else?{
????????????????????????????// 存在 where 條件 and 處理
????????????????????????????plainSelect.setWhere(new?AndExpression(plainSelect.getWhere(), inExpression));
????????????????????????}
????????????????????}?else?if?("mobile".equals(dataColumn.getName())) {
????????????????????????// 支持一個(gè)自定義條件
????????????????????????LikeExpression likeExpression =?new?LikeExpression();
????????????????????????likeExpression.setLeftExpression(new?Column(dataColumn.getAliasDotName()));
????????????????????????likeExpression.setRightExpression(new?StringValue("%1533%"));
????????????????????????plainSelect.setWhere(new?AndExpression(plainSelect.getWhere(), likeExpression));
????????????????????}
????????????????}
????????????}
????????}
????};
}

SELECT?u.*?FROM?user?u?
??WHERE?(u.department_id?IN?('1',?'2',?'3',?'5'))?
??AND?u.mobile?LIKE?'%1533%'?LIMIT?1,?10