From aefdb8eb59c5e638cba1eec73b8649e3e039c170 Mon Sep 17 00:00:00 2001
From: wenshao <szujobs@hotmail.com>
Date: Wed, 18 May 2011 10:19:04 +0000
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=B5=8B=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

git-svn-id: http://code.alibabatech.com/svn/druid/trunk@120 b9813039-fb51-4c41-a8b9-e21c2acb5095
---
 .../druid/sql/parser/SQLLexerTest2.java       |   2 +-
 .../druid/sql/parser/SQLMergeTest.java        | 119 +++++++++++++++++-
 2 files changed, 116 insertions(+), 5 deletions(-)

diff --git a/src/test/java/com/alibaba/druid/sql/parser/SQLLexerTest2.java b/src/test/java/com/alibaba/druid/sql/parser/SQLLexerTest2.java
index ba2c0dd34b..b25c1c3947 100644
--- a/src/test/java/com/alibaba/druid/sql/parser/SQLLexerTest2.java
+++ b/src/test/java/com/alibaba/druid/sql/parser/SQLLexerTest2.java
@@ -6,7 +6,7 @@ public class SQLLexerTest2 extends TestCase {
 
     public void test_lexer() throws Exception {
         String sql = "SELECT * FROM T WHERE F1 = ? ORDER BY F2";
-        SQLLexer lexer = new SQLLexer(sql);
+        Lexer lexer = new Lexer(sql);
         for (;;) {
             lexer.nextToken();
             Token tok = lexer.token();
diff --git a/src/test/java/com/alibaba/druid/sql/parser/SQLMergeTest.java b/src/test/java/com/alibaba/druid/sql/parser/SQLMergeTest.java
index e12fd847e8..7d56c2d04e 100644
--- a/src/test/java/com/alibaba/druid/sql/parser/SQLMergeTest.java
+++ b/src/test/java/com/alibaba/druid/sql/parser/SQLMergeTest.java
@@ -2,16 +2,38 @@ package com.alibaba.druid.sql.parser;
 
 import java.util.List;
 
+import junit.framework.TestCase;
+
 import com.alibaba.druid.sql.ast.SQLStatement;
 import com.alibaba.druid.sql.ast.expr.SQLInListExpr;
+import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlInsertStatement;
+import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlOutputVisitor;
 import com.alibaba.druid.sql.dialect.oracle.ast.visitor.OracleOutputVisitor;
 import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
 
-import junit.framework.TestCase;
-
 public class SQLMergeTest extends TestCase {
 
-    public void test_merge() throws Exception {
+    public void test_merge_1() throws Exception {
+        String result = merge();
+        System.out.println(result);
+    }
+    
+    public void test_merge_2() throws Exception {
+        String result = merge2();
+        System.out.println(result);
+    }
+
+    void perf() {
+        long startMillis = System.currentTimeMillis();
+        for (int i = 0; i < 1000 * 1000; ++i) {
+            merge();
+        }
+        long millis = System.currentTimeMillis() - startMillis;
+        System.out.println(millis);
+    }
+
+    private String merge() {
         String sql = "SELECT /*mark for picman*/ * FROM WP_ALBUM WHERE MEMBER_ID = ? AND ID IN (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)";
 
         StringBuilder out = new StringBuilder();
@@ -35,6 +57,95 @@ public class SQLMergeTest extends TestCase {
             visitor.println();
         }
 
-        System.out.println(out.toString());
+        return out.toString();
+    }
+    
+    private String merge2() {
+        String sql = "INSERT INTO T (F1, F2, F3, F4, F5) VALUES (?, ?, ?, ?, ?), (?, ?, ?, ?, ?), (?, ?, ?, ?, ?)";
+
+        StringBuilder out = new StringBuilder();
+        MySqlOutputVisitor visitor = new MySqlOutputVisitor(out) {
+            public boolean visit(SQLInListExpr x) {
+                x.getExpr().accept(this);
+
+                if (x.isNot()) {
+                    print(" NOT IN (##)");
+                } else {
+                    print(" IN (##)");
+                }
+                return false;
+            }
+            
+            @Override
+            public boolean visit(MySqlInsertStatement x) {
+                print("INSERT ");
+
+                if (x.isLowPriority()) {
+                    print("LOW_PRIORITY ");
+                }
+
+                if (x.isDelayed()) {
+                    print("DELAYED ");
+                }
+
+                if (x.isHighPriority()) {
+                    print("HIGH_PRIORITY ");
+                }
+
+                if (x.isIgnore()) {
+                    print("IGNORE ");
+                }
+
+                print("INTO ");
+
+                x.getTableName().accept(this);
+
+                if (x.getColumns().size() > 0) {
+                    print(" (");
+                    for (int i = 0, size = x.getColumns().size(); i < size; ++i) {
+                        if (i != 0) {
+                            print(", ");
+                        }
+                        x.getColumns().get(i).accept(this);
+                    }
+                    print(")");
+                }
+
+                if (x.getValuesList().size() != 0) {
+                    print(" VALUES ");
+                    int size = x.getValuesList().size();
+                    if (size == 0) {
+                        print("()");
+                    } else {
+                        for (int i = 0; i < 1; ++i) {
+                            if (i != 0) {
+                                print(", ");
+                            }
+                            x.getValuesList().get(i).accept(this);
+                        }
+                    }
+                }
+                if (x.getQuery() != null) {
+                    print(" ");
+                    x.getQuery().accept(this);
+                }
+
+                if (x.getDuplicateKeyUpdate().size() != 0) {
+                    print(" ON DUPLICATE KEY UPDATE ");
+                    printAndAccept(x.getDuplicateKeyUpdate(), ", ");
+                }
+
+                return false;
+            }
+        };
+        
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        for (SQLStatement statement : statementList) {
+            statement.accept(visitor);
+            visitor.println();
+        }
+
+        return out.toString();
     }
 }