[TD-10430] add dnode cfg for dnode module
This commit is contained in:
Shengliang Guan
parent
7a78d7223d
commit
9c29039ae7
|
|
@ -0,0 +1,45 @@
|
|||
/*
|
||||
* 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_DNODE_CFG_H_
|
||||
#define _TD_DNODE_CFG_H_
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
#include "dnodeInt.h"
|
||||
|
||||
typedef struct DnCfg {
|
||||
Dnode * dnode;
|
||||
int32_t dnodeId;
|
||||
int32_t dropped;
|
||||
char clusterId[TSDB_CLUSTER_ID_LEN];
|
||||
char file[PATH_MAX + 20];
|
||||
pthread_mutex_t mutex;
|
||||
} DnCfg;
|
||||
|
||||
int32_t dnodeInitCfg(Dnode *dnode, DnCfg **cfg);
|
||||
void dnodeCleanupCfg(Dnode *dnode, DnCfg **cfg);
|
||||
void dnodeUpdateCfg(DnCfg *cfg, SDnodeCfg *data);
|
||||
int32_t dnodeGetDnodeId(DnCfg *cfg);
|
||||
void dnodeGetClusterId(DnCfg *cfg, char *clusterId);
|
||||
void dnodeGetCfg(DnCfg *cfg, int32_t *dnodeId, char *clusterId);
|
||||
void dnodeSetDropped(DnCfg *cfg);
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif /*_TD_DNODE_CFG_H_*/
|
||||
|
|
@ -0,0 +1,178 @@
|
|||
/*
|
||||
* 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 "cJSON.h"
|
||||
#include "dnodeCfg.h"
|
||||
|
||||
static int32_t dnodeReadCfg(DnCfg *cfg) {
|
||||
int32_t len = 0;
|
||||
int32_t maxLen = 200;
|
||||
char * content = calloc(1, maxLen + 1);
|
||||
cJSON * root = NULL;
|
||||
FILE * fp = NULL;
|
||||
|
||||
fp = fopen(cfg->file, "r");
|
||||
if (!fp) {
|
||||
dDebug("file %s not exist", cfg->file);
|
||||
goto PARSE_CFG_OVER;
|
||||
}
|
||||
|
||||
len = (int32_t)fread(content, 1, maxLen, fp);
|
||||
if (len <= 0) {
|
||||
dError("failed to read %s since content is null", cfg->file);
|
||||
goto PARSE_CFG_OVER;
|
||||
}
|
||||
|
||||
content[len] = 0;
|
||||
root = cJSON_Parse(content);
|
||||
if (root == NULL) {
|
||||
dError("failed to read %s since invalid json format", cfg->file);
|
||||
goto PARSE_CFG_OVER;
|
||||
}
|
||||
|
||||
cJSON *dnodeId = cJSON_GetObjectItem(root, "dnodeId");
|
||||
if (!dnodeId || dnodeId->type != cJSON_Number) {
|
||||
dError("failed to read %s since dnodeId not found", cfg->file);
|
||||
goto PARSE_CFG_OVER;
|
||||
}
|
||||
cfg->dnodeId = (int32_t)dnodeId->valueint;
|
||||
|
||||
cJSON *dropped = cJSON_GetObjectItem(root, "dropped");
|
||||
if (!dropped || dropped->type != cJSON_Number) {
|
||||
dError("failed to read %s since dropped not found", cfg->file);
|
||||
goto PARSE_CFG_OVER;
|
||||
}
|
||||
cfg->dropped = (int32_t)dropped->valueint;
|
||||
|
||||
cJSON *clusterId = cJSON_GetObjectItem(root, "clusterId");
|
||||
if (!clusterId || clusterId->type != cJSON_String) {
|
||||
dError("failed to read %s since clusterId not found", cfg->file);
|
||||
goto PARSE_CFG_OVER;
|
||||
}
|
||||
tstrncpy(cfg->clusterId, clusterId->valuestring, TSDB_CLUSTER_ID_LEN);
|
||||
|
||||
dInfo("successed to read %s", cfg->file);
|
||||
|
||||
PARSE_CFG_OVER:
|
||||
if (content != NULL) free(content);
|
||||
if (root != NULL) cJSON_Delete(root);
|
||||
if (fp != NULL) fclose(fp);
|
||||
terrno = 0;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int32_t dnodeWriteCfg(DnCfg *cfg) {
|
||||
FILE *fp = fopen(cfg->file, "w");
|
||||
if (!fp) {
|
||||
dError("failed to write %s since %s", cfg->file, strerror(errno));
|
||||
return -1;
|
||||
}
|
||||
|
||||
int32_t len = 0;
|
||||
int32_t maxLen = 200;
|
||||
char * content = calloc(1, maxLen + 1);
|
||||
|
||||
len += snprintf(content + len, maxLen - len, "{\n");
|
||||
len += snprintf(content + len, maxLen - len, " \"dnodeId\": %d,\n", cfg->dnodeId);
|
||||
len += snprintf(content + len, maxLen - len, " \"dropped\": %d,\n", cfg->dropped);
|
||||
len += snprintf(content + len, maxLen - len, " \"clusterId\": \"%s\"\n", cfg->clusterId);
|
||||
len += snprintf(content + len, maxLen - len, "}\n");
|
||||
|
||||
fwrite(content, 1, len, fp);
|
||||
taosFsync(fileno(fp));
|
||||
fclose(fp);
|
||||
free(content);
|
||||
terrno = 0;
|
||||
|
||||
dInfo("successed to write %s", cfg->file);
|
||||
return 0;
|
||||
}
|
||||
|
||||
int32_t dnodeInitCfg(Dnode *dnode, DnCfg **out) {
|
||||
DnCfg* cfg = calloc(1, sizeof(DnCfg));
|
||||
if (cfg == NULL) return -1;
|
||||
|
||||
cfg->dnode = dnode;
|
||||
cfg->dnodeId = 0;
|
||||
cfg->dropped = 0;
|
||||
cfg->clusterId[0] = 0;
|
||||
snprintf(cfg->file, sizeof(cfg->file), "%s/dnodeCfg.json", tsDnodeDir);
|
||||
pthread_mutex_init(&cfg->mutex, NULL);
|
||||
*out = cfg;
|
||||
|
||||
int32_t ret = dnodeReadCfg(cfg);
|
||||
if (ret == 0) {
|
||||
dInfo("dnode cfg is initialized");
|
||||
}
|
||||
|
||||
if (cfg->dropped) {
|
||||
dInfo("dnode is dropped and start to exit");
|
||||
return -1;
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void dnodeCleanupCfg(Dnode *dnode, DnCfg **out) {
|
||||
DnCfg* cfg = *out;
|
||||
*out = NULL;
|
||||
|
||||
pthread_mutex_destroy(&cfg->mutex);
|
||||
free(cfg);
|
||||
}
|
||||
|
||||
void dnodeUpdateCfg(DnCfg *cfg, SDnodeCfg *data) {
|
||||
if (cfg == NULL || cfg->dnodeId == 0) return;
|
||||
|
||||
pthread_mutex_lock(&cfg->mutex);
|
||||
|
||||
cfg->dnodeId = data->dnodeId;
|
||||
tstrncpy(cfg->clusterId, data->clusterId, TSDB_CLUSTER_ID_LEN);
|
||||
dInfo("dnodeId is set to %d, clusterId is set to %s", cfg->dnodeId, cfg->clusterId);
|
||||
|
||||
dnodeWriteCfg(cfg);
|
||||
pthread_mutex_unlock(&cfg->mutex);
|
||||
}
|
||||
|
||||
void dnodeSetDropped(DnCfg *cfg) {
|
||||
pthread_mutex_lock(&cfg->mutex);
|
||||
cfg->dropped = 1;
|
||||
dnodeWriteCfg(cfg);
|
||||
pthread_mutex_unlock(&cfg->mutex);
|
||||
}
|
||||
|
||||
int32_t dnodeGetDnodeId(DnCfg *cfg) {
|
||||
int32_t dnodeId = 0;
|
||||
pthread_mutex_lock(&cfg->mutex);
|
||||
dnodeId = cfg->dnodeId;
|
||||
pthread_mutex_unlock(&cfg->mutex);
|
||||
return dnodeId;
|
||||
}
|
||||
|
||||
void dnodeGetClusterId(DnCfg *cfg, char *clusterId) {
|
||||
pthread_mutex_lock(&cfg->mutex);
|
||||
tstrncpy(clusterId, cfg->clusterId, TSDB_CLUSTER_ID_LEN);
|
||||
pthread_mutex_unlock(&cfg->mutex);
|
||||
}
|
||||
|
||||
void dnodeGetCfg(DnCfg *cfg, int32_t *dnodeId, char *clusterId) {
|
||||
pthread_mutex_lock(&cfg->mutex);
|
||||
*dnodeId = cfg->dnodeId;
|
||||
tstrncpy(clusterId, cfg->clusterId, TSDB_CLUSTER_ID_LEN);
|
||||
pthread_mutex_unlock(&cfg->mutex);
|
||||
}
|
||||
Loading…
Reference in New Issue