Fixed quota statements

Author: chernser

jdbc-v2/src/main/antlr4/com/clickhouse/jdbc/internal/parser/ClickHouseLexer.g4

@@ -163,6 +163,7 @@ KAFKA : K A F K A;
 KERBEROS : K E R B E R O S;
 KEY          : K E Y;
 KEYS          : K E Y S;
+KEYED       : K E Y E D;
 KILL         : K I L L;
 LAST         : L A S T;
 LAYOUT       : L A Y O U T;
@@ -172,6 +173,7 @@ LEFT         : L E F T;
 LIFETIME     : L I F E T I M E;
 LIKE         : L I K E;
 LIMIT        : L I M I T;
+LIMITS        : L I M I T S;
 LISTEN : L I S T E N ;
 LIVE         : L I V E;
 LOADING : L O A D I N G;
@@ -210,6 +212,7 @@ NULL_SQL     : N U L L; // conflicts with macro NULL
 ODBC : O D B C;
 OFFSET       : O F F S E T;
 ON           : O N;
+ONLY         : O N L Y;
 OPTIMIZE     : O P T I M I Z E;
 OPTION       : O P T I O N;
 ORDER        : O R D E R;
@@ -238,6 +241,7 @@ PROFILE      : P R O F I L E;
 PROFILES      : P R O F I L E S;
 PROJECTION   : P R O J E C T I O N;
 PULLING : P U L L I N G ;
+RANDOMIZED : R A N D O M I Z E D;
 REVOKE : R E V O K E;
 QUARTER      : Q U A R T E R;
 QUEUE : Q U E U E ;
@@ -317,6 +321,7 @@ TIMESTAMP    : T I M E S T A M P;
 TOP          : T O P;
 TOTALS       : T O T A L S;
 TO           : T O;
+TRACKING     : T R A C K I N G;
 TRAILING     : T R A I L I N G;
 TRANSACTION  : T R A N S A C T I O N;
 TREE         : T R E E;

jdbc-v2/src/main/antlr4/com/clickhouse/jdbc/internal/parser/ClickHouseParser.g4

@@ -190,6 +190,18 @@ createStmt
             | ( INHERIT identifier))? (TO identifier | ALL | ALL EXCEPT identifier)? # createProfileStmt
     | CREATE FUNCTION identifier clusterClause? AS LPAREN (identifier)? (COMMA identifier)? RPAREN ARROW .+? #createFunctionStmt
     | CREATE NAMED COLLECTION (IF NOT EXISTS)? identifier clusterClause? AS nameCollectionKey (COMMA nameCollectionKey)* #createNamedCollectionStmt
+    | CREATE QUOTA (IF NOT EXISTS | OR REPLACE)? identifier clusterClause? (IN identifier)?
+        (KEYED BY identifier | NOT KEYED)?
+        quotaForClause (COMMA quotaForClause)*
+        (TO (identifier (COMMA identifier)* | ALL | CURRENT_USER | ALL EXCEPT identifier (COMMA identifier)* ))? # createQuotaStmt
+    ;
+
+quotaMaxExpr
+    : identifier EQ_SINGLE numberLiteral
+    ;
+
+quotaForClause
+    : FOR RANDOMIZED? INTERVAL numberLiteral interval (MAX quotaMaxExpr (COMMA quotaMaxExpr)*)+?
     ;
 
 nameCollectionKey

jdbc-v2/src/test/java/com/clickhouse/jdbc/internal/SqlParserTest.java

@@ -538,6 +538,8 @@ public static Object[][] testStatementWithoutResultSetDP() {
                 {"CREATE ROW POLICY pol1 ON mydb.table1 USING b=1 TO mira, peter", 0, false},
                 {"CREATE ROW POLICY pol2 ON mydb.table1 USING c=2 TO peter, antonio", 0, false},
                 {"CREATE ROW POLICY pol2 ON mydb.table1 USING c=2 AS RESTRICTIVE TO peter, antonio", 0, false},
+                {"CREATE QUOTA qA FOR INTERVAL 15 month MAX queries = 123 TO role1, role2", 0, false},
+                {"CREATE QUOTA qA FOR INTERVAL 15 month MAX queries = 123 TO ALL EXCEPT role3", 0, false},
                 {"CREATE QUOTA qA FOR INTERVAL 15 month MAX queries = 123 TO CURRENT_USER", 0, false},
                 {"CREATE QUOTA qB FOR INTERVAL 30 minute MAX execution_time = 0.5, FOR INTERVAL 5 quarter MAX queries = 321, errors = 10 TO default", 0, false},
                 {"CREATE SETTINGS PROFILE max_memory_usage_profile SETTINGS max_memory_usage = 100000001 MIN 90000000 MAX 110000000 TO robin", 0, false},