264 lines
8.3 KiB
C++
264 lines
8.3 KiB
C++
/**
|
|
* @file queue.cpp
|
|
* @author slguan (slguan@taosdata.com)
|
|
* @brief UTIL module queue tests
|
|
* @version 1.0
|
|
* @date 2022-01-27
|
|
*
|
|
* @copyright Copyright (c) 2022
|
|
*
|
|
*/
|
|
#if 0
|
|
#include <gtest/gtest.h>
|
|
#include "tlog.h"
|
|
#include "tprocess.h"
|
|
#include "tqueue.h"
|
|
|
|
typedef struct STestMsg {
|
|
uint16_t msgType;
|
|
void *pCont;
|
|
int contLen;
|
|
int32_t code;
|
|
void *handle; // rpc handle returned to app
|
|
void *ahandle; // app handle set by client
|
|
int noResp; // has response or not(default 0, 0: resp, 1: no resp);
|
|
int persistHandle; // persist handle or not
|
|
} STestMsg;
|
|
|
|
class UtilTesProc : public ::testing::Test {
|
|
public:
|
|
void SetUp() override {
|
|
shm.id = -1;
|
|
for (int32_t i = 0; i < 4000; ++i) {
|
|
body[i] = i % 26 + 'a';
|
|
}
|
|
head.pCont = body;
|
|
head.code = 1;
|
|
head.msgType = 2;
|
|
head.noResp = 3;
|
|
head.persistHandle = 4;
|
|
|
|
taosRemoveDir(TD_TMP_DIR_PATH "td");
|
|
taosMkDir(TD_TMP_DIR_PATH "td");
|
|
tstrncpy(tsLogDir, TD_TMP_DIR_PATH "td", PATH_MAX);
|
|
if (taosInitLog("taosdlog", 1) != 0) {
|
|
printf("failed to init log file\n");
|
|
}
|
|
}
|
|
void TearDown() override { taosDropShm(&shm); }
|
|
|
|
public:
|
|
static STestMsg head;
|
|
static char body[4000];
|
|
static SShm shm;
|
|
static void SetUpTestSuite() {}
|
|
static void TearDownTestSuite() {}
|
|
};
|
|
|
|
SShm UtilTesProc::shm;
|
|
char UtilTesProc::body[4000];
|
|
STestMsg UtilTesProc::head;
|
|
|
|
TEST_F(UtilTesProc, 00_Init_Cleanup) {
|
|
ASSERT_EQ(taosCreateShm(&shm, 1234, 1024 * 1024 * 2), 0);
|
|
|
|
shm.size = 1023;
|
|
SProcCfg cfg = {(ProcConsumeFp)NULL,
|
|
(ProcMallocFp)taosAllocateQitem,
|
|
(ProcFreeFp)taosFreeQitem,
|
|
(ProcMallocFp)taosMemoryMalloc,
|
|
(ProcFreeFp)taosMemoryMalloc,
|
|
(ProcConsumeFp)NULL,
|
|
(ProcMallocFp)taosMemoryMalloc,
|
|
(ProcFreeFp)taosMemoryFree,
|
|
(ProcMallocFp)taosMemoryMalloc,
|
|
(ProcFreeFp)taosMemoryMalloc,
|
|
shm,
|
|
&shm,
|
|
"1234"};
|
|
SProc *proc = dmInitProc(&cfg);
|
|
ASSERT_EQ(proc, nullptr);
|
|
|
|
shm.size = 2468;
|
|
cfg.shm = shm;
|
|
proc = dmInitProc(&cfg);
|
|
ASSERT_NE(proc, nullptr);
|
|
|
|
ASSERT_EQ(dmRunProc(proc), 0);
|
|
dmCleanupProc(proc);
|
|
taosDropShm(&shm);
|
|
}
|
|
|
|
void ConsumeChild1(void *parent, void *pHead, int16_t headLen, void *pBody, int32_t bodyLen, EProcFuncType ftype) {
|
|
STestMsg msg;
|
|
memcpy(&msg, pHead, headLen);
|
|
char body[2000] = {0};
|
|
memcpy(body, pBody, bodyLen);
|
|
|
|
uDebug("====> parent:%" PRId64 " ftype:%d, headLen:%d bodyLen:%d head:%d:%d:%d:%d body:%s <====", (int64_t)parent,
|
|
ftype, headLen, bodyLen, msg.code, msg.msgType, msg.noResp, msg.persistHandle, body);
|
|
taosMemoryFree(pBody);
|
|
taosFreeQitem(pHead);
|
|
}
|
|
|
|
TEST_F(UtilTesProc, 01_Push_Pop_Child) {
|
|
shm.size = 3000;
|
|
ASSERT_EQ(taosCreateShm(&shm, 1235, shm.size), 0);
|
|
SProcCfg cfg = {(ProcConsumeFp)ConsumeChild1,
|
|
(ProcMallocFp)taosAllocateQitem,
|
|
(ProcFreeFp)taosFreeQitem,
|
|
(ProcMallocFp)taosMemoryMalloc,
|
|
(ProcFreeFp)taosMemoryFree,
|
|
(ProcConsumeFp)NULL,
|
|
(ProcMallocFp)taosMemoryMalloc,
|
|
(ProcFreeFp)taosMemoryFree,
|
|
(ProcMallocFp)taosMemoryMalloc,
|
|
(ProcFreeFp)taosMemoryFree,
|
|
shm,
|
|
(void *)((int64_t)1235),
|
|
"1235_c"};
|
|
SProc *cproc = dmInitProc(&cfg);
|
|
ASSERT_NE(cproc, nullptr);
|
|
|
|
ASSERT_NE(dmPutToProcCQueue(cproc, &head, 0, body, 0, 0, 0, DND_FUNC_RSP), 0);
|
|
ASSERT_NE(dmPutToProcCQueue(cproc, &head, 0, body, 0, 0, 0, DND_FUNC_REGIST), 0);
|
|
ASSERT_NE(dmPutToProcCQueue(cproc, &head, 0, body, 0, 0, 0, DND_FUNC_RELEASE), 0);
|
|
ASSERT_NE(dmPutToProcCQueue(cproc, NULL, 12, body, 0, 0, 0, DND_FUNC_REQ), 0);
|
|
ASSERT_NE(dmPutToProcCQueue(cproc, &head, 0, body, 0, 0, 0, DND_FUNC_REQ), 0);
|
|
ASSERT_NE(dmPutToProcCQueue(cproc, &head, shm.size, body, 0, 0, 0, DND_FUNC_REQ), 0);
|
|
ASSERT_NE(dmPutToProcCQueue(cproc, &head, sizeof(STestMsg), body, shm.size, 0, 0, DND_FUNC_REQ), 0);
|
|
|
|
for (int32_t j = 0; j < 1000; j++) {
|
|
int32_t i = 0;
|
|
for (i = 0; i < 20; ++i) {
|
|
ASSERT_EQ(dmPutToProcCQueue(cproc, &head, sizeof(STestMsg), body, i, 0, 0, DND_FUNC_REQ), 0);
|
|
}
|
|
ASSERT_NE(dmPutToProcCQueue(cproc, &head, sizeof(STestMsg), body, i, 0, 0, DND_FUNC_REQ), 0);
|
|
|
|
cfg.isChild = true;
|
|
cfg.name = "1235_p";
|
|
SProc *pproc = dmInitProc(&cfg);
|
|
ASSERT_NE(pproc, nullptr);
|
|
dmRunProc(pproc);
|
|
dmCleanupProc(pproc);
|
|
}
|
|
|
|
dmCleanupProc(cproc);
|
|
taosDropShm(&shm);
|
|
}
|
|
|
|
void ConsumeParent1(void *parent, void *pHead, int16_t headLen, void *pBody, int32_t bodyLen, EProcFuncType ftype) {
|
|
STestMsg msg;
|
|
memcpy(&msg, pHead, headLen);
|
|
char body[2000] = {0};
|
|
memcpy(body, pBody, bodyLen);
|
|
|
|
uDebug("----> parent:%" PRId64 " ftype:%d, headLen:%d bodyLen:%d head:%d:%d:%d:%d body:%s <----", (int64_t)parent,
|
|
ftype, headLen, bodyLen, msg.code, msg.msgType, msg.noResp, msg.persistHandle, body);
|
|
taosMemoryFree(pBody);
|
|
taosMemoryFree(pHead);
|
|
}
|
|
|
|
TEST_F(UtilTesProc, 02_Push_Pop_Parent) {
|
|
shm.size = 3000;
|
|
ASSERT_EQ(taosCreateShm(&shm, 1236, shm.size), 0);
|
|
SProcCfg cfg = {(ProcConsumeFp)NULL,
|
|
(ProcMallocFp)taosAllocateQitem,
|
|
(ProcFreeFp)taosFreeQitem,
|
|
(ProcMallocFp)taosMemoryMalloc,
|
|
(ProcFreeFp)taosMemoryFree,
|
|
(ProcConsumeFp)ConsumeParent1,
|
|
(ProcMallocFp)taosMemoryMalloc,
|
|
(ProcFreeFp)taosMemoryFree,
|
|
(ProcMallocFp)taosMemoryMalloc,
|
|
(ProcFreeFp)taosMemoryFree,
|
|
shm,
|
|
(void *)((int64_t)1236),
|
|
"1236_c"};
|
|
SProc *cproc = dmInitProc(&cfg);
|
|
ASSERT_NE(cproc, nullptr);
|
|
|
|
cfg.name = "1236_p";
|
|
cfg.isChild = true;
|
|
SProc *pproc = dmInitProc(&cfg);
|
|
ASSERT_NE(pproc, nullptr);
|
|
|
|
for (int32_t j = 0; j < 1000; j++) {
|
|
int32_t i = 0;
|
|
for (i = 0; i < 20; ++i) {
|
|
dmPutToProcPQueue(pproc, &head, sizeof(STestMsg), body, i, DND_FUNC_REQ);
|
|
}
|
|
|
|
dmRunProc(cproc);
|
|
dmStopProc(cproc);
|
|
}
|
|
|
|
dmCleanupProc(pproc);
|
|
dmCleanupProc(cproc);
|
|
taosDropShm(&shm);
|
|
}
|
|
|
|
void ConsumeChild3(void *parent, void *pHead, int16_t headLen, void *pBody, int32_t bodyLen, EProcFuncType ftype) {
|
|
STestMsg msg;
|
|
memcpy(&msg, pHead, headLen);
|
|
char body[2000] = {0};
|
|
memcpy(body, pBody, bodyLen);
|
|
|
|
uDebug("====> parent:%" PRId64 " ftype:%d, headLen:%d bodyLen:%d handle:%" PRId64 " body:%s <====", (int64_t)parent,
|
|
ftype, headLen, bodyLen, (int64_t)msg.handle, body);
|
|
taosMemoryFree(pBody);
|
|
taosFreeQitem(pHead);
|
|
}
|
|
|
|
void processHandle(void *handle) { uDebug("----> remove handle:%" PRId64 " <----", (int64_t)handle); }
|
|
|
|
TEST_F(UtilTesProc, 03_Handle) {
|
|
// uDebugFlag = 207;
|
|
shm.size = 3000;
|
|
ASSERT_EQ(taosCreateShm(&shm, 1237, shm.size), 0);
|
|
SProcCfg cfg = {(ProcConsumeFp)ConsumeChild3,
|
|
(ProcMallocFp)taosAllocateQitem,
|
|
(ProcFreeFp)taosFreeQitem,
|
|
(ProcMallocFp)taosMemoryMalloc,
|
|
(ProcFreeFp)taosMemoryFree,
|
|
(ProcConsumeFp)NULL,
|
|
(ProcMallocFp)taosMemoryMalloc,
|
|
(ProcFreeFp)taosMemoryFree,
|
|
(ProcMallocFp)taosMemoryMalloc,
|
|
(ProcFreeFp)taosMemoryFree,
|
|
shm,
|
|
(void *)((int64_t)1235),
|
|
"1237_p"};
|
|
SProc *cproc = dmInitProc(&cfg);
|
|
ASSERT_NE(cproc, nullptr);
|
|
|
|
for (int32_t j = 0; j < 1; j++) {
|
|
int32_t i = 0;
|
|
for (i = 0; i < 20; ++i) {
|
|
head.handle = (void *)((int64_t)i);
|
|
ASSERT_EQ(dmPutToProcCQueue(cproc, &head, sizeof(STestMsg), body, i, (void *)((int64_t)i), i, DND_FUNC_REQ), 0);
|
|
}
|
|
|
|
cfg.isChild = true;
|
|
cfg.name = "child_queue";
|
|
SProc *pproc = dmInitProc(&cfg);
|
|
ASSERT_NE(pproc, nullptr);
|
|
dmRunProc(pproc);
|
|
dmCleanupProc(pproc);
|
|
|
|
int64_t ref = 0;
|
|
|
|
ref = dmRemoveProcRpcHandle(cproc, (void *)((int64_t)3));
|
|
EXPECT_EQ(ref, 3);
|
|
ref = dmRemoveProcRpcHandle(cproc, (void *)((int64_t)5));
|
|
EXPECT_EQ(ref, 5);
|
|
ref = dmRemoveProcRpcHandle(cproc, (void *)((int64_t)6));
|
|
EXPECT_EQ(ref, 6);
|
|
dmCloseProcRpcHandles(cproc, processHandle);
|
|
}
|
|
|
|
dmCleanupProc(cproc);
|
|
taosDropShm(&shm);
|
|
}
|
|
|
|
#endif
|