refactor for pull request #2224

alibaba · Feb 11, 2018 · f1dd6c7 · f1dd6c7
1 parent a588725
commit f1dd6c7
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/main/java/com/alibaba/druid/sql/visitor/SQLASTOutputVisitor.java b/src/main/java/com/alibaba/druid/sql/visitor/SQLASTOutputVisitor.java
@@ -18,6 +18,8 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.Reader;
+import java.math.BigDecimal;
+import java.math.BigInteger;
 import java.sql.Blob;
 import java.sql.Clob;
 import java.sql.NClob;
@@ -1327,9 +1329,9 @@ public boolean visit(SQLIntegerExpr x) {
     }
 
     protected void printInteger(SQLIntegerExpr x, boolean parameterized) {
-        String val = x.getNumber().toString();
+        Number number = x.getNumber();
 
-        if ("1".equals(val)) {
+        if (number.equals(Integer.valueOf(1))) {
             if (JdbcConstants.ORACLE.equals(dbType)) {
                 SQLObject parent = x.getParent();
                 if (parent instanceof SQLBinaryOpExpr) {
@@ -1357,7 +1359,11 @@ protected void printInteger(SQLIntegerExpr x, boolean parameterized) {
             return;
         }
 
-        print(val);
+        if (number instanceof BigDecimal || number instanceof BigInteger) {
+            print(number.toString());
+        } else {
+            print(number.longValue());
+        }
     }
 
     public boolean visit(SQLMethodInvokeExpr x) {