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

shell query

This commit is contained in:
facetosea 2023-10-24 15:05:24 +08:00
parent d324a34d01
commit b9fa620b26
1 changed files with 143 additions and 84 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

227
tools/shell/src/shellEngine.c
View File

@ -22,6 +22,23 @@
#include "shellAuto.h" #include "shellAuto.h"
#include "shellInt.h" #include "shellInt.h"
typedef struct {
const char *sql;
bool vertical;
tsem_t sem;
int64_t numOfRows; // the num of this batch
int64_t numOfAllRows;
int32_t numFields;
TAOS_FIELD *fields;
int32_t precision;
int32_t maxColNameLen; // for vertical print
int32_t width[TSDB_MAX_COLUMNS]; // for horizontal print
uint64_t resShowMaxNum;
} tsDumpInfo;
static bool shellIsEmptyCommand(const char *cmd); static bool shellIsEmptyCommand(const char *cmd);
static int32_t shellRunSingleCommand(char *command); static int32_t shellRunSingleCommand(char *command);
static void shellRecordCommandToHistory(char *command); static void shellRecordCommandToHistory(char *command);
@ -31,8 +48,8 @@ static char *shellFormatTimestamp(char *buf, int64_t val, int32_t precision);
static int64_t shellDumpResultToFile(const char *fname, TAOS_RES *tres); static int64_t shellDumpResultToFile(const char *fname, TAOS_RES *tres);
static void shellPrintNChar(const char *str, int32_t length, int32_t width); static void shellPrintNChar(const char *str, int32_t length, int32_t width);
static void shellPrintGeometry(const unsigned char *str, int32_t length, int32_t width); static void shellPrintGeometry(const unsigned char *str, int32_t length, int32_t width);
static int64_t shellVerticalPrintResult(TAOS_RES *tres, const char *sql); static void shellVerticalPrintResult(TAOS_RES *tres, tsDumpInfo* dump_info);
static int64_t shellHorizontalPrintResult(TAOS_RES *tres, const char *sql); static void shellHorizontalPrintResult(TAOS_RES *tres, tsDumpInfo* dump_info);
static int64_t shellDumpResult(TAOS_RES *tres, char *fname, int32_t *error_no, bool vertical, const char *sql); static int64_t shellDumpResult(TAOS_RES *tres, char *fname, int32_t *error_no, bool vertical, const char *sql);
static void shellReadHistory(); static void shellReadHistory();
static void shellWriteHistory(); static void shellWriteHistory();
@ -702,64 +719,88 @@ bool shellIsShowQuery(const char *sql) {
return false; return false;
} }
int64_t shellVerticalPrintResult(TAOS_RES *tres, const char *sql) { void init_dump_info(tsDumpInfo *dump_info, TAOS_RES *tres, const char *sql, bool vertical) {
TAOS_ROW row = taos_fetch_row(tres); dump_info->sql = sql;
if (row == NULL) { dump_info->vertical = vertical;
return 0; tsem_init(&dump_info->sem, 0, 0);
dump_info->numOfAllRows = 0;
dump_info->numFields = taos_num_fields(tres);
dump_info->fields = taos_fetch_fields(tres);
dump_info->precision = taos_result_precision(tres);
dump_info->resShowMaxNum = UINT64_MAX;
if (shell.args.commands == NULL && shell.args.file[0] == 0 && !shellIsShowWhole(dump_info->sql)) {
dump_info->resShowMaxNum = SHELL_DEFAULT_RES_SHOW_NUM;
} }
int32_t num_fields = taos_num_fields(tres); if (vertical) {
TAOS_FIELD *fields = taos_fetch_fields(tres); dump_info->maxColNameLen = 0;
int32_t precision = taos_result_precision(tres); for (int32_t col = 0; col < dump_info->numFields; col++) {
int32_t len = (int32_t)strlen(dump_info->fields[col].name);
int32_t maxColNameLen = 0; if (len > dump_info->maxColNameLen) {
for (int32_t col = 0; col < num_fields; col++) { dump_info->maxColNameLen = len;
int32_t len = (int32_t)strlen(fields[col].name); }
if (len > maxColNameLen) { }
maxColNameLen = len; } else {
for (int32_t col = 0; col < dump_info->numFields; col++) {
dump_info->width[col] = shellCalcColWidth(dump_info->fields + col, dump_info->precision);
} }
} }
}
uint64_t resShowMaxNum = UINT64_MAX; void shellVerticalPrintResult(TAOS_RES *tres, tsDumpInfo *dump_info) {
if (dump_info->numOfRows <= 0) return;
if (shell.args.commands == NULL && shell.args.file[0] == 0 && !shellIsShowWhole(sql)) { if (dump_info->numOfAllRows >= dump_info->resShowMaxNum) {
resShowMaxNum = SHELL_DEFAULT_RES_SHOW_NUM; return;
} }
int64_t numOfRows = 0; TAOS_ROW row = taos_fetch_row(tres);
int32_t showMore = 1; if (row == NULL) {
do { return;
if (numOfRows < resShowMaxNum) { }
printf("*************************** %"PRId64".row ***************************\r\n", numOfRows + 1);
int32_t *length = taos_fetch_lengths(tres); int64_t numOfPintRows = dump_info->numOfAllRows;
int numOfPrintRowsThisOne = 0;
for (int32_t i = 0; i < num_fields; i++) { while (row != NULL) {
TAOS_FIELD *field = fields + i; printf("*************************** %" PRId64 ".row ***************************\r\n", numOfPintRows + 1);
int32_t padding = (int32_t)(maxColNameLen - strlen(field->name)); int32_t *length = taos_fetch_lengths(tres);
printf("%*.s%s: ", padding, " ", field->name);
shellPrintField((const char *)row[i], field, 0, length[i], precision); for (int32_t i = 0; i < dump_info->numFields; i++) {
putchar('\r'); TAOS_FIELD *field = dump_info->fields + i;
putchar('\n');
} int32_t padding = (int32_t)(dump_info->maxColNameLen - strlen(field->name));
} else if (showMore) { printf("%*.s%s: ", padding, " ", field->name);
shellPrintField((const char *)row[i], field, 0, length[i], dump_info->precision);
putchar('\r');
putchar('\n');
}
numOfPintRows++;
numOfPrintRowsThisOne++;
if (numOfPintRows == dump_info->resShowMaxNum) {
printf("\r\n"); printf("\r\n");
printf(" Notice: The result shows only the first %d rows.\r\n", SHELL_DEFAULT_RES_SHOW_NUM); printf(" Notice: The result shows only the first %d rows.\r\n", dump_info->resShowMaxNum);
printf(" You can use the `LIMIT` clause to get fewer result to show.\r\n"); printf(" You can use the `LIMIT` clause to get fewer result to show.\r\n");
printf(" Or use '>>' to redirect the whole set of the result to a specified file.\r\n"); printf(" Or use '>>' to redirect the whole set of the result to a specified file.\r\n");
printf("\r\n"); printf("\r\n");
printf(" You can use Ctrl+C to stop the underway fetching.\r\n"); printf(" You can use Ctrl+C to stop the underway fetching.\r\n");
printf("\r\n"); printf("\r\n");
showMore = 0; return;
} }
numOfRows++; if (numOfPrintRowsThisOne == dump_info->numOfRows) {
row = taos_fetch_row(tres); return;
} while (row != NULL); }
return numOfRows; row = taos_fetch_row(tres);
}
return;
} }
int32_t shellCalcColWidth(TAOS_FIELD *field, int32_t precision) { int32_t shellCalcColWidth(TAOS_FIELD *field, int32_t precision) {
@ -856,47 +897,42 @@ void shellPrintHeader(TAOS_FIELD *fields, int32_t *width, int32_t num_fields) {
putchar('\n'); putchar('\n');
} }
int64_t shellHorizontalPrintResult(TAOS_RES *tres, const char *sql) { void shellHorizontalPrintResult(TAOS_RES *tres, tsDumpInfo *dump_info) {
if (dump_info->numOfRows <= 0) return;
if (dump_info->numOfAllRows >= dump_info->resShowMaxNum) {
return;
}
TAOS_ROW row = taos_fetch_row(tres); TAOS_ROW row = taos_fetch_row(tres);
if (row == NULL) { if (row == NULL) {
return 0; return;
} }
int32_t num_fields = taos_num_fields(tres); int64_t numOfPintRows = dump_info->numOfAllRows;
TAOS_FIELD *fields = taos_fetch_fields(tres); int numOfPrintRowsThisOne = 0;
int32_t precision = taos_result_precision(tres); if (numOfPintRows == 0) {
shellPrintHeader(dump_info->fields, dump_info->width, dump_info->numFields);
int32_t width[TSDB_MAX_COLUMNS];
for (int32_t col = 0; col < num_fields; col++) {
width[col] = shellCalcColWidth(fields + col, precision);
} }
shellPrintHeader(fields, width, num_fields); while (row != NULL) {
uint64_t resShowMaxNum = UINT64_MAX;
if (shell.args.commands == NULL && shell.args.file[0] == 0 && !shellIsShowWhole(sql)) {
resShowMaxNum = SHELL_DEFAULT_RES_SHOW_NUM;
}
int64_t numOfRows = 0;
int32_t showMore = 1;
do {
int32_t *length = taos_fetch_lengths(tres); int32_t *length = taos_fetch_lengths(tres);
if (numOfRows < resShowMaxNum) { for (int32_t i = 0; i < dump_info->numFields; i++) {
for (int32_t i = 0; i < num_fields; i++) { putchar(' ');
putchar(' '); shellPrintField((const char *)row[i], dump_info->fields + i, dump_info->width[i], length[i],
shellPrintField((const char *)row[i], fields + i, width[i], length[i], precision); dump_info->precision);
putchar(' '); putchar(' ');
putchar('|'); putchar('|');
} }
putchar('\r'); putchar('\r');
putchar('\n'); putchar('\n');
} else if (showMore) {
numOfPintRows++;
numOfPrintRowsThisOne++;
if (numOfPintRows == dump_info->resShowMaxNum) {
printf("\r\n"); printf("\r\n");
printf(" Notice: The result shows only the first %d rows.\r\n", SHELL_DEFAULT_RES_SHOW_NUM); printf(" Notice: The result shows only the first %d rows.\r\n", dump_info->resShowMaxNum);
if (shellIsShowQuery(sql)) { if (shellIsShowQuery(dump_info->sql)) {
printf(" You can use '>>' to redirect the whole set of the result to a specified file.\r\n"); printf(" You can use '>>' to redirect the whole set of the result to a specified file.\r\n");
} else { } else {
printf(" You can use the `LIMIT` clause to get fewer result to show.\r\n"); printf(" You can use the `LIMIT` clause to get fewer result to show.\r\n");
@ -905,28 +941,51 @@ int64_t shellHorizontalPrintResult(TAOS_RES *tres, const char *sql) {
printf("\r\n"); printf("\r\n");
printf(" You can use Ctrl+C to stop the underway fetching.\r\n"); printf(" You can use Ctrl+C to stop the underway fetching.\r\n");
printf("\r\n"); printf("\r\n");
showMore = 0; return;
} }
numOfRows++; if (numOfPrintRowsThisOne == dump_info->numOfRows) {
row = taos_fetch_row(tres); return;
} while (row != NULL); }
return numOfRows; row = taos_fetch_row(tres);
}
return;
}
void shellDumpResultCallback(void *param, TAOS_RES *tres, int num_of_rows) {
tsDumpInfo *dump_info = (tsDumpInfo *)param;
if (num_of_rows > 0) {
dump_info->numOfRows = num_of_rows;
if (dump_info->vertical) {
shellVerticalPrintResult(tres, dump_info);
} else {
shellHorizontalPrintResult(tres, dump_info);
}
dump_info->numOfAllRows += num_of_rows;
taos_fetch_rows_a(tres, shellDumpResultCallback, param);
} else {
if (num_of_rows < 0) {
printf("\033[31masync retrieve failed, code: %d\033[0m\n", num_of_rows);
}
tsem_post(&dump_info->sem);
}
} }
int64_t shellDumpResult(TAOS_RES *tres, char *fname, int32_t *error_no, bool vertical, const char *sql) { int64_t shellDumpResult(TAOS_RES *tres, char *fname, int32_t *error_no, bool vertical, const char *sql) {
int64_t numOfRows = 0; int64_t num_of_rows = 0;
if (fname != NULL) { if (fname != NULL) {
numOfRows = shellDumpResultToFile(fname, tres); num_of_rows = shellDumpResultToFile(fname, tres);
} else if (vertical) {
numOfRows = shellVerticalPrintResult(tres, sql);
} else { } else {
numOfRows = shellHorizontalPrintResult(tres, sql); tsDumpInfo dump_info;
init_dump_info(&dump_info, tres, sql, vertical);
taos_fetch_rows_a(tres, shellDumpResultCallback, &dump_info);
tsem_wait(&dump_info.sem);
num_of_rows = dump_info.numOfAllRows;
} }
*error_no = taos_errno(tres); *error_no = taos_errno(tres);
return numOfRows; return num_of_rows;
} }
void shellReadHistory() { void shellReadHistory() {