syncSnapshot.h 3.4 KB
Newer Older
M
Minghao Li 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
/*
 * 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/>.
 */

#ifndef _TD_LIBS_SYNC_SNAPSHOT_H
#define _TD_LIBS_SYNC_SNAPSHOT_H

#ifdef __cplusplus
extern "C" {
#endif

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
M
Minghao Li 已提交
26
#include "cJSON.h"
M
Minghao Li 已提交
27
#include "syncInt.h"
M
Minghao Li 已提交
28
#include "syncMessage.h"
M
Minghao Li 已提交
29 30
#include "taosdef.h"

M
Minghao Li 已提交
31 32 33 34 35
#define SYNC_SNAPSHOT_SEQ_INVALID -1
#define SYNC_SNAPSHOT_SEQ_FORCE_CLOSE -2
#define SYNC_SNAPSHOT_SEQ_BEGIN 0
#define SYNC_SNAPSHOT_SEQ_END 0x7FFFFFFF

M
Minghao Li 已提交
36 37
#define SYNC_SNAPSHOT_RETRY_MS 5000

38
//---------------------------------------------------
M
Minghao Li 已提交
39
typedef struct SSyncSnapshotSender {
M
Minghao Li 已提交
40 41 42
  bool       start;
  int32_t    seq;
  int32_t    ack;
43 44
  void      *pReader;
  void      *pCurrentBlock;
45
  int32_t    blockLen;
M
Minghao Li 已提交
46
  SSnapshot  snapshot;
47
  SSyncCfg   lastConfig;
48
  int64_t    sendingMS;
M
Minghao Li 已提交
49
  SSyncNode *pSyncNode;
M
Minghao Li 已提交
50 51
  int32_t    replicaIndex;
  SyncTerm   term;
52
  SyncTerm   privateTerm;
53
  bool       finish;
M
Minghao Li 已提交
54 55
} SSyncSnapshotSender;

M
Minghao Li 已提交
56
SSyncSnapshotSender *snapshotSenderCreate(SSyncNode *pSyncNode, int32_t replicaIndex);
M
Minghao Li 已提交
57
void                 snapshotSenderDestroy(SSyncSnapshotSender *pSender);
M
Minghao Li 已提交
58
bool                 snapshotSenderIsStart(SSyncSnapshotSender *pSender);
59 60
int32_t              snapshotSenderStart(SSyncSnapshotSender *pSender, SSnapshot snapshot, void *pReader);
int32_t              snapshotSenderStop(SSyncSnapshotSender *pSender, bool finish);
M
Minghao Li 已提交
61 62
int32_t              snapshotSend(SSyncSnapshotSender *pSender);
int32_t              snapshotReSend(SSyncSnapshotSender *pSender);
M
Minghao Li 已提交
63

64
cJSON *snapshotSender2Json(SSyncSnapshotSender *pSender);
65 66
char  *snapshotSender2Str(SSyncSnapshotSender *pSender);
char  *snapshotSender2SimpleStr(SSyncSnapshotSender *pSender, char *event);
67

68 69 70 71
//---------------------------------------------------
typedef struct SSyncSnapshotReceiver {
  bool       start;
  int32_t    ack;
72
  void      *pWriter;
73 74 75
  SyncTerm   term;
  SyncTerm   privateTerm;
  SSnapshot  snapshot;
76
  SRaftId    fromId;
77
  SSyncNode *pSyncNode;
78

M
Minghao Li 已提交
79 80
} SSyncSnapshotReceiver;

81
SSyncSnapshotReceiver *snapshotReceiverCreate(SSyncNode *pSyncNode, SRaftId fromId);
M
Minghao Li 已提交
82
void                   snapshotReceiverDestroy(SSyncSnapshotReceiver *pReceiver);
83 84 85
int32_t snapshotReceiverStart(SSyncSnapshotReceiver *pReceiver, SyncTerm privateTerm, SyncSnapshotSend *pBeginMsg);
int32_t snapshotReceiverStop(SSyncSnapshotReceiver *pReceiver);
bool    snapshotReceiverIsStart(SSyncSnapshotReceiver *pReceiver);
86

M
Minghao Li 已提交
87
cJSON *snapshotReceiver2Json(SSyncSnapshotReceiver *pReceiver);
88 89
char  *snapshotReceiver2Str(SSyncSnapshotReceiver *pReceiver);
char  *snapshotReceiver2SimpleStr(SSyncSnapshotReceiver *pReceiver, char *event);
M
Minghao Li 已提交
90

91 92
//---------------------------------------------------
// on message
M
Minghao Li 已提交
93 94
int32_t syncNodeOnSnapshotSendCb(SSyncNode *ths, SyncSnapshotSend *pMsg);
int32_t syncNodeOnSnapshotRspCb(SSyncNode *ths, SyncSnapshotRsp *pMsg);
M
Minghao Li 已提交
95 96 97 98 99 100

#ifdef __cplusplus
}
#endif

#endif /*_TD_LIBS_SYNC_SNAPSHOT_H*/