/*

* 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/>.

*/

#define _DEFAULT_SOURCE

#include "os.h"

#include "taosdef.h"

#include "tsched.h"

#include "tbalance.h"

#include "tgrant.h"

#include "ttimer.h"

#include "tglobal.h"

#include "mnode.h"

#include "dnode.h"

#include "mnodeDef.h"

#include "mnodeInt.h"

#include "mnodeAcct.h"

#include "mnodeDnode.h"

#include "mnodeMnode.h"

#include "mnodeDb.h"

#include "mnodeSdb.h"

#include "mnodeVgroup.h"

#include "mnodeUser.h"

#include "mnodeTable.h"

#include "mnodeShow.h"

static int32_t (*tsMnodeProcessReadMsgFp[TSDB_MSG_TYPE_MAX])(SMnodeMsg *);

void mnodeAddReadMsgHandle(uint8_t msgType, int32_t (*fp)(SMnodeMsg *pMsg)) {

tsMnodeProcessReadMsgFp[msgType] = fp;

}

int32_t mnodeProcessRead(SMnodeMsg *pMsg) {

if (pMsg->rpcMsg.pCont == NULL) {

mError("%p, msg:%s in mread queue, content is null", pMsg->rpcMsg.ahandle, taosMsg[pMsg->rpcMsg.msgType]);

return TSDB_CODE_MND_INVALID_MSG_LEN;

}

if (!sdbIsMaster()) {

SMnodeRsp *rpcRsp = &pMsg->rpcRsp;

SRpcEpSet *epSet = rpcMallocCont(sizeof(SRpcEpSet));

mnodeGetMnodeEpSetForShell(epSet);

rpcRsp->rsp = epSet;

rpcRsp->len = sizeof(SRpcEpSet);

mDebug("%p, msg:%s in mread queue, will be redireced, inUse:%d", pMsg->rpcMsg.ahandle, taosMsg[pMsg->rpcMsg.msgType], epSet->inUse);

for (int32_t i = 0; i < epSet->numOfEps; ++i) {

mDebug("mnode index:%d ep:%s:%d", i, epSet->fqdn[i], htons(epSet->port[i]));

}

return TSDB_CODE_RPC_REDIRECT;

}

if (tsMnodeProcessReadMsgFp[pMsg->rpcMsg.msgType] == NULL) {

mError("%p, msg:%s in mread queue, not processed", pMsg->rpcMsg.ahandle, taosMsg[pMsg->rpcMsg.msgType]);

return TSDB_CODE_MND_MSG_NOT_PROCESSED;

}

int32_t code = mnodeInitMsg(pMsg);

if (code != TSDB_CODE_SUCCESS) {

mError("%p, msg:%s in mread queue, not processed reason:%s", pMsg->rpcMsg.ahandle, taosMsg[pMsg->rpcMsg.msgType], tstrerror(code));

return code;

}

return (*tsMnodeProcessReadMsgFp[pMsg->rpcMsg.msgType])(pMsg);

}