tmq.c 36.6 KB
Newer Older
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/>.
 */

L
Liu Jicong 已提交
16 17
#define _DEFAULT_SOURCE

18 19 20 21 22
#include "clientInt.h"
#include "clientLog.h"
#include "parser.h"
#include "planner.h"
#include "scheduler.h"
H
Haojun Liao 已提交
23
#include "tdatablock.h"
24 25 26
#include "tdef.h"
#include "tglobal.h"
#include "tmsgtype.h"
H
Haojun Liao 已提交
27
#include "tpagedbuf.h"
L
Liu Jicong 已提交
28
#include "tqueue.h"
29 30
#include "tref.h"

L
Liu Jicong 已提交
31 32 33 34 35
struct tmq_list_t {
  int32_t cnt;
  int32_t tot;
  char*   elems[];
};
L
Liu Jicong 已提交
36

L
Liu Jicong 已提交
37
struct tmq_topic_vgroup_t {
L
Liu Jicong 已提交
38
  SMqOffset offset;
L
Liu Jicong 已提交
39 40 41
};

struct tmq_topic_vgroup_list_t {
L
Liu Jicong 已提交
42 43
  int32_t             cnt;
  int32_t             size;
L
Liu Jicong 已提交
44 45 46 47
  tmq_topic_vgroup_t* elems;
};

struct tmq_conf_t {
L
Liu Jicong 已提交
48
  char           clientId[256];
L
Liu Jicong 已提交
49
  char           groupId[TSDB_CGROUP_LEN];
L
Liu Jicong 已提交
50 51 52
  int8_t         auto_commit;
  int8_t         resetOffset;
  tmq_commit_cb* commit_cb;
L
Liu Jicong 已提交
53 54 55 56 57
  /*char*          ip;*/
  /*uint16_t       port;*/
};

struct tmq_t {
L
Liu Jicong 已提交
58
  // conf
L
Liu Jicong 已提交
59
  char           groupId[TSDB_CGROUP_LEN];
L
Liu Jicong 已提交
60
  char           clientId[256];
L
Liu Jicong 已提交
61
  int8_t         autoCommit;
L
Liu Jicong 已提交
62
  int8_t         inWaiting;
L
Liu Jicong 已提交
63
  int64_t        consumerId;
L
Liu Jicong 已提交
64
  int32_t        epoch;
L
Liu Jicong 已提交
65
  int32_t        resetOffsetCfg;
L
Liu Jicong 已提交
66 67 68 69
  int64_t        status;
  STscObj*       pTscObj;
  tmq_commit_cb* commit_cb;
  int32_t        nextTopicIdx;
L
fix  
Liu Jicong 已提交
70
  int32_t        waitingRequest;
L
Liu Jicong 已提交
71
  int32_t        readyRequest;
L
Liu Jicong 已提交
72
  SArray*        clientTopics;  // SArray<SMqClientTopic>
L
Liu Jicong 已提交
73 74
  STaosQueue*    mqueue;        // queue of tmq_message_t
  STaosQall*     qall;
L
Liu Jicong 已提交
75
  tsem_t         rspSem;
L
Liu Jicong 已提交
76 77
  // stat
  int64_t pollCnt;
L
Liu Jicong 已提交
78 79
};

L
Liu Jicong 已提交
80 81 82 83 84 85 86 87
enum {
  TMQ_VG_STATUS__IDLE = 0,
  TMQ_VG_STATUS__WAIT,
};

enum {
  TMQ_CONSUMER_STATUS__INIT = 0,
  TMQ_CONSUMER_STATUS__READY,
L
Liu Jicong 已提交
88 89
};

L
Liu Jicong 已提交
90
typedef struct {
91 92 93 94
  // statistics
  int64_t pollCnt;
  // offset
  int64_t currentOffset;
L
Liu Jicong 已提交
95
  // connection info
96
  int32_t vgId;
L
Liu Jicong 已提交
97
  int32_t vgStatus;
98 99 100
  SEpSet  epSet;
} SMqClientVg;

L
Liu Jicong 已提交
101
typedef struct {
102 103 104 105 106 107
  // subscribe info
  int32_t sqlLen;
  char*   sql;
  char*   topicName;
  int64_t topicId;
  int32_t nextVgIdx;
L
Liu Jicong 已提交
108
  SArray* vgs;  // SArray<SMqClientVg>
109 110
} SMqClientTopic;

L
Liu Jicong 已提交
111
typedef struct {
L
Liu Jicong 已提交
112 113 114 115
  tmq_t*         tmq;
  tsem_t         rspSem;
  tmq_resp_err_t rspErr;
} SMqSubscribeCbParam;
L
Liu Jicong 已提交
116

L
Liu Jicong 已提交
117
typedef struct {
118
  tmq_t*  tmq;
L
Liu Jicong 已提交
119 120
  int32_t sync;
  tsem_t  rspSem;
121 122
} SMqAskEpCbParam;

L
Liu Jicong 已提交
123
typedef struct {
L
Liu Jicong 已提交
124 125 126 127 128 129
  tmq_t*          tmq;
  SMqClientVg*    pVg;
  int32_t         epoch;
  tsem_t          rspSem;
  tmq_message_t** msg;
  int32_t         sync;
L
Liu Jicong 已提交
130
} SMqPollCbParam;
131

L
Liu Jicong 已提交
132
typedef struct {
L
Liu Jicong 已提交
133 134 135
  tmq_t* tmq;
  /*SMqClientVg* pVg;*/
  int32_t        async;
L
Liu Jicong 已提交
136 137
  tsem_t         rspSem;
  tmq_resp_err_t rspErr;
L
Liu Jicong 已提交
138
} SMqCommitCbParam;
L
Liu Jicong 已提交
139

140 141
tmq_conf_t* tmq_conf_new() {
  tmq_conf_t* conf = calloc(1, sizeof(tmq_conf_t));
L
Liu Jicong 已提交
142
  conf->auto_commit = false;
L
fix  
Liu Jicong 已提交
143
  conf->resetOffset = TMQ_CONF__RESET_OFFSET__EARLIEAST;
144 145 146
  return conf;
}

L
Liu Jicong 已提交
147
void tmq_conf_destroy(tmq_conf_t* conf) {
L
Liu Jicong 已提交
148
  if (conf) free(conf);
L
Liu Jicong 已提交
149 150 151
}

tmq_conf_res_t tmq_conf_set(tmq_conf_t* conf, const char* key, const char* value) {
152 153
  if (strcmp(key, "group.id") == 0) {
    strcpy(conf->groupId, value);
L
Liu Jicong 已提交
154
    return TMQ_CONF_OK;
155 156 157
  }
  if (strcmp(key, "client.id") == 0) {
    strcpy(conf->clientId, value);
L
Liu Jicong 已提交
158 159 160 161 162 163 164 165 166 167 168 169
    return TMQ_CONF_OK;
  }
  if (strcmp(key, "enable.auto.commit") == 0) {
    if (strcmp(value, "true") == 0) {
      conf->auto_commit = true;
      return TMQ_CONF_OK;
    } else if (strcmp(value, "false") == 0) {
      conf->auto_commit = false;
      return TMQ_CONF_OK;
    } else {
      return TMQ_CONF_INVALID;
    }
170
  }
L
Liu Jicong 已提交
171 172 173 174 175 176 177 178 179 180 181 182 183 184
  if (strcmp(key, "auto.offset.reset") == 0) {
    if (strcmp(value, "none") == 0) {
      conf->resetOffset = TMQ_CONF__RESET_OFFSET__NONE;
      return TMQ_CONF_OK;
    } else if (strcmp(value, "earliest") == 0) {
      conf->resetOffset = TMQ_CONF__RESET_OFFSET__EARLIEAST;
      return TMQ_CONF_OK;
    } else if (strcmp(value, "latest") == 0) {
      conf->resetOffset = TMQ_CONF__RESET_OFFSET__LATEST;
      return TMQ_CONF_OK;
    } else {
      return TMQ_CONF_INVALID;
    }
  }
L
Liu Jicong 已提交
185
  return TMQ_CONF_UNKNOWN;
186 187 188
}

tmq_list_t* tmq_list_new() {
L
Liu Jicong 已提交
189
  tmq_list_t* ptr = malloc(sizeof(tmq_list_t) + 8 * sizeof(char*));
190 191 192 193 194 195 196 197
  if (ptr == NULL) {
    return ptr;
  }
  ptr->cnt = 0;
  ptr->tot = 8;
  return ptr;
}

