|
| 1 | +#ifndef SQL_PARSER_MYSQL_IDENTIFIER_H |
| 2 | +#define SQL_PARSER_MYSQL_IDENTIFIER_H |
| 3 | + |
| 4 | +#include "sql_parser/token.h" |
| 5 | +#include <cstddef> |
| 6 | + |
| 7 | +namespace sql_parser { |
| 8 | + |
| 9 | +// Keywords may be identifiers after a qualification dot, including reserved ones. |
| 10 | +inline bool mysql_identifier_word(const Token& token) { |
| 11 | + return token.type == TokenType::TK_IDENTIFIER || |
| 12 | + (token.type >= TokenType::TK_SELECT && token.type <= TokenType::TK_RECURSIVE); |
| 13 | +} |
| 14 | + |
| 15 | +inline bool mysql_identifier_token(const Token& token) { |
| 16 | + if (!mysql_identifier_word(token)) return false; |
| 17 | + if (token.source.ptr != token.text.ptr) return true; // delimited identifier |
| 18 | + // Sorted reserved-word baseline from INFORMATION_SCHEMA.KEYWORDS, MySQL |
| 19 | + // 8.4.8 (0896fcd61dec11a0904166911a0126f59daaa1bf), also reserved in |
| 20 | + // 9.7.2 (008e09c2834b98143a8c067d4d225c90953050cf). Version-specific |
| 21 | + // additions are not rejected without a server-version parser setting. |
| 22 | + static constexpr const char* reserved[] = { |
| 23 | + "ACCESSIBLE", "ADD", "ALL", "ALTER", "ANALYZE", |
| 24 | + "AND", "AS", "ASC", "ASENSITIVE", "BEFORE", |
| 25 | + "BETWEEN", "BIGINT", "BINARY", "BLOB", "BOTH", |
| 26 | + "BY", "CALL", "CASCADE", "CASE", "CHANGE", |
| 27 | + "CHAR", "CHARACTER", "CHECK", "COLLATE", "COLUMN", |
| 28 | + "CONDITION", "CONSTRAINT", "CONTINUE", "CONVERT", "CREATE", |
| 29 | + "CROSS", "CUME_DIST", "CURRENT_DATE", "CURRENT_TIME", "CURRENT_TIMESTAMP", |
| 30 | + "CURRENT_USER", "CURSOR", "DATABASE", "DATABASES", "DAY_HOUR", |
| 31 | + "DAY_MICROSECOND", "DAY_MINUTE", "DAY_SECOND", "DEC", "DECIMAL", |
| 32 | + "DECLARE", "DEFAULT", "DELAYED", "DELETE", "DENSE_RANK", |
| 33 | + "DESC", "DESCRIBE", "DETERMINISTIC", "DISTINCT", "DISTINCTROW", |
| 34 | + "DIV", "DOUBLE", "DROP", "DUAL", "EACH", |
| 35 | + "ELSE", "ELSEIF", "EMPTY", "ENCLOSED", "ESCAPED", |
| 36 | + "EXCEPT", "EXISTS", "EXIT", "EXPLAIN", "FALSE", |
| 37 | + "FETCH", "FIRST_VALUE", "FLOAT", "FLOAT4", "FLOAT8", |
| 38 | + "FOR", "FORCE", "FOREIGN", "FROM", "FULLTEXT", |
| 39 | + "FUNCTION", "GENERATED", "GET", "GRANT", "GROUP", |
| 40 | + "GROUPING", "GROUPS", "HAVING", "HIGH_PRIORITY", "HOUR_MICROSECOND", |
| 41 | + "HOUR_MINUTE", "HOUR_SECOND", "IF", "IGNORE", "IN", |
| 42 | + "INDEX", "INFILE", "INNER", "INOUT", "INSENSITIVE", |
| 43 | + "INSERT", "INT", "INT1", "INT2", "INT3", |
| 44 | + "INT4", "INT8", "INTEGER", "INTERSECT", "INTERVAL", |
| 45 | + "INTO", "IO_AFTER_GTIDS", "IO_BEFORE_GTIDS", "IS", "ITERATE", |
| 46 | + "JOIN", "JSON_TABLE", "KEY", "KEYS", "KILL", |
| 47 | + "LAG", "LAST_VALUE", "LATERAL", "LEAD", "LEADING", |
| 48 | + "LEAVE", "LEFT", "LIKE", "LIMIT", "LINEAR", |
| 49 | + "LINES", "LOAD", "LOCALTIME", "LOCALTIMESTAMP", "LOCK", |
| 50 | + "LONG", "LONGBLOB", "LONGTEXT", "LOOP", "LOW_PRIORITY", |
| 51 | + "MATCH", "MAXVALUE", "MEDIUMBLOB", "MEDIUMINT", "MEDIUMTEXT", |
| 52 | + "MIDDLEINT", "MINUTE_MICROSECOND", "MINUTE_SECOND", "MOD", "MODIFIES", |
| 53 | + "NATURAL", "NOT", "NO_WRITE_TO_BINLOG", "NTH_VALUE", "NTILE", |
| 54 | + "NULL", "NUMERIC", "OF", "ON", "OPTIMIZE", |
| 55 | + "OPTIMIZER_COSTS", "OPTION", "OPTIONALLY", "OR", "ORDER", |
| 56 | + "OUT", "OUTER", "OUTFILE", "OVER", "PARTITION", |
| 57 | + "PERCENT_RANK", "PRECISION", "PRIMARY", "PROCEDURE", "PURGE", |
| 58 | + "RANGE", "RANK", "READ", "READS", "READ_WRITE", |
| 59 | + "REAL", "RECURSIVE", "REFERENCES", "REGEXP", "RELEASE", |
| 60 | + "RENAME", "REPEAT", "REPLACE", "REQUIRE", "RESIGNAL", |
| 61 | + "RESTRICT", "RETURN", "REVOKE", "RIGHT", "RLIKE", |
| 62 | + "ROW", "ROWS", "ROW_NUMBER", "SCHEMA", "SCHEMAS", |
| 63 | + "SECOND_MICROSECOND", "SELECT", "SENSITIVE", "SEPARATOR", "SET", |
| 64 | + "SHOW", "SIGNAL", "SMALLINT", "SPATIAL", "SPECIFIC", |
| 65 | + "SQL", "SQLEXCEPTION", "SQLSTATE", "SQLWARNING", "SQL_BIG_RESULT", |
| 66 | + "SQL_CALC_FOUND_ROWS", "SQL_SMALL_RESULT", "SSL", "STARTING", "STORED", |
| 67 | + "STRAIGHT_JOIN", "SYSTEM", "TABLE", "TERMINATED", "THEN", |
| 68 | + "TINYBLOB", "TINYINT", "TINYTEXT", "TO", "TRAILING", |
| 69 | + "TRIGGER", "TRUE", "UNDO", "UNION", "UNIQUE", |
| 70 | + "UNLOCK", "UNSIGNED", "UPDATE", "USAGE", "USE", |
| 71 | + "USING", "UTC_DATE", "UTC_TIME", "UTC_TIMESTAMP", "VALUES", |
| 72 | + "VARBINARY", "VARCHAR", "VARCHARACTER", "VARYING", "VIRTUAL", |
| 73 | + "WHEN", "WHERE", "WHILE", "WINDOW", "WITH", |
| 74 | + "WRITE", "XOR", "YEAR_MONTH", "ZEROFILL", |
| 75 | + }; |
| 76 | + size_t lo = 0, hi = sizeof(reserved) / sizeof(reserved[0]); |
| 77 | + while (lo < hi) { |
| 78 | + size_t mid = lo + (hi - lo) / 2; |
| 79 | + const char* word = reserved[mid]; |
| 80 | + uint32_t i = 0; |
| 81 | + int difference = 0; |
| 82 | + for (; i < token.text.len && word[i]; ++i) { |
| 83 | + unsigned char ch = static_cast<unsigned char>(token.text.ptr[i]); |
| 84 | + if (ch >= 'a' && ch <= 'z') ch -= 'a' - 'A'; |
| 85 | + difference = static_cast<int>(ch) - static_cast<unsigned char>(word[i]); |
| 86 | + if (difference) break; |
| 87 | + } |
| 88 | + if (!difference) difference = i < token.text.len ? 1 : word[i] ? -1 : 0; |
| 89 | + if (!difference) return false; |
| 90 | + if (difference < 0) hi = mid; |
| 91 | + else lo = mid + 1; |
| 92 | + } |
| 93 | + return true; |
| 94 | +} |
| 95 | + |
| 96 | +} // namespace sql_parser |
| 97 | +#endif // SQL_PARSER_MYSQL_IDENTIFIER_H |
0 commit comments