org.apache.tajo.algebra.CreateTable Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tajo-algebra Show documentation
There is a newer version: 0.11.3
/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.apache.tajo.algebra;

import com.google.common.base.Objects;
import com.google.gson.*;
import com.google.gson.annotations.Expose;
import com.google.gson.annotations.SerializedName;
import org.apache.tajo.json.CommonGsonHelper;
import org.apache.tajo.util.TUtil;

import java.util.ArrayList;
import java.util.HashMap;
import java.lang.reflect.Type;
import java.util.List;
import java.util.Map;

public class CreateTable extends Expr {
  @Expose @SerializedName("IsExternal")
  private boolean external = false;
  @Expose @SerializedName("TableName")
  private String tableName;
  @Expose @SerializedName("Attributes")
  private ColumnDefinition [] tableElements;
  @Expose @SerializedName("StorageType")
  private String storageType;
  @Expose @SerializedName("Location")
  private String location;
  @Expose @SerializedName("SubPlan")
  private Expr subquery;
  @Expose @SerializedName("TableProperties")
  private Map params;
  @Expose @SerializedName("PartitionMethodDesc")
  private PartitionMethodDescExpr partition;
  @Expose @SerializedName("IfNotExists")
  private boolean ifNotExists;
  @Expose @SerializedName("LikeParentTable")
  private String likeParentTable;

  public CreateTable(final String tableName, boolean ifNotExists) {
    super(OpType.CreateTable);
    this.tableName = tableName;
    this.ifNotExists = ifNotExists;
  }

  public CreateTable(final String tableName, final Expr subQuery, boolean ifNotExists) {
    this(tableName, ifNotExists);
    this.subquery = subQuery;
  }

  public void setExternal() {
    external = true;
  }

  public boolean isExternal() {
    return external;
  }

  public String getTableName() {
    return this.tableName;
  }

  public boolean hasLocation() {
    return location != null;
  }

  public String getLocation() {
    return this.location;
  }

  public void setLocation(String location) {
    this.location = location;
  }

  public boolean hasTableElements() {
    return this.tableElements != null;
  }

  public ColumnDefinition [] getTableElements() {
    return tableElements;
  }

  public void setTableElements(ColumnDefinition [] tableElements) {
    this.tableElements = tableElements;
  }

  public boolean hasStorageType() {
    return storageType != null;
  }

  public void setStorageType(String storageType) {
    this.storageType = storageType;
  }

  public String getStorageType() {
    return storageType;
  }

  public boolean hasParams() {
    return params != null;
  }

  public Map getParams() {
    return params;
  }

  public void setParams(Map params) {
    this.params = params;
  }

  public boolean hasPartition() {
    return partition != null;
  }

  public void setPartitionMethod(PartitionMethodDescExpr partition) {
    this.partition = partition;
  }

  public  T getPartitionMethod() {
    return (T) this.partition;
  }

  public boolean hasSubQuery() {
    return subquery != null;
  }

  public void setSubQuery(Expr subquery) {
    this.subquery = subquery;
  }

  public Expr getSubQuery() {
    return subquery;
  }

  public boolean isIfNotExists() {
    return ifNotExists;
  }

  public void setLikeParentTable(String parentTable)  {
    this.likeParentTable = parentTable;
  }

  public String getLikeParentTableName()  {
    return likeParentTable;
  }


  @Override
  public int hashCode() {
    return Objects.hashCode(
        external, tableName, Objects.hashCode(tableElements),
        storageType, subquery, location, params, partition, ifNotExists);
  }

  @Override
  boolean equalsTo(Expr expr) {
    CreateTable another = (CreateTable) expr;
    return external == another.external &&
        tableName.equals(another.tableName) &&
        TUtil.checkEquals(tableElements, another.tableElements) &&
        TUtil.checkEquals(storageType, another.storageType) &&
        TUtil.checkEquals(location, another.location) &&
        TUtil.checkEquals(subquery, another.subquery) &&
        TUtil.checkEquals(params, another.params) &&
        TUtil.checkEquals(partition, another.partition) &&
        ifNotExists == another.ifNotExists;
  }

  @Override
  public Object clone() throws CloneNotSupportedException {
    CreateTable createTable = (CreateTable) super.clone();
    createTable.external = external;
    createTable.tableName = tableName;
    if (tableElements != null) {
      createTable.tableElements = new ColumnDefinition[tableElements.length];
      for (int i = 0; i < tableElements.length; i++) {
        createTable.tableElements[i] = (ColumnDefinition) tableElements[i].clone();
      }
    }
    createTable.storageType = storageType;
    createTable.location = location;
    createTable.subquery = subquery;
    createTable.params = new HashMap(params);
    createTable.partition = (PartitionMethodDescExpr) partition.clone();
    createTable.ifNotExists = ifNotExists;
    return createTable;
  }

  public static enum PartitionType {
    RANGE,
    HASH,
    LIST,
    COLUMN
  }

  public static abstract class PartitionMethodDescExpr implements Cloneable {
    @Expose @SerializedName("PartitionType")
    PartitionType type;

    public PartitionMethodDescExpr(PartitionType type) {
      this.type = type;
    }

