parTokenizer.c 19.2 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
Haojun Liao 已提交
18
#include "thash.h"
H
hjxilinx 已提交
19
#include "taosdef.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;

S
slguan 已提交
29
// keywords in sql string
H
hzcheng 已提交
30
static SKeyword keywordTable[] = {
31
    {"ACCOUNT",       TK_ACCOUNT},
32 33 34
    {"ACCOUNTS",      TK_ACCOUNTS},
    {"ADD",           TK_ADD},
    {"AGGREGATE",     TK_AGGREGATE},
35
    {"ALL",           TK_ALL},
36
    {"ALTER",         TK_ALTER},
37
    {"ANALYZE",       TK_ANALYZE},
38
    {"AND",           TK_AND},
39
    {"APPS",          TK_APPS},
40 41 42 43 44 45 46
    {"AS",            TK_AS},
    {"ASC",           TK_ASC},
    {"BETWEEN",       TK_BETWEEN},
    {"BINARY",        TK_BINARY},
    {"BIGINT",        TK_BIGINT},
    {"BLOCKS",        TK_BLOCKS},
    {"BOOL",          TK_BOOL},
47
    {"BUFSIZE",       TK_BUFSIZE},
48 49 50
    {"BY",            TK_BY},
    {"CACHE",         TK_CACHE},
    {"CACHELAST",     TK_CACHELAST},
51
    {"COLUMN",        TK_COLUMN},
52 53
    {"COMMENT",       TK_COMMENT},
    {"COMP",          TK_COMP},
54 55 56 57
    {"COMPACT",       TK_COMPACT},
    {"CONNS",         TK_CONNS},
    {"CONNECTION",    TK_CONNECTION},
    {"CONNECTIONS",   TK_CONNECTIONS},
58 59 60 61
    {"CREATE",        TK_CREATE},
    {"DATABASE",      TK_DATABASE},
    {"DATABASES",     TK_DATABASES},
    {"DAYS",          TK_DAYS},
62
    {"DBS",           TK_DBS},
X
Xiaoyu Wang 已提交
63
    {"DELAY",         TK_DELAY},
64
    {"DESC",          TK_DESC},
65
    {"DESCRIBE",      TK_DESCRIBE},
66
    {"DISTINCT",      TK_DISTINCT},
67 68
    {"DNODE",         TK_DNODE},
    {"DNODES",        TK_DNODES},
69
    {"DOUBLE",        TK_DOUBLE},
70
    {"DROP",          TK_DROP},
71
    {"EXISTS",        TK_EXISTS},
72
    {"EXPLAIN",       TK_EXPLAIN},
X
Xiaoyu Wang 已提交
73
    {"FILE_FACTOR",   TK_FILE_FACTOR},
74 75 76 77
    {"FILL",          TK_FILL},
    {"FLOAT",         TK_FLOAT},
    {"FROM",          TK_FROM},
    {"FSYNC",         TK_FSYNC},
X
Xiaoyu Wang 已提交
78
    {"FUNCTION",      TK_FUNCTION},
X
Xiaoyu Wang 已提交
79
    {"FUNCTIONS",     TK_FUNCTIONS},
80 81 82 83 84
    {"GROUP",         TK_GROUP},
    {"HAVING",        TK_HAVING},
    {"IF",            TK_IF},
    {"IMPORT",        TK_IMPORT},
    {"IN",            TK_IN},
X
Xiaoyu Wang 已提交
85
    {"INDEX",         TK_INDEX},
X
Xiaoyu Wang 已提交
86
    {"INDEXES",       TK_INDEXES},
87 88 89 90 91 92 93 94 95 96
    {"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},
97 98
    {"KILL",          TK_KILL},
    {"LICENCE",       TK_LICENCE},
99 100 101
    {"LIKE",          TK_LIKE},
    {"LIMIT",         TK_LIMIT},
    {"LINEAR",        TK_LINEAR},
102
    {"LOCAL",         TK_LOCAL},
103 104 105 106
    {"MATCH",         TK_MATCH},
    {"MAXROWS",       TK_MAXROWS},
    {"MINROWS",       TK_MINROWS},
    {"MINUS",         TK_MINUS},
X
Xiaoyu Wang 已提交
107
    {"MNODES",        TK_MNODES},
108
    {"MODIFY",        TK_MODIFY},
X
Xiaoyu Wang 已提交
109
    {"MODULES",       TK_MODULES},
110 111 112 113 114 115 116 117 118 119
    {"NCHAR",         TK_NCHAR},
    {"NMATCH",        TK_NMATCH},
    {"NONE",          TK_NONE},
    {"NOT",           TK_NOT},
    {"NOW",           TK_NOW},
    {"NULL",          TK_NULL},
    {"OFFSET",        TK_OFFSET},
    {"ON",            TK_ON},
    {"OR",            TK_OR},
    {"ORDER",         TK_ORDER},
120
    {"OUTPUTTYPE",    TK_OUTPUTTYPE},
121
    {"PARTITION",     TK_PARTITION},
122 123
    {"PASS",          TK_PASS},
    {"PORT",          TK_PORT},
124
    {"PPS",           TK_PPS},
125 126 127
    {"PRECISION",     TK_PRECISION},
    {"PRIVILEGE",     TK_PRIVILEGE},
    {"PREV",          TK_PREV},
128
    {"QENDTS",        TK_QENDTS},
X
Xiaoyu Wang 已提交
129
    {"QNODE",         TK_QNODE},
X
Xiaoyu Wang 已提交
130
    {"QNODES",        TK_QNODES},
131
    {"QSTARTTS",      TK_QSTARTTS},
132 133
    {"QTIME",         TK_QTIME},
    {"QUERIES",       TK_QUERIES},
134
    {"QUERY",         TK_QUERY},
135
    {"QUORUM",        TK_QUORUM},
136
    {"RATIO",         TK_RATIO},
137
    {"REPLICA",       TK_REPLICA},
138
    {"RESET",         TK_RESET},
X
Xiaoyu Wang 已提交
139 140
    {"RETENTIONS",    TK_RETENTIONS},
    {"ROLLUP",        TK_ROLLUP},
141
    {"ROWTS",         TK_ROWTS},
142
    {"SCORES",        TK_SCORES},
143 144
    {"SELECT",        TK_SELECT},
    {"SESSION",       TK_SESSION},
145
    {"SET",           TK_SET},
146 147 148 149 150 151 152
    {"SHOW",          TK_SHOW},
    {"SINGLE_STABLE", TK_SINGLE_STABLE},
    {"SLIDING",       TK_SLIDING},
    {"SLIMIT",        TK_SLIMIT},
    {"SMA",           TK_SMA},
    {"SMALLINT",      TK_SMALLINT},
    {"SOFFSET",       TK_SOFFSET},
153
    {"STABLE",        TK_STABLE},
154
    {"STABLES",       TK_STABLES},
155
    {"STATE",         TK_STATE},
156
    {"STATE_WINDOW",  TK_STATE_WINDOW},
157 158
    {"STORAGE",       TK_STORAGE},
    {"STREAM",       TK_STREAM},
X
Xiaoyu Wang 已提交
159
    {"STREAMS",       TK_STREAMS},
160
    {"STREAM_MODE",   TK_STREAM_MODE},
161
    {"SYNCDB",        TK_SYNCDB},
162 163
    {"TABLE",         TK_TABLE},
    {"TABLES",        TK_TABLES},
164
    {"TAG",           TK_TAG},
165
    {"TAGS",          TK_TAGS},
166
    {"TBNAME",        TK_TBNAME},
167 168
    {"TIMESTAMP",     TK_TIMESTAMP},
    {"TINYINT",       TK_TINYINT},
169
    {"TOPIC",         TK_TOPIC},
170 171
    {"TOPICS",        TK_TOPICS},
    {"TSERIES",       TK_TSERIES},
172 173 174 175
    {"TTL",           TK_TTL},
    {"UNION",         TK_UNION},
    {"UNSIGNED",      TK_UNSIGNED},
    {"USE",           TK_USE},
176 177
    {"USER",          TK_USER},
    {"USERS",         TK_USERS},
178 179 180
    {"USING",         TK_USING},
    {"VALUES",        TK_VALUES},
    {"VARCHAR",       TK_VARCHAR},
181
    {"VARIABLES",     TK_VARIABLES},
182
    {"VERBOSE",       TK_VERBOSE},
183
    {"VGROUPS",       TK_VGROUPS},
184
    {"VNODES",        TK_VNODES},
185
    {"WAL",           TK_WAL},
186 187
    {"WDURATION",     TK_WDURATION},
    {"WENDTS",        TK_WENDTS},
188
    {"WHERE",         TK_WHERE},
189
    {"WSTARTTS",      TK_WSTARTTS},
190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 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
    // {"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},
    // {"GRANTS",       TK_GRANTS},
    // {"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},
    // {"CLUSTER",      TK_CLUSTER},
    // {"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},
    // {"TRIGGER",      TK_TRIGGER},
    // {"VIEW",         TK_VIEW},
    // {"SEMI",         TK_SEMI},
256
    // {"PARTITIONS",   TK_PARTITIONS},
257
    // {"MODE",         TK_MODE},
H
hzcheng 已提交
258 259 260 261 262 263 264 265 266 267 268 269 270 271
};

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 已提交
272
static void* keywordHashTable = NULL;
H
hzcheng 已提交
273

S
TD-1057  
Shengliang Guan 已提交
274
static void doInitKeywordsTable(void) {
275 276
  int numOfEntries = tListLen(keywordTable);
  
H
Haojun Liao 已提交
277
  keywordHashTable = taosHashInit(numOfEntries, MurmurHash3_32, true, false);
278
  for (int32_t i = 0; i < numOfEntries; i++) {
S
TD-1057  
Shengliang Guan 已提交
279
    keywordTable[i].len = (uint8_t)strlen(keywordTable[i].name);
280
    void* ptr = &keywordTable[i];
H
Haojun Liao 已提交
281
    taosHashPut(keywordHashTable, keywordTable[i].name, keywordTable[i].len, (void*)&ptr, POINTER_BYTES);
H
hzcheng 已提交
282
  }
283 284
}

wafwerar's avatar
wafwerar 已提交
285
static TdThreadOnce keywordsHashTableInit = PTHREAD_ONCE_INIT;
286

287
static int32_t tKeywordCode(const char* z, int n) {
wafwerar's avatar
wafwerar 已提交
288
  taosThreadOnce(&keywordsHashTableInit, doInitKeywordsTable);
H
hjxilinx 已提交
289 290
  
  char key[512] = {0};
291
  if (n > tListLen(key)) { // too long token, can not be any other token type
292
    return TK_NK_ID;
293
  }
H
hjxilinx 已提交
294
  
H
hzcheng 已提交
295 296
  for (int32_t j = 0; j < n; ++j) {
    if (z[j] >= 'a' && z[j] <= 'z') {
297
      key[j] = (char)(z[j] & 0xDF);  // to uppercase and set the null-terminated
H
hzcheng 已提交
298 299 300 301 302
    } else {
      key[j] = z[j];
    }
  }

D
dapan1121 已提交
303
  if (keywordHashTable == NULL) {
304
    return TK_NK_ILLEGAL;
D
dapan1121 已提交
305
  }
H
Haojun Liao 已提交
306

H
Haojun Liao 已提交
307
  SKeyword** pKey = (SKeyword**)taosHashGet(keywordHashTable, key, n);
308
  return (pKey != NULL)? (*pKey)->type:TK_NK_ID;
H
hzcheng 已提交
309 310
}

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

S
slguan 已提交
469
      if (strEnd) {
470
        *tokenId = (delim == '`')? TK_NK_ID:TK_NK_STRING;
S
slguan 已提交
471 472
        return i;
      }
H
huili 已提交
473

S
slguan 已提交
474
      break;
H
hzcheng 已提交
475 476
    }
    case '.': {
S
slguan 已提交
477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493
      /*
       * 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++;
          }
        }

494
        *tokenId = TK_NK_FLOAT;
S
slguan 已提交
495 496
        return i;
      } else {
497
        *tokenId = TK_NK_DOT;
S
slguan 已提交
498 499 500 501 502 503 504 505
        return 1;
      }
    }

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

      if (next == 'b') { // bin number
506
        *tokenId = TK_NK_BIN;
S
slguan 已提交
507 508 509 510 511 512 513 514 515
        for (i = 2; (z[i] == '0' || z[i] == '1'); ++i) {
        }

        if (i == 2) {
          break;
        }

        return i;
      } else if (next == 'x') {  //hex number
516
        *tokenId = TK_NK_HEX;
S
slguan 已提交
517 518 519 520 521 522 523 524 525
        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 已提交
526 527 528 529 530 531 532 533 534 535
    }
    case '1':
    case '2':
    case '3':
    case '4':
    case '5':
    case '6':
    case '7':
    case '8':
    case '9': {
536
      *tokenId = TK_NK_INTEGER;
H
hzcheng 已提交
537 538 539
      for (i = 1; isdigit(z[i]); i++) {
      }

H
Haojun Liao 已提交
540
      /* here is the 1u/1a/2s/3m/9y */
541
      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' ||
H
Haojun Liao 已提交
542
           z[i] == 'y' || z[i] == 'w' ||
543
           z[i] == 'B' || z[i] == 'U' || z[i] == 'A' || z[i] == 'S' || z[i] == 'M' || z[i] == 'H' || z[i] == 'D' || z[i] == 'N' ||
H
Haojun Liao 已提交
544
           z[i] == 'Y' || z[i] == 'W') &&
