• Home
  • Features
  • Pricing
  • Docs
  • Announcements
  • Sign In

taosdata / TDengine / #3542

27 Nov 2024 02:52AM UTC coverage: 60.819% (+0.04%) from 60.776%
#3542

push

travis-ci

web-flow
Merge pull request #28931 from taosdata/enh/jdbc-demo-3.0

update jdbc demo, and version history

120305 of 252779 branches covered (47.59%)

Branch coverage included in aggregate %.

201010 of 275538 relevant lines covered (72.95%)

19989893.51 hits per line

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

74.48
/source/dnode/vnode/src/vnd/vnodeBufPool.c
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
#include "vnd.h"
17

18
/* ------------------------ STRUCTURES ------------------------ */
19
static int32_t vnodeBufPoolCreate(SVnode *pVnode, int32_t id, int64_t size, SVBufPool **ppPool) {
41,121✔
20
  SVBufPool *pPool;
21

22
  pPool = taosMemoryMalloc(sizeof(SVBufPool) + size);
41,121✔
23
  if (pPool == NULL) {
41,127!
24
    return terrno;
×
25
  }
26
  memset(pPool, 0, sizeof(SVBufPool));
41,127✔
27

28
  // query handle list
29
  (void)taosThreadMutexInit(&pPool->mutex, NULL);
41,127✔
30
  pPool->nQuery = 0;
41,125✔
31
  pPool->qList.pNext = &pPool->qList;
41,125✔
32
  pPool->qList.ppNext = &pPool->qList.pNext;
41,125✔
33

34
  pPool->pVnode = pVnode;
41,125✔
35
  pPool->id = id;
41,125✔
36
  pPool->ptr = pPool->node.data;
41,125✔
37
  pPool->pTail = &pPool->node;
41,125✔
38
  pPool->node.prev = NULL;
41,125✔
39
  pPool->node.pnext = &pPool->pTail;
41,125✔
40
  pPool->node.size = size;
41,125✔
41

42
  if (VND_IS_RSMA(pVnode)) {
41,125✔
43
    pPool->lock = taosMemoryMalloc(sizeof(TdThreadSpinlock));
51✔
44
    if (!pPool->lock) {
51!
45
      taosMemoryFree(pPool);
×
46
      return terrno;
×
47
    }
48
    if (taosThreadSpinInit(pPool->lock, 0) != 0) {
51!
49
      taosMemoryFree((void *)pPool->lock);
×
50
      taosMemoryFree(pPool);
×
51
      return terrno = TAOS_SYSTEM_ERROR(errno);
×
52
    }
53
  } else {
54
    pPool->lock = NULL;
41,074✔
55
  }
56

57
  *ppPool = pPool;
41,125✔
58
  return 0;
41,125✔
59
}
60

61
static void vnodeBufPoolDestroy(SVBufPool *pPool) {
41,130✔
62
  vnodeBufPoolReset(pPool);
41,130✔
63
  if (pPool->lock) {
41,130✔
64
    (void)taosThreadSpinDestroy(pPool->lock);
51✔
65
    taosMemoryFree((void *)pPool->lock);
51✔
66
  }
67
  (void)taosThreadMutexDestroy(&pPool->mutex);
41,130✔
68
  taosMemoryFree(pPool);
41,130✔
69
}
41,130✔
70

71
int vnodeOpenBufPool(SVnode *pVnode) {
13,710✔
72
  int64_t size = pVnode->config.szBuf / VNODE_BUFPOOL_SEGMENTS;
13,710✔
73

74
  for (int i = 0; i < VNODE_BUFPOOL_SEGMENTS; i++) {
54,836✔
75
    // create pool
76
    int32_t code;
77
    if ((code = vnodeBufPoolCreate(pVnode, i, size, &pVnode->aBufPool[i]))) {
41,122!
78
      vError("vgId:%d, failed to open vnode buffer pool since %s", TD_VID(pVnode), tstrerror(terrno));
×
79
      vnodeCloseBufPool(pVnode);
×
80
      return code;
×
81
    }
82

83
    // add to free list
84
    pVnode->aBufPool[i]->freeNext = pVnode->freeList;
41,126✔
85
    pVnode->freeList = pVnode->aBufPool[i];
41,126✔
86
  }
87

88
  vDebug("vgId:%d, vnode buffer pool is opened, size:%" PRId64, TD_VID(pVnode), size);
13,714✔
89
  return 0;
13,709✔
90
}
91

