smaRollup.c 53.9 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
/*
 * 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 "sma.h"

L
Liu Jicong 已提交
18 19 20 21 22
#define RSMA_QTASKEXEC_SMOOTH_SIZE (100)     // cnt
#define RSMA_SUBMIT_BATCH_SIZE     (1024)    // cnt
#define RSMA_FETCH_DELAY_MAX       (120000)  // ms
#define RSMA_FETCH_ACTIVE_MAX      (1000)    // ms
#define RSMA_FETCH_INTERVAL        (5000)    // ms
C
Cary Xu 已提交
23

K
kailixu 已提交
24 25
#define RSMA_NEED_FETCH(r) (RSMA_INFO_ITEM((r), 0)->fetchLevel || RSMA_INFO_ITEM((r), 1)->fetchLevel)

26
SSmaMgmt smaMgmt = {
C
Cary Xu 已提交
27 28
    .inited = 0,
    .rsetId = -1,
29 30
};

31 32
typedef struct SRSmaQTaskInfoItem SRSmaQTaskInfoItem;

33
static int32_t    tdUidStorePut(STbUidStore *pStore, tb_uid_t suid, tb_uid_t *uid);
K
kailixu 已提交
34
static void       tdUidStoreDestory(STbUidStore *pStore);
35
static int32_t    tdUpdateTbUidListImpl(SSma *pSma, tb_uid_t *suid, SArray *tbUids, bool isAdd);
36
static int32_t    tdSetRSmaInfoItemParams(SSma *pSma, SRSmaParam *param, SRSmaStat *pStat, SRSmaInfo *pRSmaInfo,
L
Liu Jicong 已提交
37
                                          int8_t idx);
C
Cary Xu 已提交
38
static int32_t    tdExecuteRSmaImpl(SSma *pSma, const void *pMsg, int32_t msgSize, int32_t inputType, SRSmaInfo *pInfo,
C
Cary Xu 已提交
39
                                    ERsmaExecType type, int8_t level);
C
Cary Xu 已提交
40 41
static SRSmaInfo *tdAcquireRSmaInfoBySuid(SSma *pSma, int64_t suid);
static void       tdReleaseRSmaInfo(SSma *pSma, SRSmaInfo *pInfo);
C
Cary Xu 已提交
42
static void       tdFreeRSmaSubmitItems(SArray *pItems);
43
static int32_t    tdRSmaFetchAllResult(SSma *pSma, SRSmaInfo *pInfo);
C
Cary Xu 已提交
44 45
static int32_t    tdRSmaExecAndSubmitResult(SSma *pSma, qTaskInfo_t taskInfo, SRSmaInfoItem *pItem, STSchema *pTSchema,
                                            int64_t suid);
C
Cary Xu 已提交
46
static void       tdRSmaFetchTrigger(void *param, void *tmrId);
47 48
static int32_t    tdRSmaInfoClone(SSma *pSma, SRSmaInfo *pInfo);
static void       tdRSmaQTaskInfoFree(qTaskInfo_t *taskHandle, int32_t vgId, int32_t level);
C
Cary Xu 已提交
49 50 51
static int32_t    tdRSmaRestoreQTaskInfoInit(SSma *pSma, int64_t *nTables);
static int32_t    tdRSmaRestoreQTaskInfoReload(SSma *pSma, int8_t type, int64_t qTaskFileVer);
static int32_t    tdRSmaRestoreTSDataReload(SSma *pSma);
C
Cary Xu 已提交
52

53 54 55 56 57 58 59
struct SRSmaQTaskInfoItem {
  int32_t len;
  int8_t  type;
  int64_t suid;
  void   *qTaskInfo;
};

60
static void tdRSmaQTaskInfoFree(qTaskInfo_t *taskHandle, int32_t vgId, int32_t level) {
61
  // Note: free/kill may in RC
C
Cary Xu 已提交
62
  if (!taskHandle || !(*taskHandle)) return;
63 64
  qTaskInfo_t otaskHandle = atomic_load_ptr(taskHandle);
  if (otaskHandle && atomic_val_compare_exchange_ptr(taskHandle, otaskHandle, NULL)) {
C
Cary Xu 已提交
65
    smaDebug("vgId:%d, free qTaskInfo_t %p of level %d", vgId, otaskHandle, level);
66
    qDestroyTask(otaskHandle);
C
Cary Xu 已提交
67
  } else {
C
Cary Xu 已提交
68
    smaDebug("vgId:%d, not free qTaskInfo_t %p of level %d", vgId, otaskHandle, level);
69
  }
C
Cary Xu 已提交
70
  // TODO: clear files related to qTaskInfo?
71 72
}

C
Cary Xu 已提交
73 74 75 76 77 78 79 80 81
/**
 * @brief general function to free rsmaInfo
 *
 * @param pSma
 * @param pInfo
 * @param isDeepFree Only stop tmrId and free pTSchema for deep free
 * @return void*
 */
void *tdFreeRSmaInfo(SSma *pSma, SRSmaInfo *pInfo, bool isDeepFree) {
82
  if (pInfo) {
C
Cary Xu 已提交
83
    for (int32_t i = 0; i < TSDB_RETENTION_L2; ++i) {
84
      SRSmaInfoItem *pItem = &pInfo->items[i];
C
Cary Xu 已提交
85

C
Cary Xu 已提交
86
      if (isDeepFree && pItem->tmrId) {
C
Cary Xu 已提交
87 88
        smaDebug("vgId:%d, stop fetch timer %p for table %" PRIi64 " level %d", SMA_VID(pSma), pItem->tmrId,
                 pInfo->suid, i + 1);
C
Cary Xu 已提交
89 90 91
        taosTmrStopA(&pItem->tmrId);
      }

92 93 94 95
      if (isDeepFree && pItem->pStreamState) {
        streamStateClose(pItem->pStreamState);
      }

C
Cary Xu 已提交
96
      if (isDeepFree && pInfo->taskInfo[i]) {
C
Cary Xu 已提交
97
        tdRSmaQTaskInfoFree(&pInfo->taskInfo[i], SMA_VID(pSma), i + 1);
C
Cary Xu 已提交
98
      } else {
C
Cary Xu 已提交
99
        smaDebug("vgId:%d, table %" PRIi64 " no need to destroy rsma info level %d since empty taskInfo", SMA_VID(pSma),
100
                 pInfo->suid, i + 1);
101
      }
K
kailixu 已提交
102
#if 0
C
Cary Xu 已提交
103
      if (pInfo->iTaskInfo[i]) {
C
Cary Xu 已提交
104
        tdRSmaQTaskInfoFree(&pInfo->iTaskInfo[i], SMA_VID(pSma), i + 1);
C
Cary Xu 已提交
105
      } else {
C
Cary Xu 已提交
106 107
        smaDebug("vgId:%d, table %" PRIi64 " no need to destroy rsma info level %d since empty iTaskInfo",
                 SMA_VID(pSma), pInfo->suid, i + 1);
C
Cary Xu 已提交
108
      }
K
kailixu 已提交
109
#endif
110
    }
C
Cary Xu 已提交
111
    if (isDeepFree) {
C
Cary Xu 已提交
112
      taosMemoryFreeClear(pInfo->pTSchema);
C
Cary Xu 已提交
113
    }
C
Cary Xu 已提交
114 115

    if (isDeepFree) {
K
kailixu 已提交
116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133
      if (pInfo->queue) {
        taosCloseQueue(pInfo->queue);
        pInfo->queue = NULL;
      }
      if (pInfo->qall) {
        taosFreeQall(pInfo->qall);
        pInfo->qall = NULL;
      }
#if 0
      if (pInfo->iQueue) {
        taosCloseQueue(pInfo->iQueue);
        pInfo->iQueue = NULL;
      }
      if (pInfo->iQall) {
        taosFreeQall(pInfo->iQall);
        pInfo->iQall = NULL;
      }
#endif
C
Cary Xu 已提交
134 135
    }

C
Cary Xu 已提交
136 137
    taosMemoryFree(pInfo);
  }
138

139 140 141 142
  return NULL;
}

static FORCE_INLINE int32_t tdUidStoreInit(STbUidStore **pStore) {
K
kailixu 已提交
143 144 145 146 147 148
  if (ASSERTS(*pStore == NULL, "*pStore:%p != NULL", *pStore)) {
    terrno = TSDB_CODE_APP_ERROR;
    return TSDB_CODE_FAILED;
  }

  *pStore = taosMemoryCalloc(1, sizeof(STbUidStore));
149
  if (*pStore == NULL) {
K
kailixu 已提交
150 151
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return TSDB_CODE_FAILED;
152
  }
K
kailixu 已提交
153

154 155 156
  return TSDB_CODE_SUCCESS;
}

157
static int32_t tdUpdateTbUidListImpl(SSma *pSma, tb_uid_t *suid, SArray *tbUids, bool isAdd) {
158 159 160 161
  SRSmaInfo *pRSmaInfo = NULL;

  if (!suid || !tbUids) {
    terrno = TSDB_CODE_INVALID_PTR;
C
Cary Xu 已提交
162 163
    smaError("vgId:%d, failed to get rsma info for uid:%" PRIi64 " since %s", SMA_VID(pSma), suid ? *suid : -1,
             terrstr());
164 165 166
    return TSDB_CODE_FAILED;
  }

C
Cary Xu 已提交
167 168 169 170 171
  if (!taosArrayGetSize(tbUids)) {
    smaDebug("vgId:%d, no need to update tbUidList for suid:%" PRIi64 " since Empty tbUids", SMA_VID(pSma), *suid);
    return TSDB_CODE_SUCCESS;
  }

C
Cary Xu 已提交
172
  pRSmaInfo = tdAcquireRSmaInfoBySuid(pSma, *suid);
173 174

  if (!pRSmaInfo) {
S
Shengliang Guan 已提交
175
    smaError("vgId:%d, failed to get rsma info for uid:%" PRIi64, SMA_VID(pSma), *suid);
C
Cary Xu 已提交
176
    terrno = TSDB_CODE_RSMA_INVALID_STAT;
177 178 179
    return TSDB_CODE_FAILED;
  }

C
Cary Xu 已提交
180 181
  for (int32_t i = 0; i < TSDB_RETENTION_L2; ++i) {
    if (pRSmaInfo->taskInfo[i]) {
K
kailixu 已提交
182
      if ((terrno = qUpdateQualifiedTableId(pRSmaInfo->taskInfo[i], tbUids, isAdd)) < 0) {
C
Cary Xu 已提交
183
        tdReleaseRSmaInfo(pSma, pRSmaInfo);
C
Cary Xu 已提交
184 185 186 187
        smaError("vgId:%d, update tbUidList failed for uid:%" PRIi64 " level %d since %s", SMA_VID(pSma), *suid, i,
                 terrstr());
        return TSDB_CODE_FAILED;
      }
K
kailixu 已提交
188 189
      smaDebug("vgId:%d, update tbUidList succeed for qTaskInfo:%p with suid:%" PRIi64 " uid:%" PRIi64 " level %d",
               SMA_VID(pSma), pRSmaInfo->taskInfo[i], *suid, *(int64_t *)taosArrayGet(tbUids, 0), i);
C
Cary Xu 已提交
190
    }
191 192
  }

C
Cary Xu 已提交
193
  tdReleaseRSmaInfo(pSma, pRSmaInfo);
194 195 196
  return TSDB_CODE_SUCCESS;
}

197
int32_t tdUpdateTbUidList(SSma *pSma, STbUidStore *pStore, bool isAdd) {
198 199 200 201
  if (!pStore || (taosArrayGetSize(pStore->tbUids) == 0)) {
    return TSDB_CODE_SUCCESS;
  }

202
  if (tdUpdateTbUidListImpl(pSma, &pStore->suid, pStore->tbUids, isAdd) != TSDB_CODE_SUCCESS) {
203 204 205 206 207 208 209 210
    return TSDB_CODE_FAILED;
  }

  void *pIter = taosHashIterate(pStore->uidHash, NULL);
  while (pIter) {
    tb_uid_t *pTbSuid = (tb_uid_t *)taosHashGetKey(pIter, NULL);
    SArray   *pTbUids = *(SArray **)pIter;

211
    if (tdUpdateTbUidListImpl(pSma, pTbSuid, pTbUids, isAdd) != TSDB_CODE_SUCCESS) {
212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237
      taosHashCancelIterate(pStore->uidHash, pIter);
      return TSDB_CODE_FAILED;
    }

    pIter = taosHashIterate(pStore->uidHash, pIter);
  }
  return TSDB_CODE_SUCCESS;
}

