/*
 * 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_MND_INT_H_
#define _TD_MND_INT_H_

#include "mndDef.h"
#include "sdb.h"
#include "tstep.h"

#ifdef __cplusplus
extern "C" {
#endif

typedef int32_t (*MndMsgFp)(SMnode *pMnode, SMnodeMsg *pMsg);

typedef struct SMnodeBak {
  int32_t   dnodeId;
  int64_t   clusterId;
  tmr_h     timer;
  SSteps   *pInitSteps;
  SSteps   *pStartSteps;
  SMnodeOpt para;
  MndMsgFp  msgFp[TSDB_MSG_TYPE_MAX];
} SMnodeBak;

typedef struct SMnode {
  int32_t           dnodeId;
  int64_t           clusterId;
  int8_t            replica;
  int8_t            selfIndex;
  SReplica          replicas[TSDB_MAX_REPLICA];
  tmr_h             timer;
  SSteps           *pInitSteps;
  SSteps           *pStartSteps;
  struct SSdb      *pSdb;
  struct SDnode    *pServer;
  MndMsgFp          msgFp[TSDB_MSG_TYPE_MAX];
  PutMsgToMnodeQFp  putMsgToApplyMsgFp;
  SendMsgToDnodeFp  sendMsgToDnodeFp;
  SendMsgToMnodeFp  sendMsgToMnodeFp;
  SendRedirectMsgFp sendRedirectMsgFp;
} SMnode;

tmr_h   mnodeGetTimer();
int32_t mnodeGetDnodeId();
int64_t mnodeGetClusterId();

void mnodeSendMsgToDnode(SMnode *pMnode, struct SEpSet *epSet, struct SRpcMsg *rpcMsg);
void mnodeSendMsgToMnode(SMnode *pMnode, struct SRpcMsg *rpcMsg);
void mnodeSendRedirectMsg(SMnode *pMnode, struct SRpcMsg *rpcMsg, bool forShell);
void mnodeSetMsgHandle(SMnode *pMnode, int32_t msgType, MndMsgFp fp);

#ifdef __cplusplus
}
#endif

#endif /*_TD_MND_INT_H_*/