clientMain.c 42.1 KB
Newer Older
L
Liu Jicong 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
/*
 * 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/>.
 */

#include "catalog.h"
17
#include "clientInt.h"
18
#include "clientLog.h"
L
Liu Jicong 已提交
19
#include "clientStmt.h"
L
Liu Jicong 已提交
20
#include "functionMgt.h"
L
Liu Jicong 已提交
21
#include "os.h"
H
Haojun Liao 已提交
22
#include "query.h"
L
Liu Jicong 已提交
23
#include "scheduler.h"
24
#include "tglobal.h"
L
Liu Jicong 已提交
25 26 27
#include "tmsg.h"
#include "tref.h"
#include "trpc.h"
S
version  
Shengliang Guan 已提交
28
#include "version.h"
H
Haojun Liao 已提交
29
#include "tdatablock.h"
30 31 32 33 34

#define TSC_VAR_NOT_RELEASE 1
#define TSC_VAR_RELEASED    0

static int32_t sentinel = TSC_VAR_NOT_RELEASE;
35
static int32_t createParseContext(const SRequestObj *pRequest, SParseContext **pCxt);
36 37 38 39 40 41 42 43 44 45 46

int taos_options(TSDB_OPTION option, const void *arg, ...) {
  static int32_t lock = 0;

  for (int i = 1; atomic_val_compare_exchange_32(&lock, 0, 1) != 0; ++i) {
    if (i % 1000 == 0) {
      tscInfo("haven't acquire lock after spin %d times.", i);
      sched_yield();
    }
  }

L
Liu Jicong 已提交
47
  int ret = taos_options_imp(option, (const char *)arg);
48 49 50 51 52
  atomic_store_32(&lock, 0);
  return ret;
}
// this function may be called by user or system, or by both simultaneously.
void taos_cleanup(void) {
D
dapan1121 已提交
53
  tscDebug("start to cleanup client environment");
54 55 56 57
  if (atomic_val_compare_exchange_32(&sentinel, TSC_VAR_NOT_RELEASE, TSC_VAR_RELEASED) != TSC_VAR_NOT_RELEASE) {
    return;
  }

H
Haojun Liao 已提交
58 59
  int32_t id = clientReqRefPool;
  clientReqRefPool = -1;
60 61
  taosCloseRef(id);

D
dapan1121 已提交
62 63 64 65
  hbMgrCleanUp();

  catalogDestroy();
  schedulerDestroy();
66

X
Xiaoyu Wang 已提交
67 68
  fmFuncMgtDestroy();
  qCleanupKeywordsTable();
69
  nodesDestroyAllocatorSet();
X
Xiaoyu Wang 已提交
70

71 72
  id = clientConnRefPool;
  clientConnRefPool = -1;
73 74
  taosCloseRef(id);

D
dapan1121 已提交
75 76
  rpcCleanup();
  tscDebug("rpc cleanup");
77

D
dapan1121 已提交
78
  cleanupTaskQueue();
H
Haojun Liao 已提交
79

D
dapan1121 已提交
80
  taosConvDestroy();
dengyihao's avatar
dengyihao 已提交
81

H
Haojun Liao 已提交
82
  tscInfo("all local resources released");
83
  taosCleanupCfg();
84
  taosCloseLog();
85 86
}

L
Liu Jicong 已提交
87
static setConfRet taos_set_config_imp(const char *config) {
88 89 90 91 92
  setConfRet ret = {SET_CONF_RET_SUCC, {0}};
  // TODO: need re-implementation
  return ret;
}

L
Liu Jicong 已提交
93 94
setConfRet taos_set_config(const char *config) {
  // TODO  pthread_mutex_lock(&setConfMutex);
95
  setConfRet ret = taos_set_config_imp(config);
L
Liu Jicong 已提交
96
  //  pthread_mutex_unlock(&setConfMutex);
97 98 99
  return ret;
}

100
TAOS *taos_connect(const char *ip, const char *user, const char *pass, const char *db, uint16_t port) {
S
Shengliang Guan 已提交
101
  tscDebug("try to connect to %s:%u, user:%s db:%s", ip, port, user, db);
H
Haojun Liao 已提交
102 103 104
  if (user == NULL) {
    user = TSDB_DEFAULT_USER;
  }
105

H
Haojun Liao 已提交
106 107 108
  if (pass == NULL) {
    pass = TSDB_DEFAULT_PASS;
  }
109

L
Liu Jicong 已提交
110
  STscObj *pObj = taos_connect_internal(ip, user, pass, NULL, db, port, CONN_TYPE__QUERY);
111
  if (pObj) {
D
dapan1121 已提交
112 113
    int64_t *rid = taosMemoryCalloc(1, sizeof(int64_t));
    *rid = pObj->id;
dengyihao's avatar
dengyihao 已提交
114
    return (TAOS *)rid;
115
  }
L
Liu Jicong 已提交
116

D
dapan1121 已提交
117
  return NULL;
118 119
}

120
void taos_close_internal(void *taos) {
121 122 123 124
  if (taos == NULL) {
    return;
  }

H
Haojun Liao 已提交
125
  STscObj *pTscObj = (STscObj *)taos;
D
dapan1121 已提交
126
  tscDebug("0x%" PRIx64 " try to close connection, numOfReq:%d", pTscObj->id, pTscObj->numOfReqs);
127

D
dapan1121 已提交
128
  taosRemoveRef(clientConnRefPool, pTscObj->id);
129 130
}

131
void taos_close(TAOS *taos) {
D
dapan1121 已提交
132 133 134
  if (taos == NULL) {
    return;
  }
L
Liu Jicong 已提交
135 136

  STscObj *pObj = acquireTscObj(*(int64_t *)taos);
137
  if (NULL == pObj) {
D
dapan1121 已提交
138
    taosMemoryFree(taos);
139 140
    return;
  }
L
Liu Jicong 已提交
141

142
  taos_close_internal(pObj);
L
Liu Jicong 已提交
143
  releaseTscObj(*(int64_t *)taos);
D
dapan1121 已提交
144
  taosMemoryFree(taos);
145 146
}

L
Liu Jicong 已提交
147 148
int taos_errno(TAOS_RES *res) {
  if (res == NULL || TD_RES_TMQ_META(res)) {
149 150 151
    return terrno;
  }

152
  if (TD_RES_TMQ(res) || TD_RES_TMQ_METADATA(res)) {
153 154 155
    return 0;
  }

156
  return ((SRequestObj *)res)->code;
157 158
}

159
const char *taos_errstr(TAOS_RES *res) {
L
Liu Jicong 已提交
160
  if (res == NULL || TD_RES_TMQ_META(res)) {
L
Liu Jicong 已提交
161
    return (const char *)tstrerror(terrno);
162 163
  }

164
  if (TD_RES_TMQ(res) || TD_RES_TMQ_METADATA(res)) {
165 166 167 168
    return "success";
  }

  SRequestObj *pRequest = (SRequestObj *)res;
169
  if (NULL != pRequest->msgBuf && (strlen(pRequest->msgBuf) > 0 || pRequest->code == TSDB_CODE_RPC_FQDN_ERROR)) {
170 171
    return pRequest->msgBuf;
  } else {
172
    return (const char *)tstrerror(pRequest->code);
173
  }
174 175 176
}

void taos_free_result(TAOS_RES *res) {
D
stmt  
dapan1121 已提交
177 178 179
  if (NULL == res) {
    return;
  }
180

L
Liu Jicong 已提交
181 182
  if (TD_RES_QUERY(res)) {
    SRequestObj *pRequest = (SRequestObj *)res;
D
dapan1121 已提交
183
    tscDebug("0x%" PRIx64 " taos_free_result start to free query", pRequest->requestId);
L
Liu Jicong 已提交
184
    destroyRequest(pRequest);
185
  } else if (TD_RES_TMQ_METADATA(res)) {
L
Liu Jicong 已提交
186 187 188 189 190 191 192 193 194 195 196 197
    SMqTaosxRspObj *pRsp = (SMqTaosxRspObj *)res;
    if (pRsp->rsp.blockData) taosArrayDestroyP(pRsp->rsp.blockData, taosMemoryFree);
    if (pRsp->rsp.blockDataLen) taosArrayDestroy(pRsp->rsp.blockDataLen);
    if (pRsp->rsp.withTbName) taosArrayDestroyP(pRsp->rsp.blockTbName, taosMemoryFree);
    if (pRsp->rsp.withSchema) taosArrayDestroyP(pRsp->rsp.blockSchema, (FDelete)tDeleteSSchemaWrapper);
    // taosx
    taosArrayDestroy(pRsp->rsp.createTableLen);
    taosArrayDestroyP(pRsp->rsp.createTableReq, taosMemoryFree);

    pRsp->resInfo.pRspMsg = NULL;
    doFreeReqResultInfo(&pRsp->resInfo);
    taosMemoryFree(pRsp);
L
Liu Jicong 已提交
198 199 200 201
  } else if (TD_RES_TMQ(res)) {
    SMqRspObj *pRsp = (SMqRspObj *)res;
    if (pRsp->rsp.blockData) taosArrayDestroyP(pRsp->rsp.blockData, taosMemoryFree);
    if (pRsp->rsp.blockDataLen) taosArrayDestroy(pRsp->rsp.blockDataLen);
L
Liu Jicong 已提交
202 203
    if (pRsp->rsp.withTbName) taosArrayDestroyP(pRsp->rsp.blockTbName, taosMemoryFree);
    if (pRsp->rsp.withSchema) taosArrayDestroyP(pRsp->rsp.blockSchema, (FDelete)tDeleteSSchemaWrapper);
L
Liu Jicong 已提交
204 205
    pRsp->resInfo.pRspMsg = NULL;
    doFreeReqResultInfo(&pRsp->resInfo);
206
    taosMemoryFree(pRsp);
L
Liu Jicong 已提交
207 208 209 210
  } else if (TD_RES_TMQ_META(res)) {
    SMqMetaRspObj *pRspObj = (SMqMetaRspObj *)res;
    taosMemoryFree(pRspObj->metaRsp.metaRsp);
    taosMemoryFree(pRspObj);
L
Liu Jicong 已提交
211
  }
212 213
}

