org.verdictdb.sqlsyntax.MysqlSyntax 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.sqlsyntax;

import com.google.common.collect.Lists;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;

public class MysqlSyntax extends SqlSyntax {

  @Override
  public Collection getCandidateJDBCDriverClassNames() {
    List candidates = Lists.newArrayList("com.mysql.jdbc.Driver");
    return candidates;
  }

  @Override
  public boolean doesSupportTablePartitioning() {
    return true;
  }

  @Override
  public void dropTable(String schema, String tablename) {}

  @Override
  public int getColumnNameColumnIndex() {
    return 0;
  }

  @Override
  public String getFallbackDefaultSchema() {
    return "test";
  }

  @Override
  public String getColumnsCommand(String schema, String table) {
    return "show columns in " + quoteName(table) + " in " + quoteName(schema);
  }

  @Override
  public int getColumnTypeColumnIndex() {
    return 1;
  }

  @Override
  public String getPartitionByInCreateTable(
      List partitionColumns, List partitionCounts) {

    for (int count : partitionCounts) {
      if (count == 0) {
        return "";
      }
    }

    StringBuilder sql = new StringBuilder();
    sql.append("partition by list columns (");

    // use a single column
    for (int i = 0; i < partitionColumns.size(); i++) {
      if (i != 0) {
        sql.append(", ");
      }
      sql.append(String.format("`%s`", partitionColumns.get(i)));
    }
    sql.append(") (");

    // add list
    List currentPart = new ArrayList<>(Collections.nCopies(partitionCounts.size(), 0));
    int partNum = 0;
    while (true) {
      //      System.out.println(sql.toString());
      //      System.out.println(partitionCounts);

      if (partNum != 0) {
        sql.append(", ");
      }
      sql.append(String.format("partition p%d values in (", partNum));
      for (int i = 0; i < currentPart.size(); i++) {
        if (i == 0) {
          if (currentPart.size() > 1) {
            sql.append("(");
          }
        } else {
          sql.append(",");
        }

        sql.append(String.format("%d", currentPart.get(i)));

        if (i == currentPart.size() - 1 && currentPart.size() > 1) {
          sql.append(")");
        }
      }
      sql.append(")");

      // increase currentPart by one
      boolean carry = true;
      for (int j = partitionCounts.size() - 1; j >= 0; j--) {
        if (carry) {
          currentPart.set(j, currentPart.get(j) + 1);
          carry = false;
        }
        if (currentPart.get(j) == partitionCounts.get(j)) {
          carry = true;
          currentPart.set(j, 0);
        }
      }

      // overflow
      if (carry) {
        break;
      }

      partNum += 1;
    }

    sql.append(")");
    return sql.toString();
  }

  @Override
  public String getPartitionCommand(String schema, String table) {
    return "SELECT DISTINCT REPLACE(PARTITION_EXPRESSION, '`', '') FROM INFORMATION_SCHEMA.PARTITIONS "
        + "WHERE TABLE_NAME='"
        + table
        + "' AND TABLE_SCHEMA='"
        + schema
        + "'";
  }

  @Override
  public String getQuoteString() {
    // https://dev.mysql.com/doc/refman/8.0/en/identifiers.html
    return "`";
  }

  @Override
  public String getSchemaCommand() {
    return "show schemas";
  }

  @Override
  public int getSchemaNameColumnIndex() {
    return 0;
  }

  @Override
  public String getTableCommand(String schema) {
    return "show tables in " + quoteName(schema);
  }

  @Override
  public int getTableNameColumnIndex() {
    return 0;
  }

  @Override
  public String randFunction() {
    return "rand()";
  }

  @Override
  public boolean isAsRequiredBeforeSelectInCreateTable() {
    return false;
  }

  @Override
  public boolean equals(Object obj) {
    if (obj == null) {
      return false;
    }
    if (obj == this) {
      return true;
    }
    if (obj.getClass() != getClass()) {
      return false;
    }
    return true;
  }
}