132 lines
4.7 KiB
C
132 lines
4.7 KiB
C
/*
|
|
* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
|
|
*
|
|
* This program is free software: you can use, redistribute, and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3
|
|
* or later ("AGPL"), as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#define _DEFAULT_SOURCE
|
|
#include "syncReplication.h"
|
|
#include "syncIndexMgr.h"
|
|
#include "syncPipeline.h"
|
|
#include "syncRaftEntry.h"
|
|
#include "syncRaftStore.h"
|
|
#include "syncUtil.h"
|
|
|
|
// TLA+ Spec
|
|
// AppendEntries(i, j) ==
|
|
// /\ i /= j
|
|
// /\ state[i] = Leader
|
|
// /\ LET prevLogIndex == nextIndex[i][j] - 1
|
|
// prevLogTerm == IF prevLogIndex > 0 THEN
|
|
// log[i][prevLogIndex].term
|
|
// ELSE
|
|
// 0
|
|
// \* Send up to 1 entry, constrained by the end of the log.
|
|
// lastEntry == Min({Len(log[i]), nextIndex[i][j]})
|
|
// entries == SubSeq(log[i], nextIndex[i][j], lastEntry)
|
|
// IN Send([mtype |-> AppendEntriesRequest,
|
|
// mterm |-> currentTerm[i],
|
|
// mprevLogIndex |-> prevLogIndex,
|
|
// mprevLogTerm |-> prevLogTerm,
|
|
// mentries |-> entries,
|
|
// \* mlog is used as a history variable for the proof.
|
|
// \* It would not exist in a real implementation.
|
|
// mlog |-> log[i],
|
|
// mcommitIndex |-> Min({commitIndex[i], lastEntry}),
|
|
// msource |-> i,
|
|
// mdest |-> j])
|
|
// /\ UNCHANGED <<serverVars, candidateVars, leaderVars, logVars>>
|
|
|
|
int32_t syncNodeMaybeSendAppendEntries(SSyncNode* pSyncNode, const SRaftId* destRaftId, SRpcMsg* pRpcMsg);
|
|
|
|
int32_t syncNodeReplicateReset(SSyncNode* pNode, SRaftId* pDestId) {
|
|
SSyncLogBuffer* pBuf = pNode->pLogBuf;
|
|
taosThreadMutexLock(&pBuf->mutex);
|
|
SSyncLogReplMgr* pMgr = syncNodeGetLogReplMgr(pNode, pDestId);
|
|
syncLogReplReset(pMgr);
|
|
taosThreadMutexUnlock(&pBuf->mutex);
|
|
return 0;
|
|
}
|
|
|
|
int32_t syncNodeReplicate(SSyncNode* pNode) {
|
|
SSyncLogBuffer* pBuf = pNode->pLogBuf;
|
|
taosThreadMutexLock(&pBuf->mutex);
|
|
int32_t ret = syncNodeReplicateWithoutLock(pNode);
|
|
taosThreadMutexUnlock(&pBuf->mutex);
|
|
return ret;
|
|
}
|
|
|
|
int32_t syncNodeReplicateWithoutLock(SSyncNode* pNode) {
|
|
if (pNode->state != TAOS_SYNC_STATE_LEADER || pNode->replicaNum == 1) {
|
|
return -1;
|
|
}
|
|
for (int32_t i = 0; i < pNode->replicaNum; i++) {
|
|
if (syncUtilSameId(&pNode->replicasId[i], &pNode->myRaftId)) {
|
|
continue;
|
|
}
|
|
SSyncLogReplMgr* pMgr = pNode->logReplMgrs[i];
|
|
(void)syncLogReplReplicateOnce(pMgr, pNode);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int32_t syncNodeSendAppendEntries(SSyncNode* pSyncNode, const SRaftId* destRaftId, SRpcMsg* pRpcMsg) {
|
|
SyncAppendEntries* pMsg = pRpcMsg->pCont;
|
|
pMsg->destId = *destRaftId;
|
|
syncNodeSendMsgById(destRaftId, pSyncNode, pRpcMsg);
|
|
return 0;
|
|
}
|
|
|
|
int32_t syncNodeMaybeSendAppendEntries(SSyncNode* pSyncNode, const SRaftId* destRaftId, SRpcMsg* pRpcMsg) {
|
|
int32_t ret = 0;
|
|
SyncAppendEntries* pMsg = pRpcMsg->pCont;
|
|
|
|
if (syncNodeNeedSendAppendEntries(pSyncNode, destRaftId, pMsg)) {
|
|
ret = syncNodeSendAppendEntries(pSyncNode, destRaftId, pRpcMsg);
|
|
} else {
|
|
sNTrace(pSyncNode, "do not repcate to dnode:%d for index:%" PRId64, DID(destRaftId), pMsg->prevLogIndex + 1);
|
|
rpcFreeCont(pRpcMsg->pCont);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int32_t syncNodeSendHeartbeat(SSyncNode* pSyncNode, const SRaftId* destId, SRpcMsg* pMsg) {
|
|
return syncNodeSendMsgById(destId, pSyncNode, pMsg);
|
|
}
|
|
|
|
int32_t syncNodeHeartbeatPeers(SSyncNode* pSyncNode) {
|
|
int64_t ts = taosGetTimestampMs();
|
|
for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
|
|
SRpcMsg rpcMsg = {0};
|
|
if (syncBuildHeartbeat(&rpcMsg, pSyncNode->vgId) != 0) {
|
|
sError("vgId:%d, build sync-heartbeat error", pSyncNode->vgId);
|
|
continue;
|
|
}
|
|
|
|
SyncHeartbeat* pSyncMsg = rpcMsg.pCont;
|
|
pSyncMsg->srcId = pSyncNode->myRaftId;
|
|
pSyncMsg->destId = pSyncNode->peersId[i];
|
|
pSyncMsg->term = raftStoreGetTerm(pSyncNode);
|
|
pSyncMsg->commitIndex = pSyncNode->commitIndex;
|
|
pSyncMsg->minMatchIndex = syncMinMatchIndex(pSyncNode);
|
|
pSyncMsg->privateTerm = 0;
|
|
pSyncMsg->timeStamp = ts;
|
|
|
|
// send msg
|
|
syncLogSendHeartbeat(pSyncNode, pSyncMsg, true, 0, 0);
|
|
syncNodeSendHeartbeat(pSyncNode, &pSyncMsg->destId, &rpcMsg);
|
|
}
|
|
|
|
return 0;
|
|
}
|