Examples with SQLJoin com.yahoo.elide.datastores.aggregation.queryengines.sql.metadata.SQLJoin used on opensource projects

Example 1 with SQLJoin

use of com.yahoo.elide.datastores.aggregation.queryengines.sql.metadata.SQLJoin in project elide by yahoo.

the class ColumnContext method getJoinContext.

protected ColumnContext getJoinContext(String key) {
    SQLJoin sqlJoin = this.queryable.getJoin(key);
    String joinExpression = sqlJoin.getJoinExpression();
    PhysicalRefColumnContext context = PhysicalRefColumnContext.physicalRefContextBuilder().queryable(this.getQueryable()).metaDataStore(this.getMetaDataStore()).column(this.getColumn()).tableArguments(this.getTableArguments()).build();
    // This will resolve everything within join expression except Physical References, Use this resolved value
    // to create alias for join dynamically.
    String resolvedJoinExpr = context.resolve(joinExpression);
    String joinAlias = createSafeAlias(appendAlias(this.alias, key), resolvedJoinExpr);
    Queryable joinQueryable = metaDataStore.getTable(sqlJoin.getJoinTableType());
    ColumnContext joinCtx = ColumnContext.builder().queryable(joinQueryable).alias(joinAlias).metaDataStore(this.metaDataStore).column(this.column).tableArguments(mergedArgumentMap(joinQueryable.getArguments(), this.getTableArguments())).build();
    return joinCtx;
}

Example 2 with SQLJoin

use of com.yahoo.elide.datastores.aggregation.queryengines.sql.metadata.SQLJoin in project elide by yahoo.

the class JoinExpressionExtractor method visitJoinReference.

@Override
public Set<String> visitJoinReference(JoinReference reference) {
    JoinPath joinPath = reference.getPath();
    List<PathElement> pathElements = joinPath.getPathElements();
    ColumnContext currentCtx = this.context;
    for (int i = 0; i < pathElements.size() - 1; i++) {
        PathElement pathElement = pathElements.get(i);
        Type<?> joinClass = pathElement.getFieldType();
        String joinFieldName = pathElement.getFieldName();
        SQLJoin sqlJoin = currentCtx.getQueryable().getJoin(joinFieldName);
        ColumnContext joinCtx;
        String onClause;
        JoinType joinType;
        String fullExpression;
        if (sqlJoin != null) {
            joinType = sqlJoin.getJoinType();
            joinCtx = (ColumnContext) currentCtx.get(joinFieldName);
            if (joinType.equals(JoinType.CROSS)) {
                onClause = EMPTY;
            } else {
                onClause = ON + currentCtx.resolve(sqlJoin.getJoinExpression());
            }
        } else {
            joinType = JoinType.LEFT;
            SQLTable table = metaDataStore.getTable(joinClass);
            joinCtx = ColumnContext.builder().queryable(table).alias(appendAlias(currentCtx.getAlias(), joinFieldName)).metaDataStore(currentCtx.getMetaDataStore()).column(currentCtx.getColumn()).tableArguments(mergedArgumentMap(table.getArguments(), currentCtx.getTableArguments())).build();
            onClause = ON + String.format("%s.%s = %s.%s", currentCtx.getAlias(), dictionary.getAnnotatedColumnName(pathElement.getType(), joinFieldName), joinCtx.getAlias(), dictionary.getAnnotatedColumnName(joinClass, dictionary.getIdFieldName(joinClass)));
        }
        SQLDialect sqlDialect = currentCtx.getQueryable().getDialect();
        String joinAlias = applyQuotes(joinCtx.getAlias(), sqlDialect);
        String joinKeyword = currentCtx.getQueryable().getDialect().getJoinKeyword(joinType);
        String joinSource = constructTableOrSubselect(joinCtx, joinClass);
        if (sqlDialect.useASBeforeTableAlias()) {
            fullExpression = String.format("%s %s AS %s %s", joinKeyword, joinSource, joinAlias, onClause);
        } else {
            fullExpression = String.format("%s %s %s %s", joinKeyword, joinSource, joinAlias, onClause);
        }
        joinExpressions.add(fullExpression);
        /**
         * If this `for` loop runs more than once, context should be switched to join context.
         */
        currentCtx = joinCtx;
    }
    // If reference within current join reference is of type PhysicalReference, then below visitor doesn't matter.
    // If it is of type LogicalReference, then visitLogicalReference method will recreate visitor with correct
    // value of ColumnProjection in context.
    JoinExpressionExtractor visitor = new JoinExpressionExtractor(currentCtx);
    joinExpressions.addAll(reference.getReference().accept(visitor));
    return joinExpressions;
}

Example 3 with SQLJoin

use of com.yahoo.elide.datastores.aggregation.queryengines.sql.metadata.SQLJoin in project elide by yahoo.

the class PhysicalRefColumnContext method getJoinContext.

@Override
protected ColumnContext getJoinContext(String key) {
    SQLJoin sqlJoin = this.queryable.getJoin(key);
    Queryable joinQueryable = metaDataStore.getTable(sqlJoin.getJoinTableType());
    String joinPath = isBlank(this.alias) ? key : this.alias + PERIOD + key;
    PhysicalRefColumnContext joinCtx = PhysicalRefColumnContext.physicalRefContextBuilder().queryable(joinQueryable).alias(joinPath).metaDataStore(this.metaDataStore).column(this.column).tableArguments(mergedArgumentMap(joinQueryable.getArguments(), this.getTableArguments())).build();
    return joinCtx;
}

Example 4 with SQLJoin

use of com.yahoo.elide.datastores.aggregation.queryengines.sql.metadata.SQLJoin in project elide by yahoo.

the class ReferenceExtractor method visitJoinReference.

@Override
public Set<T> visitJoinReference(JoinReference reference) {
    if (referenceType.equals(JoinReference.class)) {
        references.add((T) reference);
    }
    JoinPath path = reference.getPath();
    int pathLimit = (mode == Mode.SAME_QUERY) ? 1 : path.getPathElements().size() - 1;
    for (int idx = 0; idx < pathLimit; idx++) {
        Path.PathElement pathElement = path.getPathElements().get(idx);
        String fieldName = pathElement.getFieldName();
        Type<?> parentClass = pathElement.getType();
        SQLTable table = metaDataStore.getTable(parentClass);
        SQLJoin join = table.getJoin(fieldName);
        if (visitedJoins.contains(join)) {
            continue;
        }
        visitedJoins.add(join);
        parser.parse(table, join.getJoinExpression()).stream().forEach(ref -> ref.accept(this));
    }
    if (mode != Mode.SAME_QUERY) {
        return reference.getReference().accept(this);
    }
    return references;
}