92
void vnodeCloseBufPool(SVnode *pVnode) {
13,709✔
93
  for (int32_t i = 0; i < VNODE_BUFPOOL_SEGMENTS; i++) {
54,839✔
94
    if (pVnode->aBufPool[i]) {
41,130!
95
      vnodeBufPoolDestroy(pVnode->aBufPool[i]);
41,130✔
96
      pVnode->aBufPool[i] = NULL;
41,130✔
97
    }
98
  }
99

100
  vDebug("vgId:%d, vnode buffer pool is closed", TD_VID(pVnode));
13,709✔
101
}
13,709✔
102

103
void vnodeBufPoolReset(SVBufPool *pPool) {
72,387✔
104
  if (pPool->nQuery != 0) {
72,387!
105
    vError("vgId:%d, buffer pool %p of id %d has %d queries, reset it may cause problem", TD_VID(pPool->pVnode), pPool,
×
106
           pPool->id, pPool->nQuery);
107
  }
108

109
  for (SVBufPoolNode *pNode = pPool->pTail; pNode->prev; pNode = pPool->pTail) {
10,931,307✔
110
    pNode->prev->pnext = &pPool->pTail;
10,859,003✔
111
    pPool->pTail = pNode->prev;
10,859,003✔
112
    pPool->size = pPool->size - sizeof(*pNode) - pNode->size;
10,859,003✔
113
    taosMemoryFree(pNode);
10,859,003✔
114
  }
115

116
  pPool->size = 0;
72,304✔
117
  pPool->ptr = pPool->node.data;
72,304✔
118
}
72,304✔
119

120
void *vnodeBufPoolMallocAligned(SVBufPool *pPool, int size) {
1,078,738,441✔
121
  SVBufPoolNode *pNode;
122
  void          *p = NULL;
1,078,738,441✔
123
  uint8_t       *ptr = NULL;
1,078,738,441✔
124
  int            paddingLen = 0;
1,078,738,441✔
125

126
  if (pPool == NULL) {
1,078,738,441!
127
    terrno = TSDB_CODE_INVALID_PARA;
×
128
    return NULL;
×
129
  }
130

131
  if (pPool->lock) taosThreadSpinLock(pPool->lock);
1,078,738,441✔
132

133
  ptr = pPool->ptr;
1,078,785,231✔
134
  paddingLen = (((long)ptr + 7) & ~7) - (long)ptr;
1,078,785,231✔
135

136
  if (pPool->node.size >= pPool->ptr - pPool->node.data + size + paddingLen) {
1,078,785,231✔
137
    // allocate from the anchor node
138
    p = pPool->ptr + paddingLen;
1,067,924,500✔
139
    size += paddingLen;
1,067,924,500✔
140
    pPool->ptr = pPool->ptr + size;
1,067,924,500✔
141
    pPool->size += size;
1,067,924,500✔
142
  } else {
143
    // allocate a new node
144
    pNode = taosMemoryMalloc(sizeof(*pNode) + size);
10,860,731✔
145
    if (pNode == NULL) {
10,860,731!
146
      if (pPool->lock) {
×
147
        (void)taosThreadSpinUnlock(pPool->lock);
×
148
      }
149
      return NULL;
×
150
    }
151

152
    p = pNode->data;
10,860,731✔
153
    pNode->size = size;
10,860,731✔
154
    pNode->prev = pPool->pTail;
10,860,731✔
155
    pNode->pnext = &pPool->pTail;
10,860,731✔
156
    pPool->pTail->pnext = &pNode->prev;
10,860,731✔
157
    pPool->pTail = pNode;
10,860,731✔
158

159
    pPool->size = pPool->size + sizeof(*pNode) + size;
10,860,731✔
160
  }
161
  if (pPool->lock) (void)taosThreadSpinUnlock(pPool->lock);
1,078,785,231✔
162
  return p;
1,078,776,160✔
163
}
164

