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

add syncElectTest, config replicaNum myIndex

This commit is contained in:
Minghao Li 2022-04-21 13:59:42 +08:00
parent 672be6f4ff
commit 52b6a9c058
1 changed files with 49 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

101
source/libs/sync/test/syncElectTest.cpp
View File

@ -3,9 +3,8 @@
#include "syncEnv.h" #include "syncEnv.h"
#include "syncIO.h" #include "syncIO.h"
#include "syncInt.h" #include "syncInt.h"
#include "syncRaftLog.h"
#include "syncRaftStore.h"
#include "syncUtil.h" #include "syncUtil.h"
#include "wal.h"
void logTest() { void logTest() {
sTrace("--- sync log test: trace"); sTrace("--- sync log test: trace");
@ -16,43 +15,41 @@ void logTest() {
sFatal("--- sync log test: fatal"); sFatal("--- sync log test: fatal");
} }
uint16_t ports[] = {7010, 7110, 7210, 7310, 7410}; uint16_t gPorts[] = {7010, 7110, 7210, 7310, 7410};
int32_t replicaNum = 3; const char* gDir = "./syncElectTest";
int32_t myIndex = 0; int32_t gVgId = 1234;
SRaftId ids[TSDB_MAX_REPLICA];
SSyncInfo syncInfo;
SSyncFSM* pFsm;
SWal* pWal;
SSyncNode* gSyncNode;
const char* pDir = "./syncElectTest";
SSyncNode* syncNodeInit() {
syncInfo.vgId = 1234;
syncInfo.rpcClient = gSyncIO->clientRpc;
syncInfo.FpSendMsg = syncIOSendMsg;
syncInfo.queue = gSyncIO->pMsgQ;
syncInfo.FpEqMsg = syncIOEqMsg;
syncInfo.pFsm = pFsm;
snprintf(syncInfo.path, sizeof(syncInfo.path), "%s_sync_%d", pDir, myIndex);
void init() {
int code = walInit(); int code = walInit();
assert(code == 0); assert(code == 0);
}
void cleanup() { walCleanUp(); }
SWal* createWal(char* path, int32_t vgId) {
SWalCfg walCfg; SWalCfg walCfg;
memset(&walCfg, 0, sizeof(SWalCfg)); memset(&walCfg, 0, sizeof(SWalCfg));
walCfg.vgId = syncInfo.vgId; walCfg.vgId = vgId;
walCfg.fsyncPeriod = 1000; walCfg.fsyncPeriod = 1000;
walCfg.retentionPeriod = 1000; walCfg.retentionPeriod = 1000;
walCfg.rollPeriod = 1000; walCfg.rollPeriod = 1000;
walCfg.retentionSize = 1000; walCfg.retentionSize = 1000;
walCfg.segSize = 1000; walCfg.segSize = 1000;
walCfg.level = TAOS_WAL_FSYNC; walCfg.level = TAOS_WAL_FSYNC;
SWal* pWal = walOpen(path, &walCfg);
char tmpdir[128];
snprintf(tmpdir, sizeof(tmpdir), "%s_wal_%d", pDir, myIndex);
pWal = walOpen(tmpdir, &walCfg);
assert(pWal != NULL); assert(pWal != NULL);
return pWal;
}
SSyncNode* createSyncNode(int32_t replicaNum, int32_t myIndex, int32_t vgId, SWal* pWal, char* path) {
SSyncInfo syncInfo;
syncInfo.vgId = vgId;
syncInfo.rpcClient = gSyncIO->clientRpc;
syncInfo.FpSendMsg = syncIOSendMsg;
syncInfo.queue = gSyncIO->pMsgQ;
syncInfo.FpEqMsg = syncIOEqMsg;
syncInfo.pFsm = NULL;
snprintf(syncInfo.path, sizeof(syncInfo.path), "%s_sync_replica%d_index%d", path, replicaNum, myIndex);
syncInfo.pWal = pWal; syncInfo.pWal = pWal;
SSyncCfg* pCfg = &syncInfo.syncCfg; SSyncCfg* pCfg = &syncInfo.syncCfg;
@ -60,9 +57,9 @@ SSyncNode* syncNodeInit() {
pCfg->replicaNum = replicaNum; pCfg->replicaNum = replicaNum;
for (int i = 0; i < replicaNum; ++i) { for (int i = 0; i < replicaNum; ++i) {
pCfg->nodeInfo[i].nodePort = ports[i]; pCfg->nodeInfo[i].nodePort = gPorts[i];
snprintf(pCfg->nodeInfo[i].nodeFqdn, sizeof(pCfg->nodeInfo[i].nodeFqdn), "%s", "127.0.0.1"); taosGetFqdn(pCfg->nodeInfo[i].nodeFqdn);
// taosGetFqdn(pCfg->nodeInfo[0].nodeFqdn); // snprintf(pCfg->nodeInfo[i].nodeFqdn, sizeof(pCfg->nodeInfo[i].nodeFqdn), "%s", "127.0.0.1");
} }
SSyncNode* pSyncNode = syncNodeOpen(&syncInfo); SSyncNode* pSyncNode = syncNodeOpen(&syncInfo);
@ -84,46 +81,46 @@ SSyncNode* syncNodeInit() {
return pSyncNode; return pSyncNode;
} }
SSyncNode* syncInitTest() { return syncNodeInit(); } void usage(char* exe) { printf("usage: %s replicaNum myIndex \n", exe); }
void initRaftId(SSyncNode* pSyncNode) {
for (int i = 0; i < replicaNum; ++i) {
ids[i] = pSyncNode->replicasId[i];
char* s = syncUtilRaftId2Str(&ids[i]);
printf("raftId[%d] : %s\n", i, s);
taosMemoryFree(s);
}
}
int main(int argc, char** argv) { int main(int argc, char** argv) {
// taosInitLog((char *)"syncTest.log", 100000, 10);
tsAsyncLog = 0; tsAsyncLog = 0;
sDebugFlag = 143 + 64; sDebugFlag = DEBUG_TRACE + DEBUG_SCREEN + DEBUG_FILE;
if (argc != 3) {
myIndex = 0; usage(argv[0]);
if (argc >= 2) { exit(-1);
myIndex = atoi(argv[1]);
} }
int32_t replicaNum = atoi(argv[1]);
int32_t myIndex = atoi(argv[2]);
int32_t ret = syncIOStart((char*)"127.0.0.1", ports[myIndex]); assert(replicaNum >= 1 && replicaNum <= 5);
assert(myIndex >= 0 && myIndex < replicaNum);
init();
int32_t ret = syncIOStart((char*)"127.0.0.1", gPorts[myIndex]);
assert(ret == 0); assert(ret == 0);
ret = syncEnvStart(); ret = syncEnvStart();
assert(ret == 0); assert(ret == 0);
gSyncNode = syncInitTest(); char walPath[128];
assert(gSyncNode != NULL); snprintf(walPath, sizeof(walPath), "%s_wal_replica%d_index%d", gDir, replicaNum, myIndex);
syncNodeLog2((char*)"", gSyncNode); SWal* pWal = createWal(walPath, gVgId);
initRaftId(gSyncNode); SSyncNode* pSyncNode = createSyncNode(replicaNum, myIndex, gVgId, pWal, (char*)gDir);
assert(pSyncNode != NULL);
syncNodeLog2((char*)"==syncElectTest==", pSyncNode);
//--------------------------- //---------------------------
while (1) { while (1) {
char* s = syncNode2SimpleStr(gSyncNode); char* s = syncNode2SimpleStr(pSyncNode);
sTrace("%s", s); sTrace("%s", s);
taosMemoryFree(s); taosMemoryFree(s);
taosMsleep(1000); taosMsleep(1000);
} }
syncNodeClose(pSyncNode);
walClose(pWal);
syncIOStop();
cleanup();
return 0; return 0;
} }