/**
 * @brief fetch suid/uids when create child tables of rollup SMA
 *
 * @param pTsdb
 * @param ppStore
 * @param suid
 * @param uid
 * @return int32_t
 */
int32_t tdFetchTbUidList(SSma *pSma, STbUidStore **ppStore, tb_uid_t suid, tb_uid_t uid) {
  SSmaEnv *pEnv = SMA_RSMA_ENV(pSma);

  // only applicable to rollup SMA ctables
  if (!pEnv) {
    return TSDB_CODE_SUCCESS;
  }

C
Cary Xu 已提交
238 239 240
  SRSmaStat *pStat = (SRSmaStat *)SMA_ENV_STAT(pEnv);
  SHashObj  *infoHash = NULL;
  if (!pStat || !(infoHash = RSMA_INFO_HASH(pStat))) {
C
Cary Xu 已提交
241
    terrno = TSDB_CODE_RSMA_INVALID_STAT;
242 243 244 245 246 247 248 249 250
    return TSDB_CODE_FAILED;
  }

  // info cached when create rsma stable and return directly for non-rsma ctables
  if (!taosHashGet(infoHash, &suid, sizeof(tb_uid_t))) {
    return TSDB_CODE_SUCCESS;
  }

  if (!(*ppStore)) {
251
    if (tdUidStoreInit(ppStore) < 0) {
252 253 254 255
      return TSDB_CODE_FAILED;
    }
  }

256
  if (tdUidStorePut(*ppStore, suid, &uid) < 0) {
257 258 259 260 261 262 263
    *ppStore = tdUidStoreFree(*ppStore);
    return TSDB_CODE_FAILED;
  }

  return TSDB_CODE_SUCCESS;
}

264
static int32_t tdSetRSmaInfoItemParams(SSma *pSma, SRSmaParam *param, SRSmaStat *pStat, SRSmaInfo *pRSmaInfo,
L
Liu Jicong 已提交
265
                                       int8_t idx) {
C
Cary Xu 已提交
266 267 268
  if ((param->qmsgLen > 0) && param->qmsg[idx]) {
    SRetention *pRetention = SMA_RETENTION(pSma);
    STsdbCfg   *pTsdbCfg = SMA_TSDB_CFG(pSma);
C
Cary Xu 已提交
269
    SVnode     *pVnode = pSma->pVnode;
270 271 272 273 274 275
    char        taskInfDir[TSDB_FILENAME_LEN] = {0};
    void       *pStreamState = NULL;

    // set the backend of stream state
    tdRSmaQTaskInfoGetFullPathEx(TD_VID(pVnode), pRSmaInfo->suid, idx + 1, tfsGetPrimaryPath(pVnode->pTfs), taskInfDir);
    if (!taosCheckExistFile(taskInfDir)) {
276
      char *s = taosStrdup(taskInfDir);
277 278 279 280 281 282 283
      if (taosMulMkDir(taosDirName(s)) != 0) {
        terrno = TAOS_SYSTEM_ERROR(errno);
        taosMemoryFree(s);
        return TSDB_CODE_FAILED;
      }
      taosMemoryFree(s);
    }
284
    pStreamState = streamStateOpen(taskInfDir, NULL, true, -1, -1);
285 286 287 288 289
    if (!pStreamState) {
      terrno = TSDB_CODE_RSMA_STREAM_STATE_OPEN;
      return TSDB_CODE_FAILED;
    }

C
Cary Xu 已提交
290
    SReadHandle handle = {
C
Cary Xu 已提交
291 292
        .meta = pVnode->pMeta,
        .vnode = pVnode,
C
Cary Xu 已提交
293
        .initTqReader = 1,
294
        .pStateBackend = pStreamState,
C
Cary Xu 已提交
295
    };
296
    pRSmaInfo->taskInfo[idx] = qCreateStreamExecTaskInfo(param->qmsg[idx], &handle, TD_VID(pVnode));
C
Cary Xu 已提交
297
    if (!pRSmaInfo->taskInfo[idx]) {
C
Cary Xu 已提交
298
      terrno = TSDB_CODE_RSMA_QTASKINFO_CREATE;
C
Cary Xu 已提交
299
      return TSDB_CODE_FAILED;
C
Cary Xu 已提交
300
    }
C
Cary Xu 已提交
301
    SRSmaInfoItem *pItem = &(pRSmaInfo->items[idx]);
302
    pItem->triggerStat = TASK_TRIGGER_STAT_ACTIVE;  // fetch the data when reboot
303
    pItem->pStreamState = pStreamState;
C
Cary Xu 已提交
304
    if (param->maxdelay[idx] < TSDB_MIN_ROLLUP_MAX_DELAY) {
K
kailixu 已提交
305 306
      int64_t msInterval =
          convertTimeFromPrecisionToUnit(pRetention[idx + 1].freq, pTsdbCfg->precision, TIME_UNIT_MILLISECOND);
C
Cary Xu 已提交
307
      pItem->maxDelay = (int32_t)msInterval;
C
Cary Xu 已提交
308
    } else {
C
Cary Xu 已提交
309
      pItem->maxDelay = (int32_t)param->maxdelay[idx];
C
Cary Xu 已提交
310
    }
C
Cary Xu 已提交
311 312
    if (pItem->maxDelay > TSDB_MAX_ROLLUP_MAX_DELAY) {
      pItem->maxDelay = TSDB_MAX_ROLLUP_MAX_DELAY;
C
Cary Xu 已提交
313
    }
C
Cary Xu 已提交
314

K
kailixu 已提交
315 316 317 318 319 320
    pItem->level = idx == 0 ? TSDB_RETENTION_L1 : TSDB_RETENTION_L2;

    if (ASSERTS(pItem->level > 0, "pItem level:%" PRIi8 " should > 0", pItem->level)) {
      terrno = TSDB_CODE_APP_ERROR;
      return TSDB_CODE_FAILED;
    }
321 322 323 324

    SRSmaRef rsmaRef = {.refId = pStat->refId, .suid = pRSmaInfo->suid};
    taosHashPut(smaMgmt.refHash, &pItem, POINTER_BYTES, &rsmaRef, sizeof(rsmaRef));

325
    taosTmrReset(tdRSmaFetchTrigger, RSMA_FETCH_INTERVAL, pItem, smaMgmt.tmrHandle, &pItem->tmrId);
326 327

    smaInfo("vgId:%d, item:%p table:%" PRIi64 " level:%" PRIi8 " maxdelay:%" PRIi64 " watermark:%" PRIi64
328
            ", finally maxdelay:%" PRIi32,
329
            TD_VID(pVnode), pItem, pRSmaInfo->suid, (int8_t)(idx + 1), param->maxdelay[idx], param->watermark[idx],
330
            pItem->maxDelay);
C
Cary Xu 已提交
331 332 333 334
  }
  return TSDB_CODE_SUCCESS;
}

335
/**
336
 * @brief for rsam create or restore
337
 *
338 339 340 341
 * @param pSma
 * @param param
 * @param suid
 * @param tbName
342 343
 * @return int32_t
 */
344
int32_t tdRSmaProcessCreateImpl(SSma *pSma, SRSmaParam *param, int64_t suid, const char *tbName) {
345
  if ((param->qmsgLen[0] == 0) && (param->qmsgLen[1] == 0)) {
346
    smaDebug("vgId:%d, no qmsg1/qmsg2 for rollup table %s %" PRIi64, SMA_VID(pSma), tbName, suid);
347 348 349
    return TSDB_CODE_SUCCESS;
  }

K
kailixu 已提交
350
#if 0
C
Cary Xu 已提交
351
  if (tdCheckAndInitSmaEnv(pSma, TSDB_SMA_TYPE_ROLLUP) != TSDB_CODE_SUCCESS) {
352 353 354
    terrno = TSDB_CODE_TDB_INIT_FAILED;
    return TSDB_CODE_FAILED;
  }
K
kailixu 已提交
355
#endif
356 357

  SSmaEnv   *pEnv = SMA_RSMA_ENV(pSma);
C
Cary Xu 已提交
358
  SRSmaStat *pStat = (SRSmaStat *)SMA_ENV_STAT(pEnv);
359 360
  SRSmaInfo *pRSmaInfo = NULL;

361
  pRSmaInfo = taosHashGet(RSMA_INFO_HASH(pStat), &suid, sizeof(tb_uid_t));
362
  if (pRSmaInfo) {
363 364
    smaInfo("vgId:%d, rsma info already exists for table %s, %" PRIi64, SMA_VID(pSma), tbName, suid);
    return TSDB_CODE_SUCCESS;
365 366
  }

367
  // from write queue: single thead
368 369 370 371 372 373
  pRSmaInfo = (SRSmaInfo *)taosMemoryCalloc(1, sizeof(SRSmaInfo));
  if (!pRSmaInfo) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return TSDB_CODE_FAILED;
  }

374
  STSchema *pTSchema = metaGetTbTSchema(SMA_META(pSma), suid, -1, 1);
375 376 377 378
  if (!pTSchema) {
    terrno = TSDB_CODE_TDB_IVD_TB_SCHEMA_VERSION;
    goto _err;
  }
C
Cary Xu 已提交
379
  pRSmaInfo->pSma = pSma;
380
  pRSmaInfo->pTSchema = pTSchema;
C
Cary Xu 已提交
381 382
  pRSmaInfo->suid = suid;
  T_REF_INIT_VAL(pRSmaInfo, 1);
C
Cary Xu 已提交
383 384 385
  if (!(pRSmaInfo->queue = taosOpenQueue())) {
    goto _err;
  }
C
Cary Xu 已提交
386

C
Cary Xu 已提交
387 388 389
  if (!(pRSmaInfo->qall = taosAllocateQall())) {
    goto _err;
  }
K
kailixu 已提交
390
#if 0
C
Cary Xu 已提交
391 392 393 394 395 396
  if (!(pRSmaInfo->iQueue = taosOpenQueue())) {
    goto _err;
  }
  if (!(pRSmaInfo->iQall = taosAllocateQall())) {
    goto _err;
  }
K
kailixu 已提交
397
#endif
L
Liu Jicong 已提交
398
  if (tdSetRSmaInfoItemParams(pSma, param, pStat, pRSmaInfo, 0) < 0) {
C
Cary Xu 已提交
399 400
    goto _err;
  }
C
Cary Xu 已提交
401

L
Liu Jicong 已提交
402
  if (tdSetRSmaInfoItemParams(pSma, param, pStat, pRSmaInfo, 1) < 0) {
C
Cary Xu 已提交
403 404
    goto _err;
  }
405

406
  if (taosHashPut(RSMA_INFO_HASH(pStat), &suid, sizeof(tb_uid_t), &pRSmaInfo, sizeof(pRSmaInfo)) < 0) {
407
    goto _err;
408 409
  }

C
Cary Xu 已提交
410
  smaDebug("vgId:%d, register rsma info succeed for table %" PRIi64, SMA_VID(pSma), suid);
C
Cary Xu 已提交
411

412
  return TSDB_CODE_SUCCESS;
413
_err:
C
Cary Xu 已提交
414
  tdFreeRSmaInfo(pSma, pRSmaInfo, true);
415
  return TSDB_CODE_FAILED;
416 417
}

418
/**
C
Cary Xu 已提交
419
 * @brief Check and init qTaskInfo_t, only applicable to stable with SRSmaParam currently
420
 *
421
 * @param pSma
422 423 424
 * @param pReq
 * @return int32_t
 */
425 426
int32_t tdProcessRSmaCreate(SSma *pSma, SVCreateStbReq *pReq) {
  SVnode *pVnode = pSma->pVnode;
427
  if (!pReq->rollup) {
428 429 430 431 432 433
    smaTrace("vgId:%d, not create rsma for stable %s %" PRIi64 " since no rollup in req", TD_VID(pVnode), pReq->name,
             pReq->suid);
    return TSDB_CODE_SUCCESS;
  }

  if (!VND_IS_RSMA(pVnode)) {
K
kailixu 已提交
434 435
    smaWarn("vgId:%d, not create rsma for stable %s %" PRIi64 " since vnd is not rsma", TD_VID(pVnode), pReq->name,
            pReq->suid);
436 437 438
    return TSDB_CODE_SUCCESS;
  }

439
  return tdRSmaProcessCreateImpl(pSma, &pReq->rsmaParam, pReq->suid, pReq->name);
440 441
}

