Merge pull request #13096 from taosdata/feature/3.0_mhli

enh(sync) snapshot sender, receiver

source/libs/sync/inc/syncInt.h

@@ -55,6 +55,8 @@ typedef struct SVotesRespond          SVotesRespond;
 typedef struct SSyncIndexMgr          SSyncIndexMgr;
 typedef struct SRaftCfg               SRaftCfg;
 typedef struct SSyncRespMgr           SSyncRespMgr;
+typedef struct SSyncSnapshotSender    SSyncSnapshotSender;
+typedef struct SSyncSnapshotReceiver  SSyncSnapshotReceiver;
 
 typedef struct SSyncNode {
   // init by SSyncInfo
@@ -148,9 +150,11 @@ typedef struct SSyncNode {
   SSyncRespMgr* pSyncRespMgr;
 
   // restore state
-  bool restoreFinish;
   // sem_t      restoreSem;
-  SSnapshot* pSnapshot;
+  bool                   restoreFinish;
+  SSnapshot*             pSnapshot;
+  SSyncSnapshotSender*   pSender;
+  SSyncSnapshotReceiver* pReceiver;
 
 } SSyncNode;
 

source/libs/sync/inc/syncRaftLog.h

@@ -32,20 +32,21 @@ typedef struct SSyncLogStoreData {
   SWal*      pWal;
 } SSyncLogStoreData;
 
-SSyncLogStore*  logStoreCreate(SSyncNode* pSyncNode);
-void            logStoreDestory(SSyncLogStore* pLogStore);
-int32_t         logStoreAppendEntry(SSyncLogStore* pLogStore, SSyncRaftEntry* pEntry);
-SSyncRaftEntry* logStoreGetEntry(SSyncLogStore* pLogStore, SyncIndex index);
-int32_t         logStoreTruncate(SSyncLogStore* pLogStore, SyncIndex fromIndex);
-SyncIndex       logStoreLastIndex(SSyncLogStore* pLogStore);
-SyncTerm        logStoreLastTerm(SSyncLogStore* pLogStore);
-int32_t         logStoreUpdateCommitIndex(SSyncLogStore* pLogStore, SyncIndex index);
-SyncIndex       logStoreGetCommitIndex(SSyncLogStore* pLogStore);
-SSyncRaftEntry* logStoreGetLastEntry(SSyncLogStore* pLogStore);
-cJSON*          logStore2Json(SSyncLogStore* pLogStore);
-char*           logStore2Str(SSyncLogStore* pLogStore);
-cJSON*          logStoreSimple2Json(SSyncLogStore* pLogStore);
-char*           logStoreSimple2Str(SSyncLogStore* pLogStore);
+SSyncLogStore* logStoreCreate(SSyncNode* pSyncNode);
+void           logStoreDestory(SSyncLogStore* pLogStore);
+cJSON*         logStore2Json(SSyncLogStore* pLogStore);
+char*          logStore2Str(SSyncLogStore* pLogStore);
+cJSON*         logStoreSimple2Json(SSyncLogStore* pLogStore);
+char*          logStoreSimple2Str(SSyncLogStore* pLogStore);
+
+// SSyncRaftEntry* logStoreGetLastEntry(SSyncLogStore* pLogStore);
+// SyncIndex       logStoreLastIndex(SSyncLogStore* pLogStore);
+// SyncTerm        logStoreLastTerm(SSyncLogStore* pLogStore);
+// SSyncRaftEntry* logStoreGetEntry(SSyncLogStore* pLogStore, SyncIndex index);
+// int32_t         logStoreAppendEntry(SSyncLogStore* pLogStore, SSyncRaftEntry* pEntry);
+// int32_t         logStoreTruncate(SSyncLogStore* pLogStore, SyncIndex fromIndex);
+// int32_t         logStoreUpdateCommitIndex(SSyncLogStore* pLogStore, SyncIndex index);
+// SyncIndex       logStoreGetCommitIndex(SSyncLogStore* pLogStore);
 
 // for debug
 void logStorePrint(SSyncLogStore* pLogStore);

source/libs/sync/inc/syncSnapshot.h

@@ -23,11 +23,39 @@ extern "C" {
 #include <stdint.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include "cJSON.h"
 #include "syncInt.h"
 #include "taosdef.h"
 
-int32_t takeSnapshot(SSyncFSM *pFsm, SSnapshot *pSnapshot);
-int32_t restoreSnapshot(SSyncFSM *pFsm, SSnapshot *pSnapshot);
+typedef struct SSyncSnapshotSender {
+  bool       isStart;
+  int32_t    progressIndex;
+  void *     pCurrentBlock;
+  int32_t    len;
+  SSnapshot *pSnapshot;
+  SSyncNode *pSyncNode;
+} SSyncSnapshotSender;
+
+SSyncSnapshotSender *snapshotSenderCreate(SSyncNode *pSyncNode);
+void                 snapshotSenderDestroy(SSyncSnapshotSender *pSender);
+int32_t              snapshotSend(SSyncSnapshotSender *pSender);
+cJSON *              snapshotSender2Json(SSyncSnapshotSender *pSender);
+char *               snapshotSender2Str(SSyncSnapshotSender *pSender);
+
+typedef struct SSyncSnapshotReceiver {
+  bool       isStart;
+  int32_t    progressIndex;
+  void *     pCurrentBlock;
+  int32_t    len;
+  SSnapshot *pSnapshot;
+  SSyncNode *pSyncNode;
+} SSyncSnapshotReceiver;
+
+SSyncSnapshotReceiver *snapshotReceiverCreate(SSyncNode *pSyncNode);
+void                   snapshotReceiverDestroy(SSyncSnapshotReceiver *pReceiver);
+int32_t                snapshotReceive(SSyncSnapshotReceiver *pReceiver);
+cJSON *                snapshotReceiver2Json(SSyncSnapshotReceiver *pReceiver);
+char *                 snapshotReceiver2Str(SSyncSnapshotReceiver *pReceiver);
 
 #ifdef __cplusplus
 }

source/libs/sync/src/syncAppendEntries.c

@@ -107,7 +107,7 @@ int32_t syncNodeOnAppendEntriesCb(SSyncNode* ths, SyncAppendEntries* pMsg) {
 
   SyncTerm localPreLogTerm = 0;
   if (pMsg->prevLogIndex >= SYNC_INDEX_BEGIN && pMsg->prevLogIndex <= ths->pLogStore->getLastIndex(ths->pLogStore)) {
-    SSyncRaftEntry* pEntry = logStoreGetEntry(ths->pLogStore, pMsg->prevLogIndex);
+    SSyncRaftEntry* pEntry = ths->pLogStore->getEntry(ths->pLogStore, pMsg->prevLogIndex);
     assert(pEntry != NULL);
     localPreLogTerm = pEntry->term;
     syncEntryDestory(pEntry);
@@ -175,7 +175,7 @@ int32_t syncNodeOnAppendEntriesCb(SSyncNode* ths, SyncAppendEntries* pMsg) {
       bool conflict = false;
 
       SyncIndex       extraIndex = pMsg->prevLogIndex + 1;
-      SSyncRaftEntry* pExtraEntry = logStoreGetEntry(ths->pLogStore, extraIndex);
+      SSyncRaftEntry* pExtraEntry = ths->pLogStore->getEntry(ths->pLogStore, extraIndex);
       assert(pExtraEntry != NULL);
 
       SSyncRaftEntry* pAppendEntry = syncEntryDeserialize(pMsg->data, pMsg->dataLen);
@@ -197,7 +197,7 @@ int32_t syncNodeOnAppendEntriesCb(SSyncNode* ths, SyncAppendEntries* pMsg) {
         // notice! reverse roll back!
         for (SyncIndex index = delEnd; index >= delBegin; --index) {
           if (ths->pFsm->FpRollBackCb != NULL) {
-            SSyncRaftEntry* pRollBackEntry = logStoreGetEntry(ths->pLogStore, index);
+            SSyncRaftEntry* pRollBackEntry = ths->pLogStore->getEntry(ths->pLogStore, index);
             assert(pRollBackEntry != NULL);
 
             // if (pRollBackEntry->msgType != TDMT_VND_SYNC_NOOP) {
@@ -365,7 +365,7 @@ int32_t syncNodeOnAppendEntriesCb(SSyncNode* ths, SyncAppendEntries* pMsg) {
                 }
 
                 SReConfigCbMeta cbMeta = {0};
-                bool isDrop;
+                bool            isDrop;
 
                 // I am in newConfig
                 if (hit) {
@@ -388,7 +388,7 @@ int32_t syncNodeOnAppendEntriesCb(SSyncNode* ths, SyncAppendEntries* pMsg) {
                 }
 
                 // always call FpReConfigCb
-                if (ths->pFsm->FpReConfigCb != NULL) {     
+                if (ths->pFsm->FpReConfigCb != NULL) {
                   cbMeta.code = 0;
                   cbMeta.currentTerm = ths->pRaftStore->currentTerm;
                   cbMeta.index = pEntry->index;

source/libs/sync/src/syncRaftLog.c

@@ -16,6 +16,15 @@
 #include "syncRaftLog.h"
 #include "wal.h"
 
+static SSyncRaftEntry* logStoreGetLastEntry(SSyncLogStore* pLogStore);
+static SyncIndex       logStoreLastIndex(SSyncLogStore* pLogStore);
+static SyncTerm        logStoreLastTerm(SSyncLogStore* pLogStore);
+static SSyncRaftEntry* logStoreGetEntry(SSyncLogStore* pLogStore, SyncIndex index);
+static int32_t         logStoreAppendEntry(SSyncLogStore* pLogStore, SSyncRaftEntry* pEntry);
+static int32_t         logStoreTruncate(SSyncLogStore* pLogStore, SyncIndex fromIndex);
+static int32_t         logStoreUpdateCommitIndex(SSyncLogStore* pLogStore, SyncIndex index);
+static SyncIndex       logStoreGetCommitIndex(SSyncLogStore* pLogStore);
+
 SSyncLogStore* logStoreCreate(SSyncNode* pSyncNode) {
   SSyncLogStore* pLogStore = taosMemoryMalloc(sizeof(SSyncLogStore));
   assert(pLogStore != NULL);
@@ -78,7 +87,9 @@ SSyncRaftEntry* logStoreGetEntry(SSyncLogStore* pLogStore, SyncIndex index) {
 
   if (index >= SYNC_INDEX_BEGIN && index <= logStoreLastIndex(pLogStore)) {
     SWalReadHandle* pWalHandle = walOpenReadHandle(pWal);
-    int32_t         code = walReadWithHandle(pWalHandle, index);
+    ASSERT(pWalHandle != NULL);
+
+    int32_t code = walReadWithHandle(pWalHandle, index);
     if (code != 0) {
       int32_t     err = terrno;
       const char* errStr = tstrerror(err);

source/libs/sync/src/syncReplication.c

@@ -75,7 +75,7 @@ int32_t syncNodeAppendEntriesPeers(SSyncNode* pSyncNode) {
     // SyncIndex lastIndex = syncUtilMinIndex(pSyncNode->pLogStore->getLastIndex(pSyncNode->pLogStore), nextIndex);
 
     SyncAppendEntries* pMsg = NULL;
-    SSyncRaftEntry*    pEntry = logStoreGetEntry(pSyncNode->pLogStore, nextIndex);
+    SSyncRaftEntry*    pEntry = pSyncNode->pLogStore->getEntry(pSyncNode->pLogStore, nextIndex);
     if (pEntry != NULL) {
       pMsg = syncAppendEntriesBuild(pEntry->bytes, pSyncNode->vgId);
       assert(pMsg != NULL);

source/libs/sync/src/syncSnapshot.c

@@ -15,6 +15,22 @@
 
 #include "syncSnapshot.h"
 
-int32_t takeSnapshot(SSyncFSM *pFsm, SSnapshot *pSnapshot) { return 0; }
+SSyncSnapshotSender *snapshotSenderCreate(SSyncNode *pSyncNode) { return NULL; }
 
-int32_t restoreSnapshot(SSyncFSM *pFsm, SSnapshot *pSnapshot) { return 0; }
+void snapshotSenderDestroy(SSyncSnapshotSender *pSender) {}
\ No newline at end of file
+
+int32_t snapshotSend(SSyncSnapshotSender *pSender) { return 0; }
+
+cJSON *snapshotSender2Json(SSyncSnapshotSender *pSender) { return NULL; }
+
+char *snapshotSender2Str(SSyncSnapshotSender *pSender) { return NULL; }
+
+SSyncSnapshotReceiver *snapshotReceiverCreate(SSyncNode *pSyncNode) { return NULL; }
+
+void snapshotReceiverDestroy(SSyncSnapshotReceiver *pReceiver) {}
+
+int32_t snapshotReceive(SSyncSnapshotReceiver *pReceiver) { return 0; }
+
+cJSON *snapshotReceiver2Json(SSyncSnapshotReceiver *pReceiver) { return NULL; }
+
+char *snapshotReceiver2Str(SSyncSnapshotReceiver *pReceiver) { return NULL; }

source/libs/sync/test/syncConfigChangeTest.cpp

@@ -78,7 +78,8 @@ int32_t GetSnapshotCb(struct SSyncFSM* pFsm, SSnapshot* pSnapshot) {
 void RestoreFinishCb(struct SSyncFSM* pFsm) { sTrace("==callback== ==RestoreFinishCb=="); }
 
 void ReConfigCb(struct SSyncFSM* pFsm, SSyncCfg newCfg, SReConfigCbMeta cbMeta) {
-  sTrace("==callback== ==ReConfigCb== flag:0x%lX, isDrop:%d, index:%ld, code:%d, currentTerm:%lu, term:%lu", cbMeta.flag, cbMeta.isDrop, cbMeta.index, cbMeta.code, cbMeta.currentTerm, cbMeta.term);
+  sTrace("==callback== ==ReConfigCb== flag:0x%lX, isDrop:%d, index:%ld, code:%d, currentTerm:%lu, term:%lu",
+         cbMeta.flag, cbMeta.isDrop, cbMeta.index, cbMeta.code, cbMeta.currentTerm, cbMeta.term);
 }
 
 SSyncFSM* createFsm() {