OLTP場景數據分布式實踐- 技術篇

之前設計篇講了數據拆分的方式、場景、優缺點以及實施步驟,偏方法與理論。技術篇會介紹分布式數據服務平臺設計與實現,講述如何通過技術手段解決數據拆分帶來的各種問題,以及各中間件的架構與原理。
平臺主要包括分布式數據訪問中間件(SDK、Proxy)、平滑擴容、數據集成、管控平臺等四部分。

一、分布式數據訪問中間件

數據拆分后,分散在多個庫與表中,但應用開發時怎樣才能準確訪問數據庫,換言之,如何才能拿到準確的數據庫連接,拼接出正確的sql(主要是實際表名),然后執行返回結果集呢?
為了盡可能減少業務侵入性,應用少做改造,往往都會抽象出一個數據訪問層負責上述功能。數據訪問層按實現方式不同,可分為應用自定義、數據中間件、分布式數據庫三種方式,在我們項目中采用的是中間件方式,其技術架構如下:


分布式數據訪問層

按照接入方式不同,數據訪問中間件可以分為SDK、Proxy(云原生架構下可能還會有sidecar方式)。

一個典型的分庫分表中間件由JDBC接口實現(SDK模式)、MySQL報文解析(Proxy、Sider模式)、SQL解析器,路由計算、SQL重寫, SQL執行、聚合處理、結果集合并、數據源管理、配置管理等部分構成。

JDBC接口實現

JDBC接口實現起來并不太難,數據庫連接池都是基于此實現,本質上就是一種裝飾器模式,主要就是java.sql與javax.sql包下DataSource、Connection、Statement,PreparedStatement,ResultSet、DatabaseMetaData、ResultSetMetaData等接口。這些接口也并不是都需要實現,不常用的接口可在集成一些框架時根據需要再實現。

MySQL報文解析

MySQL報文解析比JDBC接口復雜些,它包含了很多MySQL的命令,需要對照MySQL報文規范分別進行解析,另外由于proxy還要支持常見DBA工具接入,比如MySQL CLI、Navicat、Dbvisualizer、MySQL workbench等,這些工具甚至不同版本使用的MySQL報文都不完全一樣,這塊的兼容性也是一個繁瑣的工作,考驗對Mysql報文的支持的完整度。這部分像Sharding-Proxy、Mycat等都有實現,如果要自行研發或者擴展優化,可參考其實現細節。

SQL解析

SQL解析是個繁瑣復雜的活兒,對應就是詞法Lexer與語法分析Parser,因為要最大程度兼容各數據庫廠商SQL,這塊是需要不斷的迭代增強的。開源的手寫解析器有阿里開源的druid,也可以使用javacc、antlr等進行實現,相比手寫解析器速度要慢些,但擴展定制化能力更好。這類解析器在使用方式上,多采用vistor設計模式,如果需要可以編寫自己的vistor從而獲取所需AST(Abstract Syntax Tree)中的各類值。

路由計算

路由計算是根據SQL解析后AST,提取分庫分表列值(提取規則是預先配置好的),然后根據應用指定的運算表達式或者函數進行計算,分別得到數據庫與表對應的序號(一般就是一個整型數值,類似一個數組下標)或者是真正的物理表名。讀寫分離模式下,只涉及庫路由,會根據一個負載均衡算法選取一個合適的物理庫,如果寫SQL則會選擇主庫,如果是讀則會按照隨機、輪詢或者權重等算法選擇一個從庫。

SQL重寫

SQL重寫主要為表名添加后綴(應用寫SQL時是邏輯表名,實際表名往往是邏輯表名+序號),根據路由計算環節得到的物理表名,替換原SQL中的邏輯表名。另外SQL中有聚合函數、多庫表分頁等操作時,也會涉及到對SQL的改寫,這部分有的開源中間件里也叫做SQL優化。注意這里最好不要簡單的用字符串匹配去替換表名,例如當存在列名與表名一樣的情況下會出現問題。

SQL執行

SQL執行負責SQL的真正執行,對應的就是執行連接池或數據庫驅動中Statement的execute、executeQuery、executeUpdate、executeBatch等方法。當然如果是涉及到多庫多表的SQL,例如where條件不包含分庫分表鍵,這時會涉及到庫表掃描,則需要考慮是連接優先還是內存優先,即采用多少個并發數據庫連接執行,連接數太大則會可能耗盡連接池,給內存以及數據庫帶來很大壓力;但連接數太小則會拉長SQL執行時間,很有可能帶來超時問題,所以一個強大的SQL執行器還會根據SQL類型、數據分布、連接數等因素生成一個到合適的執行計劃。