442 443 444 445 446 447 448
/**
 * @brief drop cache for stb
 *
 * @param pSma
 * @param pReq
 * @return int32_t
 */
L
Liu Jicong 已提交
449
int32_t tdProcessRSmaDrop(SSma *pSma, SVDropStbReq *pReq) {
450 451
  SVnode *pVnode = pSma->pVnode;
  if (!VND_IS_RSMA(pVnode)) {
C
Cary Xu 已提交
452
    smaTrace("vgId:%d, not drop rsma for stable %s %" PRIi64 " since vnd is not rsma", TD_VID(pVnode), pReq->name,
453 454 455 456
             pReq->suid);
    return TSDB_CODE_SUCCESS;
  }

C
Cary Xu 已提交
457 458 459 460 461
  SSmaEnv *pSmaEnv = SMA_RSMA_ENV(pSma);
  if (!pSmaEnv) {
    return TSDB_CODE_SUCCESS;
  }

C
Cary Xu 已提交
462
  SRSmaStat *pRSmaStat = (SRSmaStat *)SMA_ENV_STAT(pSmaEnv);
C
Cary Xu 已提交
463

C
Cary Xu 已提交
464
  SRSmaInfo *pRSmaInfo = tdAcquireRSmaInfoBySuid(pSma, pReq->suid);
C
Cary Xu 已提交
465 466 467 468 469 470 471 472

  if (!pRSmaInfo) {
    smaWarn("vgId:%d, drop rsma for stable %s %" PRIi64 " failed no rsma in hash", TD_VID(pVnode), pReq->name,
            pReq->suid);
    return TSDB_CODE_SUCCESS;
  }

  // set del flag for data in mem
473
  atomic_store_8(&pRSmaStat->delFlag, 1);
C
Cary Xu 已提交
474 475 476
  RSMA_INFO_SET_DEL(pRSmaInfo);
  tdUnRefRSmaInfo(pSma, pRSmaInfo);

C
Cary Xu 已提交
477
  tdReleaseRSmaInfo(pSma, pRSmaInfo);
C
Cary Xu 已提交
478

K
kailixu 已提交
479 480
  // no need to save to file as triggered by dropping stable
  smaDebug("vgId:%d, drop rsma for stable %" PRIi64 " succeed", TD_VID(pVnode), pReq->suid);
481
  return TSDB_CODE_SUCCESS;
L
Liu Jicong 已提交
482
}
483

484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528
/**
 * @brief store suid/[uids], prefer to use array and then hash
 *
 * @param pStore
 * @param suid
 * @param uid
 * @return int32_t
 */
static int32_t tdUidStorePut(STbUidStore *pStore, tb_uid_t suid, tb_uid_t *uid) {
  // prefer to store suid/uids in array
  if ((suid == pStore->suid) || (pStore->suid == 0)) {
    if (pStore->suid == 0) {
      pStore->suid = suid;
    }
    if (uid) {
      if (!pStore->tbUids) {
        if (!(pStore->tbUids = taosArrayInit(1, sizeof(tb_uid_t)))) {
          terrno = TSDB_CODE_OUT_OF_MEMORY;
          return TSDB_CODE_FAILED;
        }
      }
      if (!taosArrayPush(pStore->tbUids, uid)) {
        return TSDB_CODE_FAILED;
      }
    }
  } else {
    // store other suid/uids in hash when multiple stable/table included in 1 batch of request
    if (!pStore->uidHash) {
      pStore->uidHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
      if (!pStore->uidHash) {
        return TSDB_CODE_FAILED;
      }
    }
    if (uid) {
      SArray *uidArray = taosHashGet(pStore->uidHash, &suid, sizeof(tb_uid_t));
      if (uidArray && ((uidArray = *(SArray **)uidArray))) {
        taosArrayPush(uidArray, uid);
      } else {
        SArray *pUidArray = taosArrayInit(1, sizeof(tb_uid_t));
        if (!pUidArray) {
          terrno = TSDB_CODE_OUT_OF_MEMORY;
          return TSDB_CODE_FAILED;
        }
        if (!taosArrayPush(pUidArray, uid)) {
          terrno = TSDB_CODE_OUT_OF_MEMORY;
C
Cary Xu 已提交
529
          taosArrayDestroy(pUidArray);
530 531
          return TSDB_CODE_FAILED;
        }
532
        if (taosHashPut(pStore->uidHash, &suid, sizeof(suid), &pUidArray, sizeof(pUidArray)) < 0) {
533 534 535 536
          return TSDB_CODE_FAILED;
        }
      }
    } else {
537
      if (taosHashPut(pStore->uidHash, &suid, sizeof(suid), NULL, 0) < 0) {
538 539 540 541 542 543 544
        return TSDB_CODE_FAILED;
      }
    }
  }
  return TSDB_CODE_SUCCESS;
}

K
kailixu 已提交
545
static void tdUidStoreDestory(STbUidStore *pStore) {
546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570
  if (pStore) {
    if (pStore->uidHash) {
      if (pStore->tbUids) {
        // When pStore->tbUids not NULL, the pStore->uidHash has k/v; otherwise pStore->uidHash only has keys.
        void *pIter = taosHashIterate(pStore->uidHash, NULL);
        while (pIter) {
          SArray *arr = *(SArray **)pIter;
          taosArrayDestroy(arr);
          pIter = taosHashIterate(pStore->uidHash, pIter);
        }
      }
      taosHashCleanup(pStore->uidHash);
    }
    taosArrayDestroy(pStore->tbUids);
  }
}

void *tdUidStoreFree(STbUidStore *pStore) {
  if (pStore) {
    tdUidStoreDestory(pStore);
    taosMemoryFree(pStore);
  }
  return NULL;
}

C
Cary Xu 已提交
571 572 573 574 575 576 577 578
/**
 * @brief The SubmitReq for rsma L2/L3 is inserted by tsdbInsertData method directly while not by WriteQ, as the queue
 * would be freed when close Vnode, thus lock should be used if with race condition.
 * @param pTsdb
 * @param version
 * @param pReq
 * @return int32_t
 */
579 580 581 582 583 584
static int32_t tdProcessSubmitReq(STsdb *pTsdb, int64_t version, void *pReq) {
  if (!pReq) {
    terrno = TSDB_CODE_INVALID_PTR;
    return TSDB_CODE_FAILED;
  }

585 586
  SSubmitReq2 *pSubmitReq = (SSubmitReq2 *)pReq;
  // spin lock for race condition during insert data
587 588 589 590 591 592 593
  if (tsdbInsertData(pTsdb, version, pSubmitReq, NULL) < 0) {
    return TSDB_CODE_FAILED;
  }

  return TSDB_CODE_SUCCESS;
}

594 595 596
static int32_t tdFetchSubmitReqSuids(SSubmitReq2 *pMsg, STbUidStore *pStore) {
  SArray *pSubmitTbData = pMsg ? pMsg->aSubmitTbData : NULL;
  int32_t size = taosArrayGetSize(pSubmitTbData);
597 598 599

  terrno = TSDB_CODE_SUCCESS;

600 601
  for (int32_t i = 0; i < size; ++i) {
    SSubmitTbData *pData = TARRAY_GET_ELEM(pSubmitTbData, i);
602
    if ((terrno = tdUidStorePut(pStore, pData->suid, NULL)) < 0) {
C
Cary Xu 已提交
603 604
      return -1;
    }
605 606 607 608 609
  }

  return 0;
}

H
Hongze Cheng 已提交
610
#if 0
C
Cary Xu 已提交
611 612
/**
 * @brief retention of rsma1/rsma2
C
Cary Xu 已提交
613 614 615 616
 *
 * @param pSma
 * @param now
 * @return int32_t
C
Cary Xu 已提交
617
 */
618
int32_t smaDoRetention(SSma *pSma, int64_t now) {
C
Cary Xu 已提交
619
  int32_t code = TSDB_CODE_SUCCESS;
C
Cary Xu 已提交
620
  if (!VND_IS_RSMA(pSma->pVnode)) {
C
Cary Xu 已提交
621 622 623 624 625
    return code;
  }

  for (int32_t i = 0; i < TSDB_RETENTION_L2; ++i) {
    if (pSma->pRSmaTsdb[i]) {
626
      code = tsdbDoRetention(pSma->pRSmaTsdb[i], now);
C
Cary Xu 已提交
627 628 629 630 631 632 633
      if (code) goto _end;
    }
  }

_end:
  return code;
}
H
Hongze Cheng 已提交
634
#endif
C
Cary Xu 已提交
635

K
kailixu 已提交
636 637 638 639 640 641 642
static void tdBlockDataDestroy(SArray *pBlockArr) {
  for (int32_t i = 0; i < taosArrayGetSize(pBlockArr); ++i) {
    blockDataDestroy(taosArrayGetP(pBlockArr, i));
  }
  taosArrayDestroy(pBlockArr);
}

C
Cary Xu 已提交
643 644
static int32_t tdRSmaExecAndSubmitResult(SSma *pSma, qTaskInfo_t taskInfo, SRSmaInfoItem *pItem, STSchema *pTSchema,
                                         int64_t suid) {
H
Haojun Liao 已提交
645 646 647 648 649
  SArray *pResList = taosArrayInit(1, POINTER_BYTES);
  if (pResList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _err;
  }
650

H
Haojun Liao 已提交
651 652
  while (1) {
    uint64_t ts;
653
    bool     hasMore = false;
654
    int32_t  code = qExecTaskOpt(taskInfo, pResList, &ts, &hasMore, NULL);
C
Cary Xu 已提交
655
    if (code < 0) {
C
Cary Xu 已提交
656 657 658 659 660 661 662
      if (code == TSDB_CODE_QRY_IN_EXEC) {
        break;
      } else {
        smaError("vgId:%d, qExecTask for rsma table %" PRIi64 " level %" PRIi8 " failed since %s", SMA_VID(pSma), suid,
                 pItem->level, terrstr(code));
        goto _err;
      }
663
    }
C
Cary Xu 已提交
664

H
Haojun Liao 已提交
665 666
    if (taosArrayGetSize(pResList) == 0) {
      if (terrno == 0) {
667
        // smaDebug("vgId:%d, no rsma level %" PRIi8 " data fetched yet", SMA_VID(pSma), pItem->level);
H
Haojun Liao 已提交
668
      } else {
669
        smaDebug("vgId:%d, no rsma level %" PRIi8 " data fetched since %s", SMA_VID(pSma), pItem->level, terrstr());
H
Haojun Liao 已提交
670 671 672 673
        goto _err;
      }

      break;
C
Cary Xu 已提交
674
    } else {
675
      smaDebug("vgId:%d, rsma level %" PRIi8 " data fetched", SMA_VID(pSma), pItem->level);
H
Haojun Liao 已提交
676
    }
C
Cary Xu 已提交
677
#if 0
C
Cary Xu 已提交
678 679 680
    char flag[10] = {0};
    snprintf(flag, 10, "level %" PRIi8, pItem->level);
    blockDebugShowDataBlocks(pResList, flag);
C
Cary Xu 已提交
681
#endif
C
Cary Xu 已提交
682 683
    for (int32_t i = 0; i < taosArrayGetSize(pResList); ++i) {
      SSDataBlock *output = taosArrayGetP(pResList, i);
K
kailixu 已提交
684 685
      smaDebug("result block, uid:%" PRIu64 ", groupid:%" PRIu64 ", rows:%d", output->info.id.uid,
               output->info.id.groupId, output->info.rows);
H
Haojun Liao 已提交
686

687
      STsdb       *sinkTsdb = (pItem->level == TSDB_RETENTION_L1 ? pSma->pRSmaTsdb[0] : pSma->pRSmaTsdb[1]);
688
      SSubmitReq2 *pReq = NULL;
C
Cary Xu 已提交
689 690

      // TODO: the schema update should be handled later(TD-17965)
691
      if (buildSubmitReqFromDataBlock(&pReq, output, pTSchema, output->info.id.groupId, SMA_VID(pSma), suid) < 0) {
K
kailixu 已提交
692 693
        smaError("vgId:%d, build submit req for rsma table suid:%" PRIu64 ", uid:%" PRIu64 ", level %" PRIi8
                 " failed since %s",
H
Haojun Liao 已提交
694
                 SMA_VID(pSma), suid, output->info.id.groupId, pItem->level, terrstr());
C
Cary Xu 已提交
695 696
        goto _err;
      }
697

C
Cary Xu 已提交
698
      if (pReq && tdProcessSubmitReq(sinkTsdb, output->info.version, pReq) < 0) {
699
        tDestroySSubmitReq2(pReq, TSDB_MSG_FLG_ENCODE);
K
kailixu 已提交
700
        taosMemoryFree(pReq);
K
kailixu 已提交
701 702
        smaError("vgId:%d, process submit req for rsma suid:%" PRIu64 ", uid:%" PRIu64 " level %" PRIi8
                 " failed since %s",
H
Haojun Liao 已提交
703
                 SMA_VID(pSma), suid, output->info.id.groupId, pItem->level, terrstr());
C
Cary Xu 已提交
704 705
        goto _err;
      }
C
Cary Xu 已提交
706

707 708
      smaDebug("vgId:%d, process submit req for rsma suid:%" PRIu64 ",uid:%" PRIu64 ", level %" PRIi8 " ver %" PRIi64,
               SMA_VID(pSma), suid, output->info.id.groupId, pItem->level, output->info.version);
709

K
kailixu 已提交
710 711 712 713
      if (pReq) {
        tDestroySSubmitReq2(pReq, TSDB_MSG_FLG_ENCODE);
        taosMemoryFree(pReq);
      }
C
Cary Xu 已提交
714
    }
715 716
  }

H
Haojun Liao 已提交
717 718
  taosArrayDestroy(pResList);
  qCleanExecTaskBlockBuf(taskInfo);
C
Cary Xu 已提交
719
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
720

C
Cary Xu 已提交
721
_err:
H
Haojun Liao 已提交
722 723
  taosArrayDestroy(pResList);
  qCleanExecTaskBlockBuf(taskInfo);
C
Cary Xu 已提交
724
  return TSDB_CODE_FAILED;
725 726
}

