parTokenizer.c 20.3 KB
Newer Older
H
hzcheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

H
hjxilinx 已提交
16
#include "os.h"
X
Xiaoyu Wang 已提交
17
#include "parToken.h"
H
hjxilinx 已提交
18
#include "taosdef.h"
X
Xiaoyu Wang 已提交
19
#include "thash.h"
20
#include "ttokendef.h"
H
hzcheng 已提交
21

S
slguan 已提交
22
// All the keywords of the SQL language are stored in a hash table
H
hzcheng 已提交
23
typedef struct SKeyword {
S
slguan 已提交
24
  const char* name;  // The keyword name
H
Haojun Liao 已提交
25
  uint16_t    type;  // type
S
slguan 已提交
26
  uint8_t     len;   // length
H
hzcheng 已提交
27 28
} SKeyword;

X
Xiaoyu Wang 已提交
29
// clang-format off
S
slguan 已提交
30
// keywords in sql string
H
hzcheng 已提交
31
static SKeyword keywordTable[] = {
X
Xiaoyu Wang 已提交
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
    {"ACCOUNT",       TK_ACCOUNT},
    {"ACCOUNTS",      TK_ACCOUNTS},
    {"ADD",           TK_ADD},
    {"AGGREGATE",     TK_AGGREGATE},
    {"ALL",           TK_ALL},
    {"ALTER",         TK_ALTER},
    {"ANALYZE",       TK_ANALYZE},
    {"AND",           TK_AND},
    {"APPS",          TK_APPS},
    {"AS",            TK_AS},
    {"ASC",           TK_ASC},
    {"AT_ONCE",       TK_AT_ONCE},
    {"BETWEEN",       TK_BETWEEN},
    {"BINARY",        TK_BINARY},
    {"BIGINT",        TK_BIGINT},
    {"BNODE",         TK_BNODE},
    {"BNODES",        TK_BNODES},
    {"BOOL",          TK_BOOL},
    {"BUFFER",        TK_BUFFER},
    {"BUFSIZE",       TK_BUFSIZE},
    {"BY",            TK_BY},
    {"CACHE",         TK_CACHE},
    {"CACHELAST",     TK_CACHELAST},
    {"CAST",          TK_CAST},
    {"CLUSTER",       TK_CLUSTER},
    {"COLUMN",        TK_COLUMN},
    {"COMMENT",       TK_COMMENT},
    {"COMP",          TK_COMP},
    {"COMPACT",       TK_COMPACT},
    {"CONNS",         TK_CONNS},
    {"CONNECTION",    TK_CONNECTION},
    {"CONNECTIONS",   TK_CONNECTIONS},
    {"COUNT",         TK_COUNT},
    {"CREATE",        TK_CREATE},
    {"DATABASE",      TK_DATABASE},
    {"DATABASES",     TK_DATABASES},
    {"DAYS",          TK_DAYS},
    {"DBS",           TK_DBS},
    {"DELAY",         TK_DELAY},
    {"DESC",          TK_DESC},
    {"DESCRIBE",      TK_DESCRIBE},
    {"DISTINCT",      TK_DISTINCT},
    {"DNODE",         TK_DNODE},
    {"DNODES",        TK_DNODES},
    {"DOUBLE",        TK_DOUBLE},
    {"DROP",          TK_DROP},
    {"EXISTS",        TK_EXISTS},
    {"EXPLAIN",       TK_EXPLAIN},
    {"FILE_FACTOR",   TK_FILE_FACTOR},
    {"FILL",          TK_FILL},
    {"FIRST",         TK_FIRST},
    {"FLOAT",         TK_FLOAT},
    {"FROM",          TK_FROM},
    {"FSYNC",         TK_FSYNC},
    {"FUNCTION",      TK_FUNCTION},
    {"FUNCTIONS",     TK_FUNCTIONS},
88
    {"GRANT",         TK_GRANT},
X
Xiaoyu Wang 已提交
89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150
    {"GRANTS",        TK_GRANTS},
    {"GROUP",         TK_GROUP},
    {"HAVING",        TK_HAVING},
    {"IF",            TK_IF},
    {"IMPORT",        TK_IMPORT},
    {"IN",            TK_IN},
    {"INDEX",         TK_INDEX},
    {"INDEXES",       TK_INDEXES},
    {"INNER",         TK_INNER},
    {"INT",           TK_INT},
    {"INSERT",        TK_INSERT},
    {"INTEGER",       TK_INTEGER},
    {"INTERVAL",      TK_INTERVAL},
    {"INTO",          TK_INTO},
    {"IS",            TK_IS},
    {"JOIN",          TK_JOIN},
    {"JSON",          TK_JSON},
    {"KEEP",          TK_KEEP},
    {"KILL",          TK_KILL},
    {"LAST",          TK_LAST},
    {"LAST_ROW",      TK_LAST_ROW},
    {"LICENCE",       TK_LICENCE},
    {"LIKE",          TK_LIKE},
    {"LIMIT",         TK_LIMIT},
    {"LINEAR",        TK_LINEAR},
    {"LOCAL",         TK_LOCAL},
    {"MATCH",         TK_MATCH},
    {"MAXROWS",       TK_MAXROWS},
    {"MINROWS",       TK_MINROWS},
    {"MINUS",         TK_MINUS},
    {"MNODE",         TK_MNODE},
    {"MNODES",        TK_MNODES},
    {"MODIFY",        TK_MODIFY},
    {"MODULES",       TK_MODULES},
    {"NCHAR",         TK_NCHAR},
    {"NEXT",          TK_NEXT},
    {"NMATCH",        TK_NMATCH},
    {"NONE",          TK_NONE},
    {"NOT",           TK_NOT},
    {"NOW",           TK_NOW},
    {"NULL",          TK_NULL},
    {"NULLS",         TK_NULLS},
    {"OFFSET",        TK_OFFSET},
    {"ON",            TK_ON},
    {"OR",            TK_OR},
    {"ORDER",         TK_ORDER},
    {"OUTPUTTYPE",    TK_OUTPUTTYPE},
    {"PARTITION",     TK_PARTITION},
    {"PASS",          TK_PASS},
    {"PAGES",         TK_PAGES},
    {"PAGESIZE",      TK_PAGESIZE},
    {"PORT",          TK_PORT},
    {"PPS",           TK_PPS},
    {"PRECISION",     TK_PRECISION},
    {"PRIVILEGE",     TK_PRIVILEGE},
    {"PREV",          TK_PREV},
    {"QNODE",         TK_QNODE},
    {"QNODES",        TK_QNODES},
    {"QTIME",         TK_QTIME},
    {"QUERIES",       TK_QUERIES},
    {"QUERY",         TK_QUERY},
    {"RATIO",         TK_RATIO},
151 152
    {"READ",          TK_READ},
    {"RENAME",        TK_RENAME},
X
Xiaoyu Wang 已提交
153 154 155
    {"REPLICA",       TK_REPLICA},
    {"RESET",         TK_RESET},
    {"RETENTIONS",    TK_RETENTIONS},
156
    {"REVOKE",        TK_REVOKE},
X
Xiaoyu Wang 已提交
157 158 159 160 161 162 163
    {"ROLLUP",        TK_ROLLUP},
    {"SCHEMA",        TK_SCHEMA},
    {"SCORES",        TK_SCORES},
    {"SELECT",        TK_SELECT},
    {"SESSION",       TK_SESSION},
    {"SET",           TK_SET},
    {"SHOW",          TK_SHOW},
164
    {"SINGLE_STABLE", TK_SINGLE_STABLE},
X
Xiaoyu Wang 已提交
165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188
    {"SLIDING",       TK_SLIDING},
    {"SLIMIT",        TK_SLIMIT},
    {"SMA",           TK_SMA},
    {"SMALLINT",      TK_SMALLINT},
    {"SNODE",         TK_SNODE},
    {"SNODES",        TK_SNODES},
    {"SOFFSET",       TK_SOFFSET},
    {"STABLE",        TK_STABLE},
    {"STABLES",       TK_STABLES},
    {"STATE",         TK_STATE},
    {"STATE_WINDOW",  TK_STATE_WINDOW},
    {"STORAGE",       TK_STORAGE},
    {"STREAM",        TK_STREAM},
    {"STREAMS",       TK_STREAMS},
    {"STRICT",        TK_STRICT},
    {"SYNCDB",        TK_SYNCDB},
    {"TABLE",         TK_TABLE},
    {"TABLES",        TK_TABLES},
    {"TAG",           TK_TAG},
    {"TAGS",          TK_TAGS},
    {"TBNAME",        TK_TBNAME},
    {"TIMESTAMP",     TK_TIMESTAMP},
    {"TIMEZONE",      TK_TIMEZONE},
    {"TINYINT",       TK_TINYINT},
189
    {"TO",            TK_TO},
X
Xiaoyu Wang 已提交
190 191 192
    {"TODAY",         TK_TODAY},
    {"TOPIC",         TK_TOPIC},
    {"TOPICS",        TK_TOPICS},
193 194
    {"TRANSACTION",   TK_TRANSACTION},
    {"TRANSACTIONS",  TK_TRANSACTIONS},
X
Xiaoyu Wang 已提交
195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215
    {"TRIGGER",       TK_TRIGGER},
    {"TSERIES",       TK_TSERIES},
    {"TTL",           TK_TTL},
    {"UNION",         TK_UNION},
    {"UNSIGNED",      TK_UNSIGNED},
    {"USE",           TK_USE},
    {"USER",          TK_USER},
    {"USERS",         TK_USERS},
    {"USING",         TK_USING},
    {"VALUE",         TK_VALUE},
    {"VALUES",        TK_VALUES},
    {"VARCHAR",       TK_VARCHAR},
    {"VARIABLES",     TK_VARIABLES},
    {"VERBOSE",       TK_VERBOSE},
    {"VGROUPS",       TK_VGROUPS},
    {"VNODES",        TK_VNODES},
    {"WAL",           TK_WAL},
    {"WATERMARK",     TK_WATERMARK},
    {"WHERE",         TK_WHERE},
    {"WINDOW_CLOSE",  TK_WINDOW_CLOSE},
    {"WITH",          TK_WITH},
216
    {"WRITE",         TK_WRITE},
X
Xiaoyu Wang 已提交
217
    {"_C0",           TK_ROWTS},
X
Xiaoyu Wang 已提交
218 219 220 221 222 223
    {"_QENDTS",       TK_QENDTS},
    {"_QSTARTTS",     TK_QSTARTTS},
    {"_ROWTS",        TK_ROWTS},
    {"_WDURATION",    TK_WDURATION},
    {"_WENDTS",       TK_WENDTS},
    {"_WSTARTTS",     TK_WSTARTTS},
224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286
    // {"ID",           TK_ID},
    // {"STRING",       TK_STRING},
    // {"EQ",           TK_EQ},
    // {"NE",           TK_NE},
    // {"ISNULL",       TK_ISNULL},
    // {"NOTNULL",      TK_NOTNULL},
    // {"GLOB",         TK_GLOB},
    // {"GT",           TK_GT},
    // {"GE",           TK_GE},
    // {"LT",           TK_LT},
    // {"LE",           TK_LE},
    // {"BITAND",       TK_BITAND},
    // {"BITOR",        TK_BITOR},
    // {"LSHIFT",       TK_LSHIFT},
    // {"RSHIFT",       TK_RSHIFT},
    // {"PLUS",         TK_PLUS},
    // {"DIVIDE",       TK_DIVIDE},
    // {"TIMES",        TK_TIMES},
    // {"STAR",         TK_STAR},
    // {"SLASH",        TK_SLASH},
    // {"REM ",         TK_REM},
    // {"||",           TK_CONCAT},
    // {"UMINUS",       TK_UMINUS},
    // {"UPLUS",        TK_UPLUS},
    // {"BITNOT",       TK_BITNOT},
    // {"DOT",          TK_DOT},
    // {"CTIME",        TK_CTIME},
    // {"LP",           TK_LP},
    // {"RP",           TK_RP},
    // {"COMMA",        TK_COMMA},
    // {"EVERY",        TK_EVERY},
    // {"VARIABLE",     TK_VARIABLE},
    // {"UPDATE",       TK_UPDATE},
    // {"CHANGE",       TK_CHANGE},
    // {"COLON",        TK_COLON},
    // {"ABORT",        TK_ABORT},
    // {"AFTER",        TK_AFTER},
    // {"ATTACH",       TK_ATTACH},
    // {"BEFORE",       TK_BEFORE},
    // {"BEGIN",        TK_BEGIN},
    // {"CASCADE",      TK_CASCADE},
    // {"CONFLICT",     TK_CONFLICT},
    // {"COPY",         TK_COPY},
    // {"DEFERRED",     TK_DEFERRED},
    // {"DELIMITERS",   TK_DELIMITERS},
    // {"DETACH",       TK_DETACH},
    // {"EACH",         TK_EACH},
    // {"END",          TK_END},
    // {"FAIL",         TK_FAIL},
    // {"FOR",          TK_FOR},
    // {"IGNORE",       TK_IGNORE},
    // {"IMMEDIATE",    TK_IMMEDIATE},
    // {"INITIALLY",    TK_INITIALLY},
    // {"INSTEAD",      TK_INSTEAD},
    // {"KEY",          TK_KEY},
    // {"OF",           TK_OF},
    // {"RAISE",        TK_RAISE},
    // {"REPLACE",      TK_REPLACE},
    // {"RESTRICT",     TK_RESTRICT},
    // {"ROW",          TK_ROW},
    // {"STATEMENT",    TK_STATEMENT},
    // {"VIEW",         TK_VIEW},
    // {"SEMI",         TK_SEMI},
287
    // {"PARTITIONS",   TK_PARTITIONS},
288
    // {"MODE",         TK_MODE},
H
hzcheng 已提交
289
};
X
Xiaoyu Wang 已提交
290
// clang-format on
H
hzcheng 已提交
291 292 293 294 295 296 297 298 299 300 301 302 303

