vmInt.h 3.4 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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/>.
 */

S
shm  
Shengliang Guan 已提交
16 17
#ifndef _TD_DND_VNODES_INT_H_
#define _TD_DND_VNODES_INT_H_
S
shm  
Shengliang Guan 已提交
18

S
Shengliang 已提交
19
#include "dmUtil.h"
S
Shengliang Guan 已提交
20

S
shm  
Shengliang Guan 已提交
21
#include "sync.h"
S
shm  
Shengliang Guan 已提交
22
#include "vnode.h"
23 24 25 26 27

#ifdef __cplusplus
extern "C" {
#endif

S
Shengliang 已提交
28
typedef struct SVnodeMgmt {
29 30 31 32 33 34 35 36
  SDnodeData    *pData;
  SMsgCb         msgCb;
  const char    *path;
  const char    *name;
  SQWorkerPool   queryPool;
  SQWorkerPool   fetchPool;
  SWWorkerPool   syncPool;
  SWWorkerPool   writePool;
S
Shengliang Guan 已提交
37
  SWWorkerPool   applyPool;
38 39 40 41 42 43
  SSingleWorker  mgmtWorker;
  SSingleWorker  monitorWorker;
  SHashObj      *hash;
  TdThreadRwlock lock;
  SVnodesStat    state;
  STfs          *pTfs;
S
Shengliang 已提交
44
} SVnodeMgmt;
S
shm  
Shengliang Guan 已提交
45

S
shm  
Shengliang Guan 已提交
46
typedef struct {
S
Shengliang Guan 已提交
47 48 49 50
  int32_t vgId;
  int32_t vgVersion;
  int8_t  dropped;
  char    path[PATH_MAX + 20];
S
shm  
Shengliang Guan 已提交
51 52 53
} SWrapperCfg;

typedef struct {
S
Shengliang Guan 已提交
54 55
  int32_t     vgId;
  int32_t     vgVersion;
56
  int32_t     refCount;
S
Shengliang Guan 已提交
57 58 59 60 61 62 63 64
  int8_t      dropped;
  char       *path;
  SVnode     *pImpl;
  STaosQueue *pWriteQ;
  STaosQueue *pSyncQ;
  STaosQueue *pApplyQ;
  STaosQueue *pQueryQ;
  STaosQueue *pFetchQ;
S
shm  
Shengliang Guan 已提交
65 66 67 68 69 70 71
} SVnodeObj;

typedef struct {
  int32_t      vnodeNum;
  int32_t      opened;
  int32_t      failed;
  int32_t      threadIndex;
S
Shengliang Guan 已提交
72
  TdThread     thread;
S
Shengliang 已提交
73
  SVnodeMgmt  *pMgmt;
S
shm  
Shengliang Guan 已提交
74 75 76
  SWrapperCfg *pCfgs;
} SVnodeThread;

S
shm  
Shengliang Guan 已提交
77
// vmInt.c
S
Shengliang 已提交
78 79 80 81
SVnodeObj *vmAcquireVnode(SVnodeMgmt *pMgmt, int32_t vgId);
void       vmReleaseVnode(SVnodeMgmt *pMgmt, SVnodeObj *pVnode);
int32_t    vmOpenVnode(SVnodeMgmt *pMgmt, SWrapperCfg *pCfg, SVnode *pImpl);
void       vmCloseVnode(SVnodeMgmt *pMgmt, SVnodeObj *pVnode);
S
shm  
Shengliang Guan 已提交
82

S
Shengliang Guan 已提交
83
// vmHandle.c
S
Shengliang 已提交
84
SArray *vmGetMsgHandles();
S
Shengliang Guan 已提交
85 86 87 88
int32_t vmProcessCreateVnodeReq(SVnodeMgmt *pMgmt, SRpcMsg *pMsg);
int32_t vmProcessDropVnodeReq(SVnodeMgmt *pMgmt, SRpcMsg *pMsg);
int32_t vmProcessGetMonitorInfoReq(SVnodeMgmt *pMgmt, SRpcMsg *pMsg);
int32_t vmProcessGetLoadsReq(SVnodeMgmt *pMgmt, SRpcMsg *pMsg);
S
shm  
Shengliang Guan 已提交
89 90

// vmFile.c
S
Shengliang 已提交
91 92 93
int32_t     vmGetVnodeListFromFile(SVnodeMgmt *pMgmt, SWrapperCfg **ppCfgs, int32_t *numOfVnodes);
int32_t     vmWriteVnodeListToFile(SVnodeMgmt *pMgmt);
SVnodeObj **vmGetVnodeListFromHash(SVnodeMgmt *pMgmt, int32_t *numOfVnodes);
S
shm  
Shengliang Guan 已提交
94 95

// vmWorker.c
S
Shengliang 已提交
96 97 98 99 100 101
int32_t vmStartWorker(SVnodeMgmt *pMgmt);
void    vmStopWorker(SVnodeMgmt *pMgmt);
int32_t vmAllocQueue(SVnodeMgmt *pMgmt, SVnodeObj *pVnode);
void    vmFreeQueue(SVnodeMgmt *pMgmt, SVnodeObj *pVnode);

int32_t vmGetQueueSize(SVnodeMgmt *pMgmt, int32_t vgId, EQueueType qtype);
S
Shengliang Guan 已提交
102
int32_t vmPutRpcMsgToQueue(SVnodeMgmt *pMgmt, EQueueType qtype, SRpcMsg *pRpc);
S
Shengliang 已提交
103

S
Shengliang Guan 已提交
104 105 106 107 108 109 110
int32_t vmPutMsgToWriteQueue(SVnodeMgmt *pMgmt, SRpcMsg *pMsg);
int32_t vmPutMsgToSyncQueue(SVnodeMgmt *pMgmt, SRpcMsg *pMsg);
int32_t vmPutMsgToQueryQueue(SVnodeMgmt *pMgmt, SRpcMsg *pMsg);
int32_t vmPutMsgToFetchQueue(SVnodeMgmt *pMgmt, SRpcMsg *pMsg);
int32_t vmPutMsgToMergeQueue(SVnodeMgmt *pMgmt, SRpcMsg *pMsg);
int32_t vmPutMsgToMgmtQueue(SVnodeMgmt *pMgmt, SRpcMsg *pMsg);
int32_t vmPutMsgToMonitorQueue(SVnodeMgmt *pMgmt, SRpcMsg *pMsg);
S
Shengliang Guan 已提交
111

112 113 114 115
#ifdef __cplusplus
}
#endif

L
Liu Jicong 已提交
116
#endif /*_TD_DND_VNODES_INT_H_*/