D
dapan1121 已提交
214 215 216 217
void taos_kill_query(TAOS *taos) {
  if (NULL == taos) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
218

dengyihao's avatar
dengyihao 已提交
219
  int64_t  rid = *(int64_t *)taos;
dengyihao's avatar
dengyihao 已提交
220
  STscObj *pTscObj = acquireTscObj(rid);
D
dapan1121 已提交
221 222 223
  if (pTscObj) {
    stopAllRequests(pTscObj->pRequests);
  }
D
dapan1121 已提交
224 225 226
  releaseTscObj(rid);
}

L
Liu Jicong 已提交
227
int taos_field_count(TAOS_RES *res) {
L
Liu Jicong 已提交
228
  if (res == NULL || TD_RES_TMQ_META(res)) {
H
Haojun Liao 已提交
229 230 231
    return 0;
  }

L
Liu Jicong 已提交
232
  SReqResultInfo *pResInfo = tscGetCurResInfo(res);
H
Haojun Liao 已提交
233 234 235
  return pResInfo->numOfCols;
}

L
Liu Jicong 已提交
236
int taos_num_fields(TAOS_RES *res) { return taos_field_count(res); }
H
Haojun Liao 已提交
237 238

TAOS_FIELD *taos_fetch_fields(TAOS_RES *res) {
L
Liu Jicong 已提交
239
  if (taos_num_fields(res) == 0 || TD_RES_TMQ_META(res)) {
H
Haojun Liao 已提交
240 241 242
    return NULL;
  }

L
Liu Jicong 已提交
243
  SReqResultInfo *pResInfo = tscGetCurResInfo(res);
244
  return pResInfo->userFields;
H
Haojun Liao 已提交
245 246
}

247
TAOS_RES *taos_query(TAOS *taos, const char *sql) { return taosQueryImpl(taos, sql, false); }
dengyihao's avatar
dengyihao 已提交
248 249 250
TAOS_RES *taos_query_with_reqid(TAOS *taos, const char *sql, int64_t reqid) {
  return taosQueryImplWithReqid(taos, sql, false, reqid);
}
251

252 253
TAOS_ROW taos_fetch_row(TAOS_RES *res) {
  if (res == NULL) {
254 255 256
    return NULL;
  }

L
Liu Jicong 已提交
257 258 259
  if (TD_RES_QUERY(res)) {
    SRequestObj *pRequest = (SRequestObj *)res;
    if (pRequest->type == TSDB_SQL_RETRIEVE_EMPTY_RESULT || pRequest->type == TSDB_SQL_INSERT ||
D
dapan1121 已提交
260
        pRequest->code != TSDB_CODE_SUCCESS || taos_num_fields(res) == 0 || pRequest->killed) {
L
Liu Jicong 已提交
261 262
      return NULL;
    }
D
dapan1121 已提交
263 264

    return doAsyncFetchRows(pRequest, true, true);
265
  } else if (TD_RES_TMQ(res) || TD_RES_TMQ_METADATA(res)) {
dengyihao's avatar
dengyihao 已提交
266
    SMqRspObj      *msg = ((SMqRspObj *)res);
L
Liu Jicong 已提交
267 268 269 270 271 272
    SReqResultInfo *pResultInfo;
    if (msg->resIter == -1) {
      pResultInfo = tmqGetNextResInfo(res, true);
    } else {
      pResultInfo = tmqGetCurResInfo(res);
    }
273

L
Liu Jicong 已提交
274
    if (pResultInfo->current < pResultInfo->numOfRows) {
L
Liu Jicong 已提交
275 276
      doSetOneRowPtr(pResultInfo);
      pResultInfo->current += 1;
L
Liu Jicong 已提交
277 278
      return pResultInfo->row;
    } else {
L
Liu Jicong 已提交
279 280 281 282 283
      pResultInfo = tmqGetNextResInfo(res, true);
      if (pResultInfo == NULL) return NULL;
      doSetOneRowPtr(pResultInfo);
      pResultInfo->current += 1;
      return pResultInfo->row;
L
Liu Jicong 已提交
284
    }
L
Liu Jicong 已提交
285 286
  } else if (TD_RES_TMQ_META(res)) {
    return NULL;
L
Liu Jicong 已提交
287
  } else {
288
    // assert to avoid un-initialization error
289 290
    tscError("invalid result passed to taos_fetch_row");
    return NULL;
L
Liu Jicong 已提交
291 292
  }
  return NULL;
293
}
H
Haojun Liao 已提交
294

L
Liu Jicong 已提交
295
int taos_print_row(char *str, TAOS_ROW row, TAOS_FIELD *fields, int num_fields) {
H
Haojun Liao 已提交
296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353
  int32_t len = 0;
  for (int i = 0; i < num_fields; ++i) {
    if (i > 0) {
      str[len++] = ' ';
    }

    if (row[i] == NULL) {
      len += sprintf(str + len, "%s", TSDB_DATA_NULL_STR);
      continue;
    }

    switch (fields[i].type) {
      case TSDB_DATA_TYPE_TINYINT:
        len += sprintf(str + len, "%d", *((int8_t *)row[i]));
        break;

      case TSDB_DATA_TYPE_UTINYINT:
        len += sprintf(str + len, "%u", *((uint8_t *)row[i]));
        break;

      case TSDB_DATA_TYPE_SMALLINT:
        len += sprintf(str + len, "%d", *((int16_t *)row[i]));
        break;

      case TSDB_DATA_TYPE_USMALLINT:
        len += sprintf(str + len, "%u", *((uint16_t *)row[i]));
        break;

      case TSDB_DATA_TYPE_INT:
        len += sprintf(str + len, "%d", *((int32_t *)row[i]));
        break;

      case TSDB_DATA_TYPE_UINT:
        len += sprintf(str + len, "%u", *((uint32_t *)row[i]));
        break;

      case TSDB_DATA_TYPE_BIGINT:
        len += sprintf(str + len, "%" PRId64, *((int64_t *)row[i]));
        break;

      case TSDB_DATA_TYPE_UBIGINT:
        len += sprintf(str + len, "%" PRIu64, *((uint64_t *)row[i]));
        break;

      case TSDB_DATA_TYPE_FLOAT: {
        float fv = 0;
        fv = GET_FLOAT_VAL(row[i]);
        len += sprintf(str + len, "%f", fv);
      } break;

      case TSDB_DATA_TYPE_DOUBLE: {
        double dv = 0;
        dv = GET_DOUBLE_VAL(row[i]);
        len += sprintf(str + len, "%lf", dv);
      } break;

      case TSDB_DATA_TYPE_BINARY:
      case TSDB_DATA_TYPE_NCHAR: {
L
Liu Jicong 已提交
354
        int32_t charLen = varDataLen((char *)row[i] - VARSTR_HEADER_SIZE);
H
Haojun Liao 已提交
355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374
        if (fields[i].type == TSDB_DATA_TYPE_BINARY) {
          assert(charLen <= fields[i].bytes && charLen >= 0);
        } else {
          assert(charLen <= fields[i].bytes * TSDB_NCHAR_SIZE && charLen >= 0);
        }

        memcpy(str + len, row[i], charLen);
        len += charLen;
      } break;

      case TSDB_DATA_TYPE_TIMESTAMP:
        len += sprintf(str + len, "%" PRId64, *((int64_t *)row[i]));
        break;

      case TSDB_DATA_TYPE_BOOL:
        len += sprintf(str + len, "%d", *((int8_t *)row[i]));
      default:
        break;
    }
  }
L
Liu Jicong 已提交
375
  str[len] = 0;
H
Haojun Liao 已提交
376 377 378

  return len;
}
379

