xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

stmt

This commit is contained in:
dapan1121 2022-04-20 08:58:29 +08:00
parent 0f13cee876
commit 743ff696b8
1 changed files with 43 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
tests/script/api/batchprepare.c
View File

@ -34,8 +34,8 @@ typedef struct {
int32_t gVarCharSize = 10; int32_t gVarCharSize = 10;
int32_t gVarCharLen = 5; int32_t gVarCharLen = 5;
int insertAllCols(TAOS_STMT *stmt); int insertMBSETest(TAOS_STMT *stmt);
int insertSpecifyCols(TAOS_STMT *stmt); int insertMBMETest(TAOS_STMT *stmt);
int32_t shortColList[] = {TSDB_DATA_TYPE_TIMESTAMP, TSDB_DATA_TYPE_INT}; int32_t shortColList[] = {TSDB_DATA_TYPE_TIMESTAMP, TSDB_DATA_TYPE_INT};
int32_t longColList[] = {TSDB_DATA_TYPE_TIMESTAMP, TSDB_DATA_TYPE_BOOL, TSDB_DATA_TYPE_TINYINT, TSDB_DATA_TYPE_UTINYINT, TSDB_DATA_TYPE_SMALLINT, TSDB_DATA_TYPE_USMALLINT, TSDB_DATA_TYPE_INT, TSDB_DATA_TYPE_UINT, TSDB_DATA_TYPE_BIGINT, TSDB_DATA_TYPE_UBIGINT, TSDB_DATA_TYPE_FLOAT, TSDB_DATA_TYPE_DOUBLE, TSDB_DATA_TYPE_BINARY, TSDB_DATA_TYPE_NCHAR}; int32_t longColList[] = {TSDB_DATA_TYPE_TIMESTAMP, TSDB_DATA_TYPE_BOOL, TSDB_DATA_TYPE_TINYINT, TSDB_DATA_TYPE_UTINYINT, TSDB_DATA_TYPE_SMALLINT, TSDB_DATA_TYPE_USMALLINT, TSDB_DATA_TYPE_INT, TSDB_DATA_TYPE_UINT, TSDB_DATA_TYPE_BIGINT, TSDB_DATA_TYPE_UBIGINT, TSDB_DATA_TYPE_FLOAT, TSDB_DATA_TYPE_DOUBLE, TSDB_DATA_TYPE_BINARY, TSDB_DATA_TYPE_NCHAR};
@ -53,16 +53,17 @@ typedef struct {
int32_t tblNum; int32_t tblNum;
int32_t rowNum; int32_t rowNum;
int32_t bindRowNum; int32_t bindRowNum;
int32_t bindColNum; int32_t bindColNum; // equal colNum in full column case
int32_t bindNullNum; int32_t bindNullNum;
int32_t runTimes; int32_t runTimes;
} CaseCfg; } CaseCfg;
CaseCfg gCase[] = { CaseCfg gCase[] = {
// {"insert:all columns", tListLen(shortColList), shortColList, false, true, insertAllCols, 1, 10, 10, 2, 0, 10}, // {"insert:MBSE", tListLen(shortColList), shortColList, false, true, insertMBSETest, 1, 10, 10, 0, 0, 10},
// {"insert:all columns", tListLen(shortColList), shortColList, false, true, insertAllCols, 10, 100, 10, 2, 0, 10}, // {"insert:MBSE", tListLen(shortColList), shortColList, false, true, insertMBSETest, 10, 100, 10, 0, 0, 10},
{"insert:all columns", tListLen(longColList), longColList, false, true, insertAllCols, 10, 10, 2, 14, 0, 1}, // {"insert:MBSE", tListLen(longColList), longColList, false, true, insertMBSETest, 10, 10, 2, 0, 0, 1},
// {"insert:all columns", tListLen(longColList), longColList, false, false, insertSpecifyCols, 10, 10, 2, 6, 0, 1}, // {"insert:MBSE", tListLen(longColList), longColList, false, false, insertMBSETest, 10, 10, 2, 6, 0, 1},
{"insert:MBSE", tListLen(longColList), longColList, false, false, insertMBMETest, 10, 10, 2, 6, 0, 1},
}; };
CaseCfg *gCurCase = NULL; CaseCfg *gCurCase = NULL;
@ -111,7 +112,7 @@ bool colExists(TAOS_BIND_v2* pBind, int32_t dataType) {
} }
} }
void generateSql(BindData *data) { void generateInsertSQL(BindData *data) {
int32_t len = sprintf(data->sql, "insert into %s ", (gCurCase->tblNum > 1 ? "? " : "m0 ")); int32_t len = sprintf(data->sql, "insert into %s ", (gCurCase->tblNum > 1 ? "? " : "m0 "));
if (!gCurCase->fullCol) { if (!gCurCase->fullCol) {
len += sprintf(data->sql + len, "("); len += sprintf(data->sql + len, "(");
@ -184,12 +185,13 @@ void generateSql(BindData *data) {
void generateDataType(BindData *data, int32_t bindIdx, int32_t colIdx, int32_t *dataType) { void generateDataType(BindData *data, int32_t bindIdx, int32_t colIdx, int32_t *dataType) {
if (bindIdx < gCurCase->bindColNum) { if (bindIdx < gCurCase->bindColNum) {
if (colIdx) { if (gCurCase->fullCol) {
if (!gCurCase->multiCol) { *dataType = gCurCase->colList[bindIdx];
*dataType = TSDB_DATA_TYPE_INT; return;
return; } else if (0 == colIdx) {
} *dataType = TSDB_DATA_TYPE_TIMESTAMP;
return;
} else {
while (true) { while (true) {
*dataType = rand() % (TSDB_DATA_TYPE_MAX - 1) + 1; *dataType = rand() % (TSDB_DATA_TYPE_MAX - 1) + 1;
if (*dataType == TSDB_DATA_TYPE_JSON || *dataType == TSDB_DATA_TYPE_DECIMAL if (*dataType == TSDB_DATA_TYPE_JSON || *dataType == TSDB_DATA_TYPE_DECIMAL
@ -260,7 +262,7 @@ int32_t prepareColData(BindData *data, int32_t bindIdx, int32_t rowIdx, int32_t
break; break;
case TSDB_DATA_TYPE_VARCHAR: case TSDB_DATA_TYPE_VARCHAR:
data->pBind[bindIdx].buffer_length = gVarCharSize; data->pBind[bindIdx].buffer_length = gVarCharSize;
data->pBind[bindIdx].buffer = data->binaryData + rowIdx; data->pBind[bindIdx].buffer = data->binaryData + rowIdx * gVarCharSize;
data->pBind[bindIdx].length = data->binaryLen; data->pBind[bindIdx].length = data->binaryLen;
data->pBind[bindIdx].is_null = data->isNull; data->pBind[bindIdx].is_null = data->isNull;
break; break;
@ -272,7 +274,7 @@ int32_t prepareColData(BindData *data, int32_t bindIdx, int32_t rowIdx, int32_t
break; break;
case TSDB_DATA_TYPE_NCHAR: case TSDB_DATA_TYPE_NCHAR:
data->pBind[bindIdx].buffer_length = gVarCharSize; data->pBind[bindIdx].buffer_length = gVarCharSize;
data->pBind[bindIdx].buffer = data->binaryData + rowIdx; data->pBind[bindIdx].buffer = data->binaryData + rowIdx * gVarCharSize;
data->pBind[bindIdx].length = data->binaryLen; data->pBind[bindIdx].length = data->binaryLen;
data->pBind[bindIdx].is_null = data->isNull; data->pBind[bindIdx].is_null = data->isNull;
break; break;
@ -364,7 +366,7 @@ int32_t prepareData(BindData *data) {
} }
} }
generateSql(data); generateInsertSQL(data);
return 0; return 0;
} }
@ -389,7 +391,7 @@ void destroyData(BindData *data) {
} }
int insertAllCols(TAOS_STMT *stmt) { int insertMBSETest(TAOS_STMT *stmt) {
BindData data = {0}; BindData data = {0};
prepareData(&data); prepareData(&data);
@ -435,7 +437,7 @@ int insertAllCols(TAOS_STMT *stmt) {
} }
int insertSpecifyCols(TAOS_STMT *stmt) { int insertMBMETest(TAOS_STMT *stmt) {
BindData data = {0}; BindData data = {0};
prepareData(&data); prepareData(&data);
@ -470,12 +472,13 @@ int insertSpecifyCols(TAOS_STMT *stmt) {
exit(1); exit(1);
} }
} }
if (taos_stmt_execute(stmt) != 0) {
printf("taos_stmt_execute error:%s\n", taos_stmt_errstr(stmt));
exit(1);
}
} }
if (taos_stmt_execute(stmt) != 0) {
printf("taos_stmt_execute error:%s\n", taos_stmt_errstr(stmt));
exit(1);
}
return 0; return 0;
} }
@ -4763,9 +4766,9 @@ void prepareCheckResult(TAOS *taos) {
char buf[32]; char buf[32];
for (int32_t t = 0; t< gCurCase->tblNum; ++t) { for (int32_t t = 0; t< gCurCase->tblNum; ++t) {
if (gCurCase->tblNum > 1) { if (gCurCase->tblNum > 1) {
sprintf(buf, "m%d", t); sprintf(buf, "t%d", t);
} else { } else {
sprintf(buf, "m%d", 0); sprintf(buf, "t%d", 0);
} }
prepareCheckResultImpl(taos, buf, 1, gCurCase->rowNum); prepareCheckResultImpl(taos, buf, 1, gCurCase->rowNum);
@ -4924,6 +4927,9 @@ void generateCreateTableSQL(char *buf, int32_t tblIdx, int32_t colNum, int32_t *
if (stable) { if (stable) {
blen += sprintf(buf + blen, "tags ("); blen += sprintf(buf + blen, "tags (");
for (int c = 0; c < colNum; ++c) { for (int c = 0; c < colNum; ++c) {
if (c > 0) {
blen += sprintf(buf + blen, ",");
}
switch (colList[c]) { switch (colList[c]) {
case TSDB_DATA_TYPE_BOOL: case TSDB_DATA_TYPE_BOOL:
blen += sprintf(buf + blen, "tbooldata bool"); blen += sprintf(buf + blen, "tbooldata bool");
@ -4979,6 +4985,10 @@ void generateCreateTableSQL(char *buf, int32_t tblIdx, int32_t colNum, int32_t *
blen += sprintf(buf + blen, " ("); blen += sprintf(buf + blen, " (");
for (int c = 0; c < colNum; ++c) { for (int c = 0; c < colNum; ++c) {
if (c > 0) {
blen += sprintf(buf + blen, ",");
}
switch (colList[c]) { switch (colList[c]) {
case TSDB_DATA_TYPE_BOOL: case TSDB_DATA_TYPE_BOOL:
blen += sprintf(buf + blen, "booldata bool"); blen += sprintf(buf + blen, "booldata bool");
@ -5005,7 +5015,7 @@ void generateCreateTableSQL(char *buf, int32_t tblIdx, int32_t colNum, int32_t *
blen += sprintf(buf + blen, "binarydata binary(%d)", gVarCharSize); blen += sprintf(buf + blen, "binarydata binary(%d)", gVarCharSize);
break; break;
case TSDB_DATA_TYPE_TIMESTAMP: case TSDB_DATA_TYPE_TIMESTAMP:
blen += sprintf(buf + blen, "ts ts"); blen += sprintf(buf + blen, "ts timestamp");
break; break;
case TSDB_DATA_TYPE_NCHAR: case TSDB_DATA_TYPE_NCHAR:
blen += sprintf(buf + blen, "nchardata nchar(%d)", gVarCharSize); blen += sprintf(buf + blen, "nchardata nchar(%d)", gVarCharSize);
@ -5029,6 +5039,8 @@ void generateCreateTableSQL(char *buf, int32_t tblIdx, int32_t colNum, int32_t *
} }
blen += sprintf(buf + blen, ")"); blen += sprintf(buf + blen, ")");
printf("Create SQL:%s\n", buf);
} }
void prepare(TAOS *taos, int32_t colNum, int32_t *colList, int autoCreate) { void prepare(TAOS *taos, int32_t colNum, int32_t *colList, int autoCreate) {
@ -5087,8 +5099,12 @@ void* runcase(TAOS *taos) {
gCurCase = &gCase[i]; gCurCase = &gCase[i];
printf("Case %d Begin\n", i); printf("Case %d Begin\n", i);
if (gCurCase->fullCol) {
gCurCase->bindColNum = gCurCase->colNum;
}
for (int32_t n = 0; n < gCurCase->runTimes; ++n) { for (int32_t n = 0; n < gCurCase->runTimes; ++n) {
prepare(taos, tListLen(gCurCase->colList), gCurCase->colList, gCurCase->autoCreate); prepare(taos, gCurCase->colNum, gCurCase->colList, gCurCase->autoCreate);
stmt = taos_stmt_init(taos); stmt = taos_stmt_init(taos);
if (NULL == stmt) { if (NULL == stmt) {