C
Cary Xu 已提交
727
/**
C
Cary Xu 已提交
728
 * @brief Copy msg to rsmaQueueBuffer for batch process
C
Cary Xu 已提交
729 730
 *
 * @param pSma
K
kailixu 已提交
731
 * @param version
C
Cary Xu 已提交
732
 * @param pMsg
733
 * @param len
C
Cary Xu 已提交
734 735 736 737 738
 * @param inputType
 * @param pInfo
 * @param suid
 * @return int32_t
 */
K
kailixu 已提交
739 740 741
static int32_t tdExecuteRSmaImplAsync(SSma *pSma, int64_t version, const void *pMsg, int32_t len, int32_t inputType,
                                      SRSmaInfo *pInfo, tb_uid_t suid) {
  int32_t size = sizeof(int32_t) + sizeof(int64_t) + len;
K
kailixu 已提交
742
  void   *qItem = taosAllocateQitem(size, DEF_QITEM, 0);
C
Cary Xu 已提交
743 744 745 746 747

  if (!qItem) {
    return TSDB_CODE_FAILED;
  }

K
kailixu 已提交
748 749 750 751 752 753
  void *pItem = qItem;

  *(int32_t *)pItem = len;
  pItem = POINTER_SHIFT(pItem, sizeof(int32_t));
  *(int64_t *)pItem = version;
  memcpy(POINTER_SHIFT(pItem, sizeof(int64_t)), pMsg, len);
C
Cary Xu 已提交
754 755 756

  taosWriteQitem(pInfo->queue, qItem);

C
Cary Xu 已提交
757 758
  pInfo->lastRecv = taosGetTimestampMs();

C
Cary Xu 已提交
759
  SRSmaStat *pRSmaStat = SMA_RSMA_STAT(pSma);
C
Cary Xu 已提交
760 761

  int64_t nItems = atomic_fetch_add_64(&pRSmaStat->nBufItems, 1);
C
Cary Xu 已提交
762

763 764 765 766
  if (atomic_load_8(&pInfo->assigned) == 0) {
    tsem_post(&(pRSmaStat->notEmpty));
  }

C
Cary Xu 已提交
767
  // smoothing consume
C
Cary Xu 已提交
768
  int32_t n = nItems / RSMA_QTASKEXEC_SMOOTH_SIZE;
C
Cary Xu 已提交
769 770 771 772
  if (n > 1) {
    if (n > 10) {
      n = 10;
    }
C
Cary Xu 已提交
773 774
    taosMsleep(n << 3);
    if (n > 5) {
C
Cary Xu 已提交
775
      smaWarn("vgId:%d, pInfo->queue itemSize:%d, memSize:%" PRIi64 ", sleep %d ms", SMA_VID(pSma),
C
Cary Xu 已提交
776
              taosQueueItemSize(pInfo->queue), taosQueueMemorySize(pInfo->queue), n << 3);
C
Cary Xu 已提交
777 778
    }
  }
C
Cary Xu 已提交
779

C
Cary Xu 已提交
780 781 782
  return TSDB_CODE_SUCCESS;
}

783
#if 0
C
Cary Xu 已提交
784 785 786 787 788 789 790 791 792 793 794
static int32_t tdRsmaPrintSubmitReq(SSma *pSma, SSubmitReq *pReq) {
  SSubmitMsgIter msgIter = {0};
  SSubmitBlkIter blkIter = {0};
  STSRow        *row = NULL;
  if (tInitSubmitMsgIter(pReq, &msgIter) < 0) return -1;
  while (true) {
    SSubmitBlk *pBlock = NULL;
    if (tGetSubmitMsgNext(&msgIter, &pBlock) < 0) return -1;
    if (pBlock == NULL) break;
    tInitSubmitBlkIter(&msgIter, pBlock, &blkIter);
    while ((row = tGetSubmitBlkNext(&blkIter)) != NULL) {
C
Cary Xu 已提交
795
      smaDebug("vgId:%d, numOfRows:%d, suid:%" PRIi64 ", uid:%" PRIi64 ", version:%" PRIi64 ", ts:%" PRIi64,
C
Cary Xu 已提交
796 797 798 799 800
               SMA_VID(pSma), msgIter.numOfRows, msgIter.suid, msgIter.uid, pReq->version, row->ts);
    }
  }
  return 0;
}
801
#endif
C
Cary Xu 已提交
802

C
Cary Xu 已提交
803 804 805 806 807 808 809 810
/**
 * @brief sync mode
 *
 * @param pSma
 * @param pMsg
 * @param msgSize
 * @param inputType
 * @param pInfo
C
Cary Xu 已提交
811
 * @param type
C
Cary Xu 已提交
812 813 814 815
 * @param level
 * @return int32_t
 */
static int32_t tdExecuteRSmaImpl(SSma *pSma, const void *pMsg, int32_t msgSize, int32_t inputType, SRSmaInfo *pInfo,
C
Cary Xu 已提交
816
                                 ERsmaExecType type, int8_t level) {
C
Cary Xu 已提交
817
  int32_t idx = level - 1;
K
kailixu 已提交
818
#if 0
K
kailixu 已提交
819
  void   *qTaskInfo = (type == RSMA_EXEC_COMMIT) ? RSMA_INFO_IQTASK(pInfo, idx) : RSMA_INFO_QTASK(pInfo, idx);
K
kailixu 已提交
820 821 822
#else
  void *qTaskInfo = RSMA_INFO_QTASK(pInfo, idx);
#endif
C
Cary Xu 已提交
823 824 825
  if (!qTaskInfo) {
    smaDebug("vgId:%d, no qTaskInfo to execute rsma %" PRIi8 " task for suid:%" PRIu64, SMA_VID(pSma), level,
             pInfo->suid);
826 827
    return TSDB_CODE_SUCCESS;
  }
C
Cary Xu 已提交
828
  if (!pInfo->pTSchema) {
K
kailixu 已提交
829
    terrno = TSDB_CODE_INVALID_PTR;
C
Cary Xu 已提交
830
    smaWarn("vgId:%d, no schema to execute rsma %" PRIi8 " task for suid:%" PRIu64, SMA_VID(pSma), level, pInfo->suid);
831 832
    return TSDB_CODE_FAILED;
  }
833

K
kailixu 已提交
834 835
  smaDebug("vgId:%d, execute rsma %" PRIi8 " task for qTaskInfo:%p suid:%" PRIu64 " nMsg:%d", SMA_VID(pSma), level,
           RSMA_INFO_QTASK(pInfo, idx), pInfo->suid, msgSize);
836

C
Cary Xu 已提交
837 838 839
#if 0
  for (int32_t i = 0; i < msgSize; ++i) {
    SSubmitReq *pReq = *(SSubmitReq **)((char *)pMsg + i * sizeof(void *));
C
Cary Xu 已提交
840
    smaDebug("vgId:%d, [%d][%d] version %" PRIi64, SMA_VID(pSma), msgSize, i, pReq->version);
C
Cary Xu 已提交
841 842 843
    tdRsmaPrintSubmitReq(pSma, pReq);
  }
#endif
K
kailixu 已提交
844
  if ((terrno = qSetSMAInput(qTaskInfo, pMsg, msgSize, inputType)) < 0) {
845
    smaError("vgId:%d, rsma %" PRIi8 " qSetStreamInput failed since %s", SMA_VID(pSma), level, tstrerror(terrno));
846 847 848
    return TSDB_CODE_FAILED;
  }

C
Cary Xu 已提交
849
  SRSmaInfoItem *pItem = RSMA_INFO_ITEM(pInfo, idx);
C
Cary Xu 已提交
850
  tdRSmaExecAndSubmitResult(pSma, qTaskInfo, pItem, pInfo->pTSchema, pInfo->suid);
851 852 853 854

  return TSDB_CODE_SUCCESS;
}

K
kailixu 已提交
855
#if 0
856 857
static int32_t tdCloneQTaskInfo(SSma *pSma, qTaskInfo_t dstTaskInfo, qTaskInfo_t srcTaskInfo, SRSmaParam *param,
                                tb_uid_t suid, int8_t idx) {
K
kailixu 已提交
858 859
  int32_t code = 0;
  int32_t lino = 0;
860 861 862 863
  SVnode *pVnode = pSma->pVnode;
  char   *pOutput = NULL;
  int32_t len = 0;

K
kailixu 已提交
864
  if (!srcTaskInfo) {
K
kailixu 已提交
865
    code = TSDB_CODE_INVALID_PTR;
K
kailixu 已提交
866
    smaWarn("vgId:%d, rsma clone, table %" PRIi64 ", no need since srcTaskInfo is NULL", TD_VID(pVnode), suid);
K
kailixu 已提交
867
    TSDB_CHECK_CODE(code, lino, _exit);
K
kailixu 已提交
868 869
  }

K
kailixu 已提交
870 871
  code = qSerializeTaskStatus(srcTaskInfo, &pOutput, &len);
  TSDB_CHECK_CODE(code, lino, _exit);
872 873 874 875 876 877

  SReadHandle handle = {
      .meta = pVnode->pMeta,
      .vnode = pVnode,
      .initTqReader = 1,
  };
K
kailixu 已提交
878

K
kailixu 已提交
879
  if (ASSERTS(!dstTaskInfo, "dstTaskInfo:%p is not NULL", dstTaskInfo)) {
K
kailixu 已提交
880 881 882 883
    code = TSDB_CODE_APP_ERROR;
    TSDB_CHECK_CODE(code, lino, _exit);
  }

884
  dstTaskInfo = qCreateStreamExecTaskInfo(param->qmsg[idx], &handle, TD_VID(pVnode));
885
  if (!dstTaskInfo) {
K
kailixu 已提交
886 887
    code = TSDB_CODE_RSMA_QTASKINFO_CREATE;
    TSDB_CHECK_CODE(code, lino, _exit);
888 889
  }

K
kailixu 已提交
890 891
  code = qDeserializeTaskStatus(dstTaskInfo, pOutput, len);
  TSDB_CHECK_CODE(code, lino, _exit);
892 893 894

  smaDebug("vgId:%d, rsma clone, restore rsma task for table:%" PRIi64 " succeed", TD_VID(pVnode), suid);

K
kailixu 已提交
895
_exit:
896
  taosMemoryFreeClear(pOutput);
K
kailixu 已提交
897 898 899 900 901 902
  if (code) {
    tdRSmaQTaskInfoFree(dstTaskInfo, TD_VID(pVnode), idx + 1);
    smaError("vgId:%d, rsma clone, restore rsma task for table:%" PRIi64 " failed since %s", TD_VID(pVnode), suid,
             terrstr());
  }
  return code;
903
}
K
kailixu 已提交
904
#endif
905 906 907 908 909 910 911 912 913