    public PartitionType getPartitionType() {
      return type;
    }

    @Override
    public Object clone() throws CloneNotSupportedException {
      PartitionMethodDescExpr partition = (PartitionMethodDescExpr) super.clone();
      partition.type = type;
      return partition;
    }

    static class JsonSerDer implements JsonSerializer,
        JsonDeserializer {

      @Override
      public PartitionMethodDescExpr deserialize(JsonElement json, Type typeOfT, JsonDeserializationContext context)
          throws JsonParseException {
        JsonObject jsonObject = json.getAsJsonObject();
        PartitionType type = PartitionType.valueOf(
            CommonGsonHelper.getOrDie(jsonObject, "PartitionType").getAsString());
        switch (type) {
          case RANGE:
            return context.deserialize(json, RangePartition.class);
          case HASH:
            return context.deserialize(json, HashPartition.class);
          case LIST:
            return context.deserialize(json, ListPartition.class);
          case COLUMN:
            return context.deserialize(json, ColumnPartition.class);
        }
        return null;
      }

      @Override
      public JsonElement serialize(PartitionMethodDescExpr src, Type typeOfSrc, JsonSerializationContext context) {
        switch (src.getPartitionType()) {
          case RANGE:
            return context.serialize(src, RangePartition.class);
          case HASH:
            return context.serialize(src, HashPartition.class);
          case LIST:
            return context.serialize(src, ListPartition.class);
          case COLUMN:
            return context.serialize(src, ColumnPartition.class);
          default:
            return null;
        }
      }
    }
  }

  public static class RangePartition extends PartitionMethodDescExpr implements Cloneable {
    @Expose @SerializedName("Columns")
    ColumnReferenceExpr [] columns;
    @Expose @SerializedName("Specifiers")
    List specifiers;

    public RangePartition(ColumnReferenceExpr [] columns, List specifiers) {
      super(PartitionType.RANGE);
      this.columns = columns;
      this.specifiers = specifiers;
    }

    public ColumnReferenceExpr [] getColumns() {
      return columns;
    }

    public List getSpecifiers() {
      return specifiers;
    }

    public int hashCode() {
      return Objects.hashCode(type, Objects.hashCode(columns), specifiers);
    }

    public boolean equals(Object object) {
      if (object instanceof RangePartition) {
        RangePartition another = (RangePartition) object;
        return type == another.type && TUtil.checkEquals(columns, another.columns) &&
            specifiers.equals(another.specifiers);
      } else {
        return false;
      }
    }

    @Override
    public Object clone() throws CloneNotSupportedException {
      RangePartition range = (RangePartition) super.clone();
      range.columns = new ColumnReferenceExpr[columns.length];
      for (int i = 0; i < columns.length; i++) {
        range.columns[i] = (ColumnReferenceExpr) columns[i].clone();
      }
      if (range.specifiers != null) {
        range.specifiers = new ArrayList();
        for (int i = 0; i < specifiers.size(); i++) {
          range.specifiers.add(specifiers.get(i));
        }
      }
      return range;
    }
  }

  public static class HashPartition extends PartitionMethodDescExpr implements Cloneable {
    @Expose @SerializedName("Columns")
    ColumnReferenceExpr [] columns;
    @Expose @SerializedName("Quantity")
    Expr quantity;
    @Expose @SerializedName("Specifiers")
    List specifiers;

    public HashPartition(ColumnReferenceExpr [] columns, Expr quantity) {
      super(PartitionType.HASH);
      this.columns = columns;
      this.quantity = quantity;
    }

    public HashPartition(ColumnReferenceExpr [] columns, List specifier) {
      super(PartitionType.HASH);
      this.columns = columns;
      this.specifiers = specifier;
    }

    public ColumnReferenceExpr [] getColumns() {
      return columns;
    }

    public boolean hasQuantifier() {
      return quantity != null;
    }

    public Expr getQuantifier() {
      return quantity;
    }

    public boolean hasSpecifiers() {
      return specifiers != null;
    }

    public List getSpecifiers() {
      return specifiers;
    }

    public int hashCode() {
      return Objects.hashCode(type, columns, specifiers);
    }

    public boolean equals(Object object) {
      if (object instanceof HashPartition) {
        HashPartition another = (HashPartition) object;
        return type == another.type && TUtil.checkEquals(columns, another.columns) &&
            specifiers.equals(another.specifiers);
      } else {
        return false;
      }
    }

    @Override
    public Object clone() throws CloneNotSupportedException {
      HashPartition hash = (HashPartition) super.clone();
      hash.columns = new ColumnReferenceExpr[columns.length];
      for (int i = 0; i < columns.length; i++) {
        hash.columns[i] = (ColumnReferenceExpr) columns[i].clone();
      }
      hash.quantity = quantity;
      if (specifiers != null) {
        hash.specifiers = new ArrayList();
        for (PartitionSpecifier specifier : specifiers) {
          hash.specifiers.add(specifier);
        }
      }
      return hash;
    }
  }

