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;

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

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

M
Minghao Li 已提交
54
  void *pSyncNode;
M
sync io  
Minghao Li 已提交
55 56
  int32_t (*FpOnSyncPing)(SSyncNode *pSyncNode, SyncPing *pMsg);
  int32_t (*FpOnSyncPingReply)(SSyncNode *pSyncNode, SyncPingReply *pMsg);
57
  int32_t (*FpOnSyncClientRequest)(SSyncNode *pSyncNode, SRpcMsg *pMsg, SyncIndex *pRetIndex);
M
sync io  
Minghao Li 已提交
58 59 60 61
  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 已提交
62
  int32_t (*FpOnSyncTimeout)(SSyncNode *pSyncNode, SyncTimeout *pMsg);
M
Minghao Li 已提交
63 64
  int32_t (*FpOnSyncSnapshot)(SSyncNode *pSyncNode, SyncSnapshotSend *pMsg);
  int32_t (*FpOnSyncSnapshotReply)(SSyncNode *pSyncNode, SyncSnapshotRsp *pMsg);
M
Minghao Li 已提交
65

M
sync io  
Minghao Li 已提交
66
  int8_t isStart;
M
Minghao Li 已提交
67

M
Minghao Li 已提交
68 69
} SSyncIO;

M
Minghao Li 已提交
70
extern SSyncIO *gSyncIO;
M
Minghao Li 已提交
71

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

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

M
Minghao Li 已提交
82 83 84 85 86
#ifdef __cplusplus
}
#endif

#endif /*_TD_LIBS_IO_H*/