/**
 * @brief Clone qTaskInfo of SRSmaInfo
 *
 * @param pSma
 * @param pInfo
 * @return int32_t
 */
static int32_t tdRSmaInfoClone(SSma *pSma, SRSmaInfo *pInfo) {
K
kailixu 已提交
914 915
  int32_t     code = 0;
  int32_t     lino = 0;
916
  SRSmaParam *param = NULL;
K
kailixu 已提交
917 918
  SMetaReader mr = {0};

919 920 921 922 923 924
  if (!pInfo) {
    return TSDB_CODE_SUCCESS;
  }

  metaReaderInit(&mr, SMA_META(pSma), 0);
  smaDebug("vgId:%d, rsma clone qTaskInfo for suid:%" PRIi64, SMA_VID(pSma), pInfo->suid);
H
Haojun Liao 已提交
925
  if (metaGetTableEntryByUidCache(&mr, pInfo->suid) < 0) {
K
kailixu 已提交
926 927 928
    code = terrno;
    TSDB_CHECK_CODE(code, lino, _exit);
  }
K
kailixu 已提交
929

K
kailixu 已提交
930 931 932 933 934 935 936
  if (mr.me.type != TSDB_SUPER_TABLE) {
    code = TSDB_CODE_RSMA_INVALID_SCHEMA;
    TSDB_CHECK_CODE(code, lino, _exit);
  }
  if (mr.me.uid != pInfo->suid) {
    code = TSDB_CODE_RSMA_INVALID_SCHEMA;
    TSDB_CHECK_CODE(code, lino, _exit);
937
  }
K
kailixu 已提交
938

939 940
  if (TABLE_IS_ROLLUP(mr.me.flags)) {
    param = &mr.me.stbEntry.rsmaParam;
K
kailixu 已提交
941
#if 0
942 943 944 945
    for (int32_t i = 0; i < TSDB_RETENTION_L2; ++i) {
      if (!pInfo->iTaskInfo[i]) {
        continue;
      }
K
kailixu 已提交
946 947
      code = tdCloneQTaskInfo(pSma, pInfo->taskInfo[i], pInfo->iTaskInfo[i], param, pInfo->suid, i);
      TSDB_CHECK_CODE(code, lino, _exit);
948 949
    }
    smaDebug("vgId:%d, rsma clone env success for %" PRIi64, SMA_VID(pSma), pInfo->suid);
K
kailixu 已提交
950
#endif
951
  } else {
K
kailixu 已提交
952 953
    code = TSDB_CODE_RSMA_INVALID_SCHEMA;
    TSDB_CHECK_CODE(code, lino, _exit);
954 955
  }

K
kailixu 已提交
956 957
_exit:
  if (code) {
K
kailixu 已提交
958 959
    smaError("vgId:%d, %s failed at line %d since %s, suid:%" PRIi64 ", flags:%" PRIi8 ",type:%" PRIi8 ", uid:%" PRIi64,
             SMA_VID(pSma), __func__, lino, tstrerror(code), pInfo->suid, mr.me.flags, mr.me.type, mr.me.uid);
K
kailixu 已提交
960
  }
961
  metaReaderClear(&mr);
K
kailixu 已提交
962
  return code;
963 964
}

C
Cary Xu 已提交
965 966
/**
 * @brief During async commit, the SRSmaInfo object would be COW from iRSmaInfoHash and write lock should be applied.
C
Cary Xu 已提交
967 968 969 970
 *
 * @param pSma
 * @param suid
 * @return SRSmaInfo*
C
Cary Xu 已提交
971
 */
C
Cary Xu 已提交
972
static SRSmaInfo *tdAcquireRSmaInfoBySuid(SSma *pSma, int64_t suid) {
K
kailixu 已提交
973 974
  int32_t    code = 0;
  int32_t    lino = 0;
975 976
  SSmaEnv   *pEnv = SMA_RSMA_ENV(pSma);
  SRSmaStat *pStat = NULL;
C
Cary Xu 已提交
977 978
  SRSmaInfo *pRSmaInfo = NULL;

K
kailixu 已提交
979 980
  terrno = 0;

981
  if (!pEnv) {
C
Cary Xu 已提交
982
    terrno = TSDB_CODE_RSMA_INVALID_ENV;
983
    return NULL;
984 985
  }

986 987
  pStat = (SRSmaStat *)SMA_ENV_STAT(pEnv);
  if (!pStat || !RSMA_INFO_HASH(pStat)) {
C
Cary Xu 已提交
988
    terrno = TSDB_CODE_RSMA_INVALID_STAT;
989 990
    return NULL;
  }
991

992
  taosRLockLatch(SMA_ENV_LOCK(pEnv));
C
Cary Xu 已提交
993 994
  pRSmaInfo = taosHashGet(RSMA_INFO_HASH(pStat), &suid, sizeof(tb_uid_t));
  if (pRSmaInfo && (pRSmaInfo = *(SRSmaInfo **)pRSmaInfo)) {
C
Cary Xu 已提交
995
    if (RSMA_INFO_IS_DEL(pRSmaInfo)) {
996
      taosRUnLockLatch(SMA_ENV_LOCK(pEnv));
C
Cary Xu 已提交
997 998
      return NULL;
    }
C
Cary Xu 已提交
999
    if (!pRSmaInfo->taskInfo[0]) {
K
kailixu 已提交
1000
      if ((terrno = tdRSmaInfoClone(pSma, pRSmaInfo)) < 0) {
1001
        taosRUnLockLatch(SMA_ENV_LOCK(pEnv));
C
Cary Xu 已提交
1002 1003 1004
        return NULL;
      }
    }
C
Cary Xu 已提交
1005
    tdRefRSmaInfo(pSma, pRSmaInfo);
1006
    taosRUnLockLatch(SMA_ENV_LOCK(pEnv));
K
kailixu 已提交
1007
    if (ASSERTS(pRSmaInfo->suid == suid, "suid:%" PRIi64 " != %" PRIi64, pRSmaInfo->suid, suid)) {
K
kailixu 已提交
1008 1009 1010
      terrno = TSDB_CODE_APP_ERROR;
      return NULL;
    }
C
Cary Xu 已提交
1011 1012
    return pRSmaInfo;
  }
1013
  taosRUnLockLatch(SMA_ENV_LOCK(pEnv));
C
Cary Xu 已提交
1014

C
Cary Xu 已提交
1015
  return NULL;
1016 1017
}

C
Cary Xu 已提交
1018 1019 1020
static FORCE_INLINE void tdReleaseRSmaInfo(SSma *pSma, SRSmaInfo *pInfo) {
  if (pInfo) {
    tdUnRefRSmaInfo(pSma, pInfo);
C
Cary Xu 已提交
1021 1022 1023
  }
}

C
Cary Xu 已提交
1024 1025 1026 1027
/**
 * @brief async mode
 *
 * @param pSma
K
kailixu 已提交
1028
 * @param version
C
Cary Xu 已提交
1029 1030 1031 1032 1033
 * @param pMsg
 * @param inputType
 * @param suid
 * @return int32_t
 */
K
kailixu 已提交
1034 1035
static int32_t tdExecuteRSmaAsync(SSma *pSma, int64_t version, const void *pMsg, int32_t len, int32_t inputType,
                                  tb_uid_t suid) {
C
Cary Xu 已提交
1036 1037 1038 1039 1040 1041 1042
  SRSmaInfo *pRSmaInfo = tdAcquireRSmaInfoBySuid(pSma, suid);
  if (!pRSmaInfo) {
    smaDebug("vgId:%d, execute rsma, no rsma info for suid:%" PRIu64, SMA_VID(pSma), suid);
    return TSDB_CODE_SUCCESS;
  }

  if (inputType == STREAM_INPUT__DATA_SUBMIT) {
K
kailixu 已提交
1043
    if (tdExecuteRSmaImplAsync(pSma, version, pMsg, len, inputType, pRSmaInfo, suid) < 0) {
C
Cary Xu 已提交
1044 1045 1046
      tdReleaseRSmaInfo(pSma, pRSmaInfo);
      return TSDB_CODE_FAILED;
    }
C
Cary Xu 已提交
1047 1048 1049 1050 1051 1052 1053 1054 1055 1056
    if (smaMgmt.tmrHandle) {
      SRSmaInfoItem *pItem = RSMA_INFO_ITEM(pRSmaInfo, 0);
      if (pItem->level > 0) {
        atomic_store_8(&pItem->triggerStat, TASK_TRIGGER_STAT_ACTIVE);
      }
      pItem = RSMA_INFO_ITEM(pRSmaInfo, 1);
      if (pItem->level > 0) {
        atomic_store_8(&pItem->triggerStat, TASK_TRIGGER_STAT_ACTIVE);
      }
    }
C
Cary Xu 已提交
1057
  } else {
K
kailixu 已提交
1058 1059 1060 1061 1062
    terrno = TSDB_CODE_APP_ERROR;
    tdReleaseRSmaInfo(pSma, pRSmaInfo);
    smaError("vgId:%d, execute rsma, failed for suid:%" PRIu64 " since %s, type:%d", SMA_VID(pSma), suid,
             tstrerror(terrno), inputType);
    return TSDB_CODE_FAILED;
C
Cary Xu 已提交
1063 1064 1065 1066 1067 1068
  }

  tdReleaseRSmaInfo(pSma, pRSmaInfo);
  return TSDB_CODE_SUCCESS;
}

K
kailixu 已提交
1069
int32_t tdProcessRSmaSubmit(SSma *pSma, int64_t version, void *pReq, void *pMsg, int32_t len, int32_t inputType) {
1070 1071 1072 1073 1074
  SSmaEnv *pEnv = SMA_RSMA_ENV(pSma);
  if (!pEnv) {
    // only applicable when rsma env exists
    return TSDB_CODE_SUCCESS;
  }
K
kailixu 已提交
1075

C
Cary Xu 已提交
1076
  STbUidStore uidStore = {0};
1077

L
Liu Jicong 已提交
1078
  if (inputType == STREAM_INPUT__DATA_SUBMIT) {
K
kailixu 已提交
1079
    if (tdFetchSubmitReqSuids(pReq, &uidStore) < 0) {
1080
      smaError("vgId:%d, failed to process rsma submit fetch suid since: %s", SMA_VID(pSma), terrstr());
C
Cary Xu 已提交
1081 1082
      goto _err;
    }
1083 1084

    if (uidStore.suid != 0) {
K
kailixu 已提交
1085
      if (tdExecuteRSmaAsync(pSma, version, pMsg, len, inputType, uidStore.suid) < 0) {
1086
        smaError("vgId:%d, failed to process rsma submit exec 1 since: %s", SMA_VID(pSma), terrstr());
C
Cary Xu 已提交
1087 1088
        goto _err;
      }
1089

C
Cary Xu 已提交
1090 1091
      void *pIter = NULL;
      while ((pIter = taosHashIterate(uidStore.uidHash, pIter))) {
1092
        tb_uid_t *pTbSuid = (tb_uid_t *)taosHashGetKey(pIter, NULL);
K
kailixu 已提交
1093
        if (tdExecuteRSmaAsync(pSma, version, pMsg, len, inputType, *pTbSuid) < 0) {
1094
          smaError("vgId:%d, failed to process rsma submit exec 2 since: %s", SMA_VID(pSma), terrstr());
C
Cary Xu 已提交
1095 1096
          goto _err;
        }
1097 1098 1099
      }
    }
  }
C
Cary Xu 已提交
1100
  tdUidStoreDestory(&uidStore);
1101
  return TSDB_CODE_SUCCESS;
C
Cary Xu 已提交
1102 1103 1104
_err:
  tdUidStoreDestory(&uidStore);
  return TSDB_CODE_FAILED;
1105
}
C
Cary Xu 已提交
1106

C
Cary Xu 已提交
1107 1108 1109 1110 1111 1112 1113
/**
 * @brief retrieve rsma meta and init
 *
 * @param pSma
 * @param nTables number of tables of rsma
 * @return int32_t
 */
