syncIO.h 2.5 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 26 27
/*
 * 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_IO_H
#define _TD_LIBS_IO_H

#ifdef __cplusplus
extern "C" {
#endif

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include "os.h"
#include "syncInt.h"
28
#include "syncTest.h"
M
Minghao Li 已提交
29 30 31 32
#include "taosdef.h"
#include "tqueue.h"
#include "trpc.h"

H
Hongze Cheng 已提交
33
#define TICK_Q_TIMER_MS    1000
M
Minghao Li 已提交
34 35
#define TICK_Ping_TIMER_MS 1000

36 37 38
typedef struct SyncPing      SyncPing;
typedef struct SyncPingReply SyncPingReply;

S
Shengliang Guan 已提交
39 40
extern bool gRaftDetailLog;

M
Minghao Li 已提交
41 42
typedef struct SSyncIO {
  STaosQueue *pMsgQ;
H
Hongze Cheng 已提交
43
  STaosQset  *pQset;
M
Minghao Li 已提交
44
  TdThread    consumerTid;
H
Hongze Cheng 已提交
45 46
  void       *serverRpc;
  void       *clientRpc;
S
Shengliang Guan 已提交
47 48
  SEpSet      myAddr;
  SMsgCb      msgcb;
M
Minghao Li 已提交
49

M
Minghao Li 已提交
50 51 52 53 54
  tmr_h   qTimer;
  int32_t qTimerMS;
  tmr_h   pingTimer;
  int32_t pingTimerMS;
  tmr_h   timerMgr;
M
Minghao Li 已提交
55

M
Minghao Li 已提交
56
  void *pSyncNode;
M
sync io  
Minghao Li 已提交
57 58
  int32_t (*FpOnSyncPing)(SSyncNode *pSyncNode, SyncPing *pMsg);
  int32_t (*FpOnSyncPingReply)(SSyncNode *pSyncNode, SyncPingReply *pMsg);
59
  int32_t (*FpOnSyncClientRequest)(SSyncNode *pSyncNode, SRpcMsg *pMsg, SyncIndex *pRetIndex);
M
sync io  
Minghao Li 已提交
60 61 62 63
  int32_t (*FpOnSyncRequestVote)(SSyncNode *pSyncNode, SyncRequestVote *pMsg);
  int32_t (*FpOnSyncRequestVoteReply)(SSyncNode *pSyncNode, SyncRequestVoteReply *pMsg);
  int32_t (*FpOnSyncAppendEntries)(SSyncNode *pSyncNode, SyncAppendEntries *pMsg);
  int32_t (*FpOnSyncAppendEntriesReply)(SSyncNode *pSyncNode, SyncAppendEntriesReply *pMsg);
M
Minghao Li 已提交
64
  int32_t (*FpOnSyncTimeout)(SSyncNode *pSyncNode, SyncTimeout *pMsg);
M
Minghao Li 已提交
65 66
  int32_t (*FpOnSyncSnapshot)(SSyncNode *pSyncNode, SyncSnapshotSend *pMsg);
  int32_t (*FpOnSyncSnapshotReply)(SSyncNode *pSyncNode, SyncSnapshotRsp *pMsg);
M
Minghao Li 已提交
67

M
sync io  
Minghao Li 已提交
68
  int8_t isStart;
M
Minghao Li 已提交
69

M
Minghao Li 已提交
70 71
} SSyncIO;

M
Minghao Li 已提交
72
extern SSyncIO *gSyncIO;
M
Minghao Li 已提交
73

M
sync io  
Minghao Li 已提交
74
int32_t syncIOStart(char *host, uint16_t port);
M
Minghao Li 已提交
75
int32_t syncIOStop();
S
Shengliang Guan 已提交
76 77
int32_t syncIOSendMsg(const SEpSet *pEpSet, SRpcMsg *pMsg);
int32_t syncIOEqMsg(const SMsgCb *msgcb, SRpcMsg *pMsg);
M
Minghao Li 已提交
78

M
Minghao Li 已提交
79 80 81 82 83
int32_t syncIOQTimerStart();
int32_t syncIOQTimerStop();
int32_t syncIOPingTimerStart();
int32_t syncIOPingTimerStop();

M
Minghao Li 已提交
84 85 86 87 88
#ifdef __cplusplus
}
#endif

#endif /*_TD_LIBS_IO_H*/