L
Liu Jicong 已提交
380
int *taos_fetch_lengths(TAOS_RES *res) {
L
Liu Jicong 已提交
381
  if (res == NULL || TD_RES_TMQ_META(res)) {
382 383 384
    return NULL;
  }

L
Liu Jicong 已提交
385 386
  SReqResultInfo *pResInfo = tscGetCurResInfo(res);
  return pResInfo->length;
387 388
}

389
TAOS_ROW *taos_result_block(TAOS_RES *res) {
L
Liu Jicong 已提交
390
  if (res == NULL || TD_RES_TMQ_META(res)) {
391 392 393 394 395 396 397 398
    terrno = TSDB_CODE_INVALID_PARA;
    return NULL;
  }

  if (taos_is_update_query(res)) {
    return NULL;
  }

L
Liu Jicong 已提交
399 400
  SReqResultInfo *pResInfo = tscGetCurResInfo(res);
  return &pResInfo->row;
401 402
}

403
// todo intergrate with tDataTypes
404 405
const char *taos_data_type(int type) {
  switch (type) {
L
Liu Jicong 已提交
406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432
    case TSDB_DATA_TYPE_NULL:
      return "TSDB_DATA_TYPE_NULL";
    case TSDB_DATA_TYPE_BOOL:
      return "TSDB_DATA_TYPE_BOOL";
    case TSDB_DATA_TYPE_TINYINT:
      return "TSDB_DATA_TYPE_TINYINT";
    case TSDB_DATA_TYPE_SMALLINT:
      return "TSDB_DATA_TYPE_SMALLINT";
    case TSDB_DATA_TYPE_INT:
      return "TSDB_DATA_TYPE_INT";
    case TSDB_DATA_TYPE_BIGINT:
      return "TSDB_DATA_TYPE_BIGINT";
    case TSDB_DATA_TYPE_FLOAT:
      return "TSDB_DATA_TYPE_FLOAT";
    case TSDB_DATA_TYPE_DOUBLE:
      return "TSDB_DATA_TYPE_DOUBLE";
    case TSDB_DATA_TYPE_VARCHAR:
      return "TSDB_DATA_TYPE_VARCHAR";
      //    case TSDB_DATA_TYPE_BINARY:          return "TSDB_DATA_TYPE_VARCHAR";
    case TSDB_DATA_TYPE_TIMESTAMP:
      return "TSDB_DATA_TYPE_TIMESTAMP";
    case TSDB_DATA_TYPE_NCHAR:
      return "TSDB_DATA_TYPE_NCHAR";
    case TSDB_DATA_TYPE_JSON:
      return "TSDB_DATA_TYPE_JSON";
    default:
      return "UNKNOWN";
433 434
  }
}
435 436 437

const char *taos_get_client_info() { return version; }

438
// return int32_t
X
Xiaoyu Wang 已提交
439
int taos_affected_rows(TAOS_RES *res) {
440
  if (res == NULL || TD_RES_TMQ(res) || TD_RES_TMQ_META(res) || TD_RES_TMQ_METADATA(res)) {
H
Haojun Liao 已提交
441 442 443
    return 0;
  }

444 445 446 447 448 449
  SRequestObj    *pRequest = (SRequestObj *)res;
  SReqResultInfo *pResInfo = &pRequest->body.resInfo;
  return (int)pResInfo->numOfRows;
}

// return int64_t
A
Alex Duan 已提交
450
int64_t taos_affected_rows64(TAOS_RES *res) {
451 452 453 454
  if (res == NULL || TD_RES_TMQ(res) || TD_RES_TMQ_META(res) || TD_RES_TMQ_METADATA(res)) {
    return 0;
  }

dengyihao's avatar
dengyihao 已提交
455
  SRequestObj    *pRequest = (SRequestObj *)res;
L
Liu Jicong 已提交
456
  SReqResultInfo *pResInfo = &pRequest->body.resInfo;
H
Haojun Liao 已提交
457
  return pResInfo->numOfRows;
X
Xiaoyu Wang 已提交
458
}
459

460
int taos_result_precision(TAOS_RES *res) {
L
Liu Jicong 已提交
461
  if (res == NULL || TD_RES_TMQ_META(res)) {
H
Haojun Liao 已提交
462 463
    return TSDB_TIME_PRECISION_MILLI;
  }
464

L
Liu Jicong 已提交
465 466 467
  if (TD_RES_QUERY(res)) {
    SRequestObj *pRequest = (SRequestObj *)res;
    return pRequest->body.resInfo.precision;
468
  } else if (TD_RES_TMQ(res) || TD_RES_TMQ_METADATA(res)) {
L
Liu Jicong 已提交
469 470 471 472
    SReqResultInfo *info = tmqGetCurResInfo(res);
    return info->precision;
  }
  return TSDB_TIME_PRECISION_MILLI;
473
}
474 475

int taos_select_db(TAOS *taos, const char *db) {
L
Liu Jicong 已提交
476
  STscObj *pObj = acquireTscObj(*(int64_t *)taos);
477
  if (pObj == NULL) {
L
Liu Jicong 已提交
478
    releaseTscObj(*(int64_t *)taos);
479 480 481 482 483
    terrno = TSDB_CODE_TSC_DISCONNECTED;
    return TSDB_CODE_TSC_DISCONNECTED;
  }

  if (db == NULL || strlen(db) == 0) {
L
Liu Jicong 已提交
484
    releaseTscObj(*(int64_t *)taos);
485 486 487 488 489 490 491
    terrno = TSDB_CODE_TSC_INVALID_INPUT;
    return terrno;
  }

  char sql[256] = {0};
  snprintf(sql, tListLen(sql), "use %s", db);

L
Liu Jicong 已提交
492 493
  TAOS_RES *pRequest = taos_query(taos, sql);
  int32_t   code = taos_errno(pRequest);
494 495

  taos_free_result(pRequest);
L
Liu Jicong 已提交
496
  releaseTscObj(*(int64_t *)taos);
497 498 499 500
  return code;
}

void taos_stop_query(TAOS_RES *res) {
501
  if (res == NULL || TD_RES_TMQ(res) || TD_RES_TMQ_META(res) || TD_RES_TMQ_METADATA(res)) {
502 503 504
    return;
  }

L
Liu Jicong 已提交
505
  SRequestObj *pRequest = (SRequestObj *)res;
D
dapan1121 已提交
506
  pRequest->killed = true;
L
Liu Jicong 已提交
507 508

  int32_t numOfFields = taos_num_fields(pRequest);
509 510
  // It is not a query, no need to stop.
  if (numOfFields == 0) {
D
dapan1121 已提交
511
    tscDebug("request 0x%" PRIx64 " no need to be killed since not query", pRequest->requestId);
512 513 514
    return;
  }

D
dapan1121 已提交
515
  schedulerFreeJob(&pRequest->body.queryJob, TSDB_CODE_TSC_QUERY_KILLED);
D
dapan1121 已提交
516
  tscDebug("request %" PRIx64 " killed", pRequest->requestId);
517 518 519
}

bool taos_is_null(TAOS_RES *res, int32_t row, int32_t col) {
L
Liu Jicong 已提交
520 521 522
  if (res == NULL || TD_RES_TMQ_META(res)) {
    return true;
  }
L
Liu Jicong 已提交
523
  SReqResultInfo *pResultInfo = tscGetCurResInfo(res);
524 525 526 527
  if (col >= pResultInfo->numOfCols || col < 0 || row >= pResultInfo->numOfRows || row < 0) {
    return true;
  }

L
Liu Jicong 已提交
528
  SResultColumn *pCol = &pResultInfo->pCol[col];
529 530 531 532 533
  if (IS_VAR_DATA_TYPE(pResultInfo->fields[col].type)) {
    return (pCol->offset[row] == -1);
  } else {
    return colDataIsNull_f(pCol->nullbitmap, row);
  }
534 535
}

L
Liu Jicong 已提交
536
bool taos_is_update_query(TAOS_RES *res) { return taos_num_fields(res) == 0; }
537

L
Liu Jicong 已提交
538
int taos_fetch_block(TAOS_RES *res, TAOS_ROW *rows) {
H
Haojun Liao 已提交
539
  int32_t numOfRows = 0;
L
Liu Jicong 已提交
540
  /*int32_t code = */ taos_fetch_block_s(res, &numOfRows, rows);
H
Haojun Liao 已提交
541 542 543
  return numOfRows;
}

