xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

test performance

This commit is contained in:
pengrongkun94@qq.com 2025-02-24 15:20:40 +08:00
parent 6f5f375fcf
commit dd5bee0120
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
source/client/src/clientStmt.c
View File

@ -39,12 +39,19 @@ static FORCE_INLINE int32_t stmtAllocQNodeFromBuf(STableBufInfo* pTblBuf, void**
} }
bool stmtDequeue(STscStmt* pStmt, SStmtQNode** param) { bool stmtDequeue(STscStmt* pStmt, SStmtQNode** param) {
(void)taosThreadMutexLock(&pStmt->queue.mutex); int i = 0;
while (0 == atomic_load_64((int64_t*)&pStmt->queue.qRemainNum)) { while (0 == atomic_load_64((int64_t*)&pStmt->queue.qRemainNum)) {
(void)taosThreadCondWait(&pStmt->queue.waitCond, &pStmt->queue.mutex); if (i < 5000) {
if (atomic_load_8((int8_t*)&pStmt->queue.stopQueue)) { taosUsleep(1);
i++;
} else {
(void)taosThreadMutexLock(&pStmt->queue.mutex);
(void)taosThreadCondWait(&pStmt->queue.waitCond, &pStmt->queue.mutex);
(void)taosThreadMutexUnlock(&pStmt->queue.mutex); (void)taosThreadMutexUnlock(&pStmt->queue.mutex);
return false; if (atomic_load_8((int8_t*)&pStmt->queue.stopQueue)) {
(void)taosThreadMutexUnlock(&pStmt->queue.mutex);
return false;
}
} }
} }
SStmtQNode* orig = pStmt->queue.head; SStmtQNode* orig = pStmt->queue.head;
@ -53,7 +60,6 @@ bool stmtDequeue(STscStmt* pStmt, SStmtQNode** param) {
*param = node; *param = node;
(void)atomic_sub_fetch_64((int64_t*)&pStmt->queue.qRemainNum, 1); (void)atomic_sub_fetch_64((int64_t*)&pStmt->queue.qRemainNum, 1);
(void)taosThreadMutexUnlock(&pStmt->queue.mutex);
*param = node; *param = node;
@ -62,15 +68,15 @@ bool stmtDequeue(STscStmt* pStmt, SStmtQNode** param) {
} }
void stmtEnqueue(STscStmt* pStmt, SStmtQNode* param) { void stmtEnqueue(STscStmt* pStmt, SStmtQNode* param) {
(void)taosThreadMutexLock(&pStmt->queue.mutex);
pStmt->queue.tail->next = param; pStmt->queue.tail->next = param;
pStmt->queue.tail = param; pStmt->queue.tail = param;
pStmt->stat.bindDataNum++; pStmt->stat.bindDataNum++;
(void)atomic_add_fetch_64(&pStmt->queue.qRemainNum, 1); (void)atomic_add_fetch_64(&pStmt->queue.qRemainNum, 1);
(void)taosThreadCondSignal(&(pStmt->queue.waitCond));
(void)taosThreadMutexLock(&pStmt->queue.mutex);
(void)taosThreadCondSignal(&(pStmt->queue.waitCond));
(void)taosThreadMutexUnlock(&pStmt->queue.mutex); (void)taosThreadMutexUnlock(&pStmt->queue.mutex);
} }
@ -423,11 +429,9 @@ void stmtResetQueueTableBuf(STableBufInfo* pTblBuf, SStmtQueue* pQueue) {
pTblBuf->buffIdx = 1; pTblBuf->buffIdx = 1;
pTblBuf->buffOffset = sizeof(*pQueue->head); pTblBuf->buffOffset = sizeof(*pQueue->head);
(void)taosThreadMutexLock(&pQueue->mutex);
pQueue->head = pQueue->tail = pTblBuf->pCurBuff; pQueue->head = pQueue->tail = pTblBuf->pCurBuff;
pQueue->qRemainNum = 0; pQueue->qRemainNum = 0;
pQueue->head->next = NULL; pQueue->head->next = NULL;
(void)taosThreadMutexUnlock(&pQueue->mutex);
} }
int32_t stmtCleanExecInfo(STscStmt* pStmt, bool keepTable, bool deepClean) { int32_t stmtCleanExecInfo(STscStmt* pStmt, bool keepTable, bool deepClean) {