C
Cary Xu 已提交
1114
static int32_t tdRSmaRestoreQTaskInfoInit(SSma *pSma, int64_t *nTables) {
K
kailixu 已提交
1115 1116
  int32_t     code = 0;
  int32_t     lino = 0;
C
Cary Xu 已提交
1117 1118 1119 1120
  SVnode     *pVnode = pSma->pVnode;
  SArray     *suidList = NULL;
  STbUidStore uidStore = {0};
  SMetaReader mr = {0};
K
kailixu 已提交
1121
  tb_uid_t    suid = 0;
1122

C
Cary Xu 已提交
1123
  if (!(suidList = taosArrayInit(1, sizeof(tb_uid_t)))) {
K
kailixu 已提交
1124 1125
    code = TSDB_CODE_OUT_OF_MEMORY;
    TSDB_CHECK_CODE(code, lino, _exit);
C
Cary Xu 已提交
1126 1127 1128
  }

  if (vnodeGetStbIdList(pSma->pVnode, 0, suidList) < 0) {
K
kailixu 已提交
1129 1130
    code = terrno;
    TSDB_CHECK_CODE(code, lino, _exit);
1131 1132
  }

C
Cary Xu 已提交
1133 1134
  int64_t arrSize = taosArrayGetSize(suidList);

C
Cary Xu 已提交
1135
  if (arrSize == 0) {
C
Cary Xu 已提交
1136 1137 1138
    if (nTables) {
      *nTables = 0;
    }
C
Cary Xu 已提交
1139 1140
    taosArrayDestroy(suidList);
    smaDebug("vgId:%d, no need to restore rsma env since empty stb id list", TD_VID(pVnode));
1141 1142 1143
    return TSDB_CODE_SUCCESS;
  }

C
Cary Xu 已提交
1144
  int64_t nRsmaTables = 0;
1145
  metaReaderInit(&mr, SMA_META(pSma), 0);
C
Cary Xu 已提交
1146
  if (!(uidStore.tbUids = taosArrayInit(1024, sizeof(tb_uid_t)))) {
K
kailixu 已提交
1147 1148
    code = TSDB_CODE_OUT_OF_MEMORY;
    TSDB_CHECK_CODE(code, lino, _exit);
C
Cary Xu 已提交
1149
  }
K
kailixu 已提交
1150

C
Cary Xu 已提交
1151
  for (int64_t i = 0; i < arrSize; ++i) {
K
kailixu 已提交
1152
    suid = *(tb_uid_t *)taosArrayGet(suidList, i);
C
Cary Xu 已提交
1153
    smaDebug("vgId:%d, rsma restore, suid is %" PRIi64, TD_VID(pVnode), suid);
H
Haojun Liao 已提交
1154
    if (metaGetTableEntryByUidCache(&mr, suid) < 0) {
K
kailixu 已提交
1155 1156
      code = terrno;
      TSDB_CHECK_CODE(code, lino, _exit);
1157
    }
1158
    tDecoderClear(&mr.coder);
K
kailixu 已提交
1159 1160 1161 1162 1163 1164 1165 1166
    if (mr.me.type != TSDB_SUPER_TABLE) {
      code = TSDB_CODE_RSMA_INVALID_SCHEMA;
      TSDB_CHECK_CODE(code, lino, _exit);
    }
    if (mr.me.uid != suid) {
      code = TSDB_CODE_RSMA_INVALID_SCHEMA;
      TSDB_CHECK_CODE(code, lino, _exit);
    }
1167
    if (TABLE_IS_ROLLUP(mr.me.flags)) {
C
Cary Xu 已提交
1168
      ++nRsmaTables;
1169
      SRSmaParam *param = &mr.me.stbEntry.rsmaParam;
C
Cary Xu 已提交
1170 1171 1172 1173
      for (int i = 0; i < TSDB_RETENTION_L2; ++i) {
        smaDebug("vgId:%d, rsma restore, table:%" PRIi64 " level:%d, maxdelay:%" PRIi64 " watermark:%" PRIi64
                 " qmsgLen:%" PRIi32,
                 TD_VID(pVnode), suid, i, param->maxdelay[i], param->watermark[i], param->qmsgLen[i]);
1174
      }
1175
      if (tdRSmaProcessCreateImpl(pSma, &mr.me.stbEntry.rsmaParam, suid, mr.me.name) < 0) {
K
kailixu 已提交
1176 1177
        code = terrno;
        TSDB_CHECK_CODE(code, lino, _exit);
1178
      }
C
Cary Xu 已提交
1179 1180 1181 1182

      // reload all ctbUids for suid
      uidStore.suid = suid;
      if (vnodeGetCtbIdList(pVnode, suid, uidStore.tbUids) < 0) {
K
kailixu 已提交
1183 1184
        code = terrno;
        TSDB_CHECK_CODE(code, lino, _exit);
C
Cary Xu 已提交
1185 1186
      }

1187
      if (tdUpdateTbUidList(pVnode->pSma, &uidStore, true) < 0) {
K
kailixu 已提交
1188 1189
        code = terrno;
        TSDB_CHECK_CODE(code, lino, _exit);
C
Cary Xu 已提交
1190 1191 1192 1193
      }

      taosArrayClear(uidStore.tbUids);

C
Cary Xu 已提交
1194
      smaDebug("vgId:%d, rsma restore env success for %" PRIi64, TD_VID(pVnode), suid);
1195 1196 1197
    }
  }

C
Cary Xu 已提交
1198 1199 1200
  if (nTables) {
    *nTables = nRsmaTables;
  }
K
kailixu 已提交
1201 1202 1203 1204 1205
_exit:
  if (code) {
    smaError("vgId:%d, %s failed at line %d since %s, suid:%" PRIi64 ", type:%" PRIi8 ", uid:%" PRIi64, TD_VID(pVnode),
             __func__, lino, tstrerror(code), suid, mr.me.type, mr.me.uid);
  }
C
Cary Xu 已提交
1206 1207
  metaReaderClear(&mr);
  taosArrayDestroy(suidList);
C
Cary Xu 已提交
1208
  tdUidStoreDestory(&uidStore);
K
kailixu 已提交
1209
  return code;
C
Cary Xu 已提交
1210 1211 1212
}

/**
K
kailixu 已提交
1213
 * N.B. the data would be restored from the unified WAL replay procedure
C
Cary Xu 已提交
1214
 */
K
kailixu 已提交
1215
int32_t tdRSmaProcessRestoreImpl(SSma *pSma, int8_t type, int64_t qtaskFileVer, int8_t rollback) {
K
kailixu 已提交
1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227
  // step 1: init env
  if (tdCheckAndInitSmaEnv(pSma, TSDB_SMA_TYPE_ROLLUP) != TSDB_CODE_SUCCESS) {
    terrno = TSDB_CODE_TDB_INIT_FAILED;
    return TSDB_CODE_FAILED;
  }

  // step 2: open SRSmaFS for qTaskFiles
  if (tdRSmaFSOpen(pSma, qtaskFileVer, rollback) < 0) {
    goto _err;
  }

  // step 3: iterate all stables to restore the rsma env
C
Cary Xu 已提交
1228
  int64_t nTables = 0;
C
Cary Xu 已提交
1229
  if (tdRSmaRestoreQTaskInfoInit(pSma, &nTables) < 0) {
C
Cary Xu 已提交
1230 1231
    goto _err;
  }
C
Cary Xu 已提交
1232
  if (nTables <= 0) {
1233
    smaDebug("vgId:%d, no need to restore rsma task %" PRIi8 " since no tables", SMA_VID(pSma), type);
C
Cary Xu 已提交
1234 1235
    return TSDB_CODE_SUCCESS;
  }
C
Cary Xu 已提交
1236

1237
  smaInfo("vgId:%d, restore rsma task %" PRIi8 " from qtaskf %" PRIi64 " succeed", SMA_VID(pSma), type, qtaskFileVer);
1238 1239
  return TSDB_CODE_SUCCESS;
_err:
1240 1241
  smaError("vgId:%d, restore rsma task %" PRIi8 "from qtaskf %" PRIi64 " failed since %s", SMA_VID(pSma), type,
           qtaskFileVer, terrstr());
1242 1243 1244
  return TSDB_CODE_FAILED;
}

1245
int32_t tdRSmaPersistExecImpl(SRSmaStat *pRSmaStat, SHashObj *pInfoHash) {
K
kailixu 已提交
1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256
  int32_t   code = 0;
  int32_t   lino = 0;
  SSma     *pSma = pRSmaStat->pSma;
  SVnode   *pVnode = pSma->pVnode;
  SArray   *qTaskFArray = NULL;
  int64_t   version = pRSmaStat->commitAppliedVer;
  TdFilePtr pOutFD = NULL;
  TdFilePtr pInFD = NULL;
  char      fname[TSDB_FILENAME_LEN];
  char      fnameVer[TSDB_FILENAME_LEN];
  SRSmaFS   fs = {0};
1257 1258 1259 1260 1261

  if (taosHashGetSize(pInfoHash) <= 0) {
    return TSDB_CODE_SUCCESS;
  }

K
kailixu 已提交
1262 1263 1264 1265
  qTaskFArray = taosArrayInit(taosHashGetSize(pInfoHash) << 1, sizeof(SQTaskFile));
  if (!qTaskFArray) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    TSDB_CHECK_CODE(code, lino, _exit);
1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279
  }

  void *infoHash = NULL;
  while ((infoHash = taosHashIterate(pInfoHash, infoHash))) {
    SRSmaInfo *pRSmaInfo = *(SRSmaInfo **)infoHash;

    if (RSMA_INFO_IS_DEL(pRSmaInfo)) {
      continue;
    }

    for (int32_t i = 0; i < TSDB_RETENTION_L2; ++i) {
      SRSmaInfoItem *pItem = RSMA_INFO_ITEM(pRSmaInfo, i);
      if (pItem && pItem->pStreamState) {
        if (streamStateCommit(pItem->pStreamState) < 0) {
K
kailixu 已提交
1280 1281 1282
          code = TSDB_CODE_RSMA_STREAM_STATE_COMMIT;
          TSDB_CHECK_CODE(code, lino, _exit);
        }
1283
        smaDebug("vgId:%d, rsma persist, stream state commit success, table %" PRIi64 ", level %d", TD_VID(pVnode),
K
kailixu 已提交
1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304
                 pRSmaInfo->suid, i + 1);

        // qTaskInfo file
        tdRSmaQTaskInfoGetFullName(TD_VID(pVnode), pRSmaInfo->suid, i + 1, -1, tfsGetPrimaryPath(pVnode->pTfs), fname);
        tdRSmaQTaskInfoGetFullName(TD_VID(pVnode), pRSmaInfo->suid, i + 1, version, tfsGetPrimaryPath(pVnode->pTfs),
                                   fnameVer);
        if (taosCheckExistFile(fnameVer)) {
          smaWarn("vgId:%d, rsma persist, duplicate file %s exist", TD_VID(pVnode), fnameVer);
        }

        pOutFD = taosCreateFile(fnameVer, TD_FILE_WRITE | TD_FILE_CREATE | TD_FILE_TRUNC);
        if (pOutFD == NULL) {
          code = TAOS_SYSTEM_ERROR(errno);
          TSDB_CHECK_CODE(code, lino, _exit);
        }
        pInFD = taosOpenFile(fname, TD_FILE_READ);
        if (pInFD == NULL) {
          code = TAOS_SYSTEM_ERROR(errno);
          TSDB_CHECK_CODE(code, lino, _exit);
        }

1305 1306 1307
        int64_t  size = 0;
        uint32_t mtime = 0;
        if (taosFStatFile(pInFD, &size, &mtime) < 0) {
K
kailixu 已提交
1308 1309
          code = TAOS_SYSTEM_ERROR(errno);
          TSDB_CHECK_CODE(code, lino, _exit);
1310
        }
K
kailixu 已提交
1311 1312 1313 1314

        int64_t offset = 0;
        if (taosFSendFile(pOutFD, pInFD, &offset, size) < 0) {
          code = TAOS_SYSTEM_ERROR(errno);
1315 1316
          smaError("vgId:%d, rsma persist, send qtaskinfo file %s to %s failed since %s", TD_VID(pVnode), fname,
                   fnameVer, tstrerror(code));
K
kailixu 已提交
1317 1318 1319 1320 1321
          TSDB_CHECK_CODE(code, lino, _exit);
        }
        taosCloseFile(&pOutFD);
        taosCloseFile(&pInFD);

1322 1323 1324
        SQTaskFile qTaskF = {
            .nRef = 1, .level = i + 1, .suid = pRSmaInfo->suid, .version = version, .size = size, .mtime = mtime};

K
kailixu 已提交
1325
        taosArrayPush(qTaskFArray, &qTaskF);
1326 1327 1328 1329
      }
    }
  }