L
Liu Jicong 已提交
544
int taos_fetch_block_s(TAOS_RES *res, int *numOfRows, TAOS_ROW *rows) {
L
Liu Jicong 已提交
545
  if (res == NULL || TD_RES_TMQ_META(res)) {
546 547
    return 0;
  }
548

L
Liu Jicong 已提交
549 550
  if (TD_RES_QUERY(res)) {
    SRequestObj *pRequest = (SRequestObj *)res;
551

L
Liu Jicong 已提交
552 553
    (*rows) = NULL;
    (*numOfRows) = 0;
H
Haojun Liao 已提交
554

L
Liu Jicong 已提交
555 556 557 558
    if (pRequest->type == TSDB_SQL_RETRIEVE_EMPTY_RESULT || pRequest->type == TSDB_SQL_INSERT ||
        pRequest->code != TSDB_CODE_SUCCESS || taos_num_fields(res) == 0) {
      return 0;
    }
559

560
    doAsyncFetchRows(pRequest, false, true);
561

L
Liu Jicong 已提交
562 563 564
    // TODO refactor
    SReqResultInfo *pResultInfo = &pRequest->body.resInfo;
    pResultInfo->current = pResultInfo->numOfRows;
565

L
Liu Jicong 已提交
566 567 568
    (*rows) = pResultInfo->row;
    (*numOfRows) = pResultInfo->numOfRows;
    return pRequest->code;
569
  } else if (TD_RES_TMQ(res) || TD_RES_TMQ_METADATA(res)) {
L
Liu Jicong 已提交
570
    SReqResultInfo *pResultInfo = tmqGetNextResInfo(res, true);
L
Liu Jicong 已提交
571
    if (pResultInfo == NULL) return -1;
H
Haojun Liao 已提交
572

L
Liu Jicong 已提交
573 574 575
    pResultInfo->current = pResultInfo->numOfRows;
    (*rows) = pResultInfo->row;
    (*numOfRows) = pResultInfo->numOfRows;
H
Haojun Liao 已提交
576
    return 0;
L
Liu Jicong 已提交
577 578 579
  } else {
    ASSERT(0);
    return -1;
H
Haojun Liao 已提交
580
  }
L
Liu Jicong 已提交
581
}
H
Haojun Liao 已提交
582

L
Liu Jicong 已提交
583
int taos_fetch_raw_block(TAOS_RES *res, int *numOfRows, void **pData) {
L
Liu Jicong 已提交
584
  if (res == NULL || TD_RES_TMQ_META(res)) {
H
Haojun Liao 已提交
585 586
    return 0;
  }
587

588
  if (TD_RES_TMQ(res) || TD_RES_TMQ_METADATA(res)) {
L
Liu Jicong 已提交
589
    SReqResultInfo *pResultInfo = tmqGetNextResInfo(res, false);
L
Liu Jicong 已提交
590 591 592 593
    if (pResultInfo == NULL) {
      (*numOfRows) = 0;
      return 0;
    }
H
Haojun Liao 已提交
594

L
Liu Jicong 已提交
595 596 597 598
    pResultInfo->current = pResultInfo->numOfRows;
    (*numOfRows) = pResultInfo->numOfRows;
    (*pData) = (void *)pResultInfo->pData;
    return 0;
599
  }
L
Liu Jicong 已提交
600

601 602 603 604 605
  SRequestObj *pRequest = (SRequestObj *)res;

  if (pRequest->type == TSDB_SQL_RETRIEVE_EMPTY_RESULT || pRequest->type == TSDB_SQL_INSERT ||
      pRequest->code != TSDB_CODE_SUCCESS || taos_num_fields(res) == 0) {
    return 0;
L
Liu Jicong 已提交
606
  }
607

608
  doAsyncFetchRows(pRequest, false, false);
609 610 611 612 613 614 615 616

  SReqResultInfo *pResultInfo = &pRequest->body.resInfo;

  pResultInfo->current = pResultInfo->numOfRows;
  (*numOfRows) = pResultInfo->numOfRows;
  (*pData) = (void *)pResultInfo->pData;

  return 0;
617 618
}

H
Haojun Liao 已提交
619
int *taos_get_column_data_offset(TAOS_RES *res, int columnIndex) {
L
Liu Jicong 已提交
620
  if (res == NULL || TD_RES_TMQ_META(res)) {
H
Haojun Liao 已提交
621 622 623
    return 0;
  }

L
Liu Jicong 已提交
624
  int32_t numOfFields = taos_num_fields(res);
H
Haojun Liao 已提交
625 626 627 628
  if (columnIndex < 0 || columnIndex >= numOfFields || numOfFields == 0) {
    return 0;
  }

L
Liu Jicong 已提交
629
  SReqResultInfo *pResInfo = tscGetCurResInfo(res);
dengyihao's avatar
dengyihao 已提交
630
  TAOS_FIELD     *pField = &pResInfo->userFields[columnIndex];
H
Haojun Liao 已提交
631 632 633 634
  if (!IS_VAR_DATA_TYPE(pField->type)) {
    return 0;
  }

L
Liu Jicong 已提交
635
  return pResInfo->pCol[columnIndex].offset;
H
Haojun Liao 已提交
636 637
}

638 639
int taos_validate_sql(TAOS *taos, const char *sql) {
  TAOS_RES *pObj = taosQueryImpl(taos, sql, true);
D
dapan1121 已提交
640 641

  int code = taos_errno(pObj);
642

D
dapan1121 已提交
643 644 645
  taos_free_result(pObj);
  return code;
}
646

647
void taos_reset_current_db(TAOS *taos) {
L
Liu Jicong 已提交
648
  STscObj *pTscObj = acquireTscObj(*(int64_t *)taos);
649 650
  if (pTscObj == NULL) {
    terrno = TSDB_CODE_TSC_DISCONNECTED;
H
Haojun Liao 已提交
651 652 653
    return;
  }

654 655
  resetConnectDB(pTscObj);

L
Liu Jicong 已提交
656
  releaseTscObj(*(int64_t *)taos);
657 658
}

659
const char *taos_get_server_info(TAOS *taos) {
L
Liu Jicong 已提交
660
  STscObj *pTscObj = acquireTscObj(*(int64_t *)taos);
661 662
  if (pTscObj == NULL) {
    terrno = TSDB_CODE_TSC_DISCONNECTED;
663 664 665
    return NULL;
  }

L
Liu Jicong 已提交
666
  releaseTscObj(*(int64_t *)taos);
667

D
dapan1121 已提交
668
  return pTscObj->sDetailVer;
669 670
}

671 672 673 674 675
static void destoryTablesReq(void *p) {
  STablesReq *pRes = (STablesReq *)p;
  taosArrayDestroy(pRes->pTables);
}

676 677 678 679 680 681 682 683 684 685 686 687 688
static void destoryCatalogReq(SCatalogReq *pCatalogReq) {
  if (NULL == pCatalogReq) {
    return;
  }
  taosArrayDestroy(pCatalogReq->pDbVgroup);
  taosArrayDestroy(pCatalogReq->pDbCfg);
  taosArrayDestroy(pCatalogReq->pDbInfo);
  taosArrayDestroyEx(pCatalogReq->pTableMeta, destoryTablesReq);
  taosArrayDestroyEx(pCatalogReq->pTableHash, destoryTablesReq);
  taosArrayDestroy(pCatalogReq->pUdf);
  taosArrayDestroy(pCatalogReq->pIndex);
  taosArrayDestroy(pCatalogReq->pUser);
  taosArrayDestroy(pCatalogReq->pTableIndex);
X
Xiaoyu Wang 已提交
689
  taosArrayDestroy(pCatalogReq->pTableCfg);
690 691 692 693 694 695 696 697 698
  taosMemoryFree(pCatalogReq);
}

void destorySqlCallbackWrapper(SSqlCallbackWrapper *pWrapper) {
  if (NULL == pWrapper) {
    return;
  }
  destoryCatalogReq(pWrapper->pCatalogReq);
  qDestroyParseContext(pWrapper->pParseCtx);
699 700 701
  taosMemoryFree(pWrapper);
}

X
Xiaoyu Wang 已提交
702
static void doAsyncQueryFromAnalyse(SMetaData *pResultMeta, void *param, int32_t code) {
703 704 705
  SSqlCallbackWrapper *pWrapper = (SSqlCallbackWrapper *)param;
  SRequestObj         *pRequest = pWrapper->pRequest;
  SQuery              *pQuery = pRequest->pQuery;
706

D
dapan1121 已提交
707
  pRequest->metric.ctgEnd = taosGetTimestampUs();
708
  qDebug("0x%" PRIx64 " start to semantic analysis, reqId:0x%" PRIx64, pRequest->self, pRequest->requestId);
D
dapan1121 已提交
709

710
  if (code == TSDB_CODE_SUCCESS) {
711
    code = qAnalyseSqlSemantic(pWrapper->pParseCtx, pWrapper->pCatalogReq, pResultMeta, pQuery);
712
    pRequest->stableQuery = pQuery->stableQuery;
D
dapan1121 已提交
713
    if (pQuery->pRoot) {
dengyihao's avatar
dengyihao 已提交
714
      pRequest->stmtType = pQuery->pRoot->type;
D
dapan1121 已提交
715
    }
716 717
  }

D
dapan1121 已提交
718 719
  pRequest->metric.semanticEnd = taosGetTimestampUs();

720 721 722 723 724
  if (code == TSDB_CODE_SUCCESS) {
    if (pQuery->haveResultSet) {
      setResSchemaInfo(&pRequest->body.resInfo, pQuery->pResSchema, pQuery->numOfResCols);
      setResPrecision(&pRequest->body.resInfo, pQuery->precision);
    }
725

726 727
    TSWAP(pRequest->dbList, (pQuery)->pDbList);
    TSWAP(pRequest->tableList, (pQuery)->pTableList);
D
dapan1121 已提交
728
    TSWAP(pRequest->targetTableList, (pQuery)->pTargetTableList);
729

730
    double el = (pRequest->metric.semanticEnd - pRequest->metric.ctgEnd) / 1000.0;
731 732 733
    tscDebug("0x%" PRIx64 " analysis semantics completed, start async query, elapsed time:%.2f ms, reqId:0x%" PRIx64,
             pRequest->self, el, pRequest->requestId);

734
    launchAsyncQuery(pRequest, pQuery, pResultMeta, pWrapper);
735
  } else {
736
    destorySqlCallbackWrapper(pWrapper);
X
Xiaoyu Wang 已提交
737 738
    qDestroyQuery(pRequest->pQuery);
    pRequest->pQuery = NULL;
739

740
    if (NEED_CLIENT_HANDLE_ERROR(code)) {
741 742
      tscDebug("0x%" PRIx64 " client retry to handle the error, code:%d - %s, tryCount:%d, reqId:0x%" PRIx64,
               pRequest->self, code, tstrerror(code), pRequest->retry, pRequest->requestId);
743 744 745 746
      pRequest->prevCode = code;
      doAsyncQuery(pRequest, true);
      return;
    }
747

748
    // return to app directly
749 750
    tscError("0x%" PRIx64 " error occurs, code:%s, return to user app, reqId:0x%" PRIx64, pRequest->self,
             tstrerror(code), pRequest->requestId);
751 752 753
    pRequest->code = code;
    pRequest->body.queryFp(pRequest->body.param, pRequest, code);
  }
754 755
}