H
hjxilinx 已提交
545
          (isIdChar[(uint8_t)z[i + 1]] == 0)) {
546
        *tokenId = TK_NK_VARIABLE;
H
hzcheng 已提交
547 548 549 550 551 552 553 554 555 556
        i += 1;
        return i;
      }

      int32_t seg = 1;
      while (z[i] == '.' && isdigit(z[i + 1])) {
        i += 2;
        while (isdigit(z[i])) {
          i++;
        }
557
        *tokenId = TK_NK_FLOAT;
H
hzcheng 已提交
558 559 560 561
        seg++;
      }

      if (seg == 4) {  // ip address
562
        *tokenId = TK_NK_IPTOKEN;
H
hzcheng 已提交
563 564 565 566 567 568 569 570 571
        return 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++;
        }
572
        *tokenId = TK_NK_FLOAT;
H
hzcheng 已提交
573 574 575 576 577 578
      }
      return i;
    }
    case '[': {
      for (i = 1; z[i] && z[i - 1] != ']'; i++) {
      }
579
      *tokenId = TK_NK_ID;
H
hzcheng 已提交
580 581 582 583 584 585
      return i;
    }
    case 'T':
    case 't':
    case 'F':
    case 'f': {
H
hjxilinx 已提交
586
      for (i = 1; ((z[i] & 0x80) == 0) && isIdChar[(uint8_t) z[i]]; i++) {
H
hzcheng 已提交
587 588 589
      }

      if ((i == 4 && strncasecmp(z, "true", 4) == 0) || (i == 5 && strncasecmp(z, "false", 5) == 0)) {
590
        *tokenId = TK_NK_BOOL;
H
hzcheng 已提交
591 592 593 594
        return i;
      }
    }
    default: {
H
hjxilinx 已提交
595
      if (((*z & 0x80) != 0) || !isIdChar[(uint8_t) *z]) {
H
hzcheng 已提交
596 597
        break;
      }
H
hjxilinx 已提交
598
      for (i = 1; ((z[i] & 0x80) == 0) && isIdChar[(uint8_t) z[i]]; i++) {
H
hzcheng 已提交
599
      }
600
      *tokenId = tKeywordCode(z, i);
H
hzcheng 已提交
601 602 603 604
      return i;
    }
  }

