296 lines
7.4 KiB
C
296 lines
7.4 KiB
C
#include <common.h>
|
|
#include "os.h"
|
|
#include "tutil.h"
|
|
|
|
#include "tname.h"
|
|
#include "tmsg.h"
|
|
|
|
#define VALID_NAME_TYPE(x) ((x) == TSDB_DB_NAME_T || (x) == TSDB_TABLE_NAME_T)
|
|
|
|
bool tscValidateTableNameLength(size_t len) {
|
|
return len < TSDB_TABLE_NAME_LEN;
|
|
}
|
|
|
|
// TODO refactor
|
|
SColumnFilterInfo* tFilterInfoDup(const SColumnFilterInfo* src, int32_t numOfFilters) {
|
|
if (numOfFilters == 0 || src == NULL) {
|
|
assert(src == NULL);
|
|
return NULL;
|
|
}
|
|
|
|
SColumnFilterInfo* pFilter = calloc(1, numOfFilters * sizeof(SColumnFilterInfo));
|
|
|
|
memcpy(pFilter, src, sizeof(SColumnFilterInfo) * numOfFilters);
|
|
for (int32_t j = 0; j < numOfFilters; ++j) {
|
|
if (pFilter[j].filterstr) {
|
|
size_t len = (size_t) pFilter[j].len + 1 * TSDB_NCHAR_SIZE;
|
|
pFilter[j].pz = (int64_t) calloc(1, len);
|
|
|
|
memcpy((char*)pFilter[j].pz, (char*)src[j].pz, (size_t) pFilter[j].len);
|
|
}
|
|
}
|
|
|
|
assert(src->filterstr == 0 || src->filterstr == 1);
|
|
assert(!(src->lowerRelOptr == TSDB_RELATION_INVALID && src->upperRelOptr == TSDB_RELATION_INVALID));
|
|
|
|
return pFilter;
|
|
}
|
|
|
|
#if 0
|
|
int64_t taosGetIntervalStartTimestamp(int64_t startTime, int64_t slidingTime, int64_t intervalTime, char timeUnit, int16_t precision) {
|
|
if (slidingTime == 0) {
|
|
return startTime;
|
|
}
|
|
int64_t start = startTime;
|
|
if (timeUnit == 'n' || timeUnit == 'y') {
|
|
start /= 1000;
|
|
if (precision == TSDB_TIME_PRECISION_MICRO) {
|
|
start /= 1000;
|
|
}
|
|
struct tm tm;
|
|
time_t t = (time_t)start;
|
|
localtime_r(&t, &tm);
|
|
tm.tm_sec = 0;
|
|
tm.tm_min = 0;
|
|
tm.tm_hour = 0;
|
|
tm.tm_mday = 1;
|
|
|
|
if (timeUnit == 'y') {
|
|
tm.tm_mon = 0;
|
|
tm.tm_year = (int)(tm.tm_year / slidingTime * slidingTime);
|
|
} else {
|
|
int mon = tm.tm_year * 12 + tm.tm_mon;
|
|
mon = (int)(mon / slidingTime * slidingTime);
|
|
tm.tm_year = mon / 12;
|
|
tm.tm_mon = mon % 12;
|
|
}
|
|
|
|
start = mktime(&tm) * 1000L;
|
|
if (precision == TSDB_TIME_PRECISION_MICRO) {
|
|
start *= 1000L;
|
|
}
|
|
} else {
|
|
int64_t delta = startTime - intervalTime;
|
|
int32_t factor = delta > 0? 1:-1;
|
|
|
|
start = (delta / slidingTime + factor) * slidingTime;
|
|
|
|
if (timeUnit == 'd' || timeUnit == 'w') {
|
|
/*
|
|
* here we revised the start time of day according to the local time zone,
|
|
* but in case of DST, the start time of one day need to be dynamically decided.
|
|
*/
|
|
// todo refactor to extract function that is available for Linux/Windows/Mac platform
|
|
#if defined(WINDOWS) && _MSC_VER >= 1900
|
|
// see https://docs.microsoft.com/en-us/cpp/c-runtime-library/daylight-dstbias-timezone-and-tzname?view=vs-2019
|
|
int64_t timezone = _timezone;
|
|
int32_t daylight = _daylight;
|
|
char** tzname = _tzname;
|
|
#endif
|
|
|
|
int64_t t = (precision == TSDB_TIME_PRECISION_MILLI) ? MILLISECOND_PER_SECOND : MILLISECOND_PER_SECOND * 1000L;
|
|
start += timezone * t;
|
|
}
|
|
|
|
int64_t end = start + intervalTime - 1;
|
|
if (end < startTime) {
|
|
start += slidingTime;
|
|
}
|
|
}
|
|
|
|
return start;
|
|
}
|
|
|
|
#endif
|
|
|
|
int32_t tNameExtractFullName(const SName* name, char* dst) {
|
|
assert(name != NULL && dst != NULL);
|
|
|
|
// invalid full name format, abort
|
|
if (!tNameIsValid(name)) {
|
|
return -1;
|
|
}
|
|
|
|
int32_t len = snprintf(dst, TSDB_DB_FNAME_LEN, "%d.%s", name->acctId, name->dbname);
|
|
|
|
size_t tnameLen = strlen(name->tname);
|
|
if (tnameLen > 0) {
|
|
assert(name->type == TSDB_TABLE_NAME_T);
|
|
dst[len] = TS_PATH_DELIMITER[0];
|
|
|
|
memcpy(dst + len + 1, name->tname, tnameLen);
|
|
dst[len + tnameLen + 1] = 0;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int32_t tNameLen(const SName* name) {
|
|
assert(name != NULL);
|
|
|
|
char tmp[12] = {0};
|
|
int32_t len = sprintf(tmp, "%d", name->acctId);
|
|
int32_t len1 = (int32_t) strlen(name->dbname);
|
|
int32_t len2 = (int32_t) strlen(name->tname);
|
|
|
|
if (name->type == TSDB_DB_NAME_T) {
|
|
assert(len2 == 0);
|
|
return len + len1 + TSDB_NAME_DELIMITER_LEN;
|
|
} else {
|
|
assert(len2 > 0);
|
|
return len + len1 + len2 + TSDB_NAME_DELIMITER_LEN * 2;
|
|
}
|
|
}
|
|
|
|
bool tNameIsValid(const SName* name) {
|
|
assert(name != NULL);
|
|
|
|
if (!VALID_NAME_TYPE(name->type)) {
|
|
return false;
|
|
}
|
|
|
|
if (name->type == TSDB_DB_NAME_T) {
|
|
return strlen(name->dbname) > 0;
|
|
} else {
|
|
return strlen(name->dbname) > 0 && strlen(name->tname) > 0;
|
|
}
|
|
}
|
|
|
|
SName* tNameDup(const SName* name) {
|
|
assert(name != NULL);
|
|
|
|
SName* p = malloc(sizeof(SName));
|
|
memcpy(p, name, sizeof(SName));
|
|
return p;
|
|
}
|
|
|
|
int32_t tNameGetDbName(const SName* name, char* dst) {
|
|
assert(name != NULL && dst != NULL);
|
|
strncpy(dst, name->dbname, tListLen(name->dbname));
|
|
return 0;
|
|
}
|
|
|
|
int32_t tNameGetFullDbName(const SName* name, char* dst) {
|
|
assert(name != NULL && dst != NULL);
|
|
snprintf(dst, TSDB_DB_FNAME_LEN, "%d.%s", name->acctId, name->dbname);
|
|
return 0;
|
|
}
|
|
|
|
bool tNameIsEmpty(const SName* name) {
|
|
assert(name != NULL);
|
|
return name->type == 0 || name->acctId == 0;
|
|
}
|
|
|
|
const char* tNameGetTableName(const SName* name) {
|
|
assert(name != NULL && name->type == TSDB_TABLE_NAME_T);
|
|
return &name->tname[0];
|
|
}
|
|
|
|
void tNameAssign(SName* dst, const SName* src) {
|
|
memcpy(dst, src, sizeof(SName));
|
|
}
|
|
|
|
int32_t tNameSetDbName(SName* dst, int32_t acct, const char* dbName, size_t nameLen) {
|
|
assert(dst != NULL && dbName != NULL && nameLen > 0);
|
|
|
|
// too long account id or too long db name
|
|
if (nameLen >= tListLen(dst->dbname)) {
|
|
return -1;
|
|
}
|
|
|
|
dst->type = TSDB_DB_NAME_T;
|
|
dst->acctId = acct;
|
|
tstrncpy(dst->dbname, dbName, nameLen + 1);
|
|
return 0;
|
|
}
|
|
|
|
int32_t tNameSetAcctId(SName* dst, int32_t acctId) {
|
|
assert(dst != NULL && acct != NULL);
|
|
dst->acctId = acctId;
|
|
return 0;
|
|
}
|
|
|
|
int32_t tNameFromString(SName* dst, const char* str, uint32_t type) {
|
|
assert(dst != NULL && str != NULL && strlen(str) > 0);
|
|
|
|
char* p = NULL;
|
|
if ((type & T_NAME_ACCT) == T_NAME_ACCT) {
|
|
p = strstr(str, TS_PATH_DELIMITER);
|
|
if (p == NULL) {
|
|
return -1;
|
|
}
|
|
|
|
dst->acctId = strtoll(str, NULL, 10);
|
|
}
|
|
|
|
if ((type & T_NAME_DB) == T_NAME_DB) {
|
|
dst->type = TSDB_DB_NAME_T;
|
|
char* start = (char*)((p == NULL)? str:(p+1));
|
|
|
|
int32_t len = 0;
|
|
p = strstr(start, TS_PATH_DELIMITER);
|
|
if (p == NULL) {
|
|
len = (int32_t) strlen(start);
|
|
} else {
|
|
len = (int32_t) (p - start);
|
|
}
|
|
|
|
// too long account id or too long db name
|
|
if ((len >= tListLen(dst->dbname)) || (len <= 0)) {
|
|
return -1;
|
|
}
|
|
|
|
memcpy (dst->dbname, start, len);
|
|
dst->dbname[len] = 0;
|
|
}
|
|
|
|
if ((type & T_NAME_TABLE) == T_NAME_TABLE) {
|
|
dst->type = TSDB_TABLE_NAME_T;
|
|
char* start = (char*) ((p == NULL)? str: (p+1));
|
|
|
|
// too long account id or too long db name
|
|
int32_t len = (int32_t) strlen(start);
|
|
if ((len >= tListLen(dst->tname)) || (len <= 0)) {
|
|
return -1;
|
|
}
|
|
|
|
memcpy (dst->tname, start, len);
|
|
dst->tname[len] = 0;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
SSchema createSchema(uint8_t type, int32_t bytes, int32_t colId, const char* name) {
|
|
SSchema s = {0};
|
|
s.type = type;
|
|
s.bytes = bytes;
|
|
s.colId = colId;
|
|
|
|
tstrncpy(s.name, name, tListLen(s.name));
|
|
return s;
|
|
}
|
|
|
|
void* destroySDataBlock(SSDataBlock* pBlock) {
|
|
if (pBlock == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
int32_t numOfOutput = pBlock->info.numOfCols;
|
|
for(int32_t i = 0; i < numOfOutput; ++i) {
|
|
SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);
|
|
if (IS_VAR_DATA_TYPE(pColInfoData->info.type)) {
|
|
tfree(pColInfoData->varmeta.offset);
|
|
} else {
|
|
tfree(pColInfoData->nullbitmap);
|
|
}
|
|
|
|
tfree(pColInfoData->pData);
|
|
}
|
|
|
|
taosArrayDestroy(pBlock->pDataBlock);
|
|
tfree(pBlock->pBlockAgg);
|
|
tfree(pBlock);
|
|
return NULL;
|
|
}
|