Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

improve wall filter

Browse Source
This commit is contained in:
wenshao
2013-04-11 19:30:32 +08:00
parent eddb42ae53
commit 76a963ac46
7 changed files with 89 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/visitor/SQLEvalVisitor.java
+1
View File
@@ -25,6 +25,7 @@ public interface SQLEvalVisitor extends SQLASTVisitor {
public static final String EVAL_EXPR = "eval.expr";
public static final Object EVAL_ERROR = new Object();
public static final Object EVAL_VALUE_COUNT = new Object();
public static final Object EVAL_VALUE_NULL = new Object();
Function getFunction(String funcName);
src/main/java/com/alibaba/druid/sql/visitor/SQLEvalVisitorUtils.java
+25 -8
View File
@@ -62,10 +62,14 @@ import com.alibaba.druid.sql.visitor.functions.Char;
import com.alibaba.druid.sql.visitor.functions.Concat;
import com.alibaba.druid.sql.visitor.functions.Elt;
import com.alibaba.druid.sql.visitor.functions.Function;
import com.alibaba.druid.sql.visitor.functions.Greatest;
import com.alibaba.druid.sql.visitor.functions.Hex;
import com.alibaba.druid.sql.visitor.functions.If;
import com.alibaba.druid.sql.visitor.functions.Insert;
import com.alibaba.druid.sql.visitor.functions.Instr;
import com.alibaba.druid.sql.visitor.functions.Isnull;
import com.alibaba.druid.sql.visitor.functions.Lcase;
import com.alibaba.druid.sql.visitor.functions.Least;
import com.alibaba.druid.sql.visitor.functions.Left;
import com.alibaba.druid.sql.visitor.functions.Length;
import com.alibaba.druid.sql.visitor.functions.Locate;
@@ -185,6 +189,10 @@ public class SQLEvalVisitorUtils {
functions.put("lower", Lcase.instance);
functions.put("hex", Hex.instance);
functions.put("unhex", Unhex.instance);
functions.put("greatest", Greatest.instance);
functions.put("least", Least.instance);
functions.put("isnull", Isnull.instance);
functions.put("if", If.instance);
}
public static boolean visit(SQLEvalVisitor visitor, SQLMethodInvokeExpr x) {
@@ -518,6 +526,8 @@ public class SQLEvalVisitorUtils {
char ch = (char) intValue;
x.putAttribute(EVAL_VALUE, Character.toString(ch));
}
} else if ("CURRENT_USER".equals(methodName)) {
x.putAttribute(EVAL_VALUE, "CURRENT_USER");
}
return false;
}
@@ -954,7 +964,7 @@ public class SQLEvalVisitorUtils {
if (val instanceof Short) {
return (Short) val;
}
if (val instanceof String) {
return Short.parseShort((String) val);
}
@@ -975,7 +985,7 @@ public class SQLEvalVisitorUtils {
if (val instanceof String) {
return Integer.parseInt((String) val);
}
if (val instanceof List) {
List list = (List) val;
if (list.size() == 1) {
@@ -983,6 +993,14 @@ public class SQLEvalVisitorUtils {
}
}
if (val instanceof Boolean) {
if (((Boolean) val).booleanValue()) {
return 1;
} else {
return 0;
}
}
return ((Number) val).intValue();
}
@@ -995,12 +1013,11 @@ public class SQLEvalVisitorUtils {
if (val instanceof Long) {
return (Long) val;
}
if (val instanceof String) {
return Long.parseLong((String) val);
}
if (val instanceof List) {
List list = (List) val;
if (list.size() == 1) {
@@ -1424,7 +1441,7 @@ public class SQLEvalVisitorUtils {
if (b == null) {
return a;
}
if (a instanceof Date || b instanceof Date) {
return SQLEvalVisitor.EVAL_ERROR;
}
@@ -1460,8 +1477,8 @@ public class SQLEvalVisitorUtils {
if (a instanceof Byte || b instanceof Byte) {
return castToByte(a) - castToByte(b);
}
if(a instanceof String && b instanceof String) {
if (a instanceof String && b instanceof String) {
return castToLong(a) - castToLong(b);
}