static const char isIdChar[] = {
    /* x0 x1 x2 x3 x4 x5 x6 x7 x8 x9 xA xB xC xD xE xF */
    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 0x */
    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 1x */
    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 2x */
    1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, /* 3x */
    0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, /* 4x */
    1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 1, /* 5x */
    0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, /* 6x */
    1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, /* 7x */
};

H
Haojun Liao 已提交
304
static void* keywordHashTable = NULL;
H
hzcheng 已提交
305

S
TD-1057  
Shengliang Guan 已提交
306
static void doInitKeywordsTable(void) {
307
  int numOfEntries = tListLen(keywordTable);
X
Xiaoyu Wang 已提交
308

H
Haojun Liao 已提交
309
  keywordHashTable = taosHashInit(numOfEntries, MurmurHash3_32, true, false);
310
  for (int32_t i = 0; i < numOfEntries; i++) {
S
TD-1057  
Shengliang Guan 已提交
311
    keywordTable[i].len = (uint8_t)strlen(keywordTable[i].name);
312
    void* ptr = &keywordTable[i];
H
Haojun Liao 已提交
313
    taosHashPut(keywordHashTable, keywordTable[i].name, keywordTable[i].len, (void*)&ptr, POINTER_BYTES);
H
hzcheng 已提交
314
  }
315 316
}

