diff --git a/source/libs/sync/test/CMakeLists.txt b/source/libs/sync/test/CMakeLists.txt index 0c7608f902..3fcb9087ab 100644 --- a/source/libs/sync/test/CMakeLists.txt +++ b/source/libs/sync/test/CMakeLists.txt @@ -31,6 +31,7 @@ add_executable(syncElectTest "") add_executable(syncEncodeTest "") add_executable(syncWriteTest "") add_executable(syncReplicateTest "") +add_executable(syncRefTest "") target_sources(syncTest @@ -165,6 +166,10 @@ target_sources(syncReplicateTest PRIVATE "syncReplicateTest.cpp" ) +target_sources(syncRefTest + PRIVATE + "syncRefTest.cpp" +) target_include_directories(syncTest @@ -337,6 +342,11 @@ target_include_directories(syncReplicateTest "${CMAKE_SOURCE_DIR}/include/libs/sync" "${CMAKE_CURRENT_SOURCE_DIR}/../inc" ) +target_include_directories(syncRefTest + PUBLIC + "${CMAKE_SOURCE_DIR}/include/libs/sync" + "${CMAKE_CURRENT_SOURCE_DIR}/../inc" +) target_link_libraries(syncTest @@ -471,6 +481,10 @@ target_link_libraries(syncReplicateTest sync gtest_main ) +target_link_libraries(syncRefTest + sync + gtest_main +) enable_testing() diff --git a/source/libs/sync/test/syncElectTest.cpp b/source/libs/sync/test/syncElectTest.cpp index 0e0b57a025..47f55aa660 100644 --- a/source/libs/sync/test/syncElectTest.cpp +++ b/source/libs/sync/test/syncElectTest.cpp @@ -116,9 +116,10 @@ int main(int argc, char** argv) { //--------------------------- while (1) { - sTrace("elect sleep, state: %d, %s, term:%lu electTimerLogicClock:%lu, electTimerLogicClockUser:%lu, electTimerMS:%d", - gSyncNode->state, syncUtilState2String(gSyncNode->state), gSyncNode->pRaftStore->currentTerm, gSyncNode->electTimerLogicClock, - gSyncNode->electTimerLogicClockUser, gSyncNode->electTimerMS); + sTrace( + "elect sleep, state: %d, %s, term:%lu electTimerLogicClock:%lu, electTimerLogicClockUser:%lu, electTimerMS:%d", + gSyncNode->state, syncUtilState2String(gSyncNode->state), gSyncNode->pRaftStore->currentTerm, + gSyncNode->electTimerLogicClock, gSyncNode->electTimerLogicClockUser, gSyncNode->electTimerMS); } return 0; diff --git a/source/libs/sync/test/syncRefTest.cpp b/source/libs/sync/test/syncRefTest.cpp new file mode 100644 index 0000000000..8e6061a8ae --- /dev/null +++ b/source/libs/sync/test/syncRefTest.cpp @@ -0,0 +1,134 @@ +#include +#include +#include "syncIO.h" +#include "syncInt.h" +#include "syncRaftStore.h" +#include "tref.h" + +void logTest() { + sTrace("--- sync log test: trace"); + sDebug("--- sync log test: debug"); + sInfo("--- sync log test: info"); + sWarn("--- sync log test: warn"); + sError("--- sync log test: error"); + sFatal("--- sync log test: fatal"); +} + +static void syncFreeObj(void *param); +int32_t init(); +void cleanup(); +int64_t start(); +void stop(int64_t rid); + +static int32_t tsNodeRefId = -1; +int g = 100; + +typedef struct SyncObj { + int64_t rid; + void * data; + char name[32]; + int counter; +} SyncObj; + +static void syncFreeObj(void *param) { + SyncObj *pObj = (SyncObj *)param; + printf("syncFreeObj name:%s rid:%ld \n", pObj->name, pObj->rid); + free(pObj); +} + +int32_t init() { + tsNodeRefId = taosOpenRef(200, syncFreeObj); + if (tsNodeRefId < 0) { + sError("failed to init node ref"); + cleanup(); + return -1; + } + return 0; +} + +void cleanup() { + if (tsNodeRefId != -1) { + taosCloseRef(tsNodeRefId); + tsNodeRefId = -1; + } +} + +int64_t start() { + SyncObj *pObj = (SyncObj *)malloc(sizeof(SyncObj)); + assert(pObj != NULL); + + pObj->data = &g; + snprintf(pObj->name, sizeof(pObj->name), "%s", "hello"); + + pObj->rid = taosAddRef(tsNodeRefId, pObj); + if (pObj->rid < 0) { + syncFreeObj(pObj); + return -1; + } + + printf("start name:%s rid:%ld \n", pObj->name, pObj->rid); + return pObj->rid; +} + +void stop(int64_t rid) { + SyncObj *pObj = (SyncObj *)taosAcquireRef(tsNodeRefId, rid); + if (pObj == NULL) return; + + printf("stop name:%s rid:%ld \n", pObj->name, pObj->rid); + pObj->data = NULL; + + taosReleaseRef(tsNodeRefId, pObj->rid); + taosRemoveRef(tsNodeRefId, rid); +} + +void *func(void *param) { + int64_t rid = (int64_t)param; + + int32_t ms = taosRand() % 10000; + taosMsleep(ms); + + SyncObj *pObj = (SyncObj *)taosAcquireRef(tsNodeRefId, rid); + if (pObj != NULL) { + printf("taosAcquireRef sleep:%d, name:%s, rid:%ld \n", ms, pObj->name, pObj->rid); + } else { + printf("taosAcquireRef sleep:%d, NULL! \n", ms); + } + + taosReleaseRef(tsNodeRefId, rid); +} + +int main() { + // taosInitLog((char *)"syncTest.log", 100000, 10); + tsAsyncLog = 0; + sDebugFlag = 143 + 64; + logTest(); + + taosSeedRand(taosGetTimestampSec()); + + int32_t ret; + + ret = init(); + assert(ret == 0); + + int64_t rid = start(); + assert(rid > 0); + + for (int i = 0; i < 20; ++i) { + pthread_t tid; + pthread_create(&tid, NULL, func, (void*)rid); + } + + int32_t ms = taosRand() % 10000; + taosMsleep(ms); + printf("main sleep %d, stop and clean ", ms); + + stop(rid); + cleanup(); + + while (1) { + taosMsleep(1000); + printf("sleep 1 ... \n"); + } + + return 0; +} diff --git a/source/libs/sync/test/syncReplicateTest.cpp b/source/libs/sync/test/syncReplicateTest.cpp index 6c4fab2425..4d6e6f3a25 100644 --- a/source/libs/sync/test/syncReplicateTest.cpp +++ b/source/libs/sync/test/syncReplicateTest.cpp @@ -172,15 +172,19 @@ int main(int argc, char **argv) { gSyncNode->FpEqMsg(gSyncNode->queue, &rpcMsg); taosMsleep(1000); - sTrace("replicate sleep, state: %d, %s, term:%lu electTimerLogicClock:%lu, electTimerLogicClockUser:%lu, electTimerMS:%d", - gSyncNode->state, syncUtilState2String(gSyncNode->state), gSyncNode->pRaftStore->currentTerm, gSyncNode->electTimerLogicClock, - gSyncNode->electTimerLogicClockUser, gSyncNode->electTimerMS); + sTrace( + "replicate sleep, state: %d, %s, term:%lu electTimerLogicClock:%lu, electTimerLogicClockUser:%lu, " + "electTimerMS:%d", + gSyncNode->state, syncUtilState2String(gSyncNode->state), gSyncNode->pRaftStore->currentTerm, + gSyncNode->electTimerLogicClock, gSyncNode->electTimerLogicClockUser, gSyncNode->electTimerMS); } while (1) { - sTrace("replicate sleep, state: %d, %s, term:%lu electTimerLogicClock:%lu, electTimerLogicClockUser:%lu, electTimerMS:%d", - gSyncNode->state, syncUtilState2String(gSyncNode->state), gSyncNode->pRaftStore->currentTerm, gSyncNode->electTimerLogicClock, - gSyncNode->electTimerLogicClockUser, gSyncNode->electTimerMS); + sTrace( + "replicate sleep, state: %d, %s, term:%lu electTimerLogicClock:%lu, electTimerLogicClockUser:%lu, " + "electTimerMS:%d", + gSyncNode->state, syncUtilState2String(gSyncNode->state), gSyncNode->pRaftStore->currentTerm, + gSyncNode->electTimerLogicClock, gSyncNode->electTimerLogicClockUser, gSyncNode->electTimerMS); taosMsleep(1000); }