L
Liu Jicong 已提交
198
int32_t tmq_list_append(tmq_list_t* ptr, const char* src) {
L
Liu Jicong 已提交
199
  if (ptr->cnt >= ptr->tot - 1) return -1;
200 201 202 203 204
  ptr->elems[ptr->cnt] = strdup(src);
  ptr->cnt++;
  return 0;
}

L
Liu Jicong 已提交
205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224
void tmqClearUnhandleMsg(tmq_t* tmq) {
  tmq_message_t* msg;
  while (1) {
    taosGetQitem(tmq->qall, (void**)&msg);
    if (msg)
      taosFreeQitem(msg);
    else
      break;
  }

  taosReadAllQitems(tmq->mqueue, tmq->qall);
  while (1) {
    taosGetQitem(tmq->qall, (void**)&msg);
    if (msg)
      taosFreeQitem(msg);
    else
      break;
  }
}

L
Liu Jicong 已提交
225 226 227 228 229 230
int32_t tmqSubscribeCb(void* param, const SDataBuf* pMsg, int32_t code) {
  SMqSubscribeCbParam* pParam = (SMqSubscribeCbParam*)param;
  pParam->rspErr = code;
  tsem_post(&pParam->rspSem);
  return 0;
}
231

L
Liu Jicong 已提交
232
int32_t tmqCommitCb(void* param, const SDataBuf* pMsg, int32_t code) {
L
Liu Jicong 已提交
233
  SMqCommitCbParam* pParam = (SMqCommitCbParam*)param;
L
Liu Jicong 已提交
234
  pParam->rspErr = code == 0 ? TMQ_RESP_ERR__SUCCESS : TMQ_RESP_ERR__FAIL;
L
Liu Jicong 已提交
235
  if (pParam->tmq->commit_cb) {
L
Liu Jicong 已提交
236
    pParam->tmq->commit_cb(pParam->tmq, pParam->rspErr, NULL, NULL);
L
Liu Jicong 已提交
237
  }
L
fix  
Liu Jicong 已提交
238
  if (!pParam->async) tsem_post(&pParam->rspSem);
L
Liu Jicong 已提交
239 240 241
  return 0;
}

L
Liu Jicong 已提交
242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257
tmq_resp_err_t tmq_subscription(tmq_t* tmq, tmq_list_t** topics) {
  if (*topics == NULL) {
    *topics = tmq_list_new();
  }
  for (int i = 0; i < taosArrayGetSize(tmq->clientTopics); i++) {
    SMqClientTopic* topic = taosArrayGetP(tmq->clientTopics, i);
    tmq_list_append(*topics, strdup(topic->topicName));
  }
  return TMQ_RESP_ERR__SUCCESS;
}

tmq_resp_err_t tmq_unsubscribe(tmq_t* tmq) {
  tmq_list_t* lst = tmq_list_new();
  return tmq_subscribe(tmq, lst);
}

258 259 260 261 262 263
tmq_t* tmq_consumer_new(void* conn, tmq_conf_t* conf, char* errstr, int32_t errstrLen) {
  tmq_t* pTmq = calloc(sizeof(tmq_t), 1);
  if (pTmq == NULL) {
    return NULL;
  }
  pTmq->pTscObj = (STscObj*)conn;
L
Liu Jicong 已提交
264
  pTmq->inWaiting = 0;
265 266
  pTmq->status = 0;
  pTmq->pollCnt = 0;
L
Liu Jicong 已提交
267
  pTmq->epoch = 0;
L
fix  
Liu Jicong 已提交
268
  pTmq->waitingRequest = 0;
L
Liu Jicong 已提交
269
  pTmq->readyRequest = 0;
L
Liu Jicong 已提交
270
  // set conf
271 272
  strcpy(pTmq->clientId, conf->clientId);
  strcpy(pTmq->groupId, conf->groupId);
L
Liu Jicong 已提交
273
  pTmq->autoCommit = conf->auto_commit;
274
  pTmq->commit_cb = conf->commit_cb;
L
Liu Jicong 已提交
275
  pTmq->resetOffsetCfg = conf->resetOffset;
L
Liu Jicong 已提交
276

L
Liu Jicong 已提交
277 278
  tsem_init(&pTmq->rspSem, 0, 0);

L
Liu Jicong 已提交
279
  pTmq->consumerId = generateRequestId() & (((uint64_t)-1) >> 1);
280
  pTmq->clientTopics = taosArrayInit(0, sizeof(SMqClientTopic));
L
Liu Jicong 已提交
281 282 283

  pTmq->mqueue = taosOpenQueue();
  pTmq->qall = taosAllocateQall();
284 285 286
  return pTmq;
}

L
Liu Jicong 已提交
287 288 289 290
tmq_resp_err_t tmq_commit(tmq_t* tmq, const tmq_topic_vgroup_list_t* offsets, int32_t async) {
  // TODO: add read write lock
  SRequestObj*   pRequest = NULL;
  tmq_resp_err_t resp = TMQ_RESP_ERR__SUCCESS;
L
Liu Jicong 已提交
291 292
  // build msg
  // send to mnode
L
Liu Jicong 已提交
293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315
  SMqCMCommitOffsetReq req;
  SArray*              pArray = NULL;

  if (offsets == NULL) {
    pArray = taosArrayInit(0, sizeof(SMqOffset));
    for (int i = 0; i < taosArrayGetSize(tmq->clientTopics); i++) {
      SMqClientTopic* pTopic = taosArrayGet(tmq->clientTopics, i);
      for (int j = 0; j < taosArrayGetSize(pTopic->vgs); j++) {
        SMqClientVg* pVg = taosArrayGet(pTopic->vgs, j);
        SMqOffset    offset;
        strcpy(offset.topicName, pTopic->topicName);
        strcpy(offset.cgroup, tmq->groupId);
        offset.vgId = pVg->vgId;
        offset.offset = pVg->currentOffset;
        taosArrayPush(pArray, &offset);
      }
    }
    req.num = pArray->size;
    req.offsets = pArray->pData;
  } else {
    req.num = offsets->cnt;
    req.offsets = (SMqOffset*)offsets->elems;
  }
L
Liu Jicong 已提交
316 317 318 319

  SCoder encoder;

  tCoderInit(&encoder, TD_LITTLE_ENDIAN, NULL, 0, TD_ENCODER);
L
Liu Jicong 已提交
320
  tEncodeSMqCMCommitOffsetReq(&encoder, &req);
L
Liu Jicong 已提交
321 322 323 324 325 326 327 328 329
  int32_t tlen = encoder.pos;
  void*   buf = malloc(tlen);
  if (buf == NULL) {
    tCoderClear(&encoder);
    return -1;
  }
  tCoderClear(&encoder);

  tCoderInit(&encoder, TD_LITTLE_ENDIAN, buf, tlen, TD_ENCODER);
L
Liu Jicong 已提交
330
  tEncodeSMqCMCommitOffsetReq(&encoder, &req);
L
Liu Jicong 已提交
331 332
  tCoderClear(&encoder);

L
Liu Jicong 已提交
333
  pRequest = createRequest(tmq->pTscObj, NULL, NULL, TDMT_MND_MQ_COMMIT_OFFSET);
L
Liu Jicong 已提交
334 335 336 337
  if (pRequest == NULL) {
    tscError("failed to malloc request");
  }

L
Liu Jicong 已提交
338 339 340 341 342 343
  SMqCommitCbParam* pParam = malloc(sizeof(SMqCommitCbParam));
  if (pParam == NULL) {
    return -1;
  }
  pParam->tmq = tmq;
  tsem_init(&pParam->rspSem, 0, 0);
L
fix  
Liu Jicong 已提交
344
  pParam->async = async;
L
Liu Jicong 已提交
345

L
Liu Jicong 已提交
346 347 348 349 350
  pRequest->body.requestMsg = (SDataBuf){
      .pData = buf,
      .len = tlen,
      .handle = NULL,
  };
L
Liu Jicong 已提交
351 352

  SMsgSendInfo* sendInfo = buildMsgInfoImpl(pRequest);
L
Liu Jicong 已提交
353 354
  sendInfo->param = pParam;
  sendInfo->fp = tmqCommitCb;
L
Liu Jicong 已提交
355 356 357 358 359
  SEpSet epSet = getEpSet_s(&tmq->pTscObj->pAppInfo->mgmtEp);

  int64_t transporterId = 0;
  asyncSendMsgToServer(tmq->pTscObj->pAppInfo->pTransporter, &epSet, &transporterId, sendInfo);

L
Liu Jicong 已提交
360 361 362 363
  if (!async) {
    tsem_wait(&pParam->rspSem);
    resp = pParam->rspErr;
  }
L
Liu Jicong 已提交
364

L
fix  
Liu Jicong 已提交
365 366 367
  tsem_destroy(&pParam->rspSem);
  free(pParam);

L
Liu Jicong 已提交
368 369 370 371 372
  if (pArray) {
    taosArrayDestroy(pArray);
  }

  return resp;
L
Liu Jicong 已提交
373 374
}

