org.verdictdb.sqlreader.NonValidatingSQLParser Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of verdictdb-core Show documentation

Platform-independent, interactive-speed data analytics system

There is a newer version: 0.5.8

/*
 *    Copyright 2018 University of Michigan
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

package org.verdictdb.sqlreader;

import java.util.List;

import org.antlr.v4.runtime.ANTLRInputStream;
import org.antlr.v4.runtime.CommonTokenStream;
import org.apache.commons.lang3.tuple.Pair;
import org.verdictdb.core.sqlobject.AbstractRelation;
import org.verdictdb.core.sqlobject.CreateScrambleQuery;
import org.verdictdb.parser.VerdictSQLLexer;
import org.verdictdb.parser.VerdictSQLParser;

/**
 * This is the entry point to both the select query and the scrambling query.
 *
 * 1. select query: SELECT ... 2. scrambling query: CREATE SCRAMBLE newSchema.newTable FROM
 * originalSchema.originalTable [METHOD method_name] [SIZE percent] ;
 *
 * method_name := 'uniform' | 'fastconverge' percent := NOT SUPPORTED YET
 *
 * @author Yongjoo Park
 */
public class NonValidatingSQLParser {

  //  MetaData meta;

  public NonValidatingSQLParser() {}
  //    this.meta = meta;
  //  }

  public static VerdictSQLParser parserOf(String text) {
    VerdictDBErrorListener verdictDBErrorListener = new VerdictDBErrorListener();
    VerdictSQLLexer l = new VerdictSQLLexer(new ANTLRInputStream(text));
    l.removeErrorListeners();
    l.addErrorListener(verdictDBErrorListener);

    VerdictSQLParser p = new VerdictSQLParser(new CommonTokenStream(l));
    p.removeErrorListeners();
    p.addErrorListener(verdictDBErrorListener);
    return p;
  }

  public AbstractRelation toRelation(String sql) {
    VerdictSQLParser p = parserOf(sql);
    RelationGen g = new RelationGen();
    return g.visit(p.select_statement());
  }

  public CreateScrambleQuery toCreateScrambleQuery(String sql) {
    VerdictSQLParser p = parserOf(sql);
    ScramblingQueryGenerator generator = new ScramblingQueryGenerator();
    CreateScrambleQuery query = generator.visit(p.create_scramble_statement());
    return query;
  }

  /**
   * @param rel
   * @return Pairs of (schemaName, tableName) that appear in the argument.
   */
  public static List> extractInvolvedTables(AbstractRelation rel) {
    return null;
  }
}