io.wizzie.enricher.query.internal.EnricherQLBaseVisitorImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of query-compiler Show documentation

query-compiler

There is a newer version: 0.6.1

package io.wizzie.enricher.query.internal;

import io.wizzie.enricher.query.antlr4.Join;
import io.wizzie.enricher.query.antlr4.Query;
import io.wizzie.enricher.query.antlr4.Select;
import io.wizzie.enricher.query.antlr4.Stream;
import io.wizzie.enricher.query.compiler.EnricherQLBaseVisitor;
import io.wizzie.enricher.query.compiler.EnricherQLParser;
import io.wizzie.enricher.query.utils.Constants;

import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;

public class EnricherQLBaseVisitorImpl extends EnricherQLBaseVisitor {

    @Override
    public Query visitQuery(EnricherQLParser.QueryContext ctx) {

        List selectDimensions = ctx.dimensions().id().stream().map(id -> id.getText()).collect(Collectors.toList());

        if(selectDimensions.isEmpty())
            selectDimensions = Collections.singletonList("*");

        List selectInputStreams = ctx.streams().id().stream().map(id -> new Stream(id.getText(), ctx.type().getText().equals("TABLE"))).collect(Collectors.toList());

        List joins = ctx.query_join().stream().map(joinContext -> {

            boolean table = joinContext.type().getText().equals("TABLE");

            List joinDimensions = joinContext.dimensions().id().stream().map(id -> id.getText()).collect(Collectors.toList());

            if(joinDimensions.isEmpty())
                joinDimensions = Collections.singletonList("*");

            String className = joinContext.className().getText();
            String joinStream = joinContext.id().getText();

            EnricherQLParser.PartitionKeyContext partitionKeyContext;

            String partitionKey = ( partitionKeyContext = joinContext.partitionKey()) != null ? partitionKeyContext.getText() : Constants.__KEY;

            return new Join(new Stream(joinStream, table), className, joinDimensions, partitionKey);
        }).collect(Collectors.toList());

        List enrichWiths = ctx.query_enrich_with().stream().map(enrichContext -> new String(enrichContext.className().getText())).collect(Collectors.toList());

        Stream output = new Stream(ctx.query_output().id().getText(), ctx.query_output().type().getText().matches("TABLE"));

        return new Query(new Select(selectDimensions, selectInputStreams), output, joins, enrichWiths);
    }

}