L
Liu Jicong 已提交
375
tmq_resp_err_t tmq_subscribe(tmq_t* tmq, tmq_list_t* topic_list) {
L
Liu Jicong 已提交
376 377 378
  SRequestObj* pRequest = NULL;
  int32_t      sz = topic_list->cnt;
  // destroy ex
379 380 381 382 383 384 385 386 387 388 389 390 391 392
  taosArrayDestroy(tmq->clientTopics);
  tmq->clientTopics = taosArrayInit(sz, sizeof(SMqClientTopic));

  SCMSubscribeReq req;
  req.topicNum = sz;
  req.consumerId = tmq->consumerId;
  req.consumerGroup = strdup(tmq->groupId);
  req.topicNames = taosArrayInit(sz, sizeof(void*));

  for (int i = 0; i < sz; i++) {
    char* topicName = topic_list->elems[i];

    SName name = {0};
    char* dbName = getDbOfConnection(tmq->pTscObj);
L
Liu Jicong 已提交
393 394 395
    if (dbName == NULL) {
      return TMQ_RESP_ERR__FAIL;
    }
L
Liu Jicong 已提交
396
    tNameSetDbName(&name, tmq->pTscObj->acctId, dbName, strlen(dbName));
397 398 399 400
    tNameFromString(&name, topicName, T_NAME_TABLE);

    char* topicFname = calloc(1, TSDB_TOPIC_FNAME_LEN);
    if (topicFname == NULL) {
L
Liu Jicong 已提交
401
      goto _return;
402 403 404 405
    }
    tNameExtractFullName(&name, topicFname);
    tscDebug("subscribe topic: %s", topicFname);
    SMqClientTopic topic = {
L
Liu Jicong 已提交
406
        .nextVgIdx = 0, .sql = NULL, .sqlLen = 0, .topicId = 0, .topicName = topicFname, .vgs = NULL};
407
    topic.vgs = taosArrayInit(0, sizeof(SMqClientVg));
L
Liu Jicong 已提交
408
    taosArrayPush(tmq->clientTopics, &topic);
409
    taosArrayPush(req.topicNames, &topicFname);
L
Liu Jicong 已提交
410
    free(dbName);
411 412
  }

L
Liu Jicong 已提交
413
  int   tlen = tSerializeSCMSubscribeReq(NULL, &req);
414
  void* buf = malloc(tlen);
L
Liu Jicong 已提交
415
  if (buf == NULL) {
416 417 418 419 420 421 422 423 424
    goto _return;
  }

  void* abuf = buf;
  tSerializeSCMSubscribeReq(&abuf, &req);
  /*printf("formatted: %s\n", dagStr);*/

  pRequest = createRequest(tmq->pTscObj, NULL, NULL, TDMT_MND_SUBSCRIBE);
  if (pRequest == NULL) {
L
Liu Jicong 已提交
425
    tscError("failed to malloc request");
426 427
  }

L
Liu Jicong 已提交
428 429 430 431
  SMqSubscribeCbParam param = {
      .rspErr = TMQ_RESP_ERR__SUCCESS,
      .tmq = tmq,
  };
L
Liu Jicong 已提交
432 433
  tsem_init(&param.rspSem, 0, 0);

L
Liu Jicong 已提交
434 435 436 437 438
  pRequest->body.requestMsg = (SDataBuf){
      .pData = buf,
      .len = tlen,
      .handle = NULL,
  };
439 440

  SMsgSendInfo* sendInfo = buildMsgInfoImpl(pRequest);
L
Liu Jicong 已提交
441 442
  sendInfo->param = &param;
  sendInfo->fp = tmqSubscribeCb;
443 444 445 446 447
  SEpSet epSet = getEpSet_s(&tmq->pTscObj->pAppInfo->mgmtEp);

  int64_t transporterId = 0;
  asyncSendMsgToServer(tmq->pTscObj->pAppInfo->pTransporter, &epSet, &transporterId, sendInfo);

L
Liu Jicong 已提交
448 449
  tsem_wait(&param.rspSem);
  tsem_destroy(&param.rspSem);
450 451 452

_return:
  /*if (sendInfo != NULL) {*/
L
Liu Jicong 已提交
453
  /*destroySendMsgInfo(sendInfo);*/
454 455
  /*}*/

L
Liu Jicong 已提交
456
  return param.rspErr;
457 458
}

L
Liu Jicong 已提交
459
void tmq_conf_set_offset_commit_cb(tmq_conf_t* conf, tmq_commit_cb* cb) { conf->commit_cb = cb; }
460

L
Liu Jicong 已提交
461 462 463 464 465
TAOS_RES* tmq_create_topic(TAOS* taos, const char* topicName, const char* sql, int sqlLen) {
  STscObj*     pTscObj = (STscObj*)taos;
  SRequestObj* pRequest = NULL;
  SQueryNode*  pQueryNode = NULL;
  char*        pStr = NULL;
466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490

  terrno = TSDB_CODE_SUCCESS;
  if (taos == NULL || topicName == NULL || sql == NULL) {
    tscError("invalid parameters for creating topic, connObj:%p, topic name:%s, sql:%s", taos, topicName, sql);
    terrno = TSDB_CODE_TSC_INVALID_INPUT;
    goto _return;
  }

  if (strlen(topicName) >= TSDB_TOPIC_NAME_LEN) {
    tscError("topic name too long, max length:%d", TSDB_TOPIC_NAME_LEN - 1);
    terrno = TSDB_CODE_TSC_INVALID_INPUT;
    goto _return;
  }

  if (sqlLen > TSDB_MAX_ALLOWED_SQL_LEN) {
    tscError("sql string exceeds max length:%d", TSDB_MAX_ALLOWED_SQL_LEN);
    terrno = TSDB_CODE_TSC_EXCEED_SQL_LIMIT;
    goto _return;
  }

  tscDebug("start to create topic, %s", topicName);

  CHECK_CODE_GOTO(buildRequest(pTscObj, sql, sqlLen, &pRequest), _return);
  CHECK_CODE_GOTO(parseSql(pRequest, &pQueryNode), _return);

L
Liu Jicong 已提交
491
  SQueryStmtInfo* pQueryStmtInfo = (SQueryStmtInfo*)pQueryNode;
492 493 494 495
  pQueryStmtInfo->info.continueQuery = true;

  // todo check for invalid sql statement and return with error code

L
Liu Jicong 已提交
496
  SSchema* schema = NULL;
497
  int32_t  numOfCols = 0;
L
Liu Jicong 已提交
498 499
  CHECK_CODE_GOTO(qCreateQueryDag(pQueryNode, &pRequest->body.pDag, &schema, &numOfCols, NULL, pRequest->requestId),
                  _return);
500 501

  pStr = qDagToString(pRequest->body.pDag);
L
Liu Jicong 已提交
502
  if (pStr == NULL) {
503 504 505
    goto _return;
  }

L
Liu Jicong 已提交
506
  /*printf("%s\n", pStr);*/
507 508 509

  // The topic should be related to a database that the queried table is belonged to.
  SName name = {0};
L
Liu Jicong 已提交
510 511
  char  dbName[TSDB_DB_FNAME_LEN] = {0};
  tNameGetFullDbName(&((SQueryStmtInfo*)pQueryNode)->pTableMetaInfo[0]->name, dbName);
512

L
Liu Jicong 已提交
513
  tNameFromString(&name, dbName, T_NAME_ACCT | T_NAME_DB);
514 515
  tNameFromString(&name, topicName, T_NAME_TABLE);

S
Shengliang Guan 已提交
516
  SMCreateTopicReq req = {
L
Liu Jicong 已提交
517 518 519
      .igExists = 1,
      .physicalPlan = (char*)pStr,
      .sql = (char*)sql,
L
Liu Jicong 已提交
520
      .logicalPlan = (char*)"no logic plan",
521
  };
L
Liu Jicong 已提交
522
  tNameExtractFullName(&name, req.name);
523

S
Shengliang Guan 已提交
524
  int   tlen = tSerializeMCreateTopicReq(NULL, 0, &req);
525 526 527 528 529
  void* buf = malloc(tlen);
  if (buf == NULL) {
    goto _return;
  }

S
Shengliang Guan 已提交
530
  tSerializeMCreateTopicReq(buf, tlen, &req);
531 532
  /*printf("formatted: %s\n", dagStr);*/

dengyihao's avatar
dengyihao 已提交
533
  pRequest->body.requestMsg = (SDataBuf){.pData = buf, .len = tlen, .handle = NULL};
534 535 536
  pRequest->type = TDMT_MND_CREATE_TOPIC;

  SMsgSendInfo* sendInfo = buildMsgInfoImpl(pRequest);
L
Liu Jicong 已提交
537
  SEpSet        epSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
538 539 540 541 542 543 544 545 546

  int64_t transporterId = 0;
  asyncSendMsgToServer(pTscObj->pAppInfo->pTransporter, &epSet, &transporterId, sendInfo);

  tsem_wait(&pRequest->body.rspSem);

_return:
  qDestroyQuery(pQueryNode);
  /*if (sendInfo != NULL) {*/
L
Liu Jicong 已提交
547
  /*destroySendMsgInfo(sendInfo);*/
548 549 550 551 552 553 554 555 556
  /*}*/

  if (pRequest != NULL && terrno != TSDB_CODE_SUCCESS) {
    pRequest->code = terrno;
  }

  return pRequest;
}