K
kailixu 已提交
1330
  // prepare
K
kailixu 已提交
1331
  code = tdRSmaFSCopy(pSma, &fs);
K
kailixu 已提交
1332
  TSDB_CHECK_CODE(code, lino, _exit);
1333

K
kailixu 已提交
1334
  code = tdRSmaFSUpsertQTaskFile(pSma, &fs, qTaskFArray->pData, taosArrayGetSize(qTaskFArray));
K
kailixu 已提交
1335
  TSDB_CHECK_CODE(code, lino, _exit);
1336

K
kailixu 已提交
1337 1338 1339 1340 1341 1342 1343
  code = tdRSmaFSPrepareCommit(pSma, &fs);
  TSDB_CHECK_CODE(code, lino, _exit);

_exit:

  taosArrayDestroy(fs.aQTaskInf);
  taosArrayDestroy(qTaskFArray);
1344

K
kailixu 已提交
1345 1346 1347 1348 1349
  if (code) {
    if (pOutFD) taosCloseFile(&pOutFD);
    if (pInFD) taosCloseFile(&pInFD);
    smaError("vgId:%d, %s failed at line %d since %s", TD_VID(pVnode), __func__, lino, tstrerror(code));
  }
1350

K
kailixu 已提交
1351 1352
  terrno = code;
  return code;
1353 1354
}

C
Cary Xu 已提交
1355
/**
C
Cary Xu 已提交
1356
 * @brief trigger to get rsma result in async mode
C
Cary Xu 已提交
1357 1358 1359 1360
 *
 * @param param
 * @param tmrId
 */
C
Cary Xu 已提交
1361
static void tdRSmaFetchTrigger(void *param, void *tmrId) {
1362
  SRSmaRef      *pRSmaRef = NULL;
C
Cary Xu 已提交
1363
  SSma          *pSma = NULL;
1364 1365 1366
  SRSmaStat     *pStat = NULL;
  SRSmaInfo     *pRSmaInfo = NULL;
  SRSmaInfoItem *pItem = NULL;
C
Cary Xu 已提交
1367

1368
  if (!(pRSmaRef = taosHashGet(smaMgmt.refHash, &param, POINTER_BYTES))) {
C
Cary Xu 已提交
1369 1370
    smaDebug("rsma fetch task not start since rsma info item:%p not exist in refHash:%p, rsetId:%d", param,
             smaMgmt.refHash, smaMgmt.rsetId);
C
Cary Xu 已提交
1371 1372 1373
    return;
  }

1374
  if (!(pStat = (SRSmaStat *)tdAcquireSmaRef(smaMgmt.rsetId, pRSmaRef->refId))) {
K
kailixu 已提交
1375
    smaWarn("rsma fetch task not start since rsma stat already destroyed, rsetId:%d refId:%" PRIi64 ")",
C
Cary Xu 已提交
1376 1377
             smaMgmt.rsetId, pRSmaRef->refId);  // pRSmaRef freed in taosHashRemove
    taosHashRemove(smaMgmt.refHash, &param, POINTER_BYTES);
C
Cary Xu 已提交
1378
    return;
C
Cary Xu 已提交
1379 1380
  }

C
Cary Xu 已提交
1381 1382
  pSma = pStat->pSma;

1383
  if (!(pRSmaInfo = tdAcquireRSmaInfoBySuid(pSma, pRSmaRef->suid))) {
C
Cary Xu 已提交
1384
    smaDebug("rsma fetch task not start since rsma info not exist, rsetId:%d refId:%" PRIi64 ")", smaMgmt.rsetId,
1385 1386 1387 1388 1389 1390 1391
             pRSmaRef->refId);  // pRSmaRef freed in taosHashRemove
    tdReleaseSmaRef(smaMgmt.rsetId, pRSmaRef->refId);
    taosHashRemove(smaMgmt.refHash, &param, POINTER_BYTES);
    return;
  }

  if (RSMA_INFO_IS_DEL(pRSmaInfo)) {
C
Cary Xu 已提交
1392
    smaDebug("rsma fetch task not start since rsma info already deleted, rsetId:%d refId:%" PRIi64 ")", smaMgmt.rsetId,
1393 1394 1395 1396 1397 1398 1399 1400 1401
             pRSmaRef->refId);  // pRSmaRef freed in taosHashRemove
    tdReleaseRSmaInfo(pSma, pRSmaInfo);
    tdReleaseSmaRef(smaMgmt.rsetId, pRSmaRef->refId);
    taosHashRemove(smaMgmt.refHash, &param, POINTER_BYTES);
    return;
  }

  pItem = *(SRSmaInfoItem **)&param;

C
Cary Xu 已提交
1402 1403 1404 1405 1406
  // if rsma trigger stat in paused, cancelled or finished, not start fetch task
  int8_t rsmaTriggerStat = atomic_load_8(RSMA_TRIGGER_STAT(pStat));
  switch (rsmaTriggerStat) {
    case TASK_TRIGGER_STAT_PAUSED:
    case TASK_TRIGGER_STAT_CANCELLED: {
C
Cary Xu 已提交
1407
      smaDebug("vgId:%d, rsma fetch task not start for level %" PRIi8 " since stat is %" PRIi8
C
Cary Xu 已提交
1408
               ", rsetId:%d refId:%" PRIi64,
1409
               SMA_VID(pSma), pItem->level, rsmaTriggerStat, smaMgmt.rsetId, pRSmaRef->refId);
C
Cary Xu 已提交
1410
      if (rsmaTriggerStat == TASK_TRIGGER_STAT_PAUSED) {
1411
        taosTmrReset(tdRSmaFetchTrigger, RSMA_FETCH_INTERVAL, pItem, smaMgmt.tmrHandle, &pItem->tmrId);
C
Cary Xu 已提交
1412
      }
1413 1414
      tdReleaseRSmaInfo(pSma, pRSmaInfo);
      tdReleaseSmaRef(smaMgmt.rsetId, pRSmaRef->refId);
C
Cary Xu 已提交
1415
      return;
C
Cary Xu 已提交
1416
    }
C
Cary Xu 已提交
1417 1418 1419 1420 1421 1422 1423 1424
    default:
      break;
  }

  int8_t fetchTriggerStat =
      atomic_val_compare_exchange_8(&pItem->triggerStat, TASK_TRIGGER_STAT_ACTIVE, TASK_TRIGGER_STAT_INACTIVE);
  switch (fetchTriggerStat) {
    case TASK_TRIGGER_STAT_ACTIVE: {
1425
      smaDebug("vgId:%d, rsma fetch task planned for level:%" PRIi8 " suid:%" PRIi64 " since stat is active",
C
Cary Xu 已提交
1426
               SMA_VID(pSma), pItem->level, pRSmaInfo->suid);
C
Cary Xu 已提交
1427
      // async process
C
Cary Xu 已提交
1428
      pItem->fetchLevel = pItem->level;
C
Cary Xu 已提交
1429
#if 0
K
kailixu 已提交
1430
      // debugging codes
C
Cary Xu 已提交
1431 1432
      SRSmaInfo     *qInfo = tdAcquireRSmaInfoBySuid(pSma, pRSmaInfo->suid);
      SRSmaInfoItem *qItem = RSMA_INFO_ITEM(qInfo, pItem->level - 1);
K
kailixu 已提交
1433 1434
      make sure(qItem->level == pItem->level);
      make sure(qItem->fetchLevel == pItem->fetchLevel);
C
Cary Xu 已提交
1435
#endif
1436 1437 1438
      if (atomic_load_8(&pRSmaInfo->assigned) == 0) {
        tsem_post(&(pStat->notEmpty));
      }
C
Cary Xu 已提交
1439 1440
    } break;
    case TASK_TRIGGER_STAT_INACTIVE: {
K
kailixu 已提交
1441
      smaDebug("vgId:%d, rsma fetch task not start for level:%" PRIi8 " suid:%" PRIi64 " since stat is inactive ",
C
Cary Xu 已提交
1442 1443 1444
               SMA_VID(pSma), pItem->level, pRSmaInfo->suid);
    } break;
    case TASK_TRIGGER_STAT_INIT: {
C
Cary Xu 已提交
1445 1446
      smaDebug("vgId:%d, rsma fetch task not start for level:%" PRIi8 " suid::%" PRIi64 " since stat is init",
               SMA_VID(pSma), pItem->level, pRSmaInfo->suid);
C
Cary Xu 已提交
1447 1448
    } break;
    default: {
K
kailixu 已提交
1449 1450 1451
      smaDebug("vgId:%d, rsma fetch task not start for level:%" PRIi8 " suid:%" PRIi64 " since stat:%" PRIi8
               " is unknown",
               SMA_VID(pSma), pItem->level, pRSmaInfo->suid, fetchTriggerStat);
C
Cary Xu 已提交
1452
    } break;
C
Cary Xu 已提交
1453 1454 1455
  }

_end:
C
Cary Xu 已提交
1456
  taosTmrReset(tdRSmaFetchTrigger, pItem->maxDelay, pItem, smaMgmt.tmrHandle, &pItem->tmrId);
1457 1458
  tdReleaseRSmaInfo(pSma, pRSmaInfo);
  tdReleaseSmaRef(smaMgmt.rsetId, pRSmaRef->refId);
C
Cary Xu 已提交
1459
}
C
Cary Xu 已提交
1460

C
Cary Xu 已提交
1461 1462
static void tdFreeRSmaSubmitItems(SArray *pItems) {
  for (int32_t i = 0; i < taosArrayGetSize(pItems); ++i) {
K
kailixu 已提交
1463
    SPackedData *packData = taosArrayGet(pItems, i);
K
kailixu 已提交
1464
    taosFreeQitem(POINTER_SHIFT(packData->msgStr, -sizeof(int32_t) - sizeof(int64_t)));
C
Cary Xu 已提交
1465
  }
C
Cary Xu 已提交
1466
  taosArrayClear(pItems);
C
Cary Xu 已提交
1467 1468
}

C
Cary Xu 已提交
1469 1470 1471 1472 1473 1474 1475
/**
 * @brief fetch rsma result(consider the efficiency and functionality)
 *
 * @param pSma
 * @param pInfo
 * @return int32_t
 */
