Zebra 分库分表 API 介绍
优质
小牛编辑
133浏览
2023-12-01
zebra API是针对分库分表场景(ShardDataSource),提供的一套可以通过手动指定一些路由参数 的方式来改变zebra默认路由行为的API。通过API,用户可以手动指定SQL本身以外的列 或值 来实现将SQL路由到任意用户想要路由的表上。
zebra API依靠ShardDataSourceHelper类提供的static接口,来实现相关的功能。
zebra API功能有哪些
zebra API目前支持指定路由用的值。
指定用于路由的值
通过ShardDataSourceHelper.setShardParams(column, value)接口,可以设置某个shardKey的值为多少,然后这个值就可以参与到路由的计算当中。
用法场景
在我们的实际使用场景中,偶尔会有需要将某些sql(SQL不带shard key或者想指定shard key的值)发送到某张具体的表执行,这个时候就需要使用zebra的api接口,来虚拟分表值来达到这种目的。
下面我们就举例说明该功能怎么使用:
后面的例子都以下面规则为例
表名:table
分库规则:(#id#.toLong())%4 zebrasample[0-3]
分表规则:(#sid#.toLong())%2 alldb:[0,7]
批量写入
仅支持批量写入的SQL确定都在同一库同一张表上,不支持批量写入的SQL在不同的库上。
//这里指定一下shardKey和shardValue分别是多少
ShardDataSourceHelper.setShardParams(“id”, 0);
ShardDataSourceHelper.setShardParams(“sid”, "100");
// 上面两个配置,就指定了,接下来我们执行的SQL会以【id=0,sid="100"】这组规则来路由SQL,即zebrasample0库,table0表。
// **注:如果只设置了一个规则,比如id或者sid,那么这个设置会被忽略掉。
// 如果以上设置的参数需要用在同一个线程的多条SQL中,请执行以下代码,否则zebra内部会默认调用 ShardDataSourceHelper.clearAllThreadLocal() 清理
ShardDataSourceHelper.setGlobalParams(true);
//指定框架无需从SQL中获取shardKey和shardValue了,直接从用户的传入获得
ShardDataSourceHelper.setExtractParamsOnlyFromThreadLocal(true);
//业务调用batch insert
dao.batchInsert();
//调用完毕后,请清理一下threadlocal变量
ShardDataSourceHelper.clearAllThreadLocal();
自定义分表Key
有的时候,业务对某个表进行分表时,分表的键可以不存在SQL中,可以自定义一个分表键。举例来说,对于以下SQL
select * from table where name = "test"
其中在这个SQL中并没有分表键id和sid,跟上面的使用方式类型,此时可以调用zebra api来虚拟一组分表键,来指定SQL到哪个库表上执行:
//这里指定一下shardKey和shardValue分别是多少
ShardDataSourceHelper.setShardParams(“id”, 1);
ShardDataSourceHelper.setShardParams(“sid”, 1);
// 后面的SQL将在zebrasample1,table1上执行
// 如果以上设置的参数需要用在同一个线程的多条SQL中,请执行以下代码,否则zebra内部会默认调用 ShardDataSourceHelper.clearAllThreadLocal() 清理
ShardDataSourceHelper.setGlobalParams(true);
//如果dbRule或tbRule中所有的维度都是自定义传入,那么还需要调用以下API;如果还有其他数据从SQL中获得,那么无需调用。
ShardDataSourceHelper.setExtractParamsOnlyFromThreadLocal(true);
// 执行业务SQL
dao.execute();
//调用完毕后,请清理一下threadlocal变量
ShardDataSourceHelper.clearAllThreadLocal();
对于查询,除了上述用法外,还可以指定多组shardkey来执行:
//这里指定一下shardKey和shardValue分别是多少
ShardDataSourceHelper.setShardParams(“id”, Arrays.asList(1,2,3););
ShardDataSourceHelper.setShardParams(“sid”, Arrays.asList(4,5,6););
// 上面这种用法,zebra将解释为【id=1,sid=4】,【id=2,sid=5】,【id=3,sid=6】这样一一对应的三组路由规则,然后将SQL分别发送到
// zebrasample1.table2
// zebrasample2.table5
// zebrasample3.table6 这三张表里执行
// 两个分表键的值list长度不等,则以短的为准,其余的将被抛弃,比如id:[1,2,3,4], sid:[4,5],那么最后只有[1,4],[2,5]这两组值会生效
// 如果以上设置的参数需要用在同一个线程的多条SQL中,请执行以下代码,否则zebra内部会默认调用 ShardDataSourceHelper.clearAllThreadLocal() 清理
ShardDataSourceHelper.setGlobalParams(true);
//如果dbRule或tbRule中所有的维度都是自定义传入,那么还需要调用以下API;如果还有其他数据从SQL中获得,那么无需调用。
ShardDataSourceHelper.setExtractParamsOnlyFromThreadLocal(true);
// 执行业务SQL
dao.execute();
//调用完毕后,请清理一下threadlocal变量
ShardDataSourceHelper.clearAllThreadLocal();