L
Liu Jicong 已提交
557
static char* formatTimestamp(char* buf, int64_t val, int precision) {
558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592
  time_t  tt;
  int32_t ms = 0;
  if (precision == TSDB_TIME_PRECISION_NANO) {
    tt = (time_t)(val / 1000000000);
    ms = val % 1000000000;
  } else if (precision == TSDB_TIME_PRECISION_MICRO) {
    tt = (time_t)(val / 1000000);
    ms = val % 1000000;
  } else {
    tt = (time_t)(val / 1000);
    ms = val % 1000;
  }

  /* comment out as it make testcases like select_with_tags.sim fail.
    but in windows, this may cause the call to localtime crash if tt < 0,
    need to find a better solution.
    if (tt < 0) {
      tt = 0;
    }
    */

#ifdef WINDOWS
  if (tt < 0) tt = 0;
#endif
  if (tt <= 0 && ms < 0) {
    tt--;
    if (precision == TSDB_TIME_PRECISION_NANO) {
      ms += 1000000000;
    } else if (precision == TSDB_TIME_PRECISION_MICRO) {
      ms += 1000000;
    } else {
      ms += 1000;
    }
  }

L
Liu Jicong 已提交
593
  struct tm* ptm = localtime(&tt);
594 595 596 597 598 599 600 601 602 603 604 605 606
  size_t     pos = strftime(buf, 35, "%Y-%m-%d %H:%M:%S", ptm);

  if (precision == TSDB_TIME_PRECISION_NANO) {
    sprintf(buf + pos, ".%09d", ms);
  } else if (precision == TSDB_TIME_PRECISION_MICRO) {
    sprintf(buf + pos, ".%06d", ms);
  } else {
    sprintf(buf + pos, ".%03d", ms);
  }

  return buf;
}

L
Liu Jicong 已提交
607 608
int32_t tmqGetSkipLogNum(tmq_message_t* tmq_message) {
  if (tmq_message == NULL) return 0;
L
fix  
Liu Jicong 已提交
609
  SMqConsumeRsp* pRsp = &tmq_message->consumeRsp;
L
Liu Jicong 已提交
610 611 612
  return pRsp->skipLogNum;
}

L
Liu Jicong 已提交
613 614 615
void tmqShowMsg(tmq_message_t* tmq_message) {
  if (tmq_message == NULL) return;

L
Liu Jicong 已提交
616 617
  static bool    noPrintSchema;
  char           pBuf[128];
L
Liu Jicong 已提交
618
  SMqConsumeRsp* pRsp = &tmq_message->consumeRsp;
L
Liu Jicong 已提交
619
  int32_t        colNum = pRsp->schemas->nCols;
L
Liu Jicong 已提交
620 621 622 623 624 625 626 627 628 629 630
  if (!noPrintSchema) {
    printf("|");
    for (int32_t i = 0; i < colNum; i++) {
      if (i == 0)
        printf(" %25s |", pRsp->schemas->pSchema[i].name);
      else
        printf(" %15s |", pRsp->schemas->pSchema[i].name);
    }
    printf("\n");
    printf("===============================================\n");
    noPrintSchema = true;
631
  }
L
Liu Jicong 已提交
632
  int32_t sz = taosArrayGetSize(pRsp->pBlockData);
633
  for (int32_t i = 0; i < sz; i++) {
L
Liu Jicong 已提交
634 635
    SSDataBlock* pDataBlock = taosArrayGet(pRsp->pBlockData, i);
    int32_t      rows = pDataBlock->info.rows;
636 637 638 639
    for (int32_t j = 0; j < rows; j++) {
      printf("|");
      for (int32_t k = 0; k < colNum; k++) {
        SColumnInfoData* pColInfoData = taosArrayGet(pDataBlock->pDataBlock, k);
L
Liu Jicong 已提交
640 641
        void*            var = POINTER_SHIFT(pColInfoData->pData, j * pColInfoData->info.bytes);
        switch (pColInfoData->info.type) {
642 643 644 645 646 647 648 649 650 651 652 653 654
          case TSDB_DATA_TYPE_TIMESTAMP:
            formatTimestamp(pBuf, *(uint64_t*)var, TSDB_TIME_PRECISION_MILLI);
            printf(" %25s |", pBuf);
            break;
          case TSDB_DATA_TYPE_INT:
          case TSDB_DATA_TYPE_UINT:
            printf(" %15u |", *(uint32_t*)var);
            break;
        }
      }
      printf("\n");
    }
  }
L
Liu Jicong 已提交
655 656 657
}

int32_t tmqPollCb(void* param, const SDataBuf* pMsg, int32_t code) {
L
fix  
Liu Jicong 已提交
658
  /*printf("recv poll\n");*/
L
Liu Jicong 已提交
659 660
  SMqPollCbParam* pParam = (SMqPollCbParam*)param;
  SMqClientVg*    pVg = pParam->pVg;
L
Liu Jicong 已提交
661
  tmq_t*          tmq = pParam->tmq;
L
Liu Jicong 已提交
662
  if (code != 0) {
663
    printf("msg discard\n");
L
Liu Jicong 已提交
664
    goto WRITE_QUEUE_FAIL;
L
Liu Jicong 已提交
665 666
  }

L
Liu Jicong 已提交
667 668 669
  int32_t msgEpoch = ((SMqRspHead*)pMsg->pData)->epoch;
  int32_t tmqEpoch = atomic_load_32(&tmq->epoch);
  if (msgEpoch < tmqEpoch) {
L
Liu Jicong 已提交
670
    tsem_post(&tmq->rspSem);
L
Liu Jicong 已提交
671 672 673 674 675 676
    printf("discard rsp epoch %d, current epoch %d\n", msgEpoch, tmqEpoch);
    return 0;
  }

  if (msgEpoch != tmqEpoch) {
    printf("mismatch rsp epoch %d, current epoch %d\n", msgEpoch, tmqEpoch);
L
fix  
Liu Jicong 已提交
677 678
  } else {
    atomic_sub_fetch_32(&tmq->waitingRequest, 1);
L
Liu Jicong 已提交
679
  }
L
Liu Jicong 已提交
680 681

#if 0
L
Liu Jicong 已提交
682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697
  if (pParam->sync == 1) {
    /**pParam->msg = malloc(sizeof(tmq_message_t));*/
    *pParam->msg = taosAllocateQitem(sizeof(tmq_message_t));
    if (*pParam->msg) {
      memcpy(*pParam->msg, pMsg->pData, sizeof(SMqRspHead));
      tDecodeSMqConsumeRsp(POINTER_SHIFT(pMsg->pData, sizeof(SMqRspHead)), &((*pParam->msg)->consumeRsp));
      if ((*pParam->msg)->consumeRsp.numOfTopics != 0) {
        pVg->currentOffset = (*pParam->msg)->consumeRsp.rspOffset;
      }
      taosWriteQitem(tmq->mqueue, *pParam->msg);
      tsem_post(&pParam->rspSem);
      return 0;
    }
    tsem_post(&pParam->rspSem);
    return -1;
  }
L
Liu Jicong 已提交
698
#endif
L
Liu Jicong 已提交
699 700 701

  /*SMqConsumeRsp* pRsp = calloc(1, sizeof(SMqConsumeRsp));*/
  tmq_message_t* pRsp = taosAllocateQitem(sizeof(tmq_message_t));
L
Liu Jicong 已提交
702
  if (pRsp == NULL) {
L
Liu Jicong 已提交
703
    goto WRITE_QUEUE_FAIL;
L
Liu Jicong 已提交
704
  }
L
Liu Jicong 已提交
705 706
  memcpy(pRsp, pMsg->pData, sizeof(SMqRspHead));
  tDecodeSMqConsumeRsp(POINTER_SHIFT(pMsg->pData, sizeof(SMqRspHead)), &pRsp->consumeRsp);
707
  /*printf("rsp commit off:%ld rsp off:%ld has data:%d\n", pRsp->committedOffset, pRsp->rspOffset, pRsp->numOfTopics);*/
L
Liu Jicong 已提交
708
  if (pRsp->consumeRsp.numOfTopics == 0) {
L
fix  
Liu Jicong 已提交
709
    /*printf("no data\n");*/
L
Liu Jicong 已提交
710
    taosFreeQitem(pRsp);
L
Liu Jicong 已提交
711
    goto WRITE_QUEUE_FAIL;
L
Liu Jicong 已提交
712
  }
L
Liu Jicong 已提交
713

L
Liu Jicong 已提交
714 715
  pRsp->extra = pParam->pVg;
  taosWriteQitem(tmq->mqueue, pRsp);
L
Liu Jicong 已提交
716
  atomic_add_fetch_32(&tmq->readyRequest, 1);
L
Liu Jicong 已提交
717
  tsem_post(&tmq->rspSem);
718
  return 0;
L
Liu Jicong 已提交
719 720 721 722 723 724 725

WRITE_QUEUE_FAIL:
  if (pParam->epoch == tmq->epoch) {
    atomic_store_32(&pVg->vgStatus, TMQ_VG_STATUS__IDLE);
  }
  tsem_post(&tmq->rspSem);
  return code;
726 727
}