  public static class ListPartition extends PartitionMethodDescExpr {
    @Expose @SerializedName("Columns")
    ColumnReferenceExpr [] columns;
    @Expose @SerializedName("Specifiers")
    List specifiers;

    public ListPartition(ColumnReferenceExpr [] columns, List specifers) {
      super(PartitionType.LIST);
      this.columns = columns;
      this.specifiers = specifers;
    }

    public ColumnReferenceExpr [] getColumns() {
      return columns;
    }

    public List getSpecifiers() {
      return specifiers;
    }

    public int hashCode() {
      return Objects.hashCode(Objects.hashCode(columns), specifiers);
    }

    public boolean equals(Object object) {
      if (object instanceof ListPartition) {
        ListPartition another = (ListPartition) object;
        return type == another.type && TUtil.checkEquals(columns, another.columns) &&
            specifiers.equals(another.specifiers);
      } else {
        return false;
      }
    }

    @Override
    public Object clone() throws CloneNotSupportedException {
      ListPartition listPartition = (ListPartition) super.clone();
      listPartition.columns = new ColumnReferenceExpr[columns.length];
      for (int i = 0; i < columns.length; i++) {
        listPartition.columns[i] = (ColumnReferenceExpr) columns[i].clone();
      }
      if (specifiers != null) {
        listPartition.specifiers = new ArrayList();
        for (ListPartitionSpecifier specifier : specifiers) {
          listPartition.specifiers.add(specifier);
        }
      }
      return listPartition;
    }
  }

  public static class ColumnPartition extends PartitionMethodDescExpr {
    @Expose @SerializedName("Columns")
    private ColumnDefinition [] columns;
    @Expose @SerializedName("IsOmitValues")
    private boolean isOmitValues;

    public ColumnPartition(ColumnDefinition [] columns) {
      super(PartitionType.COLUMN);
      this.columns = columns;
    }

    public ColumnDefinition [] getColumns() {
      return columns;
    }

    public int hashCode() {
      return Objects.hashCode(columns);
    }

    public boolean equals(Object object) {
      if (object instanceof ColumnPartition) {
        ColumnPartition another = (ColumnPartition) object;
        return type == another.type && TUtil.checkEquals(columns, another.columns);
      }
      return false;
    }

    @Override
    public Object clone() throws CloneNotSupportedException {
      ColumnPartition columnPartition = (ColumnPartition) super.clone();
      columnPartition.columns = new ColumnDefinition[columns.length];
      for (int i = 0; i < columns.length; i++) {
        columnPartition.columns[i] = (ColumnDefinition) columns[i].clone();
      }
      return columnPartition;
    }
  }

  public static class RangePartitionSpecifier extends PartitionSpecifier implements Cloneable {
    @Expose @SerializedName("End")
    Expr end;
    @Expose @SerializedName("IsMaxValue")
    boolean maxValue;

    public RangePartitionSpecifier(String name, Expr end) {
      super(name);
      this.end = end;
    }

    public RangePartitionSpecifier(String name) {
      super(name);
      maxValue = true;
    }

    public Expr getEnd() {
      return end;
    }

    public boolean isEndMaxValue() {
      return this.maxValue;
    }

    public int hashCode() {
      return Objects.hashCode(end, maxValue);
    }

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

      RangePartitionSpecifier that = (RangePartitionSpecifier) o;

      if (maxValue != that.maxValue) return false;
      if (!end.equals(that.end)) return false;

      return true;
    }

    @Override
    public Object clone() throws CloneNotSupportedException {
      RangePartitionSpecifier specifier = (RangePartitionSpecifier) super.clone();
      specifier.end = (Expr) end.clone();
      specifier.maxValue = maxValue;
      return specifier;
    }
  }

  public static class ListPartitionSpecifier extends PartitionSpecifier implements Cloneable {
    @Expose @SerializedName("ValueList")
    ValueListExpr valueList;

    public ListPartitionSpecifier(String name, ValueListExpr valueList) {
      super(name);
      this.valueList = valueList;
    }

    public ValueListExpr getValueList() {
      return valueList;
    }

    @Override
    public int hashCode() {
      return valueList.hashCode();
    }

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

      ListPartitionSpecifier that = (ListPartitionSpecifier) o;

      return valueList.equals(that.valueList);
    }

    @Override
    public Object clone() throws CloneNotSupportedException {
      ListPartitionSpecifier specifier = (ListPartitionSpecifier) super.clone();
      specifier.valueList = (ValueListExpr) valueList.clone();
      return specifier;
    }
  }

  public static class PartitionSpecifier implements Cloneable {
    @Expose @SerializedName("PartitionSpecName")
    private String name;

    public PartitionSpecifier(String name) {
      this.name = name;
    }

    public String getName() {
      return this.name;
    }

    public int hashCode() {
      return name.hashCode();
    }

    public boolean equals(Object obj) {
      if (obj instanceof PartitionSpecifier ) {
        return name.equals(((PartitionSpecifier)obj).name);
      } else {
        return false;
      }
    }

    @Override
    public Object clone() throws CloneNotSupportedException {
      PartitionSpecifier specifier = (PartitionSpecifier) super.clone();
      specifier.name = name;
      return specifier;
    }
  }
}