209 lines
7.2 KiB
C++
209 lines
7.2 KiB
C++
/*
|
|
* 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 <function.h>
|
|
#include <gtest/gtest.h>
|
|
#include <tglobal.h>
|
|
#include <iostream>
|
|
|
|
#pragma GCC diagnostic push
|
|
#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 "os.h"
|
|
|
|
#include "astGenerator.h"
|
|
#include "parserInt.h"
|
|
#include "taos.h"
|
|
#include "tdef.h"
|
|
#include "tvariant.h"
|
|
#include "planner.h"
|
|
#include "../../planner/inc/plannerInt.h"
|
|
|
|
namespace {
|
|
void setSchema(SSchema* p, int32_t type, int32_t bytes, const char* name, int32_t colId) {
|
|
p->colId = colId;
|
|
p->bytes = bytes;
|
|
p->type = type;
|
|
strcpy(p->name, name);
|
|
}
|
|
|
|
void setTableMetaInfo(SQueryStmtInfo* pQueryInfo, SCatalogReq *req) {
|
|
pQueryInfo->numOfTables = 1;
|
|
|
|
pQueryInfo->pTableMetaInfo = (STableMetaInfo**)calloc(1, POINTER_BYTES);
|
|
STableMetaInfo* pTableMetaInfo = (STableMetaInfo*)calloc(1, sizeof(STableMetaInfo));
|
|
pQueryInfo->pTableMetaInfo[0] = pTableMetaInfo;
|
|
|
|
SName* name = (SName*)taosArrayGet(req->pTableName, 0);
|
|
|
|
memcpy(&pTableMetaInfo->name, taosArrayGet(req->pTableName, 0), sizeof(SName));
|
|
pTableMetaInfo->pTableMeta = (STableMeta*)calloc(1, sizeof(STableMeta) + 4 * sizeof(SSchema));
|
|
strcpy(pTableMetaInfo->aliasName, name->tname);
|
|
STableMeta* pTableMeta = pTableMetaInfo->pTableMeta;
|
|
pTableMeta->tableType = TSDB_NORMAL_TABLE;
|
|
pTableMeta->tableInfo.numOfColumns = 4;
|
|
pTableMeta->tableInfo.rowSize = 28;
|
|
pTableMeta->uid = 110;
|
|
|
|
pTableMetaInfo->tagColList = (SArray*) taosArrayInit(4, POINTER_BYTES);
|
|
|
|
SSchema* pSchema = pTableMetaInfo->pTableMeta->schema;
|
|
setSchema(&pSchema[0], TSDB_DATA_TYPE_TIMESTAMP, 8, "ts", 0);
|
|
setSchema(&pSchema[1], TSDB_DATA_TYPE_INT, 4, "a", 1);
|
|
setSchema(&pSchema[2], TSDB_DATA_TYPE_DOUBLE, 8, "b", 2);
|
|
setSchema(&pSchema[3], TSDB_DATA_TYPE_DOUBLE, 8, "col", 3);
|
|
}
|
|
|
|
void generateLogicplan(const char* sql) {
|
|
SSqlInfo info1 = doGenerateAST(sql);
|
|
ASSERT_EQ(info1.valid, true);
|
|
|
|
char msg[128] = {0};
|
|
SMsgBuf buf;
|
|
buf.len = 128;
|
|
buf.buf = msg;
|
|
|
|
SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.sub.node), 0);
|
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
|
ASSERT_EQ(code, 0);
|
|
|
|
SCatalogReq req = {0};
|
|
SParseContext ctx = {0};
|
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, &ctx, msg, 128);
|
|
ASSERT_EQ(ret, 0);
|
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
|
|
|
SQueryStmtInfo* pQueryInfo = createQueryInfo();
|
|
setTableMetaInfo(pQueryInfo, &req);
|
|
|
|
SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.sub.node, 0);
|
|
ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
|
|
ASSERT_EQ(ret, 0);
|
|
|
|
struct SQueryPlanNode* n = nullptr;
|
|
code = createQueryPlan((const SQueryNode*)pQueryInfo, &n);
|
|
|
|
char* str = NULL;
|
|
queryPlanToString(n, &str);
|
|
|
|
printf("--------SQL:%s\n", sql);
|
|
printf("%s\n", str);
|
|
|
|
destroyQueryInfo(pQueryInfo);
|
|
qParserCleanupMetaRequestInfo(&req);
|
|
destroySqlInfo(&info1);
|
|
}
|
|
}
|
|
|
|
TEST(testCase, planner_test) {
|
|
SSqlInfo info1 = doGenerateAST("select top(a*b / 99, 20) from `t.1abc` interval(10s, 1s)");
|
|
ASSERT_EQ(info1.valid, true);
|
|
|
|
char msg[128] = {0};
|
|
SMsgBuf buf;
|
|
buf.len = 128;
|
|
buf.buf = msg;
|
|
|
|
SSqlNode* pNode = (SSqlNode*) taosArrayGetP(((SArray*)info1.sub.node), 0);
|
|
int32_t code = evaluateSqlNode(pNode, TSDB_TIME_PRECISION_NANO, &buf);
|
|
ASSERT_EQ(code, 0);
|
|
|
|
SCatalogReq req = {0};
|
|
SParseContext ctx = {0};
|
|
|
|
int32_t ret = qParserExtractRequestedMetaInfo(&info1, &req, &ctx, msg, 128);
|
|
ASSERT_EQ(ret, 0);
|
|
ASSERT_EQ(taosArrayGetSize(req.pTableName), 1);
|
|
|
|
SQueryStmtInfo* pQueryInfo = createQueryInfo();
|
|
setTableMetaInfo(pQueryInfo, &req);
|
|
|
|
SSqlNode* pSqlNode = (SSqlNode*)taosArrayGetP(info1.sub.node, 0);
|
|
ret = validateSqlNode(pSqlNode, pQueryInfo, &buf);
|
|
ASSERT_EQ(ret, 0);
|
|
|
|
SArray* pExprList = pQueryInfo->exprList[0];
|
|
|
|
int32_t num = tsCompatibleModel? 2:1;
|
|
ASSERT_EQ(taosArrayGetSize(pExprList), num);
|
|
|
|
SExprInfo* p1 = (SExprInfo*) taosArrayGetP(pExprList, 1);
|
|
ASSERT_EQ(p1->base.pColumns->uid, 110);
|
|
ASSERT_EQ(p1->base.numOfParams, 1);
|
|
ASSERT_EQ(p1->base.resSchema.type, TSDB_DATA_TYPE_DOUBLE);
|
|
ASSERT_STRCASEEQ(p1->base.resSchema.name, "top(a*b / 99, 20)");
|
|
ASSERT_EQ(p1->base.pColumns->flag, TSDB_COL_TMP);
|
|
ASSERT_STRCASEEQ(p1->base.token, "top(a*b / 99, 20)");
|
|
ASSERT_EQ(p1->base.interBytes, 16);
|
|
|
|
ASSERT_EQ(p1->pExpr->nodeType, TEXPR_FUNCTION_NODE);
|
|
ASSERT_STREQ(p1->pExpr->_function.functionName, "top");
|
|
|
|
tExprNode* pParam = p1->pExpr->_function.pChild[0];
|
|
|
|
ASSERT_EQ(pParam->nodeType, TEXPR_COL_NODE);
|
|
ASSERT_EQ(taosArrayGetSize(pQueryInfo->colList), 3);
|
|
ASSERT_EQ(pQueryInfo->fieldsInfo.numOfOutput, 2);
|
|
|
|
struct SQueryPlanNode* n = nullptr;
|
|
code = createQueryPlan((const SQueryNode*)pQueryInfo, &n);
|
|
|
|
char* str = NULL;
|
|
queryPlanToString(n, &str);
|
|
printf("%s\n", str);
|
|
|
|
destroyQueryInfo(pQueryInfo);
|
|
qParserCleanupMetaRequestInfo(&req);
|
|
destroySqlInfo(&info1);
|
|
}
|
|
|
|
TEST(testCase, displayPlan) {
|
|
generateLogicplan("select count(*) from `t.1abc`");
|
|
generateLogicplan("select count(*)+ 22 from `t.1abc`");
|
|
generateLogicplan("select count(*)+ 22 from `t.1abc` interval(1h, 20s) sliding(10m) limit 20,30");
|
|
generateLogicplan("select count(*) from `t.1abc` group by a");
|
|
generateLogicplan("select count(A+B) from `t.1abc` group by a");
|
|
generateLogicplan("select count(length(a)+b) from `t.1abc` group by a");
|
|
generateLogicplan("select count(*) from `t.1abc` interval(10s, 5s) sliding(7s)");
|
|
generateLogicplan("select count(*) from `t.1abc` interval(10s, 5s) sliding(7s) order by 1 desc ");
|
|
generateLogicplan("select count(*),sum(a),avg(b),min(a+b)+99 from `t.1abc`");
|
|
generateLogicplan("select count(*), min(a) + 99 from `t.1abc`");
|
|
generateLogicplan("select count(length(count(*) + 22)) from `t.1abc`");
|
|
generateLogicplan("select concat(concat(a,b), concat(a,b)) from `t.1abc` limit 20");
|
|
generateLogicplan("select count(*), first(a), last(b) from `t.1abc` state_window(a)");
|
|
generateLogicplan("select count(*), first(a), last(b) from `t.1abc` session(ts, 20s)");
|
|
|
|
// order by + group by column + limit offset
|
|
generateLogicplan("select top(a, 20) k from `t.1abc` order by k asc limit 3 offset 1");
|
|
|
|
// fill
|
|
generateLogicplan("select min(a) from `t.1abc` where ts>now and ts<now+2h interval(1s) fill(linear)");
|
|
|
|
// union + union all
|
|
|
|
|
|
|
|
// join
|
|
|
|
// Aggregate(count(*) [count(*) #5056], sum(a) [sum(a) #5057], avg(b) [avg(b) #5058], min(a+b) [min(a+b) #5060])
|
|
// Projection(cols: [a+b #5059]) filters:(nil)
|
|
// Projection(cols: [ts #0], [a #1], [b #2]) filters:(nil)
|
|
// TableScan(t.1abc #110) time_range: -9223372036854775808 - 9223372036854775807
|
|
|
|
}
|
|
|
|
#pragma GCC diagnostic pop |