L
Liu Jicong 已提交
728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756
bool tmqUpdateEp(tmq_t* tmq, int32_t epoch, SMqCMGetSubEpRsp* pRsp) {
  bool    set = false;
  int32_t sz = taosArrayGetSize(pRsp->topics);
  if (tmq->clientTopics) taosArrayDestroy(tmq->clientTopics);
  tmq->clientTopics = taosArrayInit(sz, sizeof(SMqClientTopic));
  for (int32_t i = 0; i < sz; i++) {
    SMqClientTopic topic = {0};
    SMqSubTopicEp* pTopicEp = taosArrayGet(pRsp->topics, i);
    topic.topicName = strdup(pTopicEp->topic);
    int32_t vgSz = taosArrayGetSize(pTopicEp->vgs);
    topic.vgs = taosArrayInit(vgSz, sizeof(SMqClientVg));
    for (int32_t j = 0; j < vgSz; j++) {
      SMqSubVgEp* pVgEp = taosArrayGet(pTopicEp->vgs, j);
      SMqClientVg clientVg = {
          .pollCnt = 0,
          .currentOffset = pVgEp->offset,
          .vgId = pVgEp->vgId,
          .epSet = pVgEp->epSet,
          .vgStatus = TMQ_VG_STATUS__IDLE,
      };
      taosArrayPush(topic.vgs, &clientVg);
      set = true;
    }
    taosArrayPush(tmq->clientTopics, &topic);
  }
  atomic_store_32(&tmq->epoch, epoch);
  return set;
}

L
Liu Jicong 已提交
757
int32_t tmqAskEpCb(void* param, const SDataBuf* pMsg, int32_t code) {
758
  SMqAskEpCbParam* pParam = (SMqAskEpCbParam*)param;
L
Liu Jicong 已提交
759
  tmq_t*           tmq = pParam->tmq;
760
  if (code != 0) {
L
Liu Jicong 已提交
761
    printf("get topic endpoint error, not ready, wait:%d\n", pParam->sync);
L
Liu Jicong 已提交
762
    goto END;
763
  }
L
Liu Jicong 已提交
764 765 766 767 768 769 770 771 772 773

  // tmq's epoch is monotomically increase,
  // so it's safe to discard any old epoch msg.
  // epoch will only increase when received newer epoch ep msg
  SMqRspHead* head = pMsg->pData;
  int32_t     epoch = atomic_load_32(&tmq->epoch);
  if (head->epoch <= epoch) {
    goto END;
  }

L
Liu Jicong 已提交
774 775 776 777 778
  if (pParam->sync) {
    SMqCMGetSubEpRsp rsp;
    tDecodeSMqCMGetSubEpRsp(POINTER_SHIFT(pMsg->pData, sizeof(SMqRspHead)), &rsp);
    /*printf("rsp epoch %ld sz %ld\n", rsp.epoch, rsp.topics->size);*/
    /*printf("tmq epoch %ld sz %ld\n", tmq->epoch, tmq->clientTopics->size);*/
L
Liu Jicong 已提交
779
    if (tmqUpdateEp(tmq, head->epoch, &rsp)) {
L
Liu Jicong 已提交
780
      atomic_store_64(&tmq->status, TMQ_CONSUMER_STATUS__READY);
781
    }
L
Liu Jicong 已提交
782 783 784
    tDeleteSMqCMGetSubEpRsp(&rsp);
  } else {
    tmq_message_t* pRsp = taosAllocateQitem(sizeof(tmq_message_t));
L
Liu Jicong 已提交
785 786
    if (pRsp == NULL) {
      terrno = TSDB_CODE_OUT_OF_MEMORY;
L
Liu Jicong 已提交
787 788
      code = -1;
      goto END;
L
Liu Jicong 已提交
789
    }
L
Liu Jicong 已提交
790
    memcpy(pRsp, pMsg->pData, sizeof(SMqRspHead));
L
fix  
Liu Jicong 已提交
791
    tDecodeSMqCMGetSubEpRsp(POINTER_SHIFT(pMsg->pData, sizeof(SMqRspHead)), &pRsp->getEpRsp);
L
Liu Jicong 已提交
792

L
Liu Jicong 已提交
793 794
    taosWriteQitem(tmq->mqueue, pRsp);
    tsem_post(&tmq->rspSem);
795
  }
L
Liu Jicong 已提交
796 797 798 799 800 801

END:
  if (pParam->sync) {
    tsem_post(&pParam->rspSem);
  }
  return code;
802 803
}

L
Liu Jicong 已提交
804
int32_t tmqAskEp(tmq_t* tmq, bool sync) {
L
Liu Jicong 已提交
805
  int32_t           tlen = sizeof(SMqCMGetSubEpReq);
L
Liu Jicong 已提交
806 807
  SMqCMGetSubEpReq* req = malloc(tlen);
  if (req == NULL) {
L
Liu Jicong 已提交
808
    tscError("failed to malloc get subscribe ep buf");
L
Liu Jicong 已提交
809
    return -1;
L
Liu Jicong 已提交
810
  }
L
Liu Jicong 已提交
811 812 813
  req->consumerId = htobe64(tmq->consumerId);
  req->epoch = htonl(tmq->epoch);
  strcpy(req->cgroup, tmq->groupId);
814

L
Liu Jicong 已提交
815
  SMqAskEpCbParam* pParam = malloc(sizeof(SMqAskEpCbParam));
L
Liu Jicong 已提交
816 817
  if (pParam == NULL) {
    tscError("failed to malloc subscribe param");
L
Liu Jicong 已提交
818 819
    free(req);
    return -1;
L
Liu Jicong 已提交
820 821
  }
  pParam->tmq = tmq;
L
Liu Jicong 已提交
822 823
  pParam->sync = sync;
  tsem_init(&pParam->rspSem, 0, 0);
824

L
Liu Jicong 已提交
825 826 827 828 829 830 831 832 833 834 835 836 837 838 839
  SMsgSendInfo* sendInfo = malloc(sizeof(SMsgSendInfo));
  if (sendInfo == NULL) {
    tsem_destroy(&pParam->rspSem);
    free(pParam);
    free(req);
    return -1;
  }

  sendInfo->msgInfo = (SDataBuf){
      .pData = req,
      .len = tlen,
      .handle = NULL,
  };

  sendInfo->requestId = generateRequestId();
L
Liu Jicong 已提交
840 841 842
  sendInfo->requestObjRefId = 0;
  sendInfo->param = pParam;
  sendInfo->fp = tmqAskEpCb;
L
Liu Jicong 已提交
843
  sendInfo->msgType = TDMT_MND_GET_SUB_EP;
844

L
Liu Jicong 已提交
845 846 847 848
  SEpSet epSet = getEpSet_s(&tmq->pTscObj->pAppInfo->mgmtEp);

  int64_t transporterId = 0;
  asyncSendMsgToServer(tmq->pTscObj->pAppInfo->pTransporter, &epSet, &transporterId, sendInfo);
849

L
Liu Jicong 已提交
850
  if (sync) tsem_wait(&pParam->rspSem);
L
Liu Jicong 已提交
851
  return 0;
852 853
}