X
Xiaoyu Wang 已提交
756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798
static int32_t getAllMetaAsync(SSqlCallbackWrapper *pWrapper, catalogCallback fp) {
  SRequestConnInfo conn = {.pTrans = pWrapper->pParseCtx->pTransporter,
                           .requestId = pWrapper->pParseCtx->requestId,
                           .requestObjRefId = pWrapper->pParseCtx->requestRid,
                           .mgmtEps = pWrapper->pParseCtx->mgmtEpSet};

  pWrapper->pRequest->metric.ctgStart = taosGetTimestampUs();

  return catalogAsyncGetAllMeta(pWrapper->pParseCtx->pCatalog, &conn, pWrapper->pCatalogReq, fp, pWrapper,
                                &pWrapper->pRequest->body.queryJob);
}

static void doAsyncQueryFromParse(SMetaData *pResultMeta, void *param, int32_t code);

static int32_t phaseAsyncQuery(SSqlCallbackWrapper *pWrapper) {
  int32_t code = TSDB_CODE_SUCCESS;
  switch (pWrapper->pRequest->pQuery->execStage) {
    case QUERY_EXEC_STAGE_PARSE: {
      // continue parse after get metadata
      code = getAllMetaAsync(pWrapper, doAsyncQueryFromParse);
      break;
    }
    case QUERY_EXEC_STAGE_ANALYSE: {
      // analysis after get metadata
      code = getAllMetaAsync(pWrapper, doAsyncQueryFromAnalyse);
      break;
    }
    case QUERY_EXEC_STAGE_SCHEDULE: {
      launchAsyncQuery(pWrapper->pRequest, pWrapper->pRequest->pQuery, NULL, pWrapper);
      break;
    }
    default:
      break;
  }
  return code;
}

static void doAsyncQueryFromParse(SMetaData *pResultMeta, void *param, int32_t code) {
  SSqlCallbackWrapper *pWrapper = (SSqlCallbackWrapper *)param;
  SRequestObj         *pRequest = pWrapper->pRequest;
  SQuery              *pQuery = pRequest->pQuery;

  pRequest->metric.ctgEnd = taosGetTimestampUs();
799
  qDebug("0x%" PRIx64 " start to continue parse, reqId:0x%" PRIx64 ", code:%s", pRequest->self, pRequest->requestId, tstrerror(code));
X
Xiaoyu Wang 已提交
800 801

  if (code == TSDB_CODE_SUCCESS) {
802
    pWrapper->pCatalogReq->forceUpdate = false;
X
Xiaoyu Wang 已提交
803 804 805 806 807 808 809 810 811 812 813 814
    code = qContinueParseSql(pWrapper->pParseCtx, pWrapper->pCatalogReq, pResultMeta, pQuery);
  }

  if (TSDB_CODE_SUCCESS == code) {
    code = phaseAsyncQuery(pWrapper);
  }

  if (TSDB_CODE_SUCCESS != code) {
    tscError("0x%" PRIx64 " error happens, code:%d - %s, reqId:0x%" PRIx64, pWrapper->pRequest->self, code,
             tstrerror(code), pWrapper->pRequest->requestId);
    destorySqlCallbackWrapper(pWrapper);
    terrno = code;
X
Xiaoyu Wang 已提交
815 816
    pRequest->code = code;
    pRequest->body.queryFp(pRequest->body.param, pRequest, code);
X
Xiaoyu Wang 已提交
817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832
  }
}

void continueInsertFromCsv(SSqlCallbackWrapper *pWrapper, SRequestObj *pRequest) {
  int32_t code = qParseSqlSyntax(pWrapper->pParseCtx, &pRequest->pQuery, pWrapper->pCatalogReq);
  if (TSDB_CODE_SUCCESS == code) {
    code = phaseAsyncQuery(pWrapper);
  }

  if (TSDB_CODE_SUCCESS != code) {
    tscError("0x%" PRIx64 " error happens, code:%d - %s, reqId:0x%" PRIx64, pWrapper->pRequest->self, code,
             tstrerror(code), pWrapper->pRequest->requestId);
    destorySqlCallbackWrapper(pWrapper);
    terrno = code;
    pWrapper->pRequest->code = code;
    pWrapper->pRequest->body.queryFp(pWrapper->pRequest->body.param, pWrapper->pRequest, code);
833 834 835
  }
}

836
void taos_query_a(TAOS *taos, const char *sql, __taos_async_fn_t fp, void *param) {
dengyihao's avatar
dengyihao 已提交
837
  int64_t connId = *(int64_t *)taos;
838
  taosAsyncQueryImpl(connId, sql, fp, param, false);
839
}
840

dengyihao's avatar
dengyihao 已提交
841 842 843 844 845
void taos_query_a_with_reqid(TAOS *taos, const char *sql, __taos_async_fn_t fp, void *param, int64_t reqid) {
  int64_t connId = *(int64_t *)taos;
  taosAsyncQueryImplWithReqid(connId, sql, fp, param, false, reqid);
}

846
int32_t createParseContext(const SRequestObj *pRequest, SParseContext **pCxt) {
847 848 849 850 851 852 853
  const STscObj *pTscObj = pRequest->pTscObj;

  *pCxt = taosMemoryCalloc(1, sizeof(SParseContext));
  if (*pCxt == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

dengyihao's avatar
dengyihao 已提交
854 855 856 857 858 859 860 861 862 863 864 865 866 867
  **pCxt = (SParseContext){.requestId = pRequest->requestId,
                           .requestRid = pRequest->self,
                           .acctId = pTscObj->acctId,
                           .db = pRequest->pDb,
                           .topicQuery = false,
                           .pSql = pRequest->sqlstr,
                           .sqlLen = pRequest->sqlLen,
                           .pMsg = pRequest->msgBuf,
                           .msgLen = ERROR_MSG_BUF_DEFAULT_SIZE,
                           .pTransporter = pTscObj->pAppInfo->pTransporter,
                           .pStmtCb = NULL,
                           .pUser = pTscObj->user,
                           .schemalessType = pTscObj->schemalessType,
                           .isSuperUser = (0 == strcmp(pTscObj->user, TSDB_DEFAULT_USER)),
868
                           .enableSysInfo = pTscObj->sysInfo,
dengyihao's avatar
dengyihao 已提交
869 870
                           .async = true,
                           .svrVer = pTscObj->sVer,
871 872
                           .nodeOffline = (pTscObj->pAppInfo->onlineDnodes < pTscObj->pAppInfo->totalDnodes),
                           .allocatorId = pRequest->allocatorRefId};
873 874 875
  return TSDB_CODE_SUCCESS;
}

876
void doAsyncQuery(SRequestObj *pRequest, bool updateMetaForce) {
877 878 879
  STscObj             *pTscObj = pRequest->pTscObj;
  SSqlCallbackWrapper *pWrapper = NULL;
  int32_t              code = TSDB_CODE_SUCCESS;
880 881

  if (pRequest->retry++ > REQUEST_TOTAL_EXEC_TIMES) {
882
    code = pRequest->prevCode;
883 884
  }

885 886 887 888 889 890 891
  if (TSDB_CODE_SUCCESS == code) {
    pWrapper = taosMemoryCalloc(1, sizeof(SSqlCallbackWrapper));
    if (pWrapper == NULL) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    } else {
      pWrapper->pRequest = pRequest;
    }
892
  }
893

894 895
  if (TSDB_CODE_SUCCESS == code) {
    code = createParseContext(pRequest, &pWrapper->pParseCtx);
896 897
  }

898 899 900
  if (TSDB_CODE_SUCCESS == code) {
    pWrapper->pParseCtx->mgmtEpSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
    code = catalogGetHandle(pTscObj->pAppInfo->clusterId, &pWrapper->pParseCtx->pCatalog);
901 902
  }

903 904 905 906 907 908 909 910 911 912 913 914 915 916
  if (TSDB_CODE_SUCCESS == code) {
    pRequest->metric.syntaxStart = taosGetTimestampUs();

    pWrapper->pCatalogReq = taosMemoryCalloc(1, sizeof(SCatalogReq));
    if (pWrapper->pCatalogReq == NULL) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    } else {
      pWrapper->pCatalogReq->forceUpdate = updateMetaForce;
      pWrapper->pCatalogReq->qNodeRequired = qnodeRequired(pRequest);
      code = qParseSqlSyntax(pWrapper->pParseCtx, &pRequest->pQuery, pWrapper->pCatalogReq);
    }

    pRequest->metric.syntaxEnd = taosGetTimestampUs();
  }
