/**
* 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.hadoop.hive.ql.plan;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import com.facebook.presto.hive.shaded.org.apache.commons.lang.builder.HashCodeBuilder;
import org.apache.hadoop.hive.serde2.typeinfo.TypeInfo;
import org.apache.hadoop.hive.serde2.typeinfo.TypeInfoFactory;
/**
* ExprNodeColumnDesc.
*
*/
public class ExprNodeColumnDesc extends ExprNodeDesc implements Serializable {
private static final long serialVersionUID = 1L;
/**
* The column name.
*/
private String column;
/**
* The alias of the table.
*/
private String tabAlias;
/**
* Is the column a partitioned column.
*/
private boolean isPartitionColOrVirtualCol;
/**
* Is the column a skewed column
*/
private boolean isSkewedCol;
public ExprNodeColumnDesc() {
}
public ExprNodeColumnDesc(TypeInfo typeInfo, String column, String tabAlias,
boolean isPartitionColOrVirtualCol) {
super(typeInfo);
this.column = column;
this.tabAlias = tabAlias;
this.isPartitionColOrVirtualCol = isPartitionColOrVirtualCol;
}
public ExprNodeColumnDesc(Class<?> c, String column, String tabAlias,
boolean isPartitionColOrVirtualCol) {
super(TypeInfoFactory.getPrimitiveTypeInfoFromJavaPrimitive(c));
this.column = column;
this.tabAlias = tabAlias;
this.isPartitionColOrVirtualCol = isPartitionColOrVirtualCol;
}
public ExprNodeColumnDesc(TypeInfo typeInfo, String column, String tabAlias,
boolean isPartitionColOrVirtualCol, boolean isSkewedCol) {
super(typeInfo);
this.column = column;
this.tabAlias = tabAlias;
this.isPartitionColOrVirtualCol = isPartitionColOrVirtualCol;
this.isSkewedCol = isSkewedCol;
}
public String getColumn() {
return column;
}
public void setColumn(String column) {
this.column = column;
}
public String getTabAlias() {
return tabAlias;
}
public void setTabAlias(String tabAlias) {
this.tabAlias = tabAlias;
}
public boolean getIsPartitionColOrVirtualCol() {
return isPartitionColOrVirtualCol;
}
public void setIsPartitionColOrVirtualCol(boolean isPartitionCol) {
this.isPartitionColOrVirtualCol = isPartitionCol;
}
@Override
public String toString() {
return "Column[" + column + "]";
}
@Override
public String getExprString() {
return getColumn();
}
@Override
public List<String> getCols() {
List<String> lst = new ArrayList<String>();
lst.add(column);
return lst;
}
@Override
public ExprNodeDesc clone() {
return new ExprNodeColumnDesc(typeInfo, column, tabAlias, isPartitionColOrVirtualCol,
isSkewedCol);
}
@Override
public boolean isSame(Object o) {
if (!(o instanceof ExprNodeColumnDesc)) {
return false;
}
ExprNodeColumnDesc dest = (ExprNodeColumnDesc) o;
if (!column.equals(dest.getColumn())) {
return false;
}
if (!typeInfo.equals(dest.getTypeInfo())) {
return false;
}
if ( tabAlias != null && dest.tabAlias != null ) {
if ( !tabAlias.equals(dest.tabAlias) ) {
return false;
}
}
return true;
}
/**
* @return the isSkewedCol
*/
public boolean isSkewedCol() {
return isSkewedCol;
}
/**
* @param isSkewedCol the isSkewedCol to set
*/
public void setSkewedCol(boolean isSkewedCol) {
this.isSkewedCol = isSkewedCol;
}
@Override
public int hashCode() {
int superHashCode = super.hashCode();
HashCodeBuilder builder = new HashCodeBuilder();
builder.appendSuper(superHashCode);
builder.append(column);
builder.append(tabAlias);
return builder.toHashCode();
}
}