wafwerar's avatar
wafwerar 已提交
317
static TdThreadOnce keywordsHashTableInit = PTHREAD_ONCE_INIT;
318

319
static int32_t tKeywordCode(const char* z, int n) {
wafwerar's avatar
wafwerar 已提交
320
  taosThreadOnce(&keywordsHashTableInit, doInitKeywordsTable);
X
Xiaoyu Wang 已提交
321

H
hjxilinx 已提交
322
  char key[512] = {0};
X
Xiaoyu Wang 已提交
323
  if (n > tListLen(key)) {  // too long token, can not be any other token type
324
    return TK_NK_ID;
325
  }
X
Xiaoyu Wang 已提交
326

H
hzcheng 已提交
327 328
  for (int32_t j = 0; j < n; ++j) {
    if (z[j] >= 'a' && z[j] <= 'z') {
329
      key[j] = (char)(z[j] & 0xDF);  // to uppercase and set the null-terminated
H
hzcheng 已提交
330 331 332 333 334
    } else {
      key[j] = z[j];
    }
  }

D
dapan1121 已提交
335
  if (keywordHashTable == NULL) {
336
    return TK_NK_ILLEGAL;
D
dapan1121 已提交
337
  }
H
Haojun Liao 已提交
338

H
Haojun Liao 已提交
339
  SKeyword** pKey = (SKeyword**)taosHashGet(keywordHashTable, key, n);
X
Xiaoyu Wang 已提交
340
  return (pKey != NULL) ? (*pKey)->type : TK_NK_ID;
H
hzcheng 已提交
341 342
}

