impala parser support upsert followed by with clause

core/src/main/java/com/alibaba/druid/sql/dialect/impala/parser/ImpalaLexer.java

@@ -4,11 +4,17 @@
 import com.alibaba.druid.sql.dialect.hive.parser.HiveLexer;
 import com.alibaba.druid.sql.parser.Keywords;
 import com.alibaba.druid.sql.parser.SQLParserFeature;
+import com.alibaba.druid.sql.parser.Token;
+
+import java.util.Map;
 
 public class ImpalaLexer extends HiveLexer {
     @Override
     protected Keywords loadKeywords() {
-        return super.loadKeywords();
+        Keywords keywords = super.loadKeywords();
+        Map<String, Token> map = keywords.getKeywords();
+        map.put("UPSERT", Token.UPSERT);
+        return new Keywords(map);
     }
 
     public ImpalaLexer(String input) {

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

@@ -5826,6 +5826,10 @@ public SQLStatement parseWith() {
             SQLInsertStatement insert = (SQLInsertStatement) this.parseInsert();
             insert.setWith(with);
             stmt = insert;
+        } else if (lexer.token == Token.UPSERT) {
+            SQLInsertStatement insert = (SQLInsertStatement) this.parseUpsert();
+            insert.setWith(with);
+            stmt = insert;
         } else if (lexer.token == Token.FROM) {
             HiveMultiInsertStatement insert = (HiveMultiInsertStatement) this.parseInsert();
             insert.setWith(with);

core/src/test/resources/bvt/parser/impala/5.txt

@@ -15,4 +15,14 @@ UPSERT [SHUFFLE] INTO production_table
 [NOSHUFFLE] SELECT *
 FROM staging_table
 WHERE c1 IS NOT NULL
-	AND c2 > 0;
+	AND c2 > 0;
+------------------------------------------------------------------------------------------------------------------------
+with tmp as (select c,d from prj1.kudu_table2) upsert into prj1.kudu_table select * from tmp
+--------------------
+WITH tmp AS (
+		SELECT c, d
+		FROM prj1.kudu_table2
+	)
+UPSERT INTO prj1.kudu_table
+SELECT *
+FROM tmp