數據源管理

數據源管理負責維護各數據庫的連接,這塊實現起來比較簡單,一般維護一個數據庫連接池DataSource對象的Map就可以,只要根據數據源下標或者名稱可以拿到對應的數據庫連接即可。

聚合處理

聚合處理負責對聚合類函數的處理,因為分庫分表后,實際執行的SQL都是面向單庫的,而對于max、min、sum、count、avg等聚合操作,需要將各單庫返回的結果進行二次處理才能計算出準確的值,例如max、min、sum、count需要遍歷個各庫結果,然后分別取最大、最小、累加,對于avg操作,還需要將原SQL修改為select sum,count,然后分別累加,最后用累積后的sum除以累加后count才能得到準確值。另外對于多庫表的分頁操作,例如limit 1,10,則將單庫SQL的起始頁都修改為第一頁即limit 0,10,然后再整體排序取出前10個才是正確的數據。

結果集合并

結果集合并負責將多個SQL執行單元返回的數據集進行合并,然后返回給調用客戶端。一般當進行庫表遍歷、或者涉及多個庫SQL(例如使用in
時)會需要進行合并。當然并不一定需要把數據全部讀到內存再合并,有時基于數據庫驅動實現的ResultSet.next()函數,逐條從數據庫獲取數據即可滿足要求。關于結果集合并,sharding-jdbc對此有一個更豐富的抽象與分類,支持流式歸并、內存歸并、分組歸并等,具體可參見歸并引擎

配置管理

配置管理負責分庫分表的規則以及數據源的定義,這塊是面向應用開發者的,在使用體驗上應當簡單、易用、靈活。其中會涉及到物理數據源(參數跟連接池類似)、邏輯表、路由規則(庫路由、表路由,庫表分布,支持指定java函數或者groovy表達式),邏輯表->路由規則的映射關系。另外我們在實踐時還包括了一些元數據信息,包括shardID->庫表序號,這樣做有個好處,業務在配置路由規則時只需要關注業務對象->shardID即可。配置管理在具體形式方面,可以支持xml、yaml、也支持在管控平臺上在線進行配置,后者會通過將配置同步到配置中心,進而支持數據訪問層進行編排(orchestration),例如在線擴容時需要動態增加數據源、修改路由規則、元數據信息等。

一個完整的分布式數據訪問中間件,在架構上和數據庫的計算層很像,尤其如果涉及到DB協議報文與SQL的解析,還是一個復雜和工作量較大的工程,因此一般應用團隊建議還是采用開源成熟的方案,基于此做定制優化即可,沒必要重復造輪子。

SDK和Proxy方式各有優缺點,在我們項目中分別用在不同的場景,簡單總結如下:

  • 聯機交易 高頻、高并發,查詢帶拆分鍵,數據量小,sdk方式
  • 運維 低頻、查詢條件靈活,數據量大,以查詢為主 proxy方式
  • 批量 不攜帶分庫分表列,數據量大,查詢、更新、插入、刪除都有,通過API指定庫表方式。

接下來介紹下我們在開源中間件方面的實踐,分為三個階段:

第一階段

早些年這類開源中間件還挺多,但其實都沒有一個穩定的社區支持。2015年時我們基于一個類似TDDL的組件,對其事務、數據連接池、SQL解析等方面進行了優化,修復了數十個開發遇到的bug,實現SDK版本的數據訪問中間件,暫就叫做DAL。

第二階段

2017年,系統上線后發現,開發測試以及運維還需要一個執行分庫分表SQL的平臺,于是我們調研了Mycat,但當時1.6版本只支持單維度拆分(單庫內分表或者只分庫),因此我們重寫了其后端SQL路由模塊,結合原SDK版本數據組件,利用Mycat的報文解析實現了Proxy的數據訪問層。

Proxy模式的數據訪問層上線后,可以很好的應對帶分庫分表鍵的SQL操作,但在涉及到庫表遍歷時,由于并發連接太多,經常會導致連接數不夠,但如果串行執行則經常導致執行時間太長,最后超時報錯。針對這個問題,我們做了個新的優化:
在將這類庫表遍歷的查詢在生成執行計劃時,通過union all進行了改寫,類似map-reduce,同一庫上的不同表的sql通過union all合并,然后發到數據庫執行,這樣連接數=物理數據庫總數,同時盡可能的利用了數據庫的計算能力,在損耗較少連接數的前提下,大大提升了這類SQL的執行效率。(注意order by 和limit需要加在union all的最后,為了不影響主庫,可以將這類查詢在從庫執行)。例如user表拆分成1024表,分布在4個庫,SQL拆分與合并示意圖如下:

