Java ISelectable类代码示例

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
private void replaceAliasInFilter(Object filter, String alias) {
    if (filter instanceof IFunction) {
        for (Object sub : ((IFunction) filter).getArgs()) {
            this.replaceAliasInFilter(sub, alias);
        }
    }

    if (filter instanceof ISelectable) {
        if (alias != null) {
            ((ISelectable) filter).setTableName(alias);
        }

        if (((ISelectable) filter).getAlias() != null) {
            ((ISelectable) filter).setColumnName(((ISelectable) filter).getAlias());
            ((ISelectable) filter).setAlias(null);
        }
    }

}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
private void buildColumnRefered() {
    List<ISelectable> columnRefered = new ArrayList();
    columnRefered.addAll(this.getNode().getColumnsSelected());

    for (IBooleanFilter f : this.getNode().getJoinFilter()) {
        ISelectable left = (ISelectable) f.getColumn();
        ISelectable right = (ISelectable) f.getValue();

        if (!columnRefered.contains(left)) {
            columnRefered.add(left);
        }

        if (!columnRefered.contains(right)) {
            columnRefered.add(right);
        }
    }

    for (IColumn c : this.columnInAggregate) {
        if (!columnRefered.contains(c)) {
            columnRefered.add(c);
        }
    }

    this.getNode().setColumnsRefered(columnRefered);
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
@Override
public ISelectable getSelectableFromChild(ISelectable c) {
    // 如果存在表名，则进行强校验，比如字段为A.ID，否则直接进行ID名字匹配
    if (c.getTableName() != null && !c.getTableName().equals(this.getNode().getTableName())
        && !c.getTableName().equals(this.getNode().getAlias())) {
        return null;
    }

    if (IColumn.STAR.equals(c.getColumnName())) {
        return c;
    }

    if (c instanceof IFunction) {
        c.setTableName(this.getNode().getTableName());
        return c;
    }

    if (SchemaManager.DUAL.equals(getNode().getTableName())) {
        throw new TddlRuntimeException(ErrorCode.ERR_PARSER, "can't not visit column in dual");
    }
    return this.getSelectableFromChild(c.getColumnName());
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
protected ISchematicCursor executeAgg(ISchematicCursor cursor, IDataNodeExecutor executor,
                                      boolean closeResultCursor, IRepository repo, List<IFunction> aggregates,
                                      List<IOrderBy> groupBycols, ExecutionContext executionContext)
                                                                                                    throws TddlException {
    List<ISelectable> _retColumns = null;
    if (executor instanceof IQuery) {
        _retColumns = ((IQuery) executor).getColumns();
    } else if (executor instanceof IJoin) {
        _retColumns = ((IJoin) executor).getColumns();
    } else if (executor instanceof IMerge) {
        _retColumns = ((IMerge) executor).getColumns();
    }

    if (_retColumns != null) {
        if ((aggregates != null && !aggregates.isEmpty()) || (groupBycols != null && !groupBycols.isEmpty())) {
            cursor = repo.getCursorFactory().aggregateCursor(executionContext,
                cursor,
                aggregates,
                groupBycols,
                _retColumns,
                false);
        }
    }
    return cursor;
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
private void initSchema(IRowSet firstRowSet, IFilter lf) {
    if (schemaInited) {
        return;
    }
    schemaInited = true;

    List<ColumnMeta> columnMetas = to.getParentCursorMeta().getColumns();
    List<ISelectable> iColumns = ExecUtils.getIColumnsWithISelectable(columnMetas.toArray(new ColumnMeta[0]));
    toComparator = ExecUtils.getComp(iColumns,
        iColumns,
        to.getParentCursorMeta(),
        firstRowSet.getParentCursorMeta());
    fromComparator = ExecUtils.getComp(iColumns,
        iColumns,
        from.getParentCursorMeta(),
        firstRowSet.getParentCursorMeta());
    // init(rangeFilters);
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
private boolean findOrderByInKey(List<IOrderBy> orderBys, List<ColumnMeta> columns, ColumnMeta cm,
                                 Set<IOrderBy> hashOrderBys) {
    for (IOrderBy ob : orderBys) {
        ISelectable iSelectable = ob.getColumn();
        String orderByTable = iSelectable.getTableName();

        orderByTable = ExecUtils.getLogicTableName(orderByTable);

        if (cm != null && TStringUtil.equals(ExecUtils.getLogicTableName(cm.getTableName()), orderByTable)) {
            if (TStringUtil.equals(cm.getName(), iSelectable.getColumnName())) {
                ColumnMeta cm2 = new ColumnMeta(cm.getTableName(),
                    cm.getTableName() + "_ANDOR_TABLENAME_" + cm.getName(),
                    cm.getDataType(),
                    cm.getAlias(),
                    cm.isNullable());
                // 列名与order by Match.放到key里
                columns.add(cm2);
                hashOrderBys.add(ob);
                return true;
            }
        }
    }
    return false;
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
public static List<ISelectable> copySelectables(List<ISelectable> selects, String oldTableName, String tableName) {
    if (tableName == null) {
        return copySelectables(selects);
    }

    if (selects == null) {
        return null;
    }

    List<ISelectable> news = new ArrayList(selects.size());
    for (ISelectable s : selects) {
        ISelectable a = s.copy();
        if (a instanceof IColumn) {
            setColumn((IColumn) a, oldTableName, tableName);
        } else if (a instanceof IFilter) {
            setFilter((IFilter) a, oldTableName, tableName);
        } else if (a instanceof IFunction) {
            setFunction((IFunction) a, oldTableName, tableName);
        }

        news.add(a);
    }

    return news;
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
@Override
public DataType getReturnType() {
    Object[] args = function.getArgs().toArray();
    DataType type = null;
    if (args[0] instanceof ISelectable) {
        type = ((ISelectable) args[0]).getDataType();
    }
    if (type == null) {
        type = DataTypeUtil.getTypeOfObject(args[0]);
    }

    if (type == DataType.BigIntegerType || type == DataType.BigDecimalType) {
        return DataType.BigIntegerType;
    } else {
        return DataType.LongType;
    }
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
@Override
public void visit(FunctionExpression node) {
    boolean argDistinct = isDistinct(node);
    IFunction ifunc = ASTNodeFactory.getInstance().createFunction();
    ifunc.setFunctionName(node.getFunctionName());
    List<Expression> expressions = node.getArguments();

    if (expressions != null) {
        ArrayList<Object> args = new ArrayList<Object>();
        for (Expression expr : expressions) {
            MySqlExprVisitor v = new MySqlExprVisitor(parent);
            expr.accept(v);
            Object cv = v.getColumnOrValue();
            if ((cv instanceof ISelectable)) {
                ((ISelectable) cv).setDistinct(argDistinct);
            }
            args.add(v.getColumnOrValue());
        }
        ifunc.setArgs(args);
    } else {
        ifunc.setArgs(new ArrayList());
    }

    ifunc.setColumnName(getSqlExprStr(node));
    columnOrValue = ifunc;
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
/**
 * 列的tablename会设为表别名
 * 
 * @return
 */
public List<ISelectable> getColumnsSelectedForParent() {
    List<ISelectable> res = new ArrayList<ISelectable>(this.getColumnsSelected().size());
    for (ISelectable s : this.getColumnsSelected()) {
        ISelectable a = s.copy();
        if (this.getAlias() != null) {
            a.setTableName(this.getAlias());
        }

        if (s.getAlias() != null) {
            a.setColumnName(s.getAlias()); // 设置为alias name
            a.setAlias(null);
        }
        res.add(a);
    }
    return res;
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
public void buildFunction(IFunction f, boolean findInSelectList) {
    if (FunctionType.Aggregate == f.getFunctionType()) {
        setExistAggregate();
    }

    if (f.getArgs().size() == 0) {
        return;
    }

    List<Object> args = f.getArgs();
    for (int i = 0; i < args.size(); i++) {
        if (args.get(i) instanceof ISelectable) {
            args.set(i, this.buildSelectable((ISelectable) args.get(i), findInSelectList));
        }
    }
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
public DataType getSumType() {
    Object[] args = function.getArgs().toArray();
    DataType type = null;
    if (args[0] instanceof ISelectable) {
        type = ((ISelectable) args[0]).getDataType();
    }

    if (type == null) {
        type = DataTypeUtil.getTypeOfObject(args[0]);
    }
    if (type == DataType.IntegerType || type == DataType.ShortType) {
        return DataType.LongType;
    } else {
        return type;
    }
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
private static void pushJoinOrder(List<IOrderBy> orders, QueryTreeNode qn, boolean isUedForIndexJoinPK) {
    if (orders != null && !orders.isEmpty()) {
        for (IOrderBy order : orders) {
            if (qn.hasColumn(order.getColumn())) {
                qn.orderBy(order.getColumn(), order.getDirection());
            } else if (isUedForIndexJoinPK) {
                // 尝试忽略下表名查找一下
                ISelectable newC = order.getColumn().copy().setTableName(null);
                if (qn.hasColumn(newC)) {
                    qn.orderBy(order.getColumn(), order.getDirection());
                }
            }
        }

        qn.build();
    }
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
/**
 * 判断insert..select的分区规则是否一致
 */
private static PartitionJoinResult isInsertSelectOnPartition(QueryTreeNode left, QueryTreeNode right) {
    PartitionJoinResult leftResult = isJoinOnPartitionOneSide(new ArrayList<ISelectable>(), left);
    if (!leftResult.flag) {
        return leftResult;
    }

    PartitionJoinResult rightResult = isJoinOnPartitionOneSide(new ArrayList<ISelectable>(), right);
    if (!rightResult.flag) {
        return rightResult;
    }

    PartitionJoinResult result = new PartitionJoinResult();
    result.broadcast = leftResult.broadcast || rightResult.broadcast;
    result.flag = StringUtils.equalsIgnoreCase(leftResult.joinGroup, rightResult.joinGroup) || result.broadcast;
    result.joinGroup = leftResult.joinGroup;
    result.joinFilters = null;
    return result;
}
 

import com.taobao.tddl.optimizer.core.expression.ISelectable; //导入依赖的package包/类
private static boolean tryPushColumn(IFilter filter, boolean isColumn, QueryTreeNode qtn) {
    Object value = null;
    if (isColumn) {
        value = ((IBooleanFilter) filter).getColumn();
    } else {
        value = ((IBooleanFilter) filter).getValue();
    }

    if (value instanceof ISelectable) {
        ISelectable c = qtn.findColumn((ISelectable) value);
        if (c instanceof IColumn) {
            if (isColumn) {
                ((IBooleanFilter) filter).setColumn(c.copy());
            } else {
                ((IBooleanFilter) filter).setValue(c.copy());
            }
            return true;
        } else {
            return false;
        }
    } else {
        return true;
    }
}