605
  *tokenId = TK_NK_ILLEGAL;
H
hzcheng 已提交
606 607 608
  return 0;
}

H
Haojun Liao 已提交
609
SToken tscReplaceStrToken(char **str, SToken *token, const char* newToken) {
D
dapan1121 已提交
610
  char *src = *str;
D
dapan1121 已提交
611 612 613
  size_t nsize = strlen(newToken);
  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);
H
Haojun Liao 已提交
614
  SToken ntoken;
D
dapan1121 已提交
615

wafwerar's avatar
wafwerar 已提交
616
  *str = taosMemoryCalloc(1, size);
D
dapan1121 已提交
617 618

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

D
dapan1121 已提交
622
  ntoken.n = (uint32_t)nsize;
D
dapan1121 已提交
623 624
  ntoken.z = *str + bsize;

wafwerar's avatar
wafwerar 已提交
625
  taosMemoryFreeClear(src);
D
dapan1121 已提交
626 627 628 629

  return ntoken;
}

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

H
hzcheng 已提交
633 634
  // here we reach the end of sql string, null-terminated string
  if (str[*i] == 0) {
S
slguan 已提交
635 636
    t0.n = 0;
    return t0;
H
hzcheng 已提交
637 638
  }

639
  // IGNORE TK_NK_SPACE, TK_NK_COMMA, and specified tokens