通過union all實現庫表遍歷

第三階段

這兩個中間件在運行3年左右后,也暴露出來了很多問題,例如SQL限制太多,兼容性太差,開源社區不活躍,部分核心代碼設計結構不夠清晰等,這給后續更復雜場景的使用帶來了很多桎梏。因此在19年,我們決定對數據訪問層進行升級重構,將底層分庫分表組件與上層配置、編排進行剝離,改成插拔式設計,增加更加多元的分庫分表組件。在那時開源社區已經涌現了一些優秀的分庫分表項目,目前來看做的最好的就是shardingshpere(后面簡稱ss)了,ss的設計與使用手冊其官網都有詳細介紹,這里主要簡單介紹下我們集成ss的一些實踐。

shardingsphere整體設計架構清晰,內核各個引擎設計職責明確,jdbc 與proxy版本共享內核,接入端支持的多種實現方式。治理、事務、SQL解析器分別單獨抽象出來,都可以hook方式進行集成,通過SPI進行擴展。這種靈活的設計也為我們定制帶來了很大的方便,代碼實現上比較優雅。我們在集成時開始是3.0.0版本,后來升級到4.0.0-RC1版本,目前ss已發布4.0.0的release版本。

1. 配置兼容

因為要在上層應用無感知的情況下更換底層分庫分表引擎,所以改造的第一個問題就是兼容以前的配置?;诖?,也就無法直接使用sharding-jdbc的spring或者yaml配置方式,而改用API方式,將原配置都轉換為sharding-jdbc的配置對象。這塊工作量時改造里最大的,但如果項目之前并沒有分庫分表配置,則直接在sharding-jdbc提供的方式中選擇一種即可。由于我們項目中需要支持規則鏈、讀權重等ss不支持功能,所以我們是基于ComplexKeysShardingAlgorithm接口進行的實現。
更簡潔的yaml配置形式:

ds:
  master_0:
    blockingTimeoutMillis: 5000
    borrowConnectionTimeout: 30
    connectionProperties: {}
    idleTimeoutMinutes: 30
    jdbcUrl: jdbc:mysql://localhost:3306/shard_0
    logAbandoned: false
    maintenanceInterval: 60
    maxConn: 10
    maxIdleTime: 61
    minConn: 1
    userName: root
    password: 123456
    queryTimeout: 30
    testOnBorrow: false
    testOnReturn: false
    testQuery: null
    testWhileIdle: true
    timeBetweenEvictionRunsMillis: 60000
  master_1:
    jdbcUrl: jdbc:mysql://localhost:3306/shard_1
    parent: master_0
