sql server parser

src/main/java/com/alibaba/druid/sql/dialect/sqlserver/ast/SQLServerStatement.java

@@ -0,0 +1,7 @@
+package com.alibaba.druid.sql.dialect.sqlserver.ast;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+
+public interface SQLServerStatement extends SQLStatement {
+
+}

src/main/java/com/alibaba/druid/sql/dialect/sqlserver/ast/SQLServerUseStatement.java

@@ -0,0 +1,27 @@
+package com.alibaba.druid.sql.dialect.sqlserver.ast;
+
+import com.alibaba.druid.sql.ast.SQLName;
+import com.alibaba.druid.sql.dialect.sqlserver.visitor.SQLServerASTVisitor;
+
+public class SQLServerUseStatement extends SQLServerObjectImpl implements SQLServerStatement {
+
+    private static final long serialVersionUID = 1L;
+    private SQLName           database;
+
+    public SQLName getDatabase() {
+        return database;
+    }
+
+    public void setDatabase(SQLName database) {
+        this.database = database;
+    }
+
+    @Override
+    public void accept0(SQLServerASTVisitor visitor) {
+        if (visitor.visit(this)) {
+            acceptChild(visitor, database);
+        }
+        visitor.endVisit(this);
+    }
+
+}

src/main/java/com/alibaba/druid/sql/dialect/sqlserver/parser/SQLServerLexer.java

@@ -86,6 +86,7 @@ public class SQLServerLexer extends Lexer {
         map.put("WITH", Token.WITH);
         map.put("CURSOR", Token.CURSOR);
         map.put("TOP", Token.TOP);
+        map.put("USE", Token.USE);
 
         DEFAULT_SQL_SERVER_KEYWORDS = new Keywords(map);
     }

src/main/java/com/alibaba/druid/sql/dialect/sqlserver/visitor/SQLServerASTVisitor.java

@@ -1,6 +1,7 @@
 package com.alibaba.druid.sql.dialect.sqlserver.visitor;
 
 import com.alibaba.druid.sql.dialect.sqlserver.ast.SQLServerSelectQueryBlock;
+import com.alibaba.druid.sql.dialect.sqlserver.ast.SQLServerUseStatement;
 import com.alibaba.druid.sql.dialect.sqlserver.ast.Top;
 import com.alibaba.druid.sql.visitor.SQLASTVisitor;
 
@@ -13,4 +14,8 @@ public interface SQLServerASTVisitor extends SQLASTVisitor {
     boolean visit(Top x);
 
     void endVisit(Top x);
+    
+    boolean visit(SQLServerUseStatement x);
+    
+    void endVisit(SQLServerUseStatement x);
 }

src/main/java/com/alibaba/druid/sql/dialect/sqlserver/visitor/SQLServerASTVisitorAdapter.java

@@ -1,6 +1,7 @@
 package com.alibaba.druid.sql.dialect.sqlserver.visitor;
 
 import com.alibaba.druid.sql.dialect.sqlserver.ast.SQLServerSelectQueryBlock;
+import com.alibaba.druid.sql.dialect.sqlserver.ast.SQLServerUseStatement;
 import com.alibaba.druid.sql.dialect.sqlserver.ast.Top;
 import com.alibaba.druid.sql.visitor.SQLASTVisitorAdapter;
 
@@ -26,4 +27,14 @@ public class SQLServerASTVisitorAdapter extends SQLASTVisitorAdapter implements
 
     }
 
+    @Override
+    public boolean visit(SQLServerUseStatement x) {
+        return true;
+    }
+
+    @Override
+    public void endVisit(SQLServerUseStatement x) {
+        
+    }
+
 }

src/main/java/com/alibaba/druid/sql/dialect/sqlserver/visitor/SQLServerOutputVisitor.java

@@ -2,6 +2,7 @@ package com.alibaba.druid.sql.dialect.sqlserver.visitor;
 
 import com.alibaba.druid.sql.ast.SQLSetQuantifier;
 import com.alibaba.druid.sql.dialect.sqlserver.ast.SQLServerSelectQueryBlock;
+import com.alibaba.druid.sql.dialect.sqlserver.ast.SQLServerUseStatement;
 import com.alibaba.druid.sql.dialect.sqlserver.ast.Top;
 import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor;
 
@@ -67,4 +68,16 @@ public class SQLServerOutputVisitor extends SQLASTOutputVisitor implements SQLSe
 
     }
 
+    @Override
+    public boolean visit(SQLServerUseStatement x) {
+        print("USE ");
+        x.getDatabase().accept(this);
+        return false;
+    }
+
+    @Override
+    public void endVisit(SQLServerUseStatement x) {
+        
+    }
+
 }

src/main/java/com/alibaba/druid/sql/dialect/sqlserver/visitor/SQLServerSchemaStatVisitor.java

@@ -2,6 +2,7 @@ package com.alibaba.druid.sql.dialect.sqlserver.visitor;
 
 import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
 import com.alibaba.druid.sql.dialect.sqlserver.ast.SQLServerSelectQueryBlock;
+import com.alibaba.druid.sql.dialect.sqlserver.ast.SQLServerUseStatement;
 import com.alibaba.druid.sql.dialect.sqlserver.ast.Top;
 import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
 
@@ -28,4 +29,14 @@ public class SQLServerSchemaStatVisitor extends SchemaStatVisitor implements SQL
         
     }
 
+    @Override
+    public boolean visit(SQLServerUseStatement x) {
+        return false;
+    }
+
+    @Override
+    public void endVisit(SQLServerUseStatement x) {
+        
+    }
+
 }

src/main/java/com/alibaba/druid/sql/parser/SQLStatementParser.java

@@ -134,6 +134,12 @@ public class SQLStatementParser extends SQLParser {
                 statementList.add(stmt);
                 continue;
             }
+            
+            if (lexer.token() == Token.USE) {
+                SQLStatement stmt = parseUse();
+                statementList.add(stmt);
+                continue;
+            }
 
             if (identifierEquals("CALL")) {
                 SQLCallStatement stmt = parseCall();
@@ -148,6 +154,10 @@ public class SQLStatementParser extends SQLParser {
             throw new ParserException("TODO " + lexer.token());
         }
     }
+    
+    public SQLStatement parseUse() {
+        throw new ParserException("TODO " + lexer.token());
+    }
 
     protected SQLDropTableStatement parseDropTable(boolean acceptDrop) {
         if (acceptDrop) {

src/main/java/com/alibaba/druid/sql/parser/Token.java

@@ -70,6 +70,7 @@ public enum Token {
     CAST("CAST"),
     USER("USER"),
     COLUMN("COLUMN"),
+    USE("USE"),
 
     AND("AND"), 
     OR("OR"),