D
dapan1121 已提交
917

918
  if (TSDB_CODE_SUCCESS == code && !updateMetaForce) {
D
dapan1121 已提交
919
    SAppClusterSummary *pActivity = &pTscObj->pAppInfo->summary;
X
Xiaoyu Wang 已提交
920
    if (QUERY_NODE_INSERT_STMT == nodeType(pRequest->pQuery->pRoot)) {
dengyihao's avatar
dengyihao 已提交
921
      atomic_add_fetch_64((int64_t *)&pActivity->numOfInsertsReq, 1);
X
Xiaoyu Wang 已提交
922
    } else if (QUERY_NODE_SELECT_STMT == nodeType(pRequest->pQuery->pRoot)) {
dengyihao's avatar
dengyihao 已提交
923
      atomic_add_fetch_64((int64_t *)&pActivity->numOfQueryReq, 1);
D
dapan1121 已提交
924 925 926
    }
  }

927
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
928 929
    phaseAsyncQuery(pWrapper);
  } else {
930 931 932 933 934 935 936
    tscError("0x%" PRIx64 " error happens, code:%d - %s, reqId:0x%" PRIx64, pRequest->self, code, tstrerror(code),
             pRequest->requestId);
    destorySqlCallbackWrapper(pWrapper);
    terrno = code;
    pRequest->code = code;
    pRequest->body.queryFp(pRequest->body.param, pRequest, code);
  }
937 938
}

939 940
static void fetchCallback(void *pResult, void *param, int32_t code) {
  SRequestObj *pRequest = (SRequestObj *)param;
941

942
  SReqResultInfo *pResultInfo = &pRequest->body.resInfo;
943
  pRequest->metric.resultReady = taosGetTimestampUs();
944

L
Liu Jicong 已提交
945 946
  tscDebug("0x%" PRIx64 " enter scheduler fetch cb, code:%d - %s, reqId:0x%" PRIx64, pRequest->self, code,
           tstrerror(code), pRequest->requestId);
D
dapan1121 已提交
947

948 949 950 951 952
  pResultInfo->pData = pResult;
  pResultInfo->numOfRows = 0;

  if (code != TSDB_CODE_SUCCESS) {
    pRequest->code = code;
953
    taosMemoryFreeClear(pResultInfo->pData);
954 955 956 957 958
    pRequest->body.fetchFp(pRequest->body.param, pRequest, 0);
    return;
  }

  if (pRequest->code != TSDB_CODE_SUCCESS) {
959
    taosMemoryFreeClear(pResultInfo->pData);
960
    pRequest->body.fetchFp(pRequest->body.param, pRequest, 0);
961
    return;
962 963
  }

964
  pRequest->code =
D
dapan1121 已提交
965
      setQueryResultFromRsp(pResultInfo, (SRetrieveTableRsp *)pResultInfo->pData, pResultInfo->convertUcs4, true);
966 967 968
  if (pRequest->code != TSDB_CODE_SUCCESS) {
    pResultInfo->numOfRows = 0;
    pRequest->code = code;
969 970 971
    tscError("0x%" PRIx64 " fetch results failed, code:%s, reqId:0x%" PRIx64, pRequest->self, tstrerror(code),
             pRequest->requestId);
  } else {
A
Alex Duan 已提交
972
    tscDebug("0x%" PRIx64 " fetch results, numOfRows:%" PRId64 " total Rows:%" PRId64 ", complete:%d, reqId:0x%" PRIx64,
973 974
             pRequest->self, pResultInfo->numOfRows, pResultInfo->totalRows, pResultInfo->completed,
             pRequest->requestId);
D
dapan1121 已提交
975 976 977

    STscObj            *pTscObj = pRequest->pTscObj;
    SAppClusterSummary *pActivity = &pTscObj->pAppInfo->summary;
dengyihao's avatar
dengyihao 已提交
978
    atomic_add_fetch_64((int64_t *)&pActivity->fetchBytes, pRequest->body.resInfo.payloadLen);
979 980 981 982 983
  }

  pRequest->body.fetchFp(pRequest->body.param, pRequest, pResultInfo->numOfRows);
}

984
void taos_fetch_rows_a(TAOS_RES *res, __taos_async_fn_t fp, void *param) {
985
  ASSERT(res != NULL && fp != NULL);
L
Liu Jicong 已提交
986
  ASSERT(TD_RES_QUERY(res));
987 988 989

  SRequestObj *pRequest = res;
  pRequest->body.fetchFp = fp;
990
  pRequest->body.param = param;
991 992

  SReqResultInfo *pResultInfo = &pRequest->body.resInfo;
993 994 995

  // this query has no results or error exists, return directly
  if (taos_num_fields(pRequest) == 0 || pRequest->code != TSDB_CODE_SUCCESS) {
996 997 998 999 1000
    pResultInfo->numOfRows = 0;
    pRequest->body.fetchFp(param, pRequest, pResultInfo->numOfRows);
    return;
  }

1001
  // all data has returned to App already, no need to try again
D
dapan1121 已提交
1002 1003
  if (pResultInfo->completed) {
    // it is a local executed query, no need to do async fetch
X
Xiaoyu Wang 已提交
1004
    if (QUERY_EXEC_MODE_SCHEDULE != pRequest->body.execMode) {
D
dapan1121 已提交
1005 1006 1007 1008 1009 1010
      if (pResultInfo->localResultFetched) {
        pResultInfo->numOfRows = 0;
        pResultInfo->current = 0;
      } else {
        pResultInfo->localResultFetched = true;
      }
1011
    } else {
D
dapan1121 已提交
1012
      pResultInfo->numOfRows = 0;
1013
    }
1014

D
dapan1121 已提交
1015
    pRequest->body.fetchFp(param, pRequest, pResultInfo->numOfRows);
1016
    return;
1017 1018
  }

D
dapan1121 已提交
1019
  SSchedulerReq req = {
dengyihao's avatar
dengyihao 已提交
1020 1021 1022
      .syncReq = false,
      .fetchFp = fetchCallback,
      .cbParam = pRequest,
D
dapan1121 已提交
1023
  };
1024

D
dapan1121 已提交
1025
  schedulerFetchRows(pRequest->body.queryJob, &req);
L
Liu Jicong 已提交
1026
}
1027

1028
void taos_fetch_raw_block_a(TAOS_RES *res, __taos_async_fn_t fp, void *param) {
1029
  ASSERT(res != NULL && fp != NULL);
L
Liu Jicong 已提交
1030
  ASSERT(TD_RES_QUERY(res));
1031

dengyihao's avatar
dengyihao 已提交
1032
  SRequestObj    *pRequest = res;
1033 1034 1035
  SReqResultInfo *pResultInfo = &pRequest->body.resInfo;

  // set the current block is all consumed
1036 1037
  pResultInfo->convertUcs4 = false;

1038 1039
  // it is a local executed query, no need to do async fetch
  taos_fetch_rows_a(pRequest, fp, param);
1040 1041
}

1042
const void *taos_get_raw_block(TAOS_RES *res) {
H
Haojun Liao 已提交
1043
  ASSERT(res != NULL);
L
Liu Jicong 已提交
1044
  ASSERT(TD_RES_QUERY(res));
1045
  SRequestObj *pRequest = res;
H
Haojun Liao 已提交
1046 1047 1048 1049

  return pRequest->body.resInfo.pData;
}

