修正OracleExportParameterVisitor无法解析数字类型的参数值

2023-03-04 19:17:16 +00:00 · 2017-04-27 14:52:22 +08:00
parent 1d99c00f2c
commit 126c79a8e9
2 changed files with 53 additions and 0 deletions
@@ -0,0 +1,46 @@
+package com.alibaba.druid.bvt.sql.oracle.visitor;
+
+import java.util.List;
+
+import org.junit.Assert;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
+import com.alibaba.druid.sql.dialect.oracle.visitor.OracleExportParameterVisitor;
+import com.alibaba.druid.sql.parser.SQLStatementParser;
+import com.alibaba.druid.sql.visitor.ExportParameterVisitor;
+import com.alibaba.druid.util.JdbcConstants;
+
+import junit.framework.TestCase;
+
+public class ExportParameterDotNumberTest extends TestCase {
+	/**
+	* Logger for this class
+	*/
+	private static final Logger LOG = LoggerFactory.getLogger(ExportParameterDotNumberTest.class);
+
+	String dbType = JdbcConstants.MYSQL;
+
+	public void test_exportParameter() throws Exception {
+		String[] sqls = { "INSERT INTO test_tab1 (name) VALUES ( 2.0  )",
+				 "INSERT INTO test_tab1 (name) VALUES ( 2  )",
+		};
+		for(String sql : sqls){
+			final StringBuilder out = new StringBuilder();
+			final ExportParameterVisitor visitor = new OracleExportParameterVisitor(out);
+			SQLStatementParser parser = new OracleStatementParser(sql);
+			final SQLStatement parseStatement = parser.parseStatement();
+			parseStatement.accept(visitor);
+			final List<Object> plist = visitor.getParameters();
+			LOG.info("from_sql:{}",sql);
+			sql = out.toString();
+		
+			LOG.info("to_sql:{} ==> plist:{}",sql,plist);
+			
+			Assert.assertTrue(plist.size()>0);
+		}
+	}
+
+}