/**
* 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.engine.planner.logical;
import com.google.gson.annotations.Expose;
import org.apache.tajo.engine.planner.PlanString;
import org.apache.tajo.engine.planner.Target;
import java.util.Arrays;
public class ProjectionNode extends UnaryNode implements Projectable {
/**
* the targets are always filled even if the query is 'select *'
*/
@Expose private Target [] targets;
@Expose private boolean distinct = false;
/**
* @param targets they should be all evaluated ones.
*/
public ProjectionNode(int pid, Target [] targets) {
super(pid, NodeType.PROJECTION);
this.targets = targets;
}
public boolean hasTargets() {
return this.targets != null;
}
@Override
public void setTargets(Target[] targets) {
this.targets = targets;
}
@Override
public Target [] getTargets() {
return this.targets;
}
public void setChild(LogicalNode subNode) {
super.setChild(subNode);
}
public String toString() {
StringBuilder sb = new StringBuilder();
sb.append("\"Projection\": {");
if (distinct) {
sb.append("\"distinct\": true, ");
}
sb.append("\"targets\": [");
for (int i = 0; i < targets.length; i++) {
sb.append("\"").append(targets[i]).append("\"");
if( i < targets.length - 1) {
sb.append(",");
}
}
sb.append("],");
sb.append("\n \"out schema\": ").append(getOutSchema()).append(",");
sb.append("\n \"in schema\": ").append(getInSchema());
sb.append("}");
return sb.toString()+"\n"
+ getChild().toString();
}
@Override
public boolean equals(Object obj) {
if (obj instanceof ProjectionNode) {
ProjectionNode other = (ProjectionNode) obj;
boolean b1 = super.equals(other);
boolean b2 = Arrays.equals(targets, other.targets);
boolean b3 = child.equals(other.child);
return b1 && b2 && b3;
} else {
return false;
}
}
@Override
public Object clone() throws CloneNotSupportedException {
ProjectionNode projNode = (ProjectionNode) super.clone();
projNode.targets = targets.clone();
return projNode;
}
@Override
public PlanString getPlanString() {
PlanString planStr = new PlanString("Projection: ");
if (distinct) {
planStr.appendTitle(" (distinct)");
}
StringBuilder sb = new StringBuilder("Targets: ");
for (int i = 0; i < targets.length; i++) {
sb.append(targets[i]);
if( i < targets.length - 1) {
sb.append(", ");
}
}
planStr.addExplan(sb.toString());
if (getOutSchema() != null) {
planStr.addExplan("out schema: " + getOutSchema().toString());
}
if (getInSchema() != null) {
planStr.addExplan("in schema: " + getInSchema().toString());
}
return planStr;
}
}