S
slguan 已提交
640 641 642
  while (1) {
    *i += t0.n;

H
Haojun Liao 已提交
643 644 645 646 647 648
    int32_t numOfComma = 0;
    char t = str[*i];
    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 已提交
649
      }
H
Haojun Liao 已提交
650
    
H
Haojun Liao 已提交
651
      t = str[++(*i)];
S
slguan 已提交
652
    }
H
hzcheng 已提交
653

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

H
Haojun Liao 已提交
657 658
    // not support user specfied ignored symbol list
#if 0
H
Haojun Liao 已提交
659
    bool ignore = false;
S
slguan 已提交
660 661
    for (uint32_t k = 0; k < numOfIgnoreToken; k++) {
      if (t0.type == ignoreTokenTypes[k]) {
H
Haojun Liao 已提交
662
        ignore = true;
S
slguan 已提交
663 664 665 666
        break;
      }
    }

H
Haojun Liao 已提交
667
    if (!ignore) {
S
slguan 已提交
668 669
      break;
    }
H
Haojun Liao 已提交
670
#endif
H
hzcheng 已提交
671 672
  }

673
  if (t0.type == TK_NK_SEMI) {
S
slguan 已提交
674 675 676 677 678 679 680 681 682
    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]) {
683
    len = tGetToken(&str[*i + t0.n + 1], &type);
S
slguan 已提交
684 685

    // only id and string are valid
686
    if ((TK_NK_STRING != t0.type) && (TK_NK_ID != t0.type)) {
687
      t0.type = TK_NK_ILLEGAL;
S
slguan 已提交
688 689 690 691 692 693 694 695 696
      t0.n = 0;

      return t0;
    }

    t0.n += len + 1;

  } else {
    // support parse the -/+number format
X
Xiaoyu Wang 已提交
697
    if ((isPrevOptr) && (t0.type == TK_NK_MINUS || t0.type == TK_NK_PLUS)) {
698
      len = tGetToken(&str[*i + t0.n], &type);
699
      if (type == TK_NK_INTEGER || type == TK_NK_FLOAT) {
S
slguan 已提交
700 701 702
        t0.type = type;
        t0.n += len;
      }
H
hzcheng 已提交
703 704 705
    }
  }

706
  t0.z = (char*) str + (*i);
S
slguan 已提交
707 708 709
  *i += t0.n;

  return t0;
H
hzcheng 已提交
710 711
}

712
bool taosIsKeyWordToken(const char* z, int32_t len) {
713
  return (tKeywordCode((char*)z, len) != TK_NK_ID);
714
}
H
Haojun Liao 已提交
715 716

void taosCleanupKeywordsTable() {
H
Haojun Liao 已提交
717 718 719 720
  void* m = keywordHashTable;
  if (m != NULL && atomic_val_compare_exchange_ptr(&keywordHashTable, m, 0) == m) {
    taosHashCleanup(m);
  }
Y
yihaoDeng 已提交
721
}
722

H
Haojun Liao 已提交
723
SToken taosTokenDup(SToken* pToken, char* buf, int32_t len) {
724 725 726 727 728
  assert(pToken != NULL && buf != NULL && len > pToken->n);
  
  strncpy(buf, pToken->z, pToken->n);
  buf[pToken->n] = 0;

H
Haojun Liao 已提交
729
  SToken token = *pToken;
730 731 732
  token.z = buf;
  return token;
}