H
huili 已提交
343
/*
344 345 346
 * Return the length of the token that begins at z[0].
 * Store the token type in *type before returning.
 */
347
uint32_t tGetToken(const char* z, uint32_t* tokenId) {
348
  uint32_t i;
H
hzcheng 已提交
349 350 351 352 353 354 355 356
  switch (*z) {
    case ' ':
    case '\t':
    case '\n':
    case '\f':
    case '\r': {
      for (i = 1; isspace(z[i]); i++) {
      }
357
      *tokenId = TK_NK_SPACE;
H
hzcheng 已提交
358 359 360
      return i;
    }
    case ':': {
361
      *tokenId = TK_NK_COLON;
H
hzcheng 已提交
362 363 364 365 366 367
      return 1;
    }
    case '-': {
      if (z[1] == '-') {
        for (i = 2; z[i] && z[i] != '\n'; i++) {
        }
368
        *tokenId = TK_NK_COMMENT;
H
hzcheng 已提交
369
        return i;
370 371 372
      } else if (z[1] == '>') {
        *tokenId = TK_NK_ARROW;
        return 2;
H
hzcheng 已提交
373
      }
X
Xiaoyu Wang 已提交
374
      *tokenId = TK_NK_MINUS;
H
hzcheng 已提交
375 376 377
      return 1;
    }
    case '(': {
378
      *tokenId = TK_NK_LP;
H
hzcheng 已提交
379 380 381
      return 1;
    }
    case ')': {
382
      *tokenId = TK_NK_RP;
H
hzcheng 已提交
383 384 385
      return 1;
    }
    case ';': {
386
      *tokenId = TK_NK_SEMI;
H
hzcheng 已提交
387 388 389
      return 1;
    }
    case '+': {
390
      *tokenId = TK_NK_PLUS;
H
hzcheng 已提交
391 392 393
      return 1;
    }
    case '*': {
394
      *tokenId = TK_NK_STAR;
H
hzcheng 已提交
395 396 397 398
      return 1;
    }
    case '/': {
      if (z[1] != '*' || z[2] == 0) {
399
        *tokenId = TK_NK_SLASH;
H
hzcheng 已提交
400 401 402 403 404
        return 1;
      }
      for (i = 3; z[i] && (z[i] != '/' || z[i - 1] != '*'); i++) {
      }
      if (z[i]) i++;
405
      *tokenId = TK_NK_COMMENT;
H
hzcheng 已提交
406 407 408
      return i;
    }
    case '%': {
409
      *tokenId = TK_NK_REM;
H
hzcheng 已提交
410 411 412
      return 1;
    }
    case '=': {
413
      *tokenId = TK_NK_EQ;
H
hzcheng 已提交
414 415 416 417
      return 1 + (z[1] == '=');
    }
    case '<': {
      if (z[1] == '=') {
418
        *tokenId = TK_NK_LE;
H
hzcheng 已提交
419 420
        return 2;
      } else if (z[1] == '>') {
421
        *tokenId = TK_NK_NE;
H
hzcheng 已提交
422 423
        return 2;
      } else if (z[1] == '<') {
424
        *tokenId = TK_NK_LSHIFT;
H
hzcheng 已提交
425 426
        return 2;
      } else {
427
        *tokenId = TK_NK_LT;
H
hzcheng 已提交
428 429 430 431 432
        return 1;
      }
    }
    case '>': {
      if (z[1] == '=') {
433
        *tokenId = TK_NK_GE;
H
hzcheng 已提交
434 435
        return 2;
      } else if (z[1] == '>') {
436
        *tokenId = TK_NK_RSHIFT;
H
hzcheng 已提交
437 438
        return 2;
      } else {
439
        *tokenId = TK_NK_GT;
H
hzcheng 已提交
440 441 442 443 444
        return 1;
      }
    }
    case '!': {
      if (z[1] != '=') {
445
        *tokenId = TK_NK_ILLEGAL;
H
hzcheng 已提交
446 447
        return 2;
      } else {
448
        *tokenId = TK_NK_NE;
H
hzcheng 已提交
449 450 451 452 453
        return 2;
      }
    }
    case '|': {
      if (z[1] != '|') {
454
        *tokenId = TK_NK_BITOR;
H
hzcheng 已提交
455 456
        return 1;
      } else {
457
        *tokenId = TK_NK_CONCAT;
H
hzcheng 已提交
458 459 460 461
        return 2;
      }
    }
    case ',': {
462
      *tokenId = TK_NK_COMMA;
H
hzcheng 已提交
463 464 465
      return 1;
    }
    case '&': {
466
      *tokenId = TK_NK_BITAND;
H
hzcheng 已提交
467 468 469
      return 1;
    }
    case '~': {
470
      *tokenId = TK_NK_BITNOT;
H
hzcheng 已提交
471 472
      return 1;
    }
S
slguan 已提交
473
    case '?': {
474
      *tokenId = TK_NK_QUESTION;
S
slguan 已提交
475 476
      return 1;
    }
477
    case '`':
H
hzcheng 已提交
478 479
    case '\'':
    case '"': {
S
slguan 已提交
480 481
      int  delim = z[0];
      bool strEnd = false;
H
hzcheng 已提交
482
      for (i = 1; z[i]; i++) {
483
        if (delim != '`' && z[i] == '\\') {  // ignore the escaped character that follows this backslash
L
[1292]  
lihui 已提交
484 485 486
          i++;
          continue;
        }
X
Xiaoyu Wang 已提交
487

488
        if (z[i] == delim) {
H
hzcheng 已提交
489 490 491
          if (z[i + 1] == delim) {
            i++;
          } else {
H
huili 已提交
492
            strEnd = true;
H
hzcheng 已提交
493 494 495 496
            break;
          }
        }
      }
X
Xiaoyu Wang 已提交
497

H
hzcheng 已提交
498
      if (z[i]) i++;
H
huili 已提交
499

S
slguan 已提交
500
      if (strEnd) {
X
Xiaoyu Wang 已提交
501
        *tokenId = (delim == '`') ? TK_NK_ID : TK_NK_STRING;
S
slguan 已提交
502 503
        return i;
      }
H
huili 已提交
504

S
slguan 已提交
505
      break;
H
hzcheng 已提交
506 507
    }
    case '.': {
S
slguan 已提交
508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524
      /*
       * handle the the float number with out integer part
       * .123
       * .123e4
       */
      if (isdigit(z[1])) {
        for (i = 2; isdigit(z[i]); i++) {
        }

        if ((z[i] == 'e' || z[i] == 'E') &&
            (isdigit(z[i + 1]) || ((z[i + 1] == '+' || z[i + 1] == '-') && isdigit(z[i + 2])))) {
          i += 2;
          while (isdigit(z[i])) {
            i++;
          }
        }

525
        *tokenId = TK_NK_FLOAT;
S
slguan 已提交
526 527
        return i;
      } else {
528
        *tokenId = TK_NK_DOT;
S
slguan 已提交
529 530 531 532 533 534 535
        return 1;
      }
    }

    case '0': {
      char next = z[1];

X
Xiaoyu Wang 已提交
536
      if (next == 'b') {  // bin number
537
        *tokenId = TK_NK_BIN;
S
slguan 已提交
538 539 540 541 542 543 544 545
        for (i = 2; (z[i] == '0' || z[i] == '1'); ++i) {
        }

        if (i == 2) {
          break;
        }

        return i;
X
Xiaoyu Wang 已提交
546
      } else if (next == 'x') {  // hex number
547
        *tokenId = TK_NK_HEX;
S
slguan 已提交
548 549 550 551 552 553 554 555 556
        for (i = 2; isdigit(z[i]) || (z[i] >= 'a' && z[i] <= 'f') || (z[i] >= 'A' && z[i] <= 'F'); ++i) {
        }

        if (i == 2) {
          break;
        }

        return i;
      }
H
hzcheng 已提交
557 558 559 560 561 562 563 564 565 566
    }
    case '1':
    case '2':
    case '3':
    case '4':
    case '5':
    case '6':
    case '7':
    case '8':
    case '9': {
567
      *tokenId = TK_NK_INTEGER;
H
hzcheng 已提交
568 569 570
      for (i = 1; isdigit(z[i]); i++) {
      }

H
Haojun Liao 已提交
571
      /* here is the 1u/1a/2s/3m/9y */
X
Xiaoyu Wang 已提交
572 573 574
      if ((z[i] == 'b' || z[i] == 'u' || z[i] == 'a' || z[i] == 's' || z[i] == 'm' || z[i] == 'h' || z[i] == 'd' ||
           z[i] == 'n' || z[i] == 'y' || z[i] == 'w' || z[i] == 'B' || z[i] == 'U' || z[i] == 'A' || z[i] == 'S' ||
           z[i] == 'M' || z[i] == 'H' || z[i] == 'D' || z[i] == 'N' || z[i] == 'Y' || z[i] == 'W') &&
H
hjxilinx 已提交
575
          (isIdChar[(uint8_t)z[i + 1]] == 0)) {
576
        *tokenId = TK_NK_VARIABLE;
H
hzcheng 已提交
577 578 579 580 581 582 583 584 585 586
        i += 1;
        return i;
      }

      int32_t seg = 1;
      while (z[i] == '.' && isdigit(z[i + 1])) {
        i += 2;
        while (isdigit(z[i])) {
          i++;
        }
587
        *tokenId = TK_NK_FLOAT;
H
hzcheng 已提交
588 589 590 591
        seg++;
      }

      if (seg == 4) {  // ip address
592
        *tokenId = TK_NK_IPTOKEN;
H
hzcheng 已提交
593
        return i;
X
Xiaoyu Wang 已提交
594 595
      } else if (seg > 2) {
        break;
H
hzcheng 已提交
596 597 598 599 600 601 602 603
      }

      if ((z[i] == 'e' || z[i] == 'E') &&
          (isdigit(z[i + 1]) || ((z[i + 1] == '+' || z[i + 1] == '-') && isdigit(z[i + 2])))) {
        i += 2;
        while (isdigit(z[i])) {
          i++;
        }
604
        *tokenId = TK_NK_FLOAT;
H
hzcheng 已提交
605 606 607 608 609 610
      }
      return i;
    }
    case '[': {
      for (i = 1; z[i] && z[i - 1] != ']'; i++) {
      }
611
      *tokenId = TK_NK_ID;
H
hzcheng 已提交
612 613 614 615 616 617
      return i;
    }
    case 'T':
    case 't':
    case 'F':
    case 'f': {
X
Xiaoyu Wang 已提交
618
      for (i = 1; ((z[i] & 0x80) == 0) && isIdChar[(uint8_t)z[i]]; i++) {
H
hzcheng 已提交
619 620 621
      }

      if ((i == 4 && strncasecmp(z, "true", 4) == 0) || (i == 5 && strncasecmp(z, "false", 5) == 0)) {
622
        *tokenId = TK_NK_BOOL;
H
hzcheng 已提交
623 624 625 626
        return i;
      }
    }
    default: {
X
Xiaoyu Wang 已提交
627
      if (((*z & 0x80) != 0) || !isIdChar[(uint8_t)*z]) {
H
hzcheng 已提交
628 629
        break;
      }
X
Xiaoyu Wang 已提交
630
      for (i = 1; ((z[i] & 0x80) == 0) && isIdChar[(uint8_t)z[i]]; i++) {
H
hzcheng 已提交
631
      }
632
      *tokenId = tKeywordCode(z, i);
H
hzcheng 已提交
633 634 635 636
      return i;
    }
  }

637
  *tokenId = TK_NK_ILLEGAL;
H
hzcheng 已提交
638 639 640
  return 0;
}

