[zebra源码]分组数据源GroupDataSource读写分

前面 分组数据源GroupDataSource及其初始化 已经知道 GroupDataSource 在初始化的时候会初始化两个DataSource， 读 readDataSource + 写 writeDataSource

由于zebra 也是基于 jdbc规范实现的， 所以sql执行的过程和 jdbc api最原始的api一样的

(1) 从DataSource 获取连接 #getConnection
(2) Connection 创建语句 #prepareStatement
(3) 语句的执行 Statement#executeQuery()

可以顺着上述jdbc api的执行过程查看代码执行逻辑， 发现其中读写分离的核心逻辑位于 GroupConnection#getRealConnection(sql, forceWrite) 中， 根据不同的情况 从 读 readDataSource 还是 writeDataSource 取底层连接

具体过程：

image.png

读写库判断逻辑

1. routeType 路由类型默认为 masterSlave 读写分离, 可在配置文件修改

2. forceWrite=true的场景为直接调用 Statement 的更新方法 Statement#executeUpdate 或批量执行场景
   

   image.png

3. sql hint 在sql前增加了特殊注释 /+zebra:w/ 则走写库

4. ReadWriteStrategy 则是一个预定义的java接口， 如果提供了实现类， 则会根据 #shouldReadFromMaster() 来判断强制从写库读数据， 主要用于避免 写主库然后读从库 而从库同步延迟的问题

ZebraForceMasterHelper#forceMasterInLocalContext() ****强制读主库就是基于 ReadWriteStrategy 的实现类 LocalContextReadWriteStrategy 来实现的 （内部用的ThreadLocal存储状态）

5. 最后，会直接根据 sql 字符串解析是读操作(select) 还是写操作(update/delete/select for update/ddl ..) 来从读ds还是写ds获取连接

需要debug的话，可以从单元测试入手

• LocalContextReadWriteStrategyTest
• CustomizedReadWriteStrategyWrapperTest
• com.dianping.zebra.group.jdbc.DPGroupPreparedStatementTest#test_read_and_write_perpared_statement