org.apache.tajo.catalog.AlterTableDesc.java Source code

Introduction

Here is the source code for org.apache.tajo.catalog.AlterTableDesc.java

Source

/**
 * 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.catalog;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.annotations.Expose;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.tajo.catalog.json.CatalogGsonHelper;
import org.apache.tajo.catalog.proto.CatalogProtos;
import org.apache.tajo.common.ProtoObject;
import org.apache.tajo.json.GsonObject;

public class AlterTableDesc implements ProtoObject<CatalogProtos.AlterTableDescProto>, GsonObject, Cloneable {
    private final Log LOG = LogFactory.getLog(AlterTableDesc.class);

    protected CatalogProtos.AlterTableDescProto.Builder builder = null;

    @Expose
    protected String tableName; // required
    @Expose
    protected String newTableName; // optional
    @Expose
    protected String columnName; //optional
    @Expose
    protected String newColumnName; //optional
    @Expose
    protected Column addColumn = null; //optiona
    @Expose
    protected AlterTableType alterTableType; //required

    public AlterTableDesc() {
        builder = CatalogProtos.AlterTableDescProto.newBuilder();
    }

    public String getTableName() {
        return tableName;
    }

    public void setTableName(String tableName) {
        this.tableName = tableName;
    }

    public String getNewTableName() {
        return newTableName;
    }

    public void setNewTableName(String newTableName) {
        this.newTableName = newTableName;
    }

    public String getColumnName() {
        return columnName;
    }

    public void setColumnName(String columnName) {
        this.columnName = columnName;
    }

    public String getNewColumnName() {
        return newColumnName;
    }

    public void setNewColumnName(String newColumnName) {
        this.newColumnName = newColumnName;
    }

    public Column getAddColumn() {
        return addColumn;
    }

    public void setAddColumn(Column addColumn) {
        this.addColumn = addColumn;
    }

    public AlterTableType getAlterTableType() {
        return alterTableType;
    }

    public void setAlterTableType(AlterTableType alterTableType) {
        this.alterTableType = alterTableType;
    }

    @Override
    public String toString() {
        Gson gson = new GsonBuilder().setPrettyPrinting().excludeFieldsWithoutExposeAnnotation().create();
        return gson.toJson(this);
    }

    @Override
    public String toJson() {
        return CatalogGsonHelper.toJson(this, AlterTableDesc.class);
    }

    @Override
    public CatalogProtos.AlterTableDescProto getProto() {
        if (null == builder) {
            builder = CatalogProtos.AlterTableDescProto.newBuilder();
        }
        if (null != this.tableName) {
            builder.setTableName(this.tableName);
        }
        if (null != this.newTableName) {
            builder.setNewTableName(this.newTableName);
        }
        if (null != this.columnName && null != this.newColumnName) {
            final CatalogProtos.AlterColumnProto.Builder alterColumnBuilder = CatalogProtos.AlterColumnProto
                    .newBuilder();
            alterColumnBuilder.setOldColumnName(this.columnName);
            alterColumnBuilder.setNewColumnName(this.newColumnName);
            builder.setAlterColumnName(alterColumnBuilder.build());
        }
        if (null != this.addColumn) {
            builder.setAddColumn(addColumn.getProto());
        }

        switch (alterTableType) {
        case RENAME_TABLE:
            builder.setAlterTableType(CatalogProtos.AlterTableType.RENAME_TABLE);
            break;
        case RENAME_COLUMN:
            builder.setAlterTableType(CatalogProtos.AlterTableType.RENAME_COLUMN);
            break;
        case ADD_COLUMN:
            builder.setAlterTableType(CatalogProtos.AlterTableType.ADD_COLUMN);
            break;
        default:
        }
        return builder.build();
    }

}