L
Liu Jicong 已提交
854 855 856 857 858 859 860 861 862 863 864 865 866 867
tmq_resp_err_t tmq_seek(tmq_t* tmq, const tmq_topic_vgroup_t* offset) {
  const SMqOffset* pOffset = &offset->offset;
  if (strcmp(pOffset->cgroup, tmq->groupId) != 0) {
    return TMQ_RESP_ERR__FAIL;
  }
  int32_t sz = taosArrayGetSize(tmq->clientTopics);
  for (int32_t i = 0; i < sz; i++) {
    SMqClientTopic* clientTopic = taosArrayGet(tmq->clientTopics, i);
    if (strcmp(clientTopic->topicName, pOffset->topicName) == 0) {
      int32_t vgSz = taosArrayGetSize(clientTopic->vgs);
      for (int32_t j = 0; j < vgSz; j++) {
        SMqClientVg* pVg = taosArrayGet(clientTopic->vgs, j);
        if (pVg->vgId == pOffset->vgId) {
          pVg->currentOffset = pOffset->offset;
L
Liu Jicong 已提交
868
          tmqClearUnhandleMsg(tmq);
L
Liu Jicong 已提交
869 870 871 872 873 874 875 876
          return TMQ_RESP_ERR__SUCCESS;
        }
      }
    }
  }
  return TMQ_RESP_ERR__FAIL;
}

L
fix  
Liu Jicong 已提交
877
SMqConsumeReq* tmqBuildConsumeReqImpl(tmq_t* tmq, int64_t blockingTime, SMqClientTopic* pTopic, SMqClientVg* pVg) {
L
Liu Jicong 已提交
878 879 880 881 882 883 884 885 886 887 888
  int64_t reqOffset;
  if (pVg->currentOffset >= 0) {
    reqOffset = pVg->currentOffset;
  } else {
    if (tmq->resetOffsetCfg == TMQ_CONF__RESET_OFFSET__NONE) {
      tscError("unable to poll since no committed offset but reset offset is set to none");
      return NULL;
    }
    reqOffset = tmq->resetOffsetCfg;
  }

889 890 891 892
  SMqConsumeReq* pReq = malloc(sizeof(SMqConsumeReq));
  if (pReq == NULL) {
    return NULL;
  }
L
Liu Jicong 已提交
893

L
Liu Jicong 已提交
894
  strcpy(pReq->topic, pTopic->topicName);
895 896
  strcpy(pReq->cgroup, tmq->groupId);

L
fix  
Liu Jicong 已提交
897
  pReq->blockingTime = blockingTime;
L
Liu Jicong 已提交
898
  pReq->consumerId = tmq->consumerId;
L
Liu Jicong 已提交
899
  pReq->epoch = tmq->epoch;
L
Liu Jicong 已提交
900
  pReq->currentOffset = reqOffset;
901 902 903 904 905 906

  pReq->head.vgId = htonl(pVg->vgId);
  pReq->head.contLen = htonl(sizeof(SMqConsumeReq));
  return pReq;
}

L
Liu Jicong 已提交
907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922
tmq_message_t* tmqSyncPollImpl(tmq_t* tmq, int64_t blockingTime) {
  tmq_message_t* msg = NULL;
  for (int i = 0; i < taosArrayGetSize(tmq->clientTopics); i++) {
    SMqClientTopic* pTopic = taosArrayGet(tmq->clientTopics, i);
    for (int j = 0; j < taosArrayGetSize(pTopic->vgs); j++) {
      SMqClientVg* pVg = taosArrayGet(pTopic->vgs, j);
      int32_t      vgStatus = atomic_val_compare_exchange_32(&pVg->vgStatus, TMQ_VG_STATUS__IDLE, TMQ_VG_STATUS__WAIT);
      /*if (vgStatus != TMQ_VG_STATUS__IDLE) {*/
      /*continue;*/
      /*}*/
      SMqConsumeReq* pReq = tmqBuildConsumeReqImpl(tmq, blockingTime, pTopic, pVg);
      if (pReq == NULL) {
        atomic_store_32(&pVg->vgStatus, TMQ_VG_STATUS__IDLE);
        // TODO: out of mem
        return NULL;
      }
L
Liu Jicong 已提交
923 924 925

      SMqPollCbParam* pParam = malloc(sizeof(SMqPollCbParam));
      if (pParam == NULL) {
L
Liu Jicong 已提交
926 927 928 929
        atomic_store_32(&pVg->vgStatus, TMQ_VG_STATUS__IDLE);
        // TODO: out of mem
        return NULL;
      }
L
Liu Jicong 已提交
930 931 932 933 934 935 936 937 938 939 940 941 942
      pParam->tmq = tmq;
      pParam->pVg = pVg;
      pParam->epoch = tmq->epoch;
      pParam->sync = 1;
      pParam->msg = &msg;
      tsem_init(&pParam->rspSem, 0, 0);

      SMsgSendInfo* sendInfo = malloc(sizeof(SMsgSendInfo));
      if (sendInfo == NULL) {
        return NULL;
      }

      sendInfo->msgInfo = (SDataBuf){
L
Liu Jicong 已提交
943 944 945 946
          .pData = pReq,
          .len = sizeof(SMqConsumeReq),
          .handle = NULL,
      };
L
Liu Jicong 已提交
947
      sendInfo->requestId = generateRequestId();
L
Liu Jicong 已提交
948
      sendInfo->requestObjRefId = 0;
L
Liu Jicong 已提交
949
      sendInfo->param = pParam;
L
Liu Jicong 已提交
950
      sendInfo->fp = tmqPollCb;
L
Liu Jicong 已提交
951
      sendInfo->msgType = TDMT_VND_CONSUME;
L
Liu Jicong 已提交
952 953 954 955 956 957 958 959

      int64_t transporterId = 0;
      /*printf("send poll\n");*/
      atomic_add_fetch_32(&tmq->waitingRequest, 1);
      asyncSendMsgToServer(tmq->pTscObj->pAppInfo->pTransporter, &pVg->epSet, &transporterId, sendInfo);
      pVg->pollCnt++;
      tmq->pollCnt++;

L
Liu Jicong 已提交
960
      tsem_wait(&pParam->rspSem);
L
Liu Jicong 已提交
961 962 963 964
      tmq_message_t* nmsg = NULL;
      while (1) {
        taosReadQitem(tmq->mqueue, (void**)&nmsg);
        if (nmsg == NULL) continue;
L
Liu Jicong 已提交
965 966 967
        while (nmsg->head.mqMsgType != TMQ_MSG_TYPE__POLL_RSP) {
          taosReadQitem(tmq->mqueue, (void**)&nmsg);
        }
L
Liu Jicong 已提交
968 969 970 971 972 973 974
        return nmsg;
      }
    }
  }
  return NULL;
}

