Add missing aggregate functions and cast() to functions list

Author: nene

src/languages/tidb/tidb.functions.ts

@@ -1,6 +1,8 @@
 export const functions: string[] = [
   // https://docs.pingcap.com/tidb/stable/sql-statement-show-builtins
   // https://docs.pingcap.com/tidb/stable/functions-and-operators-overview
+  // + MySQL aggregate functions: https://dev.mysql.com/doc/refman/8.0/en/aggregate-functions.html
+  // + MySQL window functions: https://dev.mysql.com/doc/refman/8.0/en/window-functions-usage.html
   'ABS',
   'ACOS',
   'ADDDATE',
@@ -13,16 +15,21 @@ export const functions: string[] = [
   'ASIN',
   'ATAN',
   'ATAN2',
+  'AVG',
   'BENCHMARK',
   'BIN',
   'BIN_TO_UUID',
+  'BIT_AND',
   'BIT_COUNT',
   'BIT_LENGTH',
+  'BIT_OR',
+  'BIT_XOR',
   'BITAND',
   'BITNEG',
   'BITOR',
   'BITXOR',
   'CASE',
+  'CAST',
   'CEIL',
   'CEILING',
   'CHAR_FUNC',
@@ -41,7 +48,9 @@ export const functions: string[] = [
   'CONVERT_TZ',
   'COS',
   'COT',
+  'COUNT',
   'CRC32',
+  'CUME_DIST',
   'CURDATE',
   'CURRENT_DATE',
   'CURRENT_RESOURCE_GROUP',
@@ -64,6 +73,7 @@ export const functions: string[] = [
   'DECODE',
   'DEFAULT_FUNC',
   'DEGREES',
+  'DENSE_RANK',
   'DES_DECRYPT',
   'DES_ENCRYPT',
   'DIV',
@@ -76,6 +86,7 @@ export const functions: string[] = [
   'EXTRACT',
   'FIELD',
   'FIND_IN_SET',
+  'FIRST_VALUE',
   'FLOOR',
   'FORMAT',
   'FORMAT_BYTES',
@@ -89,6 +100,7 @@ export const functions: string[] = [
   'GET_LOCK',
   'GETPARAM',
   'GREATEST',
+  'GROUP_CONCAT',
   'GROUPING',
   'GT',
   'HEX',
@@ -116,6 +128,7 @@ export const functions: string[] = [
   'ISNULL',
   'ISTRUE',
   'JSON_ARRAY',
+  'JSON_ARRAYAGG',
   'JSON_ARRAY_APPEND',
   'JSON_ARRAY_INSERT',
   'JSON_CONTAINS',
@@ -130,6 +143,7 @@ export const functions: string[] = [
   'JSON_MERGE_PATCH',
   'JSON_MERGE_PRESERVE',
   'JSON_OBJECT',
+  'JSON_OBJECTAGG',
   'JSON_OVERLAPS',
   'JSON_PRETTY',
   'JSON_QUOTE',
@@ -142,11 +156,14 @@ export const functions: string[] = [
   'JSON_TYPE',
   'JSON_UNQUOTE',
   'JSON_VALID',
+  'LAG',
   'LAST_DAY',
   'LAST_INSERT_ID',
+  'LAST_VALUE',
   'LASTVAL',
   'LCASE',
   'LE',
+  'LEAD',
   'LEAST',
   'LEFT',
   'LEFTSHIFT',
@@ -168,9 +185,11 @@ export const functions: string[] = [
   'MAKEDATE',
   'MAKETIME',
   'MASTER_POS_WAIT',
+  'MAX',
   'MD5',
   'MICROSECOND',
   'MID',
+  'MIN',
   'MINUS',
   'MINUTE',
   'MOD',
@@ -182,13 +201,16 @@ export const functions: string[] = [
   'NEXTVAL',
   'NOT',
   'NOW',
+  'NTH_VALUE',
+  'NTILE',
   'NULLEQ',
   'OCT',
   'OCTET_LENGTH',
   'OLD_PASSWORD',
   // 'OR',
   'ORD',
   'PASSWORD_FUNC',
+  'PERCENT_RANK',
   'PERIOD_ADD',
   'PERIOD_DIFF',
   'PI',
@@ -201,6 +223,7 @@ export const functions: string[] = [
   'RADIANS',
   'RAND',
   'RANDOM_BYTES',
+  'RANK',
   'REGEXP',
   'REGEXP_INSTR',
   'REGEXP_LIKE',
@@ -215,6 +238,7 @@ export const functions: string[] = [
   'RIGHTSHIFT',
   'ROUND',
   'ROW_COUNT',
+  'ROW_NUMBER',
   'RPAD',
   'RTRIM',
   'SCHEMA',
@@ -232,13 +256,18 @@ export const functions: string[] = [
   'SM3',
   'SPACE',
   'SQRT',
+  'STD',
+  'STDDEV',
+  'STDDEV_POP',
+  'STDDEV_SAMP',
   'STR_TO_DATE',
   'STRCMP',
   'SUBDATE',
   'SUBSTR',
   'SUBSTRING',
   'SUBSTRING_INDEX',
   'SUBTIME',
+  'SUM',
   'SYSDATE',
   'SYSTEM_USER',
   'TAN',
@@ -283,6 +312,9 @@ export const functions: string[] = [
   'UUID_SHORT',
   'UUID_TO_BIN',
   'VALIDATE_PASSWORD_STRENGTH',
+  'VAR_POP',
+  'VAR_SAMP',
+  'VARIANCE',
   'VERSION',
   'VITESS_HASH',
   'WEEK',

src/languages/tidb/tidb.keywords.ts

@@ -23,7 +23,6 @@ export const keywords: string[] = [
   'CONVERT', // (R)
   'CREATE', // (R)
   'CROSS', // (R)
-  'CUME_DIST', // (R)
   'CURRENT_DATE', // (R)
   'CURRENT_ROLE', // (R)
   'CURRENT_TIME', // (R)
@@ -39,7 +38,6 @@ export const keywords: string[] = [
   'DEFAULT', // (R)
   'DELAYED', // (R)
   'DELETE', // (R)
-  'DENSE_RANK', // (R)
   'DESC', // (R)
   'DESCRIBE', // (R)
   'DISTINCT', // (R)
@@ -58,7 +56,6 @@ export const keywords: string[] = [
   'EXPLAIN', // (R)
   'FALSE', // (R)
   'FETCH', // (R)
-  'FIRST_VALUE', // (R)
   'FOR', // (R)
   'FORCE', // (R)
   'FOREIGN', // (R)
@@ -91,9 +88,6 @@ export const keywords: string[] = [
   'KEY', // (R)
   'KEYS', // (R)
   'KILL', // (R)
-  'LAG', // (R)
-  'LAST_VALUE', // (R)
-  'LEAD', // (R)
   'LEADING', // (R)
   'LEAVE', // (R)
   'LEFT', // (R)
@@ -115,8 +109,6 @@ export const keywords: string[] = [
   'NATURAL', // (R)
   'NOT', // (R)
   'NO_WRITE_TO_BINLOG', // (R)
-  'NTH_VALUE', // (R)
-  'NTILE', // (R)
   'NULL', // (R)
   'OF', // (R)
   'ON', // (R)
@@ -130,11 +122,9 @@ export const keywords: string[] = [
   'OUTFILE', // (R)
   'OVER', // (R)
   'PARTITION', // (R)
-  'PERCENT_RANK', // (R)
   'PRIMARY', // (R)
   'PROCEDURE', // (R)
   'RANGE', // (R)
-  'RANK', // (R)
   'READ', // (R)
   'RECURSIVE', // (R)
   'REFERENCES', // (R)
@@ -150,7 +140,6 @@ export const keywords: string[] = [
   'RLIKE', // (R)
   'ROW', // (R)
   'ROWS', // (R)
-  'ROW_NUMBER', // (R)
   'SECOND_MICROSECOND', // (R)
   'SELECT', // (R)
   'SET', // (R)