Java IGroupExecutor.getRemotingExecutableObject方法代码示例

import com.taobao.tddl.executor.spi.IGroupExecutor; //导入方法依赖的package包/类
public DataSource getDatasourceByGroupNode(TopologyHandler topology, String groupNode) {
    if ("undecided".equals(groupNode)) {
        return null;
    }
    IGroupExecutor matrixExecutor = topology.get(groupNode);

    if (matrixExecutor == null) return null;
    /*
     * 这里做个hack吧。 原因是在构造topologic的时候，默认全部使用mysql作为type了。
     * 这样就造成所有的Datasource都是由RemotingExecutor
     * com.taobao.ustore.jdbc.mysql.My_Reponsitory.buildRemoting(Group
     * group) 创建的，type都是mysql.
     * 但实际上这里的type需要根据实际的DataSource来决定是个oracle还是个mysql.
     * 如果是oracle那么type需要改成oracle..
     */
    Group.GroupType type = matrixExecutor.getGroupInfo().getType();
    DataSource ds = (DataSource) matrixExecutor.getRemotingExecutableObject();

    if (isNotValidateNode(type)) {
        throw new IllegalArgumentException("target node is not a validated Jdbc node");
    }

    if (ds == null) {
        throw new IllegalArgumentException("can't find ds by group name ");
    }
    return ds;
}
 

import com.taobao.tddl.executor.spi.IGroupExecutor; //导入方法依赖的package包/类
public DatabaseMetaData getDatabaseMetaData() throws SQLException {
    Connection conn = null;
    DatabaseMetaData dbMa = null;
    try {
        // 找到默认的主库
        String defaultDbIndex = this.dataSource.getConfigHolder()
            .getOptimizerContext()
            .getRule()
            .getDefaultDbIndex(null);

        // 找到对应的执行器
        IGroupExecutor groupExecutor = this.dataSource.getConfigHolder()
            .getExecutorContext()
            .getTopologyHandler()
            .get(defaultDbIndex);

        Object groupDataSource = groupExecutor.getRemotingExecutableObject();
        if (groupDataSource instanceof DataSource) {
            // 指定到主库上执行
            conn = ((DataSource) groupDataSource).getConnection();
            dbMa = conn.getMetaData();
        }
    } finally {
        if (conn != null) {
            conn.close();
        }
    }
    return dbMa;
}
 

import com.taobao.tddl.executor.spi.IGroupExecutor; //导入方法依赖的package包/类
@Override
public ISchematicCursor handle(IDataNodeExecutor executor, ExecutionContext executionContext) throws TddlException {
    ArrayResultCursor result = new ArrayResultCursor("TRACE", executionContext);
    result.addColumn("ID", DataType.IntegerType);
    result.addColumn("SCHEMA", DataType.StringType);
    result.addColumn("NAME", DataType.StringType);

    result.addColumn("GROUP", DataType.StringType);
    result.addColumn("URL", DataType.StringType);
    result.addColumn("USER", DataType.StringType);
    result.addColumn("TYPE", DataType.StringType);
    result.addColumn("INIT", DataType.StringType);
    result.addColumn("MIN", DataType.StringType);

    result.addColumn("MAX", DataType.StringType);
    result.addColumn("IDLE_TIMEOUT", DataType.StringType);
    result.addColumn("MAX_WAIT", DataType.StringType);

    result.addColumn("ACTIVE_COUNT", DataType.StringType);
    result.addColumn("POOLING_COUNT", DataType.StringType);

    result.initMeta();
    int index = 0;

    Matrix matrix = ExecutorContext.getContext().getTopologyHandler().getMatrix();
    TopologyHandler topology = ExecutorContext.getContext().getTopologyHandler();

    for (Group group : matrix.getGroups()) {
        IGroupExecutor groupExecutor = topology.get(group.getName());

        Object o = groupExecutor.getRemotingExecutableObject();

        if (o != null && o instanceof TGroupDataSource) {
            TGroupDataSource ds = (TGroupDataSource) o;

            for (DataSource atom : ds.getDataSourceMap().values()) {
                TAtomDataSource atomDs = this.getAtomDatasource(atom);
                DruidDataSource d;
                try {

                    if (!(atomDs.getDataSource().getTargetDataSource() instanceof DruidDataSource)) {
                        logger.warn("atom datasource is not druid?"
                                    + atomDs.getDataSource().getTargetDataSource().getClass());
                        continue;
                    }
                    d = (DruidDataSource) atomDs.getDataSource().getTargetDataSource();
                    result.addRow(new Object[] { index++, topology.getAppName(), d.getName(), group.getName(),
                            d.getUrl(), d.getUsername(), d.getDbType(), d.getInitialSize(), d.getMinIdle(),
                            d.getMaxActive(), d.getTimeBetweenEvictionRunsMillis() / (1000 * 60), d.getMaxWait(),
                            d.getActiveCount(), d.getPoolingCount() });

                } catch (SQLException e) {
                    logger.error("", e);
                }

            }
        }
    }

    return result;

}
 

import com.taobao.tddl.executor.spi.IGroupExecutor; //导入方法依赖的package包/类
@Override
public CallableStatement prepareCall(String sql, int resultSetType, int resultSetConcurrency,
                                     int resultSetHoldability) throws SQLException {
    checkClosed();
    // 先检查下是否有db hint
    DirectlyRouteCondition route = (DirectlyRouteCondition) SimpleHintParser.getRouteContiongFromThreadLocal(ThreadLocalString.DB_SELECTOR);
    String defaultDbIndex = null;
    if (route != null) {
        defaultDbIndex = route.getDbId();
    } else {
        // 针对存储过程，直接下推到default库上执行
        defaultDbIndex = this.ds.getConfigHolder().getOptimizerContext().getRule().getDefaultDbIndex(null);
    }

    IGroupExecutor groupExecutor = this.ds.getConfigHolder()
        .getExecutorContext()
        .getTopologyHandler()
        .get(defaultDbIndex);

    Object groupDataSource = groupExecutor.getRemotingExecutableObject();
    if (groupDataSource instanceof DataSource) {
        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(0);
        Connection conn = ((DataSource) groupDataSource).getConnection();
        CallableStatement target = null;
        if (resultSetType != Integer.MIN_VALUE && resultSetHoldability != Integer.MIN_VALUE) {
            target = conn.prepareCall(sql, resultSetType, resultSetConcurrency, resultSetHoldability);
        } else if (resultSetType != Integer.MIN_VALUE) {
            target = conn.prepareCall(sql, resultSetType, resultSetConcurrency);
        } else {
            target = conn.prepareCall(sql);
        }

        ExecutionContext context = prepareExecutionContext();
        TCallableStatement stmt = new TCallableStatement(ds, this, sql, context, target);
        openedStatements.add(stmt);
        return stmt;
    } else {
        throw new UnsupportedOperationException();
    }

}