X
Xiaoyu Wang 已提交
641 642 643
SToken tscReplaceStrToken(char** str, SToken* token, const char* newToken) {
  char*   src = *str;
  size_t  nsize = strlen(newToken);
D
dapan1121 已提交
644 645
  int32_t size = (int32_t)strlen(*str) - token->n + (int32_t)nsize + 1;
  int32_t bsize = (int32_t)((uint64_t)token->z - (uint64_t)src);
X
Xiaoyu Wang 已提交
646
  SToken  ntoken;
D
dapan1121 已提交
647

wafwerar's avatar
wafwerar 已提交
648
  *str = taosMemoryCalloc(1, size);
D
dapan1121 已提交
649 650

  strncpy(*str, src, bsize);
H
Haojun Liao 已提交
651
  strcat(*str, newToken);
D
dapan1121 已提交
652 653
  strcat(*str, token->z + token->n);

D
dapan1121 已提交
654
  ntoken.n = (uint32_t)nsize;
D
dapan1121 已提交
655 656
  ntoken.z = *str + bsize;

wafwerar's avatar
wafwerar 已提交
657
  taosMemoryFreeClear(src);
D
dapan1121 已提交
658 659 660 661

  return ntoken;
}

662
SToken tStrGetToken(const char* str, int32_t* i, bool isPrevOptr) {
H
Haojun Liao 已提交
663
  SToken t0 = {0};
S
slguan 已提交
664

H
hzcheng 已提交
665 666
  // here we reach the end of sql string, null-terminated string
  if (str[*i] == 0) {
S
slguan 已提交
667 668
    t0.n = 0;
    return t0;
H
hzcheng 已提交
669 670
  }

671
  // IGNORE TK_NK_SPACE, TK_NK_COMMA, and specified tokens
S
slguan 已提交
672 673 674
  while (1) {
    *i += t0.n;

H
Haojun Liao 已提交
675
    int32_t numOfComma = 0;
X
Xiaoyu Wang 已提交
676
    char    t = str[*i];
H
Haojun Liao 已提交
677 678 679 680
    while (t == ' ' || t == '\n' || t == '\r' || t == '\t' || t == '\f' || t == ',') {
      if (t == ',' && (++numOfComma > 1)) {  // comma only allowed once
        t0.n = 0;
        return t0;
S
slguan 已提交
681
      }
X
Xiaoyu Wang 已提交
682

H
Haojun Liao 已提交
683
      t = str[++(*i)];
S
slguan 已提交
684
    }
H
hzcheng 已提交
685

686
    t0.n = tGetToken(&str[*i], &t0.type);
H
Haojun Liao 已提交
687
    break;
S
slguan 已提交
688

H
Haojun Liao 已提交
689 690
    // not support user specfied ignored symbol list
#if 0
H
Haojun Liao 已提交
691
    bool ignore = false;
S
slguan 已提交
692 693
    for (uint32_t k = 0; k < numOfIgnoreToken; k++) {
      if (t0.type == ignoreTokenTypes[k]) {
H
Haojun Liao 已提交
694
        ignore = true;
S
slguan 已提交
695 696 697 698
        break;
      }
    }

H
Haojun Liao 已提交
699
    if (!ignore) {
S
slguan 已提交
700 701
      break;
    }
H
Haojun Liao 已提交
702
#endif
H
hzcheng 已提交
703 704
  }

705
  if (t0.type == TK_NK_SEMI) {
S
slguan 已提交
706 707 708 709 710 711 712 713 714
    t0.n = 0;
    return t0;
  }

  uint32_t type = 0;
  int32_t  len;

  // support parse the 'db.tbl' format, notes: There should be no space on either side of the dot!
  if ('.' == str[*i + t0.n]) {
715
    len = tGetToken(&str[*i + t0.n + 1], &type);
S
slguan 已提交
716 717

    // only id and string are valid
718
    if ((TK_NK_STRING != t0.type) && (TK_NK_ID != t0.type)) {
719
      t0.type = TK_NK_ILLEGAL;
S
slguan 已提交
720 721 722 723 724 725 726 727 728
      t0.n = 0;

      return t0;
    }

    t0.n += len + 1;

  } else {
    // support parse the -/+number format
X
Xiaoyu Wang 已提交
729
    if ((isPrevOptr) && (t0.type == TK_NK_MINUS || t0.type == TK_NK_PLUS)) {
730
      len = tGetToken(&str[*i + t0.n], &type);
731
      if (type == TK_NK_INTEGER || type == TK_NK_FLOAT) {
S
slguan 已提交
732 733 734
        t0.type = type;
        t0.n += len;
      }
H
hzcheng 已提交
735 736 737
    }
  }

X
Xiaoyu Wang 已提交
738
  t0.z = (char*)str + (*i);
S
slguan 已提交
739 740 741
  *i += t0.n;

  return t0;
H
hzcheng 已提交
742 743
}

X
Xiaoyu Wang 已提交
744
bool taosIsKeyWordToken(const char* z, int32_t len) { return (tKeywordCode((char*)z, len) != TK_NK_ID); }
H
Haojun Liao 已提交
745 746

void taosCleanupKeywordsTable() {
H
Haojun Liao 已提交
747 748 749 750
  void* m = keywordHashTable;
  if (m != NULL && atomic_val_compare_exchange_ptr(&keywordHashTable, m, 0) == m) {
    taosHashCleanup(m);
  }
Y
yihaoDeng 已提交
751
}
752

H
Haojun Liao 已提交
753
SToken taosTokenDup(SToken* pToken, char* buf, int32_t len) {
754
  assert(pToken != NULL && buf != NULL && len > pToken->n);
X
Xiaoyu Wang 已提交
755

756 757 758
  strncpy(buf, pToken->z, pToken->n);
  buf[pToken->n] = 0;

H
Haojun Liao 已提交
759
  SToken token = *pToken;
760 761 762
  token.z = buf;
  return token;
}