groupRule: null
shardRule:
  bindingTables: 
  - user,name 
  rules:
    userTableRule:
      dbIndexs: master_0,master_1
      dataNodes: master_0.user_${['00','01']},master_1.user_${['02','03']}
      dbRules:
       - return test.dal.jdbc.shardingjdbc.YamlShardRuleParser.parserDbIndex(#user_id#)
       - return test.dal.jdbc.shardingjdbc.YamlShardRuleParser.parserDbIndexByName(#name#,#address#)
      tbRules:
       - return test.dal.jdbc.shardingjdbc.YamlShardRuleParser.parserTbIndex(#user_id#)
       - return test.dal.jdbc.shardingjdbc.YamlShardRuleParser.parserTbIndexByName(#name#,#address#)
       
  tableRuleMap: {name: nameTableRule, user: userTableRule}
2. 事務級別

sharding-jdbc的默認事務是local,即最大努力一階段提交,或者叫鏈式提交,這種方式的好處是對應用透明,性能也還不錯,互聯網中使用較多。但這種方式可能會由于網絡等原因導致部分提交成功,部分失敗。雖然這種概率可能并不高,但一旦出現則會產生事務不一致的問題,這在金融關鍵場景下風險是很高的。所以我們在聯機交易場景下禁止使用這種方式,而是要求必須嚴格單庫事務,我們在先前SDK版本的數據訪問中間件增加了校驗,一旦跨庫就直接拋異常。因此切換到sharding-jdbc,這種事務級別也要繼續支持。實現代碼片段:

/**
 * Single DB Transaction Manager
 *  SPI:  org.apache.shardingsphere.transaction.spi.ShardingTransactionManager
 */
@NoArgsConstructor
public class SingleDBTransactionManager implements ShardingTransactionManager {
  private Map<String, DataSource> dataSources = new HashMap<String, DataSource>();
  private ThreadLocal<String> targetDataSourceName = new ThreadLocal<String>() {
    protected String initialValue() {
      return null;
    }
  };
  private ThreadLocal<Connection> connection = new ThreadLocal<Connection>() {
    protected Connection initialValue() {
      return null;
    }
  };
  private ThreadLocal<Boolean> autoCommitted = new ThreadLocal<Boolean>() {
    protected Boolean initialValue() {
      return true;
    }
  };
  @Override
  public void close() throws Exception {
    if (connection.get() != null) {
      connection.get().close();
    }
  }
  @Override
  public void init(DatabaseType databaseType, Collection<ResourceDataSource> resourceDataSources) {
    for (ResourceDataSource res : resourceDataSources) {
      dataSources.put(res.getOriginalName(), res.getDataSource());
    }
  }
  @Override
  public TransactionType getTransactionType() {
    return TransactionType.SINGLEDB;
  }
  @Override
  public Connection getConnection(String dataSourceName) throws SQLException {
    if (!ConditionChecker.getInstance().isMultiDbTxAllowed() && targetDataSourceName.get() != null
        && !targetDataSourceName.get().equals(dataSourceName)) {
      throw new TransactionException(
          "Don't allow multi-db transaction currently.previous dataSource key="
              + targetDataSourceName.get() + ", new dataSource key=" + dataSourceName);
    }
    targetDataSourceName.set(dataSourceName);
    if (connection.get() == null) {
      connection.set(dataSources.get(dataSourceName).getConnection());
    }
    return connection.get();
  }
…
}
3. 讀庫權重

雖然多個從庫(一個主一般都要掛兩個或者三個從,從庫的數量由RPO、多活甚至監管要求等因素決定)可以提供讀功能,但細分的話,這些從庫其實是有“差別”的,這種差異性有可能是由于機器硬件配置,也可能是由于所在機房、網絡原因導致,這種時候就會需要支持讀權限的權重配置,例如我們項目中有單元化的設計,需要根據當前所在單元及權重配置路由到當前機房的從庫。另外也可以通過調整權重,支持在線對數據庫進行維護或者升級等運維操作。實現代碼片段:

/**
 * Weight based slave database load-balance algorithm.
 * SPI: org.apache.shardingsphere.spi.masterslave.MasterSlaveLoadBalanceAlgorithm
 */
public final class WeightMasterSlaveLoadBalanceAlgorithm implements MasterSlaveLoadBalanceAlgorithm {
  public final static String TYPE = "WEIGHT";
    
  protected DataSource dataSource;
  public WeightMasterSlaveLoadBalanceAlgorithm(DataSource ds) {
    this.dataSource = ds;
  }
  
  public  WeightMasterSlaveLoadBalanceAlgorithm(){
    
  }
    @Override
    public String getDataSource(final String name, final String masterDataSourceName, final List<String> slaveDataSourceNames) {
      String selectReadDb = dataSource.getTableRuleContext().getGroupRule(name).selectReadDb();
      return slaveDataSourceNames.contains(selectReadDb) ? selectReadDb : null;
    }
  @Override
  public String getType() {
    return TYPE;
  }
4. SQL開關

SDK模式的數據訪問中間件,主要用在聯機交易中,在這類場景下,是沒有DDL操作需求的,也是不允許的,但shading-jdbc作為一個通用的數據分片中間件。對此并沒有相應的開關配置,因此我們增加開關功能,應用在默認情況下,對DDL、DCL等語句進行了校驗,不允許執行該類SQL,在技術層面杜絕了應用的誤用。實現代碼片段:

//SPI: org.apache.shardingsphere.core.parse.hook.ParsingHook
public class AccessPrevilegeCheckHook implements ParsingHook {
  @Override
  public void start(String sql) {

  }
  @Override
  public void finishSuccess(SQLStatement sqlStatement, ShardingTableMetaData shardingTableMetaData) {
    ConditionChecker.getInstance().checkDdlAndDcl(sqlStatement);
  }
…
}

//SPI:org.apache.shardingsphere.core.rewrite.hook.RewriteHook
@NoArgsConstructor
public class TableScanCheckHook implements RewriteHook {
  
  private List<TableUnit> tableUnits = new LinkedList<TableUnit>();
  @Override
  public void start(TableUnit tableUnit) {
    if(tableUnits.size() > 0 && !ConditionChecker.getInstance().isTableScanAllowed()){
      throw new RouteException("Don't allow table scan.");
    }
    tableUnits.add(tableUnit);  
  }
…
}

public class ConditionChecker {
  private static ThreadLocal<SQLType> sqlTypeSnapshot = new ThreadLocal<SQLType>();
  private boolean defalutTableScanAllowed = true;
  private boolean defalutMultiDbTxAllowed = true;
  private boolean defalutDdlAndDclAllowed = true;
  private static ConditionChecker checker = new ConditionChecker();
  public static ConditionChecker getInstance() {
    return checker;
  }
  private ConditionChecker() {
  }
  private ThreadLocal<Boolean> tableScanAllowed = new ThreadLocal<Boolean>() {
    protected Boolean initialValue() {
      return defalutTableScanAllowed;
    }
  };
  private ThreadLocal<Boolean> multiDbTxAllowed = new ThreadLocal<Boolean>() {
    protected Boolean initialValue() {
      return defalutMultiDbTxAllowed;
    }
  };
  private ThreadLocal<Boolean> ddlAndDclAllowed = new ThreadLocal<Boolean>() {
    protected Boolean initialValue() {
      return defalutDdlAndDclAllowed;
    }
  };
  public void setDefaultCondtion(boolean tableScanAllowed, boolean multiDbTxAllowed, boolean ddlAndDclAllowed) {
    defalutTableScanAllowed = tableScanAllowed;
    defalutMultiDbTxAllowed = multiDbTxAllowed;
    defalutDdlAndDclAllowed = ddlAndDclAllowed;
  }
  public boolean isTableScanAllowed() {
    return tableScanAllowed.get();
  }
  public void setTableScanAllowed(boolean tableScanAllowed) {
    this.tableScanAllowed.set(tableScanAllowed);
  }
  public boolean isMultiDbTxAllowed() {
    return multiDbTxAllowed.get();
  }
  public void setMultiDbTxAllowed(boolean multiDbTxAllowed) {
    this.multiDbTxAllowed.set(multiDbTxAllowed);
  }
  public boolean isDdlAndDclAllowed() {
    return ddlAndDclAllowed.get();
  }
  public void setDdlAndDclAllowed(boolean ddlAllowed) {
    this.ddlAndDclAllowed.set(ddlAllowed);
  }
  public SQLType getSqlTypeSnapshot() {
    return sqlTypeSnapshot.get();
  }
  public void checkTableScan(boolean isTableScan) {
    if (!isTableScanAllowed())
      throw new ConditionCheckException("Don't allow table scan.");
  }
  public void checkDdlAndDcl(SQLStatement sqlStatement) {
    sqlTypeSnapshot.set(sqlStatement.getType());
    if (!isDdlAndDclAllowed()
        && (sqlStatement.getType().equals(SQLType.DDL) || sqlStatement.getType().equals(SQLType.DCL))) {
      throw new ConditionCheckException("Don't allow DDL or DCL.");
    }
  }
  public void checkMultiDbTx(Map<String, Connection> cachedConnections, String newDataSource) {
    if (!isMultiDbTxAllowed() && cachedConnections.size() > 0 && !cachedConnections.containsKey(newDataSource)) {
      throw new ConditionCheckException("Don't allow multi-db transaction currently.old connection key="
          + cachedConnections.keySet() + "new connection key=" + newDataSource);
    }
  }
}
5. 路由規則鏈

在我們項目中,對于一張表,在不同場景下可能會使用不同的分庫分表列,例如有的是賬號、有的是客戶號(這兩列都可路由到同一庫表中),這時候就需要路由模塊可以依次匹配搭配多個規則,例如SQL中有賬號則用account-rule,有客戶號則用customer-rule,因此我們支持了規則鏈配置功能,但sharding-jdbc只支持配置一個路由規則,因此在自定義路由算法函數中,我們增加了對規則鏈的支持。實現代碼片段:

public abstract class ChainedRuleShardingAlgorithm implements ComplexKeysShardingAlgorithm {
  protected final DataSource dataSource;
  public ChainedRuleShardingAlgorithm(DataSource ds) {
    this.dataSource = ds;
  }
  @Override
  public Collection<String> doSharding(Collection availableTargetNames, ComplexKeysShardingValue shardingValue) {
    List<String> targets = new ArrayList<String>();
    Set<String> actualNames = HintManager.isDatabaseShardingOnly() ? getHintActualName(shardingValue)
        : calculateActualNames(shardingValue);
    for (String each : actualNames) {
      if (availableTargetNames.contains(each)) {
        targets.add(each);
      }
    }
    clear();
    return targets;
  }
  @SuppressWarnings({ "serial", "unchecked" })
  protected Set<String> calculateActualNames(ComplexKeysShardingValue shardingValue) {
    Set<String> target = new HashSet<String>();
    Map<String/* table */, Map<String/* column */, Collection/* value */>> shardingMap = new HashMap<String, Map<String, Collection>>();
    String logicalTableName = shardingValue.getLogicTableName();
    Map<String, Collection> shardingValuesMap = shardingValue.getColumnNameAndShardingValuesMap();
    for (final Entry<String, Collection> entry : shardingValuesMap.entrySet()) {
      if (shardingMap.containsKey(logicalTableName)) {
        shardingMap.get(logicalTableName).put(entry.getKey(), entry.getValue());
      } else {
        shardingMap.put(logicalTableName, new HashMap<String, Collection>() {
          {
            put(entry.getKey(), entry.getValue());
          }
        });
      }
    }
    // 遍歷規則鏈,查詢匹配規則
    for (String tableName : shardingMap.keySet()) {
      RuleChain ruleChain = dataSource.getTableRuleContext().getRuleChain(tableName);
      for (GroovyListRuleEngine engine : getRuleEngine(ruleChain)) {
        Set<String> parameters = engine.getParameters();
        Map<String, Collection> columnValues = shardingMap.get(tableName);
        Set<String> eval = eval(columnValues, parameters, engine, ruleChain);
        if (eval.size() > 0) {// 匹配即中止
          target.addAll(eval);
          return target;
        }
      }
    }
    return target;
  }
  @SuppressWarnings("unchecked")
  protected Set<String> eval(final Map<String, Collection> columnValues, Set<String> parameters,
      GroovyListRuleEngine engine, RuleChain ruleChain) {
    Set<String> targetNames = new HashSet<String>();
    if (columnValues.keySet().containsAll(parameters)) {// 匹配
      List<Set<Object>> list = new LinkedList<Set<Object>>();// 參數集合
      List<String> columns = new LinkedList<String>();// 列名集合
      for (final String requireParam : parameters) {
        list.add(convertToSet(columnValues.get(requireParam)));
        columns.add(requireParam);
      }
      Set<List<Object>> cartesianProduct = Sets.cartesianProduct(list);
      for (List<Object> values : cartesianProduct) {
        Map<String, Object> arugmentMap = createArugmentMap(values, columns);
        int index = engine.evaluate(arugmentMap);
        targetNames.add(getActualName(ruleChain, index));
      }
    }
    return targetNames;
  }
  private Set<Object> convertToSet(final Collection<Object> values) {
    return Sets.newLinkedHashSet(values);
  }
  private Map<String, Object> createArugmentMap(List<Object> values, List<String> columns) {
    HashMap<String, Object> map = new HashMap<String, Object>();
    for (int i = 0; i < columns.size(); i++) {
      map.put(columns.get(i).toLowerCase(), values.get(i));
    }
    return map;
  }

  protected abstract List<GroovyListRuleEngine> getRuleEngine(RuleChain ruleChain);
  protected abstract String getActualName(RuleChain ruleChain, int index);
}

/**
 * 庫路由算法
 */

public class ChainedRuleDbShardingAlgorithm extends ChainedRuleShardingAlgorithm {
  
  public ChainedRuleDbShardingAlgorithm(DataSource ds) {
    super(ds);
  }
  @Override
  protected List<GroovyListRuleEngine> getRuleEngine(RuleChain ruleChain) {
    return ruleChain.getDbRuleList();
  }
  @Override
  protected String getActualName(RuleChain ruleChain,int index) {
    //add mapping from shard metadata
    String dbIndex = dataSource.getHintSupport().getShardingDb(String.valueOf(index));
    if(StringUtils.isEmpty(dbIndex)){
      return ruleChain.getTableRule().getDbIndexArray()[index];
    }else{
      return ruleChain.getTableRule().getDbIndexArray()[Integer.valueOf(dbIndex)];
    }
    
  }
}

/**
 * 表路由算法
 */
public class ChainedRuleTableShardingAlgorithm extends ChainedRuleShardingAlgorithm {
  public ChainedRuleTableShardingAlgorithm(DataSource ds) {
    super(ds);
  }
  @Override
  protected List<GroovyListRuleEngine> getRuleEngine(RuleChain ruleChain) {
    return ruleChain.getTableRuleList();
  }
  @Override
  protected String getActualName(RuleChain ruleChain, int index) {
    //add mapping from shard metadata
    String tbShardIndex = dataSource.getHintSupport().getShardingTable(String.valueOf(index));
    int tbIndex = index;
    if(!StringUtils.isEmpty(tbShardIndex)){
      tbIndex = Integer.valueOf(tbShardIndex);
    }
    SuffixManager suffixManager = ruleChain.getTableRule().getSuffixManager();
    if(suffixManager.isInlineExpression()){
      return ruleChain.getTbIndexs()[tbIndex];
    }else{
      Suffix suffix = suffixManager.getSuffix(0);
      return String.format("%s%s%0"+suffix.getTbSuffixWidth() +"d", ruleChain.getLogicTable(),suffix.getTbSuffixPadding(), suffix.getTbSuffixFrom() + tbIndex);
    }
  }
6. 管控平臺對接

我們提供了一個管控平臺,支持分布式數據相關組件在線配置,這些通過配置中心統一下發到各應用,而且支持動態變更。不管是SDK模式還是Proxy模式的數據訪問中間件都使用的是同一份分庫分表配置,只是接入方式不同而已。因此在集成ss的時候,還需要增加從配置中心獲取配置的功能,這塊主要涉及的調用配置中心API獲取配置,這里就不貼具體代碼了。

數據訪問中間件的發展演進方向,未來其將會是多種形態的混合存在。

分布式數據訪問中間件的三種模式

二、平滑擴容

在設計篇中已經介紹了擴容的機制,簡單的說,平滑擴容就是通過異步復制,等數據接近追平后禁寫,修改路由,然后恢復業務。主要目的是自動化、以及盡可能縮短停機窗口,目前一些云產品比如阿里云DRDS、騰訊云TDSQL等的一鍵在線擴容本質上都是基于此機制。
但實踐中這個過程需要多個步驟,數據庫數量越多,操作風險越大,而且需要停機完成。為此我們與數據庫團隊一起設計與開發了平滑擴容功能。
我們將整個擴容環節,分為配置、遷移、校驗、切換、清理五個大的步驟,每個步驟里又由多個任務構成。擴容任務在管控平臺上建立,平滑擴容模塊自動依次觸發各個任務。


平滑擴容
  • 配置環節,主要是應用系統方定義擴容后的分庫分表配置;
  • 遷移環節,依次自動完成從復制,同時進行數據校驗;
  • *切換環節,首先進行禁寫,斷開主從只從,然后修改路由規則,最后再解除擴容庫禁寫。整個過程應用無需停機,僅僅會有一段時間禁寫,這個時間一般來說也就十來秒;
  • 清理環節,清理環節是在后臺異步處理,即清理數據庫冗余表。

在分布式數據服務管控平臺定義好擴容前后分庫分表配置后,即可啟動一鍵在線擴容。在擴容過程中可實時監控擴容進度,同時支持擴容中斷恢復以及回滾。

三、數據集成

微服務架構下,有大量需要數據集成的場景:

  1. 業務系統之間,例如下訂單后需要通知庫存、商家,然后還要推送到大數據等下游系統;
  2. 分庫分表后,為了應對其它維度查詢,會需要建立異構索引,這樣就需要數據傳輸到另外一套數據庫中;
  3. 系統內應用與中間件之間,例如如果使用redis等緩存,在操作完數據庫還要更新緩存,類似這類數據集成需求,最樸素的解決方式就是雙寫,但雙寫一個問題是增加了應用復雜性,另外當發生不一致的情況是難以處理。

這類問題本質上也屬于分布式事務場景,一種簡單的方式就是基于MQ可靠消息,即在應用端寫消息表,然后通過MQ消費消息進行數據集成處理。但這導致應用代碼耦合大量雙寫邏輯,給應用開發帶來很多復雜度。
針對雙寫問題,業界一種更優雅、先進的設計是基于日志的集成架構,在OLTP場景下,可以通過解析數據庫日志類似CDC,這種方式的好處是數據集成工作從應用代碼中進行了剝離。

關于雙寫以及基于日志集成架構可參考Using logs to build a solid data infrastructure (or: why dual writes are a bad idea

數據集成

這類CDC的開源軟件,java類的有shyiko、canal、debezium等。這類項目的實現原理是主要模擬從庫從主庫異步獲取日志事件,然后經過ETL發送到MQ或者其它下游系統。

如何模擬一個從庫可參照MySQL復制協議:https://dev.mysql.com/doc/internals/en/replication-protocol.html

考慮功能完整性和社區活躍度,我們選擇了基于canal構建數據集成中間件,具體工作原理這里就不介紹了,可參見canal github。這里主要介紹下我們對其做的一些定制和優化。

在線配置:

canal的配置非常繁雜,很容易配錯,所以最先開始做的就是提供了一個更簡單、易用的在線配置定義功能,用戶是只需進行一些核心關鍵的配置,例如數據庫的IP、用戶、密碼、訂閱表、MQ地址,其它不常修改的配置通過模板形式提供,大大降低了配置復雜度和工作量,當然如果需要也完全支持自定義。

性能優化

數據庫binlog是有序的,但如果寫MQ或者目標庫,仍完全保證該順序,那么則無法進行并發,這樣同步的TPS是肯定上不去的,因此如何在保證一定順序的前提下最大程度提高并發性能是一個需要結合業務場景解決的問題。

我們當時用的是canal1.1.3的版本,經過我們性能測試,數據寫入kafka的TPS也就5000+,這對于結息等大量數據變更的場景是不能滿足要求的。
另外canal寫入MQ的并發維度是表的主鍵,但我們項目中表的主鍵都是自增列(這個是我們項目中數據庫開發規范,主要目的是保證MySQL寫性能),如果根據此列進行并發控制,那么則無法保證MQ寫入時的業務順序性。例如支付流水表,如根據主鍵(自增列)則無法保證同一賬戶流水的順序性。

canal自身寫入機制

針對此問題,我們對canal進行了改造,將原來只支持根據主鍵進行并發控制,修改為支持應用指定,例如我們項目采用業務唯一鍵;原來流程是順序從canal server端讀取binlog->寫MQ->再確認->再讀取下一binlog事件;調整后改為并發讀取binlog,一旦在執行事件集中沒有當前業務唯一鍵,就可直接寫入MQ,后臺開啟一個線程,按照batchID依次進行ack,通過并行拉取binlog事件、分階段無阻塞處理,單庫數據同步kafak的TPS可以達到1.2W+,已可以滿足結息等場景。

canal寫入改造后機制
Serverless化

如果通過安裝包部署,在用戶配置完數據集成相關參數后,需要手工將canal server以及adapter包上傳至服務器上??紤]到高可用,還得在備機上進行部署,在分庫分表下,數據庫拆分成多個,需要部署多個實列到多個服務器上(canal支持同個實例部署在統一server節點,但性能會受影響)。因此我們將canal server與adapter進行了容器化改造,然后部署到了統一的k8s集群中,這樣用戶在配置完后,點擊“啟動實列”按鈕,即可在k8s環境中自動部署高可用的canal集群,從而實現了數據集成功能的serverless化。

通過數據集成中間件,可以在應用無侵入下解決分庫分表后一個很典型的問題:多維度拆分與多庫查詢。例如將分庫分表的數據再集中到一個匯總庫,然后一些復雜的查詢統計就可以放在匯總庫上;還有一些多維度拆分場景,類似電商里的賣家庫、商家庫,需要創建“二級索引”,也可以通過數據中間件自動創建;另外也可以方便實現諸如小表廣播等需要保證數據一致性的功能。

四、管控平臺

前面提到的各種中間件,涉及到大量配置定義、實例管理、監控等功能,這些功能分散在各組件內部,缺少一個統一的視圖,而且應用開發人員需要重復定義。因此我們設計開發了數據服務管控平臺,將分庫分表配置定義、在線擴容、運維、監控等功能統一集成,最大程度降低開發以及運維人員對數據拆分帶來的復雜度。同時提供開放API,可以對接目前公司已有數據庫以及云管理系統。
配置信息統一存放在配置中心,各中間件直接從配置中心拉取配置,在管控平臺修改配置后,也可以實時通知各應用進行動態加載。管控平臺相當于一個數據服務云管理平臺,提供多租戶,各應用無需自行部署,直接接入使用即可。在技術架構方面管控平臺是個前后端分離架構,前端基于vue.js,后端按照功能模塊拆分成微服務,都部署在k8s集群中。

管控平臺

五、 感受

上面介紹了我們在數據服務平臺建設中各技術組件的設計原理和實踐,限于篇幅,更多實現細節就不展開介紹。

在企業軟件這塊,有兩個不同的思路,一個是購買商業產品,一種是基于開源軟件自行構建。在金融領域,早些年以前者為主,近些年后者則變成了趨勢。開源軟件有點是開放,因為有源碼所以有自主掌控的可能與條件,缺點是不像商業產品功能完整,往往需要自行定制、優化、擴展。作為軟件開發人員,我們更喜歡使用白盒而不是黑盒。

當然開源并不代表免費,有可能付出的成本比商業軟件更高。一方面需要投入精力學習開源項目,只有熟悉源代碼后才可能具備修改定制能力;另一方面要積極關注開源界技術的發展,與時俱進,要有開放的心態,吸取開源先進的設計的東西,大膽驗證,謹慎使用。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容