L
Liu Jicong 已提交
975
int32_t tmqPollImpl(tmq_t* tmq, int64_t blockingTime) {
L
fix  
Liu Jicong 已提交
976
  /*printf("call poll\n");*/
L
Liu Jicong 已提交
977 978 979 980 981 982 983 984 985 986 987
  for (int i = 0; i < taosArrayGetSize(tmq->clientTopics); i++) {
    SMqClientTopic* pTopic = taosArrayGet(tmq->clientTopics, i);
    for (int j = 0; j < taosArrayGetSize(pTopic->vgs); j++) {
      SMqClientVg* pVg = taosArrayGet(pTopic->vgs, j);
      int32_t      vgStatus = atomic_val_compare_exchange_32(&pVg->vgStatus, TMQ_VG_STATUS__IDLE, TMQ_VG_STATUS__WAIT);
      if (vgStatus != TMQ_VG_STATUS__IDLE) {
        continue;
      }
      SMqConsumeReq* pReq = tmqBuildConsumeReqImpl(tmq, blockingTime, pTopic, pVg);
      if (pReq == NULL) {
        atomic_store_32(&pVg->vgStatus, TMQ_VG_STATUS__IDLE);
L
Liu Jicong 已提交
988
        tsem_post(&tmq->rspSem);
L
Liu Jicong 已提交
989 990
        return -1;
      }
L
Liu Jicong 已提交
991 992 993
      SMqPollCbParam* pParam = malloc(sizeof(SMqPollCbParam));
      if (pParam == NULL) {
        free(pReq);
L
Liu Jicong 已提交
994
        atomic_store_32(&pVg->vgStatus, TMQ_VG_STATUS__IDLE);
L
Liu Jicong 已提交
995
        tsem_post(&tmq->rspSem);
L
Liu Jicong 已提交
996 997
        return -1;
      }
L
Liu Jicong 已提交
998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012
      pParam->tmq = tmq;
      pParam->pVg = pVg;
      pParam->epoch = tmq->epoch;
      pParam->sync = 0;

      SMsgSendInfo* sendInfo = malloc(sizeof(SMsgSendInfo));
      if (sendInfo == NULL) {
        free(pReq);
        free(pParam);
        atomic_store_32(&pVg->vgStatus, TMQ_VG_STATUS__IDLE);
        tsem_post(&tmq->rspSem);
        return -1;
      }

      sendInfo->msgInfo = (SDataBuf){
L
Liu Jicong 已提交
1013 1014 1015 1016
          .pData = pReq,
          .len = sizeof(SMqConsumeReq),
          .handle = NULL,
      };
L
Liu Jicong 已提交
1017
      sendInfo->requestId = generateRequestId();
L
Liu Jicong 已提交
1018
      sendInfo->requestObjRefId = 0;
L
Liu Jicong 已提交
1019
      sendInfo->param = pParam;
L
Liu Jicong 已提交
1020
      sendInfo->fp = tmqPollCb;
L
Liu Jicong 已提交
1021
      sendInfo->msgType = TDMT_VND_CONSUME;
L
Liu Jicong 已提交
1022 1023

      int64_t transporterId = 0;
L
fix  
Liu Jicong 已提交
1024 1025
      /*printf("send poll\n");*/
      atomic_add_fetch_32(&tmq->waitingRequest, 1);
L
Liu Jicong 已提交
1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036
      asyncSendMsgToServer(tmq->pTscObj->pAppInfo->pTransporter, &pVg->epSet, &transporterId, sendInfo);
      pVg->pollCnt++;
      tmq->pollCnt++;
    }
  }
  return 0;
}

// return
int32_t tmqHandleRes(tmq_t* tmq, tmq_message_t* rspMsg, bool* pReset) {
  if (rspMsg->head.mqMsgType == TMQ_MSG_TYPE__EP_RSP) {
L
fix  
Liu Jicong 已提交
1037
    /*printf("ep %d %d\n", rspMsg->head.epoch, tmq->epoch);*/
L
Liu Jicong 已提交
1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050
    if (rspMsg->head.epoch > atomic_load_32(&tmq->epoch)) {
      tmqUpdateEp(tmq, rspMsg->head.epoch, &rspMsg->getEpRsp);
      tmqClearUnhandleMsg(tmq);
      *pReset = true;
    } else {
      *pReset = false;
    }
  } else {
    return -1;
  }
  return 0;
}

L
Liu Jicong 已提交
1051 1052 1053 1054 1055
tmq_message_t* tmqHandleAllRsp(tmq_t* tmq, int64_t blockingTime, bool pollIfReset) {
  while (1) {
    tmq_message_t* rspMsg = NULL;
    taosGetQitem(tmq->qall, (void**)&rspMsg);
    if (rspMsg == NULL) {
L
Liu Jicong 已提交
1056 1057 1058
      taosReadAllQitems(tmq->mqueue, tmq->qall);
      taosGetQitem(tmq->qall, (void**)&rspMsg);
      if (rspMsg == NULL) return NULL;
L
Liu Jicong 已提交
1059 1060 1061
    }

    if (rspMsg->head.mqMsgType == TMQ_MSG_TYPE__POLL_RSP) {
L
Liu Jicong 已提交
1062
      atomic_sub_fetch_32(&tmq->readyRequest, 1);
L
fix  
Liu Jicong 已提交
1063
      /*printf("handle poll rsp %d\n", rspMsg->head.mqMsgType);*/
L
Liu Jicong 已提交
1064
      if (rspMsg->head.epoch == atomic_load_32(&tmq->epoch)) {
L
fix  
Liu Jicong 已提交
1065
        /*printf("epoch match\n");*/
L
Liu Jicong 已提交
1066 1067 1068 1069 1070
        SMqClientVg* pVg = rspMsg->extra;
        pVg->currentOffset = rspMsg->consumeRsp.rspOffset;
        atomic_store_32(&pVg->vgStatus, TMQ_VG_STATUS__IDLE);
        return rspMsg;
      } else {
L
Liu Jicong 已提交
1071
        /*printf("epoch mismatch\n");*/
L
Liu Jicong 已提交
1072 1073 1074
        taosFreeQitem(rspMsg);
      }
    } else {
L
fix  
Liu Jicong 已提交
1075
      /*printf("handle ep rsp %d\n", rspMsg->head.mqMsgType);*/
L
Liu Jicong 已提交
1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086
      bool reset = false;
      tmqHandleRes(tmq, rspMsg, &reset);
      taosFreeQitem(rspMsg);
      if (pollIfReset && reset) {
        printf("reset and repoll\n");
        tmqPollImpl(tmq, blockingTime);
      }
    }
  }
}

L
Liu Jicong 已提交
1087
tmq_message_t* tmq_consumer_poll_v1(tmq_t* tmq, int64_t blocking_time) {
L
Liu Jicong 已提交
1088 1089
  tmq_message_t* rspMsg = NULL;
  int64_t        startTime = taosGetTimestampMs();
1090

L
Liu Jicong 已提交
1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109
  int64_t status = atomic_load_64(&tmq->status);
  tmqAskEp(tmq, status == TMQ_CONSUMER_STATUS__INIT);

  while (1) {
    rspMsg = tmqSyncPollImpl(tmq, blocking_time);
    if (rspMsg && rspMsg->consumeRsp.numOfTopics) {
      return rspMsg;
    }

    if (blocking_time != 0) {
      int64_t endTime = taosGetTimestampMs();
      if (endTime - startTime > blocking_time) {
        return NULL;
      }
    } else
      return NULL;
  }
}

L
Liu Jicong 已提交
1110 1111
tmq_message_t* tmq_consumer_poll(tmq_t* tmq, int64_t blocking_time) {
  tmq_message_t* rspMsg;
L
Liu Jicong 已提交
1112 1113
  int64_t        startTime = taosGetTimestampMs();

L
Liu Jicong 已提交
1114
  // TODO: put into another thread or delayed queue
1115
  int64_t status = atomic_load_64(&tmq->status);
L
Liu Jicong 已提交
1116 1117
  tmqAskEp(tmq, status == TMQ_CONSUMER_STATUS__INIT);

L
fix  
Liu Jicong 已提交
1118 1119 1120 1121
  rspMsg = tmqHandleAllRsp(tmq, blocking_time, false);
  if (rspMsg) {
    return rspMsg;
  }
L
Liu Jicong 已提交
1122 1123

  while (1) {
L
fix  
Liu Jicong 已提交
1124
    /*printf("cycle\n");*/
L
Liu Jicong 已提交
1125
    tmqPollImpl(tmq, blocking_time);
L
Liu Jicong 已提交
1126 1127 1128 1129

    tsem_wait(&tmq->rspSem);

    rspMsg = tmqHandleAllRsp(tmq, blocking_time, false);
L
fix  
Liu Jicong 已提交
1130 1131 1132
    if (rspMsg) {
      return rspMsg;
    }
L
Liu Jicong 已提交
1133 1134 1135 1136 1137 1138
    if (blocking_time != 0) {
      int64_t endTime = taosGetTimestampMs();
      if (endTime - startTime > blocking_time) {
        return NULL;
      }
    }
L
Liu Jicong 已提交
1139 1140 1141 1142
  }
}

#if 0
1143

L
Liu Jicong 已提交
1144
  if (blocking_time <= 0) blocking_time = 1;
L
Liu Jicong 已提交
1145 1146
  if (blocking_time > 1000) blocking_time = 1000;
  /*blocking_time = 1;*/
