com.sharksharding.core.shard.HorizontalFacade Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of shark Show documentation
shark mysql sharding
The newest version!
/*
 * Copyright 2015-2101 gaoxianglong
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.sharksharding.core.shard;

import java.util.List;

/**
 * 水平分库(对垂直分库后的单一业务库进行水平化),水平分片的一种关系型数据库分库分表模式
 *
 * @author gaoxianglong
 * 
 * @author 2.0.1
 */
public class HorizontalFacade extends RouteImpl {
	private DataSourceHolder dataSourceHolder;

	public HorizontalFacade() {
		dataSourceHolder = SharkDataSourceHolder.getDataSourceHolder();
	}

	@Override
	public Object[] route(String sql, Object[] params, boolean indexType) {
		String newSql = null;
		final String DB_RULE_ARRAY = shardConfigInfo.getDbRuleArray();
		final String TB_RULE_ARRAY = shardConfigInfo.getTbRuleArray();
		List keyNames = ResolveShardkey.getShardKeys(shardConfigInfo);
		if (keyNames.isEmpty())
			return null;
		/* 解析sql语句中的路由条件 */
		final long SHARD_VALUE = ResolveRouteValue.getRoute(sql, keyNames);
		Rule dbRule = getDbRule();
		int dbIndex = dbRule.getIndex(SHARD_VALUE, DB_RULE_ARRAY);
		Rule tbRule = getTbRule();
		int tbIndex = tbRule.getIndex(SHARD_VALUE, TB_RULE_ARRAY);
		/* 解析配置文件中数据库和数据库表的数量 */
		String values[] = TB_RULE_ARRAY.split("[\\%]");
		final int TB_SIZE = Integer.parseInt(values[1]);
		final int DB_SIZE = Integer.parseInt(values[2]);
		/* 解析数据库表名 */
		final String TB_NAME = ResolveTbName.getTbName(sql);
		/* 单库多表模式下设定真正的数据库表名 */
		newSql = SetTbName.setName(shardConfigInfo, dbIndex, tbIndex, DB_SIZE, TB_SIZE, TB_NAME, sql);
		final int DB_BEGIN_INDEX = ResolveIndex.getBeginIndex(shardConfigInfo.getWr_index(), indexType);
		/* 切换数据源索引 */
		SetDatasource.setIndex(dbIndex += DB_BEGIN_INDEX, dataSourceHolder);
		return updateParam(newSql, params);
	}
}