/* * Copyright (c) 2019 TAOS Data, Inc. * * 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 . */ #define _DEFAULT_SOURCE #include "mndCluster.h" #include "mndTrans.h" #include "mndShow.h" #define SDB_CLUSTER_VER 1 static SSdbRaw *mndClusterActionEncode(SClusterObj *pCluster) { SSdbRaw *pRaw = sdbAllocRaw(SDB_CLUSTER, SDB_CLUSTER_VER, sizeof(SClusterObj)); if (pRaw == NULL) return NULL; int32_t dataPos = 0; SDB_SET_INT32(pRaw, dataPos, pCluster->id); SDB_SET_INT64(pRaw, dataPos, pCluster->createdTime) SDB_SET_INT64(pRaw, dataPos, pCluster->updateTime) SDB_SET_BINARY(pRaw, dataPos, pCluster->uid, TSDB_CLUSTER_ID_LEN) return pRaw; } static SSdbRow *mndClusterActionDecode(SSdbRaw *pRaw) { int8_t sver = 0; if (sdbGetRawSoftVer(pRaw, &sver) != 0) return NULL; if (sver != SDB_CLUSTER_VER) { terrno = TSDB_CODE_SDB_INVALID_DATA_VER; mError("failed to decode cluster since %s", terrstr()); return NULL; } SSdbRow *pRow = sdbAllocRow(sizeof(SClusterObj)); SClusterObj *pCluster = sdbGetRowObj(pRow); if (pCluster == NULL) return NULL; int32_t dataPos = 0; SDB_GET_INT32(pRaw, pRow, dataPos, &pCluster->id) SDB_GET_INT64(pRaw, pRow, dataPos, &pCluster->createdTime) SDB_GET_INT64(pRaw, pRow, dataPos, &pCluster->updateTime) SDB_GET_BINARY(pRaw, pRow, dataPos, pCluster->uid, TSDB_CLUSTER_ID_LEN) return pRow; } static int32_t mndClusterActionInsert(SSdb *pSdb, SClusterObj *pCluster) { mTrace("cluster:%d, perform insert action", pCluster->id); return 0; } static int32_t mndClusterActionDelete(SSdb *pSdb, SClusterObj *pCluster) { mTrace("cluster:%d, perform delete action", pCluster->id); return 0; } static int32_t mndClusterActionUpdate(SSdb *pSdb, SClusterObj *pSrcCluster, SClusterObj *pDstCluster) { mTrace("cluster:%d, perform update action", pSrcCluster->id); return 0; } static int32_t mndCreateDefaultCluster(SMnode *pMnode) { SClusterObj clusterObj = {0}; clusterObj.createdTime = taosGetTimestampMs(); clusterObj.updateTime = clusterObj.createdTime; int32_t code = taosGetSystemUid(clusterObj.uid, TSDB_CLUSTER_ID_LEN); if (code != 0) { strcpy(clusterObj.uid, "tdengine2.0"); mError("failed to get uid from system, set to default val %s", clusterObj.uid); } else { mDebug("cluster:%d, uid is %s", clusterObj.id, clusterObj.uid); } clusterObj.id = MurmurHash3_32(clusterObj.uid, TSDB_CLUSTER_ID_LEN); clusterObj.id = abs(clusterObj.id); pMnode->clusterId = clusterObj.id; SSdbRaw *pRaw = mndClusterActionEncode(&clusterObj); if (pRaw == NULL) return -1; sdbSetRawStatus(pRaw, SDB_STATUS_READY); mTrace("cluster:%d, will be created while deploy sdb", clusterObj.id); return sdbWrite(pMnode->pSdb, pRaw); } // static int32_t mnodeGetClusterMeta(STableMetaMsg *pMeta, SShowObj *pShow, void *pConn) { // int32_t cols = 0; // SSchema *pSchema = pMeta->schema; // pShow->bytes[cols] = TSDB_CLUSTER_ID_LEN + VARSTR_HEADER_SIZE; // pSchema[cols].type = TSDB_DATA_TYPE_BINARY; // strcpy(pSchema[cols].name, "clusterId"); // pSchema[cols].bytes = htons(pShow->bytes[cols]); // cols++; // pShow->bytes[cols] = 8; // pSchema[cols].type = TSDB_DATA_TYPE_TIMESTAMP; // strcpy(pSchema[cols].name, "create_time"); // pSchema[cols].bytes = htons(pShow->bytes[cols]); // cols++; // pMeta->numOfColumns = htons(cols); // strcpy(pMeta->tableFname, "show cluster"); // pShow->numOfColumns = cols; // pShow->offset[0] = 0; // for (int32_t i = 1; i < cols; ++i) { // pShow->offset[i] = pShow->offset[i - 1] + pShow->bytes[i - 1]; // } // pShow->numOfRows = 1; // pShow->rowSize = pShow->offset[cols - 1] + pShow->bytes[cols - 1]; // return 0; // } // static int32_t mnodeRetrieveClusters(SShowObj *pShow, char *data, int32_t rows, void *pConn) { // int32_t numOfRows = 0; // int32_t cols = 0; // char * pWrite; // SClusterObj *pCluster = NULL; // while (numOfRows < rows) { // pShow->pIter = mnodeGetNextCluster(pShow->pIter, &pCluster); // if (pCluster == NULL) break; // cols = 0; // pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows; // STR_WITH_MAXSIZE_TO_VARSTR(pWrite, pCluster->uid, TSDB_CLUSTER_ID_LEN); // cols++; // pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows; // *(int64_t *) pWrite = pCluster->createdTime; // cols++; // mnodeDecClusterRef(pCluster); // numOfRows++; // } // mnodeVacuumResult(data, pShow->numOfColumns, numOfRows, rows, pShow); // pShow->numOfReads += numOfRows; // return numOfRows; // } // static void mnodeCancelGetNextCluster(void *pIter) { // sdbFreeIter(tsClusterSdb, pIter); // } int32_t mndInitCluster(SMnode *pMnode) { SSdbTable table = {.sdbType = SDB_CLUSTER, .keyType = SDB_KEY_INT32, .deployFp = (SdbDeployFp)mndCreateDefaultCluster, .encodeFp = (SdbEncodeFp)mndClusterActionEncode, .decodeFp = (SdbDecodeFp)mndClusterActionDecode, .insertFp = (SdbInsertFp)mndClusterActionInsert, .updateFp = (SdbUpdateFp)mndClusterActionUpdate, .deleteFp = (SdbDeleteFp)mndClusterActionDelete}; // mndAddShowMetaHandle(TSDB_MGMT_TABLE_CLUSTER, mnodeGetClusterMeta); // mndAddShowRetrieveHandle(TSDB_MGMT_TABLE_CLUSTER, mnodeRetrieveClusters); // mndAddShowFreeIterHandle(TSDB_MGMT_TABLE_CLUSTER, mnodeCancelGetNextCluster); return sdbSetTable(pMnode->pSdb, table); } void mndCleanupCluster(SMnode *pMnode) {}