147 lines
3.9 KiB
C
147 lines
3.9 KiB
C
/*
|
|
* Copyright (c) 2019 TAOS Data, Inc. <cli@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/>.
|
|
*/
|
|
|
|
#ifndef _TD_LIBS_SYNC_RAFT_MESSAGE_H
|
|
#define _TD_LIBS_SYNC_RAFT_MESSAGE_H
|
|
|
|
#include "sync.h"
|
|
#include "sync_type.h"
|
|
|
|
/**
|
|
* below define message type which handled by Raft node thread
|
|
* internal message, which communicate in threads, start with RAFT_MSG_INTERNAL_*,
|
|
* internal message use pointer only, need not to be decode/encode
|
|
* outter message start with RAFT_MSG_*, need to implement its decode/encode functions
|
|
**/
|
|
typedef enum RaftMessageType {
|
|
// client propose a cmd
|
|
RAFT_MSG_INTERNAL_PROP = 1,
|
|
|
|
// node election timeout
|
|
RAFT_MSG_INTERNAL_ELECTION = 2,
|
|
|
|
RAFT_MSG_VOTE = 3,
|
|
RAFT_MSG_VOTE_RESP = 4,
|
|
|
|
RAFT_MSG_APPEND = 5,
|
|
} RaftMessageType;
|
|
|
|
typedef struct RaftMsgInternal_Prop {
|
|
const SSyncBuffer *pBuf;
|
|
bool isWeak;
|
|
void* pData;
|
|
} RaftMsgInternal_Prop;
|
|
|
|
typedef struct RaftMsgInternal_Election {
|
|
|
|
} RaftMsgInternal_Election;
|
|
|
|
typedef struct RaftMsg_Vote {
|
|
SyncRaftElectionType cType;
|
|
SyncIndex lastIndex;
|
|
SyncTerm lastTerm;
|
|
} RaftMsg_Vote;
|
|
|
|
typedef struct RaftMsg_VoteResp {
|
|
bool reject;
|
|
SyncRaftElectionType cType;
|
|
} RaftMsg_VoteResp;
|
|
|
|
typedef struct SSyncMessage {
|
|
RaftMessageType msgType;
|
|
SyncTerm term;
|
|
SyncGroupId groupId;
|
|
SyncNodeId from;
|
|
SyncNodeId to;
|
|
|
|
union {
|
|
RaftMsgInternal_Prop propose;
|
|
|
|
RaftMsgInternal_Election election;
|
|
|
|
RaftMsg_Vote vote;
|
|
RaftMsg_VoteResp voteResp;
|
|
};
|
|
} SSyncMessage;
|
|
|
|
static FORCE_INLINE SSyncMessage* syncInitPropMsg(SSyncMessage* pMsg, const SSyncBuffer* pBuf, void* pData, bool isWeak) {
|
|
*pMsg = (SSyncMessage) {
|
|
.msgType = RAFT_MSG_INTERNAL_PROP,
|
|
.term = 0,
|
|
.propose = (RaftMsgInternal_Prop) {
|
|
.isWeak = isWeak,
|
|
.pBuf = pBuf,
|
|
.pData = pData,
|
|
},
|
|
};
|
|
|
|
return pMsg;
|
|
}
|
|
|
|
static FORCE_INLINE SSyncMessage* syncInitElectionMsg(SSyncMessage* pMsg, SyncNodeId from) {
|
|
*pMsg = (SSyncMessage) {
|
|
.msgType = RAFT_MSG_INTERNAL_ELECTION,
|
|
.term = 0,
|
|
.from = from,
|
|
.election = (RaftMsgInternal_Election) {
|
|
|
|
},
|
|
};
|
|
|
|
return pMsg;
|
|
}
|
|
|
|
static FORCE_INLINE SSyncMessage* syncNewVoteMsg(SyncGroupId groupId, SyncNodeId from, SyncNodeId to,
|
|
SyncTerm term, SyncRaftElectionType cType,
|
|
SyncIndex lastIndex, SyncTerm lastTerm) {
|
|
SSyncMessage* pMsg = (SSyncMessage*)malloc(sizeof(SSyncMessage));
|
|
if (pMsg == NULL) {
|
|
return NULL;
|
|
}
|
|
*pMsg = (SSyncMessage) {
|
|
.groupId = groupId,
|
|
.from = from,
|
|
.to = to,
|
|
.term = term,
|
|
.vote = (RaftMsg_Vote) {
|
|
.cType = cType,
|
|
.lastIndex = lastIndex,
|
|
.lastTerm = lastTerm,
|
|
},
|
|
};
|
|
|
|
return pMsg;
|
|
}
|
|
|
|
static FORCE_INLINE bool syncIsInternalMsg(RaftMessageType msgType) {
|
|
return msgType == RAFT_MSG_INTERNAL_PROP ||
|
|
msgType == RAFT_MSG_INTERNAL_ELECTION;
|
|
}
|
|
|
|
static FORCE_INLINE bool syncIsPreVoteRespMsg(SSyncMessage* pMsg) {
|
|
return pMsg->msgType == RAFT_MSG_VOTE_RESP && pMsg->voteResp.cType == SYNC_RAFT_CAMPAIGN_PRE_ELECTION;
|
|
}
|
|
|
|
static FORCE_INLINE bool syncIsPreVoteMsg(const SSyncMessage* pMsg) {
|
|
return pMsg->msgType == RAFT_MSG_VOTE && pMsg->voteResp.cType == SYNC_RAFT_CAMPAIGN_PRE_ELECTION;
|
|
}
|
|
|
|
void syncFreeMessage(const SSyncMessage* pMsg);
|
|
|
|
// message handlers
|
|
int syncRaftHandleElectionMessage(SSyncRaft* pRaft, const SSyncMessage* pMsg);
|
|
int syncRaftHandleVoteRespMessage(SSyncRaft* pRaft, const SSyncMessage* pMsg);
|
|
|
|
#endif /* _TD_LIBS_SYNC_RAFT_MESSAGE_H */ |