X
Xiaoyu Wang 已提交
1050
int taos_get_db_route_info(TAOS *taos, const char *db, TAOS_DB_ROUTE_INFO *dbInfo) {
1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061
  if (NULL == taos) {
    terrno = TSDB_CODE_TSC_DISCONNECTED;
    return terrno;
  }

  if (NULL == db || NULL == dbInfo) {
    tscError("invalid input param, db:%p, dbInfo:%p", db, dbInfo);
    terrno = TSDB_CODE_TSC_INVALID_INPUT;
    return terrno;
  }

X
Xiaoyu Wang 已提交
1062 1063 1064
  int64_t      connId = *(int64_t *)taos;
  SRequestObj *pRequest = NULL;
  char        *sql = "taos_get_db_route_info";
dengyihao's avatar
dengyihao 已提交
1065
  int32_t      code = buildRequest(connId, sql, strlen(sql), NULL, false, &pRequest, 0);
1066
  if (code != TSDB_CODE_SUCCESS) {
1067
    terrno = code;
1068 1069 1070
    return terrno;
  }

X
Xiaoyu Wang 已提交
1071
  STscObj  *pTscObj = pRequest->pTscObj;
1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084
  SCatalog *pCtg = NULL;
  code = catalogGetHandle(pTscObj->pAppInfo->clusterId, &pCtg);
  if (code != TSDB_CODE_SUCCESS) {
    goto _return;
  }

  SRequestConnInfo conn = {
      .pTrans = pTscObj->pAppInfo->pTransporter, .requestId = pRequest->requestId, .requestObjRefId = pRequest->self};

  conn.mgmtEps = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);

  char dbFName[TSDB_DB_FNAME_LEN] = {0};
  snprintf(dbFName, sizeof(dbFName), "%d.%s", pTscObj->acctId, db);
X
Xiaoyu Wang 已提交
1085

1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098
  code = catalogGetDBVgInfo(pCtg, &conn, dbFName, dbInfo);
  if (code) {
    goto _return;
  }

_return:

  terrno = code;

  destroyRequest(pRequest);
  return code;
}

X
Xiaoyu Wang 已提交
1099
int taos_get_table_vgId(TAOS *taos, const char *db, const char *table, int *vgId) {
1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110
  if (NULL == taos) {
    terrno = TSDB_CODE_TSC_DISCONNECTED;
    return terrno;
  }

  if (NULL == db || NULL == table || NULL == vgId) {
    tscError("invalid input param, db:%p, table:%p, vgId:%p", db, table, vgId);
    terrno = TSDB_CODE_TSC_INVALID_INPUT;
    return terrno;
  }

X
Xiaoyu Wang 已提交
1111 1112 1113
  int64_t      connId = *(int64_t *)taos;
  SRequestObj *pRequest = NULL;
  char        *sql = "taos_get_table_vgId";
dengyihao's avatar
dengyihao 已提交
1114
  int32_t      code = buildRequest(connId, sql, strlen(sql), NULL, false, &pRequest, 0);
1115 1116 1117 1118
  if (code != TSDB_CODE_SUCCESS) {
    return terrno;
  }

1119 1120
  pRequest->syncQuery = true;

X
Xiaoyu Wang 已提交
1121
  STscObj  *pTscObj = pRequest->pTscObj;
1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146
  SCatalog *pCtg = NULL;
  code = catalogGetHandle(pTscObj->pAppInfo->clusterId, &pCtg);
  if (code != TSDB_CODE_SUCCESS) {
    goto _return;
  }

  SRequestConnInfo conn = {
      .pTrans = pTscObj->pAppInfo->pTransporter, .requestId = pRequest->requestId, .requestObjRefId = pRequest->self};

  conn.mgmtEps = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);

  SName tableName;
  toName(pTscObj->acctId, db, table, &tableName);

  SVgroupInfo vgInfo;
  code = catalogGetTableHashVgroup(pCtg, &conn, &tableName, &vgInfo);
  if (code) {
    goto _return;
  }

  *vgId = vgInfo.vgId;

_return:

  terrno = code;
1147

1148 1149
  destroyRequest(pRequest);
  return code;
1150 1151
}

1152
int taos_load_table_info(TAOS *taos, const char *tableNameList) {
D
dapan1121 已提交
1153 1154 1155 1156 1157
  if (NULL == taos) {
    terrno = TSDB_CODE_TSC_DISCONNECTED;
    return terrno;
  }

1158
  int64_t       connId = *(int64_t *)taos;
D
dapan1121 已提交
1159
  const int32_t MAX_TABLE_NAME_LENGTH = 12 * 1024 * 1024;  // 12MB list
1160
  int32_t       code = 0;
dengyihao's avatar
dengyihao 已提交
1161
  SRequestObj  *pRequest = NULL;
1162 1163
  SCatalogReq   catalogReq = {0};

D
dapan1121 已提交
1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175
  if (NULL == tableNameList) {
    return TSDB_CODE_SUCCESS;
  }

  int32_t length = (int32_t)strlen(tableNameList);
  if (0 == length) {
    return TSDB_CODE_SUCCESS;
  } else if (length > MAX_TABLE_NAME_LENGTH) {
    tscError("tableNameList too long, length:%d, maximum allowed:%d", length, MAX_TABLE_NAME_LENGTH);
    return TSDB_CODE_TSC_INVALID_OPERATION;
  }

1176
  char *sql = "taos_load_table_info";
dengyihao's avatar
dengyihao 已提交
1177
  code = buildRequest(connId, sql, strlen(sql), NULL, false, &pRequest, 0);
1178 1179 1180
  if (code != TSDB_CODE_SUCCESS) {
    terrno = code;
    goto _return;
D
dapan1121 已提交
1181 1182
  }

dengyihao's avatar
dengyihao 已提交
1183
  STscObj *pTscObj = pRequest->pTscObj;
D
dapan1121 已提交
1184 1185 1186 1187 1188
  code = transferTableNameList(tableNameList, pTscObj->acctId, pTscObj->db, &catalogReq.pTableMeta);
  if (code) {
    goto _return;
  }

1189
  SCatalog *pCtg = NULL;
D
dapan1121 已提交
1190 1191 1192 1193 1194
  code = catalogGetHandle(pTscObj->pAppInfo->clusterId, &pCtg);
  if (code != TSDB_CODE_SUCCESS) {
    goto _return;
  }

1195 1196
  SRequestConnInfo conn = {
      .pTrans = pTscObj->pAppInfo->pTransporter, .requestId = pRequest->requestId, .requestObjRefId = pRequest->self};
D
dapan1121 已提交
1197 1198 1199

  conn.mgmtEps = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);

D
dapan1121 已提交
1200
  code = catalogAsyncGetAllMeta(pCtg, &conn, &catalogReq, syncCatalogFn, pRequest->body.param, NULL);
D
dapan1121 已提交
1201 1202 1203 1204
  if (code) {
    goto _return;
  }

dengyihao's avatar
dengyihao 已提交
1205
  SSyncQueryParam *pParam = pRequest->body.param;
1206
  tsem_wait(&pParam->sem);
D
dapan1121 已提交
1207 1208 1209 1210 1211

_return:
  taosArrayDestroy(catalogReq.pTableMeta);
  destroyRequest(pRequest);
  return code;
1212 1213
}

L
Liu Jicong 已提交
1214
TAOS_STMT *taos_stmt_init(TAOS *taos) {
L
Liu Jicong 已提交
1215
  STscObj *pObj = acquireTscObj(*(int64_t *)taos);
1216 1217 1218
  if (NULL == pObj) {
    tscError("invalid parameter for %s", __FUNCTION__);
    terrno = TSDB_CODE_TSC_DISCONNECTED;
D
stmt  
dapan1121 已提交
1219 1220 1221
    return NULL;
  }

dengyihao's avatar
dengyihao 已提交
1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237
  TAOS_STMT *pStmt = stmtInit(pObj, 0);

  releaseTscObj(*(int64_t *)taos);

  return pStmt;
}

TAOS_STMT *taos_stmt_init_with_reqid(TAOS *taos, int64_t reqid) {
  STscObj *pObj = acquireTscObj(*(int64_t *)taos);
  if (NULL == pObj) {
    tscError("invalid parameter for %s", __FUNCTION__);
    terrno = TSDB_CODE_TSC_DISCONNECTED;
    return NULL;
  }

  TAOS_STMT *pStmt = stmtInit(pObj, reqid);
L
Liu Jicong 已提交
1238 1239

  releaseTscObj(*(int64_t *)taos);
1240 1241

  return pStmt;
1242 1243
}

