taosdata/TDengine | Build 4873 | source/dnode/mnode/sdb/src/sdbHash.c | Coveralls

21

const char *sdbTableName(ESdbType type) {

388,586,359✔

22

  switch (type) {

388,586,359✔

23

    case SDB_TRANS:

61,435,594✔

24

      return "trans";

61,435,594✔

25

    case SDB_CLUSTER:

3,735,254✔

26

      return "cluster";

3,735,254✔

27

    case SDB_MNODE:

4,207,998✔

28

      return "mnode";

4,207,998✔

29

    case SDB_QNODE:

2,198,904✔

30

      return "qnode";

2,198,904✔

31

    case SDB_SNODE:

4,102,298✔

32

      return "snode";

4,102,298✔

33

    case SDB_DNODE:

8,220,474✔

34

      return "dnode";

8,220,474✔

35

    case SDB_USER:

7,890,106✔

36

      return "user";

7,890,106✔

37

    case SDB_AUTH:

1,578✔

38

      return "auth";

1,578✔

39

    case SDB_ACCT:

3,538,999✔

40

      return "acct";

3,538,999✔

41

    case SDB_STREAM_CK:

1,578✔

42

      return "stream_ck";

1,578✔

43

    case SDB_STREAM:

2,709,863✔

44

      return "stream";

2,709,863✔

45

    case SDB_OFFSET:

1,578✔

46

      return "offset";

1,578✔

47

    case SDB_SUBSCRIBE:

2,379,501✔

48

      return "subscribe";

2,379,501✔

49

    case SDB_CONSUMER:

2,709,058✔

50

      return "consumer";

2,709,058✔

51

    case SDB_TOPIC:

2,298,469✔

52

      return "topic";

2,298,469✔

53

    case SDB_VGROUP:

31,831,051✔

54

      return "vgroup";

31,831,051✔

55

    case SDB_SMA:

2,053,438✔

56

      return "sma";

2,053,438✔

57

    case SDB_STB:

49,548,443✔

58

      return "stb";

49,548,443✔

59

    case SDB_DB:

18,453,995✔

60

      return "db";

18,453,995✔

61

    case SDB_FUNC:

2,132,754✔

62

      return "func";

2,132,754✔

63

    case SDB_IDX:

13,040,322✔

64

      return "idx";

13,040,322✔

65

    case SDB_VIEW:

3,280,153✔

66

      return "view";

3,280,153✔

67

    case SDB_STREAM_SEQ:

1,578✔

68

      return "stream_seq";

1,578✔

69

    case SDB_COMPACT:

2,280,907✔

70

      return "compact";

2,280,907✔

71

    case SDB_COMPACT_DETAIL:

3,066,911✔

72

      return "compact_detail";

3,066,911✔

73

    case SDB_GRANT:

4,211,824✔

74

      return "grant";

4,211,824✔

75

    case SDB_ARBGROUP:

2,134,234✔

76

      return "arb_group";

2,134,234✔

77

    case SDB_ANODE:

2,053,438✔

78

      return "anode";

2,053,438✔

79

    case SDB_BNODE:

2,518,178✔

80

      return "bnode";

2,518,178✔

81

    case SDB_CFG:

128,039,943✔

82

      return "config";

128,039,943✔

83

    case SDB_MOUNT:

2,061,769✔

84

      return "mount";

2,061,769✔

85

    case SDB_MOUNT_LOG:

2,058,241✔

86

      return "mount_log";

2,058,241✔

87

    case SDB_SSMIGRATE:

2,053,438✔

88

      return "ssmigrate";

2,053,438✔

89

    case SDB_SCAN:

2,057,038✔

90

      return "scan";

2,057,038✔

91

    case SDB_SCAN_DETAIL:

2,058,838✔

92

      return "scan_detail";

2,058,838✔

93

    case SDB_RSMA:

2,345,254✔

94

      return "rsma";

2,345,254✔

95

    case SDB_INSTANCE:

1,188,228✔

96

      return "instance";

1,188,228✔

97

    default:

4,685,132✔

98

      return "undefine";

4,685,132✔

102

const char *sdbStatusName(ESdbStatus status) {

159,208,715✔

103

  switch (status) {

159,208,715✔

104

    case SDB_STATUS_CREATING:

8,344,574✔

105

      return "creating";

8,344,574✔

106

    case SDB_STATUS_DROPPING:

2,710,570✔

107

      return "dropping";

2,710,570✔

108

    case SDB_STATUS_READY:

133,553,395✔

109

      return "ready";

133,553,395✔

110

    case SDB_STATUS_DROPPED:

14,513,214✔

111

      return "dropped";

14,513,214✔

112

    case SDB_STATUS_INIT:

86,962✔

113

      return "init";

86,962✔

121

void sdbPrintOper(SSdb *pSdb, SSdbRow *pRow, const char *oper) {

2,147,483,647✔

123

  EKeyType keyType = pSdb->keyTypes[pRow->type];

2,147,483,647✔

125

  if (keyType == SDB_KEY_BINARY) {

2,147,483,647✔

126

    mTrace("%s:%s, ref:%d oper:%s row:%p row->pObj:%p status:%s", sdbTableName(pRow->type), (char *)pRow->pObj,

1,074,114,363✔

128

  } else if (keyType == SDB_KEY_INT32) {

2,147,483,647✔

129

    mTrace("%s:%d, ref:%d oper:%s row:%p row->pObj:%p status:%s", sdbTableName(pRow->type), *(int32_t *)pRow->pObj,

2,147,483,647✔

131

  } else if (keyType == SDB_KEY_INT64) {

115,287,551✔

132

    mTrace("%s:%" PRId64 ", ref:%d oper:%s row:%p row->pObj:%p status:%s", sdbTableName(pRow->type),

115,287,551✔

137

2,147,483,647✔

139

static SHashObj *sdbGetHash(SSdb *pSdb, int32_t type) {

2,147,483,647✔

140

  if (type >= SDB_MAX || type < 0) {

2,147,483,647✔

141

    terrno = TSDB_CODE_SDB_INVALID_TABLE_TYPE;

8✔

145

  SHashObj *hash = pSdb->hashObjs[type];

2,147,483,647✔

146

  if (hash == NULL) {

2,147,483,647✔

151

  return hash;

2,147,483,647✔

154

static int32_t sdbGetkeySize(SSdb *pSdb, ESdbType type, const void *pKey) {

1,030,695,546✔

155

  int32_t  keySize = 0;

1,030,695,546✔

156

  EKeyType keyType = pSdb->keyTypes[type];

1,030,695,546✔

158

  if (keyType == SDB_KEY_INT32) {

1,030,698,095✔

159

    keySize = sizeof(int32_t);

692,508,919✔

160

  } else if (keyType == SDB_KEY_BINARY) {

338,189,176✔

161

    keySize = strlen(pKey) + 1;

335,365,661✔

163

    keySize = sizeof(int64_t);

2,823,515✔

166

  return keySize;

1,030,698,095✔

169

static int32_t sdbInsertRow(SSdb *pSdb, SHashObj *hash, SSdbRaw *pRaw, SSdbRow *pRow, int32_t keySize) {

87,311,884✔

170

  int32_t type = pRow->type;

87,311,884✔

171

  sdbWriteLock(pSdb, type);

87,311,884✔

173

  SSdbRow *pOldRow = taosHashGet(hash, pRow->pObj, keySize);

87,311,884✔

174

  if (pOldRow != NULL) {

87,311,884✔

181

  pRow->refCount = 0;

87,311,884✔

182

  pRow->status = pRaw->status;

87,311,884✔

183

  sdbPrintOper(pSdb, pRow, "insert");

87,311,884✔

185

  int32_t code = 0;

87,311,884✔

186

  if ((code = taosHashPut(hash, pRow->pObj, keySize, &pRow, sizeof(void *))) != 0) {

87,311,884✔

192

  SdbInsertFp insertFp = pSdb->insertFps[pRow->type];

87,311,884✔

193

  if (insertFp != NULL) {

87,311,884✔

194

    code = (*insertFp)(pSdb, pRow->pObj);

87,311,884✔

195

    if (code != 0) {

87,311,884✔

206

  sdbUnLock(pSdb, type);

87,311,884✔

208

  if (pSdb->keyTypes[pRow->type] == SDB_KEY_INT32) {

87,311,884✔

209

    pSdb->maxId[pRow->type] = TMAX(pSdb->maxId[pRow->type], *((int32_t *)pRow->pObj));

23,805,005✔

211

  if (pSdb->keyTypes[pRow->type] == SDB_KEY_INT64) {

87,311,884✔

212

    pSdb->maxId[pRow->type] = TMAX(pSdb->maxId[pRow->type], *((int64_t *)pRow->pObj));

773,034✔

214

  pSdb->tableVer[pRow->type]++;

87,311,884✔

216

  return 0;

87,311,884✔

219

static int32_t sdbUpdateRow(SSdb *pSdb, SHashObj *hash, SSdbRaw *pRaw, SSdbRow *pNewRow, int32_t keySize) {

127,100,737✔

220

  int32_t type = pNewRow->type;

127,100,737✔

221

  sdbWriteLock(pSdb, type);

127,100,737✔

223

  SSdbRow **ppOldRow = taosHashGet(hash, pNewRow->pObj, keySize);

127,100,737✔

224

  if (ppOldRow == NULL || *ppOldRow == NULL) {

127,100,737✔

225

    sdbUnLock(pSdb, type);

82,994,411✔

226

    return sdbInsertRow(pSdb, hash, pRaw, pNewRow, keySize);

82,994,411✔

229

  SSdbRow *pOldRow = *ppOldRow;

44,106,326✔

230

  pOldRow->status = pRaw->status;

44,106,326✔

231

  sdbPrintOper(pSdb, pOldRow, "update");

44,106,326✔

233

  int32_t     code = 0;

44,106,326✔

234

  SdbUpdateFp updateFp = pSdb->updateFps[type];

44,106,326✔

235

  if (updateFp != NULL) {

44,106,326✔

236

    code = (*updateFp)(pSdb, pOldRow->pObj, pNewRow->pObj);

44,106,326✔

238

  sdbUnLock(pSdb, type);

44,106,326✔

241

  sdbFreeRow(pSdb, pNewRow, false);

44,106,326✔

243

  pSdb->tableVer[pOldRow->type]++;

44,106,326✔

244

  return code;

44,106,326✔

247

static int32_t sdbDeleteRow(SSdb *pSdb, SHashObj *hash, SSdbRaw *pRaw, SSdbRow *pRow, int32_t keySize) {

23,554,196✔

248

  int32_t type = pRow->type;

23,554,196✔

249

  sdbWriteLock(pSdb, type);

23,554,196✔

251

  SSdbRow **ppOldRow = taosHashGet(hash, pRow->pObj, keySize);

23,554,196✔

252

  if (ppOldRow == NULL || *ppOldRow == NULL) {

23,554,196✔

253

    sdbUnLock(pSdb, type);

9,842✔

254

    sdbFreeRow(pSdb, pRow, false);

9,842✔

255

    terrno = TSDB_CODE_SDB_OBJ_NOT_THERE;

9,842✔

256

    return terrno;

9,842✔

258

  SSdbRow *pOldRow = *ppOldRow;

23,544,354✔

259

  pOldRow->status = pRaw->status;

23,544,354✔

261

  (void)atomic_add_fetch_32(&pOldRow->refCount, 1);

23,544,354✔

262

  sdbPrintOper(pSdb, pOldRow, "delete");

23,544,354✔

264

  if (taosHashRemove(hash, pOldRow->pObj, keySize) != 0) {

23,544,354✔

270

  pSdb->tableVer[pOldRow->type]++;

23,544,354✔

271

  sdbUnLock(pSdb, type);

23,544,354✔

273

  sdbFreeRow(pSdb, pRow, false);

23,544,354✔

275

  sdbCheckRow(pSdb, pOldRow);

23,544,354✔

276

  return 0;

23,544,354✔

279

int32_t sdbWriteWithoutFree(SSdb *pSdb, SSdbRaw *pRaw) {

154,972,406✔

280

  if (pRaw->type == SDB_CFG) {

154,972,406✔

281

    mTrace("sdb write cfg");

52,695,780✔

283

  SHashObj *hash = sdbGetHash(pSdb, pRaw->type);

154,972,406✔

284

  if (hash == NULL) return terrno;

154,972,406✔

286

  SdbDecodeFp decodeFp = pSdb->decodeFps[pRaw->type];

154,972,406✔

287

  SSdbRow    *pRow = (*decodeFp)(pRaw);

154,972,406✔

288

  if (pRow == NULL) return terrno;

154,972,406✔

290

  pRow->type = pRaw->type;

154,972,406✔

292

  int32_t keySize = sdbGetkeySize(pSdb, pRow->type, pRow->pObj);

154,972,406✔

293

  int32_t code = TSDB_CODE_SDB_INVALID_ACTION_TYPE;

154,972,406✔

295

  switch (pRaw->status) {

154,972,406✔

296

    case SDB_STATUS_CREATING:

4,317,473✔

297

      code = sdbInsertRow(pSdb, hash, pRaw, pRow, keySize);

4,317,473✔

298

      break;

4,317,473✔

299

    case SDB_STATUS_READY:

127,100,737✔

302

      code = sdbUpdateRow(pSdb, hash, pRaw, pRow, keySize);

127,100,737✔

303

      break;

127,100,737✔

304

    case SDB_STATUS_DROPPED:

23,554,196✔

305

      code = sdbDeleteRow(pSdb, hash, pRaw, pRow, keySize);

23,554,196✔

306

      break;

23,554,196✔

309

  return code;

154,972,406✔

312

int32_t sdbWrite(SSdb *pSdb, SSdbRaw *pRaw) {

18,553,333✔

313

  int32_t code = sdbWriteWithoutFree(pSdb, pRaw);

18,553,333✔

314

  sdbFreeRaw(pRaw);

18,553,333✔

315

  return code;

18,553,333✔

318

void *sdbAcquireAll(SSdb *pSdb, ESdbType type, const void *pKey, bool onlyReady) {

875,278,368✔

319

  terrno = 0;

875,278,368✔

321

  SHashObj *hash = sdbGetHash(pSdb, type);

875,276,935✔

322

  if (hash == NULL) return NULL;

875,276,179✔

324

  void   *pRet = NULL;

875,276,179✔

325

  int32_t keySize = sdbGetkeySize(pSdb, type, pKey);

875,276,179✔

327

  sdbReadLock(pSdb, type);

875,272,629✔

329

  SSdbRow **ppRow = taosHashGet(hash, pKey, keySize);

875,277,020✔

330

  if (ppRow == NULL || *ppRow == NULL) {

875,279,638✔

331

    sdbUnLock(pSdb, type);

215,327,448✔

332

    terrno = TSDB_CODE_SDB_OBJ_NOT_THERE;

215,327,923✔

333

    return NULL;

215,327,923✔

336

  SSdbRow *pRow = *ppRow;

659,952,856✔

337

  switch (pRow->status) {

659,952,856✔

338

    case SDB_STATUS_READY:

657,309,387✔

339

      (void)atomic_add_fetch_32(&pRow->refCount, 1);

657,309,387✔

340

      pRet = pRow->pObj;

657,309,875✔

341

      sdbPrintOper(pSdb, pRow, "acquire");

657,310,476✔

342

      break;

657,309,288✔

343

    case SDB_STATUS_CREATING:

2,095,841✔

344

      terrno = TSDB_CODE_SDB_OBJ_CREATING;

2,095,841✔

345

      break;

2,095,841✔

346

    case SDB_STATUS_DROPPING:

545,726✔

347

      terrno = TSDB_CODE_SDB_OBJ_DROPPING;

545,726✔

348

      break;

545,726✔

354

  if (pRet == NULL) {

659,950,855✔

355

    if (!onlyReady) {

2,641,567✔

363

  sdbUnLock(pSdb, type);

659,950,855✔

364

  return pRet;

659,953,180✔

367

void *sdbAcquire(SSdb *pSdb, ESdbType type, const void *pKey) { return sdbAcquireAll(pSdb, type, pKey, true); }

874,598,314✔

368

void *sdbAcquireNotReadyObj(SSdb *pSdb, ESdbType type, const void *pKey) {

677,642✔

369

  return sdbAcquireAll(pSdb, type, pKey, false);

677,642✔

372

static void sdbCheckRow(SSdb *pSdb, SSdbRow *pRow) {

23,544,354✔

373

  int32_t type = pRow->type;

23,544,354✔

374

  sdbWriteLock(pSdb, type);

23,544,354✔

376

  int32_t ref = atomic_sub_fetch_32(&pRow->refCount, 1);

23,544,354✔

377

  sdbPrintOper(pSdb, pRow, "check");

23,544,354✔

378

  if (ref <= 0 && pRow->status == SDB_STATUS_DROPPED) {

23,544,354✔

379

    sdbFreeRow(pSdb, pRow, true);

4,666,776✔

382

  sdbUnLock(pSdb, type);

23,544,354✔

383

23,544,354✔

385

void sdbReleaseLock(SSdb *pSdb, void *pObj, bool lock) {

1,505,959,757✔

386

  if (pObj == NULL) return;

1,505,959,757✔

388

  SSdbRow *pRow = (SSdbRow *)((char *)pObj - sizeof(SSdbRow));

1,489,476,356✔

389

  if (pRow->type >= SDB_MAX) return;

1,489,479,441✔

391

  int32_t type = pRow->type;

1,489,475,850✔

392

  if (lock) {

1,489,477,319✔

393

    sdbWriteLock(pSdb, type);

1,488,342,800✔

396

  int32_t ref = atomic_sub_fetch_32(&pRow->refCount, 1);

1,489,489,670✔

397

  sdbPrintOper(pSdb, pRow, "release");

1,489,497,006✔

398

  if (ref <= 0 && pRow->status == SDB_STATUS_DROPPED) {

1,489,490,533✔

399

    sdbFreeRow(pSdb, pRow, true);

18,877,578✔

401

  if (ref < 0) {

1,489,490,540✔

402

    mError("row:%p, ref:%d, type:%s", pRow, ref, sdbTableName(type));

1,758,329✔

405

  if (lock) {

1,489,490,466✔

406

    sdbUnLock(pSdb, type);

1,488,355,572✔

410

void sdbRelease(SSdb *pSdb, void *pObj) { sdbReleaseLock(pSdb, pObj, true); }

1,504,824,706✔

412

void *sdbFetch(SSdb *pSdb, ESdbType type, void *pIter, void **ppObj) {

910,011,192✔

413

  *ppObj = NULL;

910,011,192✔

415

  SHashObj *hash = sdbGetHash(pSdb, type);

910,011,393✔

416

  if (hash == NULL) return NULL;

910,006,262✔

418

  sdbReadLock(pSdb, type);

910,006,262✔

420

  SSdbRow **ppRow = taosHashIterate(hash, pIter);

910,011,792✔

421

  while (ppRow != NULL) {

922,898,122✔

422

    SSdbRow *pRow = *ppRow;

676,746,046✔

423

    if (pRow == NULL || pRow->status != SDB_STATUS_READY) {

676,747,674✔

424

      ppRow = taosHashIterate(hash, ppRow);

12,888,097✔

425

      continue;

12,884,360✔

428

    (void) atomic_add_fetch_32(&pRow->refCount, 1);

663,859,577✔

430

    sdbPrintOper(pSdb, pRow, "fetch");

663,862,864✔

431

    *ppObj = pRow->pObj;

663,853,406✔

432

    break;

663,856,374✔

434

  sdbUnLock(pSdb, type);

910,008,450✔

436

  return ppRow;

910,008,268✔

439

void *sdbFetchAll(SSdb *pSdb, ESdbType type, void *pIter, void **ppObj, ESdbStatus *status, bool lock) {

207,554,578✔

440

  *ppObj = NULL;

207,554,578✔

442

  SHashObj *hash = sdbGetHash(pSdb, type);

207,554,578✔

443

  if (hash == NULL) return NULL;

207,554,578✔

445

  if (lock) {

207,554,578✔

446

    sdbReadLock(pSdb, type);

205,652,604✔

449

  SSdbRow **ppRow = taosHashIterate(hash, pIter);

207,554,578✔

450

  while (ppRow != NULL) {

207,554,578✔

451

    SSdbRow *pRow = *ppRow;

168,337,018✔

452

    if (pRow == NULL) {

168,337,018✔

457

    (void)atomic_add_fetch_32(&pRow->refCount, 1);

168,337,018✔

458

    sdbPrintOper(pSdb, pRow, "fetch");

168,337,018✔

459

    *ppObj = pRow->pObj;

168,337,018✔

460

    *status = pRow->status;

168,337,018✔

461

    break;

168,337,018✔

463

  if (lock) {

207,554,578✔

464

    sdbUnLock(pSdb, type);

205,652,604✔

467

  return ppRow;

207,554,578✔

470

void sdbCancelFetch(SSdb *pSdb, void *pIter) {

97,915,964✔

471

  if (pIter == NULL) return;

97,915,964✔

472

  SSdbRow *pRow = *(SSdbRow **)pIter;

97,826,220✔

473

  mTrace("cancel fetch row:%p", pRow);

97,826,220✔

474

  SHashObj *hash = sdbGetHash(pSdb, pRow->type);

97,826,220✔

475

  if (hash == NULL) return;

97,825,448✔

477

  int32_t type = pRow->type;

97,825,448✔

478

  sdbReadLock(pSdb, type);

97,826,220✔

479

  taosHashCancelIterate(hash, pIter);

97,826,220✔

480

  sdbUnLock(pSdb, type);

97,826,220✔

483

void sdbCancelFetchByType(SSdb *pSdb, void *pIter, ESdbType type) {

117✔

484

  if (pIter == NULL) return;

117✔

485

  if (type >= SDB_MAX || type < 0) return;

117✔

486

  SHashObj *hash = sdbGetHash(pSdb, type);

117✔

487

  if (hash == NULL) return;

117✔

489

  sdbReadLock(pSdb, type);

117✔

490

  taosHashCancelIterate(hash, pIter);

117✔

491

  sdbUnLock(pSdb, type);

117✔

494

void sdbTraverse(SSdb *pSdb, ESdbType type, sdbTraverseFp fp, void *p1, void *p2, void *p3) {

10,571,945✔

495

  SHashObj *hash = sdbGetHash(pSdb, type);

10,571,945✔

496

  if (hash == NULL) return;

10,571,945✔

498

  sdbReadLock(pSdb, type);

10,571,945✔

500

  SSdbRow **ppRow = taosHashIterate(hash, NULL);

10,571,945✔

501

  while (ppRow != NULL) {

43,801,019✔

502

    SSdbRow *pRow = *ppRow;

33,243,198✔

503

    if (pRow->status == SDB_STATUS_READY) {

33,243,198✔

504

      bool isContinue = (*fp)(pSdb->pMnode, pRow->pObj, p1, p2, p3);

31,141,062✔

505

      if (!isContinue) {

31,141,062✔

506

        taosHashCancelIterate(hash, ppRow);

14,124✔

507

        break;

14,124✔

511

    ppRow = taosHashIterate(hash, ppRow);

33,229,074✔

514

  sdbUnLock(pSdb, type);

10,571,945✔

517

int32_t sdbGetSize(SSdb *pSdb, ESdbType type) {

202,460,192✔

518

  SHashObj *hash = sdbGetHash(pSdb, type);

202,460,192✔

519

  if (hash == NULL) return 0;

202,459,470✔

521

  sdbReadLock(pSdb, type);

202,459,470✔

522

  int32_t size = taosHashGetSize(hash);

202,458,208✔

523

  sdbUnLock(pSdb, type);

202,458,288✔

525

  return size;

202,460,390✔

528

int32_t sdbGetMaxId(SSdb *pSdb, ESdbType type) {

20,251,312✔

529

  SHashObj *hash = sdbGetHash(pSdb, type);

20,251,312✔

530

  if (hash == NULL) return -1;

20,251,312✔

532

  if (pSdb->keyTypes[type] != SDB_KEY_INT32) return -1;

20,251,312✔

534

  int32_t maxId = 0;

20,251,312✔

535

  sdbReadLock(pSdb, type);

20,251,312✔

537

  SSdbRow **ppRow = taosHashIterate(hash, NULL);

20,251,312✔

538

  while (ppRow != NULL) {

34,718,656✔

539

    SSdbRow *pRow = *ppRow;

14,467,344✔

540

    int32_t  id = *(int32_t *)pRow->pObj;

14,467,344✔

541

    maxId = TMAX(id, maxId);

14,467,344✔

542

    ppRow = taosHashIterate(hash, ppRow);

14,467,344✔

545

  sdbUnLock(pSdb, type);

20,251,312✔

546

  maxId = TMAX(maxId, pSdb->maxId[type]);

20,251,312✔

547

  return maxId + 1;

20,251,312✔

550

int64_t sdbGetTableVer(SSdb *pSdb, ESdbType type) {

150,146,779✔

551

  if (type >= SDB_MAX || type < 0) {

150,146,779✔

556

  return pSdb->tableVer[type];

150,146,779✔

572

bool sdbCheckExists(SSdb *pSdb, ESdbType type, const void *pKey) {

447,898✔

573

  SHashObj *hash = sdbGetHash(pSdb, type);

447,898✔

574

  if (hash == NULL) return false;

447,898✔

575

  int32_t keySize = sdbGetkeySize(pSdb, type, pKey);

447,898✔

577

  sdbReadLock(pSdb, type);

447,898✔

578

  void* p = taosHashGet(hash, pKey, keySize);

447,898✔

579

  sdbUnLock(pSdb, type);

447,898✔

581

  return (NULL != p);

447,898✔

taosdata / TDengine / #4873

Source File
Press 'n' to go to next uncovered line, 'b' for previous

1	/*
2	* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
3	*
4	* This program is free software: you can use, redistribute, and/or modify
5	* it under the terms of the GNU Affero General Public License, version 3
6	* or later ("AGPL"), as published by the Free Software Foundation.
7	*
8	* This program is distributed in the hope that it will be useful, but WITHOUT
9	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10	* FITNESS FOR A PARTICULAR PURPOSE.
11	*
12	* You should have received a copy of the GNU Affero General Public License
13	* along with this program. If not, see <http://www.gnu.org/licenses/>.
14	*/
15
16	#define _DEFAULT_SOURCE
17	#include "sdb.h"
18
19	static void sdbCheckRow(SSdb pSdb, SSdbRow pRow);
20
21	const char *sdbTableName(ESdbType type) {	388,586,359✔
22	switch (type) {	388,586,359✔
23	case SDB_TRANS:	61,435,594✔
24	return "trans";	61,435,594✔
25	case SDB_CLUSTER:	3,735,254✔
26	return "cluster";	3,735,254✔
27	case SDB_MNODE:	4,207,998✔
28	return "mnode";	4,207,998✔
29	case SDB_QNODE:	2,198,904✔
30	return "qnode";	2,198,904✔
31	case SDB_SNODE:	4,102,298✔
32	return "snode";	4,102,298✔
33	case SDB_DNODE:	8,220,474✔
34	return "dnode";	8,220,474✔
35	case SDB_USER:	7,890,106✔
36	return "user";	7,890,106✔
37	case SDB_AUTH:	1,578✔
38	return "auth";	1,578✔
39	case SDB_ACCT:	3,538,999✔
40	return "acct";	3,538,999✔
41	case SDB_STREAM_CK:	1,578✔
42	return "stream_ck";	1,578✔
43	case SDB_STREAM:	2,709,863✔
44	return "stream";	2,709,863✔
45	case SDB_OFFSET:	1,578✔
46	return "offset";	1,578✔
47	case SDB_SUBSCRIBE:	2,379,501✔
48	return "subscribe";	2,379,501✔
49	case SDB_CONSUMER:	2,709,058✔
50	return "consumer";	2,709,058✔
51	case SDB_TOPIC:	2,298,469✔
52	return "topic";	2,298,469✔
53	case SDB_VGROUP:	31,831,051✔
54	return "vgroup";	31,831,051✔
55	case SDB_SMA:	2,053,438✔
56	return "sma";	2,053,438✔
57	case SDB_STB:	49,548,443✔
58	return "stb";	49,548,443✔
59	case SDB_DB:	18,453,995✔
60	return "db";	18,453,995✔
61	case SDB_FUNC:	2,132,754✔
62	return "func";	2,132,754✔
63	case SDB_IDX:	13,040,322✔
64	return "idx";	13,040,322✔
65	case SDB_VIEW:	3,280,153✔
66	return "view";	3,280,153✔
67	case SDB_STREAM_SEQ:	1,578✔
68	return "stream_seq";	1,578✔
69	case SDB_COMPACT:	2,280,907✔
70	return "compact";	2,280,907✔
71	case SDB_COMPACT_DETAIL:	3,066,911✔
72	return "compact_detail";	3,066,911✔
73	case SDB_GRANT:	4,211,824✔
74	return "grant";	4,211,824✔
75	case SDB_ARBGROUP:	2,134,234✔
76	return "arb_group";	2,134,234✔
77	case SDB_ANODE:	2,053,438✔
78	return "anode";	2,053,438✔
79	case SDB_BNODE:	2,518,178✔
80	return "bnode";	2,518,178✔
81	case SDB_CFG:	128,039,943✔
82	return "config";	128,039,943✔
83	case SDB_MOUNT:	2,061,769✔
84	return "mount";	2,061,769✔
85	case SDB_MOUNT_LOG:	2,058,241✔
86	return "mount_log";	2,058,241✔
87	case SDB_SSMIGRATE:	2,053,438✔
88	return "ssmigrate";	2,053,438✔
89	case SDB_SCAN:	2,057,038✔
90	return "scan";	2,057,038✔
91	case SDB_SCAN_DETAIL:	2,058,838✔
92	return "scan_detail";	2,058,838✔
93	case SDB_RSMA:	2,345,254✔
94	return "rsma";	2,345,254✔
95	case SDB_INSTANCE:	1,188,228✔
96	return "instance";	1,188,228✔
97	default:	4,685,132✔
98	return "undefine";	4,685,132✔
99	}
100	}
101
102	const char *sdbStatusName(ESdbStatus status) {	159,208,715✔
103	switch (status) {	159,208,715✔
104	case SDB_STATUS_CREATING:	8,344,574✔
105	return "creating";	8,344,574✔
106	case SDB_STATUS_DROPPING:	2,710,570✔
107	return "dropping";	2,710,570✔
108	case SDB_STATUS_READY:	133,553,395✔
109	return "ready";	133,553,395✔
110	case SDB_STATUS_DROPPED:	14,513,214✔
111	return "dropped";	14,513,214✔
112	case SDB_STATUS_INIT:	86,962✔
113	return "init";	86,962✔
UNCOV 114	case SDB_STATUS_UPDATE:	×
UNCOV 115	return "update";	×
116	default:	×
117	return "undefine";	×
118	}
119	}
120
121	void sdbPrintOper(SSdb pSdb, SSdbRow pRow, const char *oper) {	2,147,483,647✔
122	#if 1
123	EKeyType keyType = pSdb->keyTypes[pRow->type];	2,147,483,647✔
124
125	if (keyType == SDB_KEY_BINARY) {	2,147,483,647✔
126	mTrace("%s:%s, ref:%d oper:%s row:%p row->pObj:%p status:%s", sdbTableName(pRow->type), (char *)pRow->pObj,	1,074,114,363✔
127	pRow->refCount, oper, pRow, pRow->pObj, sdbStatusName(pRow->status));
128	} else if (keyType == SDB_KEY_INT32) {	2,147,483,647✔
129	mTrace("%s:%d, ref:%d oper:%s row:%p row->pObj:%p status:%s", sdbTableName(pRow->type), (int32_t )pRow->pObj,	2,147,483,647✔
130	pRow->refCount, oper, pRow, pRow->pObj, sdbStatusName(pRow->status));
131	} else if (keyType == SDB_KEY_INT64) {	115,287,551✔
132	mTrace("%s:%" PRId64 ", ref:%d oper:%s row:%p row->pObj:%p status:%s", sdbTableName(pRow->type),	115,287,551✔
133	(int64_t )pRow->pObj, pRow->refCount, oper, pRow, pRow->pObj, sdbStatusName(pRow->status));
134	} else {
135	}
136	#endif
137	}	2,147,483,647✔
138
139	static SHashObj sdbGetHash(SSdb pSdb, int32_t type) {	2,147,483,647✔
140	if (type >= SDB_MAX \|\| type < 0) {	2,147,483,647✔
141	terrno = TSDB_CODE_SDB_INVALID_TABLE_TYPE;	8✔
UNCOV 142	return NULL;	×
143	}
144
145	SHashObj *hash = pSdb->hashObjs[type];	2,147,483,647✔
146	if (hash == NULL) {	2,147,483,647✔
UNCOV 147	terrno = TSDB_CODE_APP_ERROR;	×
UNCOV 148	return NULL;	×
149	}
150
151	return hash;	2,147,483,647✔
152	}
153
154	static int32_t sdbGetkeySize(SSdb pSdb, ESdbType type, const void pKey) {	1,030,695,546✔
155	int32_t keySize = 0;	1,030,695,546✔
156	EKeyType keyType = pSdb->keyTypes[type];	1,030,695,546✔
157
158	if (keyType == SDB_KEY_INT32) {	1,030,698,095✔
159	keySize = sizeof(int32_t);	692,508,919✔
160	} else if (keyType == SDB_KEY_BINARY) {	338,189,176✔
161	keySize = strlen(pKey) + 1;	335,365,661✔
162	} else {
163	keySize = sizeof(int64_t);	2,823,515✔
164	}
165
166	return keySize;	1,030,698,095✔
167	}
168
169	static int32_t sdbInsertRow(SSdb pSdb, SHashObj hash, SSdbRaw pRaw, SSdbRow pRow, int32_t keySize) {	87,311,884✔
170	int32_t type = pRow->type;	87,311,884✔
171	sdbWriteLock(pSdb, type);	87,311,884✔
172
173	SSdbRow *pOldRow = taosHashGet(hash, pRow->pObj, keySize);	87,311,884✔
174	if (pOldRow != NULL) {	87,311,884✔
UNCOV 175	sdbUnLock(pSdb, type);	×
UNCOV 176	sdbFreeRow(pSdb, pRow, false);	×
177	terrno = TSDB_CODE_SDB_OBJ_ALREADY_THERE;	×
178	return terrno;	×
179	}
180
181	pRow->refCount = 0;	87,311,884✔
182	pRow->status = pRaw->status;	87,311,884✔
183	sdbPrintOper(pSdb, pRow, "insert");	87,311,884✔
184
185	int32_t code = 0;	87,311,884✔
186	if ((code = taosHashPut(hash, pRow->pObj, keySize, &pRow, sizeof(void *))) != 0) {	87,311,884✔
UNCOV 187	sdbUnLock(pSdb, type);	×
UNCOV 188	sdbFreeRow(pSdb, pRow, false);	×
189	return code;	×
190	}
191
192	SdbInsertFp insertFp = pSdb->insertFps[pRow->type];	87,311,884✔
193	if (insertFp != NULL) {	87,311,884✔
194	code = (*insertFp)(pSdb, pRow->pObj);	87,311,884✔
195	if (code != 0) {	87,311,884✔
UNCOV 196	if (taosHashRemove(hash, pRow->pObj, keySize) != 0) {	×
UNCOV 197	mError("failed to remove row from hash");	×
198	}
199	sdbFreeRow(pSdb, pRow, false);	×
UNCOV 200	sdbUnLock(pSdb, type);	×
201	terrno = code;	×
202	return terrno;	×
203	}
204	}
205
206	sdbUnLock(pSdb, type);	87,311,884✔
207
208	if (pSdb->keyTypes[pRow->type] == SDB_KEY_INT32) {	87,311,884✔
209	pSdb->maxId[pRow->type] = TMAX(pSdb->maxId[pRow->type], ((int32_t )pRow->pObj));	23,805,005✔
210	}
211	if (pSdb->keyTypes[pRow->type] == SDB_KEY_INT64) {	87,311,884✔
212	pSdb->maxId[pRow->type] = TMAX(pSdb->maxId[pRow->type], ((int64_t )pRow->pObj));	773,034✔
213	}
214	pSdb->tableVer[pRow->type]++;	87,311,884✔
215
216	return 0;	87,311,884✔
217	}
218
219	static int32_t sdbUpdateRow(SSdb pSdb, SHashObj hash, SSdbRaw pRaw, SSdbRow pNewRow, int32_t keySize) {	127,100,737✔
220	int32_t type = pNewRow->type;	127,100,737✔
221	sdbWriteLock(pSdb, type);	127,100,737✔
222
223	SSdbRow **ppOldRow = taosHashGet(hash, pNewRow->pObj, keySize);	127,100,737✔
224	if (ppOldRow == NULL \|\| *ppOldRow == NULL) {	127,100,737✔
225	sdbUnLock(pSdb, type);	82,994,411✔
226	return sdbInsertRow(pSdb, hash, pRaw, pNewRow, keySize);	82,994,411✔
227	}
228
229	SSdbRow pOldRow = ppOldRow;	44,106,326✔
230	pOldRow->status = pRaw->status;	44,106,326✔
231	sdbPrintOper(pSdb, pOldRow, "update");	44,106,326✔
232
233	int32_t code = 0;	44,106,326✔
234	SdbUpdateFp updateFp = pSdb->updateFps[type];	44,106,326✔
235	if (updateFp != NULL) {	44,106,326✔
236	code = (*updateFp)(pSdb, pOldRow->pObj, pNewRow->pObj);	44,106,326✔
237	}
238	sdbUnLock(pSdb, type);	44,106,326✔
239
240	// sdbUnLock(pSdb, type);
241	sdbFreeRow(pSdb, pNewRow, false);	44,106,326✔
242
243	pSdb->tableVer[pOldRow->type]++;	44,106,326✔
244	return code;	44,106,326✔
245	}
246
247	static int32_t sdbDeleteRow(SSdb pSdb, SHashObj hash, SSdbRaw pRaw, SSdbRow pRow, int32_t keySize) {	23,554,196✔
248	int32_t type = pRow->type;	23,554,196✔
249	sdbWriteLock(pSdb, type);	23,554,196✔
250
251	SSdbRow **ppOldRow = taosHashGet(hash, pRow->pObj, keySize);	23,554,196✔
252	if (ppOldRow == NULL \|\| *ppOldRow == NULL) {	23,554,196✔
253	sdbUnLock(pSdb, type);	9,842✔
254	sdbFreeRow(pSdb, pRow, false);	9,842✔
255	terrno = TSDB_CODE_SDB_OBJ_NOT_THERE;	9,842✔
256	return terrno;	9,842✔
257	}
258	SSdbRow pOldRow = ppOldRow;	23,544,354✔
259	pOldRow->status = pRaw->status;	23,544,354✔
260
261	(void)atomic_add_fetch_32(&pOldRow->refCount, 1);	23,544,354✔
262	sdbPrintOper(pSdb, pOldRow, "delete");	23,544,354✔
263
264	if (taosHashRemove(hash, pOldRow->pObj, keySize) != 0) {	23,544,354✔
UNCOV 265	sdbUnLock(pSdb, type);	×
UNCOV 266	sdbFreeRow(pSdb, pRow, false);	×
267	terrno = TSDB_CODE_SDB_OBJ_NOT_THERE;	×
268	return terrno;	×
269	}
270	pSdb->tableVer[pOldRow->type]++;	23,544,354✔
271	sdbUnLock(pSdb, type);	23,544,354✔
272
273	sdbFreeRow(pSdb, pRow, false);	23,544,354✔
274
275	sdbCheckRow(pSdb, pOldRow);	23,544,354✔
276	return 0;	23,544,354✔
277	}
278
279	int32_t sdbWriteWithoutFree(SSdb pSdb, SSdbRaw pRaw) {	154,972,406✔
280	if (pRaw->type == SDB_CFG) {	154,972,406✔
281	mTrace("sdb write cfg");	52,695,780✔
282	}
283	SHashObj *hash = sdbGetHash(pSdb, pRaw->type);	154,972,406✔
284	if (hash == NULL) return terrno;	154,972,406✔
285
286	SdbDecodeFp decodeFp = pSdb->decodeFps[pRaw->type];	154,972,406✔
287	SSdbRow pRow = (decodeFp)(pRaw);	154,972,406✔
288	if (pRow == NULL) return terrno;	154,972,406✔
289
290	pRow->type = pRaw->type;	154,972,406✔
291
292	int32_t keySize = sdbGetkeySize(pSdb, pRow->type, pRow->pObj);	154,972,406✔
293	int32_t code = TSDB_CODE_SDB_INVALID_ACTION_TYPE;	154,972,406✔
294
295	switch (pRaw->status) {	154,972,406✔
296	case SDB_STATUS_CREATING:	4,317,473✔
297	code = sdbInsertRow(pSdb, hash, pRaw, pRow, keySize);	4,317,473✔
298	break;	4,317,473✔
299	case SDB_STATUS_READY:	127,100,737✔
300	case SDB_STATUS_UPDATE:
301	case SDB_STATUS_DROPPING:
302	code = sdbUpdateRow(pSdb, hash, pRaw, pRow, keySize);	127,100,737✔
303	break;	127,100,737✔
304	case SDB_STATUS_DROPPED:	23,554,196✔
305	code = sdbDeleteRow(pSdb, hash, pRaw, pRow, keySize);	23,554,196✔
306	break;	23,554,196✔
307	}
308
309	return code;	154,972,406✔
310	}
311
312	int32_t sdbWrite(SSdb pSdb, SSdbRaw pRaw) {	18,553,333✔
313	int32_t code = sdbWriteWithoutFree(pSdb, pRaw);	18,553,333✔
314	sdbFreeRaw(pRaw);	18,553,333✔
315	return code;	18,553,333✔
316	}
317
318	void sdbAcquireAll(SSdb pSdb, ESdbType type, const void *pKey, bool onlyReady) {	875,278,368✔
319	terrno = 0;	875,278,368✔
320
321	SHashObj *hash = sdbGetHash(pSdb, type);	875,276,935✔
322	if (hash == NULL) return NULL;	875,276,179✔
323
324	void *pRet = NULL;	875,276,179✔
325	int32_t keySize = sdbGetkeySize(pSdb, type, pKey);	875,276,179✔
326
327	sdbReadLock(pSdb, type);	875,272,629✔
328
329	SSdbRow **ppRow = taosHashGet(hash, pKey, keySize);	875,277,020✔
330	if (ppRow == NULL \|\| *ppRow == NULL) {	875,279,638✔
331	sdbUnLock(pSdb, type);	215,327,448✔
332	terrno = TSDB_CODE_SDB_OBJ_NOT_THERE;	215,327,923✔
333	return NULL;	215,327,923✔
334	}
335
336	SSdbRow pRow = ppRow;	659,952,856✔
337	switch (pRow->status) {	659,952,856✔
338	case SDB_STATUS_READY:	657,309,387✔
339	(void)atomic_add_fetch_32(&pRow->refCount, 1);	657,309,387✔
340	pRet = pRow->pObj;	657,309,875✔
341	sdbPrintOper(pSdb, pRow, "acquire");	657,310,476✔
342	break;	657,309,288✔
343	case SDB_STATUS_CREATING:	2,095,841✔
344	terrno = TSDB_CODE_SDB_OBJ_CREATING;	2,095,841✔
345	break;	2,095,841✔
346	case SDB_STATUS_DROPPING:	545,726✔
347	terrno = TSDB_CODE_SDB_OBJ_DROPPING;	545,726✔
348	break;	545,726✔
UNCOV 349	default:	×
UNCOV 350	terrno = TSDB_CODE_APP_ERROR;	×
351	break;	×
352	}
353
354	if (pRet == NULL) {	659,950,855✔
355	if (!onlyReady) {	2,641,567✔
UNCOV 356	terrno = 0;	×
UNCOV 357	(void)atomic_add_fetch_32(&pRow->refCount, 1);	×
358	pRet = pRow->pObj;	×
359	sdbPrintOper(pSdb, pRow, "acquire");	×
360	}
361	}
362
363	sdbUnLock(pSdb, type);	659,950,855✔
364	return pRet;	659,953,180✔
365	}
366
367	void sdbAcquire(SSdb pSdb, ESdbType type, const void *pKey) { return sdbAcquireAll(pSdb, type, pKey, true); }	874,598,314✔
368	void sdbAcquireNotReadyObj(SSdb pSdb, ESdbType type, const void *pKey) {	677,642✔
369	return sdbAcquireAll(pSdb, type, pKey, false);	677,642✔
370	}
371
372	static void sdbCheckRow(SSdb pSdb, SSdbRow pRow) {	23,544,354✔
373	int32_t type = pRow->type;	23,544,354✔
374	sdbWriteLock(pSdb, type);	23,544,354✔
375
376	int32_t ref = atomic_sub_fetch_32(&pRow->refCount, 1);	23,544,354✔
377	sdbPrintOper(pSdb, pRow, "check");	23,544,354✔
378	if (ref <= 0 && pRow->status == SDB_STATUS_DROPPED) {	23,544,354✔
379	sdbFreeRow(pSdb, pRow, true);	4,666,776✔
380	}
381
382	sdbUnLock(pSdb, type);	23,544,354✔
383	}	23,544,354✔
384
385	void sdbReleaseLock(SSdb pSdb, void pObj, bool lock) {	1,505,959,757✔
386	if (pObj == NULL) return;	1,505,959,757✔
387
388	SSdbRow pRow = (SSdbRow )((char *)pObj - sizeof(SSdbRow));	1,489,476,356✔
389	if (pRow->type >= SDB_MAX) return;	1,489,479,441✔
390
391	int32_t type = pRow->type;	1,489,475,850✔
392	if (lock) {	1,489,477,319✔
393	sdbWriteLock(pSdb, type);	1,488,342,800✔
394	}
395
396	int32_t ref = atomic_sub_fetch_32(&pRow->refCount, 1);	1,489,489,670✔
397	sdbPrintOper(pSdb, pRow, "release");	1,489,497,006✔
398	if (ref <= 0 && pRow->status == SDB_STATUS_DROPPED) {	1,489,490,533✔
399	sdbFreeRow(pSdb, pRow, true);	18,877,578✔
400	}
401	if (ref < 0) {	1,489,490,540✔
402	mError("row:%p, ref:%d, type:%s", pRow, ref, sdbTableName(type));	1,758,329✔
403	}
404
405	if (lock) {	1,489,490,466✔
406	sdbUnLock(pSdb, type);	1,488,355,572✔
407	}
408	}
409
410	void sdbRelease(SSdb pSdb, void pObj) { sdbReleaseLock(pSdb, pObj, true); }	1,504,824,706✔
411
412	void sdbFetch(SSdb pSdb, ESdbType type, void pIter, void *ppObj) {	910,011,192✔
413	*ppObj = NULL;	910,011,192✔
414
415	SHashObj *hash = sdbGetHash(pSdb, type);	910,011,393✔
416	if (hash == NULL) return NULL;	910,006,262✔
417
418	sdbReadLock(pSdb, type);	910,006,262✔
419
420	SSdbRow **ppRow = taosHashIterate(hash, pIter);	910,011,792✔
421	while (ppRow != NULL) {	922,898,122✔
422	SSdbRow pRow = ppRow;	676,746,046✔
423	if (pRow == NULL \|\| pRow->status != SDB_STATUS_READY) {	676,747,674✔
424	ppRow = taosHashIterate(hash, ppRow);	12,888,097✔
425	continue;	12,884,360✔
426	}
427
428	(void) atomic_add_fetch_32(&pRow->refCount, 1);	663,859,577✔
429
430	sdbPrintOper(pSdb, pRow, "fetch");	663,862,864✔
431	*ppObj = pRow->pObj;	663,853,406✔
432	break;	663,856,374✔
433	}
434	sdbUnLock(pSdb, type);	910,008,450✔
435
436	return ppRow;	910,008,268✔
437	}
438
439	void sdbFetchAll(SSdb pSdb, ESdbType type, void pIter, void ppObj, ESdbStatus status, bool lock) {	207,554,578✔
440	*ppObj = NULL;	207,554,578✔
441
442	SHashObj *hash = sdbGetHash(pSdb, type);	207,554,578✔
443	if (hash == NULL) return NULL;	207,554,578✔
444
445	if (lock) {	207,554,578✔
446	sdbReadLock(pSdb, type);	205,652,604✔
447	}
448
449	SSdbRow **ppRow = taosHashIterate(hash, pIter);	207,554,578✔
450	while (ppRow != NULL) {	207,554,578✔
451	SSdbRow pRow = ppRow;	168,337,018✔
452	if (pRow == NULL) {	168,337,018✔
UNCOV 453	ppRow = taosHashIterate(hash, ppRow);	×
UNCOV 454	continue;	×
455	}
456
457	(void)atomic_add_fetch_32(&pRow->refCount, 1);	168,337,018✔
458	sdbPrintOper(pSdb, pRow, "fetch");	168,337,018✔
459	*ppObj = pRow->pObj;	168,337,018✔
460	*status = pRow->status;	168,337,018✔
461	break;	168,337,018✔
462	}
463	if (lock) {	207,554,578✔
464	sdbUnLock(pSdb, type);	205,652,604✔
465	}
466
467	return ppRow;	207,554,578✔
468	}
469
470	void sdbCancelFetch(SSdb pSdb, void pIter) {	97,915,964✔
471	if (pIter == NULL) return;	97,915,964✔
472	SSdbRow pRow = (SSdbRow **)pIter;	97,826,220✔
473	mTrace("cancel fetch row:%p", pRow);	97,826,220✔
474	SHashObj *hash = sdbGetHash(pSdb, pRow->type);	97,826,220✔
475	if (hash == NULL) return;	97,825,448✔
476
477	int32_t type = pRow->type;	97,825,448✔
478	sdbReadLock(pSdb, type);	97,826,220✔
479	taosHashCancelIterate(hash, pIter);	97,826,220✔
480	sdbUnLock(pSdb, type);	97,826,220✔
481	}
482
483	void sdbCancelFetchByType(SSdb pSdb, void pIter, ESdbType type) {	117✔
484	if (pIter == NULL) return;	117✔
485	if (type >= SDB_MAX \|\| type < 0) return;	117✔
486	SHashObj *hash = sdbGetHash(pSdb, type);	117✔
487	if (hash == NULL) return;	117✔
488
489	sdbReadLock(pSdb, type);	117✔
490	taosHashCancelIterate(hash, pIter);	117✔
491	sdbUnLock(pSdb, type);	117✔
492	}
493
494	void sdbTraverse(SSdb pSdb, ESdbType type, sdbTraverseFp fp, void p1, void p2, void p3) {	10,571,945✔
495	SHashObj *hash = sdbGetHash(pSdb, type);	10,571,945✔
496	if (hash == NULL) return;	10,571,945✔
497
498	sdbReadLock(pSdb, type);	10,571,945✔
499
500	SSdbRow **ppRow = taosHashIterate(hash, NULL);	10,571,945✔
501	while (ppRow != NULL) {	43,801,019✔
502	SSdbRow pRow = ppRow;	33,243,198✔
503	if (pRow->status == SDB_STATUS_READY) {	33,243,198✔
504	bool isContinue = (*fp)(pSdb->pMnode, pRow->pObj, p1, p2, p3);	31,141,062✔
505	if (!isContinue) {	31,141,062✔
506	taosHashCancelIterate(hash, ppRow);	14,124✔
507	break;	14,124✔
508	}
509	}
510
511	ppRow = taosHashIterate(hash, ppRow);	33,229,074✔
512	}
513
514	sdbUnLock(pSdb, type);	10,571,945✔
515	}
516
517	int32_t sdbGetSize(SSdb *pSdb, ESdbType type) {	202,460,192✔
518	SHashObj *hash = sdbGetHash(pSdb, type);	202,460,192✔
519	if (hash == NULL) return 0;	202,459,470✔
520
521	sdbReadLock(pSdb, type);	202,459,470✔
522	int32_t size = taosHashGetSize(hash);	202,458,208✔
523	sdbUnLock(pSdb, type);	202,458,288✔
524
525	return size;	202,460,390✔
526	}
527
528	int32_t sdbGetMaxId(SSdb *pSdb, ESdbType type) {	20,251,312✔
529	SHashObj *hash = sdbGetHash(pSdb, type);	20,251,312✔
530	if (hash == NULL) return -1;	20,251,312✔
531
532	if (pSdb->keyTypes[type] != SDB_KEY_INT32) return -1;	20,251,312✔
533
534	int32_t maxId = 0;	20,251,312✔
535	sdbReadLock(pSdb, type);	20,251,312✔
536
537	SSdbRow **ppRow = taosHashIterate(hash, NULL);	20,251,312✔
538	while (ppRow != NULL) {	34,718,656✔
539	SSdbRow pRow = ppRow;	14,467,344✔
540	int32_t id = (int32_t )pRow->pObj;	14,467,344✔
541	maxId = TMAX(id, maxId);	14,467,344✔
542	ppRow = taosHashIterate(hash, ppRow);	14,467,344✔
543	}
544
545	sdbUnLock(pSdb, type);	20,251,312✔
546	maxId = TMAX(maxId, pSdb->maxId[type]);	20,251,312✔
547	return maxId + 1;	20,251,312✔
548	}
549
550	int64_t sdbGetTableVer(SSdb *pSdb, ESdbType type) {	150,146,779✔
551	if (type >= SDB_MAX \|\| type < 0) {	150,146,779✔
UNCOV 552	terrno = TSDB_CODE_SDB_INVALID_TABLE_TYPE;	×
UNCOV 553	return -1;	×
554	}
555
556	return pSdb->tableVer[type];	150,146,779✔
557	}
558
UNCOV 559	bool countValid(SMnode pMnode, void pObj, void p1, void p2, void *p3) {	×
UNCOV 560	int32_t *pInt = p1;	×
561	(*pInt) += 1;	×
562	return true;	×
563	}
564
UNCOV 565	int32_t sdbGetValidSize(SSdb *pSdb, ESdbType type) {	×
UNCOV 566	int32_t num = 0;	×
567	sdbTraverse(pSdb, type, countValid, &num, 0, 0);	×
568	return num;	×
569	}
570
571
572	bool sdbCheckExists(SSdb pSdb, ESdbType type, const void pKey) {	447,898✔
573	SHashObj *hash = sdbGetHash(pSdb, type);	447,898✔
574	if (hash == NULL) return false;	447,898✔
575	int32_t keySize = sdbGetkeySize(pSdb, type, pKey);	447,898✔
576
577	sdbReadLock(pSdb, type);	447,898✔
578	void* p = taosHashGet(hash, pKey, keySize);	447,898✔
579	sdbUnLock(pSdb, type);	447,898✔
580
581	return (NULL != p);	447,898✔
582	}
583
584

taosdata / TDengine / #4873

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous