101 lines
2.9 KiB
C
101 lines
2.9 KiB
C
/*
|
|
* 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 "smInt.h"
|
|
#include "libs/function/function.h"
|
|
#include "libs/function/tudf.h"
|
|
|
|
static int32_t smRequire(const SMgmtInputOpt *pInput, bool *required) {
|
|
return dmReadFile(pInput->path, pInput->name, required);
|
|
}
|
|
|
|
static void smInitOption(SSnodeMgmt *pMgmt, SSnodeOpt *pOption) { pOption->msgCb = pMgmt->msgCb; }
|
|
|
|
static void smClose(SSnodeMgmt *pMgmt) {
|
|
if (pMgmt->pSnode != NULL) {
|
|
smStopWorker(pMgmt);
|
|
sndClose(pMgmt->pSnode);
|
|
pMgmt->pSnode = NULL;
|
|
}
|
|
|
|
taosMemoryFree(pMgmt);
|
|
}
|
|
int32_t sndOpenWrapper(const char *path, SSnodeOpt *pOption, SSnode **pNode) {
|
|
*pNode = sndOpen(path, pOption);
|
|
if (*pNode == NULL) {
|
|
return terrno;
|
|
}
|
|
return 0;
|
|
}
|
|
int32_t smOpen(SMgmtInputOpt *pInput, SMgmtOutputOpt *pOutput) {
|
|
int32_t code = 0;
|
|
SSnodeMgmt *pMgmt = taosMemoryCalloc(1, sizeof(SSnodeMgmt));
|
|
if (pMgmt == NULL) {
|
|
code = terrno;
|
|
return code;
|
|
}
|
|
|
|
pMgmt->pData = pInput->pData;
|
|
pMgmt->path = pInput->path;
|
|
pMgmt->name = pInput->name;
|
|
pMgmt->msgCb = pInput->msgCb;
|
|
pMgmt->msgCb.mgmt = pMgmt;
|
|
pMgmt->msgCb.putToQueueFp = (PutToQueueFp)smPutMsgToQueue;
|
|
|
|
SSnodeOpt option = {0};
|
|
smInitOption(pMgmt, &option);
|
|
|
|
code = sndOpenWrapper(pMgmt->path, &option, &pMgmt->pSnode);
|
|
if (code != 0) {
|
|
dError("failed to open snode since %s", tstrerror(code));
|
|
smClose(pMgmt);
|
|
return code;
|
|
}
|
|
|
|
tmsgReportStartup("snode-impl", "initialized");
|
|
|
|
if ((code = smStartWorker(pMgmt)) != 0) {
|
|
dError("failed to start snode worker since %s", tstrerror(code));
|
|
smClose(pMgmt);
|
|
return code;
|
|
}
|
|
tmsgReportStartup("snode-worker", "initialized");
|
|
|
|
if ((code = udfcOpen()) != 0) {
|
|
dError("failed to open udfc in snode since:%s", tstrerror(code));
|
|
smClose(pMgmt);
|
|
return code;
|
|
}
|
|
|
|
pOutput->pMgmt = pMgmt;
|
|
return 0;
|
|
}
|
|
|
|
static int32_t smStartSnodes(SSnodeMgmt *pMgmt) { return sndInit(pMgmt->pSnode); }
|
|
|
|
SMgmtFunc smGetMgmtFunc() {
|
|
SMgmtFunc mgmtFunc = {0};
|
|
mgmtFunc.openFp = smOpen;
|
|
mgmtFunc.startFp = (NodeStartFp)smStartSnodes;
|
|
mgmtFunc.closeFp = (NodeCloseFp)smClose;
|
|
mgmtFunc.createFp = (NodeCreateFp)smProcessCreateReq;
|
|
mgmtFunc.dropFp = (NodeDropFp)smProcessDropReq;
|
|
mgmtFunc.requiredFp = smRequire;
|
|
mgmtFunc.getHandlesFp = smGetMsgHandles;
|
|
|
|
return mgmtFunc;
|
|
}
|