sharding sphere 4.1.1 - “路由装饰器RouteDecorator-主从” 之 “控制强制读主库”

最新推荐文章于 2023-12-19 10:08:34 发布

_jan_

最新推荐文章于 2023-12-19 10:08:34 发布

阅读量528

点赞数

分类专栏： Java 文章标签： java

本文链接：https://blog.csdn.net/zhouzhiande/article/details/114745605

版权

Java 专栏收录该内容

133 篇文章 1 订阅

订阅专栏

MasterVisitedManager.setMasterVisited();

org.apache.shardingsphere.masterslave.route.engine.MasterSlaveRouteDecorator#decorate
org.apache.shardingsphere.masterslave.route.engine.impl.MasterSlaveDataSourceRouter#route
org.apache.shardingsphere.masterslave.route.engine.impl.MasterSlaveDataSourceRouter#isMasterRoute

public final class MasterSlaveDataSourceRouter {
    
    private final MasterSlaveRule masterSlaveRule;
    
    /**
     */
    public String route(final SQLStatement sqlStatement) {
    	// 1、检查是否要执行走主库逻辑（非select、sql语句存在锁标记、强制标记走主库、是否有hint标记）
        if (isMasterRoute(sqlStatement)) {
            MasterVisitedManager.setMasterVisited();
            return masterSlaveRule.getMasterDataSourceName();
        }

        // 创建负载均衡器，选择一个从库
        return masterSlaveRule.getLoadBalanceAlgorithm().getDataSource(
                masterSlaveRule.getName(), masterSlaveRule.getMasterDataSourceName(), new ArrayList<>(masterSlaveRule.getSlaveDataSourceNames()));
    }
    
    private boolean isMasterRoute(final SQLStatement sqlStatement) {
        return containsLockSegment(sqlStatement) || !(sqlStatement instanceof SelectStatement) || MasterVisitedManager.isMasterVisited() || HintManager.isMasterRouteOnly();
    }
    
    private boolean containsLockSegment(final SQLStatement sqlStatement) {
        return sqlStatement instanceof SelectStatement && ((SelectStatement) sqlStatement).getLock().isPresent();
    }
}