improve sqlparser

src/main/java/com/alibaba/druid/sql/dialect/mysql/ast/statement/MySqlLoadDataInFileStatement.java

@@ -48,7 +48,9 @@ public class MySqlLoadDataInFileStatement extends MySqlStatementImpl {
 
     private SQLExpr             ignoreLinesNumber;
 
-    private final List<SQLExpr> setList                   = new ArrayList<SQLExpr>();
+    private List<SQLExpr>  setList                   = new ArrayList<SQLExpr>();
+
+    private List<SQLExpr>  columns                   = new ArrayList<SQLExpr>();
 
     public boolean isLowPriority() {
         return lowPriority;
@@ -188,4 +190,19 @@ public class MySqlLoadDataInFileStatement extends MySqlStatementImpl {
         }
         visitor.endVisit(this);
     }
+
+    
+    public List<SQLExpr> getColumns() {
+        return columns;
+    }
+
+    
+    public void setColumns(List<SQLExpr> columns) {
+        this.columns = columns;
+    }
+
+    
+    public void setSetList(List<SQLExpr> setList) {
+        this.setList = setList;
+    }
 }

src/main/java/com/alibaba/druid/sql/dialect/mysql/parser/MySqlStatementParser.java

@@ -1787,19 +1787,61 @@ public class MySqlStatementParser extends SQLStatementParser {
         }
 
         if (identifierEquals("FIELDS") || identifierEquals("COLUMNS")) {
-            throw new ParserException("TODO");
+            lexer.nextToken();
+            if (identifierEquals("TERMINATED")) {
+                lexer.nextToken();
+                accept(Token.BY);
+                stmt.setColumnsTerminatedBy((SQLLiteralExpr) this.exprParser.expr());
+            }
+
+            if (identifierEquals("OPTIONALLY")) {
+                stmt.setColumnsEnclosedOptionally(true);
+                lexer.nextToken();
+            }
+
+            if (identifierEquals("ENCLOSED")) {
+                lexer.nextToken();
+                accept(Token.BY);
+                stmt.setColumnsEnclosedBy((SQLLiteralExpr) this.exprParser.expr());
+            }
+
+            if (identifierEquals("ESCAPED")) {
+                lexer.nextToken();
+                accept(Token.BY);
+                stmt.setColumnsEscaped((SQLLiteralExpr) this.exprParser.expr());
+            }
         }
 
         if (identifierEquals("LINES")) {
-            throw new ParserException("TODO");
+            lexer.nextToken();
+            if (identifierEquals("STARTING")) {
+                lexer.nextToken();
+                accept(Token.BY);
+                stmt.setLinesStartingBy((SQLLiteralExpr) this.exprParser.expr());
+            }
+
+            if (identifierEquals("TERMINATED")) {
+                lexer.nextToken();
+                accept(Token.BY);
+                stmt.setLinesTerminatedBy((SQLLiteralExpr) this.exprParser.expr());
+            }
         }
 
         if (identifierEquals(IGNORE)) {
-            throw new ParserException("TODO");
+            lexer.nextToken();
+            stmt.setIgnoreLinesNumber((SQLLiteralExpr) this.exprParser.expr());
+            acceptIdentifier("LINES");
+        }
+
+        if (lexer.token() == Token.LPAREN) {
+            lexer.nextToken();
+            this.exprParser.exprList(stmt.getColumns(), stmt);
+            accept(Token.RPAREN);
         }
 
         if (lexer.token() == Token.SET) {
-            throw new ParserException("TODO");
+            lexer.nextToken();
+            this.exprParser.exprList(stmt.getSetList(), stmt);
         }
 
         return stmt;

src/main/java/com/alibaba/druid/sql/dialect/mysql/visitor/MySqlOutputVisitor.java

@@ -1037,7 +1037,19 @@ public class MySqlOutputVisitor extends SQLASTOutputVisitor implements MySqlASTV
                 x.getLinesTerminatedBy().accept(this);
             }
         }
+        
+        if(x.getIgnoreLinesNumber() != null) {
+            print(" IGNORE ");
+            x.getIgnoreLinesNumber().accept(this);
+            print(" LINES");
+        }
 
+        if (x.getColumns().size() != 0) {
+            print(" (");
+            printAndAccept(x.getColumns(), ", ");
+            print(")");
+        }
+        
         if (x.getSetList().size() != 0) {
             print(" SET ");
             printAndAccept(x.getSetList(), ", ");