mirror of
https://github.com/alibaba/druid.git
synced 2023-03-04 19:17:16 +00:00
sql parser direct support OVER
This commit is contained in:
@@ -0,0 +1,74 @@
|
||||
package com.alibaba.druid.sql.ast;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
import com.alibaba.druid.sql.visitor.SQLASTVisitor;
|
||||
|
||||
public class SQLOver extends SQLObjectImpl {
|
||||
|
||||
private static final long serialVersionUID = 1L;
|
||||
|
||||
protected final List<SQLExpr> partitionBy = new ArrayList<SQLExpr>();
|
||||
protected SQLOrderBy orderBy;
|
||||
|
||||
@Override
|
||||
protected void accept0(SQLASTVisitor visitor) {
|
||||
if (visitor.visit(this)) {
|
||||
acceptChild(visitor, this.partitionBy);
|
||||
acceptChild(visitor, this.orderBy);
|
||||
}
|
||||
visitor.endVisit(this);
|
||||
}
|
||||
|
||||
public SQLOrderBy getOrderBy() {
|
||||
return orderBy;
|
||||
}
|
||||
|
||||
public void setOrderBy(SQLOrderBy orderBy) {
|
||||
this.orderBy = orderBy;
|
||||
}
|
||||
|
||||
public List<SQLExpr> getPartitionBy() {
|
||||
return partitionBy;
|
||||
}
|
||||
|
||||
@Override
|
||||
public int hashCode() {
|
||||
final int prime = 31;
|
||||
int result = 1;
|
||||
result = prime * result + ((orderBy == null) ? 0 : orderBy.hashCode());
|
||||
result = prime * result + ((partitionBy == null) ? 0 : partitionBy.hashCode());
|
||||
return result;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean equals(Object obj) {
|
||||
if (this == obj) {
|
||||
return true;
|
||||
}
|
||||
if (obj == null) {
|
||||
return false;
|
||||
}
|
||||
if (getClass() != obj.getClass()) {
|
||||
return false;
|
||||
}
|
||||
SQLOver other = (SQLOver) obj;
|
||||
if (orderBy == null) {
|
||||
if (other.orderBy != null) {
|
||||
return false;
|
||||
}
|
||||
} else if (!orderBy.equals(other.orderBy)) {
|
||||
return false;
|
||||
}
|
||||
if (partitionBy == null) {
|
||||
if (other.partitionBy != null) {
|
||||
return false;
|
||||
}
|
||||
} else if (!partitionBy.equals(other.partitionBy)) {
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -21,6 +21,7 @@ import java.util.List;
|
||||
|
||||
import com.alibaba.druid.sql.ast.SQLExpr;
|
||||
import com.alibaba.druid.sql.ast.SQLExprImpl;
|
||||
import com.alibaba.druid.sql.ast.SQLOver;
|
||||
import com.alibaba.druid.sql.visitor.SQLASTVisitor;
|
||||
|
||||
public class SQLAggregateExpr extends SQLExprImpl implements Serializable {
|
||||
@@ -29,6 +30,7 @@ public class SQLAggregateExpr extends SQLExprImpl implements Serializable {
|
||||
protected String methodName;
|
||||
protected Option option;
|
||||
protected final List<SQLExpr> arguments = new ArrayList<SQLExpr>();
|
||||
protected SQLOver over;
|
||||
|
||||
public SQLAggregateExpr(String methodName){
|
||||
|
||||
@@ -60,20 +62,19 @@ public class SQLAggregateExpr extends SQLExprImpl implements Serializable {
|
||||
return this.arguments;
|
||||
}
|
||||
|
||||
public void output(StringBuffer buf) {
|
||||
buf.append(this.methodName);
|
||||
buf.append("(");
|
||||
int i = 0;
|
||||
for (int size = this.arguments.size(); i < size; ++i) {
|
||||
((SQLExpr) this.arguments.get(i)).output(buf);
|
||||
}
|
||||
buf.append(")");
|
||||
public SQLOver getOver() {
|
||||
return over;
|
||||
}
|
||||
|
||||
public void setOver(SQLOver over) {
|
||||
this.over = over;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void accept0(SQLASTVisitor visitor) {
|
||||
if (visitor.visit(this)) {
|
||||
acceptChild(visitor, this.arguments);
|
||||
acceptChild(visitor, this.over);
|
||||
}
|
||||
|
||||
visitor.endVisit(this);
|
||||
@@ -86,6 +87,7 @@ public class SQLAggregateExpr extends SQLExprImpl implements Serializable {
|
||||
result = prime * result + ((arguments == null) ? 0 : arguments.hashCode());
|
||||
result = prime * result + ((methodName == null) ? 0 : methodName.hashCode());
|
||||
result = prime * result + ((option == null) ? 0 : option.hashCode());
|
||||
result = prime * result + ((over == null) ? 0 : over.hashCode());
|
||||
return result;
|
||||
}
|
||||
|
||||
@@ -115,6 +117,13 @@ public class SQLAggregateExpr extends SQLExprImpl implements Serializable {
|
||||
} else if (!methodName.equals(other.methodName)) {
|
||||
return false;
|
||||
}
|
||||
if (over == null) {
|
||||
if (other.over != null) {
|
||||
return false;
|
||||
}
|
||||
} else if (!over.equals(other.over)) {
|
||||
return false;
|
||||
}
|
||||
if (option != other.option) {
|
||||
return false;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user