D
stmt  
dapan1121 已提交
1244 1245
int taos_stmt_prepare(TAOS_STMT *stmt, const char *sql, unsigned long length) {
  if (stmt == NULL || sql == NULL) {
D
stmt  
dapan1121 已提交
1246
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1247 1248 1249 1250
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

D
stmt  
dapan1121 已提交
1251
  return stmtPrepare(stmt, sql, length);
1252 1253
}

D
dapan1121 已提交
1254
int taos_stmt_set_tbname_tags(TAOS_STMT *stmt, const char *name, TAOS_MULTI_BIND *tags) {
D
stmt  
dapan1121 已提交
1255
  if (stmt == NULL || name == NULL) {
D
stmt  
dapan1121 已提交
1256
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1257 1258 1259 1260
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

D
stmt  
dapan1121 已提交
1261 1262 1263 1264 1265 1266 1267 1268 1269 1270
  int32_t code = stmtSetTbName(stmt, name);
  if (code) {
    return code;
  }

  if (tags) {
    return stmtSetTbTags(stmt, tags);
  }

  return TSDB_CODE_SUCCESS;
1271 1272
}

D
stmt  
dapan1121 已提交
1273 1274
int taos_stmt_set_tbname(TAOS_STMT *stmt, const char *name) {
  if (stmt == NULL || name == NULL) {
D
stmt  
dapan1121 已提交
1275
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1276
    terrno = TSDB_CODE_INVALID_PARA;
D
stmt  
dapan1121 已提交
1277
    return terrno;
D
stmt  
dapan1121 已提交
1278 1279
  }

D
stmt  
dapan1121 已提交
1280
  return stmtSetTbName(stmt, name);
1281 1282
}

D
dapan1121 已提交
1283 1284 1285 1286 1287 1288 1289 1290 1291 1292
int taos_stmt_set_tags(TAOS_STMT *stmt, TAOS_MULTI_BIND *tags) {
  if (stmt == NULL || tags == NULL) {
    tscError("NULL parameter for %s", __FUNCTION__);
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

  return stmtSetTbTags(stmt, tags);
}

1293
int taos_stmt_set_sub_tbname(TAOS_STMT *stmt, const char *name) { return taos_stmt_set_tbname(stmt, name); }
D
dapan1121 已提交
1294

1295
int taos_stmt_get_tag_fields(TAOS_STMT *stmt, int *fieldNum, TAOS_FIELD_E **fields) {
D
dapan1121 已提交
1296 1297 1298 1299 1300
  if (stmt == NULL || NULL == fieldNum) {
    tscError("NULL parameter for %s", __FUNCTION__);
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }
1301

D
dapan1121 已提交
1302 1303 1304
  return stmtGetTagFields(stmt, fieldNum, fields);
}

1305
int taos_stmt_get_col_fields(TAOS_STMT *stmt, int *fieldNum, TAOS_FIELD_E **fields) {
D
dapan1121 已提交
1306 1307 1308 1309 1310
  if (stmt == NULL || NULL == fieldNum) {
    tscError("NULL parameter for %s", __FUNCTION__);
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }
1311

D
dapan1121 已提交
1312 1313 1314
  return stmtGetColFields(stmt, fieldNum, fields);
}

D
dapan1121 已提交
1315
int taos_stmt_bind_param(TAOS_STMT *stmt, TAOS_MULTI_BIND *bind) {
D
stmt  
dapan1121 已提交
1316
  if (stmt == NULL || bind == NULL) {
D
stmt  
dapan1121 已提交
1317
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1318
    terrno = TSDB_CODE_INVALID_PARA;
D
stmt  
dapan1121 已提交
1319
    return terrno;
D
stmt  
dapan1121 已提交
1320
  }
1321

D
stmt  
dapan1121 已提交
1322 1323 1324 1325 1326
  if (bind->num > 1) {
    tscError("invalid bind number %d for %s", bind->num, __FUNCTION__);
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }
1327

D
stmt  
dapan1121 已提交
1328
  return stmtBindBatch(stmt, bind, -1);
1329 1330
}

D
dapan1121 已提交
1331
int taos_stmt_bind_param_batch(TAOS_STMT *stmt, TAOS_MULTI_BIND *bind) {
D
stmt  
dapan1121 已提交
1332
  if (stmt == NULL || bind == NULL) {
D
stmt  
dapan1121 已提交
1333
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1334 1335 1336 1337
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

D
stmt  
dapan1121 已提交
1338 1339
  if (bind->num <= 0 || bind->num > INT16_MAX) {
    tscError("invalid bind num %d", bind->num);
D
stmt  
dapan1121 已提交
1340 1341 1342 1343
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

D
dapan1121 已提交
1344 1345 1346 1347 1348 1349 1350 1351
  int32_t insert = 0;
  stmtIsInsert(stmt, &insert);
  if (0 == insert && bind->num > 1) {
    tscError("only one row data allowed for query");
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

D
stmt  
dapan1121 已提交
1352
  return stmtBindBatch(stmt, bind, -1);
1353 1354
}

D
dapan1121 已提交
1355
int taos_stmt_bind_single_param_batch(TAOS_STMT *stmt, TAOS_MULTI_BIND *bind, int colIdx) {
D
stmt  
dapan1121 已提交
1356
  if (stmt == NULL || bind == NULL) {
D
stmt  
dapan1121 已提交
1357
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1358 1359 1360 1361
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

D
stmt  
dapan1121 已提交
1362
  if (colIdx < 0) {
D
stmt  
dapan1121 已提交
1363 1364 1365 1366
    tscError("invalid bind column idx %d", colIdx);
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }
D
dapan1121 已提交
1367 1368 1369 1370 1371 1372 1373 1374

  int32_t insert = 0;
  stmtIsInsert(stmt, &insert);
  if (0 == insert && bind->num > 1) {
    tscError("only one row data allowed for query");
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }
1375

D
stmt  
dapan1121 已提交
1376
  return stmtBindBatch(stmt, bind, colIdx);
1377 1378
}

D
stmt  
dapan1121 已提交
1379 1380
int taos_stmt_add_batch(TAOS_STMT *stmt) {
  if (stmt == NULL) {
D
stmt  
dapan1121 已提交
1381
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1382 1383 1384 1385
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

D
stmt  
dapan1121 已提交
1386
  return stmtAddBatch(stmt);
1387 1388
}

D
stmt  
dapan1121 已提交
1389 1390
int taos_stmt_execute(TAOS_STMT *stmt) {
  if (stmt == NULL) {
D
stmt  
dapan1121 已提交
1391
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1392 1393 1394 1395
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

D
stmt  
dapan1121 已提交
1396
  return stmtExec(stmt);
1397 1398
}

1399
int taos_stmt_is_insert(TAOS_STMT *stmt, int *insert) {
D
stmt  
dapan1121 已提交
1400
  if (stmt == NULL || insert == NULL) {
D
stmt  
dapan1121 已提交
1401
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1402 1403 1404 1405 1406
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

  return stmtIsInsert(stmt, insert);
1407 1408
}

1409
int taos_stmt_num_params(TAOS_STMT *stmt, int *nums) {
D
stmt  
dapan1121 已提交
1410
  if (stmt == NULL || nums == NULL) {
D
stmt  
dapan1121 已提交
1411
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1412 1413 1414 1415 1416
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

  return stmtGetParamNum(stmt, nums);
1417 1418
}

D
dapan1121 已提交
1419 1420 1421 1422 1423 1424 1425 1426 1427 1428
int taos_stmt_get_param(TAOS_STMT *stmt, int idx, int *type, int *bytes) {
  if (stmt == NULL || type == NULL || NULL == bytes || idx < 0) {
    tscError("invalid parameter for %s", __FUNCTION__);
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

  return stmtGetParam(stmt, idx, type, bytes);
}

D
stmt  
dapan1121 已提交
1429
TAOS_RES *taos_stmt_use_result(TAOS_STMT *stmt) {
D
stmt  
dapan1121 已提交
1430
  if (stmt == NULL) {
D
stmt  
dapan1121 已提交
1431
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1432
    terrno = TSDB_CODE_INVALID_PARA;
D
stmt  
dapan1121 已提交
1433
    return NULL;
D
stmt  
dapan1121 已提交
1434 1435
  }

D
stmt  
dapan1121 已提交
1436
  return stmtUseResult(stmt);
1437 1438
}

1439
char *taos_stmt_errstr(TAOS_STMT *stmt) { return (char *)stmtErrstr(stmt); }
1440

D
stmt  
dapan1121 已提交
1441
int taos_stmt_affected_rows(TAOS_STMT *stmt) {
D
stmt  
dapan1121 已提交
1442
  if (stmt == NULL) {
D
stmt  
dapan1121 已提交
1443
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1444
    terrno = TSDB_CODE_INVALID_PARA;
D
stmt  
dapan1121 已提交
1445
    return 0;
D
stmt  
dapan1121 已提交
1446 1447
  }

D
stmt  
dapan1121 已提交
1448
  return stmtAffectedRows(stmt);
1449 1450
}

D
dapan1121 已提交
1451 1452 1453 1454 1455 1456 1457 1458 1459 1460
int taos_stmt_affected_rows_once(TAOS_STMT *stmt) {
  if (stmt == NULL) {
    tscError("NULL parameter for %s", __FUNCTION__);
    terrno = TSDB_CODE_INVALID_PARA;
    return 0;
  }

  return stmtAffectedRowsOnce(stmt);
}

D
stmt  
dapan1121 已提交
1461 1462
int taos_stmt_close(TAOS_STMT *stmt) {
  if (stmt == NULL) {
D
stmt  
dapan1121 已提交
1463
    tscError("NULL parameter for %s", __FUNCTION__);
D
stmt  
dapan1121 已提交
1464 1465 1466 1467
    terrno = TSDB_CODE_INVALID_PARA;
    return terrno;
  }

D
stmt  
dapan1121 已提交
1468
  return stmtClose(stmt);
D
stmt  
dapan1121 已提交
1469
}