165
void *vnodeBufPoolMalloc(SVBufPool *pPool, int size) {
63,789✔
166
  SVBufPoolNode *pNode;
167
  void          *p = NULL;
63,789✔
168

169
  if (pPool == NULL) {
63,789!
170
    terrno = TSDB_CODE_INVALID_PARA;
×
171
    return NULL;
×
172
  }
173

174
  if (pPool->lock) taosThreadSpinLock(pPool->lock);
63,789✔
175
  if (pPool->node.size >= pPool->ptr - pPool->node.data + size) {
63,791!
176
    // allocate from the anchor node
177
    p = pPool->ptr;
63,791✔
178
    pPool->ptr = pPool->ptr + size;
63,791✔
179
    pPool->size += size;
63,791✔
180
  } else {
181
    // allocate a new node
182
    pNode = taosMemoryMalloc(sizeof(*pNode) + size);
×
183
    if (pNode == NULL) {
×
184
      if (pPool->lock) (void)taosThreadSpinUnlock(pPool->lock);
×
185
      return NULL;
×
186
    }
187

188
    p = pNode->data;
×
189
    pNode->size = size;
×
190
    pNode->prev = pPool->pTail;
×
191
    pNode->pnext = &pPool->pTail;
×
192
    pPool->pTail->pnext = &pNode->prev;
×
193
    pPool->pTail = pNode;
×
194

195
    pPool->size = pPool->size + sizeof(*pNode) + size;
×
196
  }
197
  if (pPool->lock) (void)taosThreadSpinUnlock(pPool->lock);
63,791✔
198
  return p;
63,790✔
199
}
200

201
void vnodeBufPoolFree(SVBufPool *pPool, void *p) {
13,543✔
202
  // uint8_t       *ptr = (uint8_t *)p;
203
  // SVBufPoolNode *pNode;
204

205
  // if (ptr < pPool->node.data || ptr >= pPool->node.data + pPool->node.size) {
206
  //   pNode = &((SVBufPoolNode *)p)[-1];
207
  //   *pNode->pnext = pNode->prev;
208
  //   pNode->prev->pnext = pNode->pnext;
209

210
  //   pPool->size = pPool->size - sizeof(*pNode) - pNode->size;
211
  //   taosMemoryFree(pNode);
212
  // }
213
}
13,543✔
214

215
void vnodeBufPoolRef(SVBufPool *pPool) {
45,030✔
216
  int32_t nRef = atomic_fetch_add_32(&pPool->nRef, 1);
45,030✔
217
  if (nRef <= 0) {
45,031!
218
    vError("vgId:%d, buffer pool %p of id %d is referenced by %d", TD_VID(pPool->pVnode), pPool, pPool->id, nRef);
×
219
  }
220
}
45,031✔
221

222
void vnodeBufPoolAddToFreeList(SVBufPool *pPool) {
31,259✔
223
  SVnode *pVnode = pPool->pVnode;
31,259✔
224

225
  int64_t size = pVnode->config.szBuf / VNODE_BUFPOOL_SEGMENTS;
31,259✔
226
  if (pPool->node.size != size) {
31,259!
227
    SVBufPool *pNewPool = NULL;
×
228
    if (vnodeBufPoolCreate(pVnode, pPool->id, size, &pNewPool) < 0) {
×
229
      vWarn("vgId:%d, failed to change buffer pool of id %d size from %" PRId64 " to %" PRId64 " since %s",
×
230
            TD_VID(pVnode), pPool->id, pPool->node.size, size, tstrerror(errno));
231
    } else {
232
      vInfo("vgId:%d, buffer pool of id %d size changed from %" PRId64 " to %" PRId64, TD_VID(pVnode), pPool->id,
×
233
            pPool->node.size, size);
234

235
      vnodeBufPoolDestroy(pPool);
×
236
      pPool = pNewPool;
×
237
      pVnode->aBufPool[pPool->id] = pPool;
×
238
    }
239
  }
240

241
  // add to free list
242
  vDebug("vgId:%d, buffer pool %p of id %d is added to free list", TD_VID(pVnode), pPool, pPool->id);
31,259✔
243
  vnodeBufPoolReset(pPool);
31,259✔
244
  pPool->freeNext = pVnode->freeList;
31,258✔
245
  pVnode->freeList = pPool;
31,258✔
246
  (void)taosThreadCondSignal(&pVnode->poolNotEmpty);
31,258✔
247
}
31,258✔
248