1476
static int32_t tdRSmaFetchAllResult(SSma *pSma, SRSmaInfo *pInfo) {
C
Cary Xu 已提交
1477 1478
  SSDataBlock dataBlock = {.info.type = STREAM_GET_ALL};
  for (int8_t i = 1; i <= TSDB_RETENTION_L2; ++i) {
C
Cary Xu 已提交
1479 1480 1481
    SRSmaInfoItem *pItem = RSMA_INFO_ITEM(pInfo, i - 1);
    if (pItem->fetchLevel) {
      pItem->fetchLevel = 0;
C
Cary Xu 已提交
1482 1483 1484 1485
      qTaskInfo_t taskInfo = RSMA_INFO_QTASK(pInfo, i - 1);
      if (!taskInfo) {
        continue;
      }
C
Cary Xu 已提交
1486

C
Cary Xu 已提交
1487
      if ((++pItem->nScanned * pItem->maxDelay) > RSMA_FETCH_DELAY_MAX) {
1488
        smaDebug("vgId:%d, suid:%" PRIi64 " level:%" PRIi8 " nScanned:%" PRIi16 " maxDelay:%d, fetch executed",
C
Cary Xu 已提交
1489
                 SMA_VID(pSma), pInfo->suid, i, pItem->nScanned, pItem->maxDelay);
C
Cary Xu 已提交
1490
      } else {
C
Cary Xu 已提交
1491 1492 1493 1494 1495 1496 1497 1498 1499 1500
        int64_t curMs = taosGetTimestampMs();
        if ((curMs - pInfo->lastRecv) < RSMA_FETCH_ACTIVE_MAX) {
          smaTrace("vgId:%d, suid:%" PRIi64 " level:%" PRIi8 " curMs:%" PRIi64 " lastRecv:%" PRIi64 ", fetch skipped ",
                   SMA_VID(pSma), pInfo->suid, i, curMs, pInfo->lastRecv);
          atomic_store_8(&pItem->triggerStat, TASK_TRIGGER_STAT_ACTIVE);  // restore the active stat
          continue;
        } else {
          smaDebug("vgId:%d, suid:%" PRIi64 " level:%" PRIi8 " curMs:%" PRIi64 " lastRecv:%" PRIi64 ", fetch executed ",
                   SMA_VID(pSma), pInfo->suid, i, curMs, pInfo->lastRecv);
        }
C
Cary Xu 已提交
1501 1502
      }

C
Cary Xu 已提交
1503
      pItem->nScanned = 0;
C
Cary Xu 已提交
1504

L
Liu Jicong 已提交
1505
      if ((terrno = qSetSMAInput(taskInfo, &dataBlock, 1, STREAM_INPUT__DATA_BLOCK)) < 0) {
C
Cary Xu 已提交
1506 1507
        goto _err;
      }
C
Cary Xu 已提交
1508
      if (tdRSmaExecAndSubmitResult(pSma, taskInfo, pItem, pInfo->pTSchema, pInfo->suid) < 0) {
C
Cary Xu 已提交
1509 1510 1511
        goto _err;
      }

1512
      smaDebug("vgId:%d, suid:%" PRIi64 " level:%" PRIi8 " nScanned:%" PRIi16 " maxDelay:%d, fetch finished",
C
Cary Xu 已提交
1513
               SMA_VID(pSma), pInfo->suid, i, pItem->nScanned, pItem->maxDelay);
C
Cary Xu 已提交
1514
    } else {
1515
      smaDebug("vgId:%d, suid:%" PRIi64 " level:%" PRIi8 " nScanned:%" PRIi16
C
Cary Xu 已提交
1516
               " maxDelay:%d, fetch not executed as fetch level is %" PRIi8,
C
Cary Xu 已提交
1517
               SMA_VID(pSma), pInfo->suid, i, pItem->nScanned, pItem->maxDelay, pItem->fetchLevel);
C
Cary Xu 已提交
1518 1519 1520 1521 1522 1523 1524 1525 1526
    }
  }

_end:
  return TSDB_CODE_SUCCESS;
_err:
  return TSDB_CODE_FAILED;
}

C
Cary Xu 已提交
1527 1528 1529 1530 1531 1532
static int32_t tdRSmaBatchExec(SSma *pSma, SRSmaInfo *pInfo, STaosQall *qall, SArray *pSubmitArr, ERsmaExecType type) {
  taosArrayClear(pSubmitArr);
  while (1) {
    void *msg = NULL;
    taosGetQitem(qall, (void **)&msg);
    if (msg) {
K
kailixu 已提交
1533 1534 1535 1536
      SPackedData packData = {.msgLen = *(int32_t *)msg,
                              .ver = *(int64_t *)POINTER_SHIFT(msg, sizeof(int32_t)),
                              .msgStr = POINTER_SHIFT(msg, sizeof(int32_t) + sizeof(int64_t))};

K
kailixu 已提交
1537
      if (!taosArrayPush(pSubmitArr, &packData)) {
C
Cary Xu 已提交
1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556
        tdFreeRSmaSubmitItems(pSubmitArr);
        goto _err;
      }
    } else {
      break;
    }
  }

  int32_t size = taosArrayGetSize(pSubmitArr);
  if (size > 0) {
    for (int32_t i = 1; i <= TSDB_RETENTION_L2; ++i) {
      if (tdExecuteRSmaImpl(pSma, pSubmitArr->pData, size, STREAM_INPUT__MERGED_SUBMIT, pInfo, type, i) < 0) {
        goto _err;
      }
    }
    tdFreeRSmaSubmitItems(pSubmitArr);
  }
  return TSDB_CODE_SUCCESS;
_err:
K
kailixu 已提交
1557 1558 1559
  smaError("vgId:%d, batch exec for suid:%" PRIi64 " execType:%d size:%d failed since %s", SMA_VID(pSma), pInfo->suid,
           type, (int32_t)taosArrayGetSize(pSubmitArr), terrstr());
  tdFreeRSmaSubmitItems(pSubmitArr);
C
Cary Xu 已提交
1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571
  while (1) {
    void *msg = NULL;
    taosGetQitem(qall, (void **)&msg);
    if (msg) {
      taosFreeQitem(msg);
    } else {
      break;
    }
  }
  return TSDB_CODE_FAILED;
}

C
Cary Xu 已提交
1572 1573 1574 1575
/**
 * @brief
 *
 * @param pSma
C
Cary Xu 已提交
1576
 * @param type
C
Cary Xu 已提交
1577 1578
 * @return int32_t
 */
C
Cary Xu 已提交
1579

C
Cary Xu 已提交
1580
int32_t tdRSmaProcessExecImpl(SSma *pSma, ERsmaExecType type) {
K
kailixu 已提交
1581 1582
  int32_t    code = 0;
  int32_t    lino = 0;
C
Cary Xu 已提交
1583
  SVnode    *pVnode = pSma->pVnode;
C
Cary Xu 已提交
1584 1585 1586 1587
  SSmaEnv   *pEnv = SMA_RSMA_ENV(pSma);
  SRSmaStat *pRSmaStat = (SRSmaStat *)SMA_ENV_STAT(pEnv);
  SHashObj  *infoHash = NULL;
  SArray    *pSubmitArr = NULL;
C
Cary Xu 已提交
1588
  bool       isFetchAll = false;
C
Cary Xu 已提交
1589 1590

  if (!pRSmaStat || !(infoHash = RSMA_INFO_HASH(pRSmaStat))) {
K
kailixu 已提交
1591 1592
    code = TSDB_CODE_RSMA_INVALID_STAT;
    TSDB_CHECK_CODE(code, lino, _exit);
C
Cary Xu 已提交
1593 1594
  }

C
Cary Xu 已提交
1595
  if (!(pSubmitArr =
K
kailixu 已提交
1596
            taosArrayInit(TMIN(RSMA_SUBMIT_BATCH_SIZE, atomic_load_64(&pRSmaStat->nBufItems)), sizeof(SPackedData)))) {
K
kailixu 已提交
1597 1598
    code = TSDB_CODE_OUT_OF_MEMORY;
    TSDB_CHECK_CODE(code, lino, _exit);
C
Cary Xu 已提交
1599 1600
  }

C
Cary Xu 已提交
1601 1602
  while (true) {
    // step 1: rsma exec - consume data in buffer queue for all suids
C
Cary Xu 已提交
1603
    if (type == RSMA_EXEC_OVERFLOW) {
1604 1605
      void *pIter = NULL;
      while ((pIter = taosHashIterate(infoHash, pIter))) {
C
Cary Xu 已提交
1606
        SRSmaInfo *pInfo = *(SRSmaInfo **)pIter;
1607
        if (atomic_val_compare_exchange_8(&pInfo->assigned, 0, 1) == 0) {
K
kailixu 已提交
1608
          if ((taosQueueItemSize(pInfo->queue) > 0) || RSMA_NEED_FETCH(pInfo)) {
1609 1610 1611 1612 1613
            int32_t batchCnt = -1;
            int32_t batchMax = taosHashGetSize(infoHash) / tsNumOfVnodeRsmaThreads;
            bool    occupied = (batchMax <= 1);
            if (batchMax > 1) {
              batchMax = 100 / batchMax;
C
Cary Xu 已提交
1614
              batchMax = TMAX(batchMax, 4);
C
Cary Xu 已提交
1615
            }
C
Cary Xu 已提交
1616
            while (occupied || (++batchCnt < batchMax)) {    // greedy mode
1617 1618 1619 1620
              taosReadAllQitems(pInfo->queue, pInfo->qall);  // queue has mutex lock
              int32_t qallItemSize = taosQallItemSize(pInfo->qall);
              if (qallItemSize > 0) {
                tdRSmaBatchExec(pSma, pInfo, pInfo->qall, pSubmitArr, type);
1621
                smaDebug("vgId:%d, batchSize:%d, execType:%" PRIi32, SMA_VID(pSma), qallItemSize, type);
1622 1623
              }

K
kailixu 已提交
1624
              if (RSMA_NEED_FETCH(pInfo)) {
1625 1626 1627
                int8_t oldStat = atomic_val_compare_exchange_8(RSMA_COMMIT_STAT(pRSmaStat), 0, 2);
                if (oldStat == 0 ||
                    ((oldStat == 2) && atomic_load_8(RSMA_TRIGGER_STAT(pRSmaStat)) < TASK_TRIGGER_STAT_PAUSED)) {
C
Cary Xu 已提交
1628
                  int32_t oldVal = atomic_fetch_add_32(&pRSmaStat->nFetchAll, 1);
K
kailixu 已提交
1629 1630

                  if (ASSERTS(oldVal >= 0, "oldVal of nFetchAll: %d < 0", oldVal)) {
K
kailixu 已提交
1631 1632 1633
                    code = TSDB_CODE_APP_ERROR;
                    TSDB_CHECK_CODE(code, lino, _exit);
                  }
K
kailixu 已提交
1634 1635 1636 1637 1638 1639 1640 1641

                  int8_t curStat = atomic_load_8(RSMA_COMMIT_STAT(pRSmaStat));
                  if (curStat == 1) {
                    smaDebug("vgId:%d, fetch all not exec as commit stat is %" PRIi8, SMA_VID(pSma), curStat);
                  } else {
                    tdRSmaFetchAllResult(pSma, pInfo);
                  }

1642 1643 1644
                  if (0 == atomic_sub_fetch_32(&pRSmaStat->nFetchAll, 1)) {
                    atomic_store_8(RSMA_COMMIT_STAT(pRSmaStat), 0);
                  }
C
Cary Xu 已提交
1645
                }
1646 1647 1648 1649 1650
              }

              if (qallItemSize > 0) {
                atomic_fetch_sub_64(&pRSmaStat->nBufItems, qallItemSize);
                continue;
K
kailixu 已提交
1651 1652 1653
              }
              if (RSMA_NEED_FETCH(pInfo)) {
                continue;
1654 1655 1656
              }

              break;
C
Cary Xu 已提交
1657 1658
            }
          }
1659
          atomic_val_compare_exchange_8(&pInfo->assigned, 1, 0);
C
Cary Xu 已提交
1660
        }
C
Cary Xu 已提交
1661
      }
C
Cary Xu 已提交
1662
    } else {
K
kailixu 已提交
1663
      ASSERTS(0, "unknown rsma exec type:%d", (int32_t)type);
K
kailixu 已提交
1664 1665
      code = TSDB_CODE_APP_ERROR;
      TSDB_CHECK_CODE(code, lino, _exit);
C
Cary Xu 已提交
1666 1667
    }

C
Cary Xu 已提交
1668
    if (atomic_load_64(&pRSmaStat->nBufItems) <= 0) {
C
Cary Xu 已提交
1669
      if (pEnv->flag & SMA_ENV_FLG_CLOSE) {
C
Cary Xu 已提交
1670 1671
        break;
      }
1672

1673
      tsem_wait(&pRSmaStat->notEmpty);
C
Cary Xu 已提交
1674

1675
      if ((pEnv->flag & SMA_ENV_FLG_CLOSE) && (atomic_load_64(&pRSmaStat->nBufItems) <= 0)) {
1676
        smaDebug("vgId:%d, exec task end, flag:%" PRIi8 ", nBufItems:%" PRIi64, SMA_VID(pSma), pEnv->flag,
1677
                 atomic_load_64(&pRSmaStat->nBufItems));
1678 1679
        break;
      }
C
Cary Xu 已提交
1680
    }
1681

C
Cary Xu 已提交
1682
  }  // end of while(true)
C
Cary Xu 已提交
1683

K
kailixu 已提交
1684
_exit:
C
Cary Xu 已提交
1685
  taosArrayDestroy(pSubmitArr);
K
kailixu 已提交
1686
  if (code) {
K
kailixu 已提交
1687
    smaError("vgId:%d, %s failed at line %d since %s", TD_VID(pVnode), __func__, lino, tstrerror(code));
K
kailixu 已提交
1688 1689
  }
  return code;
C
Cary Xu 已提交
1690
}