126 lines
3.5 KiB
C++
126 lines
3.5 KiB
C++
#include <gtest/gtest.h>
|
|
#include <stdio.h>
|
|
#include "syncEnv.h"
|
|
#include "syncIO.h"
|
|
#include "syncInt.h"
|
|
#include "syncUtil.h"
|
|
#include "wal.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");
|
|
}
|
|
|
|
uint16_t gPorts[] = {7010, 7110, 7210, 7310, 7410};
|
|
const char* gDir = "./syncElectTest";
|
|
int32_t gVgId = 1234;
|
|
|
|
void init() {
|
|
int code = walInit();
|
|
assert(code == 0);
|
|
}
|
|
|
|
void cleanup() { walCleanUp(); }
|
|
|
|
SWal* createWal(char* path, int32_t vgId) {
|
|
SWalCfg walCfg;
|
|
memset(&walCfg, 0, sizeof(SWalCfg));
|
|
walCfg.vgId = vgId;
|
|
walCfg.fsyncPeriod = 1000;
|
|
walCfg.retentionPeriod = 1000;
|
|
walCfg.rollPeriod = 1000;
|
|
walCfg.retentionSize = 1000;
|
|
walCfg.segSize = 1000;
|
|
walCfg.level = TAOS_WAL_FSYNC;
|
|
SWal* pWal = walOpen(path, &walCfg);
|
|
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.msgcb = &gSyncIO->msgcb;
|
|
syncInfo.FpSendMsg = syncIOSendMsg;
|
|
syncInfo.FpEqMsg = syncIOEqMsg;
|
|
syncInfo.pFsm = NULL;
|
|
snprintf(syncInfo.path, sizeof(syncInfo.path), "%s_sync_replica%d_index%d", path, replicaNum, myIndex);
|
|
syncInfo.pWal = pWal;
|
|
|
|
SSyncCfg* pCfg = &syncInfo.syncCfg;
|
|
pCfg->myIndex = myIndex;
|
|
pCfg->replicaNum = replicaNum;
|
|
|
|
for (int i = 0; i < replicaNum; ++i) {
|
|
pCfg->nodeInfo[i].nodePort = gPorts[i];
|
|
taosGetFqdn(pCfg->nodeInfo[i].nodeFqdn);
|
|
// snprintf(pCfg->nodeInfo[i].nodeFqdn, sizeof(pCfg->nodeInfo[i].nodeFqdn), "%s", "127.0.0.1");
|
|
}
|
|
|
|
SSyncNode* pSyncNode = syncNodeOpen(&syncInfo);
|
|
assert(pSyncNode != NULL);
|
|
|
|
gSyncIO->FpOnSyncPing = pSyncNode->FpOnPing;
|
|
gSyncIO->FpOnSyncPingReply = pSyncNode->FpOnPingReply;
|
|
gSyncIO->FpOnSyncRequestVote = pSyncNode->FpOnRequestVote;
|
|
gSyncIO->FpOnSyncRequestVoteReply = pSyncNode->FpOnRequestVoteReply;
|
|
gSyncIO->FpOnSyncAppendEntries = pSyncNode->FpOnAppendEntries;
|
|
gSyncIO->FpOnSyncAppendEntriesReply = pSyncNode->FpOnAppendEntriesReply;
|
|
gSyncIO->FpOnSyncPing = pSyncNode->FpOnPing;
|
|
gSyncIO->FpOnSyncPingReply = pSyncNode->FpOnPingReply;
|
|
gSyncIO->FpOnSyncTimeout = pSyncNode->FpOnTimeout;
|
|
gSyncIO->pSyncNode = pSyncNode;
|
|
|
|
syncNodeStart(pSyncNode);
|
|
|
|
return pSyncNode;
|
|
}
|
|
|
|
void usage(char* exe) { printf("usage: %s replicaNum myIndex \n", exe); }
|
|
|
|
int main(int argc, char** argv) {
|
|
tsAsyncLog = 0;
|
|
sDebugFlag = DEBUG_TRACE + DEBUG_SCREEN + DEBUG_FILE;
|
|
if (argc != 3) {
|
|
usage(argv[0]);
|
|
exit(-1);
|
|
}
|
|
int32_t replicaNum = atoi(argv[1]);
|
|
int32_t myIndex = atoi(argv[2]);
|
|
|
|
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);
|
|
ret = syncEnvStart();
|
|
assert(ret == 0);
|
|
|
|
char walPath[128];
|
|
snprintf(walPath, sizeof(walPath), "%s_wal_replica%d_index%d", gDir, replicaNum, myIndex);
|
|
SWal* pWal = createWal(walPath, gVgId);
|
|
|
|
SSyncNode* pSyncNode = createSyncNode(replicaNum, myIndex, gVgId, pWal, (char*)gDir);
|
|
assert(pSyncNode != NULL);
|
|
syncNodeLog2((char*)"==syncElectTest==", pSyncNode);
|
|
|
|
//---------------------------
|
|
while (1) {
|
|
char* s = syncNode2SimpleStr(pSyncNode);
|
|
sTrace("%s", s);
|
|
taosMemoryFree(s);
|
|
taosMsleep(1000);
|
|
}
|
|
|
|
syncNodeClose(pSyncNode);
|
|
walClose(pWal);
|
|
syncIOStop();
|
|
cleanup();
|
|
return 0;
|
|
}
|