Merge pull request #29764 from taosdata/enh/TD-33678-MAIN

fix: oem taosdump taosBenchmark --version support -main
This commit is contained in:
Linhe Huo 2025-02-17 10:41:06 +08:00 committed by GitHub
commit f21ffcb510
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
17 changed files with 48 additions and 101 deletions

View File

@ -15,6 +15,18 @@ IF (TD_PRODUCT_NAME)
ADD_DEFINITIONS(-DTD_PRODUCT_NAME="${TD_PRODUCT_NAME}")
ENDIF ()
IF (CUS_NAME)
ADD_DEFINITIONS(-DCUS_NAME="${CUS_NAME}")
ENDIF ()
IF (CUS_PROMPT)
ADD_DEFINITIONS(-DCUS_PROMPT="${CUS_PROMPT}")
ENDIF ()
IF (CUS_EMAIL)
ADD_DEFINITIONS(-DCUS_EMAIL="${CUS_EMAIL}")
ENDIF ()
find_program(HAVE_GIT NAMES git)
IF (DEFINED GITINFO)

View File

@ -31,9 +31,7 @@
extern "C" {
#endif
#if defined(CUS_NAME) || defined(CUS_PROMPT) || defined(CUS_EMAIL)
#include "cus_name.h"
#endif
#ifdef WINDOWS

View File

@ -16,12 +16,23 @@
#ifndef _CUS_NAME_H_
#define _CUS_NAME_H_
//
// support OEM
//
#ifndef TD_PRODUCT_NAME
#ifdef TD_ENTERPRISE
#define TD_PRODUCT_NAME "TDengine Enterprise Edition"
#else
#define TD_PRODUCT_NAME "TDengine Community Edition"
#endif
#endif
#ifndef CUS_NAME
#define CUS_NAME "TDengine"
#endif
#ifndef CUS_PROMPT
#define CUS_PROMPT "taos"
#define CUS_PROMPT "taos"
#endif
#ifndef CUS_EMAIL

View File

@ -17,6 +17,7 @@
#define _TD_UTIL_DEF_H_
#include "os.h"
#include "cus_name.h"
#ifdef __cplusplus
extern "C" {
@ -79,14 +80,6 @@ extern const int32_t TYPE_BYTES[21];
#define TSDB_DEFAULT_PASS "taosdata"
#endif
#ifndef TD_PRODUCT_NAME
#ifdef TD_ENTERPRISE
#define TD_PRODUCT_NAME "TDengine Enterprise Edition"
#else
#define TD_PRODUCT_NAME "TDengine Community Edition"
#endif
#endif
#define TSDB_TRUE 1
#define TSDB_FALSE 0
#define TSDB_OK 0

View File

@ -38,13 +38,7 @@
#include "tversion.h"
#include "tconv.h"
#if defined(CUS_NAME) || defined(CUS_PROMPT) || defined(CUS_EMAIL)
#include "cus_name.h"
#endif
#ifndef CUS_PROMPT
#define CUS_PROMPT "taos"
#endif
#define TSC_VAR_NOT_RELEASE 1
#define TSC_VAR_RELEASED 0

View File

@ -28,9 +28,6 @@
#include "tutil.h"
#if defined(CUS_NAME) || defined(CUS_PROMPT) || defined(CUS_EMAIL)
#include "cus_name.h"
#endif
#define CONFIG_PATH_LEN (TSDB_FILENAME_LEN + 12)
#define CONFIG_FILE_LEN (CONFIG_PATH_LEN + 32)

View File

@ -30,21 +30,8 @@
#include "jemalloc/jemalloc.h"
#endif
#if defined(CUS_NAME) || defined(CUS_PROMPT) || defined(CUS_EMAIL)
#include "cus_name.h"
#else
#ifndef CUS_NAME
#define CUS_NAME "TDengine"
#endif
#ifndef CUS_PROMPT
#define CUS_PROMPT "taos"
#endif
#ifndef CUS_EMAIL
#define CUS_EMAIL "<support@taosdata.com>"
#endif
#endif
// clang-format off
#define DM_APOLLO_URL "The apollo string to use when configuring the server, such as: -a 'jsonFile:./tests/cfg.json', cfg.json text can be '{\"fqdn\":\"td1\"}'."
#define DM_CFG_DIR "Configuration directory."

View File

@ -16,14 +16,7 @@
#define _DEFAULT_SOURCE
#include "os.h"
#include "taoserror.h"
#if defined(CUS_NAME) || defined(CUS_PROMPT) || defined(CUS_EMAIL)
#include "cus_name.h"
#else
#ifndef CUS_PROMPT
#define CUS_PROMPT "taos"
#endif
#endif
#define PROCESS_ITEM 12
#define UUIDLEN37 37

View File

@ -31,18 +31,16 @@ class TDTestCase(TBase):
# run
outputs = etool.runBinFile("taosBenchmark", "-V")
print(outputs)
if len(outputs) != 3:
tdLog.exit(f"checkVersion return lines count {len(outputs) != 3}")
if len(outputs) != 4:
tdLog.exit(f"checkVersion return lines count {len(outputs)} != 4")
# version string len
assert len(outputs[0]) > 27
assert outputs[0][:22] == "taosBenchmark version:"
assert len(outputs[1]) > 24
# commit id
assert len(outputs[1]) > 43
assert outputs[1][:4] == "git:"
assert len(outputs[2]) > 43
assert outputs[2][:4] == "git:"
# build info
assert len(outputs[2]) > 36
assert outputs[2][:6] == "build:"
assert len(outputs[3]) > 36
assert outputs[3][:6] == "build:"
tdLog.info("check taosBenchmark version successfully.")

View File

@ -34,7 +34,7 @@ class TDTestCase(TBase):
sql = "select count(*) from meters"
tdSql.query(sql)
allCnt = tdSql.getData(0, 0)
if allCnt < 2000000:
if allCnt < 200000:
tdLog.exit(f"taosbenchmark insert row small. row count={allCnt} sql={sql}")
return

View File

@ -30,7 +30,7 @@
"name": "meters",
"child_table_exists": "no",
"childtable_count": 10,
"insert_rows": 300000,
"insert_rows": 30000,
"childtable_prefix": "d",
"insert_mode": "taosc",
"insert_interval": 0,

View File

@ -31,17 +31,16 @@ class TDTestCase(TBase):
# run
outputs = etool.runBinFile("taosdump", "-V")
print(outputs)
if len(outputs) != 3:
tdLog.exit(f"checkVersion return lines count {len(outputs) != 3}")
if len(outputs) != 4:
tdLog.exit(f"checkVersion return lines count {len(outputs)} != 4")
# version string len
assert len(outputs[0]) > 22
assert outputs[0][:17] == "taosdump version:"
assert len(outputs[1]) > 19
# commit id
assert len(outputs[1]) > 43
assert outputs[1][:4] == "git:"
assert len(outputs[2]) > 43
assert outputs[2][:4] == "git:"
# build info
assert len(outputs[2]) > 36
assert outputs[2][:6] == "build:"
assert len(outputs[3]) > 36
assert outputs[3][:6] == "build:"
tdLog.info("check taosdump version successfully.")

View File

@ -17,16 +17,10 @@
#include <pwd.h>
#endif
#include "cus_name.h"
#include "shellInt.h"
#include "version.h"
#if defined(CUS_NAME) || defined(CUS_PROMPT) || defined(CUS_EMAIL)
#include "cus_name.h"
#else
#ifndef CUS_PROMPT
#define CUS_PROMPT "taos"
#endif
#endif
#define TAOS_CONSOLE_PROMPT_CONTINUE " -> "

View File

@ -49,22 +49,6 @@
// ---------------- define ----------------
//
#if defined(CUS_NAME) || defined(CUS_PROMPT) || defined(CUS_EMAIL)
#include <cus_name.h>
#else
#ifndef CUS_NAME
#define CUS_NAME "TDengine"
#endif
#ifndef CUS_PROMPT
#define CUS_PROMPT "taos"
#endif
#ifndef CUS_EMAIL
#define CUS_EMAIL "<support@taosdata.com>"
#endif
#endif
// use 256 as normal buffer length
#define BUFFER_LEN 256

View File

@ -10,7 +10,8 @@
* FITNESS FOR A PARTICULAR PURPOSE.
*/
#include <bench.h>
#include "cus_name.h" // include/util/
#include <bench.h>
#include "benchLog.h"
#include <toolsdef.h>
@ -29,7 +30,7 @@ char *g_aggreFunc[] = {"*", "count(*)", "avg(C0)", "sum(C0)",
void printVersion() {
// version, macro define in src/CMakeLists.txt
printf("taosBenchmark version: %s\n", TD_VER_NUMBER);
printf("%s\n%sBenchmark version: %s\n", TD_PRODUCT_NAME, CUS_PROMPT, TD_VER_NUMBER);
printf("git: %s\n", TAOSBENCHMARK_COMMIT_ID);
printf("build: %s\n", BUILD_INFO);
}

View File

@ -12,6 +12,7 @@
#include <stdlib.h>
#include <bench.h>
#include "benchLog.h"
#include "cus_name.h"
#ifdef LINUX
#include <argp.h>
@ -21,23 +22,7 @@
#endif
#endif
extern char version[];
#if defined(CUS_NAME) || defined(CUS_PROMPT) || defined(CUS_EMAIL)
#include "cus_name.h"
#else
#ifndef CUS_NAME
#define CUS_NAME "TDengine"
#endif
#ifndef CUS_PROMPT
#define CUS_PROMPT "taos"
#endif
#ifndef CUS_EMAIL
#define CUS_EMAIL "<support@taosdata.com>"
#endif
#endif
#ifdef WINDOWS
char g_configDir[MAX_PATH_LEN] = {0}; // "C:\\TDengine\\cfg"};

View File

@ -11,6 +11,7 @@
#define _GNU_SOURCE
#include "cus_name.h" // include/util/
#include "dump.h"
#include "dumpUtil.h"
#ifdef WEBSOCKET
@ -262,7 +263,7 @@ static void printVersion(FILE *file) {
}
// version, macro define in src/CMakeLists.txt
fprintf(file, "taosdump version: %s\n", TD_VER_NUMBER);
fprintf(file, "%s\n%sdump version: %s\n", TD_PRODUCT_NAME, CUS_PROMPT, TD_VER_NUMBER);
fprintf(file, "git: %s\n", TAOSDUMP_COMMIT_ID);
fprintf(file, "build: %s\n", BUILD_INFO);
}