249
void vnodeBufPoolUnRef(SVBufPool *pPool, bool proactive) {
45,031✔
250
  if (pPool == NULL) return;
45,031!
251

252
  SVnode *pVnode = pPool->pVnode;
45,031✔
253

254
  if (proactive) {
45,031✔
255
    (void)taosThreadMutexLock(&pVnode->mutex);
45,029✔
256
  }
257

258
  if (atomic_sub_fetch_32(&pPool->nRef, 1) > 0) goto _exit;
45,030✔
259

260
  // remove from recycle queue or on-recycle position
261
  if (pVnode->onRecycle == pPool) {
1,530✔
262
    pVnode->onRecycle = NULL;
2✔
263
  } else {
264
    if (pPool->recyclePrev) {
1,528✔
265
      pPool->recyclePrev->recycleNext = pPool->recycleNext;
2✔
266
    } else {
267
      pVnode->recycleHead = pPool->recycleNext;
1,526✔
268
    }
269

270
    if (pPool->recycleNext) {
1,528✔
271
      pPool->recycleNext->recyclePrev = pPool->recyclePrev;
21✔
272
    } else {
273
      pVnode->recycleTail = pPool->recyclePrev;
1,507✔
274
    }
275
    pPool->recyclePrev = pPool->recycleNext = NULL;
1,528✔
276
  }
277

278
  vnodeBufPoolAddToFreeList(pPool);
1,530✔
279

280
_exit:
45,030✔
281
  if (proactive) {
45,030✔
282
    (void)taosThreadMutexUnlock(&pVnode->mutex);
45,028✔
283
  }
284
  return;
45,030✔
285
}
286

287
void vnodeBufPoolRegisterQuery(SVBufPool *pPool, SQueryNode *pQNode) {
4,269,783✔
288
  (void)taosThreadMutexLock(&pPool->mutex);
4,269,783✔
289

290
  pQNode->pNext = pPool->qList.pNext;
4,273,571✔
291
  pQNode->ppNext = &pPool->qList.pNext;
4,273,571✔
292
  pPool->qList.pNext->ppNext = &pQNode->pNext;
4,273,571✔
293
  pPool->qList.pNext = pQNode;
4,273,571✔
294
  pPool->nQuery++;
4,273,571✔
295

296
  (void)taosThreadMutexUnlock(&pPool->mutex);
4,273,571✔
297
}
4,274,354✔
298

299
void vnodeBufPoolDeregisterQuery(SVBufPool *pPool, SQueryNode *pQNode, bool proactive) {
4,274,305✔
300
  int32_t code = 0;
4,274,305✔
301

302
  if (proactive) {
4,274,305!
303
    (void)taosThreadMutexLock(&pPool->mutex);
4,274,454✔
304
  }
305

306
  pQNode->pNext->ppNext = pQNode->ppNext;
4,274,442✔
307
  *pQNode->ppNext = pQNode->pNext;
4,274,442✔
308
  pPool->nQuery--;
4,274,442✔
309

310
  if (proactive) {
4,274,442✔
311
    (void)taosThreadMutexUnlock(&pPool->mutex);
4,274,316✔
312
  }
313
}
4,274,753✔
314

315
int32_t vnodeBufPoolRecycle(SVBufPool *pPool) {
2✔
316
  int32_t code = 0;
2✔
317

318
  SVnode *pVnode = pPool->pVnode;
2✔
319

320
  vDebug("vgId:%d, recycle buffer pool %p of id %d", TD_VID(pVnode), pPool, pPool->id);
2!
321

322
  (void)taosThreadMutexLock(&pPool->mutex);
2✔
323

324
  SQueryNode *pNode = pPool->qList.pNext;
2✔
325
  while (pNode != &pPool->qList) {
4✔
326
    SQueryNode *pTNode = pNode->pNext;
2✔
327

328
    int32_t rc = pNode->reseek(pNode->pQHandle);
2✔
329
    if (rc == 0 || rc == TSDB_CODE_VND_QUERY_BUSY) {
2!
330
      pNode = pTNode;
2✔
331
    } else {
332
      code = rc;
×
333
      goto _exit;
×
334
    }
335
  }
336

337
_exit:
2✔
338
  (void)taosThreadMutexUnlock(&pPool->mutex);
2✔
339
  return code;
2✔
340
}
STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2025 Coveralls, Inc