1147 1148 1149

  if (taosArrayGetSize(tmq->clientTopics) == 0) {
    tscDebug("consumer:%ld poll but not assigned", tmq->consumerId);
L
Liu Jicong 已提交
1150
    /*printf("over1\n");*/
1151 1152 1153 1154 1155
    usleep(blocking_time * 1000);
    return NULL;
  }
  SMqClientTopic* pTopic = taosArrayGet(tmq->clientTopics, tmq->nextTopicIdx);
  if (taosArrayGetSize(pTopic->vgs) == 0) {
L
Liu Jicong 已提交
1156
    /*printf("over2\n");*/
1157 1158 1159 1160
    usleep(blocking_time * 1000);
    return NULL;
  }

L
Liu Jicong 已提交
1161
  tmq->nextTopicIdx = (tmq->nextTopicIdx + 1) % taosArrayGetSize(tmq->clientTopics);
L
Liu Jicong 已提交
1162
  int32_t beginVgIdx = pTopic->nextVgIdx;
L
Liu Jicong 已提交
1163
  while (1) {
L
Liu Jicong 已提交
1164 1165 1166
    pTopic->nextVgIdx = (pTopic->nextVgIdx + 1) % taosArrayGetSize(pTopic->vgs);
    SMqClientVg* pVg = taosArrayGet(pTopic->vgs, pTopic->nextVgIdx);
    /*printf("consume vg %d, offset %ld\n", pVg->vgId, pVg->currentOffset);*/
L
Liu Jicong 已提交
1167
    SMqConsumeReq* pReq = tmqBuildConsumeReqImpl(tmq, blocking_time, pTopic, pVg);
L
Liu Jicong 已提交
1168 1169 1170 1171 1172
    if (pReq == NULL) {
      ASSERT(false);
      usleep(blocking_time * 1000);
      return NULL;
    }
1173

L
Liu Jicong 已提交
1174
    SMqPollCbParam* param = malloc(sizeof(SMqPollCbParam));
L
Liu Jicong 已提交
1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185
    if (param == NULL) {
      ASSERT(false);
      usleep(blocking_time * 1000);
      return NULL;
    }
    param->tmq = tmq;
    param->retMsg = &tmq_message;
    param->pVg = pVg;
    tsem_init(&param->rspSem, 0, 0);

    SRequestObj* pRequest = createRequest(tmq->pTscObj, NULL, NULL, TDMT_VND_CONSUME);
L
Liu Jicong 已提交
1186 1187 1188 1189 1190
    pRequest->body.requestMsg = (SDataBuf){
        .pData = pReq,
        .len = sizeof(SMqConsumeReq),
        .handle = NULL,
    };
1191

L
Liu Jicong 已提交
1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213
    SMsgSendInfo* sendInfo = buildMsgInfoImpl(pRequest);
    sendInfo->requestObjRefId = 0;
    sendInfo->param = param;
    sendInfo->fp = tmqPollCb;

    /*printf("req offset: %ld\n", pReq->offset);*/

    int64_t transporterId = 0;
    asyncSendMsgToServer(tmq->pTscObj->pAppInfo->pTransporter, &pVg->epSet, &transporterId, sendInfo);
    tmq->pollCnt++;

    tsem_wait(&param->rspSem);
    tsem_destroy(&param->rspSem);
    free(param);

    if (tmq_message == NULL) {
      if (beginVgIdx == pTopic->nextVgIdx) {
        usleep(blocking_time * 1000);
      } else {
        continue;
      }
    }
L
Liu Jicong 已提交
1214

L
Liu Jicong 已提交
1215
    return tmq_message;
L
Liu Jicong 已提交
1216
  }
1217 1218 1219 1220

  /*tsem_wait(&pRequest->body.rspSem);*/

  /*if (body != NULL) {*/
L
Liu Jicong 已提交
1221
  /*destroySendMsgInfo(body);*/
1222 1223 1224
  /*}*/

  /*if (pRequest != NULL && terrno != TSDB_CODE_SUCCESS) {*/
L
Liu Jicong 已提交
1225
  /*pRequest->code = terrno;*/
1226 1227 1228 1229
  /*}*/

  /*return pRequest;*/
}
L
Liu Jicong 已提交
1230
#endif
1231

L
Liu Jicong 已提交
1232
#if 0
L
Liu Jicong 已提交
1233
tmq_resp_err_t tmq_commit(tmq_t* tmq, const tmq_topic_vgroup_list_t* tmq_topic_vgroup_list, int32_t async) {
L
Liu Jicong 已提交
1234
  if (tmq_topic_vgroup_list != NULL) {
L
Liu Jicong 已提交
1235
    // TODO
L
Liu Jicong 已提交
1236 1237
  }

L
Liu Jicong 已提交
1238
  // TODO: change semaphore to gate
L
Liu Jicong 已提交
1239 1240 1241
  for (int i = 0; i < taosArrayGetSize(tmq->clientTopics); i++) {
    SMqClientTopic* pTopic = taosArrayGet(tmq->clientTopics, i);
    for (int j = 0; j < taosArrayGetSize(pTopic->vgs); j++) {
L
Liu Jicong 已提交
1242
      SMqClientVg*   pVg = taosArrayGet(pTopic->vgs, j);
L
Liu Jicong 已提交
1243
      SMqConsumeReq* pReq = tmqBuildConsumeReqImpl(tmq, 0, pTopic, pVg);
L
Liu Jicong 已提交
1244

L
Liu Jicong 已提交
1245 1246
      SRequestObj* pRequest = createRequest(tmq->pTscObj, NULL, NULL, TDMT_VND_CONSUME);
      pRequest->body.requestMsg = (SDataBuf){.pData = pReq, .len = sizeof(SMqConsumeReq)};
L
Liu Jicong 已提交
1247
      SMqCommitCbParam* pParam = malloc(sizeof(SMqCommitCbParam));
L
Liu Jicong 已提交
1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267
      if (pParam == NULL) {
        continue;
      }
      pParam->tmq = tmq;
      pParam->pVg = pVg;
      pParam->async = async;
      if (!async) tsem_init(&pParam->rspSem, 0, 0);

      SMsgSendInfo* sendInfo = buildMsgInfoImpl(pRequest);
      sendInfo->requestObjRefId = 0;
      sendInfo->param = pParam;
      sendInfo->fp = tmqCommitCb;

      int64_t transporterId = 0;
      asyncSendMsgToServer(tmq->pTscObj->pAppInfo->pTransporter, &pVg->epSet, &transporterId, sendInfo);

      if (!async) tsem_wait(&pParam->rspSem);
    }
  }

L
Liu Jicong 已提交
1268
  return 0;
1269
}
L
Liu Jicong 已提交
1270
#endif
1271 1272 1273

void tmq_message_destroy(tmq_message_t* tmq_message) {
  if (tmq_message == NULL) return;
L
Liu Jicong 已提交
1274
  SMqConsumeRsp* pRsp = &tmq_message->consumeRsp;
L
Liu Jicong 已提交
1275
  tDeleteSMqConsumeRsp(pRsp);
L
Liu Jicong 已提交
1276
  /*free(tmq_message);*/
L
Liu Jicong 已提交
1277
  taosFreeQitem(tmq_message);
1278 1279
}

L
Liu Jicong 已提交
1280
tmq_resp_err_t tmq_consumer_close(tmq_t* tmq) { return TMQ_RESP_ERR__SUCCESS; }
L
Liu Jicong 已提交
1281 1282 1283 1284 1285 1286 1287

const char* tmq_err2str(tmq_resp_err_t err) {
  if (err == TMQ_RESP_ERR__SUCCESS) {
    return "success";
  }
  return "fail";
}
L
Liu Jicong 已提交
1288

L
Liu Jicong 已提交
1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302
#if 0
tmq_t* tmqCreateConsumerImpl(TAOS* conn, tmq_conf_t* conf) {
  tmq_t* pTmq = malloc(sizeof(tmq_t));
  if (pTmq == NULL) {
    return NULL;
  }
  strcpy(pTmq->groupId, conf->groupId);
  strcpy(pTmq->clientId, conf->clientId);
  pTmq->pTscObj = (STscObj*)conn;
  pTmq->pTscObj->connType = HEARTBEAT_TYPE_MQ;
  return pTmq;
}


1303 1304 1305 1306 1307
static void destroySendMsgInfo(SMsgSendInfo* pMsgBody) {
  assert(pMsgBody != NULL);
  tfree(pMsgBody->msgInfo.pData);
  tfree(pMsgBody);
}
L
Liu Jicong 已提交
1308
#endif