Merge remote-tracking branch 'origin/3.0' into feature/dnode3
This commit is contained in:
commit
d97620101e
|
@ -101,4 +101,4 @@ TAGS
|
||||||
|
|
||||||
contrib/*
|
contrib/*
|
||||||
!contrib/CMakeLists.txt
|
!contrib/CMakeLists.txt
|
||||||
!contrib/test
|
!contrib/test
|
|
@ -10,7 +10,7 @@ set(CMAKE_SUPPORT_DIR "${CMAKE_SOURCE_DIR}/cmake")
|
||||||
set(CMAKE_CONTRIB_DIR "${CMAKE_SOURCE_DIR}/contrib")
|
set(CMAKE_CONTRIB_DIR "${CMAKE_SOURCE_DIR}/contrib")
|
||||||
include(${CMAKE_SUPPORT_DIR}/cmake.options)
|
include(${CMAKE_SUPPORT_DIR}/cmake.options)
|
||||||
|
|
||||||
SET(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -fPIC -gdwarf-2 -msse4.2 -mfma")
|
SET(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -fPIC -gdwarf-2 -msse4.2 -mfma -g3")
|
||||||
|
|
||||||
# contrib
|
# contrib
|
||||||
add_subdirectory(contrib)
|
add_subdirectory(contrib)
|
||||||
|
@ -20,9 +20,13 @@ add_library(api INTERFACE)
|
||||||
target_include_directories(api INTERFACE "include/client")
|
target_include_directories(api INTERFACE "include/client")
|
||||||
|
|
||||||
# src
|
# src
|
||||||
|
if(${BUILD_TEST})
|
||||||
|
include(CTest)
|
||||||
|
enable_testing()
|
||||||
|
endif(${BUILD_TEST})
|
||||||
add_subdirectory(source)
|
add_subdirectory(source)
|
||||||
|
|
||||||
# docs
|
# docs
|
||||||
add_subdirectory(docs)
|
add_subdirectory(docs)
|
||||||
|
|
||||||
# tests (TODO)
|
# tests (TODO)
|
|
@ -73,10 +73,12 @@ typedef struct taosField {
|
||||||
#define DLL_EXPORT
|
#define DLL_EXPORT
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
DLL_EXPORT int taos_init();
|
typedef void (*__taos_async_fn_t)(void *param, TAOS_RES *, int code);
|
||||||
|
|
||||||
DLL_EXPORT void taos_cleanup(void);
|
DLL_EXPORT void taos_cleanup(void);
|
||||||
DLL_EXPORT int taos_options(TSDB_OPTION option, const void *arg, ...);
|
DLL_EXPORT int taos_options(TSDB_OPTION option, const void *arg, ...);
|
||||||
DLL_EXPORT TAOS *taos_connect(const char *ip, const char *user, const char *pass, const char *db, uint16_t port);
|
DLL_EXPORT TAOS *taos_connect(const char *ip, const char *user, const char *pass, const char *db, uint16_t port);
|
||||||
|
DLL_EXPORT TAOS *taos_connect_l(const char *ip, int ipLen, const char *user, int userLen, const char *pass, int passLen, const char *db, int dbLen, uint16_t port);
|
||||||
DLL_EXPORT TAOS *taos_connect_auth(const char *ip, const char *user, const char *auth, const char *db, uint16_t port);
|
DLL_EXPORT TAOS *taos_connect_auth(const char *ip, const char *user, const char *auth, const char *db, uint16_t port);
|
||||||
DLL_EXPORT void taos_close(TAOS *taos);
|
DLL_EXPORT void taos_close(TAOS *taos);
|
||||||
|
|
||||||
|
@ -154,14 +156,14 @@ DLL_EXPORT int* taos_fetch_lengths(TAOS_RES *res);
|
||||||
// TAOS_RES *taos_list_dbs(TAOS *mysql, const char *wild);
|
// TAOS_RES *taos_list_dbs(TAOS *mysql, const char *wild);
|
||||||
|
|
||||||
// TODO: the return value should be `const`
|
// TODO: the return value should be `const`
|
||||||
DLL_EXPORT char *taos_get_server_info(TAOS *taos);
|
DLL_EXPORT const char *taos_get_server_info(TAOS *taos);
|
||||||
DLL_EXPORT char *taos_get_client_info();
|
DLL_EXPORT const char *taos_get_client_info();
|
||||||
DLL_EXPORT char *taos_errstr(TAOS_RES *tres);
|
DLL_EXPORT const char *taos_errstr(TAOS_RES *tres);
|
||||||
|
|
||||||
DLL_EXPORT int taos_errno(TAOS_RES *tres);
|
DLL_EXPORT int taos_errno(TAOS_RES *tres);
|
||||||
|
|
||||||
DLL_EXPORT void taos_query_a(TAOS *taos, const char *sql, void (*fp)(void *param, TAOS_RES *, int code), void *param);
|
DLL_EXPORT void taos_query_a(TAOS *taos, const char *sql, __taos_async_fn_t fp, void *param);
|
||||||
DLL_EXPORT void taos_fetch_rows_a(TAOS_RES *res, void (*fp)(void *param, TAOS_RES *, int numOfRows), void *param);
|
DLL_EXPORT void taos_fetch_rows_a(TAOS_RES *res, __taos_async_fn_t fp, void *param);
|
||||||
|
|
||||||
typedef void (*TAOS_SUBSCRIBE_CALLBACK)(TAOS_SUB* tsub, TAOS_RES *res, void* param, int code);
|
typedef void (*TAOS_SUBSCRIBE_CALLBACK)(TAOS_SUB* tsub, TAOS_RES *res, void* param, int code);
|
||||||
DLL_EXPORT TAOS_SUB *taos_subscribe(TAOS* taos, int restart, const char* topic, const char *sql, TAOS_SUBSCRIBE_CALLBACK fp, void *param, int interval);
|
DLL_EXPORT TAOS_SUB *taos_subscribe(TAOS* taos, int restart, const char* topic, const char *sql, TAOS_SUBSCRIBE_CALLBACK fp, void *param, int interval);
|
||||||
|
|
|
@ -77,7 +77,7 @@ TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_DROP_FUNCTION, "drop-function" )
|
||||||
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_CREATE_STB, "create-stb" )
|
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_CREATE_STB, "create-stb" )
|
||||||
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_ALTER_STB, "alter-stb" )
|
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_ALTER_STB, "alter-stb" )
|
||||||
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_DROP_STB, "drop-stb" )
|
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_DROP_STB, "drop-stb" )
|
||||||
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_STB_VGROUP, "stb-vgroup" )
|
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_VGROUP_LIST, "vgroup-list" )
|
||||||
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_KILL_QUERY, "kill-query" )
|
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_KILL_QUERY, "kill-query" )
|
||||||
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_KILL_STREAM, "kill-stream" )
|
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_KILL_STREAM, "kill-stream" )
|
||||||
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_KILL_CONN, "kill-conn" )
|
TAOS_DEFINE_MESSAGE_TYPE( TSDB_MSG_TYPE_KILL_CONN, "kill-conn" )
|
||||||
|
@ -214,6 +214,11 @@ typedef enum _mgmt_table {
|
||||||
|
|
||||||
extern char *taosMsg[];
|
extern char *taosMsg[];
|
||||||
|
|
||||||
|
typedef struct SBuildTableMetaInput {
|
||||||
|
int32_t vgId;
|
||||||
|
char *tableFullName;
|
||||||
|
} SBuildTableMetaInput;
|
||||||
|
|
||||||
#pragma pack(push, 1)
|
#pragma pack(push, 1)
|
||||||
|
|
||||||
// null-terminated string instead of char array to avoid too many memory consumption in case of more than 1M tableMeta
|
// null-terminated string instead of char array to avoid too many memory consumption in case of more than 1M tableMeta
|
||||||
|
@ -358,6 +363,7 @@ typedef struct {
|
||||||
int32_t pid;
|
int32_t pid;
|
||||||
char app[TSDB_APP_NAME_LEN];
|
char app[TSDB_APP_NAME_LEN];
|
||||||
char db[TSDB_DB_NAME_LEN];
|
char db[TSDB_DB_NAME_LEN];
|
||||||
|
int64_t startTime;
|
||||||
} SConnectMsg;
|
} SConnectMsg;
|
||||||
|
|
||||||
typedef struct SEpSet {
|
typedef struct SEpSet {
|
||||||
|
@ -368,19 +374,17 @@ typedef struct SEpSet {
|
||||||
} SEpSet;
|
} SEpSet;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
int32_t acctId;
|
int32_t acctId;
|
||||||
int32_t clusterId;
|
uint32_t clusterId;
|
||||||
int32_t connId;
|
int32_t connId;
|
||||||
int8_t superAuth;
|
int8_t superUser;
|
||||||
int8_t readAuth;
|
int8_t reserved[5];
|
||||||
int8_t writeAuth;
|
SEpSet epSet;
|
||||||
int8_t reserved[5];
|
|
||||||
SEpSet epSet;
|
|
||||||
} SConnectRsp;
|
} SConnectRsp;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
char user[TSDB_USER_LEN];
|
char user[TSDB_USER_LEN];
|
||||||
char pass[TSDB_KEY_LEN];
|
char pass[TSDB_PASSWORD_LEN];
|
||||||
int32_t maxUsers;
|
int32_t maxUsers;
|
||||||
int32_t maxDbs;
|
int32_t maxDbs;
|
||||||
int32_t maxTimeSeries;
|
int32_t maxTimeSeries;
|
||||||
|
@ -395,7 +399,7 @@ typedef struct {
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
char user[TSDB_USER_LEN];
|
char user[TSDB_USER_LEN];
|
||||||
char pass[TSDB_KEY_LEN];
|
char pass[TSDB_PASSWORD_LEN];
|
||||||
} SCreateUserMsg, SAlterUserMsg;
|
} SCreateUserMsg, SAlterUserMsg;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
|
@ -774,9 +778,8 @@ typedef struct {
|
||||||
} SStbInfoMsg;
|
} SStbInfoMsg;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
|
SMsgHead msgHead;
|
||||||
char tableFname[TSDB_TABLE_FNAME_LEN];
|
char tableFname[TSDB_TABLE_FNAME_LEN];
|
||||||
int8_t createFlag;
|
|
||||||
char tags[];
|
|
||||||
} STableInfoMsg;
|
} STableInfoMsg;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
|
@ -791,6 +794,20 @@ typedef struct SSTableVgroupMsg {
|
||||||
int32_t numOfTables;
|
int32_t numOfTables;
|
||||||
} SSTableVgroupMsg, SSTableVgroupRspMsg;
|
} SSTableVgroupMsg, SSTableVgroupRspMsg;
|
||||||
|
|
||||||
|
typedef struct SVgroupInfo {
|
||||||
|
int32_t vgId;
|
||||||
|
int8_t numOfEps;
|
||||||
|
SEpAddrMsg epAddr[TSDB_MAX_REPLICA];
|
||||||
|
} SVgroupInfo;
|
||||||
|
|
||||||
|
typedef struct SVgroupListRspMsg {
|
||||||
|
int32_t vgroupNum;
|
||||||
|
int32_t vgroupVersion;
|
||||||
|
SVgroupInfo vgroupInfo[];
|
||||||
|
} SVgroupListRspMsg;
|
||||||
|
|
||||||
|
typedef SVgroupListRspMsg SVgroupListInfo;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
int32_t vgId;
|
int32_t vgId;
|
||||||
int8_t numOfEps;
|
int8_t numOfEps;
|
||||||
|
@ -961,8 +978,8 @@ typedef struct {
|
||||||
char user[TSDB_USER_LEN];
|
char user[TSDB_USER_LEN];
|
||||||
char spi;
|
char spi;
|
||||||
char encrypt;
|
char encrypt;
|
||||||
char secret[TSDB_KEY_LEN];
|
char secret[TSDB_PASSWORD_LEN];
|
||||||
char ckey[TSDB_KEY_LEN];
|
char ckey[TSDB_PASSWORD_LEN];
|
||||||
} SAuthMsg, SAuthRsp;
|
} SAuthMsg, SAuthRsp;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
|
|
|
@ -0,0 +1,17 @@
|
||||||
|
#ifndef TDENGINE_TEP_H
|
||||||
|
#define TDENGINE_TEP_H
|
||||||
|
|
||||||
|
#include "os.h"
|
||||||
|
#include "taosmsg.h"
|
||||||
|
|
||||||
|
typedef struct SCorEpSet {
|
||||||
|
int32_t version;
|
||||||
|
SEpSet epSet;
|
||||||
|
} SCorEpSet;
|
||||||
|
|
||||||
|
int taosGetFqdnPortFromEp(const char *ep, char *fqdn, uint16_t *port);
|
||||||
|
bool isEpsetEqual(const SEpSet *s1, const SEpSet *s2);
|
||||||
|
|
||||||
|
void updateEpSet_s(SCorEpSet *pEpSet, SEpSet *pNewEpSet);
|
||||||
|
|
||||||
|
#endif // TDENGINE_TEP_H
|
|
@ -81,8 +81,6 @@ extern int64_t tsMaxRetentWindow;
|
||||||
// db parameters in client
|
// db parameters in client
|
||||||
extern int32_t tsCacheBlockSize;
|
extern int32_t tsCacheBlockSize;
|
||||||
extern int32_t tsBlocksPerVnode;
|
extern int32_t tsBlocksPerVnode;
|
||||||
extern int32_t tsMinTablePerVnode;
|
|
||||||
extern int32_t tsMaxTablePerVnode;
|
|
||||||
extern int32_t tsTableIncStepPerVnode;
|
extern int32_t tsTableIncStepPerVnode;
|
||||||
extern int32_t tsMaxVgroupsPerDb;
|
extern int32_t tsMaxVgroupsPerDb;
|
||||||
extern int16_t tsDaysPerFile;
|
extern int16_t tsDaysPerFile;
|
||||||
|
@ -113,16 +111,8 @@ extern int8_t tsEnableSlaveQuery;
|
||||||
extern int8_t tsEnableAdjustMaster;
|
extern int8_t tsEnableAdjustMaster;
|
||||||
|
|
||||||
// restful
|
// restful
|
||||||
extern int8_t tsEnableHttpModule;
|
|
||||||
extern int32_t tsRestRowLimit;
|
extern int32_t tsRestRowLimit;
|
||||||
extern uint16_t tsHttpPort;
|
|
||||||
extern int32_t tsHttpCacheSessions;
|
|
||||||
extern int32_t tsHttpSessionExpire;
|
|
||||||
extern int32_t tsHttpMaxThreads;
|
|
||||||
extern int8_t tsHttpEnableCompress;
|
|
||||||
extern int8_t tsHttpEnableRecordSql;
|
|
||||||
extern int8_t tsTelegrafUseFieldNum;
|
extern int8_t tsTelegrafUseFieldNum;
|
||||||
extern int8_t tsHttpDbNameMandatory;
|
|
||||||
|
|
||||||
// mqtt
|
// mqtt
|
||||||
extern int8_t tsEnableMqttModule;
|
extern int8_t tsEnableMqttModule;
|
||||||
|
@ -144,7 +134,6 @@ extern int8_t tsEnableStream;
|
||||||
|
|
||||||
// internal
|
// internal
|
||||||
extern int8_t tsPrintAuth;
|
extern int8_t tsPrintAuth;
|
||||||
extern int8_t tscEmbedded;
|
|
||||||
extern char tsVnodeDir[];
|
extern char tsVnodeDir[];
|
||||||
extern char tsMnodeDir[];
|
extern char tsMnodeDir[];
|
||||||
extern int64_t tsTickPerDay[3];
|
extern int64_t tsTickPerDay[3];
|
||||||
|
@ -193,7 +182,7 @@ extern SDiskCfg tsDiskCfg[];
|
||||||
#define NEEDTO_COMPRESSS_MSG(size) (tsCompressMsgSize != -1 && (size) > tsCompressMsgSize)
|
#define NEEDTO_COMPRESSS_MSG(size) (tsCompressMsgSize != -1 && (size) > tsCompressMsgSize)
|
||||||
|
|
||||||
void taosInitGlobalCfg();
|
void taosInitGlobalCfg();
|
||||||
int32_t taosCheckGlobalCfg();
|
int32_t taosCheckAndPrintCfg();
|
||||||
int32_t taosCfgDynamicOptions(char *msg);
|
int32_t taosCfgDynamicOptions(char *msg);
|
||||||
bool taosCheckBalanceCfgOptions(const char *option, int32_t *vnodeId, int32_t *dnodeId);
|
bool taosCheckBalanceCfgOptions(const char *option, int32_t *vnodeId, int32_t *dnodeId);
|
||||||
void taosAddDataDir(int index, char *v1, int level, int primary);
|
void taosAddDataDir(int index, char *v1, int level, int primary);
|
||||||
|
|
|
@ -256,7 +256,7 @@ typedef struct STQ {
|
||||||
// the collection of group handle
|
// the collection of group handle
|
||||||
// the handle of kvstore
|
// the handle of kvstore
|
||||||
char* path;
|
char* path;
|
||||||
STqCfg* tqConfig;
|
STqCfg* tqConfig;
|
||||||
TqLogReader* tqLogReader;
|
TqLogReader* tqLogReader;
|
||||||
TqMemRef tqMemRef;
|
TqMemRef tqMemRef;
|
||||||
TqMetaStore* tqMeta;
|
TqMetaStore* tqMeta;
|
||||||
|
|
|
@ -36,6 +36,8 @@ typedef struct SVnodeCfg {
|
||||||
struct {
|
struct {
|
||||||
/** write buffer size */
|
/** write buffer size */
|
||||||
uint64_t wsize;
|
uint64_t wsize;
|
||||||
|
uint64_t ssize;
|
||||||
|
uint64_t lsize;
|
||||||
/** use heap allocator or arena allocator */
|
/** use heap allocator or arena allocator */
|
||||||
bool isHeapAllocator;
|
bool isHeapAllocator;
|
||||||
};
|
};
|
||||||
|
@ -66,9 +68,11 @@ typedef struct SVnodeCfg {
|
||||||
/**
|
/**
|
||||||
* @brief Initialize the vnode module
|
* @brief Initialize the vnode module
|
||||||
*
|
*
|
||||||
|
* @param nthreads number of commit threads. 0 for no threads and
|
||||||
|
* a schedule queue should be given (TODO)
|
||||||
* @return int 0 for success and -1 for failure
|
* @return int 0 for success and -1 for failure
|
||||||
*/
|
*/
|
||||||
int vnodeInit();
|
int vnodeInit(uint16_t nthreads);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief clear a vnode
|
* @brief clear a vnode
|
||||||
|
|
|
@ -30,16 +30,23 @@ extern "C" {
|
||||||
|
|
||||||
struct SCatalog;
|
struct SCatalog;
|
||||||
|
|
||||||
typedef struct SMetaReq {
|
typedef struct SDBVgroupInfo {
|
||||||
char clusterId[TSDB_CLUSTER_ID_LEN];
|
int32_t vgroupVersion;
|
||||||
|
SArray *vgId;
|
||||||
|
int32_t hashRange;
|
||||||
|
int32_t hashNum;
|
||||||
|
} SDBVgroupInfo;
|
||||||
|
|
||||||
|
typedef struct SCatalogReq {
|
||||||
|
char clusterId[TSDB_CLUSTER_ID_LEN]; //????
|
||||||
SArray *pTableName; // table full name
|
SArray *pTableName; // table full name
|
||||||
SArray *pUdf; // udf name
|
SArray *pUdf; // udf name
|
||||||
bool qNodeEpset; // valid qnode
|
bool qNodeRequired; // valid qnode
|
||||||
} SMetaReq;
|
} SCatalogReq;
|
||||||
|
|
||||||
typedef struct SMetaData {
|
typedef struct SMetaData {
|
||||||
SArray *pTableMeta; // tableMeta
|
SArray *pTableMeta; // STableMeta array
|
||||||
SArray *pVgroupInfo; // vgroupInfo list
|
SArray *pVgroupInfo; // SVgroupInfo list
|
||||||
SArray *pUdfList; // udf info list
|
SArray *pUdfList; // udf info list
|
||||||
SEpSet *pEpSet; // qnode epset list
|
SEpSet *pEpSet; // qnode epset list
|
||||||
} SMetaData;
|
} SMetaData;
|
||||||
|
@ -78,32 +85,71 @@ typedef struct STableMeta {
|
||||||
SSchema schema[];
|
SSchema schema[];
|
||||||
} STableMeta;
|
} STableMeta;
|
||||||
|
|
||||||
|
typedef struct SCatalogCfg {
|
||||||
|
|
||||||
|
} SCatalogCfg;
|
||||||
|
|
||||||
|
int32_t catalogInit(SCatalogCfg *cfg);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Catalog service object, which is utilized to hold tableMeta (meta/vgroupInfo/udfInfo) at the client-side.
|
* Catalog service object, which is utilized to hold tableMeta (meta/vgroupInfo/udfInfo) at the client-side.
|
||||||
* There is ONLY one SCatalog object for one process space, and this function returns a singleton.
|
* There is ONLY one SCatalog object for one process space, and this function returns a singleton.
|
||||||
* @param pMgmtEps
|
* @param clusterId
|
||||||
* @return
|
* @return
|
||||||
*/
|
*/
|
||||||
struct SCatalog* getCatalogHandle(const SEpSet* pMgmtEps);
|
int32_t catalogGetHandle(const char *clusterId, struct SCatalog** catalogHandle);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
int32_t catalogGetVgroupVersion(struct SCatalog* pCatalog, int32_t* version);
|
||||||
|
int32_t catalogGetVgroup(struct SCatalog* pCatalog, void *pRpc, const SEpSet* pMgmtEps, SArray** pVgroupList);
|
||||||
|
int32_t catalogUpdateVgroup(struct SCatalog* pCatalog, SVgroupListInfo* pVgroup);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
int32_t catalogGetDBVgroupVersion(struct SCatalog* pCatalog, const char* dbName, int32_t* version);
|
||||||
|
int32_t catalogGetDBVgroup(struct SCatalog* pCatalog, void *pRpc, const SEpSet* pMgmtEps, const char* dbName, int32_t forceUpdate, SDBVgroupInfo** dbInfo);
|
||||||
|
int32_t catalogUpdateDBVgroup(struct SCatalog* pCatalog, const char* dbName, SDBVgroupInfo* dbInfo);
|
||||||
|
|
||||||
|
|
||||||
|
int32_t catalogGetTableMeta(struct SCatalog* pCatalog, void *pRpc, const SEpSet* pMgmtEps, const char* pTableName, STableMeta* pTableMeta);
|
||||||
|
int32_t catalogRenewTableMeta(struct SCatalog* pCatalog, void *pRpc, const SEpSet* pMgmtEps, const STableMeta* pTableMeta);
|
||||||
|
int32_t catalogRenewAndGetTableMeta(struct SCatalog* pCatalog, void *pRpc, const SEpSet* pMgmtEps, const STableMeta* pTableMeta, STableMeta* pNewTableMeta);
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* get table's vgroup list.
|
||||||
|
* @param clusterId
|
||||||
|
* @pVgroupList - array of SVgroupInfo
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
int32_t catalogGetTableVgroup(struct SCatalog* pCatalog, void *pRpc, const SEpSet* pMgmtEps, const char* pTableName, SArray* pVgroupList);
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Get the required meta data from mnode.
|
* Get the required meta data from mnode.
|
||||||
* Note that this is a synchronized API and is also thread-safety.
|
* Note that this is a synchronized API and is also thread-safety.
|
||||||
* @param pCatalog
|
* @param pCatalog
|
||||||
|
* @param pMgmtEps
|
||||||
* @param pMetaReq
|
* @param pMetaReq
|
||||||
* @param pMetaData
|
* @param pMetaData
|
||||||
* @return
|
* @return
|
||||||
*/
|
*/
|
||||||
int32_t catalogGetMetaData(struct SCatalog* pCatalog, const SMetaReq* pMetaReq, SMetaData* pMetaData);
|
int32_t catalogGetAllMeta(struct SCatalog* pCatalog, const SEpSet* pMgmtEps, const SCatalogReq* pReq, SMetaData* pRsp);
|
||||||
|
|
||||||
|
|
||||||
|
int32_t catalogGetQnodeList(struct SCatalog* pCatalog, const SEpSet* pMgmtEps, SEpSet* pQnodeEpSet);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Destroy catalog service handle
|
* Destroy catalog and relase all resources
|
||||||
* @param pCatalog
|
* @param pCatalog
|
||||||
*/
|
*/
|
||||||
void destroyCatalog(struct SCatalog* pCatalog);
|
void catalogDestroy(void);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#endif /*_TD_CATALOG_H_*/
|
#endif /*_TD_CATALOG_H_*/
|
||||||
|
|
|
@ -162,7 +162,7 @@ typedef struct SVgDataBlocks {
|
||||||
int64_t vgId; // virtual group id
|
int64_t vgId; // virtual group id
|
||||||
int32_t numOfTables; // number of tables in current submit block
|
int32_t numOfTables; // number of tables in current submit block
|
||||||
uint32_t size;
|
uint32_t size;
|
||||||
char *pData;
|
char *pData; // SMsgDesc + SSubmitMsg + SSubmitBlk + ...
|
||||||
} SVgDataBlocks;
|
} SVgDataBlocks;
|
||||||
|
|
||||||
typedef struct SInsertStmtInfo {
|
typedef struct SInsertStmtInfo {
|
||||||
|
|
|
@ -13,11 +13,20 @@
|
||||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifndef _TD_TSDB_COMMIT_QUEUE_H_
|
#ifndef _TD_QUERY_H_
|
||||||
#define _TD_TSDB_COMMIT_QUEUE_H_
|
#define _TD_QUERY_H_
|
||||||
|
|
||||||
typedef enum { COMMIT_REQ, COMPACT_REQ,COMMIT_CONFIG_REQ } TSDB_REQ_T;
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
|
||||||
int tsdbScheduleCommit(STsdbRepo *pRepo, TSDB_REQ_T req);
|
|
||||||
|
|
||||||
#endif /* _TD_TSDB_COMMIT_QUEUE_H_ */
|
extern int32_t (*queryBuildMsg[TSDB_MSG_TYPE_MAX])(void* input, char **msg, int32_t msgSize, int32_t *msgLen);
|
||||||
|
extern int32_t (*queryProcessMsgRsp[TSDB_MSG_TYPE_MAX])(void* output, char *msg, int32_t msgSize);
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /*_TD_QUERY_H_*/
|
|
@ -29,11 +29,6 @@ extern "C" {
|
||||||
|
|
||||||
extern int tsRpcHeadSize;
|
extern int tsRpcHeadSize;
|
||||||
|
|
||||||
typedef struct SRpcCorEpSet {
|
|
||||||
int32_t version;
|
|
||||||
SEpSet epSet;
|
|
||||||
} SRpcCorEpSet;
|
|
||||||
|
|
||||||
typedef struct SRpcConnInfo {
|
typedef struct SRpcConnInfo {
|
||||||
uint32_t clientIp;
|
uint32_t clientIp;
|
||||||
uint16_t clientPort;
|
uint16_t clientPort;
|
||||||
|
|
|
@ -32,6 +32,23 @@ extern int32_t wDebugFlag;
|
||||||
#define wDebug(...) { if (wDebugFlag & DEBUG_DEBUG) { taosPrintLog("WAL ", wDebugFlag, __VA_ARGS__); }}
|
#define wDebug(...) { if (wDebugFlag & DEBUG_DEBUG) { taosPrintLog("WAL ", wDebugFlag, __VA_ARGS__); }}
|
||||||
#define wTrace(...) { if (wDebugFlag & DEBUG_TRACE) { taosPrintLog("WAL ", wDebugFlag, __VA_ARGS__); }}
|
#define wTrace(...) { if (wDebugFlag & DEBUG_TRACE) { taosPrintLog("WAL ", wDebugFlag, __VA_ARGS__); }}
|
||||||
|
|
||||||
|
#define WAL_PREFIX "wal"
|
||||||
|
#define WAL_PREFIX_LEN 3
|
||||||
|
#define WAL_NOSUFFIX_LEN 20
|
||||||
|
#define WAL_SUFFIX_AT (WAL_NOSUFFIX_LEN+1)
|
||||||
|
#define WAL_LOG_SUFFIX "log"
|
||||||
|
#define WAL_INDEX_SUFFIX "idx"
|
||||||
|
#define WAL_REFRESH_MS 1000
|
||||||
|
#define WAL_MAX_SIZE (TSDB_MAX_WAL_SIZE + sizeof(SWalHead) + 16)
|
||||||
|
#define WAL_PATH_LEN (TSDB_FILENAME_LEN + 12)
|
||||||
|
#define WAL_FILE_LEN (WAL_PATH_LEN + 32)
|
||||||
|
|
||||||
|
#define WAL_IDX_ENTRY_SIZE (sizeof(int64_t)*2)
|
||||||
|
#define WAL_CUR_POS_WRITABLE 1
|
||||||
|
#define WAL_CUR_FILE_WRITABLE 2
|
||||||
|
#define WAL_CUR_FAILED 4
|
||||||
|
|
||||||
|
#pragma pack(push,1)
|
||||||
typedef enum {
|
typedef enum {
|
||||||
TAOS_WAL_NOLOG = 0,
|
TAOS_WAL_NOLOG = 0,
|
||||||
TAOS_WAL_WRITE = 1,
|
TAOS_WAL_WRITE = 1,
|
||||||
|
@ -43,8 +60,9 @@ typedef struct SWalReadHead {
|
||||||
uint8_t msgType;
|
uint8_t msgType;
|
||||||
int8_t reserved[2];
|
int8_t reserved[2];
|
||||||
int32_t len;
|
int32_t len;
|
||||||
|
//int64_t ingestTs; //not implemented
|
||||||
int64_t version;
|
int64_t version;
|
||||||
char cont[];
|
char body[];
|
||||||
} SWalReadHead;
|
} SWalReadHead;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
|
@ -52,8 +70,9 @@ typedef struct {
|
||||||
int32_t fsyncPeriod; // millisecond
|
int32_t fsyncPeriod; // millisecond
|
||||||
int32_t retentionPeriod; // secs
|
int32_t retentionPeriod; // secs
|
||||||
int32_t rollPeriod; // secs
|
int32_t rollPeriod; // secs
|
||||||
|
int64_t retentionSize;
|
||||||
int64_t segSize;
|
int64_t segSize;
|
||||||
EWalType walLevel; // wal level
|
EWalType level; // wal level
|
||||||
} SWalCfg;
|
} SWalCfg;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
|
@ -70,64 +89,48 @@ typedef struct {
|
||||||
SWalReadHead head;
|
SWalReadHead head;
|
||||||
} SWalHead;
|
} SWalHead;
|
||||||
|
|
||||||
#define WAL_PREFIX "wal"
|
typedef struct SWalVer {
|
||||||
#define WAL_PREFIX_LEN 3
|
int64_t firstVer;
|
||||||
#define WAL_NOSUFFIX_LEN 20
|
int64_t verInSnapshotting;
|
||||||
#define WAL_SUFFIX_AT (WAL_NOSUFFIX_LEN+1)
|
int64_t snapshotVer;
|
||||||
#define WAL_LOG_SUFFIX "log"
|
int64_t commitVer;
|
||||||
#define WAL_INDEX_SUFFIX "idx"
|
int64_t lastVer;
|
||||||
#define WAL_REFRESH_MS 1000
|
} SWalVer;
|
||||||
#define WAL_MAX_SIZE (TSDB_MAX_WAL_SIZE + sizeof(SWalHead) + 16)
|
|
||||||
#define WAL_SIGNATURE ((uint32_t)(0xFAFBFDFEUL))
|
|
||||||
#define WAL_PATH_LEN (TSDB_FILENAME_LEN + 12)
|
|
||||||
#define WAL_FILE_LEN (WAL_PATH_LEN + 32)
|
|
||||||
//#define WAL_FILE_NUM 1 // 3
|
|
||||||
#define WAL_FILESET_MAX 128
|
|
||||||
|
|
||||||
#define WAL_IDX_ENTRY_SIZE (sizeof(int64_t)*2)
|
|
||||||
#define WAL_CUR_POS_WRITABLE 1
|
|
||||||
#define WAL_CUR_FILE_WRITABLE 2
|
|
||||||
#define WAL_CUR_FAILED 4
|
|
||||||
|
|
||||||
typedef struct SWal {
|
typedef struct SWal {
|
||||||
// cfg
|
// cfg
|
||||||
int32_t vgId;
|
SWalCfg cfg;
|
||||||
int32_t fsyncPeriod; // millisecond
|
SWalVer vers;
|
||||||
int32_t rollPeriod; // second
|
|
||||||
int64_t segSize;
|
|
||||||
int64_t retentionSize;
|
|
||||||
int32_t retentionPeriod;
|
|
||||||
EWalType level;
|
|
||||||
//total size
|
|
||||||
int64_t totSize;
|
|
||||||
//fsync seq
|
|
||||||
int32_t fsyncSeq;
|
|
||||||
//reference
|
|
||||||
int64_t refId;
|
|
||||||
//write tfd
|
|
||||||
int64_t writeLogTfd;
|
|
||||||
int64_t writeIdxTfd;
|
|
||||||
//wal lifecycle
|
|
||||||
int64_t firstVersion;
|
|
||||||
int64_t snapshotVersion;
|
|
||||||
int64_t commitVersion;
|
|
||||||
int64_t lastVersion;
|
|
||||||
//snapshotting version
|
|
||||||
int64_t snapshottingVer;
|
|
||||||
//roll status
|
|
||||||
int64_t lastRollSeq;
|
|
||||||
//file set
|
//file set
|
||||||
int32_t writeCur;
|
int32_t writeCur;
|
||||||
|
int64_t writeLogTfd;
|
||||||
|
int64_t writeIdxTfd;
|
||||||
SArray* fileInfoSet;
|
SArray* fileInfoSet;
|
||||||
//ctl
|
//ctl
|
||||||
int32_t curStatus;
|
int32_t curStatus;
|
||||||
|
int32_t fsyncSeq;
|
||||||
|
int64_t totSize;
|
||||||
|
int64_t refId;
|
||||||
|
int64_t lastRollSeq;
|
||||||
pthread_mutex_t mutex;
|
pthread_mutex_t mutex;
|
||||||
//path
|
//path
|
||||||
char path[WAL_PATH_LEN];
|
char path[WAL_PATH_LEN];
|
||||||
//reusable write head
|
//reusable write head
|
||||||
SWalHead head;
|
SWalHead writeHead;
|
||||||
} SWal; // WAL HANDLE
|
} SWal; // WAL HANDLE
|
||||||
|
|
||||||
|
typedef struct SWalReadHandle {
|
||||||
|
SWal* pWal;
|
||||||
|
int64_t readLogTfd;
|
||||||
|
int64_t readIdxTfd;
|
||||||
|
int64_t curFileFirstVer;
|
||||||
|
int64_t curVersion;
|
||||||
|
int64_t capacity;
|
||||||
|
int64_t status; //if cursor valid
|
||||||
|
SWalHead* pHead;
|
||||||
|
} SWalReadHandle;
|
||||||
|
#pragma pack(pop)
|
||||||
|
|
||||||
typedef int32_t (*FWalWrite)(void *ahandle, void *pHead);
|
typedef int32_t (*FWalWrite)(void *ahandle, void *pHead);
|
||||||
|
|
||||||
// module initialization
|
// module initialization
|
||||||
|
@ -153,6 +156,10 @@ int32_t walEndTakeSnapshot(SWal *);
|
||||||
//int32_t walDataCorrupted(SWal*);
|
//int32_t walDataCorrupted(SWal*);
|
||||||
|
|
||||||
// read
|
// read
|
||||||
|
SWalReadHandle* walOpenReadHandle(SWal *);
|
||||||
|
void walCloseReadHandle(SWalReadHandle *);
|
||||||
|
int32_t walReadWithHandle(SWalReadHandle *pRead, int64_t ver);
|
||||||
|
|
||||||
int32_t walRead(SWal *, SWalHead **, int64_t ver);
|
int32_t walRead(SWal *, SWalHead **, int64_t ver);
|
||||||
int32_t walReadWithFp(SWal *, FWalWrite writeFp, int64_t verStart, int32_t readNum);
|
int32_t walReadWithFp(SWal *, FWalWrite writeFp, int64_t verStart, int32_t readNum);
|
||||||
|
|
||||||
|
|
|
@ -57,7 +57,7 @@ int64_t taosGetPthreadId(pthread_t thread);
|
||||||
void taosResetPthread(pthread_t* thread);
|
void taosResetPthread(pthread_t* thread);
|
||||||
bool taosComparePthread(pthread_t first, pthread_t second);
|
bool taosComparePthread(pthread_t first, pthread_t second);
|
||||||
int32_t taosGetPId();
|
int32_t taosGetPId();
|
||||||
int32_t taosGetCurrentAPPName(char* name, int32_t* len);
|
int32_t taosGetAppName(char* name, int32_t* len);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
|
|
|
@ -117,6 +117,7 @@ int32_t* taosGetErrno();
|
||||||
#define TSDB_CODE_TSC_INVALID_JSON TAOS_DEF_ERROR_CODE(0, 0x0221) //"Invalid JSON format")
|
#define TSDB_CODE_TSC_INVALID_JSON TAOS_DEF_ERROR_CODE(0, 0x0221) //"Invalid JSON format")
|
||||||
#define TSDB_CODE_TSC_INVALID_JSON_TYPE TAOS_DEF_ERROR_CODE(0, 0x0222) //"Invalid JSON data type")
|
#define TSDB_CODE_TSC_INVALID_JSON_TYPE TAOS_DEF_ERROR_CODE(0, 0x0222) //"Invalid JSON data type")
|
||||||
#define TSDB_CODE_TSC_VALUE_OUT_OF_RANGE TAOS_DEF_ERROR_CODE(0, 0x0223) //"Value out of range")
|
#define TSDB_CODE_TSC_VALUE_OUT_OF_RANGE TAOS_DEF_ERROR_CODE(0, 0x0223) //"Value out of range")
|
||||||
|
#define TSDB_CODE_TSC_INVALID_INPUT TAOS_DEF_ERROR_CODE(0, 0X0224) //"Invalid tsc input")
|
||||||
|
|
||||||
// mnode
|
// mnode
|
||||||
#define TSDB_CODE_MND_MSG_NOT_PROCESSED TAOS_DEF_ERROR_CODE(0, 0x0300)
|
#define TSDB_CODE_MND_MSG_NOT_PROCESSED TAOS_DEF_ERROR_CODE(0, 0x0300)
|
||||||
|
@ -501,6 +502,13 @@ int32_t* taosGetErrno();
|
||||||
// monitor
|
// monitor
|
||||||
#define TSDB_CODE_MON_CONNECTION_INVALID TAOS_DEF_ERROR_CODE(0, 0x2300) //"monitor invalid monitor db connection")
|
#define TSDB_CODE_MON_CONNECTION_INVALID TAOS_DEF_ERROR_CODE(0, 0x2300) //"monitor invalid monitor db connection")
|
||||||
|
|
||||||
|
// catalog
|
||||||
|
#define TSDB_CODE_CTG_INTERNAL_ERROR TAOS_DEF_ERROR_CODE(0, 0x2400) //catalog interval error
|
||||||
|
#define TSDB_CODE_CTG_INVALID_INPUT TAOS_DEF_ERROR_CODE(0, 0x2401) //invalid catalog input parameters
|
||||||
|
#define TSDB_CODE_CTG_NOT_READY TAOS_DEF_ERROR_CODE(0, 0x2402) //catalog is not ready
|
||||||
|
#define TSDB_CODE_CTG_MEM_ERROR TAOS_DEF_ERROR_CODE(0, 0x2403) //catalog memory error
|
||||||
|
#define TSDB_CODE_CTG_SYS_ERROR TAOS_DEF_ERROR_CODE(0, 0x2404) //catalog system error
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -43,6 +43,13 @@ typedef struct SArray {
|
||||||
*/
|
*/
|
||||||
void* taosArrayInit(size_t size, size_t elemSize);
|
void* taosArrayInit(size_t size, size_t elemSize);
|
||||||
|
|
||||||
|
/**
|
||||||
|
*
|
||||||
|
* @param tsize
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
int32_t taosArrayEnsureCap(SArray* pArray, size_t tsize);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
*
|
*
|
||||||
* @param pArray
|
* @param pArray
|
||||||
|
|
|
@ -20,7 +20,7 @@
|
||||||
extern "C" {
|
extern "C" {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#define TSDB_CFG_MAX_NUM 123
|
#define TSDB_CFG_MAX_NUM 115
|
||||||
#define TSDB_CFG_PRINT_LEN 23
|
#define TSDB_CFG_PRINT_LEN 23
|
||||||
#define TSDB_CFG_OPTION_LEN 24
|
#define TSDB_CFG_OPTION_LEN 24
|
||||||
#define TSDB_CFG_VALUE_LEN 41
|
#define TSDB_CFG_VALUE_LEN 41
|
||||||
|
@ -83,11 +83,11 @@ extern int32_t tsGlobalConfigNum;
|
||||||
extern char * tsCfgStatusStr[];
|
extern char * tsCfgStatusStr[];
|
||||||
|
|
||||||
void taosReadGlobalLogCfg();
|
void taosReadGlobalLogCfg();
|
||||||
int32_t taosReadGlobalCfg();
|
int32_t taosReadCfgFromFile();
|
||||||
void taosPrintGlobalCfg();
|
void taosPrintCfg();
|
||||||
void taosDumpGlobalCfg();
|
void taosDumpGlobalCfg();
|
||||||
|
|
||||||
void taosInitConfigOption(SGlobalCfg cfg);
|
void taosAddConfigOption(SGlobalCfg cfg);
|
||||||
SGlobalCfg *taosGetConfigOption(const char *option);
|
SGlobalCfg *taosGetConfigOption(const char *option);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
|
|
|
@ -161,7 +161,7 @@ do { \
|
||||||
|
|
||||||
#define TSDB_NODE_NAME_LEN 64
|
#define TSDB_NODE_NAME_LEN 64
|
||||||
#define TSDB_TABLE_NAME_LEN 193 // it is a null-terminated string
|
#define TSDB_TABLE_NAME_LEN 193 // it is a null-terminated string
|
||||||
#define TSDB_DB_NAME_LEN 33
|
#define TSDB_DB_NAME_LEN 65
|
||||||
#define TSDB_FULL_DB_NAME_LEN (TSDB_ACCT_ID_LEN + TSDB_DB_NAME_LEN)
|
#define TSDB_FULL_DB_NAME_LEN (TSDB_ACCT_ID_LEN + TSDB_DB_NAME_LEN)
|
||||||
|
|
||||||
#define TSDB_FUNC_NAME_LEN 65
|
#define TSDB_FUNC_NAME_LEN 65
|
||||||
|
@ -193,7 +193,7 @@ do { \
|
||||||
#define TSDB_MAX_TAG_CONDITIONS 1024
|
#define TSDB_MAX_TAG_CONDITIONS 1024
|
||||||
|
|
||||||
#define TSDB_AUTH_LEN 16
|
#define TSDB_AUTH_LEN 16
|
||||||
#define TSDB_KEY_LEN 16
|
#define TSDB_PASSWORD_LEN 32
|
||||||
#define TSDB_VERSION_LEN 12
|
#define TSDB_VERSION_LEN 12
|
||||||
#define TSDB_LABEL_LEN 8
|
#define TSDB_LABEL_LEN 8
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,169 @@
|
||||||
|
/*
|
||||||
|
* 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_UTIL_TDLIST_H_
|
||||||
|
#define _TD_UTIL_TDLIST_H_
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Single linked list
|
||||||
|
#define TD_SLIST_NODE(TYPE) \
|
||||||
|
struct { \
|
||||||
|
struct TYPE *sl_next_; \
|
||||||
|
}
|
||||||
|
|
||||||
|
#define TD_SLIST(TYPE) \
|
||||||
|
struct { \
|
||||||
|
struct TYPE *sl_head_; \
|
||||||
|
int sl_neles_; \
|
||||||
|
}
|
||||||
|
|
||||||
|
#define TD_SLIST_HEAD(sl) ((sl)->sl_head_)
|
||||||
|
#define TD_SLIST_NELES(sl) ((sl)->sl_neles_)
|
||||||
|
#define TD_SLIST_NODE_NEXT(sln) ((sln)->sl_next_)
|
||||||
|
|
||||||
|
#define tSListInit(sl) \
|
||||||
|
do { \
|
||||||
|
(sl)->sl_head_ = NULL; \
|
||||||
|
(sl)->sl_neles_ = 0; \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
#define tSListPush(sl, sln) \
|
||||||
|
do { \
|
||||||
|
TD_SLIST_NODE_NEXT(sln) = TD_SLIST_HEAD(sl); \
|
||||||
|
TD_SLIST_HEAD(sl) = (sln); \
|
||||||
|
TD_SLIST_NELES(sl) += 1; \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
#define tSListPop(sl) \
|
||||||
|
do { \
|
||||||
|
TD_SLIST_HEAD(sl) = TD_SLIST_NODE_NEXT(TD_SLIST_HEAD(sl)); \
|
||||||
|
TD_SLIST_NELES(sl) -= 1; \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
// Double linked list
|
||||||
|
#define TD_DLIST_NODE(TYPE) \
|
||||||
|
struct { \
|
||||||
|
struct TYPE *dl_prev_; \
|
||||||
|
struct TYPE *dl_next_; \
|
||||||
|
}
|
||||||
|
|
||||||
|
#define TD_DLIST(TYPE) \
|
||||||
|
struct { \
|
||||||
|
struct TYPE *dl_head_; \
|
||||||
|
struct TYPE *dl_tail_; \
|
||||||
|
int dl_neles_; \
|
||||||
|
}
|
||||||
|
|
||||||
|
#define TD_DLIST_NODE_PREV(dln) ((dln)->dl_prev_)
|
||||||
|
#define TD_DLIST_NODE_NEXT(dln) ((dln)->dl_next_)
|
||||||
|
#define TD_DLIST_HEAD(dl) ((dl)->dl_head_)
|
||||||
|
#define TD_DLIST_TAIL(dl) ((dl)->dl_tail_)
|
||||||
|
#define TD_DLIST_NELES(dl) ((dl)->dl_neles_)
|
||||||
|
|
||||||
|
#define tDListInit(dl) \
|
||||||
|
do { \
|
||||||
|
TD_DLIST_HEAD(dl) = TD_DLIST_TAIL(dl) = NULL; \
|
||||||
|
TD_DLIST_NELES(dl) = 0; \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
#define tDListAppend(dl, dln) \
|
||||||
|
do { \
|
||||||
|
if (TD_DLIST_HEAD(dl) == NULL) { \
|
||||||
|
TD_DLIST_NODE_PREV(dln) = TD_DLIST_NODE_NEXT(dln) = NULL; \
|
||||||
|
TD_DLIST_HEAD(dl) = TD_DLIST_TAIL(dl) = (dln); \
|
||||||
|
} else { \
|
||||||
|
TD_DLIST_NODE_PREV(dln) = TD_DLIST_TAIL(dl); \
|
||||||
|
TD_DLIST_NODE_NEXT(dln) = NULL; \
|
||||||
|
TD_DLIST_NODE_NEXT(TD_DLIST_TAIL(dl)) = (dln); \
|
||||||
|
TD_DLIST_TAIL(dl) = (dln); \
|
||||||
|
} \
|
||||||
|
TD_DLIST_NELES(dl) += 1; \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
#define tDListPrepend(dl, dln) \
|
||||||
|
do { \
|
||||||
|
if (TD_DLIST_HEAD(dl) == NULL) { \
|
||||||
|
TD_DLIST_NODE_PREV(dln) = TD_DLIST_NODE_NEXT(dln) = NULL; \
|
||||||
|
TD_DLIST_HEAD(dl) = TD_DLIST_TAIL(dl) = (dln); \
|
||||||
|
} else { \
|
||||||
|
TD_DLIST_NODE_PREV(dln) = NULL; \
|
||||||
|
TD_DLIST_NODE_NEXT(dln) = TD_DLIST_HEAD(dl); \
|
||||||
|
TD_DLIST_NODE_PREV(TD_DLIST_HEAD(dl)) = (dln); \
|
||||||
|
TD_DLIST_HEAD(dl) = (dln); \
|
||||||
|
} \
|
||||||
|
TD_DLIST_NELES(dl) += 1; \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
#define tDListPop(dl, dln) \
|
||||||
|
do { \
|
||||||
|
if (TD_DLIST_HEAD(dl) == (dln)) { \
|
||||||
|
TD_DLIST_HEAD(dl) = TD_DLIST_NODE_NEXT(dln); \
|
||||||
|
} \
|
||||||
|
if (TD_DLIST_TAIL(dl) == (dln)) { \
|
||||||
|
TD_DLIST_TAIL(dl) = TD_DLIST_NODE_PREV(dln); \
|
||||||
|
} \
|
||||||
|
if (TD_DLIST_NODE_PREV(dln) != NULL) { \
|
||||||
|
TD_DLIST_NODE_NEXT(TD_DLIST_NODE_PREV(dln)) = TD_DLIST_NODE_NEXT(dln); \
|
||||||
|
} \
|
||||||
|
if (TD_DLIST_NODE_NEXT(dln) != NULL) { \
|
||||||
|
TD_DLIST_NODE_PREV(TD_DLIST_NODE_NEXT(dln)) = TD_DLIST_NODE_PREV(dln); \
|
||||||
|
} \
|
||||||
|
TD_DLIST_NELES(dl) -= 1; \
|
||||||
|
TD_DLIST_NODE_PREV(dln) = TD_DLIST_NODE_NEXT(dln) = NULL; \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
// List iterator
|
||||||
|
#define TD_LIST_FITER 0
|
||||||
|
#define TD_LIST_BITER 1
|
||||||
|
#define TD_LIST_ITER(S) \
|
||||||
|
struct { \
|
||||||
|
int it_dir_; \
|
||||||
|
S * it_next_; \
|
||||||
|
S * it_ptr_; \
|
||||||
|
TD_DLIST(S) * it_list_; \
|
||||||
|
}
|
||||||
|
|
||||||
|
#define tlistIterInit(it, l, dir) \
|
||||||
|
(it)->it_dir_ = (dir); \
|
||||||
|
(it)->it_list_ = l; \
|
||||||
|
if ((dir) == TD_LIST_FITER) { \
|
||||||
|
(it)->it_next_ = (l)->dl_head_; \
|
||||||
|
} else { \
|
||||||
|
(it)->it_next_ = (l)->dl_tail_; \
|
||||||
|
}
|
||||||
|
|
||||||
|
#define tlistIterNext(it) \
|
||||||
|
({ \
|
||||||
|
(it)->it_ptr_ = (it)->it_next_; \
|
||||||
|
if ((it)->it_next_ != NULL) { \
|
||||||
|
if ((it)->it_dir_ == TD_LIST_FITER) { \
|
||||||
|
(it)->it_next_ = (it)->it_next_->next_; \
|
||||||
|
} else { \
|
||||||
|
(it)->it_next_ = (it)->it_next_->prev_; \
|
||||||
|
} \
|
||||||
|
} \
|
||||||
|
(it)->it_ptr_; \
|
||||||
|
})
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /*_TD_UTIL_TDLIST_H_*/
|
|
@ -45,6 +45,8 @@ extern int32_t sDebugFlag;
|
||||||
extern int32_t tsdbDebugFlag;
|
extern int32_t tsdbDebugFlag;
|
||||||
extern int32_t cqDebugFlag;
|
extern int32_t cqDebugFlag;
|
||||||
extern int32_t debugFlag;
|
extern int32_t debugFlag;
|
||||||
|
extern int32_t ctgDebugFlag;
|
||||||
|
|
||||||
|
|
||||||
#define DEBUG_FATAL 1U
|
#define DEBUG_FATAL 1U
|
||||||
#define DEBUG_ERROR DEBUG_FATAL
|
#define DEBUG_ERROR DEBUG_FATAL
|
||||||
|
|
|
@ -0,0 +1,42 @@
|
||||||
|
/*
|
||||||
|
* 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_UTIL_MACRO_H_
|
||||||
|
#define _TD_UTIL_MACRO_H_
|
||||||
|
|
||||||
|
#include "os.h"
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Module init/clear MACRO definitions
|
||||||
|
#define TD_MOD_UNINITIALIZED 0
|
||||||
|
#define TD_MOD_INITIALIZED 1
|
||||||
|
|
||||||
|
#define TD_MOD_UNCLEARD 0
|
||||||
|
#define TD_MOD_CLEARD 1
|
||||||
|
|
||||||
|
typedef int8_t td_mode_flag_t;
|
||||||
|
|
||||||
|
#define TD_CHECK_AND_SET_MODE_INIT(FLAG) atomic_val_compare_exchange_8((FLAG), TD_MOD_UNINITIALIZED, TD_MOD_INITIALIZED)
|
||||||
|
|
||||||
|
#define TD_CHECK_AND_SET_MOD_CLEAR(FLAG) atomic_val_compare_exchange_8((FLAG), TD_MOD_UNCLEARD, TD_MOD_CLEARD)
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /*_TD_UTIL_MACRO_H_*/
|
|
@ -45,14 +45,25 @@ char *taosIpStr(uint32_t ipInt);
|
||||||
uint32_t ip2uint(const char *const ip_addr);
|
uint32_t ip2uint(const char *const ip_addr);
|
||||||
void taosIp2String(uint32_t ip, char *str);
|
void taosIp2String(uint32_t ip, char *str);
|
||||||
void taosIpPort2String(uint32_t ip, uint16_t port, char *str);
|
void taosIpPort2String(uint32_t ip, uint16_t port, char *str);
|
||||||
int32_t taosGetFqdnPortFromEp(const char *ep, char *fqdn, uint16_t *port);
|
|
||||||
|
|
||||||
static FORCE_INLINE void taosEncryptPass(uint8_t *inBuf, size_t inLen, char *target) {
|
static FORCE_INLINE void taosEncryptPass(uint8_t *inBuf, size_t inLen, char *target) {
|
||||||
T_MD5_CTX context;
|
T_MD5_CTX context;
|
||||||
tMD5Init(&context);
|
tMD5Init(&context);
|
||||||
tMD5Update(&context, inBuf, (unsigned int)inLen);
|
tMD5Update(&context, inBuf, (unsigned int)inLen);
|
||||||
tMD5Final(&context);
|
tMD5Final(&context);
|
||||||
memcpy(target, context.digest, TSDB_KEY_LEN);
|
memcpy(target, context.digest, tListLen(context.digest));
|
||||||
|
}
|
||||||
|
|
||||||
|
static FORCE_INLINE void taosEncryptPass_c(uint8_t *inBuf, size_t len, char *target) {
|
||||||
|
T_MD5_CTX context;
|
||||||
|
tMD5Init(&context);
|
||||||
|
tMD5Update(&context, inBuf, (unsigned int)len);
|
||||||
|
tMD5Final(&context);
|
||||||
|
|
||||||
|
sprintf(target, "%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x", context.digest[0], context.digest[1], context.digest[2],
|
||||||
|
context.digest[3], context.digest[4], context.digest[5], context.digest[6], context.digest[7],
|
||||||
|
context.digest[8], context.digest[9], context.digest[10], context.digest[11], context.digest[12],
|
||||||
|
context.digest[13], context.digest[14], context.digest[15]);
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
|
|
|
@ -2,9 +2,14 @@ aux_source_directory(src CLIENT_SRC)
|
||||||
add_library(taos ${CLIENT_SRC})
|
add_library(taos ${CLIENT_SRC})
|
||||||
target_include_directories(
|
target_include_directories(
|
||||||
taos
|
taos
|
||||||
PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}/inc"
|
PUBLIC "${CMAKE_SOURCE_DIR}/include/client"
|
||||||
|
PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}/inc"
|
||||||
)
|
)
|
||||||
target_link_libraries(
|
target_link_libraries(
|
||||||
taos
|
taos
|
||||||
|
PRIVATE common
|
||||||
INTERFACE api
|
INTERFACE api
|
||||||
|
PRIVATE os util common transport parser
|
||||||
)
|
)
|
||||||
|
|
||||||
|
ADD_SUBDIRECTORY(test)
|
|
@ -0,0 +1,146 @@
|
||||||
|
/*
|
||||||
|
* 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 TDENGINE_CLIENTINT_H
|
||||||
|
#define TDENGINE_CLIENTINT_H
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include "taos.h"
|
||||||
|
#include "taosmsg.h"
|
||||||
|
#include "thash.h"
|
||||||
|
#include "tlist.h"
|
||||||
|
#include "trpc.h"
|
||||||
|
#include "tdef.h"
|
||||||
|
#include "tmsgtype.h"
|
||||||
|
#include "tep.h"
|
||||||
|
|
||||||
|
typedef struct SQueryExecMetric {
|
||||||
|
int64_t start; // start timestamp
|
||||||
|
int64_t parsed; // start to parse
|
||||||
|
int64_t send; // start to send to server
|
||||||
|
int64_t rsp; // receive response from server
|
||||||
|
} SQueryExecMetric;
|
||||||
|
|
||||||
|
typedef struct SInstanceActivity {
|
||||||
|
uint64_t numOfInsertsReq;
|
||||||
|
uint64_t numOfInsertRows;
|
||||||
|
uint64_t insertElapsedTime;
|
||||||
|
uint64_t insertBytes; // submit to tsdb since launched.
|
||||||
|
|
||||||
|
uint64_t fetchBytes;
|
||||||
|
uint64_t queryElapsedTime;
|
||||||
|
uint64_t numOfSlowQueries;
|
||||||
|
uint64_t totalRequests;
|
||||||
|
uint64_t currentRequests; // the number of SRequestObj
|
||||||
|
} SInstanceActivity;
|
||||||
|
|
||||||
|
typedef struct SHeartBeatInfo {
|
||||||
|
void *pTimer; // timer, used to send request msg to mnode
|
||||||
|
} SHeartBeatInfo;
|
||||||
|
|
||||||
|
typedef struct SAppInstInfo {
|
||||||
|
int64_t numOfConns;
|
||||||
|
SCorEpSet mgmtEp;
|
||||||
|
SInstanceActivity summary;
|
||||||
|
SList *pConnList; // STscObj linked list
|
||||||
|
uint32_t clusterId;
|
||||||
|
void *pTransporter;
|
||||||
|
} SAppInstInfo;
|
||||||
|
|
||||||
|
typedef struct SAppInfo {
|
||||||
|
int64_t startTime;
|
||||||
|
char appName[TSDB_APP_NAME_LEN];
|
||||||
|
char *ep;
|
||||||
|
int32_t pid;
|
||||||
|
int32_t numOfThreads;
|
||||||
|
SHeartBeatInfo hb;
|
||||||
|
SHashObj *pInstMap;
|
||||||
|
} SAppInfo;
|
||||||
|
|
||||||
|
typedef struct STscObj {
|
||||||
|
char user[TSDB_USER_LEN];
|
||||||
|
char pass[TSDB_PASSWORD_LEN];
|
||||||
|
char acctId[TSDB_ACCT_ID_LEN];
|
||||||
|
char db[TSDB_ACCT_ID_LEN + TSDB_DB_NAME_LEN];
|
||||||
|
uint32_t connId;
|
||||||
|
uint64_t id; // ref ID returned by taosAddRef
|
||||||
|
// struct SSqlObj *sqlList;
|
||||||
|
void *pTransporter;
|
||||||
|
pthread_mutex_t mutex; // used to protect the operation on db
|
||||||
|
int32_t numOfReqs; // number of sqlObj from this tscObj
|
||||||
|
SAppInstInfo *pAppInfo;
|
||||||
|
} STscObj;
|
||||||
|
|
||||||
|
typedef struct SReqBody {
|
||||||
|
tsem_t rspSem; // not used now
|
||||||
|
void* fp;
|
||||||
|
void* param;
|
||||||
|
} SRequestBody;
|
||||||
|
|
||||||
|
typedef struct SRequestObj {
|
||||||
|
uint64_t requestId;
|
||||||
|
int32_t type; // request type
|
||||||
|
STscObj *pTscObj;
|
||||||
|
SQueryExecMetric metric;
|
||||||
|
char *sqlstr; // sql string
|
||||||
|
SRequestBody body;
|
||||||
|
int64_t self;
|
||||||
|
char *msgBuf;
|
||||||
|
int32_t code;
|
||||||
|
void *pInfo; // sql parse info, generated by parser module
|
||||||
|
} SRequestObj;
|
||||||
|
|
||||||
|
typedef struct SRequestMsgBody {
|
||||||
|
int32_t msgType;
|
||||||
|
void *pData;
|
||||||
|
int32_t msgLen;
|
||||||
|
uint64_t requestId;
|
||||||
|
uint64_t requestObjRefId;
|
||||||
|
} SRequestMsgBody;
|
||||||
|
|
||||||
|
extern SAppInfo appInfo;
|
||||||
|
extern int32_t tscReqRef;
|
||||||
|
extern void *tscQhandle;
|
||||||
|
extern int32_t tscConnRef;
|
||||||
|
|
||||||
|
extern int (*tscBuildMsg[TSDB_SQL_MAX])(SRequestObj *pRequest, SRequestMsgBody *pMsg);
|
||||||
|
extern int (*handleRequestRspFp[TSDB_SQL_MAX])(SRequestObj *pRequest, const char* pMsg, int32_t msgLen);
|
||||||
|
|
||||||
|
int taos_init();
|
||||||
|
|
||||||
|
void* createTscObj(const char* user, const char* auth, const char *ip, uint32_t port, SAppInstInfo* pAppInfo);
|
||||||
|
void destroyTscObj(void*pObj);
|
||||||
|
|
||||||
|
void* createRequest(STscObj* pObj, __taos_async_fn_t fp, void* param, int32_t type);
|
||||||
|
void destroyRequest(SRequestObj* pRequest);
|
||||||
|
|
||||||
|
TAOS *taos_connect_internal(const char *ip, const char *user, const char *pass, const char *auth, const char *db, uint16_t port);
|
||||||
|
|
||||||
|
void taos_init_imp(void);
|
||||||
|
int taos_options_imp(TSDB_OPTION option, const char *str);
|
||||||
|
|
||||||
|
void* openTransporter(const char *user, const char *auth);
|
||||||
|
|
||||||
|
void processMsgFromServer(void* parent, SRpcMsg* pMsg, SEpSet* pEpSet);
|
||||||
|
void initMsgHandleFp();
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif // TDENGINE_CLIENTINT_H
|
|
@ -0,0 +1,37 @@
|
||||||
|
/*
|
||||||
|
* 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 TDENGINE_TSCLOG_H
|
||||||
|
#define TDENGINE_TSCLOG_H
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include "tlog.h"
|
||||||
|
|
||||||
|
#define tscFatal(...) do { if (cDebugFlag & DEBUG_FATAL) { taosPrintLog("TSC FATAL ", cDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define tscError(...) do { if (cDebugFlag & DEBUG_ERROR) { taosPrintLog("TSC ERROR ", cDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define tscWarn(...) do { if (cDebugFlag & DEBUG_WARN) { taosPrintLog("TSC WARN ", cDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define tscInfo(...) do { if (cDebugFlag & DEBUG_INFO) { taosPrintLog("TSC ", cDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define tscDebug(...) do { if (cDebugFlag & DEBUG_DEBUG) { taosPrintLog("TSC ", cDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define tscTrace(...) do { if (cDebugFlag & DEBUG_TRACE) { taosPrintLog("TSC ", cDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define tscDebugL(...) do { if (cDebugFlag & DEBUG_DEBUG) { taosPrintLongString("TSC ", cDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif
|
|
@ -13,11 +13,51 @@
|
||||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
//#include "taos.h"
|
#include "os.h"
|
||||||
|
#include "tdef.h"
|
||||||
|
#include "tglobal.h"
|
||||||
|
#include "clientInt.h"
|
||||||
|
#include "tscLog.h"
|
||||||
|
|
||||||
//TAOS_RES *taos_query(TAOS *taos, const char *sql) {
|
TAOS *taos_connect(const char *ip, const char *user, const char *pass, const char *db, uint16_t port) {
|
||||||
//
|
int32_t p = (port != 0)? port:tsServerPort;
|
||||||
//}
|
|
||||||
|
tscDebug("try to connect to %s:%u, user:%s db:%s", ip, p, user, db);
|
||||||
|
if (user == NULL) {
|
||||||
|
user = TSDB_DEFAULT_USER;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pass == NULL) {
|
||||||
|
pass = TSDB_DEFAULT_PASS;
|
||||||
|
}
|
||||||
|
|
||||||
|
return taos_connect_internal(ip, user, pass, NULL, db, p);
|
||||||
|
}
|
||||||
|
|
||||||
|
TAOS *taos_connect_auth(const char *ip, const char *user, const char *auth, const char *db, uint16_t port) {
|
||||||
|
tscDebug("try to connect to %s:%u by auth, user:%s db:%s", ip, port, user, db);
|
||||||
|
if (user == NULL) {
|
||||||
|
user = TSDB_DEFAULT_USER;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (auth == NULL) {
|
||||||
|
tscError("No auth info is given, failed to connect to server");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
return taos_connect_internal(ip, user, NULL, auth, db, port);
|
||||||
|
}
|
||||||
|
|
||||||
|
TAOS *taos_connect_l(const char *ip, int ipLen, const char *user, int userLen, const char *pass, int passLen, const char *db, int dbLen, uint16_t port) {
|
||||||
|
char ipStr[TSDB_EP_LEN] = {0};
|
||||||
|
char dbStr[TSDB_DB_NAME_LEN] = {0};
|
||||||
|
char userStr[TSDB_USER_LEN] = {0};
|
||||||
|
char passStr[TSDB_PASSWORD_LEN] = {0};
|
||||||
|
|
||||||
|
strncpy(ipStr, ip, MIN(TSDB_EP_LEN - 1, ipLen));
|
||||||
|
strncpy(userStr, user, MIN(TSDB_USER_LEN - 1, userLen));
|
||||||
|
strncpy(passStr, pass, MIN(TSDB_PASSWORD_LEN - 1, passLen));
|
||||||
|
strncpy(dbStr, db, MIN(TSDB_DB_NAME_LEN - 1, dbLen));
|
||||||
|
return taos_connect(ipStr, userStr, passStr, dbStr, port);
|
||||||
|
}
|
||||||
|
|
||||||
int taos_init() { return 0; }
|
|
||||||
void taos_cleanup(void) {}
|
|
||||||
|
|
|
@ -0,0 +1,283 @@
|
||||||
|
#include <tpagedfile.h>
|
||||||
|
#include "clientInt.h"
|
||||||
|
#include "tdef.h"
|
||||||
|
#include "tep.h"
|
||||||
|
#include "tglobal.h"
|
||||||
|
#include "tmsgtype.h"
|
||||||
|
#include "tref.h"
|
||||||
|
#include "tscLog.h"
|
||||||
|
|
||||||
|
static int32_t initEpSetFromCfg(const char *firstEp, const char *secondEp, SCorEpSet *pEpSet);
|
||||||
|
static int32_t buildConnectMsg(SRequestObj *pRequest, SRequestMsgBody* pMsgBody);
|
||||||
|
static void destroyConnectMsg(SRequestMsgBody* pMsgBody);
|
||||||
|
|
||||||
|
static int32_t sendMsgToServer(void *pTransporter, SEpSet* epSet, const SRequestMsgBody *pBody, int64_t* pTransporterId);
|
||||||
|
|
||||||
|
static bool stringLengthCheck(const char* str, size_t maxsize) {
|
||||||
|
if (str == NULL) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t len = strlen(str);
|
||||||
|
if (len <= 0 || len > maxsize) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
static bool validateUserName(const char* user) {
|
||||||
|
return stringLengthCheck(user, TSDB_USER_LEN - 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
static bool validatePassword(const char* passwd) {
|
||||||
|
return stringLengthCheck(passwd, TSDB_PASSWORD_LEN - 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
static bool validateDbName(const char* db) {
|
||||||
|
return stringLengthCheck(db, TSDB_DB_NAME_LEN - 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
static char* getClusterKey(const char* user, const char* auth, const char* ip, int32_t port) {
|
||||||
|
char key[512] = {0};
|
||||||
|
snprintf(key, sizeof(key), "%s:%s:%s:%d", user, auth, ip, port);
|
||||||
|
return strdup(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
static STscObj* taosConnectImpl(const char *ip, const char *user, const char *auth, const char *db, uint16_t port, __taos_async_fn_t fp, void *param, SAppInstInfo* pAppInfo);
|
||||||
|
|
||||||
|
TAOS *taos_connect_internal(const char *ip, const char *user, const char *pass, const char *auth, const char *db, uint16_t port) {
|
||||||
|
if (taos_init() != TSDB_CODE_SUCCESS) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!validateUserName(user)) {
|
||||||
|
terrno = TSDB_CODE_TSC_INVALID_USER_LENGTH;
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
char tmp[TSDB_DB_NAME_LEN] = {0};
|
||||||
|
if (db != NULL) {
|
||||||
|
if(!validateDbName(db)) {
|
||||||
|
terrno = TSDB_CODE_TSC_INVALID_DB_LENGTH;
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
tstrncpy(tmp, db, sizeof(tmp));
|
||||||
|
strdequote(tmp);
|
||||||
|
}
|
||||||
|
|
||||||
|
char secretEncrypt[32] = {0};
|
||||||
|
if (auth == NULL) {
|
||||||
|
if (!validatePassword(pass)) {
|
||||||
|
terrno = TSDB_CODE_TSC_INVALID_PASS_LENGTH;
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
taosEncryptPass_c((uint8_t *)pass, strlen(pass), secretEncrypt);
|
||||||
|
} else {
|
||||||
|
tstrncpy(secretEncrypt, auth, tListLen(secretEncrypt));
|
||||||
|
}
|
||||||
|
|
||||||
|
SCorEpSet epSet = {0};
|
||||||
|
if (ip) {
|
||||||
|
if (initEpSetFromCfg(ip, NULL, &epSet) < 0) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (port) {
|
||||||
|
epSet.epSet.port[0] = port;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if (initEpSetFromCfg(tsFirst, tsSecond, &epSet) < 0) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
char* key = getClusterKey(user, secretEncrypt, ip, port);
|
||||||
|
|
||||||
|
SAppInstInfo* pInst = taosHashGet(appInfo.pInstMap, key, strlen(key));
|
||||||
|
if (pInst == NULL) {
|
||||||
|
pInst = calloc(1, sizeof(struct SAppInstInfo));
|
||||||
|
|
||||||
|
pInst->mgmtEp = epSet;
|
||||||
|
pInst->pTransporter = openTransporter(user, secretEncrypt);
|
||||||
|
|
||||||
|
taosHashPut(appInfo.pInstMap, key, strlen(key), &pInst, POINTER_BYTES);
|
||||||
|
}
|
||||||
|
|
||||||
|
return taosConnectImpl(ip, user, &secretEncrypt[0], db, port, NULL, NULL, pInst);
|
||||||
|
}
|
||||||
|
|
||||||
|
int initEpSetFromCfg(const char *firstEp, const char *secondEp, SCorEpSet *pEpSet) {
|
||||||
|
pEpSet->version = 0;
|
||||||
|
|
||||||
|
// init mgmt ip set
|
||||||
|
SEpSet *mgmtEpSet = &(pEpSet->epSet);
|
||||||
|
mgmtEpSet->numOfEps = 0;
|
||||||
|
mgmtEpSet->inUse = 0;
|
||||||
|
|
||||||
|
if (firstEp && firstEp[0] != 0) {
|
||||||
|
if (strlen(firstEp) >= TSDB_EP_LEN) {
|
||||||
|
terrno = TSDB_CODE_TSC_INVALID_FQDN;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
taosGetFqdnPortFromEp(firstEp, mgmtEpSet->fqdn[0], &(mgmtEpSet->port[0]));
|
||||||
|
mgmtEpSet->numOfEps++;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (secondEp && secondEp[0] != 0) {
|
||||||
|
if (strlen(secondEp) >= TSDB_EP_LEN) {
|
||||||
|
terrno = TSDB_CODE_TSC_INVALID_FQDN;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
taosGetFqdnPortFromEp(secondEp, mgmtEpSet->fqdn[mgmtEpSet->numOfEps], &(mgmtEpSet->port[mgmtEpSet->numOfEps]));
|
||||||
|
mgmtEpSet->numOfEps++;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (mgmtEpSet->numOfEps == 0) {
|
||||||
|
terrno = TSDB_CODE_TSC_INVALID_FQDN;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
STscObj* taosConnectImpl(const char *ip, const char *user, const char *auth, const char *db, uint16_t port, __taos_async_fn_t fp, void *param, SAppInstInfo* pAppInfo) {
|
||||||
|
STscObj *pTscObj = createTscObj(user, auth, ip, port, pAppInfo);
|
||||||
|
if (NULL == pTscObj) {
|
||||||
|
terrno = TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
return pTscObj;
|
||||||
|
}
|
||||||
|
|
||||||
|
SRequestObj *pRequest = createRequest(pTscObj, fp, param, TSDB_SQL_CONNECT);
|
||||||
|
if (pRequest == NULL) {
|
||||||
|
destroyTscObj(pTscObj);
|
||||||
|
terrno = TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
SRequestMsgBody body = {0};
|
||||||
|
buildConnectMsg(pRequest, &body);
|
||||||
|
|
||||||
|
int64_t transporterId = 0;
|
||||||
|
sendMsgToServer(pTscObj->pTransporter, &pTscObj->pAppInfo->mgmtEp.epSet, &body, &transporterId);
|
||||||
|
|
||||||
|
tsem_wait(&pRequest->body.rspSem);
|
||||||
|
destroyConnectMsg(&body);
|
||||||
|
|
||||||
|
if (pRequest->code != TSDB_CODE_SUCCESS) {
|
||||||
|
const char *errorMsg = (pRequest->code == TSDB_CODE_RPC_FQDN_ERROR) ? taos_errstr(pRequest) : tstrerror(terrno);
|
||||||
|
printf("failed to connect to server, reason: %s\n\n", errorMsg);
|
||||||
|
|
||||||
|
destroyRequest(pRequest);
|
||||||
|
taos_close(pTscObj);
|
||||||
|
pTscObj = NULL;
|
||||||
|
} else {
|
||||||
|
tscDebug("0x%"PRIx64" connection is opening, connId:%d, dnodeConn:%p", pTscObj->id, pTscObj->connId, pTscObj->pTransporter);
|
||||||
|
destroyRequest(pRequest);
|
||||||
|
}
|
||||||
|
|
||||||
|
return pTscObj;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t buildConnectMsg(SRequestObj *pRequest, SRequestMsgBody* pMsgBody) {
|
||||||
|
pMsgBody->msgType = TSDB_MSG_TYPE_CONNECT;
|
||||||
|
pMsgBody->msgLen = sizeof(SConnectMsg);
|
||||||
|
pMsgBody->requestObjRefId = pRequest->self;
|
||||||
|
|
||||||
|
SConnectMsg *pConnect = calloc(1, sizeof(SConnectMsg));
|
||||||
|
if (pConnect == NULL) {
|
||||||
|
terrno = TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO refactor full_name
|
||||||
|
char *db; // ugly code to move the space
|
||||||
|
|
||||||
|
STscObj *pObj = pRequest->pTscObj;
|
||||||
|
pthread_mutex_lock(&pObj->mutex);
|
||||||
|
db = strstr(pObj->db, TS_PATH_DELIMITER);
|
||||||
|
|
||||||
|
db = (db == NULL) ? pObj->db : db + 1;
|
||||||
|
tstrncpy(pConnect->db, db, sizeof(pConnect->db));
|
||||||
|
pthread_mutex_unlock(&pObj->mutex);
|
||||||
|
|
||||||
|
pConnect->pid = htonl(appInfo.pid);
|
||||||
|
pConnect->startTime = htobe64(appInfo.startTime);
|
||||||
|
tstrncpy(pConnect->app, appInfo.appName, tListLen(pConnect->app));
|
||||||
|
|
||||||
|
pMsgBody->pData = pConnect;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void destroyConnectMsg(SRequestMsgBody* pMsgBody) {
|
||||||
|
assert(pMsgBody != NULL);
|
||||||
|
tfree(pMsgBody->pData);
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t sendMsgToServer(void *pTransporter, SEpSet* epSet, const SRequestMsgBody *pBody, int64_t* pTransporterId) {
|
||||||
|
char *pMsg = rpcMallocCont(pBody->msgLen);
|
||||||
|
if (NULL == pMsg) {
|
||||||
|
tscError("0x%"PRIx64" msg:%s malloc failed", pBody->requestId, taosMsg[pBody->msgType]);
|
||||||
|
terrno = TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
memcpy(pMsg, pBody->pData, pBody->msgLen);
|
||||||
|
SRpcMsg rpcMsg = {
|
||||||
|
.msgType = pBody->msgType,
|
||||||
|
.pCont = pMsg,
|
||||||
|
.contLen = pBody->msgLen,
|
||||||
|
.ahandle = (void*) pBody->requestObjRefId,
|
||||||
|
.handle = NULL,
|
||||||
|
.code = 0
|
||||||
|
};
|
||||||
|
|
||||||
|
rpcSendRequest(pTransporter, epSet, &rpcMsg, pTransporterId);
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
void processMsgFromServer(void* parent, SRpcMsg* pMsg, SEpSet* pEpSet) {
|
||||||
|
int64_t requestRefId = (int64_t)pMsg->ahandle;
|
||||||
|
|
||||||
|
SRequestObj *pRequest = (SRequestObj *)taosAcquireRef(tscReqRef, requestRefId);
|
||||||
|
if (pRequest == NULL) {
|
||||||
|
rpcFreeCont(pMsg->pCont);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
assert(pRequest->self == requestRefId);
|
||||||
|
pRequest->metric.rsp = taosGetTimestampMs();
|
||||||
|
|
||||||
|
pRequest->code = pMsg->code;
|
||||||
|
|
||||||
|
STscObj *pTscObj = pRequest->pTscObj;
|
||||||
|
if (pEpSet) {
|
||||||
|
if (!isEpsetEqual(&pTscObj->pAppInfo->mgmtEp.epSet, pEpSet)) {
|
||||||
|
updateEpSet_s(&pTscObj->pAppInfo->mgmtEp, pEpSet);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* There is not response callback function for submit response.
|
||||||
|
* The actual inserted number of points is the first number.
|
||||||
|
*/
|
||||||
|
if (pMsg->code == TSDB_CODE_SUCCESS) {
|
||||||
|
tscDebug("0x%" PRIx64 " message:%s, code:%s rspLen:%d, elapsed:%"PRId64 " ms", pRequest->requestId, taosMsg[pMsg->msgType],
|
||||||
|
tstrerror(pMsg->code), pMsg->contLen, pRequest->metric.rsp - pRequest->metric.start);
|
||||||
|
if (handleRequestRspFp[pRequest->type]) {
|
||||||
|
pMsg->code = (*handleRequestRspFp[pRequest->type])(pRequest, pMsg->pCont, pMsg->contLen);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
tscError("0x%" PRIx64 " SQL cmd:%s, code:%s rspLen:%d", pRequest->requestId, taosMsg[pMsg->msgType],
|
||||||
|
tstrerror(pMsg->code), pMsg->contLen);
|
||||||
|
}
|
||||||
|
|
||||||
|
taosReleaseRef(tscReqRef, requestRefId);
|
||||||
|
rpcFreeCont(pMsg->pCont);
|
||||||
|
|
||||||
|
sem_post(&pRequest->body.rspSem);
|
||||||
|
}
|
File diff suppressed because it is too large
Load Diff
|
@ -0,0 +1,85 @@
|
||||||
|
#include "clientInt.h"
|
||||||
|
#include "trpc.h"
|
||||||
|
#include "os.h"
|
||||||
|
#include "taosmsg.h"
|
||||||
|
#include "tcache.h"
|
||||||
|
#include "tconfig.h"
|
||||||
|
#include "tglobal.h"
|
||||||
|
#include "tnote.h"
|
||||||
|
#include "tref.h"
|
||||||
|
#include "tscLog.h"
|
||||||
|
#include "tsched.h"
|
||||||
|
#include "ttime.h"
|
||||||
|
#include "ttimezone.h"
|
||||||
|
|
||||||
|
#define TSC_VAR_NOT_RELEASE 1
|
||||||
|
#define TSC_VAR_RELEASED 0
|
||||||
|
|
||||||
|
static int32_t sentinel = TSC_VAR_NOT_RELEASE;
|
||||||
|
static pthread_once_t tscinit = PTHREAD_ONCE_INIT;
|
||||||
|
|
||||||
|
extern int32_t tscInitRes;
|
||||||
|
|
||||||
|
int taos_options(TSDB_OPTION option, const void *arg, ...) {
|
||||||
|
static int32_t lock = 0;
|
||||||
|
|
||||||
|
for (int i = 1; atomic_val_compare_exchange_32(&lock, 0, 1) != 0; ++i) {
|
||||||
|
if (i % 1000 == 0) {
|
||||||
|
tscInfo("haven't acquire lock after spin %d times.", i);
|
||||||
|
sched_yield();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int ret = taos_options_imp(option, (const char*)arg);
|
||||||
|
|
||||||
|
atomic_store_32(&lock, 0);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int taos_init() {
|
||||||
|
pthread_once(&tscinit, taos_init_imp);
|
||||||
|
return tscInitRes;
|
||||||
|
}
|
||||||
|
|
||||||
|
// this function may be called by user or system, or by both simultaneously.
|
||||||
|
void taos_cleanup(void) {
|
||||||
|
tscDebug("start to cleanup client environment");
|
||||||
|
|
||||||
|
if (atomic_val_compare_exchange_32(&sentinel, TSC_VAR_NOT_RELEASE, TSC_VAR_RELEASED) != TSC_VAR_NOT_RELEASE) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t id = tscReqRef;
|
||||||
|
tscReqRef = -1;
|
||||||
|
taosCloseRef(id);
|
||||||
|
|
||||||
|
void* p = tscQhandle;
|
||||||
|
tscQhandle = NULL;
|
||||||
|
taosCleanUpScheduler(p);
|
||||||
|
|
||||||
|
id = tscConnRef;
|
||||||
|
tscConnRef = -1;
|
||||||
|
taosCloseRef(id);
|
||||||
|
|
||||||
|
rpcCleanup();
|
||||||
|
taosCloseLog();
|
||||||
|
}
|
||||||
|
|
||||||
|
void taos_close(TAOS* taos) {
|
||||||
|
if (taos == NULL) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
STscObj *pTscObj = (STscObj *)taos;
|
||||||
|
tscDebug("0x%"PRIx64" try to close connection, numOfReq:%d", pTscObj->id, pTscObj->numOfReqs);
|
||||||
|
|
||||||
|
taosRemoveRef(tscConnRef, pTscObj->id);
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *taos_errstr(TAOS_RES *res) {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
void taos_free_result(TAOS_RES *res) {
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,471 @@
|
||||||
|
/*
|
||||||
|
* 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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "os.h"
|
||||||
|
#include "taosmsg.h"
|
||||||
|
#include "tcache.h"
|
||||||
|
#include "tconfig.h"
|
||||||
|
#include "tglobal.h"
|
||||||
|
#include "tnote.h"
|
||||||
|
#include "tref.h"
|
||||||
|
#include "tscLog.h"
|
||||||
|
#include "tsched.h"
|
||||||
|
#include "ttime.h"
|
||||||
|
#include "trpc.h"
|
||||||
|
#include "ttimezone.h"
|
||||||
|
#include "clientInt.h"
|
||||||
|
|
||||||
|
#define TSC_VAR_NOT_RELEASE 1
|
||||||
|
#define TSC_VAR_RELEASED 0
|
||||||
|
|
||||||
|
SAppInfo appInfo;
|
||||||
|
int32_t tscReqRef = -1;
|
||||||
|
int32_t tscConnRef = -1;
|
||||||
|
void *tscQhandle = NULL;
|
||||||
|
|
||||||
|
int32_t tsNumOfThreads = 1;
|
||||||
|
volatile int32_t tscInitRes = 0;
|
||||||
|
|
||||||
|
static void registerRequest(SRequestObj* pRequest) {
|
||||||
|
STscObj *pTscObj = (STscObj *)taosAcquireRef(tscConnRef, pRequest->pTscObj->id);
|
||||||
|
assert(pTscObj != NULL);
|
||||||
|
|
||||||
|
// connection has been released already, abort creating request.
|
||||||
|
pRequest->self = taosAddRef(tscReqRef, pRequest);
|
||||||
|
|
||||||
|
int32_t num = atomic_add_fetch_32(&pTscObj->numOfReqs, 1);
|
||||||
|
|
||||||
|
if (pTscObj->pAppInfo) {
|
||||||
|
SInstanceActivity *pActivity = &pTscObj->pAppInfo->summary;
|
||||||
|
|
||||||
|
int32_t total = atomic_add_fetch_32(&pActivity->totalRequests, 1);
|
||||||
|
int32_t currentInst = atomic_add_fetch_32(&pActivity->currentRequests, 1);
|
||||||
|
tscDebug("0x%" PRIx64 " new Request from connObj:0x%" PRIx64 ", current:%d, app current:%d, total:%d", pRequest->self,
|
||||||
|
pRequest->pTscObj->id, num, currentInst, total);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void deregisterRequest(SRequestObj* pRequest) {
|
||||||
|
assert(pRequest != NULL);
|
||||||
|
|
||||||
|
STscObj* pTscObj = pRequest->pTscObj;
|
||||||
|
SInstanceActivity* pActivity = &pTscObj->pAppInfo->summary;
|
||||||
|
|
||||||
|
int32_t currentInst = atomic_sub_fetch_32(&pActivity->currentRequests, 1);
|
||||||
|
int32_t num = atomic_sub_fetch_32(&pTscObj->numOfReqs, 1);
|
||||||
|
|
||||||
|
tscDebug("0x%"PRIx64" free Request from connObj: 0x%"PRIx64", current:%d, app current:%d", pRequest->self, pTscObj->id, num, currentInst);
|
||||||
|
taosReleaseRef(tscConnRef, pTscObj->id);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void tscInitLogFile() {
|
||||||
|
taosReadGlobalLogCfg();
|
||||||
|
if (mkdir(tsLogDir, 0755) != 0 && errno != EEXIST) {
|
||||||
|
printf("failed to create log dir:%s\n", tsLogDir);
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *defaultLogFileNamePrefix = "taoslog";
|
||||||
|
const int32_t maxLogFileNum = 10;
|
||||||
|
|
||||||
|
char temp[128] = {0};
|
||||||
|
sprintf(temp, "%s/%s", tsLogDir, defaultLogFileNamePrefix);
|
||||||
|
if (taosInitLog(temp, tsNumOfLogLines, maxLogFileNum) < 0) {
|
||||||
|
printf("failed to open log file in directory:%s\n", tsLogDir);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void closeTransporter(STscObj* pTscObj) {
|
||||||
|
if (pTscObj == NULL || pTscObj->pTransporter == NULL) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
tscDebug("free transporter:%p in connObj: 0x%"PRIx64, pTscObj->pTransporter, pTscObj->id);
|
||||||
|
rpcClose(pTscObj->pTransporter);
|
||||||
|
pTscObj->pTransporter = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO refactor
|
||||||
|
void* openTransporter(const char *user, const char *auth) {
|
||||||
|
SRpcInit rpcInit;
|
||||||
|
memset(&rpcInit, 0, sizeof(rpcInit));
|
||||||
|
rpcInit.localPort = 0;
|
||||||
|
rpcInit.label = "TSC";
|
||||||
|
rpcInit.numOfThreads = tsNumOfThreads;
|
||||||
|
rpcInit.cfp = processMsgFromServer;
|
||||||
|
rpcInit.sessions = tsMaxConnections;
|
||||||
|
rpcInit.connType = TAOS_CONN_CLIENT;
|
||||||
|
rpcInit.user = (char *)user;
|
||||||
|
rpcInit.idleTime = tsShellActivityTimer * 1000;
|
||||||
|
rpcInit.ckey = "key";
|
||||||
|
// rpcInit.spi = 1;
|
||||||
|
rpcInit.secret = (char *)auth;
|
||||||
|
|
||||||
|
void* pDnodeConn = rpcOpen(&rpcInit);
|
||||||
|
if (pDnodeConn == NULL) {
|
||||||
|
tscError("failed to init connection to server");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
return pDnodeConn;
|
||||||
|
}
|
||||||
|
|
||||||
|
void destroyTscObj(void *pObj) {
|
||||||
|
STscObj *pTscObj = pObj;
|
||||||
|
|
||||||
|
atomic_sub_fetch_64(&pTscObj->pAppInfo->numOfConns, 1);
|
||||||
|
tscDebug("connObj 0x%"PRIx64" destroyed, totalConn:%"PRId64, pTscObj->id, pTscObj->pAppInfo->numOfConns);
|
||||||
|
|
||||||
|
closeTransporter(pTscObj);
|
||||||
|
pthread_mutex_destroy(&pTscObj->mutex);
|
||||||
|
tfree(pTscObj);
|
||||||
|
}
|
||||||
|
|
||||||
|
void* createTscObj(const char* user, const char* auth, const char *ip, uint32_t port, SAppInstInfo* pAppInfo) {
|
||||||
|
STscObj *pObj = (STscObj *)calloc(1, sizeof(STscObj));
|
||||||
|
if (NULL == pObj) {
|
||||||
|
terrno = TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
pObj->pAppInfo = pAppInfo;
|
||||||
|
if (pAppInfo != NULL) {
|
||||||
|
pObj->pTransporter = pAppInfo->pTransporter;
|
||||||
|
}
|
||||||
|
|
||||||
|
tstrncpy(pObj->user, user, sizeof(pObj->user));
|
||||||
|
memcpy(pObj->pass, auth, TSDB_PASSWORD_LEN);
|
||||||
|
|
||||||
|
pthread_mutex_init(&pObj->mutex, NULL);
|
||||||
|
pObj->id = taosAddRef(tscConnRef, pObj);
|
||||||
|
|
||||||
|
tscDebug("connObj created, 0x%"PRIx64, pObj->id);
|
||||||
|
return pObj;
|
||||||
|
}
|
||||||
|
|
||||||
|
void* createRequest(STscObj* pObj, __taos_async_fn_t fp, void* param, int32_t type) {
|
||||||
|
assert(pObj != NULL);
|
||||||
|
|
||||||
|
SRequestObj *pRequest = (SRequestObj *)calloc(1, sizeof(SRequestObj));
|
||||||
|
if (NULL == pRequest) {
|
||||||
|
terrno = TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO generated request uuid
|
||||||
|
pRequest->requestId = 0;
|
||||||
|
|
||||||
|
pRequest->metric.start = taosGetTimestampMs();
|
||||||
|
|
||||||
|
pRequest->type = type;
|
||||||
|
pRequest->pTscObj = pObj;
|
||||||
|
pRequest->body.fp = fp;
|
||||||
|
pRequest->body.param = param;
|
||||||
|
tsem_init(&pRequest->body.rspSem, 0, 0);
|
||||||
|
|
||||||
|
registerRequest(pRequest);
|
||||||
|
return pRequest;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void doDestroyRequest(void* p) {
|
||||||
|
assert(p != NULL);
|
||||||
|
SRequestObj* pRequest = (SRequestObj*)p;
|
||||||
|
|
||||||
|
assert(RID_VALID(pRequest->self));
|
||||||
|
|
||||||
|
tfree(pRequest->msgBuf);
|
||||||
|
tfree(pRequest->sqlstr);
|
||||||
|
tfree(pRequest->pInfo);
|
||||||
|
|
||||||
|
deregisterRequest(pRequest);
|
||||||
|
tfree(pRequest);
|
||||||
|
}
|
||||||
|
|
||||||
|
void destroyRequest(SRequestObj* pRequest) {
|
||||||
|
if (pRequest == NULL) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
taosReleaseRef(tscReqRef, pRequest->self);
|
||||||
|
}
|
||||||
|
|
||||||
|
void taos_init_imp(void) {
|
||||||
|
// In the APIs of other program language, taos_cleanup is not available yet.
|
||||||
|
// So, to make sure taos_cleanup will be invoked to clean up the allocated resource to suppress the valgrind warning.
|
||||||
|
atexit(taos_cleanup);
|
||||||
|
|
||||||
|
errno = TSDB_CODE_SUCCESS;
|
||||||
|
srand(taosGetTimestampSec());
|
||||||
|
|
||||||
|
deltaToUtcInitOnce();
|
||||||
|
taosInitGlobalCfg();
|
||||||
|
taosReadCfgFromFile();
|
||||||
|
|
||||||
|
tscInitLogFile();
|
||||||
|
if (taosCheckAndPrintCfg()) {
|
||||||
|
tscInitRes = -1;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
taosInitNotes();
|
||||||
|
initMsgHandleFp();
|
||||||
|
|
||||||
|
rpcInit();
|
||||||
|
|
||||||
|
tscDebug("starting to initialize TAOS driver, local ep: %s", tsLocalEp);
|
||||||
|
|
||||||
|
taosSetCoreDump(true);
|
||||||
|
|
||||||
|
double factor = 4.0;
|
||||||
|
int32_t numOfThreads = MAX((int)(tsNumOfCores * tsNumOfThreadsPerCore / factor), 2);
|
||||||
|
|
||||||
|
int32_t queueSize = tsMaxConnections * 2;
|
||||||
|
tscQhandle = taosInitScheduler(queueSize, numOfThreads, "tsc");
|
||||||
|
if (NULL == tscQhandle) {
|
||||||
|
tscError("failed to init task queue");
|
||||||
|
tscInitRes = -1;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
tscDebug("client task queue is initialized, numOfThreads: %d", numOfThreads);
|
||||||
|
tscConnRef = taosOpenRef(200, destroyTscObj);
|
||||||
|
tscReqRef = taosOpenRef(40960, doDestroyRequest);
|
||||||
|
|
||||||
|
taosGetAppName(appInfo.appName, NULL);
|
||||||
|
appInfo.pid = taosGetPId();
|
||||||
|
appInfo.startTime = taosGetTimestampMs();
|
||||||
|
appInfo.pInstMap = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);
|
||||||
|
|
||||||
|
tscDebug("client is initialized successfully");
|
||||||
|
}
|
||||||
|
|
||||||
|
int taos_options_imp(TSDB_OPTION option, const char *str) {
|
||||||
|
SGlobalCfg *cfg = NULL;
|
||||||
|
|
||||||
|
switch (option) {
|
||||||
|
case TSDB_OPTION_CONFIGDIR:
|
||||||
|
cfg = taosGetConfigOption("configDir");
|
||||||
|
assert(cfg != NULL);
|
||||||
|
|
||||||
|
if (cfg->cfgStatus <= TAOS_CFG_CSTATUS_OPTION) {
|
||||||
|
tstrncpy(configDir, str, TSDB_FILENAME_LEN);
|
||||||
|
cfg->cfgStatus = TAOS_CFG_CSTATUS_OPTION;
|
||||||
|
tscInfo("set config file directory:%s", str);
|
||||||
|
} else {
|
||||||
|
tscWarn("config option:%s, input value:%s, is configured by %s, use %s", cfg->option, str, tsCfgStatusStr[cfg->cfgStatus], (char *)cfg->ptr);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
case TSDB_OPTION_SHELL_ACTIVITY_TIMER:
|
||||||
|
cfg = taosGetConfigOption("shellActivityTimer");
|
||||||
|
assert(cfg != NULL);
|
||||||
|
|
||||||
|
if (cfg->cfgStatus <= TAOS_CFG_CSTATUS_OPTION) {
|
||||||
|
tsShellActivityTimer = atoi(str);
|
||||||
|
if (tsShellActivityTimer < 1) tsShellActivityTimer = 1;
|
||||||
|
if (tsShellActivityTimer > 3600) tsShellActivityTimer = 3600;
|
||||||
|
cfg->cfgStatus = TAOS_CFG_CSTATUS_OPTION;
|
||||||
|
tscInfo("set shellActivityTimer:%d", tsShellActivityTimer);
|
||||||
|
} else {
|
||||||
|
tscWarn("config option:%s, input value:%s, is configured by %s, use %d", cfg->option, str, tsCfgStatusStr[cfg->cfgStatus], *(int32_t *)cfg->ptr);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
case TSDB_OPTION_LOCALE: { // set locale
|
||||||
|
cfg = taosGetConfigOption("locale");
|
||||||
|
assert(cfg != NULL);
|
||||||
|
|
||||||
|
size_t len = strlen(str);
|
||||||
|
if (len == 0 || len > TSDB_LOCALE_LEN) {
|
||||||
|
tscInfo("Invalid locale:%s, use default", str);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (cfg->cfgStatus <= TAOS_CFG_CSTATUS_OPTION) {
|
||||||
|
char sep = '.';
|
||||||
|
|
||||||
|
if (strlen(tsLocale) == 0) { // locale does not set yet
|
||||||
|
char* defaultLocale = setlocale(LC_CTYPE, "");
|
||||||
|
|
||||||
|
// The locale of the current OS does not be set correctly, so the default locale cannot be acquired.
|
||||||
|
// The launch of current system will abort soon.
|
||||||
|
if (defaultLocale == NULL) {
|
||||||
|
tscError("failed to get default locale, please set the correct locale in current OS");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
tstrncpy(tsLocale, defaultLocale, TSDB_LOCALE_LEN);
|
||||||
|
}
|
||||||
|
|
||||||
|
// set the user specified locale
|
||||||
|
char *locale = setlocale(LC_CTYPE, str);
|
||||||
|
|
||||||
|
if (locale != NULL) { // failed to set the user specified locale
|
||||||
|
tscInfo("locale set, prev locale:%s, new locale:%s", tsLocale, locale);
|
||||||
|
cfg->cfgStatus = TAOS_CFG_CSTATUS_OPTION;
|
||||||
|
} else { // set the user specified locale failed, use default LC_CTYPE as current locale
|
||||||
|
locale = setlocale(LC_CTYPE, tsLocale);
|
||||||
|
tscInfo("failed to set locale:%s, current locale:%s", str, tsLocale);
|
||||||
|
}
|
||||||
|
|
||||||
|
tstrncpy(tsLocale, locale, TSDB_LOCALE_LEN);
|
||||||
|
|
||||||
|
char *charset = strrchr(tsLocale, sep);
|
||||||
|
if (charset != NULL) {
|
||||||
|
charset += 1;
|
||||||
|
|
||||||
|
charset = taosCharsetReplace(charset);
|
||||||
|
|
||||||
|
if (taosValidateEncodec(charset)) {
|
||||||
|
if (strlen(tsCharset) == 0) {
|
||||||
|
tscInfo("charset set:%s", charset);
|
||||||
|
} else {
|
||||||
|
tscInfo("charset changed from %s to %s", tsCharset, charset);
|
||||||
|
}
|
||||||
|
|
||||||
|
tstrncpy(tsCharset, charset, TSDB_LOCALE_LEN);
|
||||||
|
cfg->cfgStatus = TAOS_CFG_CSTATUS_OPTION;
|
||||||
|
|
||||||
|
} else {
|
||||||
|
tscInfo("charset:%s is not valid in locale, charset remains:%s", charset, tsCharset);
|
||||||
|
}
|
||||||
|
|
||||||
|
free(charset);
|
||||||
|
} else { // it may be windows system
|
||||||
|
tscInfo("charset remains:%s", tsCharset);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
tscWarn("config option:%s, input value:%s, is configured by %s, use %s", cfg->option, str, tsCfgStatusStr[cfg->cfgStatus], (char *)cfg->ptr);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TSDB_OPTION_CHARSET: {
|
||||||
|
/* set charset will override the value of charset, assigned during system locale changed */
|
||||||
|
cfg = taosGetConfigOption("charset");
|
||||||
|
assert(cfg != NULL);
|
||||||
|
|
||||||
|
size_t len = strlen(str);
|
||||||
|
if (len == 0 || len > TSDB_LOCALE_LEN) {
|
||||||
|
tscInfo("failed to set charset:%s", str);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (cfg->cfgStatus <= TAOS_CFG_CSTATUS_OPTION) {
|
||||||
|
if (taosValidateEncodec(str)) {
|
||||||
|
if (strlen(tsCharset) == 0) {
|
||||||
|
tscInfo("charset is set:%s", str);
|
||||||
|
} else {
|
||||||
|
tscInfo("charset changed from %s to %s", tsCharset, str);
|
||||||
|
}
|
||||||
|
|
||||||
|
tstrncpy(tsCharset, str, TSDB_LOCALE_LEN);
|
||||||
|
cfg->cfgStatus = TAOS_CFG_CSTATUS_OPTION;
|
||||||
|
} else {
|
||||||
|
tscInfo("charset:%s not valid", str);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
tscWarn("config option:%s, input value:%s, is configured by %s, use %s", cfg->option, str, tsCfgStatusStr[cfg->cfgStatus], (char *)cfg->ptr);
|
||||||
|
}
|
||||||
|
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TSDB_OPTION_TIMEZONE:
|
||||||
|
cfg = taosGetConfigOption("timezone");
|
||||||
|
assert(cfg != NULL);
|
||||||
|
|
||||||
|
if (cfg->cfgStatus <= TAOS_CFG_CSTATUS_OPTION) {
|
||||||
|
tstrncpy(tsTimezone, str, TSDB_TIMEZONE_LEN);
|
||||||
|
tsSetTimeZone();
|
||||||
|
cfg->cfgStatus = TAOS_CFG_CSTATUS_OPTION;
|
||||||
|
tscDebug("timezone set:%s, input:%s by taos_options", tsTimezone, str);
|
||||||
|
} else {
|
||||||
|
tscWarn("config option:%s, input value:%s, is configured by %s, use %s", cfg->option, str, tsCfgStatusStr[cfg->cfgStatus], (char *)cfg->ptr);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
// TODO return the correct error code to client in the format for taos_errstr()
|
||||||
|
tscError("Invalid option %d", option);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
#include "cJSON.h"
|
||||||
|
static setConfRet taos_set_config_imp(const char *config){
|
||||||
|
setConfRet ret = {SET_CONF_RET_SUCC, {0}};
|
||||||
|
static bool setConfFlag = false;
|
||||||
|
if (setConfFlag) {
|
||||||
|
ret.retCode = SET_CONF_RET_ERR_ONLY_ONCE;
|
||||||
|
strcpy(ret.retMsg, "configuration can only set once");
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
taosInitGlobalCfg();
|
||||||
|
cJSON *root = cJSON_Parse(config);
|
||||||
|
if (root == NULL){
|
||||||
|
ret.retCode = SET_CONF_RET_ERR_JSON_PARSE;
|
||||||
|
strcpy(ret.retMsg, "parse json error");
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int size = cJSON_GetArraySize(root);
|
||||||
|
if(!cJSON_IsObject(root) || size == 0) {
|
||||||
|
ret.retCode = SET_CONF_RET_ERR_JSON_INVALID;
|
||||||
|
strcpy(ret.retMsg, "json content is invalid, must be not empty object");
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(size >= 1000) {
|
||||||
|
ret.retCode = SET_CONF_RET_ERR_TOO_LONG;
|
||||||
|
strcpy(ret.retMsg, "json object size is too long");
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
for(int i = 0; i < size; i++){
|
||||||
|
cJSON *item = cJSON_GetArrayItem(root, i);
|
||||||
|
if(!item) {
|
||||||
|
ret.retCode = SET_CONF_RET_ERR_INNER;
|
||||||
|
strcpy(ret.retMsg, "inner error");
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
if(!taosReadConfigOption(item->string, item->valuestring, NULL, NULL, TAOS_CFG_CSTATUS_OPTION, TSDB_CFG_CTYPE_B_CLIENT)){
|
||||||
|
ret.retCode = SET_CONF_RET_ERR_PART;
|
||||||
|
if (strlen(ret.retMsg) == 0){
|
||||||
|
snprintf(ret.retMsg, RET_MSG_LENGTH, "part error|%s", item->string);
|
||||||
|
}else{
|
||||||
|
int tmp = RET_MSG_LENGTH - 1 - (int)strlen(ret.retMsg);
|
||||||
|
size_t leftSize = tmp >= 0 ? tmp : 0;
|
||||||
|
strncat(ret.retMsg, "|", leftSize);
|
||||||
|
tmp = RET_MSG_LENGTH - 1 - (int)strlen(ret.retMsg);
|
||||||
|
leftSize = tmp >= 0 ? tmp : 0;
|
||||||
|
strncat(ret.retMsg, item->string, leftSize);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
cJSON_Delete(root);
|
||||||
|
setConfFlag = true;
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
setConfRet taos_set_config(const char *config){
|
||||||
|
pthread_mutex_lock(&setConfMutex);
|
||||||
|
setConfRet ret = taos_set_config_imp(config);
|
||||||
|
pthread_mutex_unlock(&setConfMutex);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
#endif
|
|
@ -0,0 +1,18 @@
|
||||||
|
|
||||||
|
MESSAGE(STATUS "build parser unit test")
|
||||||
|
|
||||||
|
# GoogleTest requires at least C++11
|
||||||
|
SET(CMAKE_CXX_STANDARD 11)
|
||||||
|
AUX_SOURCE_DIRECTORY(${CMAKE_CURRENT_SOURCE_DIR} SOURCE_LIST)
|
||||||
|
|
||||||
|
ADD_EXECUTABLE(clientTest ${SOURCE_LIST})
|
||||||
|
TARGET_LINK_LIBRARIES(
|
||||||
|
clientTest
|
||||||
|
PUBLIC os util common transport gtest taos
|
||||||
|
)
|
||||||
|
|
||||||
|
TARGET_INCLUDE_DIRECTORIES(
|
||||||
|
clientTest
|
||||||
|
PUBLIC "${CMAKE_SOURCE_DIR}/include/libs/client/"
|
||||||
|
PRIVATE "${CMAKE_SOURCE_DIR}/source/libs/client/inc"
|
||||||
|
)
|
|
@ -0,0 +1,40 @@
|
||||||
|
/*
|
||||||
|
* 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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <gtest/gtest.h>
|
||||||
|
#include <iostream>
|
||||||
|
#include "tglobal.h"
|
||||||
|
#pragma GCC diagnostic ignored "-Wwrite-strings"
|
||||||
|
|
||||||
|
#pragma GCC diagnostic ignored "-Wunused-function"
|
||||||
|
#pragma GCC diagnostic ignored "-Wunused-variable"
|
||||||
|
#pragma GCC diagnostic ignored "-Wsign-compare"
|
||||||
|
|
||||||
|
#include "taos.h"
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
int main(int argc, char** argv) {
|
||||||
|
testing::InitGoogleTest(&argc, argv);
|
||||||
|
return RUN_ALL_TESTS();
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(testCase, driverInit_Test) {
|
||||||
|
TAOS* pConn = taos_connect("ubuntu", "root", "taosdata", NULL, 0);
|
||||||
|
assert(pConn != NULL);
|
||||||
|
|
||||||
|
taos_close(pConn);
|
||||||
|
}
|
|
@ -20,8 +20,11 @@
|
||||||
extern "C" {
|
extern "C" {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#endif /*_TD_COMMON_INT_H_*/
|
#endif /*_TD_COMMON_INT_H_*/
|
||||||
|
|
|
@ -16,3 +16,5 @@
|
||||||
#define TAOS_MESSAGE_C
|
#define TAOS_MESSAGE_C
|
||||||
|
|
||||||
#include "taosmsg.h"
|
#include "taosmsg.h"
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,41 @@
|
||||||
|
#include "tep.h"
|
||||||
|
#include "tglobal.h"
|
||||||
|
#include "tlockfree.h"
|
||||||
|
|
||||||
|
int taosGetFqdnPortFromEp(const char *ep, char *fqdn, uint16_t *port) {
|
||||||
|
*port = 0;
|
||||||
|
strcpy(fqdn, ep);
|
||||||
|
|
||||||
|
char *temp = strchr(fqdn, ':');
|
||||||
|
if (temp) {
|
||||||
|
*temp = 0;
|
||||||
|
*port = atoi(temp+1);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (*port == 0) {
|
||||||
|
*port = tsServerPort;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool isEpsetEqual(const SEpSet *s1, const SEpSet *s2) {
|
||||||
|
if (s1->numOfEps != s2->numOfEps || s1->inUse != s2->inUse) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int32_t i = 0; i < s1->numOfEps; i++) {
|
||||||
|
if (s1->port[i] != s2->port[i]
|
||||||
|
|| strncmp(s1->fqdn[i], s2->fqdn[i], TSDB_FQDN_LEN) != 0)
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void updateEpSet_s(SCorEpSet *pEpSet, SEpSet *pNewEpSet) {
|
||||||
|
taosCorBeginWrite(&pEpSet->version);
|
||||||
|
pEpSet->epSet = *pNewEpSet;
|
||||||
|
taosCorEndWrite(&pEpSet->version);
|
||||||
|
}
|
||||||
|
|
File diff suppressed because it is too large
Load Diff
|
@ -112,12 +112,12 @@ int dmnReadConfig(const char *path) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (taosReadGlobalCfg() != 0) {
|
if (taosReadCfgFromFile() != 0) {
|
||||||
uError("failed to read global config");
|
uError("failed to read global config");
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (taosCheckGlobalCfg() != 0) {
|
if (taosCheckAndPrintCfg() != 0) {
|
||||||
uError("failed to check global config");
|
uError("failed to check global config");
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
|
@ -17,6 +17,7 @@
|
||||||
#include "dndDnode.h"
|
#include "dndDnode.h"
|
||||||
#include "dndTransport.h"
|
#include "dndTransport.h"
|
||||||
#include "dndVnodes.h"
|
#include "dndVnodes.h"
|
||||||
|
#include "tep.h"
|
||||||
|
|
||||||
int32_t dndGetDnodeId(SDnode *pDnode) {
|
int32_t dndGetDnodeId(SDnode *pDnode) {
|
||||||
SDnodeMgmt *pMgmt = &pDnode->dmgmt;
|
SDnodeMgmt *pMgmt = &pDnode->dmgmt;
|
||||||
|
|
|
@ -72,7 +72,7 @@ static void dndInitMsgFp(STransMgmt *pMgmt) {
|
||||||
pMgmt->msgFp[TSDB_MSG_TYPE_CREATE_STB] = dndProcessMnodeWriteMsg;
|
pMgmt->msgFp[TSDB_MSG_TYPE_CREATE_STB] = dndProcessMnodeWriteMsg;
|
||||||
pMgmt->msgFp[TSDB_MSG_TYPE_ALTER_STB] = dndProcessMnodeWriteMsg;
|
pMgmt->msgFp[TSDB_MSG_TYPE_ALTER_STB] = dndProcessMnodeWriteMsg;
|
||||||
pMgmt->msgFp[TSDB_MSG_TYPE_DROP_STB] = dndProcessMnodeWriteMsg;
|
pMgmt->msgFp[TSDB_MSG_TYPE_DROP_STB] = dndProcessMnodeWriteMsg;
|
||||||
pMgmt->msgFp[TSDB_MSG_TYPE_STB_VGROUP] = dndProcessMnodeReadMsg;
|
pMgmt->msgFp[TSDB_MSG_TYPE_VGROUP_LIST] = dndProcessMnodeReadMsg;
|
||||||
pMgmt->msgFp[TSDB_MSG_TYPE_KILL_QUERY] = dndProcessMnodeWriteMsg;
|
pMgmt->msgFp[TSDB_MSG_TYPE_KILL_QUERY] = dndProcessMnodeWriteMsg;
|
||||||
pMgmt->msgFp[TSDB_MSG_TYPE_KILL_STREAM] = dndProcessMnodeWriteMsg;
|
pMgmt->msgFp[TSDB_MSG_TYPE_KILL_STREAM] = dndProcessMnodeWriteMsg;
|
||||||
pMgmt->msgFp[TSDB_MSG_TYPE_KILL_CONN] = dndProcessMnodeWriteMsg;
|
pMgmt->msgFp[TSDB_MSG_TYPE_KILL_CONN] = dndProcessMnodeWriteMsg;
|
||||||
|
@ -235,18 +235,18 @@ static void dndSendMsgToMnodeRecv(SDnode *pDnode, SRpcMsg *pRpcMsg, SRpcMsg *pRp
|
||||||
static int32_t dndAuthInternalMsg(SDnode *pDnode, char *user, char *spi, char *encrypt, char *secret, char *ckey) {
|
static int32_t dndAuthInternalMsg(SDnode *pDnode, char *user, char *spi, char *encrypt, char *secret, char *ckey) {
|
||||||
if (strcmp(user, INTERNAL_USER) == 0) {
|
if (strcmp(user, INTERNAL_USER) == 0) {
|
||||||
// A simple temporary implementation
|
// A simple temporary implementation
|
||||||
char pass[32] = {0};
|
char pass[TSDB_PASSWORD_LEN] = {0};
|
||||||
taosEncryptPass((uint8_t *)(INTERNAL_SECRET), strlen(INTERNAL_SECRET), pass);
|
taosEncryptPass((uint8_t *)(INTERNAL_SECRET), strlen(INTERNAL_SECRET), pass);
|
||||||
memcpy(secret, pass, TSDB_KEY_LEN);
|
memcpy(secret, pass, TSDB_PASSWORD_LEN);
|
||||||
*spi = 0;
|
*spi = 0;
|
||||||
*encrypt = 0;
|
*encrypt = 0;
|
||||||
*ckey = 0;
|
*ckey = 0;
|
||||||
return 0;
|
return 0;
|
||||||
} else if (strcmp(user, TSDB_NETTEST_USER) == 0) {
|
} else if (strcmp(user, TSDB_NETTEST_USER) == 0) {
|
||||||
// A simple temporary implementation
|
// A simple temporary implementation
|
||||||
char pass[32] = {0};
|
char pass[TSDB_PASSWORD_LEN] = {0};
|
||||||
taosEncryptPass((uint8_t *)(TSDB_NETTEST_USER), strlen(TSDB_NETTEST_USER), pass);
|
taosEncryptPass((uint8_t *)(TSDB_NETTEST_USER), strlen(TSDB_NETTEST_USER), pass);
|
||||||
memcpy(secret, pass, TSDB_KEY_LEN);
|
memcpy(secret, pass, TSDB_PASSWORD_LEN);
|
||||||
*spi = 0;
|
*spi = 0;
|
||||||
*encrypt = 0;
|
*encrypt = 0;
|
||||||
*ckey = 0;
|
*ckey = 0;
|
||||||
|
@ -288,8 +288,8 @@ static int32_t dndRetrieveUserAuthInfo(void *parent, char *user, char *spi, char
|
||||||
dError("user:%s, failed to get user auth from other mnodes since %s", user, terrstr());
|
dError("user:%s, failed to get user auth from other mnodes since %s", user, terrstr());
|
||||||
} else {
|
} else {
|
||||||
SAuthRsp *pRsp = rpcRsp.pCont;
|
SAuthRsp *pRsp = rpcRsp.pCont;
|
||||||
memcpy(secret, pRsp->secret, TSDB_KEY_LEN);
|
memcpy(secret, pRsp->secret, TSDB_PASSWORD_LEN);
|
||||||
memcpy(ckey, pRsp->ckey, TSDB_KEY_LEN);
|
memcpy(ckey, pRsp->ckey, TSDB_PASSWORD_LEN);
|
||||||
*spi = pRsp->spi;
|
*spi = pRsp->spi;
|
||||||
*encrypt = pRsp->encrypt;
|
*encrypt = pRsp->encrypt;
|
||||||
dDebug("user:%s, success to get user auth from other mnodes", user);
|
dDebug("user:%s, success to get user auth from other mnodes", user);
|
||||||
|
@ -368,4 +368,4 @@ void dndSendMsgToMnode(SDnode *pDnode, SRpcMsg *pMsg) {
|
||||||
SEpSet epSet = {0};
|
SEpSet epSet = {0};
|
||||||
dndGetMnodeEpSet(pDnode, &epSet);
|
dndGetMnodeEpSet(pDnode, &epSet);
|
||||||
dndSendMsgToDnode(pDnode, &epSet, pMsg);
|
dndSendMsgToDnode(pDnode, &epSet, pMsg);
|
||||||
}
|
}
|
||||||
|
|
|
@ -120,7 +120,7 @@ SClient* createClient(const char* user, const char* pass, const char* fqdn, uint
|
||||||
SClient* pClient = (SClient*)calloc(1, sizeof(SClient));
|
SClient* pClient = (SClient*)calloc(1, sizeof(SClient));
|
||||||
ASSERT(pClient);
|
ASSERT(pClient);
|
||||||
|
|
||||||
char secretEncrypt[32] = {0};
|
char secretEncrypt[TSDB_PASSWORD_LEN] = {0};
|
||||||
taosEncryptPass((uint8_t*)pass, strlen(pass), secretEncrypt);
|
taosEncryptPass((uint8_t*)pass, strlen(pass), secretEncrypt);
|
||||||
|
|
||||||
SRpcInit rpcInit;
|
SRpcInit rpcInit;
|
||||||
|
|
|
@ -180,13 +180,11 @@ typedef struct SAcctObj {
|
||||||
|
|
||||||
typedef struct SUserObj {
|
typedef struct SUserObj {
|
||||||
char user[TSDB_USER_LEN];
|
char user[TSDB_USER_LEN];
|
||||||
char pass[TSDB_KEY_LEN];
|
char pass[TSDB_PASSWORD_LEN];
|
||||||
char acct[TSDB_USER_LEN];
|
char acct[TSDB_USER_LEN];
|
||||||
int64_t createdTime;
|
int64_t createdTime;
|
||||||
int64_t updateTime;
|
int64_t updateTime;
|
||||||
int8_t superAuth;
|
int8_t superUser;
|
||||||
int8_t readAuth;
|
|
||||||
int8_t writeAuth;
|
|
||||||
int32_t acctId;
|
int32_t acctId;
|
||||||
SHashObj *prohibitDbHash;
|
SHashObj *prohibitDbHash;
|
||||||
} SUserObj;
|
} SUserObj;
|
||||||
|
|
|
@ -29,6 +29,7 @@ typedef struct {
|
||||||
char user[TSDB_USER_LEN];
|
char user[TSDB_USER_LEN];
|
||||||
char app[TSDB_APP_NAME_LEN]; // app name that invokes taosc
|
char app[TSDB_APP_NAME_LEN]; // app name that invokes taosc
|
||||||
int32_t pid; // pid of app that invokes taosc
|
int32_t pid; // pid of app that invokes taosc
|
||||||
|
int64_t appStartTime; // app start time
|
||||||
int32_t id;
|
int32_t id;
|
||||||
int8_t killed;
|
int8_t killed;
|
||||||
int8_t align;
|
int8_t align;
|
||||||
|
@ -44,7 +45,7 @@ typedef struct {
|
||||||
SQueryDesc *pQueries;
|
SQueryDesc *pQueries;
|
||||||
} SConnObj;
|
} SConnObj;
|
||||||
|
|
||||||
static SConnObj *mndCreateConn(SMnode *pMnode, char *user, uint32_t ip, uint16_t port, int32_t pid, const char *app);
|
static SConnObj *mndCreateConn(SMnode *pMnode, char *user, uint32_t ip, uint16_t port, int32_t pid, const char *app, int64_t startTime);
|
||||||
static void mndFreeConn(SConnObj *pConn);
|
static void mndFreeConn(SConnObj *pConn);
|
||||||
static SConnObj *mndAcquireConn(SMnode *pMnode, int32_t connId);
|
static SConnObj *mndAcquireConn(SMnode *pMnode, int32_t connId);
|
||||||
static void mndReleaseConn(SMnode *pMnode, SConnObj *pConn);
|
static void mndReleaseConn(SMnode *pMnode, SConnObj *pConn);
|
||||||
|
@ -102,13 +103,14 @@ void mndCleanupProfile(SMnode *pMnode) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static SConnObj *mndCreateConn(SMnode *pMnode, char *user, uint32_t ip, uint16_t port, int32_t pid, const char *app) {
|
static SConnObj *mndCreateConn(SMnode *pMnode, char *user, uint32_t ip, uint16_t port, int32_t pid, const char *app, int64_t startTime) {
|
||||||
SProfileMgmt *pMgmt = &pMnode->profileMgmt;
|
SProfileMgmt *pMgmt = &pMnode->profileMgmt;
|
||||||
|
|
||||||
int32_t connId = atomic_add_fetch_32(&pMgmt->connId, 1);
|
int32_t connId = atomic_add_fetch_32(&pMgmt->connId, 1);
|
||||||
if (connId == 0) atomic_add_fetch_32(&pMgmt->connId, 1);
|
if (connId == 0) atomic_add_fetch_32(&pMgmt->connId, 1);
|
||||||
|
|
||||||
SConnObj connObj = {.pid = pid,
|
SConnObj connObj = {.pid = pid,
|
||||||
|
.appStartTime = startTime,
|
||||||
.id = connId,
|
.id = connId,
|
||||||
.killed = 0,
|
.killed = 0,
|
||||||
.port = port,
|
.port = port,
|
||||||
|
@ -195,6 +197,7 @@ static int32_t mndProcessConnectMsg(SMnodeMsg *pMsg) {
|
||||||
SMnode *pMnode = pMsg->pMnode;
|
SMnode *pMnode = pMsg->pMnode;
|
||||||
SConnectMsg *pReq = pMsg->rpcMsg.pCont;
|
SConnectMsg *pReq = pMsg->rpcMsg.pCont;
|
||||||
pReq->pid = htonl(pReq->pid);
|
pReq->pid = htonl(pReq->pid);
|
||||||
|
pReq->startTime = htobe64(pReq->startTime);
|
||||||
|
|
||||||
SRpcConnInfo info = {0};
|
SRpcConnInfo info = {0};
|
||||||
if (rpcGetConnInfo(pMsg->rpcMsg.handle, &info) != 0) {
|
if (rpcGetConnInfo(pMsg->rpcMsg.handle, &info) != 0) {
|
||||||
|
@ -216,7 +219,7 @@ static int32_t mndProcessConnectMsg(SMnodeMsg *pMsg) {
|
||||||
mndReleaseDb(pMnode, pDb);
|
mndReleaseDb(pMnode, pDb);
|
||||||
}
|
}
|
||||||
|
|
||||||
SConnObj *pConn = mndCreateConn(pMnode, info.user, info.clientIp, info.clientPort, pReq->pid, pReq->app);
|
SConnObj *pConn = mndCreateConn(pMnode, info.user, info.clientIp, info.clientPort, pReq->pid, pReq->app, pReq->startTime);
|
||||||
if (pConn == NULL) {
|
if (pConn == NULL) {
|
||||||
mError("user:%s, failed to login from %s while create connection since %s", pMsg->user, ip, terrstr());
|
mError("user:%s, failed to login from %s while create connection since %s", pMsg->user, ip, terrstr());
|
||||||
return -1;
|
return -1;
|
||||||
|
@ -233,9 +236,7 @@ static int32_t mndProcessConnectMsg(SMnodeMsg *pMsg) {
|
||||||
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
||||||
if (pUser != NULL) {
|
if (pUser != NULL) {
|
||||||
pRsp->acctId = htonl(pUser->acctId);
|
pRsp->acctId = htonl(pUser->acctId);
|
||||||
pRsp->superAuth = pUser->superAuth;
|
pRsp->superUser = pUser->superUser;
|
||||||
pRsp->readAuth = pUser->readAuth;
|
|
||||||
pRsp->writeAuth = pUser->writeAuth;
|
|
||||||
mndReleaseUser(pMnode, pUser);
|
mndReleaseUser(pMnode, pUser);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -246,7 +247,8 @@ static int32_t mndProcessConnectMsg(SMnodeMsg *pMsg) {
|
||||||
|
|
||||||
pMsg->contLen = sizeof(SConnectRsp);
|
pMsg->contLen = sizeof(SConnectRsp);
|
||||||
pMsg->pCont = pRsp;
|
pMsg->pCont = pRsp;
|
||||||
mDebug("user:%s, login from %s, conn:%d", info.user, ip, pConn->id);
|
|
||||||
|
mDebug("user:%s, login from %s, conn:%d, app:%s", info.user, ip, pConn->id, pReq->app);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -301,7 +303,7 @@ static int32_t mndProcessHeartBeatMsg(SMnodeMsg *pMsg) {
|
||||||
|
|
||||||
SConnObj *pConn = mndAcquireConn(pMnode, pReq->connId);
|
SConnObj *pConn = mndAcquireConn(pMnode, pReq->connId);
|
||||||
if (pConn == NULL) {
|
if (pConn == NULL) {
|
||||||
pConn = mndCreateConn(pMnode, info.user, info.clientIp, info.clientPort, pReq->pid, pReq->app);
|
pConn = mndCreateConn(pMnode, info.user, info.clientIp, info.clientPort, pReq->pid, pReq->app, 0);
|
||||||
if (pConn == NULL) {
|
if (pConn == NULL) {
|
||||||
mError("user:%s, conn:%d is freed and failed to create new conn since %s", pMsg->user, pReq->connId, terrstr());
|
mError("user:%s, conn:%d is freed and failed to create new conn since %s", pMsg->user, pReq->connId, terrstr());
|
||||||
return -1;
|
return -1;
|
||||||
|
@ -368,7 +370,7 @@ static int32_t mndProcessKillQueryMsg(SMnodeMsg *pMsg) {
|
||||||
|
|
||||||
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
||||||
if (pUser == NULL) return 0;
|
if (pUser == NULL) return 0;
|
||||||
if (!pUser->superAuth) {
|
if (!pUser->superUser) {
|
||||||
mndReleaseUser(pMnode, pUser);
|
mndReleaseUser(pMnode, pUser);
|
||||||
terrno = TSDB_CODE_MND_NO_RIGHTS;
|
terrno = TSDB_CODE_MND_NO_RIGHTS;
|
||||||
return -1;
|
return -1;
|
||||||
|
@ -399,7 +401,7 @@ static int32_t mndProcessKillStreamMsg(SMnodeMsg *pMsg) {
|
||||||
|
|
||||||
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
||||||
if (pUser == NULL) return 0;
|
if (pUser == NULL) return 0;
|
||||||
if (!pUser->superAuth) {
|
if (!pUser->superUser) {
|
||||||
mndReleaseUser(pMnode, pUser);
|
mndReleaseUser(pMnode, pUser);
|
||||||
terrno = TSDB_CODE_MND_NO_RIGHTS;
|
terrno = TSDB_CODE_MND_NO_RIGHTS;
|
||||||
return -1;
|
return -1;
|
||||||
|
@ -430,7 +432,7 @@ static int32_t mndProcessKillConnectionMsg(SMnodeMsg *pMsg) {
|
||||||
|
|
||||||
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
||||||
if (pUser == NULL) return 0;
|
if (pUser == NULL) return 0;
|
||||||
if (!pUser->superAuth) {
|
if (!pUser->superUser) {
|
||||||
mndReleaseUser(pMnode, pUser);
|
mndReleaseUser(pMnode, pUser);
|
||||||
terrno = TSDB_CODE_MND_NO_RIGHTS;
|
terrno = TSDB_CODE_MND_NO_RIGHTS;
|
||||||
return -1;
|
return -1;
|
||||||
|
@ -459,7 +461,7 @@ static int32_t mndGetConnsMeta(SMnodeMsg *pMsg, SShowObj *pShow, STableMetaMsg *
|
||||||
|
|
||||||
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
||||||
if (pUser == NULL) return 0;
|
if (pUser == NULL) return 0;
|
||||||
if (!pUser->superAuth) {
|
if (!pUser->superUser) {
|
||||||
mndReleaseUser(pMnode, pUser);
|
mndReleaseUser(pMnode, pUser);
|
||||||
terrno = TSDB_CODE_MND_NO_RIGHTS;
|
terrno = TSDB_CODE_MND_NO_RIGHTS;
|
||||||
return -1;
|
return -1;
|
||||||
|
@ -587,7 +589,7 @@ static int32_t mndGetQueryMeta(SMnodeMsg *pMsg, SShowObj *pShow, STableMetaMsg *
|
||||||
|
|
||||||
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
||||||
if (pUser == NULL) return 0;
|
if (pUser == NULL) return 0;
|
||||||
if (!pUser->superAuth) {
|
if (!pUser->superUser) {
|
||||||
mndReleaseUser(pMnode, pUser);
|
mndReleaseUser(pMnode, pUser);
|
||||||
terrno = TSDB_CODE_MND_NO_RIGHTS;
|
terrno = TSDB_CODE_MND_NO_RIGHTS;
|
||||||
return -1;
|
return -1;
|
||||||
|
@ -803,7 +805,7 @@ static int32_t mndGetStreamMeta(SMnodeMsg *pMsg, SShowObj *pShow, STableMetaMsg
|
||||||
|
|
||||||
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
SUserObj *pUser = mndAcquireUser(pMnode, pMsg->user);
|
||||||
if (pUser == NULL) return 0;
|
if (pUser == NULL) return 0;
|
||||||
if (!pUser->superAuth) {
|
if (!pUser->superUser) {
|
||||||
mndReleaseUser(pMnode, pUser);
|
mndReleaseUser(pMnode, pUser);
|
||||||
terrno = TSDB_CODE_MND_NO_RIGHTS;
|
terrno = TSDB_CODE_MND_NO_RIGHTS;
|
||||||
return -1;
|
return -1;
|
||||||
|
|
|
@ -65,11 +65,9 @@ static int32_t mndCreateDefaultUser(SMnode *pMnode, char *acct, char *user, char
|
||||||
taosEncryptPass((uint8_t *)pass, strlen(pass), userObj.pass);
|
taosEncryptPass((uint8_t *)pass, strlen(pass), userObj.pass);
|
||||||
userObj.createdTime = taosGetTimestampMs();
|
userObj.createdTime = taosGetTimestampMs();
|
||||||
userObj.updateTime = userObj.createdTime;
|
userObj.updateTime = userObj.createdTime;
|
||||||
userObj.readAuth = 1;
|
|
||||||
userObj.writeAuth = 1;
|
|
||||||
|
|
||||||
if (strcmp(user, TSDB_DEFAULT_USER) == 0) {
|
if (strcmp(user, TSDB_DEFAULT_USER) == 0) {
|
||||||
userObj.superAuth = 1;
|
userObj.superUser = 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
SSdbRaw *pRaw = mndUserActionEncode(&userObj);
|
SSdbRaw *pRaw = mndUserActionEncode(&userObj);
|
||||||
|
@ -89,7 +87,7 @@ static int32_t mndCreateDefaultUsers(SMnode *pMnode) {
|
||||||
if (mndCreateDefaultUser(pMnode, TSDB_DEFAULT_USER, "_" TSDB_DEFAULT_USER, TSDB_DEFAULT_PASS) != 0) {
|
if (mndCreateDefaultUser(pMnode, TSDB_DEFAULT_USER, "_" TSDB_DEFAULT_USER, TSDB_DEFAULT_PASS) != 0) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
@ -100,13 +98,11 @@ static SSdbRaw *mndUserActionEncode(SUserObj *pUser) {
|
||||||
|
|
||||||
int32_t dataPos = 0;
|
int32_t dataPos = 0;
|
||||||
SDB_SET_BINARY(pRaw, dataPos, pUser->user, TSDB_USER_LEN)
|
SDB_SET_BINARY(pRaw, dataPos, pUser->user, TSDB_USER_LEN)
|
||||||
SDB_SET_BINARY(pRaw, dataPos, pUser->pass, TSDB_KEY_LEN)
|
SDB_SET_BINARY(pRaw, dataPos, pUser->pass, TSDB_PASSWORD_LEN)
|
||||||
SDB_SET_BINARY(pRaw, dataPos, pUser->acct, TSDB_USER_LEN)
|
SDB_SET_BINARY(pRaw, dataPos, pUser->acct, TSDB_USER_LEN)
|
||||||
SDB_SET_INT64(pRaw, dataPos, pUser->createdTime)
|
SDB_SET_INT64(pRaw, dataPos, pUser->createdTime)
|
||||||
SDB_SET_INT64(pRaw, dataPos, pUser->updateTime)
|
SDB_SET_INT64(pRaw, dataPos, pUser->updateTime)
|
||||||
SDB_SET_INT8(pRaw, dataPos, pUser->superAuth)
|
SDB_SET_INT8(pRaw, dataPos, pUser->superUser)
|
||||||
SDB_SET_INT8(pRaw, dataPos, pUser->readAuth)
|
|
||||||
SDB_SET_INT8(pRaw, dataPos, pUser->writeAuth)
|
|
||||||
SDB_SET_DATALEN(pRaw, dataPos);
|
SDB_SET_DATALEN(pRaw, dataPos);
|
||||||
|
|
||||||
return pRaw;
|
return pRaw;
|
||||||
|
@ -128,13 +124,11 @@ static SSdbRow *mndUserActionDecode(SSdbRaw *pRaw) {
|
||||||
|
|
||||||
int32_t dataPos = 0;
|
int32_t dataPos = 0;
|
||||||
SDB_GET_BINARY(pRaw, pRow, dataPos, pUser->user, TSDB_USER_LEN)
|
SDB_GET_BINARY(pRaw, pRow, dataPos, pUser->user, TSDB_USER_LEN)
|
||||||
SDB_GET_BINARY(pRaw, pRow, dataPos, pUser->pass, TSDB_KEY_LEN)
|
SDB_GET_BINARY(pRaw, pRow, dataPos, pUser->pass, TSDB_PASSWORD_LEN)
|
||||||
SDB_GET_BINARY(pRaw, pRow, dataPos, pUser->acct, TSDB_USER_LEN)
|
SDB_GET_BINARY(pRaw, pRow, dataPos, pUser->acct, TSDB_USER_LEN)
|
||||||
SDB_GET_INT64(pRaw, pRow, dataPos, &pUser->createdTime)
|
SDB_GET_INT64(pRaw, pRow, dataPos, &pUser->createdTime)
|
||||||
SDB_GET_INT64(pRaw, pRow, dataPos, &pUser->updateTime)
|
SDB_GET_INT64(pRaw, pRow, dataPos, &pUser->updateTime)
|
||||||
SDB_GET_INT8(pRaw, pRow, dataPos, &pUser->superAuth)
|
SDB_GET_INT8(pRaw, pRow, dataPos, &pUser->superUser)
|
||||||
SDB_GET_INT8(pRaw, pRow, dataPos, &pUser->readAuth)
|
|
||||||
SDB_GET_INT8(pRaw, pRow, dataPos, &pUser->writeAuth)
|
|
||||||
|
|
||||||
return pRow;
|
return pRow;
|
||||||
}
|
}
|
||||||
|
@ -173,13 +167,11 @@ static int32_t mndUserActionDelete(SSdb *pSdb, SUserObj *pUser) {
|
||||||
static int32_t mndUserActionUpdate(SSdb *pSdb, SUserObj *pOldUser, SUserObj *pNewUser) {
|
static int32_t mndUserActionUpdate(SSdb *pSdb, SUserObj *pOldUser, SUserObj *pNewUser) {
|
||||||
mTrace("user:%s, perform update action", pOldUser->user);
|
mTrace("user:%s, perform update action", pOldUser->user);
|
||||||
memcpy(pOldUser->user, pNewUser->user, TSDB_USER_LEN);
|
memcpy(pOldUser->user, pNewUser->user, TSDB_USER_LEN);
|
||||||
memcpy(pOldUser->pass, pNewUser->pass, TSDB_KEY_LEN);
|
memcpy(pOldUser->pass, pNewUser->pass, TSDB_PASSWORD_LEN);
|
||||||
memcpy(pOldUser->acct, pNewUser->acct, TSDB_USER_LEN);
|
memcpy(pOldUser->acct, pNewUser->acct, TSDB_USER_LEN);
|
||||||
pOldUser->createdTime = pNewUser->createdTime;
|
pOldUser->createdTime = pNewUser->createdTime;
|
||||||
pOldUser->updateTime = pNewUser->updateTime;
|
pOldUser->updateTime = pNewUser->updateTime;
|
||||||
pOldUser->superAuth = pNewUser->superAuth;
|
pOldUser->superUser = pNewUser->superUser;
|
||||||
pOldUser->readAuth = pNewUser->readAuth;
|
|
||||||
pOldUser->writeAuth = pNewUser->writeAuth;
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -200,9 +192,7 @@ static int32_t mndCreateUser(SMnode *pMnode, char *acct, char *user, char *pass,
|
||||||
taosEncryptPass((uint8_t *)pass, strlen(pass), userObj.pass);
|
taosEncryptPass((uint8_t *)pass, strlen(pass), userObj.pass);
|
||||||
userObj.createdTime = taosGetTimestampMs();
|
userObj.createdTime = taosGetTimestampMs();
|
||||||
userObj.updateTime = userObj.createdTime;
|
userObj.updateTime = userObj.createdTime;
|
||||||
userObj.superAuth = 0;
|
userObj.superUser = 0;
|
||||||
userObj.readAuth = 1;
|
|
||||||
userObj.writeAuth = 1;
|
|
||||||
|
|
||||||
STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, pMsg->rpcMsg.handle);
|
STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, pMsg->rpcMsg.handle);
|
||||||
if (pTrans == NULL) {
|
if (pTrans == NULL) {
|
||||||
|
@ -475,7 +465,7 @@ static int32_t mndRetrieveUsers(SMnodeMsg *pMsg, SShowObj *pShow, char *data, in
|
||||||
cols++;
|
cols++;
|
||||||
|
|
||||||
pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
|
pWrite = data + pShow->offset[cols] * rows + pShow->bytes[cols] * numOfRows;
|
||||||
if (pUser->superAuth) {
|
if (pUser->superUser) {
|
||||||
const char *src = "super";
|
const char *src = "super";
|
||||||
STR_WITH_SIZE_TO_VARSTR(pWrite, src, strlen(src));
|
STR_WITH_SIZE_TO_VARSTR(pWrite, src, strlen(src));
|
||||||
} else {
|
} else {
|
||||||
|
|
|
@ -18,6 +18,6 @@ target_link_libraries(
|
||||||
)
|
)
|
||||||
|
|
||||||
# test
|
# test
|
||||||
#if(${BUILD_TEST})
|
if(${BUILD_TEST})
|
||||||
# add_subdirectory(test)
|
add_subdirectory(test)
|
||||||
#endif(${BUILD_TEST})
|
endif(${BUILD_TEST})
|
|
@ -27,7 +27,10 @@ typedef struct SVBufPool SVBufPool;
|
||||||
|
|
||||||
int vnodeOpenBufPool(SVnode *pVnode);
|
int vnodeOpenBufPool(SVnode *pVnode);
|
||||||
void vnodeCloseBufPool(SVnode *pVnode);
|
void vnodeCloseBufPool(SVnode *pVnode);
|
||||||
|
int vnodeBufPoolSwitch(SVnode *pVnode);
|
||||||
|
int vnodeBufPoolRecycle(SVnode *pVnode);
|
||||||
void *vnodeMalloc(SVnode *pVnode, uint64_t size);
|
void *vnodeMalloc(SVnode *pVnode, uint64_t size);
|
||||||
|
bool vnodeBufPoolIsFull(SVnode *pVnode);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
|
|
|
@ -22,8 +22,9 @@
|
||||||
extern "C" {
|
extern "C" {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
bool vnodeShouldCommit(SVnode *pVnode);
|
#define vnodeShouldCommit vnodeBufPoolIsFull
|
||||||
int vnodeAsyncCommit(SVnode *pVnode);
|
int vnodeAsyncCommit(SVnode *pVnode);
|
||||||
|
int vnodeCommit(void *arg);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
|
|
|
@ -18,15 +18,19 @@
|
||||||
|
|
||||||
#include "mallocator.h"
|
#include "mallocator.h"
|
||||||
#include "sync.h"
|
#include "sync.h"
|
||||||
#include "tlockfree.h"
|
|
||||||
#include "wal.h"
|
|
||||||
#include "tcoding.h"
|
#include "tcoding.h"
|
||||||
|
#include "tdlist.h"
|
||||||
|
#include "tlockfree.h"
|
||||||
|
#include "tmacro.h"
|
||||||
|
#include "wal.h"
|
||||||
|
|
||||||
#include "vnode.h"
|
#include "vnode.h"
|
||||||
|
|
||||||
#include "vnodeBufferPool.h"
|
#include "vnodeBufferPool.h"
|
||||||
#include "vnodeCfg.h"
|
#include "vnodeCfg.h"
|
||||||
#include "vnodeCommit.h"
|
#include "vnodeCommit.h"
|
||||||
#include "vnodeFS.h"
|
#include "vnodeFS.h"
|
||||||
|
#include "vnodeMemAllocator.h"
|
||||||
#include "vnodeRequest.h"
|
#include "vnodeRequest.h"
|
||||||
#include "vnodeStateMgr.h"
|
#include "vnodeStateMgr.h"
|
||||||
#include "vnodeSync.h"
|
#include "vnodeSync.h"
|
||||||
|
@ -35,6 +39,27 @@
|
||||||
extern "C" {
|
extern "C" {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
typedef struct SVnodeTask {
|
||||||
|
TD_DLIST_NODE(SVnodeTask);
|
||||||
|
void* arg;
|
||||||
|
int (*execute)(void*);
|
||||||
|
} SVnodeTask;
|
||||||
|
|
||||||
|
typedef struct SVnodeMgr {
|
||||||
|
td_mode_flag_t vnodeInitFlag;
|
||||||
|
td_mode_flag_t vnodeClearFlag;
|
||||||
|
// For commit
|
||||||
|
bool stop;
|
||||||
|
uint16_t nthreads;
|
||||||
|
pthread_t* threads;
|
||||||
|
pthread_mutex_t mutex;
|
||||||
|
pthread_cond_t hasTask;
|
||||||
|
TD_DLIST(SVnodeTask) queue;
|
||||||
|
// For vnode Mgmt
|
||||||
|
} SVnodeMgr;
|
||||||
|
|
||||||
|
extern SVnodeMgr vnodeMgr;
|
||||||
|
|
||||||
struct SVnode {
|
struct SVnode {
|
||||||
char* path;
|
char* path;
|
||||||
SVnodeCfg config;
|
SVnodeCfg config;
|
||||||
|
@ -48,6 +73,8 @@ struct SVnode {
|
||||||
SVnodeFS* pFs;
|
SVnodeFS* pFs;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
int vnodeScheduleTask(SVnodeTask* task);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -16,15 +16,38 @@
|
||||||
#ifndef _TD_VNODE_MEM_ALLOCATOR_H_
|
#ifndef _TD_VNODE_MEM_ALLOCATOR_H_
|
||||||
#define _TD_VNODE_MEM_ALLOCATOR_H_
|
#define _TD_VNODE_MEM_ALLOCATOR_H_
|
||||||
|
|
||||||
#include "mallocator.h"
|
#include "os.h"
|
||||||
#include "vnode.h"
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
extern "C" {
|
extern "C" {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
SMemAllocator *vnodeCreateMemAllocator(SVnode *pVnode);
|
typedef struct SVArenaNode SVArenaNode;
|
||||||
void vnodeDestroyMemAllocator(SMemAllocator *pma);
|
typedef struct SVMemAllocator SVMemAllocator;
|
||||||
|
|
||||||
|
struct SVArenaNode {
|
||||||
|
TD_SLIST_NODE(SVArenaNode);
|
||||||
|
uint64_t size; // current node size
|
||||||
|
void * ptr;
|
||||||
|
char data[];
|
||||||
|
};
|
||||||
|
|
||||||
|
struct SVMemAllocator {
|
||||||
|
T_REF_DECLARE()
|
||||||
|
TD_DLIST_NODE(SVMemAllocator);
|
||||||
|
uint64_t capacity;
|
||||||
|
uint64_t ssize;
|
||||||
|
uint64_t lsize;
|
||||||
|
SVArenaNode *pNode;
|
||||||
|
TD_SLIST(SVArenaNode) nlist;
|
||||||
|
};
|
||||||
|
|
||||||
|
SVMemAllocator *vmaCreate(uint64_t capacity, uint64_t ssize, uint64_t lsize);
|
||||||
|
void vmaDestroy(SVMemAllocator *pVMA);
|
||||||
|
void vmaReset(SVMemAllocator *pVMA);
|
||||||
|
void * vmaMalloc(SVMemAllocator *pVMA, uint64_t size);
|
||||||
|
void vmaFree(SVMemAllocator *pVMA, void *ptr);
|
||||||
|
bool vmaIsFull(SVMemAllocator *pVMA);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,117 @@
|
||||||
|
/*
|
||||||
|
* 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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "vnodeDef.h"
|
||||||
|
|
||||||
|
static SVArenaNode *vArenaNodeNew(uint64_t capacity);
|
||||||
|
static void vArenaNodeFree(SVArenaNode *pNode);
|
||||||
|
|
||||||
|
SVMemAllocator *vmaCreate(uint64_t capacity, uint64_t ssize, uint64_t lsize) {
|
||||||
|
SVMemAllocator *pVMA = (SVMemAllocator *)malloc(sizeof(*pVMA));
|
||||||
|
if (pVMA == NULL) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
pVMA->capacity = capacity;
|
||||||
|
pVMA->ssize = ssize;
|
||||||
|
pVMA->lsize = lsize;
|
||||||
|
tSListInit(&(pVMA->nlist));
|
||||||
|
|
||||||
|
pVMA->pNode = vArenaNodeNew(capacity);
|
||||||
|
if (pVMA->pNode == NULL) {
|
||||||
|
free(pVMA);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
tSListPush(&(pVMA->nlist), pVMA->pNode);
|
||||||
|
|
||||||
|
return pVMA;
|
||||||
|
}
|
||||||
|
|
||||||
|
void vmaDestroy(SVMemAllocator *pVMA) {
|
||||||
|
if (pVMA) {
|
||||||
|
while (TD_SLIST_NELES(&(pVMA->nlist)) > 1) {
|
||||||
|
SVArenaNode *pNode = TD_SLIST_HEAD(&(pVMA->nlist));
|
||||||
|
tSListPop(&(pVMA->nlist));
|
||||||
|
vArenaNodeFree(pNode);
|
||||||
|
}
|
||||||
|
|
||||||
|
free(pVMA);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void vmaReset(SVMemAllocator *pVMA) {
|
||||||
|
while (TD_SLIST_NELES(&(pVMA->nlist)) > 1) {
|
||||||
|
SVArenaNode *pNode = TD_SLIST_HEAD(&(pVMA->nlist));
|
||||||
|
tSListPop(&(pVMA->nlist));
|
||||||
|
vArenaNodeFree(pNode);
|
||||||
|
}
|
||||||
|
|
||||||
|
SVArenaNode *pNode = TD_SLIST_HEAD(&(pVMA->nlist));
|
||||||
|
pNode->ptr = pNode->data;
|
||||||
|
}
|
||||||
|
|
||||||
|
void *vmaMalloc(SVMemAllocator *pVMA, uint64_t size) {
|
||||||
|
SVArenaNode *pNode = TD_SLIST_HEAD(&(pVMA->nlist));
|
||||||
|
void * ptr;
|
||||||
|
|
||||||
|
if (pNode->size < POINTER_DISTANCE(pNode->ptr, pNode->data) + size) {
|
||||||
|
uint64_t capacity = MAX(pVMA->ssize, size);
|
||||||
|
pNode = vArenaNodeNew(capacity);
|
||||||
|
if (pNode == NULL) {
|
||||||
|
// TODO: handle error
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
tSListPush(&(pVMA->nlist), pNode);
|
||||||
|
}
|
||||||
|
|
||||||
|
ptr = pNode->ptr;
|
||||||
|
pNode->ptr = POINTER_SHIFT(ptr, size);
|
||||||
|
|
||||||
|
return ptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
void vmaFree(SVMemAllocator *pVMA, void *ptr) {
|
||||||
|
// TODO
|
||||||
|
}
|
||||||
|
|
||||||
|
bool vmaIsFull(SVMemAllocator *pVMA) {
|
||||||
|
SVArenaNode *pNode = TD_SLIST_HEAD(&(pVMA->nlist));
|
||||||
|
|
||||||
|
return (TD_SLIST_NELES(&(pVMA->nlist)) > 1) ||
|
||||||
|
(pNode->size < POINTER_DISTANCE(pNode->ptr, pNode->data) + pVMA->lsize);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ------------------------ STATIC METHODS ------------------------ */
|
||||||
|
static SVArenaNode *vArenaNodeNew(uint64_t capacity) {
|
||||||
|
SVArenaNode *pNode = NULL;
|
||||||
|
|
||||||
|
pNode = (SVArenaNode *)malloc(sizeof(*pNode) + capacity);
|
||||||
|
if (pNode == NULL) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
pNode->size = capacity;
|
||||||
|
pNode->ptr = pNode->data;
|
||||||
|
|
||||||
|
return pNode;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void vArenaNodeFree(SVArenaNode *pNode) {
|
||||||
|
if (pNode) {
|
||||||
|
free(pNode);
|
||||||
|
}
|
||||||
|
}
|
|
@ -19,14 +19,114 @@
|
||||||
#define VNODE_BUF_POOL_SHARDS 3
|
#define VNODE_BUF_POOL_SHARDS 3
|
||||||
|
|
||||||
struct SVBufPool {
|
struct SVBufPool {
|
||||||
// buffer pool impl
|
pthread_mutex_t mutex;
|
||||||
SList free;
|
pthread_cond_t hasFree;
|
||||||
SList incycle;
|
TD_DLIST(SVMemAllocator) free;
|
||||||
SListNode *inuse;
|
TD_DLIST(SVMemAllocator) incycle;
|
||||||
|
SVMemAllocator *inuse;
|
||||||
// MAF for submodules
|
// MAF for submodules
|
||||||
SMemAllocatorFactory maf;
|
// SMemAllocatorFactory maf;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
int vnodeOpenBufPool(SVnode *pVnode) {
|
||||||
|
uint64_t capacity;
|
||||||
|
|
||||||
|
if ((pVnode->pBufPool = (SVBufPool *)calloc(1, sizeof(SVBufPool))) == NULL) {
|
||||||
|
/* TODO */
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
tDListInit(&(pVnode->pBufPool->free));
|
||||||
|
tDListInit(&(pVnode->pBufPool->incycle));
|
||||||
|
|
||||||
|
pVnode->pBufPool->inuse = NULL;
|
||||||
|
|
||||||
|
// TODO
|
||||||
|
capacity = pVnode->config.wsize / VNODE_BUF_POOL_SHARDS;
|
||||||
|
|
||||||
|
for (int i = 0; i < VNODE_BUF_POOL_SHARDS; i++) {
|
||||||
|
SVMemAllocator *pVMA = vmaCreate(capacity, pVnode->config.ssize, pVnode->config.lsize);
|
||||||
|
if (pVMA == NULL) {
|
||||||
|
// TODO: handle error
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
tDListAppend(&(pVnode->pBufPool->free), pVMA);
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void vnodeCloseBufPool(SVnode *pVnode) {
|
||||||
|
if (pVnode->pBufPool) {
|
||||||
|
vmaDestroy(pVnode->pBufPool->inuse);
|
||||||
|
|
||||||
|
while (true) {
|
||||||
|
SVMemAllocator *pVMA = TD_DLIST_HEAD(&(pVnode->pBufPool->incycle));
|
||||||
|
if (pVMA == NULL) break;
|
||||||
|
tDListPop(&(pVnode->pBufPool->incycle), pVMA);
|
||||||
|
vmaDestroy(pVMA);
|
||||||
|
}
|
||||||
|
|
||||||
|
while (true) {
|
||||||
|
SVMemAllocator *pVMA = TD_DLIST_HEAD(&(pVnode->pBufPool->free));
|
||||||
|
if (pVMA == NULL) break;
|
||||||
|
tDListPop(&(pVnode->pBufPool->free), pVMA);
|
||||||
|
vmaDestroy(pVMA);
|
||||||
|
}
|
||||||
|
|
||||||
|
free(pVnode->pBufPool);
|
||||||
|
pVnode->pBufPool = NULL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int vnodeBufPoolSwitch(SVnode *pVnode) {
|
||||||
|
SVMemAllocator *pvma = pVnode->pBufPool->inuse;
|
||||||
|
|
||||||
|
pVnode->pBufPool->inuse = NULL;
|
||||||
|
|
||||||
|
tDListAppend(&(pVnode->pBufPool->incycle), pvma);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int vnodeBufPoolRecycle(SVnode *pVnode) {
|
||||||
|
SVBufPool * pBufPool = pVnode->pBufPool;
|
||||||
|
SVMemAllocator *pvma = TD_DLIST_HEAD(&(pBufPool->incycle));
|
||||||
|
ASSERT(pvma != NULL);
|
||||||
|
|
||||||
|
tDListPop(&(pBufPool->incycle), pvma);
|
||||||
|
vmaReset(pvma);
|
||||||
|
tDListAppend(&(pBufPool->free), pvma);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void *vnodeMalloc(SVnode *pVnode, uint64_t size) {
|
||||||
|
SVBufPool *pBufPool = pVnode->pBufPool;
|
||||||
|
|
||||||
|
if (pBufPool->inuse == NULL) {
|
||||||
|
while (true) {
|
||||||
|
// TODO: add sem_wait and sem_post
|
||||||
|
pBufPool->inuse = TD_DLIST_HEAD(&(pBufPool->free));
|
||||||
|
if (pBufPool->inuse) {
|
||||||
|
tDListPop(&(pBufPool->free), pBufPool->inuse);
|
||||||
|
break;
|
||||||
|
} else {
|
||||||
|
// tsem_wait(&(pBufPool->hasFree));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return vmaMalloc(pBufPool->inuse, size);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool vnodeBufPoolIsFull(SVnode *pVnode) {
|
||||||
|
if (pVnode->pBufPool->inuse == NULL) return false;
|
||||||
|
return vmaIsFull(pVnode->pBufPool->inuse);
|
||||||
|
}
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
|
||||||
typedef enum {
|
typedef enum {
|
||||||
// Heap allocator
|
// Heap allocator
|
||||||
E_V_HEAP_ALLOCATOR = 0,
|
E_V_HEAP_ALLOCATOR = 0,
|
||||||
|
@ -57,15 +157,6 @@ typedef struct {
|
||||||
SListNode *pNode;
|
SListNode *pNode;
|
||||||
} SVMAWrapper;
|
} SVMAWrapper;
|
||||||
|
|
||||||
typedef struct {
|
|
||||||
T_REF_DECLARE()
|
|
||||||
uint64_t capacity;
|
|
||||||
EVMemAllocatorT type;
|
|
||||||
union {
|
|
||||||
SVHeapAllocator vha;
|
|
||||||
SVArenaAllocator vaa;
|
|
||||||
};
|
|
||||||
} SVMemAllocator;
|
|
||||||
|
|
||||||
static SListNode * vBufPoolNewNode(uint64_t capacity, EVMemAllocatorT type);
|
static SListNode * vBufPoolNewNode(uint64_t capacity, EVMemAllocatorT type);
|
||||||
static void vBufPoolFreeNode(SListNode *pNode);
|
static void vBufPoolFreeNode(SListNode *pNode);
|
||||||
|
@ -73,106 +164,13 @@ static SMemAllocator *vBufPoolCreateMA(SMemAllocatorFactory *pmaf);
|
||||||
static void vBufPoolDestroyMA(SMemAllocatorFactory *pmaf, SMemAllocator *pma);
|
static void vBufPoolDestroyMA(SMemAllocatorFactory *pmaf, SMemAllocator *pma);
|
||||||
static void * vBufPoolMalloc(SVMemAllocator *pvma, uint64_t size);
|
static void * vBufPoolMalloc(SVMemAllocator *pvma, uint64_t size);
|
||||||
|
|
||||||
int vnodeOpenBufPool(SVnode *pVnode) {
|
|
||||||
uint64_t capacity;
|
|
||||||
EVMemAllocatorT type = E_V_ARENA_ALLOCATOR;
|
|
||||||
|
|
||||||
if ((pVnode->pBufPool = (SVBufPool *)calloc(1, sizeof(SVBufPool))) == NULL) {
|
|
||||||
/* TODO */
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
tdListInit(&(pVnode->pBufPool->free), 0);
|
|
||||||
tdListInit(&(pVnode->pBufPool->incycle), 0);
|
|
||||||
|
|
||||||
capacity = pVnode->config.wsize / VNODE_BUF_POOL_SHARDS;
|
|
||||||
if (pVnode->config.isHeapAllocator) {
|
|
||||||
type = E_V_HEAP_ALLOCATOR;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (int i = 0; i < VNODE_BUF_POOL_SHARDS; i++) {
|
|
||||||
SListNode *pNode = vBufPoolNewNode(capacity, type);
|
|
||||||
if (pNode == NULL) {
|
|
||||||
vnodeCloseBufPool(pVnode);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
tdListAppendNode(&(pVnode->pBufPool->free), pNode);
|
|
||||||
}
|
|
||||||
|
|
||||||
pVnode->pBufPool->maf.impl = pVnode;
|
|
||||||
pVnode->pBufPool->maf.create = vBufPoolCreateMA;
|
|
||||||
pVnode->pBufPool->maf.destroy = vBufPoolDestroyMA;
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
void vnodeCloseBufPool(SVnode *pVnode) {
|
|
||||||
SListNode *pNode;
|
|
||||||
if (pVnode->pBufPool) {
|
|
||||||
// Clear free list
|
|
||||||
while ((pNode = tdListPopHead(&(pVnode->pBufPool->free))) != NULL) {
|
|
||||||
vBufPoolFreeNode(pNode);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Clear incycle list
|
|
||||||
while ((pNode = tdListPopHead(&(pVnode->pBufPool->incycle))) != NULL) {
|
|
||||||
vBufPoolFreeNode(pNode);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Free inuse node
|
|
||||||
if (pVnode->pBufPool->inuse) {
|
|
||||||
vBufPoolFreeNode(pVnode->pBufPool->inuse);
|
|
||||||
}
|
|
||||||
|
|
||||||
free(pVnode->pBufPool);
|
|
||||||
pVnode->pBufPool = NULL;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void *vnodeMalloc(SVnode *pVnode, uint64_t size) {
|
|
||||||
void *ptr;
|
|
||||||
|
|
||||||
if (pVnode->pBufPool->inuse == NULL) {
|
|
||||||
SListNode *pNode;
|
|
||||||
while ((pNode = tdListPopHead(&(pVnode->pBufPool->free))) == NULL) {
|
|
||||||
// todo
|
|
||||||
// tsem_wait();
|
|
||||||
ASSERT(0);
|
|
||||||
}
|
|
||||||
|
|
||||||
pVnode->pBufPool->inuse = pNode;
|
|
||||||
}
|
|
||||||
|
|
||||||
SVMemAllocator *pvma = (SVMemAllocator *)(pVnode->pBufPool->inuse->data);
|
|
||||||
return vBufPoolMalloc(pvma, size);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* ------------------------ STATIC METHODS ------------------------ */
|
/* ------------------------ STATIC METHODS ------------------------ */
|
||||||
static void vArenaAllocatorInit(SVArenaAllocator *pvaa, uint64_t capacity, uint64_t ssize, uint64_t lsize) { /* TODO */
|
|
||||||
pvaa->ssize = ssize;
|
|
||||||
pvaa->lsize = lsize;
|
|
||||||
pvaa->inuse = &pvaa->node;
|
|
||||||
|
|
||||||
pvaa->node.prev = NULL;
|
|
||||||
pvaa->node.size = capacity;
|
|
||||||
pvaa->node.ptr = pvaa->node.data;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void vArenaAllocatorClear(SVArenaAllocator *pvaa) { /* TODO */
|
|
||||||
while (pvaa->inuse != &(pvaa->node)) {
|
|
||||||
SVArenaNode *pANode = pvaa->inuse;
|
|
||||||
pvaa->inuse = pANode->prev;
|
|
||||||
free(pANode);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static SListNode *vBufPoolNewNode(uint64_t capacity, EVMemAllocatorT type) {
|
static SListNode *vBufPoolNewNode(uint64_t capacity, EVMemAllocatorT type) {
|
||||||
SListNode * pNode;
|
SListNode * pNode;
|
||||||
SVMemAllocator *pvma;
|
SVMemAllocator *pvma;
|
||||||
uint64_t msize;
|
uint64_t msize;
|
||||||
uint64_t ssize = 0; // TODO
|
uint64_t ssize = 4096; // TODO
|
||||||
uint64_t lsize = 0; // TODO
|
uint64_t lsize = 1024; // TODO
|
||||||
|
|
||||||
msize = sizeof(SListNode) + sizeof(SVMemAllocator);
|
msize = sizeof(SListNode) + sizeof(SVMemAllocator);
|
||||||
if (type == E_V_ARENA_ALLOCATOR) {
|
if (type == E_V_ARENA_ALLOCATOR) {
|
||||||
|
@ -317,4 +315,5 @@ static void vBufPoolDestroyMA(SMemAllocatorFactory *pmaf, SMemAllocator *pma) {
|
||||||
tdListAppendNode(&(pVnode->pBufPool->free), tdListPopNode(&(pVnode->pBufPool->incycle), pNode));
|
tdListAppendNode(&(pVnode->pBufPool->free), tdListPopNode(&(pVnode->pBufPool->incycle), pNode));
|
||||||
// tsem_post(); todo: sem_post
|
// tsem_post(); todo: sem_post
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
#endif
|
|
@ -15,7 +15,8 @@
|
||||||
|
|
||||||
#include "vnodeDef.h"
|
#include "vnodeDef.h"
|
||||||
|
|
||||||
const SVnodeCfg defaultVnodeOptions = {0}; /* TODO */
|
const SVnodeCfg defaultVnodeOptions = {
|
||||||
|
.wsize = 96 * 1024 * 1024, .ssize = 1 * 1024 * 1024, .lsize = 1024, .walCfg = {.level = TAOS_WAL_WRITE}}; /* TODO */
|
||||||
|
|
||||||
void vnodeOptionsInit(SVnodeCfg *pVnodeOptions) { /* TODO */
|
void vnodeOptionsInit(SVnodeCfg *pVnodeOptions) { /* TODO */
|
||||||
vnodeOptionsCopy(pVnodeOptions, &defaultVnodeOptions);
|
vnodeOptionsCopy(pVnodeOptions, &defaultVnodeOptions);
|
||||||
|
|
|
@ -18,31 +18,22 @@
|
||||||
static int vnodeStartCommit(SVnode *pVnode);
|
static int vnodeStartCommit(SVnode *pVnode);
|
||||||
static int vnodeEndCommit(SVnode *pVnode);
|
static int vnodeEndCommit(SVnode *pVnode);
|
||||||
|
|
||||||
bool vnodeShouldCommit(SVnode *pVnode) { return false; }
|
|
||||||
|
|
||||||
int vnodeAsyncCommit(SVnode *pVnode) {
|
int vnodeAsyncCommit(SVnode *pVnode) {
|
||||||
#if 0
|
vnodeBufPoolSwitch(pVnode);
|
||||||
if (vnodeStartCommit(pVnode) < 0) {
|
SVnodeTask *pTask = (SVnodeTask *)malloc(sizeof(*pTask));
|
||||||
// TODO
|
|
||||||
}
|
|
||||||
|
|
||||||
if (tqCommit(pVnode->pTQ) < 0) {
|
pTask->execute = vnodeCommit; // TODO
|
||||||
// TODO
|
pTask->arg = pVnode; // TODO
|
||||||
}
|
|
||||||
|
|
||||||
if (metaCommit(pVnode->pMeta) < 0) {
|
vnodeScheduleTask(pTask);
|
||||||
// TODO
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (tsdbCommit(pVnode->pTsdb) < 0) {
|
int vnodeCommit(void *arg) {
|
||||||
// TODO
|
SVnode *pVnode = (SVnode *)arg;
|
||||||
}
|
|
||||||
|
|
||||||
if (vnodeEndCommit(pVnode) < 0) {
|
vnodeBufPoolRecycle(pVnode);
|
||||||
// TODO
|
// TODO
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -20,19 +20,6 @@ static void vnodeFree(SVnode *pVnode);
|
||||||
static int vnodeOpenImpl(SVnode *pVnode);
|
static int vnodeOpenImpl(SVnode *pVnode);
|
||||||
static void vnodeCloseImpl(SVnode *pVnode);
|
static void vnodeCloseImpl(SVnode *pVnode);
|
||||||
|
|
||||||
int vnodeInit() {
|
|
||||||
// TODO
|
|
||||||
if (walInit() < 0) {
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
void vnodeClear() {
|
|
||||||
walCleanUp();
|
|
||||||
}
|
|
||||||
|
|
||||||
SVnode *vnodeOpen(const char *path, const SVnodeCfg *pVnodeCfg) {
|
SVnode *vnodeOpen(const char *path, const SVnodeCfg *pVnodeCfg) {
|
||||||
SVnode *pVnode = NULL;
|
SVnode *pVnode = NULL;
|
||||||
|
|
||||||
|
|
|
@ -1,113 +0,0 @@
|
||||||
/*
|
|
||||||
* 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/>.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include "vnodeDef.h"
|
|
||||||
|
|
||||||
SMemAllocator *vnodeCreateMemAllocator(SVnode *pVnode) {
|
|
||||||
SMemAllocator *pma = NULL;
|
|
||||||
/* TODO */
|
|
||||||
return pma;
|
|
||||||
}
|
|
||||||
|
|
||||||
void vnodeDestroyMemAllocator(SMemAllocator *pma) {
|
|
||||||
// TODO
|
|
||||||
}
|
|
||||||
|
|
||||||
#if 0
|
|
||||||
#define VNODE_HEAP_ALLOCATOR 0
|
|
||||||
#define VNODE_ARENA_ALLOCATOR 1
|
|
||||||
|
|
||||||
typedef struct {
|
|
||||||
uint64_t tsize;
|
|
||||||
uint64_t used;
|
|
||||||
} SVHeapAllocator;
|
|
||||||
|
|
||||||
typedef struct SVArenaNode {
|
|
||||||
struct SVArenaNode *prev;
|
|
||||||
void * nptr;
|
|
||||||
char data[];
|
|
||||||
} SVArenaNode;
|
|
||||||
|
|
||||||
typedef struct {
|
|
||||||
SVArenaNode *inuse;
|
|
||||||
SVArenaNode node;
|
|
||||||
} SVArenaAllocator;
|
|
||||||
|
|
||||||
typedef struct {
|
|
||||||
int8_t type;
|
|
||||||
uint64_t tsize;
|
|
||||||
T_REF_DECLARE()
|
|
||||||
union {
|
|
||||||
SVHeapAllocator vha;
|
|
||||||
SVArenaAllocator vaa;
|
|
||||||
};
|
|
||||||
} SVMemAllocator;
|
|
||||||
|
|
||||||
SMemAllocator *vnodeCreateMemAllocator(int8_t type, uint64_t tsize, uint64_t ssize /* step size only for arena */) {
|
|
||||||
SMemAllocator * pma;
|
|
||||||
uint64_t msize;
|
|
||||||
SVMemAllocator *pva;
|
|
||||||
|
|
||||||
msize = sizeof(*pma) + sizeof(SVMemAllocator);
|
|
||||||
if (type == VNODE_ARENA_ALLOCATOR) {
|
|
||||||
msize += tsize;
|
|
||||||
}
|
|
||||||
|
|
||||||
pma = (SMemAllocator *)calloc(1, msize);
|
|
||||||
if (pma == NULL) {
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
pma->impl = POINTER_SHIFT(pma, sizeof(*pma));
|
|
||||||
pva = (SVMemAllocator *)(pma->impl);
|
|
||||||
pva->type = type;
|
|
||||||
pva->tsize = tsize;
|
|
||||||
|
|
||||||
if (type == VNODE_HEAP_ALLOCATOR) {
|
|
||||||
pma->malloc = NULL;
|
|
||||||
pma->calloc = NULL;
|
|
||||||
pma->realloc = NULL;
|
|
||||||
pma->free = NULL;
|
|
||||||
pma->usage = NULL;
|
|
||||||
} else if (type == VNODE_ARENA_ALLOCATOR) {
|
|
||||||
pma->malloc = NULL;
|
|
||||||
pma->calloc = NULL;
|
|
||||||
pma->realloc = NULL;
|
|
||||||
pma->free = NULL;
|
|
||||||
pma->usage = NULL;
|
|
||||||
} else {
|
|
||||||
ASSERT(0);
|
|
||||||
}
|
|
||||||
|
|
||||||
return pma;
|
|
||||||
}
|
|
||||||
|
|
||||||
void vnodeDestroyMemAllocator(SMemAllocator *pma) {
|
|
||||||
// TODO
|
|
||||||
}
|
|
||||||
|
|
||||||
void vnodeRefMemAllocator(SMemAllocator *pma) {
|
|
||||||
// TODO
|
|
||||||
}
|
|
||||||
|
|
||||||
void vnodeUnrefMemAllocator(SMemAllocator *pma) {
|
|
||||||
// TODO
|
|
||||||
}
|
|
||||||
|
|
||||||
/* ------------------------ Heap Allocator IMPL ------------------------ */
|
|
||||||
|
|
||||||
/* ------------------------ Arena Allocator IMPL ------------------------ */
|
|
||||||
|
|
||||||
#endif
|
|
|
@ -0,0 +1,115 @@
|
||||||
|
/*
|
||||||
|
* 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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "vnodeDef.h"
|
||||||
|
|
||||||
|
SVnodeMgr vnodeMgr = {.vnodeInitFlag = TD_MOD_UNINITIALIZED, .vnodeClearFlag = TD_MOD_UNCLEARD, .stop = false};
|
||||||
|
|
||||||
|
static void* loop(void* arg);
|
||||||
|
|
||||||
|
int vnodeInit(uint16_t nthreads) {
|
||||||
|
if (TD_CHECK_AND_SET_MODE_INIT(&(vnodeMgr.vnodeInitFlag)) == TD_MOD_INITIALIZED) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start commit handers
|
||||||
|
if (nthreads > 0) {
|
||||||
|
vnodeMgr.nthreads = nthreads;
|
||||||
|
vnodeMgr.threads = (pthread_t*)calloc(nthreads, sizeof(pthread_t));
|
||||||
|
if (vnodeMgr.threads == NULL) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
pthread_mutex_init(&(vnodeMgr.mutex), NULL);
|
||||||
|
pthread_cond_init(&(vnodeMgr.hasTask), NULL);
|
||||||
|
tDListInit(&(vnodeMgr.queue));
|
||||||
|
|
||||||
|
for (uint16_t i = 0; i < nthreads; i++) {
|
||||||
|
pthread_create(&(vnodeMgr.threads[i]), NULL, loop, NULL);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// TODO: if no commit thread is set, then another mechanism should be
|
||||||
|
// given. Otherwise, it is a false.
|
||||||
|
ASSERT(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (walInit() < 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void vnodeClear() {
|
||||||
|
if (TD_CHECK_AND_SET_MOD_CLEAR(&(vnodeMgr.vnodeClearFlag)) == TD_MOD_CLEARD) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
walCleanUp();
|
||||||
|
|
||||||
|
// Stop commit handler
|
||||||
|
pthread_mutex_lock(&(vnodeMgr.mutex));
|
||||||
|
vnodeMgr.stop = true;
|
||||||
|
pthread_cond_broadcast(&(vnodeMgr.hasTask));
|
||||||
|
pthread_mutex_unlock(&(vnodeMgr.mutex));
|
||||||
|
|
||||||
|
for (uint16_t i = 0; i < vnodeMgr.nthreads; i++) {
|
||||||
|
pthread_join(vnodeMgr.threads[i], NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
tfree(vnodeMgr.threads);
|
||||||
|
pthread_cond_destroy(&(vnodeMgr.hasTask));
|
||||||
|
pthread_mutex_destroy(&(vnodeMgr.mutex));
|
||||||
|
}
|
||||||
|
|
||||||
|
int vnodeScheduleTask(SVnodeTask* pTask) {
|
||||||
|
pthread_mutex_lock(&(vnodeMgr.mutex));
|
||||||
|
|
||||||
|
tDListAppend(&(vnodeMgr.queue), pTask);
|
||||||
|
|
||||||
|
pthread_cond_signal(&(vnodeMgr.hasTask));
|
||||||
|
|
||||||
|
pthread_mutex_unlock(&(vnodeMgr.mutex));
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ------------------------ STATIC METHODS ------------------------ */
|
||||||
|
static void* loop(void* arg) {
|
||||||
|
SVnodeTask* pTask;
|
||||||
|
for (;;) {
|
||||||
|
pthread_mutex_lock(&(vnodeMgr.mutex));
|
||||||
|
for (;;) {
|
||||||
|
pTask = TD_DLIST_HEAD(&(vnodeMgr.queue));
|
||||||
|
if (pTask == NULL) {
|
||||||
|
if (vnodeMgr.stop) {
|
||||||
|
pthread_mutex_unlock(&(vnodeMgr.mutex));
|
||||||
|
return NULL;
|
||||||
|
} else {
|
||||||
|
pthread_cond_wait(&(vnodeMgr.hasTask), &(vnodeMgr.mutex));
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
tDListPop(&(vnodeMgr.queue), pTask);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pthread_mutex_unlock(&(vnodeMgr.mutex));
|
||||||
|
|
||||||
|
(*(pTask->execute))(pTask->arg);
|
||||||
|
}
|
||||||
|
|
||||||
|
return NULL;
|
||||||
|
}
|
|
@ -4,4 +4,9 @@ target_sources(vnodeApiTests
|
||||||
PRIVATE
|
PRIVATE
|
||||||
"vnodeApiTests.cpp"
|
"vnodeApiTests.cpp"
|
||||||
)
|
)
|
||||||
target_link_libraries(vnodeApiTests vnode gtest gtest_main)
|
target_link_libraries(vnodeApiTests vnode gtest gtest_main)
|
||||||
|
|
||||||
|
add_test(
|
||||||
|
NAME vnode_api_tests
|
||||||
|
COMMAND ${CMAKE_CURRENT_BINARY_DIR}/vnodeApiTests
|
||||||
|
)
|
|
@ -1,3 +1,14 @@
|
||||||
|
/**
|
||||||
|
* @file vnodeApiTests.cpp
|
||||||
|
* @author hzcheng (hzcheng@taosdata.com)
|
||||||
|
* @brief VNODE module API tests
|
||||||
|
* @version 0.1
|
||||||
|
* @date 2021-12-13
|
||||||
|
*
|
||||||
|
* @copyright Copyright (c) 2021
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
#include <gtest/gtest.h>
|
#include <gtest/gtest.h>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
|
||||||
|
@ -81,7 +92,7 @@ TEST(vnodeApiTest, test_create_table_encode_and_decode_function) {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
TEST(vnodeApiTest, vnodeOpen_vnodeClose_test) {
|
TEST(vnodeApiTest, vnodeOpen_vnodeClose_test) {
|
||||||
GTEST_ASSERT_GE(vnodeInit(), 0);
|
GTEST_ASSERT_GE(vnodeInit(2), 0);
|
||||||
|
|
||||||
// Create and open a vnode
|
// Create and open a vnode
|
||||||
SVnode *pVnode = vnodeOpen("vnode1", NULL);
|
SVnode *pVnode = vnodeOpen("vnode1", NULL);
|
||||||
|
|
|
@ -9,4 +9,5 @@ add_subdirectory(cache)
|
||||||
add_subdirectory(catalog)
|
add_subdirectory(catalog)
|
||||||
add_subdirectory(executor)
|
add_subdirectory(executor)
|
||||||
add_subdirectory(planner)
|
add_subdirectory(planner)
|
||||||
add_subdirectory(function)
|
add_subdirectory(function)
|
||||||
|
add_subdirectory(query)
|
||||||
|
|
|
@ -8,5 +8,5 @@ target_include_directories(
|
||||||
|
|
||||||
target_link_libraries(
|
target_link_libraries(
|
||||||
catalog
|
catalog
|
||||||
PRIVATE os util common transport
|
PRIVATE os util common transport query
|
||||||
)
|
)
|
||||||
|
|
|
@ -21,14 +21,58 @@ extern "C" {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include "catalog.h"
|
#include "catalog.h"
|
||||||
|
#include "common.h"
|
||||||
|
#include "tlog.h"
|
||||||
|
|
||||||
|
#define CTG_DEFAULT_CLUSTER_NUMBER 6
|
||||||
|
#define CTG_DEFAULT_VGROUP_NUMBER 100
|
||||||
|
|
||||||
|
#define CTG_DEFAULT_INVALID_VERSION (-1)
|
||||||
|
|
||||||
|
typedef struct SVgroupListCache {
|
||||||
|
int32_t vgroupVersion;
|
||||||
|
SHashObj *cache; // key:vgId, value:SVgroupInfo*
|
||||||
|
SArray *arrayCache; // SVgroupInfo
|
||||||
|
} SVgroupListCache;
|
||||||
|
|
||||||
|
typedef struct SDBVgroupCache {
|
||||||
|
SHashObj *cache; //key:dbname, value:SDBVgroupInfo
|
||||||
|
} SDBVgroupCache;
|
||||||
|
|
||||||
|
typedef struct STableMetaCache {
|
||||||
|
SHashObj *cache; //key:fulltablename, value:STableMeta
|
||||||
|
} STableMetaCache;
|
||||||
|
|
||||||
typedef struct SCatalog {
|
typedef struct SCatalog {
|
||||||
void *pMsgSender; // used to send messsage to mnode to fetch necessary metadata
|
SVgroupListCache vgroupCache;
|
||||||
SHashObj *pData; // items cached for each cluster, the hash key is the cluster-id, returned by mgmt node
|
SDBVgroupCache dbCache;
|
||||||
|
STableMetaCache tableCache;
|
||||||
} SCatalog;
|
} SCatalog;
|
||||||
|
|
||||||
|
typedef struct SCatalogMgmt {
|
||||||
|
void *pMsgSender; // used to send messsage to mnode to fetch necessary metadata
|
||||||
|
SHashObj *pCluster; // items cached for each cluster, the hash key is the cluster-id got from mgmt node
|
||||||
|
} SCatalogMgmt;
|
||||||
|
|
||||||
|
|
||||||
|
extern int32_t ctgDebugFlag;
|
||||||
|
|
||||||
|
#define ctgFatal(...) do { if (ctgDebugFlag & DEBUG_FATAL) { taosPrintLog("CTG FATAL ", ctgDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define ctgError(...) do { if (ctgDebugFlag & DEBUG_ERROR) { taosPrintLog("CTG ERROR ", ctgDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define ctgWarn(...) do { if (ctgDebugFlag & DEBUG_WARN) { taosPrintLog("CTG WARN ", ctgDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define ctgInfo(...) do { if (ctgDebugFlag & DEBUG_INFO) { taosPrintLog("CTG ", ctgDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define ctgDebug(...) do { if (ctgDebugFlag & DEBUG_DEBUG) { taosPrintLog("CTG ", ctgDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define ctgTrace(...) do { if (ctgDebugFlag & DEBUG_TRACE) { taosPrintLog("CTG ", ctgDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define ctgDebugL(...) do { if (ctgDebugFlag & DEBUG_DEBUG) { taosPrintLongString("CTG ", ctgDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
|
||||||
|
|
||||||
|
#define CTG_ERR_RET(c) do { int32_t _code = c; if (_code != TSDB_CODE_SUCCESS) { return _code; } } while (0)
|
||||||
|
#define CTG_ERR_LRET(c,...) do { int32_t _code = c; if (_code != TSDB_CODE_SUCCESS) { ctgError(__VA_ARGS__); return _code; } } while (0)
|
||||||
|
#define CTG_ERR_JRET(c) do { code = c; if (code != TSDB_CODE_SUCCESS) { goto _return; } } while (0)
|
||||||
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#endif /*_TD_CATALOG_INT_H_*/
|
#endif /*_TD_CATALOG_INT_H_*/
|
||||||
|
|
|
@ -14,11 +14,339 @@
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "catalogInt.h"
|
#include "catalogInt.h"
|
||||||
|
#include "trpc.h"
|
||||||
|
#include "query.h"
|
||||||
|
|
||||||
struct SCatalog* getCatalogHandle(const SEpSet* pMgmtEps) {
|
SCatalogMgmt ctgMgmt = {0};
|
||||||
return (struct SCatalog*) 0x1;
|
|
||||||
|
int32_t ctgGetVgroupFromMnode(struct SCatalog* pCatalog, void *pRpc, const SEpSet* pMgmtEps, SVgroupListInfo** pVgroup) {
|
||||||
|
char *msg = NULL;
|
||||||
|
SEpSet *pVnodeEpSet = NULL;
|
||||||
|
int32_t msgLen = 0;
|
||||||
|
|
||||||
|
int32_t code = queryBuildMsg[TSDB_MSG_TYPE_VGROUP_LIST](NULL, &msg, 0, &msgLen);
|
||||||
|
if (code) {
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
|
||||||
|
SRpcMsg rpcMsg = {
|
||||||
|
.msgType = TSDB_MSG_TYPE_VGROUP_LIST,
|
||||||
|
.pCont = msg,
|
||||||
|
.contLen = msgLen,
|
||||||
|
};
|
||||||
|
|
||||||
|
SRpcMsg rpcRsp = {0};
|
||||||
|
|
||||||
|
rpcSendRecv(pRpc, (SEpSet*)pMgmtEps, &rpcMsg, &rpcRsp);
|
||||||
|
|
||||||
|
code = queryProcessMsgRsp[TSDB_MSG_TYPE_VGROUP_LIST](pVgroup, rpcRsp.pCont, rpcRsp.contLen);
|
||||||
|
if (code) {
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t catalogGetMetaData(struct SCatalog* pCatalog, const SMetaReq* pMetaReq, SMetaData* pMetaData) {
|
int32_t ctgGetVgroupFromCache(SCatalog* pCatalog, SArray** pVgroupList, int32_t* exist) {
|
||||||
|
if (NULL == pCatalog->vgroupCache.arrayCache || pCatalog->vgroupCache.vgroupVersion < 0) {
|
||||||
|
*exist = 0;
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pVgroupList) {
|
||||||
|
*pVgroupList = taosArrayDup(pCatalog->vgroupCache.arrayCache);
|
||||||
|
}
|
||||||
|
|
||||||
|
*exist = 1;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t ctgGetDBVgroupFromCache(SCatalog* pCatalog, char *dbName, SDBVgroupInfo **dbInfo, int32_t *exist) {
|
||||||
|
/*
|
||||||
|
if (NULL == pCatalog->dbCache.cache) {
|
||||||
|
*exist = 0;
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
taosHashGet(SHashObj * pHashObj, const void * key, size_t keyLen)
|
||||||
|
|
||||||
|
if (dbInfo) {
|
||||||
|
*pVgroupList = taosArrayDup(pCatalog->vgroupCache.arrayCache);
|
||||||
|
}
|
||||||
|
|
||||||
|
*exist = 1;
|
||||||
|
*/
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t catalogInit(SCatalogCfg *cfg) {
|
||||||
|
ctgMgmt.pCluster = taosHashInit(CTG_DEFAULT_CLUSTER_NUMBER, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);
|
||||||
|
if (NULL == ctgMgmt.pCluster) {
|
||||||
|
CTG_ERR_LRET(TSDB_CODE_CTG_INTERNAL_ERROR, "init %d cluster cache failed", CTG_DEFAULT_CLUSTER_NUMBER);
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t catalogGetHandle(const char *clusterId, struct SCatalog** catalogHandle) {
|
||||||
|
if (NULL == clusterId || NULL == catalogHandle) {
|
||||||
|
return TSDB_CODE_CTG_INVALID_INPUT;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (NULL == ctgMgmt.pCluster) {
|
||||||
|
ctgError("cluster cache are not ready");
|
||||||
|
return TSDB_CODE_CTG_NOT_READY;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t clen = strlen(clusterId);
|
||||||
|
SCatalog *clusterCtg = (SCatalog *)taosHashGet(ctgMgmt.pCluster, clusterId, clen);
|
||||||
|
|
||||||
|
if (clusterCtg) {
|
||||||
|
*catalogHandle = clusterCtg;
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
clusterCtg = calloc(1, sizeof(*clusterCtg));
|
||||||
|
if (NULL == clusterCtg) {
|
||||||
|
ctgError("calloc %d failed", (int32_t)sizeof(*clusterCtg));
|
||||||
|
return TSDB_CODE_CTG_MEM_ERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
clusterCtg->vgroupCache.vgroupVersion = CTG_DEFAULT_INVALID_VERSION;
|
||||||
|
|
||||||
|
if (taosHashPut(ctgMgmt.pCluster, clusterId, clen, &clusterCtg, POINTER_BYTES)) {
|
||||||
|
ctgError("put cluster %s cache to hash failed", clusterId);
|
||||||
|
tfree(clusterCtg);
|
||||||
|
return TSDB_CODE_CTG_INTERNAL_ERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
*catalogHandle = clusterCtg;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t catalogGetVgroupVersion(struct SCatalog* pCatalog, int32_t* version) {
|
||||||
|
if (NULL == pCatalog || NULL == version) {
|
||||||
|
return TSDB_CODE_CTG_INVALID_INPUT;
|
||||||
|
}
|
||||||
|
|
||||||
|
*version = pCatalog->vgroupCache.vgroupVersion;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
int32_t catalogUpdateVgroup(struct SCatalog* pCatalog, SVgroupListInfo* pVgroup) {
|
||||||
|
if (NULL == pVgroup) {
|
||||||
|
ctgError("vgroup get from mnode succeed, but no output");
|
||||||
|
return TSDB_CODE_CTG_INTERNAL_ERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pVgroup->vgroupVersion < 0) {
|
||||||
|
ctgError("vgroup version[%d] is invalid", pVgroup->vgroupVersion);
|
||||||
|
return TSDB_CODE_CTG_INVALID_INPUT;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
if (NULL == pCatalog->vgroupCache.arrayCache) {
|
||||||
|
pCatalog->vgroupCache.arrayCache = taosArrayInit(pVgroup->vgroupNum, sizeof(pVgroup->vgroupInfo[0]));
|
||||||
|
if (NULL == pCatalog->vgroupCache.arrayCache) {
|
||||||
|
ctgError("init array[%d] for cluster cache failed", pVgroup->vgroupNum);
|
||||||
|
return TSDB_CODE_CTG_MEM_ERROR;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
taosArrayClear(pCatalog->vgroupCache.arrayCache);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (NULL == pCatalog->vgroupCache.cache) {
|
||||||
|
pCatalog->vgroupCache.cache = taosHashInit(CTG_DEFAULT_VGROUP_NUMBER, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_ENTRY_LOCK);
|
||||||
|
if (NULL == pCatalog->vgroupCache.cache) {
|
||||||
|
ctgError("init hash[%d] for cluster cache failed", CTG_DEFAULT_VGROUP_NUMBER);
|
||||||
|
return TSDB_CODE_CTG_MEM_ERROR;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
taosHashClear(pCatalog->vgroupCache.cache);
|
||||||
|
}
|
||||||
|
|
||||||
|
SVgroupInfo *vInfo = NULL;
|
||||||
|
for (int32_t i = 0; i < pVgroup->vgroupNum; ++i) {
|
||||||
|
vInfo = taosArrayPush(pCatalog->vgroupCache.arrayCache, &pVgroup->vgroupInfo[i]);
|
||||||
|
if (NULL == vInfo) {
|
||||||
|
ctgError("push to vgroup array cache failed");
|
||||||
|
goto error_exit;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (taosHashPut(pCatalog->vgroupCache.cache, &pVgroup->vgroupInfo[i].vgId, sizeof(pVgroup->vgroupInfo[i].vgId), &vInfo, POINTER_BYTES) != 0) {
|
||||||
|
ctgError("push to vgroup hash cache failed");
|
||||||
|
goto error_exit;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pCatalog->vgroupCache.vgroupVersion = pVgroup->vgroupVersion;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
|
||||||
|
error_exit:
|
||||||
|
if (pCatalog->vgroupCache.arrayCache) {
|
||||||
|
taosArrayDestroy(pCatalog->vgroupCache.arrayCache);
|
||||||
|
pCatalog->vgroupCache.arrayCache = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pCatalog->vgroupCache.cache) {
|
||||||
|
taosHashCleanup(pCatalog->vgroupCache.cache);
|
||||||
|
pCatalog->vgroupCache.cache = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
pCatalog->vgroupCache.vgroupVersion = CTG_DEFAULT_INVALID_VERSION;
|
||||||
|
|
||||||
|
return TSDB_CODE_CTG_INTERNAL_ERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t catalogGetVgroup(SCatalog* pCatalog, void *pRpc, const SEpSet* pMgmtEps, SArray** pVgroupList) {
|
||||||
|
if (NULL == pCatalog || NULL == pMgmtEps || NULL == pRpc) {
|
||||||
|
return TSDB_CODE_CTG_INVALID_INPUT;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t exist = 0;
|
||||||
|
|
||||||
|
CTG_ERR_RET(ctgGetVgroupFromCache(pCatalog, pVgroupList, &exist));
|
||||||
|
|
||||||
|
if (exist) {
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
SVgroupListInfo *pVgroup = NULL;
|
||||||
|
|
||||||
|
CTG_ERR_RET(ctgGetVgroupFromMnode(pCatalog, pRpc, pMgmtEps, &pVgroup));
|
||||||
|
|
||||||
|
CTG_ERR_RET(catalogUpdateVgroup(pCatalog, pVgroup));
|
||||||
|
|
||||||
|
if (pVgroupList) {
|
||||||
|
CTG_ERR_RET(ctgGetVgroupFromCache(pCatalog, pVgroupList, &exist));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (0 == exist) {
|
||||||
|
ctgError("catalog fetched but get from cache failed");
|
||||||
|
return TSDB_CODE_CTG_INTERNAL_ERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t catalogGetDBVgroupVersion(struct SCatalog* pCatalog, const char* dbName, int32_t* version) {
|
||||||
|
if (NULL == pCatalog || NULL == dbName || NULL == version) {
|
||||||
|
return TSDB_CODE_CTG_INVALID_INPUT;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (NULL == pCatalog->dbCache.cache) {
|
||||||
|
*version = CTG_DEFAULT_INVALID_VERSION;
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
SDBVgroupInfo * dbInfo = taosHashGet(pCatalog->dbCache.cache, dbName, strlen(dbName));
|
||||||
|
if (NULL == dbInfo) {
|
||||||
|
*version = CTG_DEFAULT_INVALID_VERSION;
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
*version = dbInfo->vgroupVersion;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t catalogUpdateDBVgroup(struct SCatalog* pCatalog, const char* dbName, SDBVgroupInfo* dbInfo) {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
int32_t catalogGetDBVgroup(struct SCatalog* pCatalog, void *pRpc, const SEpSet* pMgmtEps, const char* dbName, int32_t forceUpdate, SDBVgroupInfo** dbInfo) {
|
||||||
|
if (NULL == pCatalog || NULL == dbName || NULL == pRpc || NULL == pMgmtEps) {
|
||||||
|
return TSDB_CODE_CTG_INVALID_INPUT;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
int32_t exist = 0;
|
||||||
|
|
||||||
|
if (0 == forceUpdate) {
|
||||||
|
CTG_ERR_RET(ctgGetDBVgroupFromCache(pCatalog, dbName, dbInfo, &exist));
|
||||||
|
|
||||||
|
if (exist) {
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
SDBVgroupInfo* newDbInfo = NULL;
|
||||||
|
|
||||||
|
CTG_ERR_RET(ctgGetDBVgroupFromMnode(pCatalog, pRpc, pMgmtEps, dbName, &newDbInfo));
|
||||||
|
|
||||||
|
CTG_ERR_RET(catalogUpdateDBVgroup(pCatalog, dbName, newDbInfo));
|
||||||
|
|
||||||
|
if (dbInfo) {
|
||||||
|
*dbInfo = newDbInfo;
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
int32_t catalogGetTableMetaFromMnode(struct SCatalog* pCatalog, void *pRpc, const SEpSet* pMgmtEps, const char* pTableName, const STagData* tagData, STableMeta* pTableMeta) {
|
||||||
|
if (NULL == pCatalog || NULL == pMgmtEps || NULL == pTableName || NULL == pTableMeta) {
|
||||||
|
return TSDB_CODE_CTG_INVALID_INPUT;
|
||||||
|
}
|
||||||
|
|
||||||
|
SBuildTableMetaInput bInput = {0};
|
||||||
|
char *msg = NULL;
|
||||||
|
SEpSet *pVnodeEpSet = NULL;
|
||||||
|
int32_t msgLen = 0;
|
||||||
|
|
||||||
|
int32_t code = queryBuildMsg[TSDB_MSG_TYPE_TABLE_META](&bInput, &msg, 0, &msgLen);
|
||||||
|
if (code) {
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
|
||||||
|
SRpcMsg rpcMsg = {
|
||||||
|
.msgType = TSDB_MSG_TYPE_TABLE_META,
|
||||||
|
.pCont = msg,
|
||||||
|
.contLen = msgLen,
|
||||||
|
};
|
||||||
|
|
||||||
|
SRpcMsg rpcRsp = {0};
|
||||||
|
|
||||||
|
rpcSendRecv(pRpc, (SEpSet*)pMgmtEps, &rpcMsg, &rpcRsp);
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t catalogGetTableMeta(struct SCatalog* pCatalog, void *pRpc, const SEpSet* pMgmtEps, const char* pTableName, STableMeta* pTableMeta) {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t catalogGetAllMeta(struct SCatalog* pCatalog, const SEpSet* pMgmtEps, const SCatalogReq* pReq, SMetaData* pRsp) {
|
||||||
|
if (NULL == pCatalog || NULL == pMgmtEps || NULL == pReq || NULL == pRsp) {
|
||||||
|
return TSDB_CODE_CTG_INVALID_INPUT;
|
||||||
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void catalogDestroy(void) {
|
||||||
|
if (ctgMgmt.pCluster) {
|
||||||
|
taosHashCleanup(ctgMgmt.pCluster); //TBD
|
||||||
|
ctgMgmt.pCluster = NULL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -20,6 +20,8 @@ extern "C" {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include "index_fst_util.h"
|
#include "index_fst_util.h"
|
||||||
|
|
||||||
|
|
||||||
typedef struct AutomationCtx AutomationCtx;
|
typedef struct AutomationCtx AutomationCtx;
|
||||||
|
|
||||||
typedef enum AutomationType {
|
typedef enum AutomationType {
|
||||||
|
@ -42,14 +44,23 @@ typedef struct AutomationCtx {
|
||||||
} AutomationCtx;
|
} AutomationCtx;
|
||||||
|
|
||||||
|
|
||||||
|
typedef enum ValueType { FST_INT, FST_CHAR, FST_ARRAY} ValueType;
|
||||||
typedef enum StartWithStateKind { Done, Running } StartWithStateKind;
|
typedef enum StartWithStateKind { Done, Running } StartWithStateKind;
|
||||||
|
|
||||||
typedef struct StartWithStateValue {
|
typedef struct StartWithStateValue {
|
||||||
StartWithStateKind kind;
|
StartWithStateKind kind;
|
||||||
void *value;
|
ValueType type;
|
||||||
|
union {
|
||||||
|
int val;
|
||||||
|
char *ptr;
|
||||||
|
SArray *arr;
|
||||||
|
// add more type
|
||||||
|
} ;
|
||||||
} StartWithStateValue;
|
} StartWithStateValue;
|
||||||
|
|
||||||
|
StartWithStateValue *startWithStateValueDump(StartWithStateValue *sv);
|
||||||
|
|
||||||
|
|
||||||
typedef struct AutomationFunc {
|
typedef struct AutomationFunc {
|
||||||
void* (*start)(AutomationCtx *ctx) ;
|
void* (*start)(AutomationCtx *ctx) ;
|
||||||
bool (*isMatch)(AutomationCtx *ctx, void *);
|
bool (*isMatch)(AutomationCtx *ctx, void *);
|
||||||
|
|
|
@ -1322,6 +1322,7 @@ StreamWithStateResult *streamWithStateNextWith(StreamWithState *sws, StreamCallb
|
||||||
return swsResultCreate(&s, output, callback(start));
|
return swsResultCreate(&s, output, callback(start));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
SArray *nodes = taosArrayInit(8, sizeof(FstNode *));
|
||||||
while (taosArrayGetSize(sws->stack) > 0) {
|
while (taosArrayGetSize(sws->stack) > 0) {
|
||||||
StreamState *p = (StreamState *)taosArrayPop(sws->stack);
|
StreamState *p = (StreamState *)taosArrayPop(sws->stack);
|
||||||
if (p->trans >= FST_NODE_LEN(p->node) || automFuncs[aut->type].canMatch(aut, p->autState)) {
|
if (p->trans >= FST_NODE_LEN(p->node) || automFuncs[aut->type].canMatch(aut, p->autState)) {
|
||||||
|
@ -1337,8 +1338,8 @@ StreamWithStateResult *streamWithStateNextWith(StreamWithState *sws, StreamCallb
|
||||||
void* nextState = automFuncs[aut->type].accept(aut, p->autState, trn.inp);
|
void* nextState = automFuncs[aut->type].accept(aut, p->autState, trn.inp);
|
||||||
void* tState = callback(nextState);
|
void* tState = callback(nextState);
|
||||||
bool isMatch = automFuncs[aut->type].isMatch(aut, nextState);
|
bool isMatch = automFuncs[aut->type].isMatch(aut, nextState);
|
||||||
//bool isMatch = sws->aut->isMatch(nextState);
|
|
||||||
FstNode *nextNode = fstGetNode(sws->fst, trn.addr);
|
FstNode *nextNode = fstGetNode(sws->fst, trn.addr);
|
||||||
|
taosArrayPush(nodes, &nextNode);
|
||||||
taosArrayPush(sws->inp, &(trn.inp));
|
taosArrayPush(sws->inp, &(trn.inp));
|
||||||
|
|
||||||
if (FST_NODE_IS_FINAL(nextNode)) {
|
if (FST_NODE_IS_FINAL(nextNode)) {
|
||||||
|
@ -1354,26 +1355,35 @@ StreamWithStateResult *streamWithStateNextWith(StreamWithState *sws, StreamCallb
|
||||||
StreamState s2 = {.node = nextNode, .trans = 0, .out = {.null = false, .out = out}, .autState = nextState};
|
StreamState s2 = {.node = nextNode, .trans = 0, .out = {.null = false, .out = out}, .autState = nextState};
|
||||||
taosArrayPush(sws->stack, &s2);
|
taosArrayPush(sws->stack, &s2);
|
||||||
|
|
||||||
uint8_t *buf = (uint8_t *)malloc(taosArrayGetSize(sws->inp) * sizeof(uint8_t));
|
|
||||||
for (uint32_t i = 0; i < taosArrayGetSize(sws->inp); i++) {
|
size_t isz = taosArrayGetSize(sws->inp);
|
||||||
uint8_t *t = (uint8_t *)taosArrayGet(sws->inp, i);
|
uint8_t *buf = (uint8_t *)malloc(isz * sizeof(uint8_t));
|
||||||
buf[i] = *t;
|
for (uint32_t i = 0; i < isz; i++) {
|
||||||
|
buf[i] = *(uint8_t *)taosArrayGet(sws->inp, i);
|
||||||
}
|
}
|
||||||
FstSlice slice = fstSliceCreate(buf, taosArrayGetSize(sws->inp));
|
FstSlice slice = fstSliceCreate(buf, taosArrayGetSize(sws->inp));
|
||||||
if (fstBoundWithDataExceededBy(sws->endAt, &slice)) {
|
if (fstBoundWithDataExceededBy(sws->endAt, &slice)) {
|
||||||
taosArrayDestroyEx(sws->stack, streamStateDestroy);
|
taosArrayDestroyEx(sws->stack, streamStateDestroy);
|
||||||
sws->stack = (SArray *)taosArrayInit(256, sizeof(StreamState));
|
sws->stack = (SArray *)taosArrayInit(256, sizeof(StreamState));
|
||||||
|
free(buf);
|
||||||
fstSliceDestroy(&slice);
|
fstSliceDestroy(&slice);
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
if (FST_NODE_IS_FINAL(nextNode) && isMatch) {
|
if (FST_NODE_IS_FINAL(nextNode) && isMatch) {
|
||||||
FstOutput fOutput = {.null = false, .out = out + FST_NODE_FINAL_OUTPUT(nextNode)};
|
FstOutput fOutput = {.null = false, .out = out + FST_NODE_FINAL_OUTPUT(nextNode)};
|
||||||
StreamWithStateResult *result = swsResultCreate(&slice, fOutput , tState);
|
StreamWithStateResult *result = swsResultCreate(&slice, fOutput, tState);
|
||||||
|
free(buf);
|
||||||
fstSliceDestroy(&slice);
|
fstSliceDestroy(&slice);
|
||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
free(buf);
|
||||||
fstSliceDestroy(&slice);
|
fstSliceDestroy(&slice);
|
||||||
}
|
}
|
||||||
|
for (size_t i = 0; i < taosArrayGetSize(nodes); i++) {
|
||||||
|
FstNode** node = (FstNode **)taosArrayGet(nodes, i);
|
||||||
|
fstNodeDestroy(*node);
|
||||||
|
}
|
||||||
|
taosArrayDestroy(nodes);
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -16,9 +16,59 @@
|
||||||
#include "index_fst_automation.h"
|
#include "index_fst_automation.h"
|
||||||
|
|
||||||
|
|
||||||
|
StartWithStateValue *startWithStateValueCreate(StartWithStateKind kind, ValueType ty, void *val) {
|
||||||
|
StartWithStateValue *nsv = calloc(1, sizeof(StartWithStateValue));
|
||||||
|
if (nsv == NULL) { return NULL; }
|
||||||
|
|
||||||
|
nsv->kind = kind;
|
||||||
|
nsv->type = ty;
|
||||||
|
if (ty == FST_INT) {
|
||||||
|
nsv->val = *(int *)val;
|
||||||
|
} else if (ty == FST_CHAR) {
|
||||||
|
size_t len = strlen((char *)val);
|
||||||
|
nsv->ptr = (char *)calloc(1, len + 1);
|
||||||
|
memcpy(nsv->ptr, val, len);
|
||||||
|
} else if (ty == FST_ARRAY) {
|
||||||
|
//TODO,
|
||||||
|
//nsv->arr = taosArrayFromList()
|
||||||
|
}
|
||||||
|
return nsv;
|
||||||
|
}
|
||||||
|
void startWithStateValueDestroy(StartWithStateValue *sv) {
|
||||||
|
if (sv == NULL) { return; }
|
||||||
|
|
||||||
|
if (sv->type == FST_INT) {
|
||||||
|
//
|
||||||
|
} else if (sv->type == FST_CHAR) {
|
||||||
|
free(sv->ptr);
|
||||||
|
} else if (sv->type == FST_ARRAY) {
|
||||||
|
taosArrayDestroy(sv->arr);
|
||||||
|
}
|
||||||
|
free(sv);
|
||||||
|
}
|
||||||
|
StartWithStateValue *startWithStateValueDump(StartWithStateValue *sv) {
|
||||||
|
StartWithStateValue *nsv = calloc(1, sizeof(StartWithStateValue));
|
||||||
|
if (nsv == NULL) { return NULL; }
|
||||||
|
|
||||||
|
nsv->kind = sv->kind;
|
||||||
|
nsv->type= sv->type;
|
||||||
|
if (nsv->type == FST_INT) {
|
||||||
|
nsv->val = sv->val;
|
||||||
|
} else if (nsv->type == FST_CHAR) {
|
||||||
|
size_t len = strlen(sv->ptr);
|
||||||
|
nsv->ptr = (char *)calloc(1, len + 1);
|
||||||
|
memcpy(nsv->ptr, sv->ptr, len);
|
||||||
|
} else if (nsv->type == FST_ARRAY) {
|
||||||
|
}
|
||||||
|
return nsv;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
// prefix query, impl later
|
// prefix query, impl later
|
||||||
|
|
||||||
static void* prefixStart(AutomationCtx *ctx) {
|
static void* prefixStart(AutomationCtx *ctx) {
|
||||||
StartWithStateValue *data = (StartWithStateValue *)(ctx->data);
|
StartWithStateValue *data = (StartWithStateValue *)(ctx->data);
|
||||||
|
|
||||||
return data;
|
return data;
|
||||||
};
|
};
|
||||||
static bool prefixIsMatch(AutomationCtx *ctx, void *data) {
|
static bool prefixIsMatch(AutomationCtx *ctx, void *data) {
|
||||||
|
@ -86,7 +136,7 @@ AutomationCtx* automCtxCreate(void *data, AutomationType type) {
|
||||||
if (type == AUTOMATION_PREFIX) {
|
if (type == AUTOMATION_PREFIX) {
|
||||||
StartWithStateValue *swsv = (StartWithStateValue *)calloc(1, sizeof(StartWithStateValue));
|
StartWithStateValue *swsv = (StartWithStateValue *)calloc(1, sizeof(StartWithStateValue));
|
||||||
swsv->kind = Done;
|
swsv->kind = Done;
|
||||||
swsv->value = NULL;
|
//swsv->value = NULL;
|
||||||
ctx->data = (void *)swsv;
|
ctx->data = (void *)swsv;
|
||||||
} else if (type == AUTMMATION_MATCH) {
|
} else if (type == AUTMMATION_MATCH) {
|
||||||
|
|
||||||
|
|
|
@ -75,16 +75,7 @@ typedef struct {
|
||||||
SMemRowInfo *rowInfo;
|
SMemRowInfo *rowInfo;
|
||||||
} SMemRowBuilder;
|
} SMemRowBuilder;
|
||||||
|
|
||||||
typedef struct SParamInfo {
|
|
||||||
int32_t idx;
|
|
||||||
uint8_t type;
|
|
||||||
uint8_t timePrec;
|
|
||||||
int16_t bytes;
|
|
||||||
uint32_t offset;
|
|
||||||
} SParamInfo;
|
|
||||||
|
|
||||||
typedef struct STableDataBlocks {
|
typedef struct STableDataBlocks {
|
||||||
SName tableName;
|
|
||||||
int8_t tsSource; // where does the UNIX timestamp come from, server or client
|
int8_t tsSource; // where does the UNIX timestamp come from, server or client
|
||||||
bool ordered; // if current rows are ordered or not
|
bool ordered; // if current rows are ordered or not
|
||||||
int64_t vgId; // virtual group id
|
int64_t vgId; // virtual group id
|
||||||
|
@ -100,11 +91,6 @@ typedef struct STableDataBlocks {
|
||||||
STagData tagData;
|
STagData tagData;
|
||||||
|
|
||||||
SParsedDataColInfo boundColumnInfo;
|
SParsedDataColInfo boundColumnInfo;
|
||||||
|
|
||||||
// for parameter ('?') binding
|
|
||||||
uint32_t numOfAllocedParams;
|
|
||||||
uint32_t numOfParams;
|
|
||||||
SParamInfo * params;
|
|
||||||
SMemRowBuilder rowBuilder;
|
SMemRowBuilder rowBuilder;
|
||||||
} STableDataBlocks;
|
} STableDataBlocks;
|
||||||
|
|
||||||
|
@ -187,7 +173,7 @@ void destroyBoundColumnInfo(SParsedDataColInfo* pColList);
|
||||||
int32_t initMemRowBuilder(SMemRowBuilder *pBuilder, uint32_t nRows, uint32_t nCols, uint32_t nBoundCols, int32_t allNullLen);
|
int32_t initMemRowBuilder(SMemRowBuilder *pBuilder, uint32_t nRows, uint32_t nCols, uint32_t nBoundCols, int32_t allNullLen);
|
||||||
int32_t allocateMemIfNeed(STableDataBlocks *pDataBlock, int32_t rowSize, int32_t * numOfRows);
|
int32_t allocateMemIfNeed(STableDataBlocks *pDataBlock, int32_t rowSize, int32_t * numOfRows);
|
||||||
int32_t getDataBlockFromList(SHashObj* pHashList, int64_t id, int32_t size, int32_t startOffset, int32_t rowSize,
|
int32_t getDataBlockFromList(SHashObj* pHashList, int64_t id, int32_t size, int32_t startOffset, int32_t rowSize,
|
||||||
SName* name, const STableMeta* pTableMeta, STableDataBlocks** dataBlocks, SArray* pBlockList);
|
const STableMeta* pTableMeta, STableDataBlocks** dataBlocks, SArray* pBlockList);
|
||||||
int32_t mergeTableDataBlocks(SHashObj* pHashObj, int8_t schemaAttached, uint8_t payloadType, bool freeBlockMap);
|
int32_t mergeTableDataBlocks(SHashObj* pHashObj, int8_t schemaAttached, uint8_t payloadType, bool freeBlockMap);
|
||||||
|
|
||||||
#endif // TDENGINE_DATABLOCKMGT_H
|
#endif // TDENGINE_DATABLOCKMGT_H
|
||||||
|
|
|
@ -79,13 +79,13 @@ int32_t checkForInvalidExpr(SQueryStmtInfo* pQueryInfo, SMsgBuf* pMsgBuf);
|
||||||
* @param msgBufLen
|
* @param msgBufLen
|
||||||
* @return
|
* @return
|
||||||
*/
|
*/
|
||||||
int32_t qParserExtractRequestedMetaInfo(const SSqlInfo* pSqlInfo, SMetaReq* pMetaInfo, char* msg, int32_t msgBufLen);
|
int32_t qParserExtractRequestedMetaInfo(const SSqlInfo* pSqlInfo, SCatalogReq* pMetaInfo, char* msg, int32_t msgBufLen);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Destroy the meta data request structure.
|
* Destroy the meta data request structure.
|
||||||
* @param pMetaInfo
|
* @param pMetaInfo
|
||||||
*/
|
*/
|
||||||
void qParserClearupMetaRequestInfo(SMetaReq* pMetaInfo);
|
void qParserClearupMetaRequestInfo(SCatalogReq* pMetaInfo);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
|
|
|
@ -4077,18 +4077,18 @@ int32_t qParserValidateSqlNode(struct SCatalog* pCatalog, SSqlInfo* pInfo, SQuer
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
SMetaData data = {0};
|
SMetaData data = {0};
|
||||||
|
|
||||||
// TODO: check if the qnode info has been cached already
|
// TODO: check if the qnode info has been cached already
|
||||||
req.qNodeEpset = true;
|
req.qNodeRequired = true;
|
||||||
code = qParserExtractRequestedMetaInfo(pInfo, &req, msgBuf, msgBufLen);
|
code = qParserExtractRequestedMetaInfo(pInfo, &req, msgBuf, msgBufLen);
|
||||||
if (code != TSDB_CODE_SUCCESS) {
|
if (code != TSDB_CODE_SUCCESS) {
|
||||||
return code;
|
return code;
|
||||||
}
|
}
|
||||||
|
|
||||||
// load the meta data from catalog
|
// load the meta data from catalog
|
||||||
code = catalogGetMetaData(pCatalog, &req, &data);
|
code = catalogGetAllMeta(pCatalog, NULL, &req, &data);
|
||||||
if (code != TSDB_CODE_SUCCESS) {
|
if (code != TSDB_CODE_SUCCESS) {
|
||||||
return code;
|
return code;
|
||||||
}
|
}
|
||||||
|
|
|
@ -108,7 +108,7 @@ void destroyBoundColumnInfo(SParsedDataColInfo* pColList) {
|
||||||
tfree(pColList->colIdxInfo);
|
tfree(pColList->colIdxInfo);
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t createDataBlock(size_t defaultSize, int32_t rowSize, int32_t startOffset, SName* name,
|
static int32_t createDataBlock(size_t defaultSize, int32_t rowSize, int32_t startOffset,
|
||||||
const STableMeta* pTableMeta, STableDataBlocks** dataBlocks) {
|
const STableMeta* pTableMeta, STableDataBlocks** dataBlocks) {
|
||||||
STableDataBlocks* dataBuf = (STableDataBlocks*)calloc(1, sizeof(STableDataBlocks));
|
STableDataBlocks* dataBuf = (STableDataBlocks*)calloc(1, sizeof(STableDataBlocks));
|
||||||
if (dataBuf == NULL) {
|
if (dataBuf == NULL) {
|
||||||
|
@ -145,7 +145,7 @@ static int32_t createDataBlock(size_t defaultSize, int32_t rowSize, int32_t star
|
||||||
dataBuf->tsSource = -1;
|
dataBuf->tsSource = -1;
|
||||||
dataBuf->vgId = dataBuf->pTableMeta->vgId;
|
dataBuf->vgId = dataBuf->pTableMeta->vgId;
|
||||||
|
|
||||||
tNameAssign(&dataBuf->tableName, name);
|
// tNameAssign(&dataBuf->tableName, name);
|
||||||
|
|
||||||
assert(defaultSize > 0 && pTableMeta != NULL && dataBuf->pTableMeta != NULL);
|
assert(defaultSize > 0 && pTableMeta != NULL && dataBuf->pTableMeta != NULL);
|
||||||
|
|
||||||
|
@ -154,8 +154,7 @@ static int32_t createDataBlock(size_t defaultSize, int32_t rowSize, int32_t star
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t getDataBlockFromList(SHashObj* pHashList, int64_t id, int32_t size, int32_t startOffset, int32_t rowSize,
|
int32_t getDataBlockFromList(SHashObj* pHashList, int64_t id, int32_t size, int32_t startOffset, int32_t rowSize,
|
||||||
SName* name, const STableMeta* pTableMeta, STableDataBlocks** dataBlocks,
|
const STableMeta* pTableMeta, STableDataBlocks** dataBlocks, SArray* pBlockList) {
|
||||||
SArray* pBlockList) {
|
|
||||||
*dataBlocks = NULL;
|
*dataBlocks = NULL;
|
||||||
STableDataBlocks** t1 = (STableDataBlocks**)taosHashGet(pHashList, (const char*)&id, sizeof(id));
|
STableDataBlocks** t1 = (STableDataBlocks**)taosHashGet(pHashList, (const char*)&id, sizeof(id));
|
||||||
if (t1 != NULL) {
|
if (t1 != NULL) {
|
||||||
|
@ -163,7 +162,7 @@ int32_t getDataBlockFromList(SHashObj* pHashList, int64_t id, int32_t size, int3
|
||||||
}
|
}
|
||||||
|
|
||||||
if (*dataBlocks == NULL) {
|
if (*dataBlocks == NULL) {
|
||||||
int32_t ret = createDataBlock((size_t)size, rowSize, startOffset, name, pTableMeta, dataBlocks);
|
int32_t ret = createDataBlock((size_t)size, rowSize, startOffset, pTableMeta, dataBlocks);
|
||||||
if (ret != TSDB_CODE_SUCCESS) {
|
if (ret != TSDB_CODE_SUCCESS) {
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
@ -253,23 +252,13 @@ static FORCE_INLINE void convertSMemRow(SMemRow dest, SMemRow src, STableDataBlo
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void destroyDataBlock(STableDataBlocks* pDataBlock, bool removeMeta) {
|
void destroyDataBlock(STableDataBlocks* pDataBlock) {
|
||||||
if (pDataBlock == NULL) {
|
if (pDataBlock == NULL) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
tfree(pDataBlock->pData);
|
tfree(pDataBlock->pData);
|
||||||
|
|
||||||
if (removeMeta) {
|
|
||||||
char name[TSDB_TABLE_FNAME_LEN] = {0};
|
|
||||||
tNameExtractFullName(&pDataBlock->tableName, name);
|
|
||||||
|
|
||||||
// taosHashRemove(tscTableMetaMap, name, strnlen(name, TSDB_TABLE_FNAME_LEN));
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!pDataBlock->cloned) {
|
if (!pDataBlock->cloned) {
|
||||||
tfree(pDataBlock->params);
|
|
||||||
|
|
||||||
// free the refcount for metermeta
|
// free the refcount for metermeta
|
||||||
if (pDataBlock->pTableMeta != NULL) {
|
if (pDataBlock->pTableMeta != NULL) {
|
||||||
tfree(pDataBlock->pTableMeta);
|
tfree(pDataBlock->pTableMeta);
|
||||||
|
@ -277,7 +266,6 @@ void destroyDataBlock(STableDataBlocks* pDataBlock, bool removeMeta) {
|
||||||
|
|
||||||
destroyBoundColumnInfo(&pDataBlock->boundColumnInfo);
|
destroyBoundColumnInfo(&pDataBlock->boundColumnInfo);
|
||||||
}
|
}
|
||||||
|
|
||||||
tfree(pDataBlock);
|
tfree(pDataBlock);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -289,7 +277,7 @@ void* destroyBlockArrayList(SArray* pDataBlockList) {
|
||||||
size_t size = taosArrayGetSize(pDataBlockList);
|
size_t size = taosArrayGetSize(pDataBlockList);
|
||||||
for (int32_t i = 0; i < size; i++) {
|
for (int32_t i = 0; i < size; i++) {
|
||||||
void* d = taosArrayGetP(pDataBlockList, i);
|
void* d = taosArrayGetP(pDataBlockList, i);
|
||||||
destroyDataBlock(d, false);
|
destroyDataBlock(d);
|
||||||
}
|
}
|
||||||
|
|
||||||
taosArrayDestroy(pDataBlockList);
|
taosArrayDestroy(pDataBlockList);
|
||||||
|
@ -505,7 +493,7 @@ int32_t mergeTableDataBlocks(SHashObj* pHashObj, int8_t schemaAttached, uint8_t
|
||||||
if (pBlocks->numOfRows > 0) {
|
if (pBlocks->numOfRows > 0) {
|
||||||
STableDataBlocks* dataBuf = NULL;
|
STableDataBlocks* dataBuf = NULL;
|
||||||
int32_t ret = getDataBlockFromList(pVnodeDataBlockHashList, pOneTableBlock->vgId, TSDB_PAYLOAD_SIZE,
|
int32_t ret = getDataBlockFromList(pVnodeDataBlockHashList, pOneTableBlock->vgId, TSDB_PAYLOAD_SIZE,
|
||||||
INSERT_HEAD_SIZE, 0, &pOneTableBlock->tableName, pOneTableBlock->pTableMeta, &dataBuf, pVnodeDataBlockList);
|
INSERT_HEAD_SIZE, 0, pOneTableBlock->pTableMeta, &dataBuf, pVnodeDataBlockList);
|
||||||
if (ret != TSDB_CODE_SUCCESS) {
|
if (ret != TSDB_CODE_SUCCESS) {
|
||||||
taosHashCleanup(pVnodeDataBlockHashList);
|
taosHashCleanup(pVnodeDataBlockHashList);
|
||||||
destroyBlockArrayList(pVnodeDataBlockList);
|
destroyBlockArrayList(pVnodeDataBlockList);
|
||||||
|
@ -580,7 +568,6 @@ int32_t mergeTableDataBlocks(SHashObj* pHashObj, int8_t schemaAttached, uint8_t
|
||||||
extractTableNameList(pHashObj, freeBlockMap);
|
extractTableNameList(pHashObj, freeBlockMap);
|
||||||
|
|
||||||
// free the table data blocks;
|
// free the table data blocks;
|
||||||
// pInsertParam->pDataBlocks = pVnodeDataBlockList;
|
|
||||||
taosHashCleanup(pVnodeDataBlockHashList);
|
taosHashCleanup(pVnodeDataBlockHashList);
|
||||||
tfree(blkKeyInfo.pKeyTuple);
|
tfree(blkKeyInfo.pKeyTuple);
|
||||||
|
|
||||||
|
|
|
@ -167,25 +167,26 @@ static int32_t skipInsertInto(SInsertParseContext* pCxt) {
|
||||||
|
|
||||||
static int32_t buildTableName(SInsertParseContext* pCxt, SToken* pStname, SArray* tableNameList) {
|
static int32_t buildTableName(SInsertParseContext* pCxt, SToken* pStname, SArray* tableNameList) {
|
||||||
if (parserValidateIdToken(pStname) != TSDB_CODE_SUCCESS) {
|
if (parserValidateIdToken(pStname) != TSDB_CODE_SUCCESS) {
|
||||||
return buildInvalidOperationMsg(&pCxt->msg, "invalid table name");
|
return buildSyntaxErrMsg(&pCxt->msg, "invalid table name", pStname->z);
|
||||||
}
|
}
|
||||||
|
|
||||||
SName name = {0};
|
SName name = {0};
|
||||||
strndequote(name.tname, pStname->z, pStname->n);
|
strcpy(name.dbname, pCxt->pComCxt->pDbname);
|
||||||
|
strncpy(name.tname, pStname->z, pStname->n);
|
||||||
taosArrayPush(tableNameList, &name);
|
taosArrayPush(tableNameList, &name);
|
||||||
|
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t buildMetaReq(SInsertParseContext* pCxt, SToken* pStname, SMetaReq* pMetaReq) {
|
static int32_t buildMetaReq(SInsertParseContext* pCxt, SToken* pStname, SCatalogReq* pMetaReq) {
|
||||||
pMetaReq->pTableName = taosArrayInit(4, sizeof(SName));
|
pMetaReq->pTableName = taosArrayInit(4, sizeof(SName));
|
||||||
return buildTableName(pCxt, pStname, pMetaReq->pTableName);
|
return buildTableName(pCxt, pStname, pMetaReq->pTableName);
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t getTableMeta(SInsertParseContext* pCxt, SToken* pTname) {
|
static int32_t getTableMeta(SInsertParseContext* pCxt, SToken* pTname) {
|
||||||
SMetaReq req;
|
SCatalogReq req;
|
||||||
CHECK_CODE(buildMetaReq(pCxt, pTname, &req));
|
CHECK_CODE(buildMetaReq(pCxt, pTname, &req));
|
||||||
CHECK_CODE(catalogGetMetaData(pCxt->pCatalog, &req, &pCxt->meta));
|
CHECK_CODE(catalogGetTableMeta(pCxt->pCatalog, NULL, NULL, NULL, &pCxt->meta)); //TODO
|
||||||
pCxt->pTableMeta = (STableMeta*)taosArrayGetP(pCxt->meta.pTableMeta, 0);
|
pCxt->pTableMeta = (STableMeta*)taosArrayGetP(pCxt->meta.pTableMeta, 0);
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
@ -686,7 +687,6 @@ static int parseOneRow(SInsertParseContext* pCxt, STableDataBlocks* pDataBlocks,
|
||||||
// 1. set the parsed value from sql string
|
// 1. set the parsed value from sql string
|
||||||
for (int i = 0; i < spd->numOfBound; ++i) {
|
for (int i = 0; i < spd->numOfBound; ++i) {
|
||||||
NEXT_TOKEN(pCxt->pSql, sToken);
|
NEXT_TOKEN(pCxt->pSql, sToken);
|
||||||
// todo bind param
|
|
||||||
SSchema *pSchema = &schema[spd->boundedColumns[i]];
|
SSchema *pSchema = &schema[spd->boundedColumns[i]];
|
||||||
param.schema = pSchema;
|
param.schema = pSchema;
|
||||||
param.compareStat = pBuilder->compareStat;
|
param.compareStat = pBuilder->compareStat;
|
||||||
|
@ -770,14 +770,6 @@ static int32_t parseValuesClause(SInsertParseContext* pCxt, STableDataBlocks* da
|
||||||
int32_t numOfRows = 0;
|
int32_t numOfRows = 0;
|
||||||
CHECK_CODE(parseValues(pCxt, dataBuf, maxNumOfRows, &numOfRows));
|
CHECK_CODE(parseValues(pCxt, dataBuf, maxNumOfRows, &numOfRows));
|
||||||
|
|
||||||
for (uint32_t i = 0; i < dataBuf->numOfParams; ++i) {
|
|
||||||
SParamInfo *param = dataBuf->params + i;
|
|
||||||
if (param->idx == -1) {
|
|
||||||
// param->idx = pInsertParam->numOfParams++;
|
|
||||||
param->offset -= sizeof(SSubmitBlk);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
SSubmitBlk *pBlocks = (SSubmitBlk *)(dataBuf->pData);
|
SSubmitBlk *pBlocks = (SSubmitBlk *)(dataBuf->pData);
|
||||||
if (TSDB_CODE_SUCCESS != setBlockInfo(pBlocks, dataBuf->pTableMeta, numOfRows)) {
|
if (TSDB_CODE_SUCCESS != setBlockInfo(pBlocks, dataBuf->pTableMeta, numOfRows)) {
|
||||||
return buildInvalidOperationMsg(&pCxt->msg, "too many rows in sql, total number of rows should be less than 32767");
|
return buildInvalidOperationMsg(&pCxt->msg, "too many rows in sql, total number of rows should be less than 32767");
|
||||||
|
@ -815,12 +807,12 @@ static int32_t parseInsertBody(SInsertParseContext* pCxt) {
|
||||||
CHECK_CODE(parseUsingClause(pCxt, &tbnameToken));
|
CHECK_CODE(parseUsingClause(pCxt, &tbnameToken));
|
||||||
NEXT_TOKEN(pCxt->pSql, sToken);
|
NEXT_TOKEN(pCxt->pSql, sToken);
|
||||||
} else {
|
} else {
|
||||||
CHECK_CODE(getTableMeta(pCxt, &sToken));
|
CHECK_CODE(getTableMeta(pCxt, &tbnameToken));
|
||||||
}
|
}
|
||||||
|
|
||||||
STableDataBlocks *dataBuf = NULL;
|
STableDataBlocks *dataBuf = NULL;
|
||||||
CHECK_CODE(getDataBlockFromList(pCxt->pTableBlockHashObj, pCxt->pTableMeta->uid, TSDB_DEFAULT_PAYLOAD_SIZE,
|
CHECK_CODE(getDataBlockFromList(pCxt->pTableBlockHashObj, pCxt->pTableMeta->uid, TSDB_DEFAULT_PAYLOAD_SIZE,
|
||||||
sizeof(SSubmitBlk), getTableInfo(pCxt->pTableMeta).rowSize, NULL/* tbname */, pCxt->pTableMeta, &dataBuf, NULL));
|
sizeof(SSubmitBlk), getTableInfo(pCxt->pTableMeta).rowSize, pCxt->pTableMeta, &dataBuf, NULL));
|
||||||
|
|
||||||
if (TK_LP == sToken.type) {
|
if (TK_LP == sToken.type) {
|
||||||
// pSql -> field1_name, ...)
|
// pSql -> field1_name, ...)
|
||||||
|
@ -831,6 +823,7 @@ static int32_t parseInsertBody(SInsertParseContext* pCxt) {
|
||||||
if (TK_VALUES == sToken.type) {
|
if (TK_VALUES == sToken.type) {
|
||||||
// pSql -> (field1_value, ...) [(field1_value2, ...) ...]
|
// pSql -> (field1_value, ...) [(field1_value2, ...) ...]
|
||||||
CHECK_CODE(parseValuesClause(pCxt, dataBuf));
|
CHECK_CODE(parseValuesClause(pCxt, dataBuf));
|
||||||
|
pCxt->pOutput->insertType = TSDB_QUERY_TYPE_INSERT;
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -842,6 +835,7 @@ static int32_t parseInsertBody(SInsertParseContext* pCxt) {
|
||||||
return buildSyntaxErrMsg(&pCxt->msg, "file path is required following keyword FILE", sToken.z);
|
return buildSyntaxErrMsg(&pCxt->msg, "file path is required following keyword FILE", sToken.z);
|
||||||
}
|
}
|
||||||
// todo
|
// todo
|
||||||
|
pCxt->pOutput->insertType = TSDB_QUERY_TYPE_FILE_INSERT;
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -867,13 +861,15 @@ int32_t parseInsertSql(SParseContext* pContext, SInsertStmtInfo** pInfo) {
|
||||||
.pComCxt = pContext,
|
.pComCxt = pContext,
|
||||||
.pSql = pContext->pSql,
|
.pSql = pContext->pSql,
|
||||||
.msg = {.buf = pContext->pMsg, .len = pContext->msgLen},
|
.msg = {.buf = pContext->pMsg, .len = pContext->msgLen},
|
||||||
.pCatalog = getCatalogHandle(pContext->pEpSet),
|
.pCatalog = NULL,
|
||||||
.pTableMeta = NULL,
|
.pTableMeta = NULL,
|
||||||
.pTableBlockHashObj = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), true, false),
|
.pTableBlockHashObj = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), true, false),
|
||||||
.totalNum = 0,
|
.totalNum = 0,
|
||||||
.pOutput = *pInfo
|
.pOutput = *pInfo
|
||||||
};
|
};
|
||||||
|
|
||||||
|
CHECK_CODE(catalogGetHandle(NULL, &context.pCatalog)); //TODO
|
||||||
|
|
||||||
if (NULL == context.pTableBlockHashObj) {
|
if (NULL == context.pTableBlockHashObj) {
|
||||||
return TSDB_CODE_TSC_OUT_OF_MEMORY;
|
return TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
}
|
}
|
||||||
|
|
|
@ -43,7 +43,12 @@ int32_t qParseQuerySql(const char* pStr, size_t length, struct SQueryStmtInfo**
|
||||||
return TSDB_CODE_TSC_SQL_SYNTAX_ERROR;
|
return TSDB_CODE_TSC_SQL_SYNTAX_ERROR;
|
||||||
}
|
}
|
||||||
|
|
||||||
struct SCatalog* pCatalog = getCatalogHandle(NULL);
|
struct SCatalog* pCatalog = NULL;
|
||||||
|
int32_t code = catalogGetHandle(NULL, &pCatalog);
|
||||||
|
if (code) {
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
|
||||||
return qParserValidateSqlNode(pCatalog, &info, *pQueryInfo, id, msg, msgLen);
|
return qParserValidateSqlNode(pCatalog, &info, *pQueryInfo, id, msg, msgLen);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -132,7 +137,7 @@ static void freePtrElem(void* p) {
|
||||||
tfree(*(char**)p);
|
tfree(*(char**)p);
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t qParserExtractRequestedMetaInfo(const SSqlInfo* pSqlInfo, SMetaReq* pMetaInfo, char* msg, int32_t msgBufLen) {
|
int32_t qParserExtractRequestedMetaInfo(const SSqlInfo* pSqlInfo, SCatalogReq* pMetaInfo, char* msg, int32_t msgBufLen) {
|
||||||
int32_t code = TSDB_CODE_SUCCESS;
|
int32_t code = TSDB_CODE_SUCCESS;
|
||||||
SMsgBuf msgBuf = {.buf = msg, .len = msgBufLen};
|
SMsgBuf msgBuf = {.buf = msg, .len = msgBufLen};
|
||||||
|
|
||||||
|
@ -189,7 +194,7 @@ int32_t qParserExtractRequestedMetaInfo(const SSqlInfo* pSqlInfo, SMetaReq* pMet
|
||||||
return code;
|
return code;
|
||||||
}
|
}
|
||||||
|
|
||||||
void qParserClearupMetaRequestInfo(SMetaReq* pMetaReq) {
|
void qParserClearupMetaRequestInfo(SCatalogReq* pMetaReq) {
|
||||||
if (pMetaReq == NULL) {
|
if (pMetaReq == NULL) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
|
@ -1448,23 +1448,6 @@ void* vgroupInfoClear(SVgroupsInfo *vgroupList) {
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
char* serializeTagData(STagData* pTagData, char* pMsg) {
|
|
||||||
int32_t n = (int32_t) strlen(pTagData->name);
|
|
||||||
*(int32_t*) pMsg = htonl(n);
|
|
||||||
pMsg += sizeof(n);
|
|
||||||
|
|
||||||
memcpy(pMsg, pTagData->name, n);
|
|
||||||
pMsg += n;
|
|
||||||
|
|
||||||
*(int32_t*)pMsg = htonl(pTagData->dataLen);
|
|
||||||
pMsg += sizeof(int32_t);
|
|
||||||
|
|
||||||
memcpy(pMsg, pTagData->data, pTagData->dataLen);
|
|
||||||
pMsg += pTagData->dataLen;
|
|
||||||
|
|
||||||
return pMsg;
|
|
||||||
}
|
|
||||||
|
|
||||||
int32_t copyTagData(STagData* dst, const STagData* src) {
|
int32_t copyTagData(STagData* dst, const STagData* src) {
|
||||||
dst->dataLen = src->dataLen;
|
dst->dataLen = src->dataLen;
|
||||||
tstrncpy(dst->name, src->name, tListLen(dst->name));
|
tstrncpy(dst->name, src->name, tListLen(dst->name));
|
||||||
|
|
|
@ -36,51 +36,67 @@ namespace {
|
||||||
// [...];
|
// [...];
|
||||||
class InsertTest : public Test {
|
class InsertTest : public Test {
|
||||||
protected:
|
protected:
|
||||||
|
void setDatabase(const string& db) {
|
||||||
|
db_ = db;
|
||||||
|
}
|
||||||
|
|
||||||
void bind(const char* sql) {
|
void bind(const char* sql) {
|
||||||
reset();
|
reset();
|
||||||
cxt.pSql = sql;
|
cxt_.sqlLen = strlen(sql);
|
||||||
cxt.sqlLen = strlen(sql);
|
strcpy(sqlBuf_, sql);
|
||||||
|
sqlBuf_[cxt_.sqlLen] = '\0';
|
||||||
|
cxt_.pSql = sqlBuf_;
|
||||||
|
cxt_.pDbname = db_.c_str();
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t run() {
|
int32_t run() {
|
||||||
code = parseInsertSql(&cxt, &res);
|
code_ = parseInsertSql(&cxt_, &res_);
|
||||||
if (code != TSDB_CODE_SUCCESS) {
|
if (code_ != TSDB_CODE_SUCCESS) {
|
||||||
cout << "code:" << toString(code) << ", msg:" << errMagBuf << endl;
|
cout << "code:" << toString(code_) << ", msg:" << errMagBuf_ << endl;
|
||||||
}
|
}
|
||||||
return code;
|
return code_;
|
||||||
}
|
}
|
||||||
|
|
||||||
SInsertStmtInfo* reslut() {
|
SInsertStmtInfo* reslut() {
|
||||||
return res;
|
return res_;
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
static const int max_err_len = 1024;
|
static const int max_err_len = 1024;
|
||||||
|
static const int max_sql_len = 1024 * 1024;
|
||||||
|
|
||||||
void reset() {
|
void reset() {
|
||||||
memset(&cxt, 0, sizeof(cxt));
|
memset(&cxt_, 0, sizeof(cxt_));
|
||||||
memset(errMagBuf, 0, max_err_len);
|
memset(errMagBuf_, 0, max_err_len);
|
||||||
cxt.pMsg = errMagBuf;
|
cxt_.pMsg = errMagBuf_;
|
||||||
cxt.msgLen = max_err_len;
|
cxt_.msgLen = max_err_len;
|
||||||
code = TSDB_CODE_SUCCESS;
|
code_ = TSDB_CODE_SUCCESS;
|
||||||
res = nullptr;
|
res_ = nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
char errMagBuf[max_err_len];
|
string db_;
|
||||||
SParseContext cxt;
|
char errMagBuf_[max_err_len];
|
||||||
int32_t code;
|
char sqlBuf_[max_sql_len];
|
||||||
SInsertStmtInfo* res;
|
SParseContext cxt_;
|
||||||
|
int32_t code_;
|
||||||
|
SInsertStmtInfo* res_;
|
||||||
};
|
};
|
||||||
|
|
||||||
// INSERT INTO tb_name VALUES (field1_value, ...)
|
// INSERT INTO tb_name VALUES (field1_value, ...)
|
||||||
TEST_F(InsertTest, simpleTest) {
|
TEST_F(InsertTest, simpleTest) {
|
||||||
bind("insert into .. values (...)");
|
setDatabase("test");
|
||||||
|
|
||||||
|
bind("insert into t1 values (now, 1, \"wxy\")");
|
||||||
ASSERT_EQ(run(), TSDB_CODE_SUCCESS);
|
ASSERT_EQ(run(), TSDB_CODE_SUCCESS);
|
||||||
SInsertStmtInfo* res = reslut();
|
SInsertStmtInfo* res = reslut();
|
||||||
// todo check
|
// todo check
|
||||||
|
ASSERT_EQ(res->insertType, TSDB_QUERY_TYPE_INSERT);
|
||||||
|
// ASSERT_EQ(taosArrayGetSize(res->pDataBlocks), 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_F(InsertTest, toleranceTest) {
|
TEST_F(InsertTest, toleranceTest) {
|
||||||
|
setDatabase("test");
|
||||||
|
|
||||||
bind("insert into");
|
bind("insert into");
|
||||||
ASSERT_NE(run(), TSDB_CODE_SUCCESS);
|
ASSERT_NE(run(), TSDB_CODE_SUCCESS);
|
||||||
bind("insert into t");
|
bind("insert into t");
|
||||||
|
|
|
@ -17,29 +17,30 @@
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
void generateTestT1(MockCatalogService* mcs) {
|
||||||
|
ITableBuilder& builder = mcs->createTableBuilder("test", "t1", TSDB_NORMAL_TABLE, 3)
|
||||||
|
.setPrecision(TSDB_TIME_PRECISION_MILLI).setVgid(1).addColumn("ts", TSDB_DATA_TYPE_TIMESTAMP)
|
||||||
|
.addColumn("c1", TSDB_DATA_TYPE_INT).addColumn("c2", TSDB_DATA_TYPE_BINARY, 10);
|
||||||
|
builder.done();
|
||||||
|
}
|
||||||
|
|
||||||
|
void generateTestST1(MockCatalogService* mcs) {
|
||||||
|
ITableBuilder& builder = mcs->createTableBuilder("test", "st1", TSDB_SUPER_TABLE, 3, 2)
|
||||||
|
.setPrecision(TSDB_TIME_PRECISION_MILLI).addColumn("ts", TSDB_DATA_TYPE_TIMESTAMP)
|
||||||
|
.addTag("tag1", TSDB_DATA_TYPE_INT).addTag("tag2", TSDB_DATA_TYPE_BINARY, 10)
|
||||||
|
.addColumn("c1", TSDB_DATA_TYPE_INT).addColumn("c2", TSDB_DATA_TYPE_BINARY, 10);
|
||||||
|
builder.done();
|
||||||
|
mcs->createSubTable("test", "st1", "st1s1", 1);
|
||||||
|
mcs->createSubTable("test", "st1", "st1s2", 2);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
void generateMetaData(MockCatalogService* mcs) {
|
void generateMetaData(MockCatalogService* mcs) {
|
||||||
{
|
generateTestT1(mcs);
|
||||||
ITableBuilder& builder = mcs->createTableBuilder("test", "t1", TSDB_NORMAL_TABLE, MockCatalogService::numOfDataTypes)
|
generateTestST1(mcs);
|
||||||
.setPrecision(TSDB_TIME_PRECISION_MILLI).setVgid(1).addColumn("ts", TSDB_DATA_TYPE_TIMESTAMP);
|
|
||||||
for (int32_t i = 0; i < MockCatalogService::numOfDataTypes; ++i) {
|
|
||||||
if (TSDB_DATA_TYPE_NULL == tDataTypes[i].type) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
builder = builder.addColumn("c" + std::to_string(i + 1), tDataTypes[i].type);
|
|
||||||
}
|
|
||||||
builder.done();
|
|
||||||
}
|
|
||||||
{
|
|
||||||
ITableBuilder& builder = mcs->createTableBuilder("test", "st1", TSDB_SUPER_TABLE, MockCatalogService::numOfDataTypes, 2)
|
|
||||||
.setPrecision(TSDB_TIME_PRECISION_MILLI).setVgid(2).addColumn("ts", TSDB_DATA_TYPE_TIMESTAMP);
|
|
||||||
for (int32_t i = 0; i < MockCatalogService::numOfDataTypes; ++i) {
|
|
||||||
if (TSDB_DATA_TYPE_NULL == tDataTypes[i].type) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
builder = builder.addColumn("c" + std::to_string(i + 1), tDataTypes[i].type);
|
|
||||||
}
|
|
||||||
builder.done();
|
|
||||||
}
|
|
||||||
mcs->showTables();
|
mcs->showTables();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -47,6 +48,6 @@ struct SCatalog* getCatalogHandle(const SEpSet* pMgmtEps) {
|
||||||
return mockCatalogService->getCatalogHandle(pMgmtEps);
|
return mockCatalogService->getCatalogHandle(pMgmtEps);
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t catalogGetMetaData(struct SCatalog* pCatalog, const SMetaReq* pMetaReq, SMetaData* pMetaData) {
|
int32_t catalogGetMetaData(struct SCatalog* pCatalog, const SCatalogReq* pMetaReq, SMetaData* pMetaData) {
|
||||||
return mockCatalogService->catalogGetMetaData(pCatalog, pMetaReq, pMetaData);
|
return mockCatalogService->catalogGetMetaData(pCatalog, pMetaReq, pMetaData);
|
||||||
}
|
}
|
||||||
|
|
|
@ -22,6 +22,6 @@ void generateMetaData(MockCatalogService* mcs);
|
||||||
|
|
||||||
// mock
|
// mock
|
||||||
struct SCatalog* getCatalogHandle(const SEpSet* pMgmtEps);
|
struct SCatalog* getCatalogHandle(const SEpSet* pMgmtEps);
|
||||||
int32_t catalogGetMetaData(struct SCatalog* pCatalog, const SMetaReq* pMetaReq, SMetaData* pMetaData);
|
int32_t catalogGetMetaData(struct SCatalog* pCatalog, const SCatalogReq* pMetaReq, SMetaData* pMetaData);
|
||||||
|
|
||||||
#endif // MOCK_CATALOG_H
|
#endif // MOCK_CATALOG_H
|
||||||
|
|
|
@ -19,6 +19,7 @@
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <map>
|
#include <map>
|
||||||
|
|
||||||
|
#include "tname.h"
|
||||||
#include "ttypes.h"
|
#include "ttypes.h"
|
||||||
|
|
||||||
std::unique_ptr<MockCatalogService> mockCatalogService;
|
std::unique_ptr<MockCatalogService> mockCatalogService;
|
||||||
|
@ -82,12 +83,26 @@ public:
|
||||||
MockCatalogServiceImpl() {
|
MockCatalogServiceImpl() {
|
||||||
}
|
}
|
||||||
|
|
||||||
struct SCatalog* getCatalogHandle(const SEpSet* pMgmtEps) {
|
struct SCatalog* getCatalogHandle(const SEpSet* pMgmtEps) const {
|
||||||
return (struct SCatalog*)0x01;
|
return (struct SCatalog*)0x01;
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t catalogGetMetaData(struct SCatalog* pCatalog, const SMetaReq* pMetaReq, SMetaData* pMetaData) {
|
int32_t catalogGetMetaData(struct SCatalog* pCatalog, const SCatalogReq* pMetaReq, SMetaData* pMetaData) const {
|
||||||
return 0;
|
assert(nullptr != pMetaReq && 1 == taosArrayGetSize(pMetaReq->pTableName));
|
||||||
|
SName* fullName = (SName*)taosArrayGet(pMetaReq->pTableName, 0);
|
||||||
|
std::unique_ptr<STableMeta> table;
|
||||||
|
int32_t code = copyTableMeta(fullName->dbname, fullName->tname, &table);
|
||||||
|
if (TSDB_CODE_SUCCESS != code) {
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
std::unique_ptr<SArray> tables((SArray*)taosArrayInit(1, sizeof(STableMeta*)));
|
||||||
|
if (!tables) {
|
||||||
|
return TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
}
|
||||||
|
STableMeta* elem = table.release();
|
||||||
|
taosArrayPush(tables.get(), &elem);
|
||||||
|
pMetaData->pTableMeta = tables.release();
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
TableBuilder& createTableBuilder(const std::string& db, const std::string& tbname, int8_t tableType, int32_t numOfColumns, int32_t numOfTags) {
|
TableBuilder& createTableBuilder(const std::string& db, const std::string& tbname, int8_t tableType, int32_t numOfColumns, int32_t numOfTags) {
|
||||||
|
@ -97,6 +112,15 @@ public:
|
||||||
return *(builder_.get());
|
return *(builder_.get());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void createSubTable(const std::string& db, const std::string& stbname, const std::string& tbname, int16_t vgid) {
|
||||||
|
std::unique_ptr<STableMeta> table;
|
||||||
|
if (TSDB_CODE_SUCCESS != copyTableMeta(db, stbname, &table)) {
|
||||||
|
throw std::runtime_error("copyTableMeta failed");
|
||||||
|
}
|
||||||
|
meta_[db][tbname].reset(table.release());
|
||||||
|
meta_[db][tbname]->uid = id_++;
|
||||||
|
}
|
||||||
|
|
||||||
void showTables() const {
|
void showTables() const {
|
||||||
// number of forward fills
|
// number of forward fills
|
||||||
#define NOF(n) ((n) / 2)
|
#define NOF(n) ((n) / 2)
|
||||||
|
@ -120,20 +144,35 @@ public:
|
||||||
#define SL(sn, in) std::setfill('=') << std::setw((sn) * (SFL + 1) + (in) * (IFL + 1)) << "" << std::setfill(' ')
|
#define SL(sn, in) std::setfill('=') << std::setw((sn) * (SFL + 1) + (in) * (IFL + 1)) << "" << std::setfill(' ')
|
||||||
|
|
||||||
for (const auto& db : meta_) {
|
for (const auto& db : meta_) {
|
||||||
std::cout << SH("Database") << SH("Table") << SH("Type") << SH("Precision") << IH(std::string("Vgid")) << std::endl;
|
std::cout << "Databse:" << db.first << std::endl;
|
||||||
std::cout << SL(4, 1) << std::endl;
|
std::cout << SH("Table") << SH("Type") << SH("Precision") << IH("Vgid") << std::endl;
|
||||||
|
std::cout << SL(3, 1) << std::endl;
|
||||||
for (const auto& table : db.second) {
|
for (const auto& table : db.second) {
|
||||||
std::cout << SF(db.first) << SF(table.first) << SF(ttToString(table.second->tableType)) << SF(pToString(table.second->tableInfo.precision)) << IF(table.second->vgId) << std::endl;
|
std::cout << SF(table.first) << SF(ttToString(table.second->tableType)) << SF(pToString(table.second->tableInfo.precision)) << IF(table.second->vgId) << std::endl;
|
||||||
// int16_t numOfFields = table.second->tableInfo.numOfTags + table.second->tableInfo.numOfColumns;
|
}
|
||||||
// for (int16_t i = 0; i < numOfFields; ++i) {
|
std::cout << std::endl;
|
||||||
// const SSchema* schema = table.second->schema + i;
|
}
|
||||||
// std::cout << schema->name << " " << schema->type << " " << schema->bytes << std::endl;
|
|
||||||
// }
|
for (const auto& db : meta_) {
|
||||||
|
for (const auto& table : db.second) {
|
||||||
|
std::cout << "Table:" << table.first << std::endl;
|
||||||
|
std::cout << SH("Field") << SH("Type") << SH("DataType") << IH("Bytes") << std::endl;
|
||||||
|
std::cout << SL(3, 1) << std::endl;
|
||||||
|
int16_t numOfTags = table.second->tableInfo.numOfTags;
|
||||||
|
int16_t numOfFields = numOfTags + table.second->tableInfo.numOfColumns;
|
||||||
|
for (int16_t i = 0; i < numOfFields; ++i) {
|
||||||
|
const SSchema* schema = table.second->schema + i;
|
||||||
|
std::cout << SF(std::string(schema->name)) << SH(ftToString(i, numOfTags)) << SH(dtToString(schema->type)) << IF(schema->bytes) << std::endl;
|
||||||
|
}
|
||||||
|
std::cout << std::endl;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
typedef std::map<std::string, std::shared_ptr<STableMeta> > TableMetaCache;
|
||||||
|
typedef std::map<std::string, TableMetaCache> DbMetaCache;
|
||||||
|
|
||||||
std::string ttToString(int8_t tableType) const {
|
std::string ttToString(int8_t tableType) const {
|
||||||
switch (tableType) {
|
switch (tableType) {
|
||||||
case TSDB_SUPER_TABLE:
|
case TSDB_SUPER_TABLE:
|
||||||
|
@ -160,9 +199,43 @@ private:
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::string dtToString(int8_t type) const {
|
||||||
|
return tDataTypes[type].name;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string ftToString(int16_t colid, int16_t numOfTags) const {
|
||||||
|
return (0 == colid ? "column" : (colid <= numOfTags ? "tag" : "column"));
|
||||||
|
}
|
||||||
|
|
||||||
|
std::shared_ptr<STableMeta> getTableMeta(const std::string& db, const std::string& tbname) const {
|
||||||
|
DbMetaCache::const_iterator it = meta_.find(db);
|
||||||
|
if (meta_.end() == it) {
|
||||||
|
return std::shared_ptr<STableMeta>();
|
||||||
|
}
|
||||||
|
TableMetaCache::const_iterator tit = it->second.find(tbname);
|
||||||
|
if (it->second.end() == tit) {
|
||||||
|
return std::shared_ptr<STableMeta>();
|
||||||
|
}
|
||||||
|
return tit->second;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t copyTableMeta(const std::string& db, const std::string& tbname, std::unique_ptr<STableMeta>* dst) const {
|
||||||
|
std::shared_ptr<STableMeta> src = getTableMeta(db, tbname);
|
||||||
|
if (!src) {
|
||||||
|
return TSDB_CODE_TSC_INVALID_TABLE_NAME;
|
||||||
|
}
|
||||||
|
int32_t len = sizeof(STableMeta) + sizeof(SSchema) * (src->tableInfo.numOfTags + src->tableInfo.numOfColumns);
|
||||||
|
dst->reset((STableMeta*)std::calloc(1, len));
|
||||||
|
if (!dst) {
|
||||||
|
return TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
}
|
||||||
|
memcpy(dst->get(), src.get(), len);
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
uint64_t id_;
|
uint64_t id_;
|
||||||
std::unique_ptr<TableBuilder> builder_;
|
std::unique_ptr<TableBuilder> builder_;
|
||||||
std::map<std::string, std::map<std::string, std::shared_ptr<STableMeta> > > meta_;
|
DbMetaCache meta_;
|
||||||
};
|
};
|
||||||
|
|
||||||
MockCatalogService::MockCatalogService() : impl_(new MockCatalogServiceImpl()) {
|
MockCatalogService::MockCatalogService() : impl_(new MockCatalogServiceImpl()) {
|
||||||
|
@ -171,11 +244,11 @@ MockCatalogService::MockCatalogService() : impl_(new MockCatalogServiceImpl()) {
|
||||||
MockCatalogService::~MockCatalogService() {
|
MockCatalogService::~MockCatalogService() {
|
||||||
}
|
}
|
||||||
|
|
||||||
struct SCatalog* MockCatalogService::getCatalogHandle(const SEpSet* pMgmtEps) {
|
struct SCatalog* MockCatalogService::getCatalogHandle(const SEpSet* pMgmtEps) const {
|
||||||
return impl_->getCatalogHandle(pMgmtEps);
|
return impl_->getCatalogHandle(pMgmtEps);
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t MockCatalogService::catalogGetMetaData(struct SCatalog* pCatalog, const SMetaReq* pMetaReq, SMetaData* pMetaData) {
|
int32_t MockCatalogService::catalogGetMetaData(struct SCatalog* pCatalog, const SCatalogReq* pMetaReq, SMetaData* pMetaData) const {
|
||||||
return impl_->catalogGetMetaData(pCatalog, pMetaReq, pMetaData);
|
return impl_->catalogGetMetaData(pCatalog, pMetaReq, pMetaData);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -183,6 +256,10 @@ ITableBuilder& MockCatalogService::createTableBuilder(const std::string& db, con
|
||||||
return impl_->createTableBuilder(db, tbname, tableType, numOfColumns, numOfTags);
|
return impl_->createTableBuilder(db, tbname, tableType, numOfColumns, numOfTags);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void MockCatalogService::createSubTable(const std::string& db, const std::string& stbname, const std::string& tbname, int16_t vgid) {
|
||||||
|
impl_->createSubTable(db, stbname, tbname, vgid);
|
||||||
|
}
|
||||||
|
|
||||||
void MockCatalogService::showTables() const {
|
void MockCatalogService::showTables() const {
|
||||||
impl_->showTables();
|
impl_->showTables();
|
||||||
}
|
}
|
|
@ -49,9 +49,10 @@ public:
|
||||||
|
|
||||||
MockCatalogService();
|
MockCatalogService();
|
||||||
~MockCatalogService();
|
~MockCatalogService();
|
||||||
struct SCatalog* getCatalogHandle(const SEpSet* pMgmtEps);
|
struct SCatalog* getCatalogHandle(const SEpSet* pMgmtEps) const;
|
||||||
int32_t catalogGetMetaData(struct SCatalog* pCatalog, const SMetaReq* pMetaReq, SMetaData* pMetaData);
|
int32_t catalogGetMetaData(struct SCatalog* pCatalog, const SCatalogReq* pMetaReq, SMetaData* pMetaData) const;
|
||||||
ITableBuilder& createTableBuilder(const std::string& db, const std::string& tbname, int8_t tableType, int32_t numOfColumns, int32_t numOfTags = 0);
|
ITableBuilder& createTableBuilder(const std::string& db, const std::string& tbname, int8_t tableType, int32_t numOfColumns, int32_t numOfTags = 0);
|
||||||
|
void createSubTable(const std::string& db, const std::string& stbname, const std::string& tbname, int16_t vgid);
|
||||||
void showTables() const;
|
void showTables() const;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
|
|
@ -38,7 +38,7 @@ void setSchema(SSchema* p, int32_t type, int32_t bytes, const char* name, int32_
|
||||||
strcpy(p->name, name);
|
strcpy(p->name, name);
|
||||||
}
|
}
|
||||||
|
|
||||||
void setTableMetaInfo(SQueryStmtInfo* pQueryInfo, SMetaReq* req) {
|
void setTableMetaInfo(SQueryStmtInfo* pQueryInfo, SCatalogReq* req) {
|
||||||
pQueryInfo->numOfTables = 1;
|
pQueryInfo->numOfTables = 1;
|
||||||
|
|
||||||
pQueryInfo->pTableMetaInfo = (STableMetaInfo**)calloc(1, POINTER_BYTES);
|
pQueryInfo->pTableMetaInfo = (STableMetaInfo**)calloc(1, POINTER_BYTES);
|
||||||
|
@ -80,7 +80,7 @@ void sqlCheck(const char* sql, bool valid) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
@ -117,7 +117,7 @@ TEST(testCase, validateAST_test) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
@ -175,7 +175,7 @@ TEST(testCase, function_Test) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
@ -221,7 +221,7 @@ TEST(testCase, function_Test2) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
@ -267,7 +267,7 @@ TEST(testCase, function_Test3) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
@ -312,7 +312,7 @@ TEST(testCase, function_Test4) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
@ -360,7 +360,7 @@ TEST(testCase, function_Test5) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
@ -445,7 +445,7 @@ TEST(testCase, function_Test6) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
@ -523,7 +523,7 @@ TEST(testCase, function_Test6) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
@ -585,7 +585,7 @@ TEST(testCase, function_Test6) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
@ -664,7 +664,7 @@ TEST(testCase, function_Test6) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
|
|
@ -39,7 +39,7 @@ void setSchema(SSchema* p, int32_t type, int32_t bytes, const char* name, int32_
|
||||||
strcpy(p->name, name);
|
strcpy(p->name, name);
|
||||||
}
|
}
|
||||||
|
|
||||||
void setTableMetaInfo(SQueryStmtInfo* pQueryInfo, SMetaReq *req) {
|
void setTableMetaInfo(SQueryStmtInfo* pQueryInfo, SCatalogReq *req) {
|
||||||
pQueryInfo->numOfTables = 1;
|
pQueryInfo->numOfTables = 1;
|
||||||
|
|
||||||
pQueryInfo->pTableMetaInfo = (STableMetaInfo**)calloc(1, POINTER_BYTES);
|
pQueryInfo->pTableMetaInfo = (STableMetaInfo**)calloc(1, POINTER_BYTES);
|
||||||
|
@ -79,7 +79,7 @@ void generateLogicplan(const char* sql) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
@ -119,7 +119,7 @@ TEST(testCase, planner_test) {
|
||||||
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
|
|
@ -709,7 +709,7 @@ TEST(testCase, extractMeta_test) {
|
||||||
ASSERT_EQ(info1.valid, true);
|
ASSERT_EQ(info1.valid, true);
|
||||||
|
|
||||||
char msg[128] = {0};
|
char msg[128] = {0};
|
||||||
SMetaReq req = {0};
|
SCatalogReq req = {0};
|
||||||
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, msg, 128);
|
||||||
ASSERT_EQ(ret, 0);
|
ASSERT_EQ(ret, 0);
|
||||||
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
||||||
|
|
|
@ -0,0 +1,12 @@
|
||||||
|
aux_source_directory(src QUERY_SRC)
|
||||||
|
add_library(query ${QUERY_SRC})
|
||||||
|
target_include_directories(
|
||||||
|
query
|
||||||
|
PUBLIC "${CMAKE_SOURCE_DIR}/include/libs/query"
|
||||||
|
PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}/inc"
|
||||||
|
)
|
||||||
|
|
||||||
|
target_link_libraries(
|
||||||
|
query
|
||||||
|
PRIVATE os util common transport
|
||||||
|
)
|
|
@ -0,0 +1,40 @@
|
||||||
|
/*
|
||||||
|
* 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_QUERY_INT_H_
|
||||||
|
#define _TD_QUERY_INT_H_
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
#include "tlog.h"
|
||||||
|
|
||||||
|
extern int32_t qDebugFlag;
|
||||||
|
|
||||||
|
#define qFatal(...) do { if (qDebugFlag & DEBUG_FATAL) { taosPrintLog("QRY FATAL ", qDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define qError(...) do { if (qDebugFlag & DEBUG_ERROR) { taosPrintLog("QRY ERROR ", qDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define qWarn(...) do { if (qDebugFlag & DEBUG_WARN) { taosPrintLog("QRY WARN ", qDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define qInfo(...) do { if (qDebugFlag & DEBUG_INFO) { taosPrintLog("QRY ", qDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define qDebug(...) do { if (qDebugFlag & DEBUG_DEBUG) { taosPrintLog("QRY ", qDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define qTrace(...) do { if (qDebugFlag & DEBUG_TRACE) { taosPrintLog("QRY ", qDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
#define qDebugL(...) do { if (qDebugFlag & DEBUG_DEBUG) { taosPrintLongString("QRY ", qDebugFlag, __VA_ARGS__); }} while(0)
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /*_TD_QUERY_INT_H_*/
|
|
@ -0,0 +1,192 @@
|
||||||
|
/*
|
||||||
|
* 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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "taosmsg.h"
|
||||||
|
#include "queryInt.h"
|
||||||
|
|
||||||
|
|
||||||
|
int32_t (*queryBuildMsg[TSDB_MSG_TYPE_MAX])(void* input, char **msg, int32_t msgSize, int32_t *msgLen) = {0};
|
||||||
|
|
||||||
|
int32_t (*queryProcessMsgRsp[TSDB_MSG_TYPE_MAX])(void* output, char *msg, int32_t msgSize) = {0};
|
||||||
|
|
||||||
|
|
||||||
|
int32_t queryBuildVgroupListReqMsg(void* input, char **msg, int32_t msgSize, int32_t *msgLen) {
|
||||||
|
if (NULL == msg || NULL == msgLen) {
|
||||||
|
return TSDB_CODE_TSC_INVALID_INPUT;
|
||||||
|
}
|
||||||
|
|
||||||
|
*msgLen = 0;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t queryBuildTableMetaReqMsg(void* input, char **msg, int32_t msgSize, int32_t *msgLen) {
|
||||||
|
if (NULL == input || NULL == msg || NULL == msgLen) {
|
||||||
|
return TSDB_CODE_TSC_INVALID_INPUT;
|
||||||
|
}
|
||||||
|
|
||||||
|
SBuildTableMetaInput* bInput = (SBuildTableMetaInput *)input;
|
||||||
|
|
||||||
|
int32_t estimateSize = sizeof(STableInfoMsg);
|
||||||
|
if (NULL == *msg || msgSize < estimateSize) {
|
||||||
|
tfree(*msg);
|
||||||
|
*msg = calloc(1, estimateSize);
|
||||||
|
if (NULL == *msg) {
|
||||||
|
return TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
STableInfoMsg *bMsg = (STableInfoMsg *)*msg;
|
||||||
|
|
||||||
|
bMsg->msgHead.vgId = bInput->vgId;
|
||||||
|
|
||||||
|
strncpy(bMsg->tableFname, bInput->tableFullName, sizeof(bMsg->tableFname));
|
||||||
|
bMsg->tableFname[sizeof(bMsg->tableFname) - 1] = 0;
|
||||||
|
|
||||||
|
*msgLen = (int32_t)sizeof(*bMsg);
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t queryProcessVgroupListRsp(void* output, char *msg, int32_t msgSize) {
|
||||||
|
if (NULL == output || NULL == msg || msgSize <= 0) {
|
||||||
|
return TSDB_CODE_TSC_INVALID_INPUT;
|
||||||
|
}
|
||||||
|
|
||||||
|
SVgroupListRspMsg *pRsp = (SVgroupListRspMsg *)msg;
|
||||||
|
|
||||||
|
pRsp->vgroupNum = htonl(pRsp->vgroupNum);
|
||||||
|
pRsp->vgroupVersion = htonl(pRsp->vgroupVersion);
|
||||||
|
|
||||||
|
if (pRsp->vgroupNum < 0) {
|
||||||
|
qError("vgroup number[%d] in rsp is invalid", pRsp->vgroupNum);
|
||||||
|
return TSDB_CODE_TSC_VALUE_OUT_OF_RANGE;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pRsp->vgroupVersion < 0) {
|
||||||
|
qError("vgroup vgroupVersion[%d] in rsp is invalid", pRsp->vgroupVersion);
|
||||||
|
return TSDB_CODE_TSC_VALUE_OUT_OF_RANGE;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (msgSize != (pRsp->vgroupNum * sizeof(pRsp->vgroupInfo[0]) + sizeof(*pRsp))) {
|
||||||
|
qError("vgroup list msg size mis-match, msgSize:%d, vgroup number:%d", msgSize, pRsp->vgroupNum);
|
||||||
|
return TSDB_CODE_TSC_VALUE_OUT_OF_RANGE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// keep SVgroupListInfo/SVgroupListRspMsg the same
|
||||||
|
*(SVgroupListInfo **)output = (SVgroupListInfo *)msg;
|
||||||
|
|
||||||
|
if (pRsp->vgroupNum == 0) {
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int32_t i = 0; i < pRsp->vgroupNum; ++i) {
|
||||||
|
pRsp->vgroupInfo[i].vgId = htonl(pRsp->vgroupInfo[i].vgId);
|
||||||
|
for (int32_t n = 0; n < pRsp->vgroupInfo[i].numOfEps; ++n) {
|
||||||
|
pRsp->vgroupInfo[i].epAddr[n].port = htonl(pRsp->vgroupInfo[i].epAddr[n].port);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
void msgInit() {
|
||||||
|
queryBuildMsg[TSDB_MSG_TYPE_TABLE_META] = queryBuildTableMetaReqMsg;
|
||||||
|
queryBuildMsg[TSDB_MSG_TYPE_VGROUP_LIST] = queryBuildVgroupListReqMsg;
|
||||||
|
|
||||||
|
//tscProcessMsgRsp[TSDB_MSG_TYPE_TABLE_META] = tscProcessTableMetaRsp;
|
||||||
|
queryProcessMsgRsp[TSDB_MSG_TYPE_VGROUP_LIST] = queryProcessVgroupListRsp;
|
||||||
|
|
||||||
|
/*
|
||||||
|
tscBuildMsg[TSDB_SQL_SELECT] = tscBuildQueryMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_INSERT] = tscBuildSubmitMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_FETCH] = tscBuildFetchMsg;
|
||||||
|
|
||||||
|
tscBuildMsg[TSDB_SQL_CREATE_DB] = tscBuildCreateDbMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_CREATE_USER] = tscBuildUserMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_CREATE_FUNCTION] = tscBuildCreateFuncMsg;
|
||||||
|
|
||||||
|
tscBuildMsg[TSDB_SQL_CREATE_ACCT] = tscBuildAcctMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_ALTER_ACCT] = tscBuildAcctMsg;
|
||||||
|
|
||||||
|
tscBuildMsg[TSDB_SQL_CREATE_TABLE] = tscBuildCreateTableMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_DROP_USER] = tscBuildDropUserAcctMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_DROP_ACCT] = tscBuildDropUserAcctMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_DROP_DB] = tscBuildDropDbMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_DROP_FUNCTION] = tscBuildDropFuncMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_SYNC_DB_REPLICA] = tscBuildSyncDbReplicaMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_DROP_TABLE] = tscBuildDropTableMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_ALTER_USER] = tscBuildUserMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_CREATE_DNODE] = tscBuildCreateDnodeMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_DROP_DNODE] = tscBuildDropDnodeMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_CFG_DNODE] = tscBuildCfgDnodeMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_ALTER_TABLE] = tscBuildAlterTableMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_UPDATE_TAGS_VAL] = tscBuildUpdateTagMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_ALTER_DB] = tscAlterDbMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_COMPACT_VNODE] = tscBuildCompactMsg;
|
||||||
|
|
||||||
|
tscBuildMsg[TSDB_SQL_CONNECT] = tscBuildConnectMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_USE_DB] = tscBuildUseDbMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_STABLEVGROUP] = tscBuildSTableVgroupMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_RETRIEVE_FUNC] = tscBuildRetrieveFuncMsg;
|
||||||
|
|
||||||
|
tscBuildMsg[TSDB_SQL_HB] = tscBuildHeartBeatMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_SHOW] = tscBuildShowMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_RETRIEVE] = tscBuildRetrieveFromMgmtMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_KILL_QUERY] = tscBuildKillMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_KILL_STREAM] = tscBuildKillMsg;
|
||||||
|
tscBuildMsg[TSDB_SQL_KILL_CONNECTION] = tscBuildKillMsg;
|
||||||
|
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_SELECT] = tscProcessQueryRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_FETCH] = tscProcessRetrieveRspFromNode;
|
||||||
|
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_DROP_DB] = tscProcessDropDbRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_DROP_TABLE] = tscProcessDropTableRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_CONNECT] = tscProcessConnectRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_USE_DB] = tscProcessUseDbRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_META] = tscProcessTableMetaRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_STABLEVGROUP] = tscProcessSTableVgroupRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_MULTI_META] = tscProcessMultiTableMetaRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_RETRIEVE_FUNC] = tscProcessRetrieveFuncRsp;
|
||||||
|
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_SHOW] = tscProcessShowRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_RETRIEVE] = tscProcessRetrieveRspFromNode; // rsp handled by same function.
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_DESCRIBE_TABLE] = tscProcessDescribeTableRsp;
|
||||||
|
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_CURRENT_DB] = tscProcessLocalRetrieveRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_CURRENT_USER] = tscProcessLocalRetrieveRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_SERV_VERSION] = tscProcessLocalRetrieveRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_CLI_VERSION] = tscProcessLocalRetrieveRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_SERV_STATUS] = tscProcessLocalRetrieveRsp;
|
||||||
|
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_RETRIEVE_EMPTY_RESULT] = tscProcessEmptyResultRsp;
|
||||||
|
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_RETRIEVE_GLOBALMERGE] = tscProcessRetrieveGlobalMergeRsp;
|
||||||
|
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_ALTER_TABLE] = tscProcessAlterTableMsgRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_ALTER_DB] = tscProcessAlterDbMsgRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_COMPACT_VNODE] = tscProcessCompactRsp;
|
||||||
|
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_SHOW_CREATE_TABLE] = tscProcessShowCreateRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_SHOW_CREATE_STABLE] = tscProcessShowCreateRsp;
|
||||||
|
tscProcessMsgRsp[TSDB_SQL_SHOW_CREATE_DATABASE] = tscProcessShowCreateRsp;
|
||||||
|
*/
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -228,7 +228,7 @@ static int syncInitRpcServer(SSyncManager* syncManager, const SSyncCluster* pSyn
|
||||||
}
|
}
|
||||||
|
|
||||||
static int syncInitRpcClient(SSyncManager* syncManager) {
|
static int syncInitRpcClient(SSyncManager* syncManager) {
|
||||||
char secret[TSDB_KEY_LEN] = "secret";
|
char secret[TSDB_PASSWORD_LEN] = "secret";
|
||||||
SRpcInit rpcInit;
|
SRpcInit rpcInit;
|
||||||
memset(&rpcInit, 0, sizeof(rpcInit));
|
memset(&rpcInit, 0, sizeof(rpcInit));
|
||||||
rpcInit.label = "sync-client";
|
rpcInit.label = "sync-client";
|
||||||
|
|
|
@ -23,11 +23,10 @@ extern "C" {
|
||||||
#include "tlog.h"
|
#include "tlog.h"
|
||||||
|
|
||||||
extern int32_t rpcDebugFlag;
|
extern int32_t rpcDebugFlag;
|
||||||
extern int8_t tscEmbedded;
|
|
||||||
|
|
||||||
#define tFatal(...) { if (rpcDebugFlag & DEBUG_FATAL) { taosPrintLog("RPC FATAL ", tscEmbedded ? 255 : rpcDebugFlag, __VA_ARGS__); }}
|
#define tFatal(...) { if (rpcDebugFlag & DEBUG_FATAL) { taosPrintLog("RPC FATAL ", rpcDebugFlag, __VA_ARGS__); }}
|
||||||
#define tError(...) { if (rpcDebugFlag & DEBUG_ERROR) { taosPrintLog("RPC ERROR ", tscEmbedded ? 255 : rpcDebugFlag, __VA_ARGS__); }}
|
#define tError(...) { if (rpcDebugFlag & DEBUG_ERROR) { taosPrintLog("RPC ERROR ", rpcDebugFlag, __VA_ARGS__); }}
|
||||||
#define tWarn(...) { if (rpcDebugFlag & DEBUG_WARN) { taosPrintLog("RPC WARN ", tscEmbedded ? 255 : rpcDebugFlag, __VA_ARGS__); }}
|
#define tWarn(...) { if (rpcDebugFlag & DEBUG_WARN) { taosPrintLog("RPC WARN ", rpcDebugFlag, __VA_ARGS__); }}
|
||||||
#define tInfo(...) { if (rpcDebugFlag & DEBUG_INFO) { taosPrintLog("RPC ", tscEmbedded ? 255 : rpcDebugFlag, __VA_ARGS__); }}
|
#define tInfo(...) { if (rpcDebugFlag & DEBUG_INFO) { taosPrintLog("RPC ", tscEmbedded ? 255 : rpcDebugFlag, __VA_ARGS__); }}
|
||||||
#define tDebug(...) { if (rpcDebugFlag & DEBUG_DEBUG) { taosPrintLog("RPC ", rpcDebugFlag, __VA_ARGS__); }}
|
#define tDebug(...) { if (rpcDebugFlag & DEBUG_DEBUG) { taosPrintLog("RPC ", rpcDebugFlag, __VA_ARGS__); }}
|
||||||
#define tTrace(...) { if (rpcDebugFlag & DEBUG_TRACE) { taosPrintLog("RPC ", rpcDebugFlag, __VA_ARGS__); }}
|
#define tTrace(...) { if (rpcDebugFlag & DEBUG_TRACE) { taosPrintLog("RPC ", rpcDebugFlag, __VA_ARGS__); }}
|
||||||
|
|
|
@ -51,8 +51,8 @@ typedef struct {
|
||||||
char user[TSDB_UNI_LEN]; // meter ID
|
char user[TSDB_UNI_LEN]; // meter ID
|
||||||
char spi; // security parameter index
|
char spi; // security parameter index
|
||||||
char encrypt; // encrypt algorithm
|
char encrypt; // encrypt algorithm
|
||||||
char secret[TSDB_KEY_LEN]; // secret for the link
|
char secret[TSDB_PASSWORD_LEN]; // secret for the link
|
||||||
char ckey[TSDB_KEY_LEN]; // ciphering key
|
char ckey[TSDB_PASSWORD_LEN]; // ciphering key
|
||||||
|
|
||||||
void (*cfp)(void *parent, SRpcMsg *, SEpSet *);
|
void (*cfp)(void *parent, SRpcMsg *, SEpSet *);
|
||||||
int (*afp)(void *parent, char *user, char *spi, char *encrypt, char *secret, char *ckey);
|
int (*afp)(void *parent, char *user, char *spi, char *encrypt, char *secret, char *ckey);
|
||||||
|
@ -97,8 +97,8 @@ typedef struct SRpcConn {
|
||||||
char user[TSDB_UNI_LEN]; // user ID for the link
|
char user[TSDB_UNI_LEN]; // user ID for the link
|
||||||
char spi; // security parameter index
|
char spi; // security parameter index
|
||||||
char encrypt; // encryption, 0:1
|
char encrypt; // encryption, 0:1
|
||||||
char secret[TSDB_KEY_LEN]; // secret for the link
|
char secret[TSDB_PASSWORD_LEN]; // secret for the link
|
||||||
char ckey[TSDB_KEY_LEN]; // ciphering key
|
char ckey[TSDB_PASSWORD_LEN]; // ciphering key
|
||||||
char secured; // if set to 1, no authentication
|
char secured; // if set to 1, no authentication
|
||||||
uint16_t localPort; // for UDP only
|
uint16_t localPort; // for UDP only
|
||||||
uint32_t linkUid; // connection unique ID assigned by client
|
uint32_t linkUid; // connection unique ID assigned by client
|
||||||
|
@ -229,8 +229,6 @@ static void rpcInitImp(void) {
|
||||||
tsRpcOverhead = sizeof(SRpcReqContext);
|
tsRpcOverhead = sizeof(SRpcReqContext);
|
||||||
|
|
||||||
tsRpcRefId = taosOpenRef(200, rpcFree);
|
tsRpcRefId = taosOpenRef(200, rpcFree);
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t rpcInit(void) {
|
int32_t rpcInit(void) {
|
||||||
|
@ -413,7 +411,7 @@ void rpcSendRequest(void *shandle, const SEpSet *pEpSet, SRpcMsg *pMsg, int64_t
|
||||||
// for TDengine, all the query, show commands shall have TCP connection
|
// for TDengine, all the query, show commands shall have TCP connection
|
||||||
char type = pMsg->msgType;
|
char type = pMsg->msgType;
|
||||||
if (type == TSDB_MSG_TYPE_QUERY || type == TSDB_MSG_TYPE_SHOW_RETRIEVE
|
if (type == TSDB_MSG_TYPE_QUERY || type == TSDB_MSG_TYPE_SHOW_RETRIEVE
|
||||||
|| type == TSDB_MSG_TYPE_FETCH || type == TSDB_MSG_TYPE_STB_VGROUP
|
|| type == TSDB_MSG_TYPE_FETCH || type == TSDB_MSG_TYPE_VGROUP_LIST
|
||||||
|| type == TSDB_MSG_TYPE_TABLES_META || type == TSDB_MSG_TYPE_TABLE_META
|
|| type == TSDB_MSG_TYPE_TABLES_META || type == TSDB_MSG_TYPE_TABLE_META
|
||||||
|| type == TSDB_MSG_TYPE_SHOW || type == TSDB_MSG_TYPE_STATUS || type == TSDB_MSG_TYPE_ALTER_TABLE)
|
|| type == TSDB_MSG_TYPE_SHOW || type == TSDB_MSG_TYPE_STATUS || type == TSDB_MSG_TYPE_ALTER_TABLE)
|
||||||
pContext->connType = RPC_CONN_TCPC;
|
pContext->connType = RPC_CONN_TCPC;
|
||||||
|
@ -705,7 +703,7 @@ static SRpcConn *rpcAllocateClientConn(SRpcInfo *pRpc) {
|
||||||
pConn->linkUid = (uint32_t)((int64_t)pConn + taosGetPid() + (int64_t)pConn->tranId);
|
pConn->linkUid = (uint32_t)((int64_t)pConn + taosGetPid() + (int64_t)pConn->tranId);
|
||||||
pConn->spi = pRpc->spi;
|
pConn->spi = pRpc->spi;
|
||||||
pConn->encrypt = pRpc->encrypt;
|
pConn->encrypt = pRpc->encrypt;
|
||||||
if (pConn->spi) memcpy(pConn->secret, pRpc->secret, TSDB_KEY_LEN);
|
if (pConn->spi) memcpy(pConn->secret, pRpc->secret, TSDB_PASSWORD_LEN);
|
||||||
tDebug("%s %p client connection is allocated, uid:0x%x", pRpc->label, pConn, pConn->linkUid);
|
tDebug("%s %p client connection is allocated, uid:0x%x", pRpc->label, pConn, pConn->linkUid);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1534,9 +1532,9 @@ static int rpcAuthenticateMsg(void *pMsg, int msgLen, void *pAuth, void *pKey) {
|
||||||
int ret = -1;
|
int ret = -1;
|
||||||
|
|
||||||
tMD5Init(&context);
|
tMD5Init(&context);
|
||||||
tMD5Update(&context, (uint8_t *)pKey, TSDB_KEY_LEN);
|
tMD5Update(&context, (uint8_t *)pKey, TSDB_PASSWORD_LEN);
|
||||||
tMD5Update(&context, (uint8_t *)pMsg, msgLen);
|
tMD5Update(&context, (uint8_t *)pMsg, msgLen);
|
||||||
tMD5Update(&context, (uint8_t *)pKey, TSDB_KEY_LEN);
|
tMD5Update(&context, (uint8_t *)pKey, TSDB_PASSWORD_LEN);
|
||||||
tMD5Final(&context);
|
tMD5Final(&context);
|
||||||
|
|
||||||
if (memcmp(context.digest, pAuth, sizeof(context.digest)) == 0) ret = 0;
|
if (memcmp(context.digest, pAuth, sizeof(context.digest)) == 0) ret = 0;
|
||||||
|
@ -1548,9 +1546,9 @@ static void rpcBuildAuthHead(void *pMsg, int msgLen, void *pAuth, void *pKey) {
|
||||||
T_MD5_CTX context;
|
T_MD5_CTX context;
|
||||||
|
|
||||||
tMD5Init(&context);
|
tMD5Init(&context);
|
||||||
tMD5Update(&context, (uint8_t *)pKey, TSDB_KEY_LEN);
|
tMD5Update(&context, (uint8_t *)pKey, TSDB_PASSWORD_LEN);
|
||||||
tMD5Update(&context, (uint8_t *)pMsg, msgLen);
|
tMD5Update(&context, (uint8_t *)pMsg, msgLen);
|
||||||
tMD5Update(&context, (uint8_t *)pKey, TSDB_KEY_LEN);
|
tMD5Update(&context, (uint8_t *)pKey, TSDB_PASSWORD_LEN);
|
||||||
tMD5Final(&context);
|
tMD5Final(&context);
|
||||||
|
|
||||||
memcpy(pAuth, context.digest, sizeof(context.digest));
|
memcpy(pAuth, context.digest, sizeof(context.digest));
|
||||||
|
|
|
@ -33,10 +33,12 @@ typedef struct WalFileInfo {
|
||||||
int64_t fileSize;
|
int64_t fileSize;
|
||||||
} WalFileInfo;
|
} WalFileInfo;
|
||||||
|
|
||||||
|
#pragma pack(push,1)
|
||||||
typedef struct WalIdxEntry {
|
typedef struct WalIdxEntry {
|
||||||
int64_t ver;
|
int64_t ver;
|
||||||
int64_t offset;
|
int64_t offset;
|
||||||
} WalIdxEntry;
|
} WalIdxEntry;
|
||||||
|
#pragma pack(pop)
|
||||||
|
|
||||||
static inline int32_t compareWalFileInfo(const void* pLeft, const void* pRight) {
|
static inline int32_t compareWalFileInfo(const void* pLeft, const void* pRight) {
|
||||||
WalFileInfo* pInfoLeft = (WalFileInfo*)pLeft;
|
WalFileInfo* pInfoLeft = (WalFileInfo*)pLeft;
|
||||||
|
@ -78,11 +80,11 @@ static inline WalFileInfo* walGetCurFileInfo(SWal* pWal) {
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline int walBuildLogName(SWal*pWal, int64_t fileFirstVer, char* buf) {
|
static inline int walBuildLogName(SWal*pWal, int64_t fileFirstVer, char* buf) {
|
||||||
return sprintf(buf, "%s/%" PRId64 "." WAL_LOG_SUFFIX, pWal->path, fileFirstVer);
|
return sprintf(buf, "%s/%020" PRId64 "." WAL_LOG_SUFFIX, pWal->path, fileFirstVer);
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline int walBuildIdxName(SWal*pWal, int64_t fileFirstVer, char* buf) {
|
static inline int walBuildIdxName(SWal*pWal, int64_t fileFirstVer, char* buf) {
|
||||||
return sprintf(buf, "%s/%" PRId64 "." WAL_INDEX_SUFFIX, pWal->path, fileFirstVer);
|
return sprintf(buf, "%s/%020" PRId64 "." WAL_INDEX_SUFFIX, pWal->path, fileFirstVer);
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline int walValidHeadCksum(SWalHead* pHead) {
|
static inline int walValidHeadCksum(SWalHead* pHead) {
|
||||||
|
@ -90,7 +92,7 @@ static inline int walValidHeadCksum(SWalHead* pHead) {
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline int walValidBodyCksum(SWalHead* pHead) {
|
static inline int walValidBodyCksum(SWalHead* pHead) {
|
||||||
return taosCheckChecksum((uint8_t*)pHead->head.cont, pHead->head.len, pHead->cksumBody);
|
return taosCheckChecksum((uint8_t*)pHead->head.body, pHead->head.len, pHead->cksumBody);
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline int walValidCksum(SWalHead *pHead, void* body, int64_t bodyLen) {
|
static inline int walValidCksum(SWalHead *pHead, void* body, int64_t bodyLen) {
|
||||||
|
|
|
@ -25,15 +25,15 @@
|
||||||
#include <regex.h>
|
#include <regex.h>
|
||||||
|
|
||||||
int64_t walGetFirstVer(SWal *pWal) {
|
int64_t walGetFirstVer(SWal *pWal) {
|
||||||
return pWal->firstVersion;
|
return pWal->vers.firstVer;
|
||||||
}
|
}
|
||||||
|
|
||||||
int64_t walGetSnaphostVer(SWal *pWal) {
|
int64_t walGetSnaphostVer(SWal *pWal) {
|
||||||
return pWal->snapshotVersion;
|
return pWal->vers.snapshotVer;
|
||||||
}
|
}
|
||||||
|
|
||||||
int64_t walGetLastVer(SWal *pWal) {
|
int64_t walGetLastVer(SWal *pWal) {
|
||||||
return pWal->lastVersion;
|
return pWal->vers.lastVer;
|
||||||
}
|
}
|
||||||
|
|
||||||
int walRollFileInfo(SWal* pWal) {
|
int walRollFileInfo(SWal* pWal) {
|
||||||
|
@ -42,26 +42,28 @@ int walRollFileInfo(SWal* pWal) {
|
||||||
SArray* pArray = pWal->fileInfoSet;
|
SArray* pArray = pWal->fileInfoSet;
|
||||||
if(taosArrayGetSize(pArray) != 0) {
|
if(taosArrayGetSize(pArray) != 0) {
|
||||||
WalFileInfo *pInfo = taosArrayGetLast(pArray);
|
WalFileInfo *pInfo = taosArrayGetLast(pArray);
|
||||||
pInfo->lastVer = pWal->lastVersion;
|
pInfo->lastVer = pWal->vers.lastVer;
|
||||||
pInfo->closeTs = ts;
|
pInfo->closeTs = ts;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
//TODO: change to emplace back
|
||||||
WalFileInfo *pNewInfo = malloc(sizeof(WalFileInfo));
|
WalFileInfo *pNewInfo = malloc(sizeof(WalFileInfo));
|
||||||
if(pNewInfo == NULL) {
|
if(pNewInfo == NULL) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
pNewInfo->firstVer = pWal->lastVersion + 1;
|
pNewInfo->firstVer = pWal->vers.lastVer + 1;
|
||||||
pNewInfo->lastVer = -1;
|
pNewInfo->lastVer = -1;
|
||||||
pNewInfo->createTs = ts;
|
pNewInfo->createTs = ts;
|
||||||
pNewInfo->closeTs = -1;
|
pNewInfo->closeTs = -1;
|
||||||
pNewInfo->fileSize = 0;
|
pNewInfo->fileSize = 0;
|
||||||
taosArrayPush(pWal->fileInfoSet, pNewInfo);
|
taosArrayPush(pWal->fileInfoSet, pNewInfo);
|
||||||
|
free(pNewInfo);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
char* walMetaSerialize(SWal* pWal) {
|
char* walMetaSerialize(SWal* pWal) {
|
||||||
char buf[30];
|
char buf[30];
|
||||||
if(pWal == NULL || pWal->fileInfoSet == NULL) return 0;
|
ASSERT(pWal->fileInfoSet);
|
||||||
int sz = pWal->fileInfoSet->size;
|
int sz = pWal->fileInfoSet->size;
|
||||||
cJSON* pRoot = cJSON_CreateObject();
|
cJSON* pRoot = cJSON_CreateObject();
|
||||||
cJSON* pMeta = cJSON_CreateObject();
|
cJSON* pMeta = cJSON_CreateObject();
|
||||||
|
@ -72,13 +74,13 @@ char* walMetaSerialize(SWal* pWal) {
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
cJSON_AddItemToObject(pRoot, "meta", pMeta);
|
cJSON_AddItemToObject(pRoot, "meta", pMeta);
|
||||||
sprintf(buf, "%" PRId64, pWal->firstVersion);
|
sprintf(buf, "%" PRId64, pWal->vers.firstVer);
|
||||||
cJSON_AddStringToObject(pMeta, "firstVer", buf);
|
cJSON_AddStringToObject(pMeta, "firstVer", buf);
|
||||||
sprintf(buf, "%" PRId64, pWal->snapshotVersion);
|
sprintf(buf, "%" PRId64, pWal->vers.snapshotVer);
|
||||||
cJSON_AddStringToObject(pMeta, "snapshotVer", buf);
|
cJSON_AddStringToObject(pMeta, "snapshotVer", buf);
|
||||||
sprintf(buf, "%" PRId64, pWal->commitVersion);
|
sprintf(buf, "%" PRId64, pWal->vers.commitVer);
|
||||||
cJSON_AddStringToObject(pMeta, "commitVer", buf);
|
cJSON_AddStringToObject(pMeta, "commitVer", buf);
|
||||||
sprintf(buf, "%" PRId64, pWal->lastVersion);
|
sprintf(buf, "%" PRId64, pWal->vers.lastVer);
|
||||||
cJSON_AddStringToObject(pMeta, "lastVer", buf);
|
cJSON_AddStringToObject(pMeta, "lastVer", buf);
|
||||||
|
|
||||||
cJSON_AddItemToObject(pRoot, "files", pFiles);
|
cJSON_AddItemToObject(pRoot, "files", pFiles);
|
||||||
|
@ -103,7 +105,9 @@ char* walMetaSerialize(SWal* pWal) {
|
||||||
sprintf(buf, "%" PRId64, pInfo->fileSize);
|
sprintf(buf, "%" PRId64, pInfo->fileSize);
|
||||||
cJSON_AddStringToObject(pField, "fileSize", buf);
|
cJSON_AddStringToObject(pField, "fileSize", buf);
|
||||||
}
|
}
|
||||||
return cJSON_Print(pRoot);
|
char* serialized = cJSON_Print(pRoot);
|
||||||
|
cJSON_Delete(pRoot);
|
||||||
|
return serialized;
|
||||||
}
|
}
|
||||||
|
|
||||||
int walMetaDeserialize(SWal* pWal, const char* bytes) {
|
int walMetaDeserialize(SWal* pWal, const char* bytes) {
|
||||||
|
@ -112,18 +116,19 @@ int walMetaDeserialize(SWal* pWal, const char* bytes) {
|
||||||
pRoot = cJSON_Parse(bytes);
|
pRoot = cJSON_Parse(bytes);
|
||||||
pMeta = cJSON_GetObjectItem(pRoot, "meta");
|
pMeta = cJSON_GetObjectItem(pRoot, "meta");
|
||||||
pField = cJSON_GetObjectItem(pMeta, "firstVer");
|
pField = cJSON_GetObjectItem(pMeta, "firstVer");
|
||||||
pWal->firstVersion = atoll(cJSON_GetStringValue(pField));
|
pWal->vers.firstVer = atoll(cJSON_GetStringValue(pField));
|
||||||
pField = cJSON_GetObjectItem(pMeta, "snapshotVer");
|
pField = cJSON_GetObjectItem(pMeta, "snapshotVer");
|
||||||
pWal->snapshotVersion = atoll(cJSON_GetStringValue(pField));
|
pWal->vers.snapshotVer = atoll(cJSON_GetStringValue(pField));
|
||||||
pField = cJSON_GetObjectItem(pMeta, "commitVer");
|
pField = cJSON_GetObjectItem(pMeta, "commitVer");
|
||||||
pWal->commitVersion = atoll(cJSON_GetStringValue(pField));
|
pWal->vers.commitVer = atoll(cJSON_GetStringValue(pField));
|
||||||
pField = cJSON_GetObjectItem(pMeta, "lastVer");
|
pField = cJSON_GetObjectItem(pMeta, "lastVer");
|
||||||
pWal->lastVersion = atoll(cJSON_GetStringValue(pField));
|
pWal->vers.lastVer = atoll(cJSON_GetStringValue(pField));
|
||||||
|
|
||||||
pFiles = cJSON_GetObjectItem(pRoot, "files");
|
pFiles = cJSON_GetObjectItem(pRoot, "files");
|
||||||
int sz = cJSON_GetArraySize(pFiles);
|
int sz = cJSON_GetArraySize(pFiles);
|
||||||
//deserialize
|
//deserialize
|
||||||
SArray* pArray = taosArrayInit(sz, sizeof(WalFileInfo));
|
SArray* pArray = pWal->fileInfoSet;
|
||||||
|
taosArrayEnsureCap(pArray, sz);
|
||||||
WalFileInfo *pData = pArray->pData;
|
WalFileInfo *pData = pArray->pData;
|
||||||
for(int i = 0; i < sz; i++) {
|
for(int i = 0; i < sz; i++) {
|
||||||
cJSON* pInfoJson = cJSON_GetArrayItem(pFiles, i);
|
cJSON* pInfoJson = cJSON_GetArrayItem(pFiles, i);
|
||||||
|
@ -141,6 +146,7 @@ int walMetaDeserialize(SWal* pWal, const char* bytes) {
|
||||||
}
|
}
|
||||||
taosArraySetSize(pArray, sz);
|
taosArraySetSize(pArray, sz);
|
||||||
pWal->fileInfoSet = pArray;
|
pWal->fileInfoSet = pArray;
|
||||||
|
cJSON_Delete(pRoot);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -155,7 +161,7 @@ static int walFindCurMetaVer(SWal* pWal) {
|
||||||
|
|
||||||
DIR *dir = opendir(pWal->path);
|
DIR *dir = opendir(pWal->path);
|
||||||
if(dir == NULL) {
|
if(dir == NULL) {
|
||||||
wError("vgId:%d, path:%s, failed to open since %s", pWal->vgId, pWal->path, strerror(errno));
|
wError("vgId:%d, path:%s, failed to open since %s", pWal->cfg.vgId, pWal->path, strerror(errno));
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -171,6 +177,8 @@ static int walFindCurMetaVer(SWal* pWal) {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
closedir(dir);
|
||||||
|
regfree(&walMetaRegexPattern);
|
||||||
return metaVer;
|
return metaVer;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -195,6 +203,7 @@ int walWriteMeta(SWal* pWal) {
|
||||||
walBuildMetaName(pWal, metaVer, fnameStr);
|
walBuildMetaName(pWal, metaVer, fnameStr);
|
||||||
remove(fnameStr);
|
remove(fnameStr);
|
||||||
}
|
}
|
||||||
|
free(serialized);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -215,6 +224,7 @@ int walReadMeta(SWal* pWal) {
|
||||||
if(buf == NULL) {
|
if(buf == NULL) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
memset(buf, 0, size+5);
|
||||||
int tfd = tfOpenRead(fnameStr);
|
int tfd = tfOpenRead(fnameStr);
|
||||||
if(tfRead(tfd, buf, size) != size) {
|
if(tfRead(tfd, buf, size) != size) {
|
||||||
free(buf);
|
free(buf);
|
||||||
|
|
|
@ -68,9 +68,12 @@ int32_t walInit() {
|
||||||
}
|
}
|
||||||
|
|
||||||
void walCleanUp() {
|
void walCleanUp() {
|
||||||
|
int old = atomic_val_compare_exchange_8(&tsWal.inited, 1, 0);
|
||||||
|
if(old == 0) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
walStopThread();
|
walStopThread();
|
||||||
taosCloseRef(tsWal.refSetId);
|
taosCloseRef(tsWal.refSetId);
|
||||||
atomic_store_8(&tsWal.inited, 0);
|
|
||||||
wInfo("wal module is cleaned up");
|
wInfo("wal module is cleaned up");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -80,31 +83,30 @@ SWal *walOpen(const char *path, SWalCfg *pCfg) {
|
||||||
terrno = TAOS_SYSTEM_ERROR(errno);
|
terrno = TAOS_SYSTEM_ERROR(errno);
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
memset(pWal, 0, sizeof(SWal));
|
||||||
pWal->writeLogTfd = -1;
|
pWal->writeLogTfd = -1;
|
||||||
pWal->writeIdxTfd = -1;
|
pWal->writeIdxTfd = -1;
|
||||||
pWal->writeCur = -1;
|
pWal->writeCur = -1;
|
||||||
|
|
||||||
//set config
|
//set config
|
||||||
pWal->vgId = pCfg->vgId;
|
memcpy(&pWal->cfg, pCfg, sizeof(SWalCfg));
|
||||||
pWal->fsyncPeriod = pCfg->fsyncPeriod;
|
|
||||||
pWal->rollPeriod = pCfg->rollPeriod;
|
|
||||||
pWal->segSize = pCfg->segSize;
|
|
||||||
pWal->level = pCfg->walLevel;
|
|
||||||
|
|
||||||
//init version info
|
//init version info
|
||||||
pWal->firstVersion = -1;
|
pWal->vers.firstVer = -1;
|
||||||
pWal->commitVersion = -1;
|
pWal->vers.commitVer = -1;
|
||||||
pWal->snapshotVersion = -1;
|
pWal->vers.snapshotVer = -1;
|
||||||
pWal->lastVersion = -1;
|
pWal->vers.lastVer = -1;
|
||||||
|
|
||||||
pWal->snapshottingVer = -1;
|
pWal->vers.verInSnapshotting = -1;
|
||||||
|
|
||||||
|
pWal->totSize = 0;
|
||||||
|
|
||||||
//init status
|
//init status
|
||||||
pWal->lastRollSeq = -1;
|
pWal->lastRollSeq = -1;
|
||||||
|
|
||||||
//init write buffer
|
//init write buffer
|
||||||
memset(&pWal->head, 0, sizeof(SWalHead));
|
memset(&pWal->writeHead, 0, sizeof(SWalHead));
|
||||||
pWal->head.head.sver = 0;
|
pWal->writeHead.head.sver = 0;
|
||||||
|
|
||||||
tstrncpy(pWal->path, path, sizeof(pWal->path));
|
tstrncpy(pWal->path, path, sizeof(pWal->path));
|
||||||
pthread_mutex_init(&pWal->mutex, NULL);
|
pthread_mutex_init(&pWal->mutex, NULL);
|
||||||
|
@ -122,8 +124,9 @@ SWal *walOpen(const char *path, SWalCfg *pCfg) {
|
||||||
walFreeObj(pWal);
|
walFreeObj(pWal);
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
walReadMeta(pWal);
|
||||||
|
|
||||||
wDebug("vgId:%d, wal:%p is opened, level:%d fsyncPeriod:%d", pWal->vgId, pWal, pWal->level, pWal->fsyncPeriod);
|
wDebug("vgId:%d, wal:%p is opened, level:%d fsyncPeriod:%d", pWal->cfg.vgId, pWal, pWal->cfg.level, pWal->cfg.fsyncPeriod);
|
||||||
|
|
||||||
return pWal;
|
return pWal;
|
||||||
}
|
}
|
||||||
|
@ -131,17 +134,17 @@ SWal *walOpen(const char *path, SWalCfg *pCfg) {
|
||||||
int32_t walAlter(SWal *pWal, SWalCfg *pCfg) {
|
int32_t walAlter(SWal *pWal, SWalCfg *pCfg) {
|
||||||
if (pWal == NULL) return TSDB_CODE_WAL_APP_ERROR;
|
if (pWal == NULL) return TSDB_CODE_WAL_APP_ERROR;
|
||||||
|
|
||||||
if (pWal->level == pCfg->walLevel && pWal->fsyncPeriod == pCfg->fsyncPeriod) {
|
if (pWal->cfg.level == pCfg->level && pWal->cfg.fsyncPeriod == pCfg->fsyncPeriod) {
|
||||||
wDebug("vgId:%d, old walLevel:%d fsync:%d, new walLevel:%d fsync:%d not change", pWal->vgId, pWal->level,
|
wDebug("vgId:%d, old walLevel:%d fsync:%d, new walLevel:%d fsync:%d not change", pWal->cfg.vgId, pWal->cfg.level,
|
||||||
pWal->fsyncPeriod, pCfg->walLevel, pCfg->fsyncPeriod);
|
pWal->cfg.fsyncPeriod, pCfg->level, pCfg->fsyncPeriod);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
wInfo("vgId:%d, change old walLevel:%d fsync:%d, new walLevel:%d fsync:%d", pWal->vgId, pWal->level,
|
wInfo("vgId:%d, change old walLevel:%d fsync:%d, new walLevel:%d fsync:%d", pWal->cfg.vgId, pWal->cfg.level,
|
||||||
pWal->fsyncPeriod, pCfg->walLevel, pCfg->fsyncPeriod);
|
pWal->cfg.fsyncPeriod, pCfg->level, pCfg->fsyncPeriod);
|
||||||
|
|
||||||
pWal->level = pCfg->walLevel;
|
pWal->cfg.level = pCfg->level;
|
||||||
pWal->fsyncPeriod = pCfg->fsyncPeriod;
|
pWal->cfg.fsyncPeriod = pCfg->fsyncPeriod;
|
||||||
pWal->fsyncSeq = pCfg->fsyncPeriod / 1000;
|
pWal->fsyncSeq = pCfg->fsyncPeriod / 1000;
|
||||||
if (pWal->fsyncSeq <= 0) pWal->fsyncSeq = 1;
|
if (pWal->fsyncSeq <= 0) pWal->fsyncSeq = 1;
|
||||||
|
|
||||||
|
@ -153,31 +156,34 @@ void walClose(SWal *pWal) {
|
||||||
|
|
||||||
pthread_mutex_lock(&pWal->mutex);
|
pthread_mutex_lock(&pWal->mutex);
|
||||||
tfClose(pWal->writeLogTfd);
|
tfClose(pWal->writeLogTfd);
|
||||||
|
pWal->writeLogTfd = -1;
|
||||||
tfClose(pWal->writeIdxTfd);
|
tfClose(pWal->writeIdxTfd);
|
||||||
/*taosArrayDestroy(pWal->fileInfoSet);*/
|
pWal->writeIdxTfd = -1;
|
||||||
/*pWal->fileInfoSet = NULL;*/
|
walWriteMeta(pWal);
|
||||||
|
taosArrayDestroy(pWal->fileInfoSet);
|
||||||
|
pWal->fileInfoSet = NULL;
|
||||||
pthread_mutex_unlock(&pWal->mutex);
|
pthread_mutex_unlock(&pWal->mutex);
|
||||||
taosRemoveRef(tsWal.refSetId, pWal->refId);
|
taosRemoveRef(tsWal.refSetId, pWal->refId);
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t walInitObj(SWal *pWal) {
|
static int32_t walInitObj(SWal *pWal) {
|
||||||
if (taosMkDir(pWal->path) != 0) {
|
if (taosMkDir(pWal->path) != 0) {
|
||||||
wError("vgId:%d, path:%s, failed to create directory since %s", pWal->vgId, pWal->path, strerror(errno));
|
wError("vgId:%d, path:%s, failed to create directory since %s", pWal->cfg.vgId, pWal->path, strerror(errno));
|
||||||
return TAOS_SYSTEM_ERROR(errno);
|
return TAOS_SYSTEM_ERROR(errno);
|
||||||
}
|
}
|
||||||
pWal->fileInfoSet = taosArrayInit(0, sizeof(WalFileInfo));
|
pWal->fileInfoSet = taosArrayInit(8, sizeof(WalFileInfo));
|
||||||
if(pWal->fileInfoSet == NULL) {
|
if(pWal->fileInfoSet == NULL) {
|
||||||
wError("vgId:%d, path:%s, failed to init taosArray %s", pWal->vgId, pWal->path, strerror(errno));
|
wError("vgId:%d, path:%s, failed to init taosArray %s", pWal->cfg.vgId, pWal->path, strerror(errno));
|
||||||
return TAOS_SYSTEM_ERROR(errno);
|
return TAOS_SYSTEM_ERROR(errno);
|
||||||
}
|
}
|
||||||
|
|
||||||
wDebug("vgId:%d, object is initialized", pWal->vgId);
|
wDebug("vgId:%d, object is initialized", pWal->cfg.vgId);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void walFreeObj(void *wal) {
|
static void walFreeObj(void *wal) {
|
||||||
SWal *pWal = wal;
|
SWal *pWal = wal;
|
||||||
wDebug("vgId:%d, wal:%p is freed", pWal->vgId, pWal);
|
wDebug("vgId:%d, wal:%p is freed", pWal->cfg.vgId, pWal);
|
||||||
|
|
||||||
tfClose(pWal->writeLogTfd);
|
tfClose(pWal->writeLogTfd);
|
||||||
tfClose(pWal->writeIdxTfd);
|
tfClose(pWal->writeIdxTfd);
|
||||||
|
@ -188,7 +194,7 @@ static void walFreeObj(void *wal) {
|
||||||
}
|
}
|
||||||
|
|
||||||
static bool walNeedFsync(SWal *pWal) {
|
static bool walNeedFsync(SWal *pWal) {
|
||||||
if (pWal->fsyncPeriod <= 0 || pWal->level != TAOS_WAL_FSYNC) {
|
if (pWal->cfg.fsyncPeriod <= 0 || pWal->cfg.level != TAOS_WAL_FSYNC) {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -208,10 +214,10 @@ static void walFsyncAll() {
|
||||||
SWal *pWal = taosIterateRef(tsWal.refSetId, 0);
|
SWal *pWal = taosIterateRef(tsWal.refSetId, 0);
|
||||||
while (pWal) {
|
while (pWal) {
|
||||||
if (walNeedFsync(pWal)) {
|
if (walNeedFsync(pWal)) {
|
||||||
wTrace("vgId:%d, do fsync, level:%d seq:%d rseq:%d", pWal->vgId, pWal->level, pWal->fsyncSeq, atomic_load_32(&tsWal.seq));
|
wTrace("vgId:%d, do fsync, level:%d seq:%d rseq:%d", pWal->cfg.vgId, pWal->cfg.level, pWal->fsyncSeq, atomic_load_32(&tsWal.seq));
|
||||||
int32_t code = tfFsync(pWal->writeLogTfd);
|
int32_t code = tfFsync(pWal->writeLogTfd);
|
||||||
if (code != 0) {
|
if (code != 0) {
|
||||||
wError("vgId:%d, file:%"PRId64".log, failed to fsync since %s", pWal->vgId, walGetLastFileFirstVer(pWal), strerror(code));
|
wError("vgId:%d, file:%"PRId64".log, failed to fsync since %s", pWal->cfg.vgId, walGetLastFileFirstVer(pWal), strerror(code));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
pWal = taosIterateRef(tsWal.refSetId, pWal->refId);
|
pWal = taosIterateRef(tsWal.refSetId, pWal->refId);
|
||||||
|
@ -249,9 +255,8 @@ static int32_t walCreateThread() {
|
||||||
static void walStopThread() {
|
static void walStopThread() {
|
||||||
atomic_store_8(&tsWal.stop, 1);
|
atomic_store_8(&tsWal.stop, 1);
|
||||||
|
|
||||||
if (tsWal.thread != NULL && taosCheckPthreadValid(tsWal.thread)) {
|
if (taosCheckPthreadValid(tsWal.thread)) {
|
||||||
pthread_join(tsWal.thread, NULL);
|
pthread_join(tsWal.thread, NULL);
|
||||||
tsWal.thread = NULL;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
wDebug("wal thread is stopped");
|
wDebug("wal thread is stopped");
|
||||||
|
|
|
@ -16,6 +16,163 @@
|
||||||
#include "walInt.h"
|
#include "walInt.h"
|
||||||
#include "tfile.h"
|
#include "tfile.h"
|
||||||
|
|
||||||
|
SWalReadHandle* walOpenReadHandle(SWal* pWal) {
|
||||||
|
SWalReadHandle *pRead = malloc(sizeof(SWalReadHandle));
|
||||||
|
if(pRead == NULL) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
pRead->pWal = pWal;
|
||||||
|
pRead->readIdxTfd = -1;
|
||||||
|
pRead->readLogTfd = -1;
|
||||||
|
pRead->curVersion = -1;
|
||||||
|
pRead->curFileFirstVer = -1;
|
||||||
|
pRead->capacity = 0;
|
||||||
|
pRead->status = 0;
|
||||||
|
pRead->pHead = malloc(sizeof(SWalHead));
|
||||||
|
if(pRead->pHead == NULL) {
|
||||||
|
free(pRead);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
return pRead;
|
||||||
|
}
|
||||||
|
|
||||||
|
void walCloseReadHandle(SWalReadHandle *pRead) {
|
||||||
|
tfClose(pRead->readIdxTfd);
|
||||||
|
tfClose(pRead->readLogTfd);
|
||||||
|
tfree(pRead->pHead);
|
||||||
|
free(pRead);
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t walRegisterRead(SWalReadHandle *pRead, int64_t ver) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t walReadSeekFilePos(SWalReadHandle *pRead, int64_t fileFirstVer, int64_t ver) {
|
||||||
|
int code = 0;
|
||||||
|
|
||||||
|
int64_t idxTfd = pRead->readIdxTfd;
|
||||||
|
int64_t logTfd = pRead->readLogTfd;
|
||||||
|
|
||||||
|
//seek position
|
||||||
|
int64_t offset = (ver - fileFirstVer) * WAL_IDX_ENTRY_SIZE;
|
||||||
|
code = tfLseek(idxTfd, offset, SEEK_SET);
|
||||||
|
if(code < 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
WalIdxEntry entry;
|
||||||
|
if(tfRead(idxTfd, &entry, sizeof(WalIdxEntry)) != sizeof(WalIdxEntry)) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
//TODO:deserialize
|
||||||
|
ASSERT(entry.ver == ver);
|
||||||
|
code = tfLseek(logTfd, entry.offset, SEEK_SET);
|
||||||
|
if (code < 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t walReadChangeFile(SWalReadHandle *pRead, int64_t fileFirstVer) {
|
||||||
|
char fnameStr[WAL_FILE_LEN];
|
||||||
|
|
||||||
|
tfClose(pRead->readIdxTfd);
|
||||||
|
tfClose(pRead->readLogTfd);
|
||||||
|
|
||||||
|
walBuildLogName(pRead->pWal, fileFirstVer, fnameStr);
|
||||||
|
int64_t logTfd = tfOpenRead(fnameStr);
|
||||||
|
if(logTfd < 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
walBuildIdxName(pRead->pWal, fileFirstVer, fnameStr);
|
||||||
|
int64_t idxTfd = tfOpenRead(fnameStr);
|
||||||
|
if(idxTfd < 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
pRead->readLogTfd = logTfd;
|
||||||
|
pRead->readIdxTfd = idxTfd;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t walReadSeekVer(SWalReadHandle *pRead, int64_t ver) {
|
||||||
|
int code;
|
||||||
|
SWal *pWal = pRead->pWal;
|
||||||
|
if(ver == pRead->curVersion) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if(ver > pWal->vers.lastVer || ver < pWal->vers.firstVer) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(ver < pWal->vers.snapshotVer) {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
WalFileInfo tmpInfo;
|
||||||
|
tmpInfo.firstVer = ver;
|
||||||
|
//bsearch in fileSet
|
||||||
|
WalFileInfo* pRet = taosArraySearch(pWal->fileInfoSet, &tmpInfo, compareWalFileInfo, TD_LE);
|
||||||
|
ASSERT(pRet != NULL);
|
||||||
|
if(pRead->curFileFirstVer != pRet->firstVer) {
|
||||||
|
code = walReadChangeFile(pRead, pRet->firstVer);
|
||||||
|
if(code < 0) {
|
||||||
|
//TODO: set error flag
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
code = walReadSeekFilePos(pRead, pRet->firstVer, ver);
|
||||||
|
if(code < 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
pRead->curVersion = ver;
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t walReadWithHandle(SWalReadHandle *pRead, int64_t ver) {
|
||||||
|
int code;
|
||||||
|
//TODO: check wal life
|
||||||
|
if(pRead->curVersion != ver) {
|
||||||
|
code = walReadSeekVer(pRead, ver);
|
||||||
|
if(code != 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if(!tfValid(pRead->readLogTfd)) return -1;
|
||||||
|
|
||||||
|
code = tfRead(pRead->readLogTfd, pRead->pHead, sizeof(SWalHead));
|
||||||
|
if(code != sizeof(SWalHead)) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
code = walValidHeadCksum(pRead->pHead);
|
||||||
|
if(code != 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(pRead->capacity < pRead->pHead->head.len) {
|
||||||
|
void* ptr = realloc(pRead->pHead, sizeof(SWalHead) + pRead->pHead->head.len);
|
||||||
|
if(ptr == NULL) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
pRead->pHead = ptr;
|
||||||
|
pRead->capacity = pRead->pHead->head.len;
|
||||||
|
}
|
||||||
|
if(pRead->pHead->head.len != tfRead(pRead->readLogTfd, pRead->pHead->head.body, pRead->pHead->head.len)) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*code = walValidBodyCksum(pRead->pHead);*/
|
||||||
|
ASSERT(pRead->pHead->head.version == ver);
|
||||||
|
|
||||||
|
if(code != 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
pRead->curVersion++;
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
int32_t walRead(SWal *pWal, SWalHead **ppHead, int64_t ver) {
|
int32_t walRead(SWal *pWal, SWalHead **ppHead, int64_t ver) {
|
||||||
int code;
|
int code;
|
||||||
code = walSeekVer(pWal, ver);
|
code = walSeekVer(pWal, ver);
|
||||||
|
@ -42,7 +199,7 @@ int32_t walRead(SWal *pWal, SWalHead **ppHead, int64_t ver) {
|
||||||
*ppHead = NULL;
|
*ppHead = NULL;
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
if(tfRead(pWal->writeLogTfd, (*ppHead)->head.cont, (*ppHead)->head.len) != (*ppHead)->head.len) {
|
if(tfRead(pWal->writeLogTfd, (*ppHead)->head.body, (*ppHead)->head.len) != (*ppHead)->head.len) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
//TODO: endian compatibility processing after read
|
//TODO: endian compatibility processing after read
|
||||||
|
|
|
@ -78,10 +78,12 @@ int walChangeFile(SWal *pWal, int64_t ver) {
|
||||||
code = tfClose(pWal->writeLogTfd);
|
code = tfClose(pWal->writeLogTfd);
|
||||||
if(code != 0) {
|
if(code != 0) {
|
||||||
//TODO
|
//TODO
|
||||||
|
return -1;
|
||||||
}
|
}
|
||||||
code = tfClose(pWal->writeIdxTfd);
|
code = tfClose(pWal->writeIdxTfd);
|
||||||
if(code != 0) {
|
if(code != 0) {
|
||||||
//TODO
|
//TODO
|
||||||
|
return -1;
|
||||||
}
|
}
|
||||||
WalFileInfo tmpInfo;
|
WalFileInfo tmpInfo;
|
||||||
tmpInfo.firstVer = ver;
|
tmpInfo.firstVer = ver;
|
||||||
|
@ -106,24 +108,19 @@ int walChangeFile(SWal *pWal, int64_t ver) {
|
||||||
|
|
||||||
pWal->writeLogTfd = logTfd;
|
pWal->writeLogTfd = logTfd;
|
||||||
pWal->writeIdxTfd = idxTfd;
|
pWal->writeIdxTfd = idxTfd;
|
||||||
return code;
|
return fileFirstVer;
|
||||||
}
|
|
||||||
|
|
||||||
int walGetVerOffset(SWal* pWal, int64_t ver) {
|
|
||||||
int code;
|
|
||||||
return 0;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
int walSeekVer(SWal *pWal, int64_t ver) {
|
int walSeekVer(SWal *pWal, int64_t ver) {
|
||||||
int code;
|
int code;
|
||||||
if(ver == pWal->lastVersion) {
|
if(ver == pWal->vers.lastVer) {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
if(ver > pWal->lastVersion || ver < pWal->firstVersion) {
|
if(ver > pWal->vers.lastVer|| ver < pWal->vers.firstVer) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
if(ver < pWal->snapshotVersion) {
|
if(ver < pWal->vers.snapshotVer) {
|
||||||
//TODO: set flag to prevent roll back
|
|
||||||
}
|
}
|
||||||
if(ver < walGetCurFileFirstVer(pWal) || (ver > walGetCurFileLastVer(pWal))) {
|
if(ver < walGetCurFileFirstVer(pWal) || (ver > walGetCurFileLastVer(pWal))) {
|
||||||
code = walChangeFile(pWal, ver);
|
code = walChangeFile(pWal, ver);
|
||||||
|
|
|
@ -17,6 +17,7 @@
|
||||||
#include "os.h"
|
#include "os.h"
|
||||||
#include "walInt.h"
|
#include "walInt.h"
|
||||||
|
|
||||||
|
#if 0
|
||||||
int32_t walGetNextFile(SWal *pWal, int64_t *nextFileId) {
|
int32_t walGetNextFile(SWal *pWal, int64_t *nextFileId) {
|
||||||
int64_t curFileId = *nextFileId;
|
int64_t curFileId = *nextFileId;
|
||||||
int64_t minFileId = INT64_MAX;
|
int64_t minFileId = INT64_MAX;
|
||||||
|
@ -116,3 +117,4 @@ int32_t walGetNewFile(SWal *pWal, int64_t *newFileId) {
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
#endif
|
||||||
|
|
|
@ -114,22 +114,22 @@ void walRemoveAllOldFiles(void *handle) {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
int32_t walCommit(SWal *pWal, int64_t ver) {
|
int32_t walCommit(SWal *pWal, int64_t ver) {
|
||||||
ASSERT(pWal->commitVersion >= pWal->snapshotVersion);
|
ASSERT(pWal->vers.commitVer >= pWal->vers.snapshotVer);
|
||||||
ASSERT(pWal->commitVersion <= pWal->lastVersion);
|
ASSERT(pWal->vers.commitVer <= pWal->vers.lastVer);
|
||||||
if(ver < pWal->commitVersion || ver > pWal->lastVersion) {
|
if(ver < pWal->vers.commitVer || ver > pWal->vers.lastVer) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
pWal->commitVersion = ver;
|
pWal->vers.commitVer = ver;
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t walRollback(SWal *pWal, int64_t ver) {
|
int32_t walRollback(SWal *pWal, int64_t ver) {
|
||||||
int code;
|
int code;
|
||||||
char fnameStr[WAL_FILE_LEN];
|
char fnameStr[WAL_FILE_LEN];
|
||||||
if(ver == pWal->lastVersion) {
|
if(ver == pWal->vers.lastVer) {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
if(ver > pWal->lastVersion || ver < pWal->commitVersion) {
|
if(ver > pWal->vers.lastVer || ver < pWal->vers.commitVer) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
pthread_mutex_lock(&pWal->mutex);
|
pthread_mutex_lock(&pWal->mutex);
|
||||||
|
@ -220,7 +220,7 @@ int32_t walRollback(SWal *pWal, int64_t ver) {
|
||||||
if(code < 0) {
|
if(code < 0) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
pWal->lastVersion = ver - 1;
|
pWal->vers.lastVer = ver - 1;
|
||||||
((WalFileInfo*)taosArrayGetLast(pWal->fileInfoSet))->lastVer = ver - 1;
|
((WalFileInfo*)taosArrayGetLast(pWal->fileInfoSet))->lastVer = ver - 1;
|
||||||
((WalFileInfo*)taosArrayGetLast(pWal->fileInfoSet))->fileSize = entry.offset;
|
((WalFileInfo*)taosArrayGetLast(pWal->fileInfoSet))->fileSize = entry.offset;
|
||||||
|
|
||||||
|
@ -228,10 +228,11 @@ int32_t walRollback(SWal *pWal, int64_t ver) {
|
||||||
pthread_mutex_unlock(&pWal->mutex);
|
pthread_mutex_unlock(&pWal->mutex);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t walBeginTakeSnapshot(SWal* pWal, int64_t ver) {
|
int32_t walBeginTakeSnapshot(SWal* pWal, int64_t ver) {
|
||||||
pWal->snapshottingVer = ver;
|
pWal->vers.verInSnapshotting = ver;
|
||||||
//check file rolling
|
//check file rolling
|
||||||
if(pWal->retentionPeriod == 0) {
|
if(pWal->cfg.retentionPeriod == 0) {
|
||||||
walRoll(pWal);
|
walRoll(pWal);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -239,10 +240,10 @@ int32_t walBeginTakeSnapshot(SWal* pWal, int64_t ver) {
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t walEndTakeSnapshot(SWal *pWal) {
|
int32_t walEndTakeSnapshot(SWal *pWal) {
|
||||||
int64_t ver = pWal->snapshottingVer;
|
int64_t ver = pWal->vers.verInSnapshotting;
|
||||||
if(ver == -1) return -1;
|
if(ver == -1) return -1;
|
||||||
|
|
||||||
pWal->snapshotVersion = ver;
|
pWal->vers.snapshotVer = ver;
|
||||||
int ts = taosGetTimestampSec();
|
int ts = taosGetTimestampSec();
|
||||||
|
|
||||||
int deleteCnt = 0;
|
int deleteCnt = 0;
|
||||||
|
@ -256,8 +257,8 @@ int32_t walEndTakeSnapshot(SWal *pWal) {
|
||||||
}
|
}
|
||||||
//iterate files, until the searched result
|
//iterate files, until the searched result
|
||||||
for(WalFileInfo* iter = pWal->fileInfoSet->pData; iter < pInfo; iter++) {
|
for(WalFileInfo* iter = pWal->fileInfoSet->pData; iter < pInfo; iter++) {
|
||||||
if(pWal->totSize > pWal->retentionSize ||
|
if(pWal->totSize > pWal->cfg.retentionSize ||
|
||||||
iter->closeTs + pWal->retentionPeriod > ts) {
|
iter->closeTs + pWal->cfg.retentionPeriod > ts) {
|
||||||
//delete according to file size or close time
|
//delete according to file size or close time
|
||||||
deleteCnt++;
|
deleteCnt++;
|
||||||
newTotSize -= iter->fileSize;
|
newTotSize -= iter->fileSize;
|
||||||
|
@ -276,12 +277,14 @@ int32_t walEndTakeSnapshot(SWal *pWal) {
|
||||||
//make new array, remove files
|
//make new array, remove files
|
||||||
taosArrayPopFrontBatch(pWal->fileInfoSet, deleteCnt);
|
taosArrayPopFrontBatch(pWal->fileInfoSet, deleteCnt);
|
||||||
if(taosArrayGetSize(pWal->fileInfoSet) == 0) {
|
if(taosArrayGetSize(pWal->fileInfoSet) == 0) {
|
||||||
pWal->firstVersion = -1;
|
pWal->writeCur = -1;
|
||||||
|
pWal->vers.firstVer = -1;
|
||||||
} else {
|
} else {
|
||||||
pWal->firstVersion = ((WalFileInfo*)taosArrayGet(pWal->fileInfoSet, 0))->firstVer;
|
pWal->vers.firstVer = ((WalFileInfo*)taosArrayGet(pWal->fileInfoSet, 0))->firstVer;
|
||||||
}
|
}
|
||||||
|
pWal->writeCur = taosArrayGetSize(pWal->fileInfoSet) - 1;;
|
||||||
pWal->totSize = newTotSize;
|
pWal->totSize = newTotSize;
|
||||||
pWal->snapshottingVer = -1;
|
pWal->vers.verInSnapshotting = -1;
|
||||||
|
|
||||||
//save snapshot ver, commit ver
|
//save snapshot ver, commit ver
|
||||||
int code = walWriteMeta(pWal);
|
int code = walWriteMeta(pWal);
|
||||||
|
@ -308,7 +311,7 @@ int walRoll(SWal *pWal) {
|
||||||
}
|
}
|
||||||
int64_t idxTfd, logTfd;
|
int64_t idxTfd, logTfd;
|
||||||
//create new file
|
//create new file
|
||||||
int64_t newFileFirstVersion = pWal->lastVersion + 1;
|
int64_t newFileFirstVersion = pWal->vers.lastVer + 1;
|
||||||
char fnameStr[WAL_FILE_LEN];
|
char fnameStr[WAL_FILE_LEN];
|
||||||
walBuildIdxName(pWal, newFileFirstVersion, fnameStr);
|
walBuildIdxName(pWal, newFileFirstVersion, fnameStr);
|
||||||
idxTfd = tfOpenCreateWrite(fnameStr);
|
idxTfd = tfOpenCreateWrite(fnameStr);
|
||||||
|
@ -340,19 +343,10 @@ int walRoll(SWal *pWal) {
|
||||||
}
|
}
|
||||||
|
|
||||||
static int walWriteIndex(SWal *pWal, int64_t ver, int64_t offset) {
|
static int walWriteIndex(SWal *pWal, int64_t ver, int64_t offset) {
|
||||||
int code = 0;
|
|
||||||
//get index file
|
|
||||||
if(!tfValid(pWal->writeIdxTfd)) {
|
|
||||||
code = TAOS_SYSTEM_ERROR(errno);
|
|
||||||
|
|
||||||
wError("vgId:%d, file:%"PRId64".idx, failed to open since %s", pWal->vgId, walGetLastFileFirstVer(pWal), strerror(errno));
|
|
||||||
return code;
|
|
||||||
}
|
|
||||||
char fnameStr[WAL_FILE_LEN];
|
|
||||||
walBuildIdxName(pWal, walGetCurFileFirstVer(pWal), fnameStr);
|
|
||||||
WalIdxEntry entry = { .ver = ver, .offset = offset };
|
WalIdxEntry entry = { .ver = ver, .offset = offset };
|
||||||
int size = tfWrite(pWal->writeIdxTfd, &entry, sizeof(WalIdxEntry));
|
int size = tfWrite(pWal->writeIdxTfd, &entry, sizeof(WalIdxEntry));
|
||||||
if(size != sizeof(WalIdxEntry)) {
|
if(size != sizeof(WalIdxEntry)) {
|
||||||
|
//TODO truncate
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
return 0;
|
return 0;
|
||||||
|
@ -363,18 +357,18 @@ int64_t walWrite(SWal *pWal, int64_t index, uint8_t msgType, const void *body, i
|
||||||
int code = 0;
|
int code = 0;
|
||||||
|
|
||||||
// no wal
|
// no wal
|
||||||
if (pWal->level == TAOS_WAL_NOLOG) return 0;
|
if (pWal->cfg.level == TAOS_WAL_NOLOG) return 0;
|
||||||
|
|
||||||
if (index == pWal->lastVersion + 1) {
|
if (index == pWal->vers.lastVer + 1) {
|
||||||
if(taosArrayGetSize(pWal->fileInfoSet) == 0) {
|
if(taosArrayGetSize(pWal->fileInfoSet) == 0) {
|
||||||
pWal->firstVersion = index;
|
pWal->vers.firstVer = index;
|
||||||
code = walRoll(pWal);
|
code = walRoll(pWal);
|
||||||
ASSERT(code == 0);
|
ASSERT(code == 0);
|
||||||
} else {
|
} else {
|
||||||
int64_t passed = walGetSeq() - pWal->lastRollSeq;
|
int64_t passed = walGetSeq() - pWal->lastRollSeq;
|
||||||
if(pWal->rollPeriod != -1 && pWal->rollPeriod != 0 && passed > pWal->rollPeriod) {
|
if(pWal->cfg.rollPeriod != -1 && pWal->cfg.rollPeriod != 0 && passed > pWal->cfg.rollPeriod) {
|
||||||
walRoll(pWal);
|
walRoll(pWal);
|
||||||
} else if(pWal->segSize != -1 && pWal->segSize != 0 && walGetLastFileSize(pWal) > pWal->segSize) {
|
} else if(pWal->cfg.segSize != -1 && pWal->cfg.segSize != 0 && walGetLastFileSize(pWal) > pWal->cfg.segSize) {
|
||||||
walRoll(pWal);
|
walRoll(pWal);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -383,35 +377,36 @@ int64_t walWrite(SWal *pWal, int64_t index, uint8_t msgType, const void *body, i
|
||||||
//must truncate explicitly first
|
//must truncate explicitly first
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
/*if (!tfValid(pWal->curLogTfd)) return 0;*/
|
/*if (!tfValid(pWal->writeLogTfd)) return -1;*/
|
||||||
|
|
||||||
pthread_mutex_lock(&pWal->mutex);
|
pthread_mutex_lock(&pWal->mutex);
|
||||||
pWal->head.head.version = index;
|
pWal->writeHead.head.version = index;
|
||||||
|
|
||||||
pWal->head.head.len = bodyLen;
|
int64_t offset = walGetCurFileOffset(pWal);
|
||||||
pWal->head.head.msgType = msgType;
|
pWal->writeHead.head.len = bodyLen;
|
||||||
pWal->head.cksumHead = walCalcHeadCksum(&pWal->head);
|
pWal->writeHead.head.msgType = msgType;
|
||||||
pWal->head.cksumBody = walCalcBodyCksum(body, bodyLen);
|
pWal->writeHead.cksumHead = walCalcHeadCksum(&pWal->writeHead);
|
||||||
|
pWal->writeHead.cksumBody = walCalcBodyCksum(body, bodyLen);
|
||||||
|
|
||||||
if (tfWrite(pWal->writeLogTfd, &pWal->head, sizeof(SWalHead)) != sizeof(SWalHead)) {
|
if (tfWrite(pWal->writeLogTfd, &pWal->writeHead, sizeof(SWalHead)) != sizeof(SWalHead)) {
|
||||||
//ftruncate
|
//ftruncate
|
||||||
code = TAOS_SYSTEM_ERROR(errno);
|
code = TAOS_SYSTEM_ERROR(errno);
|
||||||
wError("vgId:%d, file:%"PRId64".log, failed to write since %s", pWal->vgId, walGetLastFileFirstVer(pWal), strerror(errno));
|
wError("vgId:%d, file:%"PRId64".log, failed to write since %s", pWal->cfg.vgId, walGetLastFileFirstVer(pWal), strerror(errno));
|
||||||
}
|
}
|
||||||
|
|
||||||
if (tfWrite(pWal->writeLogTfd, &body, bodyLen) != bodyLen) {
|
if (tfWrite(pWal->writeLogTfd, (char*)body, bodyLen) != bodyLen) {
|
||||||
//ftruncate
|
//ftruncate
|
||||||
code = TAOS_SYSTEM_ERROR(errno);
|
code = TAOS_SYSTEM_ERROR(errno);
|
||||||
wError("vgId:%d, file:%"PRId64".log, failed to write since %s", pWal->vgId, walGetLastFileFirstVer(pWal), strerror(errno));
|
wError("vgId:%d, file:%"PRId64".log, failed to write since %s", pWal->cfg.vgId, walGetLastFileFirstVer(pWal), strerror(errno));
|
||||||
}
|
}
|
||||||
code = walWriteIndex(pWal, index, walGetCurFileOffset(pWal));
|
code = walWriteIndex(pWal, index, offset);
|
||||||
if(code != 0) {
|
if(code != 0) {
|
||||||
//TODO
|
//TODO
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
//set status
|
//set status
|
||||||
pWal->lastVersion = index;
|
pWal->vers.lastVer = index;
|
||||||
pWal->totSize += sizeof(SWalHead) + bodyLen;
|
pWal->totSize += sizeof(SWalHead) + bodyLen;
|
||||||
walGetCurFileInfo(pWal)->lastVer = index;
|
walGetCurFileInfo(pWal)->lastVer = index;
|
||||||
walGetCurFileInfo(pWal)->fileSize += sizeof(SWalHead) + bodyLen;
|
walGetCurFileInfo(pWal)->fileSize += sizeof(SWalHead) + bodyLen;
|
||||||
|
@ -422,10 +417,10 @@ int64_t walWrite(SWal *pWal, int64_t index, uint8_t msgType, const void *body, i
|
||||||
}
|
}
|
||||||
|
|
||||||
void walFsync(SWal *pWal, bool forceFsync) {
|
void walFsync(SWal *pWal, bool forceFsync) {
|
||||||
if (forceFsync || (pWal->level == TAOS_WAL_FSYNC && pWal->fsyncPeriod == 0)) {
|
if (forceFsync || (pWal->cfg.level == TAOS_WAL_FSYNC && pWal->cfg.fsyncPeriod == 0)) {
|
||||||
wTrace("vgId:%d, fileId:%"PRId64".log, do fsync", pWal->vgId, walGetCurFileFirstVer(pWal));
|
wTrace("vgId:%d, fileId:%"PRId64".log, do fsync", pWal->cfg.vgId, walGetCurFileFirstVer(pWal));
|
||||||
if (tfFsync(pWal->writeLogTfd) < 0) {
|
if (tfFsync(pWal->writeLogTfd) < 0) {
|
||||||
wError("vgId:%d, file:%"PRId64".log, fsync failed since %s", pWal->vgId, walGetCurFileFirstVer(pWal), strerror(errno));
|
wError("vgId:%d, file:%"PRId64".log, fsync failed since %s", pWal->cfg.vgId, walGetCurFileFirstVer(pWal), strerror(errno));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -498,29 +493,29 @@ int32_t walGetWalFile(void *handle, char *fileName, int64_t *fileId) {
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
static int walValidateOffset(SWal* pWal, int64_t ver) {
|
/*static int walValidateOffset(SWal* pWal, int64_t ver) {*/
|
||||||
int code = 0;
|
/*int code = 0;*/
|
||||||
SWalHead *pHead = NULL;
|
/*SWalHead *pHead = NULL;*/
|
||||||
code = (int)walRead(pWal, &pHead, ver);
|
/*code = (int)walRead(pWal, &pHead, ver);*/
|
||||||
if(pHead->head.version != ver) {
|
/*if(pHead->head.version != ver) {*/
|
||||||
return -1;
|
/*return -1;*/
|
||||||
}
|
/*}*/
|
||||||
return 0;
|
/*return 0;*/
|
||||||
}
|
/*}*/
|
||||||
|
|
||||||
static int64_t walGetOffset(SWal* pWal, int64_t ver) {
|
/*static int64_t walGetOffset(SWal* pWal, int64_t ver) {*/
|
||||||
int code = walSeekVer(pWal, ver);
|
/*int code = walSeekVer(pWal, ver);*/
|
||||||
if(code != 0) {
|
/*if(code != 0) {*/
|
||||||
return -1;
|
/*return -1;*/
|
||||||
}
|
/*}*/
|
||||||
|
|
||||||
code = walValidateOffset(pWal, ver);
|
/*code = walValidateOffset(pWal, ver);*/
|
||||||
if(code != 0) {
|
/*if(code != 0) {*/
|
||||||
return -1;
|
/*return -1;*/
|
||||||
}
|
/*}*/
|
||||||
|
|
||||||
return 0;
|
/*return 0;*/
|
||||||
}
|
/*}*/
|
||||||
|
|
||||||
#if 0
|
#if 0
|
||||||
static int32_t walSkipCorruptedRecord(SWal *pWal, SWalHead *pHead, int64_t tfd, int64_t *offset) {
|
static int32_t walSkipCorruptedRecord(SWal *pWal, SWalHead *pHead, int64_t tfd, int64_t *offset) {
|
||||||
|
|
|
@ -5,6 +5,9 @@
|
||||||
|
|
||||||
#include "walInt.h"
|
#include "walInt.h"
|
||||||
|
|
||||||
|
const char* ranStr = "tvapq02tcp";
|
||||||
|
const int ranStrLen = strlen(ranStr);
|
||||||
|
|
||||||
class WalCleanEnv : public ::testing::Test {
|
class WalCleanEnv : public ::testing::Test {
|
||||||
protected:
|
protected:
|
||||||
static void SetUpTestCase() {
|
static void SetUpTestCase() {
|
||||||
|
@ -18,12 +21,15 @@ class WalCleanEnv : public ::testing::Test {
|
||||||
|
|
||||||
void SetUp() override {
|
void SetUp() override {
|
||||||
taosRemoveDir(pathName);
|
taosRemoveDir(pathName);
|
||||||
SWalCfg* pCfg = (SWalCfg*)malloc(sizeof(SWal));
|
SWalCfg* pCfg = (SWalCfg*)malloc(sizeof(SWalCfg));
|
||||||
memset(pCfg, 0, sizeof(SWalCfg));
|
memset(pCfg, 0, sizeof(SWalCfg));
|
||||||
pCfg->rollPeriod = -1;
|
pCfg->rollPeriod = -1;
|
||||||
pCfg->segSize = -1;
|
pCfg->segSize = -1;
|
||||||
pCfg->walLevel = TAOS_WAL_FSYNC;
|
pCfg->retentionPeriod = 0;
|
||||||
|
pCfg->retentionSize = 0;
|
||||||
|
pCfg->level = TAOS_WAL_FSYNC;
|
||||||
pWal = walOpen(pathName, pCfg);
|
pWal = walOpen(pathName, pCfg);
|
||||||
|
free(pCfg);
|
||||||
ASSERT(pWal != NULL);
|
ASSERT(pWal != NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -49,11 +55,13 @@ class WalCleanDeleteEnv : public ::testing::Test {
|
||||||
|
|
||||||
void SetUp() override {
|
void SetUp() override {
|
||||||
taosRemoveDir(pathName);
|
taosRemoveDir(pathName);
|
||||||
SWalCfg* pCfg = (SWalCfg*)malloc(sizeof(SWal));
|
SWalCfg* pCfg = (SWalCfg*)malloc(sizeof(SWalCfg));
|
||||||
memset(pCfg, 0, sizeof(SWalCfg));
|
memset(pCfg, 0, sizeof(SWalCfg));
|
||||||
pCfg->retentionPeriod = 0;
|
pCfg->retentionPeriod = 0;
|
||||||
pCfg->walLevel = TAOS_WAL_FSYNC;
|
pCfg->retentionSize = 0;
|
||||||
|
pCfg->level = TAOS_WAL_FSYNC;
|
||||||
pWal = walOpen(pathName, pCfg);
|
pWal = walOpen(pathName, pCfg);
|
||||||
|
free(pCfg);
|
||||||
ASSERT(pWal != NULL);
|
ASSERT(pWal != NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -77,13 +85,22 @@ class WalKeepEnv : public ::testing::Test {
|
||||||
walCleanUp();
|
walCleanUp();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void walResetEnv() {
|
||||||
|
TearDown();
|
||||||
|
taosRemoveDir(pathName);
|
||||||
|
SetUp();
|
||||||
|
}
|
||||||
|
|
||||||
void SetUp() override {
|
void SetUp() override {
|
||||||
SWalCfg* pCfg = (SWalCfg*)malloc(sizeof(SWal));
|
SWalCfg* pCfg = (SWalCfg*)malloc(sizeof(SWalCfg));
|
||||||
memset(pCfg, 0, sizeof(SWalCfg));
|
memset(pCfg, 0, sizeof(SWalCfg));
|
||||||
pCfg->rollPeriod = -1;
|
pCfg->rollPeriod = -1;
|
||||||
pCfg->segSize = -1;
|
pCfg->segSize = -1;
|
||||||
pCfg->walLevel = TAOS_WAL_FSYNC;
|
pCfg->retentionPeriod = 0;
|
||||||
|
pCfg->retentionSize = 0;
|
||||||
|
pCfg->level = TAOS_WAL_FSYNC;
|
||||||
pWal = walOpen(pathName, pCfg);
|
pWal = walOpen(pathName, pCfg);
|
||||||
|
free(pCfg);
|
||||||
ASSERT(pWal != NULL);
|
ASSERT(pWal != NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -124,6 +141,7 @@ TEST_F(WalCleanEnv, serialize) {
|
||||||
ASSERT(code == 0);
|
ASSERT(code == 0);
|
||||||
char*ss = walMetaSerialize(pWal);
|
char*ss = walMetaSerialize(pWal);
|
||||||
printf("%s\n", ss);
|
printf("%s\n", ss);
|
||||||
|
free(ss);
|
||||||
code = walWriteMeta(pWal);
|
code = walWriteMeta(pWal);
|
||||||
ASSERT(code == 0);
|
ASSERT(code == 0);
|
||||||
}
|
}
|
||||||
|
@ -140,94 +158,134 @@ TEST_F(WalCleanEnv, removeOldMeta) {
|
||||||
ASSERT(code == 0);
|
ASSERT(code == 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
//TEST_F(WalKeepEnv, readOldMeta) {
|
TEST_F(WalKeepEnv, readOldMeta) {
|
||||||
//int code = walRollFileInfo(pWal);
|
walResetEnv();
|
||||||
//ASSERT(code == 0);
|
int code;
|
||||||
//code = walWriteMeta(pWal);
|
|
||||||
//ASSERT(code == 0);
|
|
||||||
//code = walRollFileInfo(pWal);
|
|
||||||
//ASSERT(code == 0);
|
|
||||||
//code = walWriteMeta(pWal);
|
|
||||||
//ASSERT(code == 0);
|
|
||||||
//char*oldss = walMetaSerialize(pWal);
|
|
||||||
|
|
||||||
//TearDown();
|
for(int i = 0; i < 10; i++) {
|
||||||
//SetUp();
|
code = walWrite(pWal, i, i+1, (void*)ranStr, ranStrLen);
|
||||||
//code = walReadMeta(pWal);
|
ASSERT_EQ(code, 0);
|
||||||
//ASSERT(code == 0);
|
ASSERT_EQ(pWal->vers.lastVer, i);
|
||||||
//char* newss = walMetaSerialize(pWal);
|
code = walWrite(pWal, i+2, i, (void*)ranStr, ranStrLen);
|
||||||
|
ASSERT_EQ(code, -1);
|
||||||
|
ASSERT_EQ(pWal->vers.lastVer, i);
|
||||||
|
}
|
||||||
|
char* oldss = walMetaSerialize(pWal);
|
||||||
|
|
||||||
//int len = strlen(oldss);
|
TearDown();
|
||||||
//ASSERT_EQ(len, strlen(newss));
|
SetUp();
|
||||||
//for(int i = 0; i < len; i++) {
|
|
||||||
//EXPECT_EQ(oldss[i], newss[i]);
|
ASSERT_EQ(pWal->vers.firstVer, 0);
|
||||||
//}
|
ASSERT_EQ(pWal->vers.lastVer, 9);
|
||||||
//}
|
|
||||||
|
char* newss = walMetaSerialize(pWal);
|
||||||
|
|
||||||
|
int len = strlen(oldss);
|
||||||
|
ASSERT_EQ(len, strlen(newss));
|
||||||
|
for(int i = 0; i < len; i++) {
|
||||||
|
EXPECT_EQ(oldss[i], newss[i]);
|
||||||
|
}
|
||||||
|
free(oldss);
|
||||||
|
free(newss);
|
||||||
|
}
|
||||||
|
|
||||||
TEST_F(WalCleanEnv, write) {
|
TEST_F(WalCleanEnv, write) {
|
||||||
const char* ranStr = "tvapq02tcp";
|
|
||||||
const int len = strlen(ranStr);
|
|
||||||
int code;
|
int code;
|
||||||
for(int i = 0; i < 10; i++) {
|
for(int i = 0; i < 10; i++) {
|
||||||
code = walWrite(pWal, i, i+1, (void*)ranStr, len);
|
code = walWrite(pWal, i, i+1, (void*)ranStr, ranStrLen);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
ASSERT_EQ(pWal->lastVersion, i);
|
ASSERT_EQ(pWal->vers.lastVer, i);
|
||||||
code = walWrite(pWal, i+2, i, (void*)ranStr, len);
|
code = walWrite(pWal, i+2, i, (void*)ranStr, ranStrLen);
|
||||||
ASSERT_EQ(code, -1);
|
ASSERT_EQ(code, -1);
|
||||||
ASSERT_EQ(pWal->lastVersion, i);
|
ASSERT_EQ(pWal->vers.lastVer, i);
|
||||||
}
|
}
|
||||||
code = walWriteMeta(pWal);
|
code = walWriteMeta(pWal);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_F(WalCleanEnv, rollback) {
|
TEST_F(WalCleanEnv, rollback) {
|
||||||
const char* ranStr = "tvapq02tcp";
|
|
||||||
const int len = strlen(ranStr);
|
|
||||||
int code;
|
int code;
|
||||||
for(int i = 0; i < 10; i++) {
|
for(int i = 0; i < 10; i++) {
|
||||||
code = walWrite(pWal, i, i+1, (void*)ranStr, len);
|
code = walWrite(pWal, i, i+1, (void*)ranStr, ranStrLen);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
ASSERT_EQ(pWal->lastVersion, i);
|
ASSERT_EQ(pWal->vers.lastVer, i);
|
||||||
}
|
}
|
||||||
code = walRollback(pWal, 5);
|
code = walRollback(pWal, 5);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
ASSERT_EQ(pWal->lastVersion, 4);
|
ASSERT_EQ(pWal->vers.lastVer, 4);
|
||||||
code = walRollback(pWal, 3);
|
code = walRollback(pWal, 3);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
ASSERT_EQ(pWal->lastVersion, 2);
|
ASSERT_EQ(pWal->vers.lastVer, 2);
|
||||||
code = walWriteMeta(pWal);
|
code = walWriteMeta(pWal);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_F(WalCleanDeleteEnv, roll) {
|
TEST_F(WalCleanDeleteEnv, roll) {
|
||||||
const char* ranStr = "tvapq02tcp";
|
|
||||||
const int len = strlen(ranStr);
|
|
||||||
int code;
|
int code;
|
||||||
int i;
|
int i;
|
||||||
for(i = 0; i < 100; i++) {
|
for(i = 0; i < 100; i++) {
|
||||||
code = walWrite(pWal, i, 0, (void*)ranStr, len);
|
code = walWrite(pWal, i, 0, (void*)ranStr, ranStrLen);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
ASSERT_EQ(pWal->lastVersion, i);
|
ASSERT_EQ(pWal->vers.lastVer, i);
|
||||||
code = walCommit(pWal, i);
|
code = walCommit(pWal, i);
|
||||||
ASSERT_EQ(pWal->commitVersion, i);
|
ASSERT_EQ(pWal->vers.commitVer, i);
|
||||||
}
|
}
|
||||||
|
|
||||||
walBeginTakeSnapshot(pWal, i-1);
|
walBeginTakeSnapshot(pWal, i-1);
|
||||||
ASSERT_EQ(pWal->snapshottingVer, i-1);
|
ASSERT_EQ(pWal->vers.verInSnapshotting, i-1);
|
||||||
walEndTakeSnapshot(pWal);
|
walEndTakeSnapshot(pWal);
|
||||||
ASSERT_EQ(pWal->snapshotVersion, i-1);
|
ASSERT_EQ(pWal->vers.snapshotVer, i-1);
|
||||||
ASSERT_EQ(pWal->snapshottingVer, -1);
|
ASSERT_EQ(pWal->vers.verInSnapshotting, -1);
|
||||||
|
|
||||||
code = walWrite(pWal, 5, 0, (void*)ranStr, len);
|
code = walWrite(pWal, 5, 0, (void*)ranStr, ranStrLen);
|
||||||
ASSERT_NE(code, 0);
|
ASSERT_NE(code, 0);
|
||||||
|
|
||||||
for(; i < 200; i++) {
|
for(; i < 200; i++) {
|
||||||
code = walWrite(pWal, i, 0, (void*)ranStr, len);
|
code = walWrite(pWal, i, 0, (void*)ranStr, ranStrLen);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
code = walCommit(pWal, i);
|
code = walCommit(pWal, i);
|
||||||
ASSERT_EQ(pWal->commitVersion, i);
|
ASSERT_EQ(pWal->vers.commitVer, i);
|
||||||
}
|
}
|
||||||
|
|
||||||
code = walWriteMeta(pWal);
|
code = walBeginTakeSnapshot(pWal, i - 1);
|
||||||
|
ASSERT_EQ(code, 0);
|
||||||
|
code = walEndTakeSnapshot(pWal);
|
||||||
ASSERT_EQ(code, 0);
|
ASSERT_EQ(code, 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
TEST_F(WalKeepEnv, readHandleRead) {
|
||||||
|
walResetEnv();
|
||||||
|
int code;
|
||||||
|
SWalReadHandle* pRead = walOpenReadHandle(pWal);
|
||||||
|
ASSERT(pRead != NULL);
|
||||||
|
|
||||||
|
int i ;
|
||||||
|
for(i = 0; i < 100; i++) {
|
||||||
|
char newStr[100];
|
||||||
|
sprintf(newStr, "%s-%d", ranStr, i);
|
||||||
|
int len = strlen(newStr);
|
||||||
|
code = walWrite(pWal, i, 0, newStr, len);
|
||||||
|
ASSERT_EQ(code, 0);
|
||||||
|
}
|
||||||
|
for(int i = 0; i < 1000; i++) {
|
||||||
|
int ver = rand() % 100;
|
||||||
|
code = walReadWithHandle(pRead, ver);
|
||||||
|
ASSERT_EQ(code, 0);
|
||||||
|
|
||||||
|
//printf("rrbody: \n");
|
||||||
|
//for(int i = 0; i < pRead->pHead->head.len; i++) {
|
||||||
|
//printf("%d ", pRead->pHead->head.body[i]);
|
||||||
|
//}
|
||||||
|
//printf("\n");
|
||||||
|
|
||||||
|
ASSERT_EQ(pRead->pHead->head.version, ver);
|
||||||
|
ASSERT_EQ(pRead->curVersion, ver+1);
|
||||||
|
char newStr[100];
|
||||||
|
sprintf(newStr, "%s-%d", ranStr, ver);
|
||||||
|
int len = strlen(newStr);
|
||||||
|
ASSERT_EQ(pRead->pHead->head.len, len);
|
||||||
|
for(int j = 0; j < len; j++) {
|
||||||
|
EXPECT_EQ(newStr[j], pRead->pHead->head.body[j]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
@ -42,7 +42,7 @@ bool taosComparePthread(pthread_t first, pthread_t second) { return first.p == s
|
||||||
|
|
||||||
int32_t taosGetPId() { return GetCurrentProcessId(); }
|
int32_t taosGetPId() { return GetCurrentProcessId(); }
|
||||||
|
|
||||||
int32_t taosGetCurrentAPPName(char* name, int32_t* len) {
|
int32_t taosGetAppName(char* name, int32_t* len) {
|
||||||
char filepath[1024] = {0};
|
char filepath[1024] = {0};
|
||||||
|
|
||||||
GetModuleFileName(NULL, filepath, MAX_PATH);
|
GetModuleFileName(NULL, filepath, MAX_PATH);
|
||||||
|
@ -358,7 +358,7 @@ bool taosComparePthread(pthread_t first, pthread_t second) { return pthread_equa
|
||||||
|
|
||||||
int32_t taosGetPId() { return (int32_t)getpid(); }
|
int32_t taosGetPId() { return (int32_t)getpid(); }
|
||||||
|
|
||||||
int32_t taosGetCurrentAPPName(char *name, int32_t *len) {
|
int32_t taosGetAppName(char *name, int32_t *len) {
|
||||||
char buf[PATH_MAX + 1];
|
char buf[PATH_MAX + 1];
|
||||||
buf[0] = '\0';
|
buf[0] = '\0';
|
||||||
proc_name(getpid(), buf, sizeof(buf) - 1);
|
proc_name(getpid(), buf, sizeof(buf) - 1);
|
||||||
|
@ -392,7 +392,7 @@ void taosResetPthread(pthread_t* thread) { *thread = 0; }
|
||||||
bool taosComparePthread(pthread_t first, pthread_t second) { return first == second; }
|
bool taosComparePthread(pthread_t first, pthread_t second) { return first == second; }
|
||||||
int32_t taosGetPId() { return getpid(); }
|
int32_t taosGetPId() { return getpid(); }
|
||||||
|
|
||||||
int32_t taosGetCurrentAPPName(char* name, int32_t* len) {
|
int32_t taosGetAppName(char* name, int32_t* len) {
|
||||||
const char* self = "/proc/self/exe";
|
const char* self = "/proc/self/exe";
|
||||||
char path[PATH_MAX] = {0};
|
char path[PATH_MAX] = {0};
|
||||||
|
|
||||||
|
|
|
@ -275,13 +275,13 @@ char *strsep(char **stringp, const char *delim) {
|
||||||
}
|
}
|
||||||
|
|
||||||
char *getpass(const char *prefix) {
|
char *getpass(const char *prefix) {
|
||||||
static char passwd[TSDB_KEY_LEN] = {0};
|
static char passwd[TSDB_PASSWORD_LEN] = {0};
|
||||||
memset(passwd, 0, TSDB_KEY_LEN);
|
memset(passwd, 0, TSDB_PASSWORD_LEN);
|
||||||
//printf("%s", prefix);
|
//printf("%s", prefix);
|
||||||
|
|
||||||
int32_t index = 0;
|
int32_t index = 0;
|
||||||
char ch;
|
char ch;
|
||||||
while (index < TSDB_KEY_LEN) {
|
while (index < TSDB_PASSWORD_LEN) {
|
||||||
ch = getch();
|
ch = getch();
|
||||||
if (ch == '\n' || ch == '\r') {
|
if (ch == '\n' || ch == '\r') {
|
||||||
break;
|
break;
|
||||||
|
|
|
@ -714,7 +714,7 @@ static void taosGetSystemLocale() { // get and set default locale
|
||||||
//printf("locale not configured, set to system default:%s", tsLocale);
|
//printf("locale not configured, set to system default:%s", tsLocale);
|
||||||
}
|
}
|
||||||
|
|
||||||
/* if user does not specify the charset, extract it from locale */
|
// if user does not specify the charset, extract it from locale
|
||||||
char *str = strrchr(tsLocale, sep);
|
char *str = strrchr(tsLocale, sep);
|
||||||
if (str != NULL) {
|
if (str != NULL) {
|
||||||
str++;
|
str++;
|
||||||
|
@ -1118,13 +1118,13 @@ char *taosGetCmdlineByPID(int pid) {
|
||||||
SysNameInfo taosGetSysNameInfo() {
|
SysNameInfo taosGetSysNameInfo() {
|
||||||
SysNameInfo info = {0};
|
SysNameInfo info = {0};
|
||||||
|
|
||||||
struct utsname buf;
|
struct utsname uts;
|
||||||
if (!uname(&buf)) {
|
if (!uname(&uts)) {
|
||||||
info.sysname = buf.sysname;
|
info.sysname = strdup(uts.sysname);
|
||||||
info.sysname == buf.nodename;
|
info.nodename = strdup(uts.nodename);
|
||||||
info.sysname = buf.release;
|
info.release = strdup(uts.release);
|
||||||
info.sysname = buf.version;
|
info.version = strdup(uts.version);
|
||||||
info.sysname = buf.machine;
|
info.machine = strdup(uts.machine);
|
||||||
}
|
}
|
||||||
|
|
||||||
return info;
|
return info;
|
||||||
|
|
|
@ -58,24 +58,31 @@ static int32_t taosArrayResize(SArray* pArray) {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
void* taosArrayAddBatch(SArray* pArray, const void* pData, int nEles) {
|
int32_t taosArrayEnsureCap(SArray* pArray, size_t newCap) {
|
||||||
if (pArray == NULL || pData == NULL) {
|
if (newCap > pArray->capacity) {
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (pArray->size + nEles > pArray->capacity) {
|
|
||||||
size_t tsize = (pArray->capacity << 1u);
|
size_t tsize = (pArray->capacity << 1u);
|
||||||
while (pArray->size + nEles > tsize) {
|
while (newCap > tsize) {
|
||||||
tsize = (tsize << 1u);
|
tsize = (tsize << 1u);
|
||||||
}
|
}
|
||||||
|
|
||||||
pArray->pData = realloc(pArray->pData, tsize * pArray->elemSize);
|
pArray->pData = realloc(pArray->pData, tsize * pArray->elemSize);
|
||||||
if (pArray->pData == NULL) {
|
if (pArray->pData == NULL) {
|
||||||
return NULL;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
pArray->capacity = tsize;
|
pArray->capacity = tsize;
|
||||||
}
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void* taosArrayAddBatch(SArray* pArray, const void* pData, int nEles) {
|
||||||
|
if (pArray == NULL || pData == NULL) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(taosArrayEnsureCap(pArray, pArray->size + nEles) != 0){
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
void* dst = TARRAY_GET_ELEM(pArray, pArray->size);
|
void* dst = TARRAY_GET_ELEM(pArray, pArray->size);
|
||||||
memcpy(dst, pData, pArray->elemSize * nEles);
|
memcpy(dst, pData, pArray->elemSize * nEles);
|
||||||
|
@ -243,7 +250,7 @@ void taosArrayPopFrontBatch(SArray* pArray, size_t cnt) {
|
||||||
if(pArray->size == 0) {
|
if(pArray->size == 0) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
memmove(pArray->pData, (char*)pArray->pData + cnt * pArray->elemSize, pArray->size);
|
memmove(pArray->pData, (char*)pArray->pData + cnt * pArray->elemSize, pArray->size * pArray->elemSize);
|
||||||
}
|
}
|
||||||
|
|
||||||
void taosArrayPopTailBatch(SArray* pArray, size_t cnt) {
|
void taosArrayPopTailBatch(SArray* pArray, size_t cnt) {
|
||||||
|
|
|
@ -282,7 +282,7 @@ static void taosReadConfigOption(const char *option, char *value, char *value2,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void taosInitConfigOption(SGlobalCfg cfg) {
|
void taosAddConfigOption(SGlobalCfg cfg) {
|
||||||
tsGlobalConfig[tsGlobalConfigNum++] = cfg;
|
tsGlobalConfig[tsGlobalConfigNum++] = cfg;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -335,7 +335,7 @@ void taosReadGlobalLogCfg() {
|
||||||
fclose(fp);
|
fclose(fp);
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t taosReadGlobalCfg() {
|
int32_t taosReadCfgFromFile() {
|
||||||
char * line, *option, *value, *value2, *value3;
|
char * line, *option, *value, *value2, *value3;
|
||||||
int olen, vlen, vlen2, vlen3;
|
int olen, vlen, vlen2, vlen3;
|
||||||
char fileName[PATH_MAX] = {0};
|
char fileName[PATH_MAX] = {0};
|
||||||
|
@ -396,7 +396,7 @@ int32_t taosReadGlobalCfg() {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
void taosPrintGlobalCfg() {
|
void taosPrintCfg() {
|
||||||
uInfo(" taos config & system info:");
|
uInfo(" taos config & system info:");
|
||||||
uInfo("==================================");
|
uInfo("==================================");
|
||||||
|
|
||||||
|
@ -443,7 +443,6 @@ void taosPrintGlobalCfg() {
|
||||||
}
|
}
|
||||||
|
|
||||||
taosPrintOsInfo();
|
taosPrintOsInfo();
|
||||||
// taosPrintDataDirCfg();
|
|
||||||
uInfo("==================================");
|
uInfo("==================================");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -127,6 +127,7 @@ TAOS_DEFINE_ERROR(TSDB_CODE_TSC_DUP_TAG_NAMES, "duplicated tag names"
|
||||||
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_INVALID_JSON, "Invalid JSON format")
|
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_INVALID_JSON, "Invalid JSON format")
|
||||||
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_INVALID_JSON_TYPE, "Invalid JSON data type")
|
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_INVALID_JSON_TYPE, "Invalid JSON data type")
|
||||||
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_VALUE_OUT_OF_RANGE, "Value out of range")
|
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_VALUE_OUT_OF_RANGE, "Value out of range")
|
||||||
|
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_INVALID_INPUT, "Invalid tsc input")
|
||||||
|
|
||||||
// mnode
|
// mnode
|
||||||
TAOS_DEFINE_ERROR(TSDB_CODE_MND_MSG_NOT_PROCESSED, "Message not processed")
|
TAOS_DEFINE_ERROR(TSDB_CODE_MND_MSG_NOT_PROCESSED, "Message not processed")
|
||||||
|
@ -497,6 +498,15 @@ TAOS_DEFINE_ERROR(TSDB_CODE_FS_FILE_ALREADY_EXISTS, "tfs file already exis
|
||||||
TAOS_DEFINE_ERROR(TSDB_CODE_FS_INVLD_LEVEL, "tfs invalid level")
|
TAOS_DEFINE_ERROR(TSDB_CODE_FS_INVLD_LEVEL, "tfs invalid level")
|
||||||
TAOS_DEFINE_ERROR(TSDB_CODE_FS_NO_VALID_DISK, "tfs no valid disk")
|
TAOS_DEFINE_ERROR(TSDB_CODE_FS_NO_VALID_DISK, "tfs no valid disk")
|
||||||
|
|
||||||
|
// catalog
|
||||||
|
TAOS_DEFINE_ERROR(TSDB_CODE_CTG_INTERNAL_ERROR, "catalog interval error")
|
||||||
|
TAOS_DEFINE_ERROR(TSDB_CODE_CTG_INVALID_INPUT, "invalid catalog input parameters")
|
||||||
|
TAOS_DEFINE_ERROR(TSDB_CODE_CTG_NOT_READY, "catalog is not ready")
|
||||||
|
TAOS_DEFINE_ERROR(TSDB_CODE_CTG_MEM_ERROR, "catalog memory error")
|
||||||
|
TAOS_DEFINE_ERROR(TSDB_CODE_CTG_SYS_ERROR, "catalog system error")
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifdef TAOS_ERROR_C
|
#ifdef TAOS_ERROR_C
|
||||||
};
|
};
|
||||||
#endif
|
#endif
|
||||||
|
@ -545,4 +555,4 @@ const char* tstrerror(int32_t err) {
|
||||||
return "";
|
return "";
|
||||||
}
|
}
|
||||||
|
|
||||||
const char* terrstr() { return tstrerror(terrno); }
|
const char* terrstr() { return tstrerror(terrno); }
|
||||||
|
|
|
@ -99,6 +99,8 @@ int32_t wDebugFlag = 135;
|
||||||
int32_t tsdbDebugFlag = 131;
|
int32_t tsdbDebugFlag = 131;
|
||||||
int32_t cqDebugFlag = 131;
|
int32_t cqDebugFlag = 131;
|
||||||
int32_t fsDebugFlag = 135;
|
int32_t fsDebugFlag = 135;
|
||||||
|
int32_t ctgDebugFlag = 131;
|
||||||
|
|
||||||
|
|
||||||
int64_t dbgEmptyW = 0;
|
int64_t dbgEmptyW = 0;
|
||||||
int64_t dbgWN = 0;
|
int64_t dbgWN = 0;
|
||||||
|
@ -112,7 +114,7 @@ static int32_t taosPushLogBuffer(SLogBuff *tLogBuff, char *msg, int32_t msgLen
|
||||||
static SLogBuff *taosLogBuffNew(int32_t bufSize);
|
static SLogBuff *taosLogBuffNew(int32_t bufSize);
|
||||||
static void taosCloseLogByFd(int32_t oldFd);
|
static void taosCloseLogByFd(int32_t oldFd);
|
||||||
static int32_t taosOpenLogFile(char *fn, int32_t maxLines, int32_t maxFileNum);
|
static int32_t taosOpenLogFile(char *fn, int32_t maxLines, int32_t maxFileNum);
|
||||||
extern void taosPrintGlobalCfg();
|
extern void taosPrintCfg();
|
||||||
static int32_t taosCompressFile(char *srcFileName, char *destFileName);
|
static int32_t taosCompressFile(char *srcFileName, char *destFileName);
|
||||||
|
|
||||||
static int32_t taosStartLog() {
|
static int32_t taosStartLog() {
|
||||||
|
@ -222,7 +224,7 @@ static void *taosThreadToOpenNewFile(void *param) {
|
||||||
|
|
||||||
uInfo(" new log file:%d is opened", tsLogObj.flag);
|
uInfo(" new log file:%d is opened", tsLogObj.flag);
|
||||||
uInfo("==================================");
|
uInfo("==================================");
|
||||||
taosPrintGlobalCfg();
|
taosPrintCfg();
|
||||||
taosKeepOldLog(keepName);
|
taosKeepOldLog(keepName);
|
||||||
|
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue