From ca21807d14b46264152bb11fda7dcda29f57c2a8 Mon Sep 17 00:00:00 2001 From: jiacy-jcy <714897623@qq.com> Date: Thu, 28 Apr 2022 10:52:26 +0800 Subject: [PATCH 01/27] update --- tests/system-test/2-query/Now.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/tests/system-test/2-query/Now.py b/tests/system-test/2-query/Now.py index 99fca2e88d..9d073eb4f7 100644 --- a/tests/system-test/2-query/Now.py +++ b/tests/system-test/2-query/Now.py @@ -144,6 +144,12 @@ class TDTestCase: tdSql.error("select now()+1.5 from ntb") tdSql.error("select now()+1.5 from db.ntb") + tdSql.error("select now()-1.5 from ntb") + tdSql.error("select now()-1.5 from db.ntb") + tdSql.error("select now()*1.5 from ntb") + tdSql.error("select now()*1.5 from db.ntb") + tdSql.error("select now()/1.5 from ntb") + tdSql.error("select now()/1.5 from db.ntb") tdSql.error("select now()+'abc' from ntb") tdSql.error("select now()+'abc' from db.ntb") tdSql.error("select now()+abc from ntb") @@ -262,7 +268,10 @@ class TDTestCase: tdSql.checkRows(3) tdSql.query("select now() +1 from db.stb") tdSql.checkRows(3) - + tdSql.error("select now() +1.5 from stb") + tdSql.error("select now() -1.5 from stb") + tdSql.error("select now() *1.5 from stb") + tdSql.error("select now() /1.5 from stb") tdSql.error("select now() +'abc' from stb") tdSql.error("select now() +'abc' from db.stb") tdSql.error("select now() + ! from stb") @@ -272,6 +281,7 @@ class TDTestCase: tdSql.error("select now() -today() from stb") tdSql.error("select now() - today() from db.stb") + tdSql.query("select now()+null from stb") tdSql.checkData(0,0,None) tdSql.query("select now()+null from db.stb") From e1ac1226a24a5f5db5641f3bc3a5f948b16eb9dd Mon Sep 17 00:00:00 2001 From: jiacy-jcy <714897623@qq.com> Date: Thu, 28 Apr 2022 18:05:14 +0800 Subject: [PATCH 02/27] add case for to_iso8601 --- .../2-query/{To_iso8061.py => To_iso8601.py} | 17 ++++ tests/system-test/2-query/Today.py | 80 ++++++++++++++++++- 2 files changed, 96 insertions(+), 1 deletion(-) rename tests/system-test/2-query/{To_iso8061.py => To_iso8601.py} (88%) diff --git a/tests/system-test/2-query/To_iso8061.py b/tests/system-test/2-query/To_iso8601.py similarity index 88% rename from tests/system-test/2-query/To_iso8061.py rename to tests/system-test/2-query/To_iso8601.py index 86322ed076..cd22ffb90c 100644 --- a/tests/system-test/2-query/To_iso8061.py +++ b/tests/system-test/2-query/To_iso8601.py @@ -92,6 +92,21 @@ class TDTestCase: tdSql.query("select to_iso8601(today()) /null from db.ntb") tdSql.checkRows(3) tdSql.checkData(0,0,None) + # tdSql.query("select to_iso8601(-1) from ntb") + tdSql.query("select to_iso8601(9223372036854775807) from ntb") + tdSql.checkRows(3) + + # tdSql.query("select to_iso8601(10000000000) from ntb") + # tdSql.checkData(0,0,None) + # tdSql.query("select to_iso8601(-1) from ntb") + # tdSql.checkRows(3) + # tdSql.query("select to_iso8601(-10000000000) from ntb") + # tdSql.checkData(0,0,None) + tdSql.error("select to_iso8601(1.5) from ntb") + tdSql.error("select to_iso8601(1.5) from db.ntb") + tdSql.error("select to_iso8601('a') from ntb") + tdSql.error("select to_iso8601(c2) from ntb") + @@ -137,6 +152,8 @@ class TDTestCase: tdSql.checkRows(3) tdSql.checkData(0,0,None) + # tdSql.query("select to_iso8601(-1) from ntb") + def stop(self): diff --git a/tests/system-test/2-query/Today.py b/tests/system-test/2-query/Today.py index 09f018dc11..9eb06de9fb 100644 --- a/tests/system-test/2-query/Today.py +++ b/tests/system-test/2-query/Today.py @@ -134,6 +134,35 @@ class TDTestCase: tdSql.checkRows(3) tdSql.checkData(0, 0, str(today_date)) + tdSql.error("select today()+1.5 from ntb") + tdSql.error("select today()-1.5 from ntb") + tdSql.error("select today()*1.5 from ntb") + tdSql.error("select today()/1.5 from ntb") + tdSql.error("select today()+1.5 from db.ntb") + tdSql.error("select today()-1.5 from db.ntb") + tdSql.error("select today()*1.5 from db.ntb") + tdSql.error("select today()/1.5 from db.ntb") + tdSql.query("select today()+null from ntb") + tdSql.checkData(0,0,None) + tdSql.query("select today()+null from db.ntb") + tdSql.checkData(0,0,None) + tdSql.query("select today()-null from ntb") + tdSql.checkData(0,0,None) + tdSql.query("select today()-null from db.ntb") + tdSql.checkData(0,0,None) + tdSql.query("select today()*null from ntb") + tdSql.checkData(0,0,None) + tdSql.query("select today()*null from db.ntb") + tdSql.checkData(0,0,None) + tdSql.query("select today()/null from ntb") + tdSql.checkData(0,0,None) + tdSql.query("select today()/null from db.ntb") + tdSql.checkData(0,0,None) + tdSql.query("select today()/0 from db.ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,None) + tdSql.checkData(1,0,None) + tdSql.checkData(2,0,None) # stable tdSql.query("select today() from stb") tdSql.checkRows(3) @@ -208,6 +237,26 @@ class TDTestCase: tdSql.query("select ts from stb where ts<=today()") tdSql.checkRows(2) + tdSql.error("select today()+1.5 from stb") + tdSql.error("select today()-1.5 from stb") + tdSql.error("select today()*1.5 from stb") + tdSql.error("select today()/1.5 from stb") + tdSql.query("select today()+null from stb") + tdSql.checkData(0,0,None) + tdSql.query("select today()+null from db.stb") + tdSql.checkData(0,0,None) + tdSql.query("select today()-null from stb") + tdSql.checkData(0,0,None) + tdSql.query("select today()-null from db.stb") + tdSql.checkData(0,0,None) + tdSql.query("select today()*null from stb") + tdSql.checkData(0,0,None) + tdSql.query("select today()*null from db.stb") + tdSql.checkData(0,0,None) + tdSql.query("select today()/null from stb") + tdSql.checkData(0,0,None) + tdSql.query("select today()/null from db.stb") + tdSql.checkData(0,0,None) # # tdSql.query("select * from ntb where ts Date: Fri, 29 Apr 2022 09:32:57 +0800 Subject: [PATCH 03/27] add to_unixtimestamp case --- tests/system-test/2-query/To_unixtimestamp.py | 52 +++++++++++++++++++ 1 file changed, 52 insertions(+) create mode 100644 tests/system-test/2-query/To_unixtimestamp.py diff --git a/tests/system-test/2-query/To_unixtimestamp.py b/tests/system-test/2-query/To_unixtimestamp.py new file mode 100644 index 0000000000..7088142f9a --- /dev/null +++ b/tests/system-test/2-query/To_unixtimestamp.py @@ -0,0 +1,52 @@ +from time import sleep + +from util.log import * +from util.sql import * +from util.cases import * + + + + +class TDTestCase: + + def init(self, conn, logSql): + tdLog.debug(f"start to excute {__file__}") + tdSql.init(conn.cursor()) + + def run(self): # sourcery skip: extract-duplicate-method + tdSql.prepare() + tdLog.printNoPrefix("==========step1:create tables==========") + tdSql.execute( + '''create table if not exists ntb + (ts timestamp, c1 int, c2 float,c3 double,c4 timestamp) + ''' + ) + tdSql.execute( + '''create table if not exists stb + (ts timestamp, c1 int, c2 float,c3 double,c4 timestamp) tags(t0 int) + ''' + ) + tdSql.execute( + '''create table if not exists stb_1 using stb tags(100) + ''' + ) + tdLog.printNoPrefix("==========step2:insert data into ntb==========") + + + + + + + + + + + + + + def stop(self): + tdSql.close() + tdLog.success(f"{__file__} successfully executed") + +tdCases.addLinux(__file__, TDTestCase()) +tdCases.addWindows(__file__, TDTestCase()) \ No newline at end of file From c5b62ad05368ec232778295022d0f06af9ffa0ad Mon Sep 17 00:00:00 2001 From: jiacy-jcy <714897623@qq.com> Date: Thu, 5 May 2022 11:12:21 +0800 Subject: [PATCH 04/27] update test case --- tests/system-test/2-query/Timediff.py | 196 ++++++++++++++++++ tests/system-test/2-query/To_unixtimestamp.py | 43 +++- tests/system-test/2-query/count.py | 182 ++++++++++++++++ 3 files changed, 410 insertions(+), 11 deletions(-) create mode 100644 tests/system-test/2-query/Timediff.py create mode 100644 tests/system-test/2-query/count.py diff --git a/tests/system-test/2-query/Timediff.py b/tests/system-test/2-query/Timediff.py new file mode 100644 index 0000000000..2824fea5a2 --- /dev/null +++ b/tests/system-test/2-query/Timediff.py @@ -0,0 +1,196 @@ +from util.log import * +from util.sql import * +from util.cases import * + +class TDTestCase: + + def init(self, conn, logSql): + tdLog.debug(f"start to excute {__file__}") + tdSql.init(conn.cursor()) + + def run(self): # sourcery skip: extract-duplicate-method + tdSql.prepare() + tdLog.printNoPrefix("==========step1:create tables==========") + tdSql.execute( + '''create table if not exists ntb + (ts timestamp, c1 int, c2 float,c3 double,c4 timestamp) + ''' + ) + tdSql.execute( + '''create table if not exists stb + (ts timestamp, c1 int, c2 float,c3 double,c4 timestamp) tags(t0 int) + ''' + ) + tdSql.execute( + '''create table if not exists stb_1 using stb tags(100) + ''' + ) + tdLog.printNoPrefix("==========step2:insert data into ntb==========") + + # RFC3339:2020-01-01T00:00:00+8:00 + # ISO8601:2020-01-01T00:00:00.000+0800 + tdSql.execute( + 'insert into ntb values(now,1,1.55,100.555555,today())("2020-1-1 00:00:00",10,11.11,99.999999,now())(today(),3,3.333,333.333333,now())') + tdSql.execute( + 'insert into stb_1 values(now,1,1.55,100.555555,today())("2020-1-1 00:00:00",10,11.11,99.999999,now())(today(),3,3.333,333.333333,now())') + + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00') from ntb") + tdSql.checkRows(3) + tdSql.query("select timediff(1,0,1d) from ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff(1,0,1d) from db.ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff(1,0,1s) from ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,1) + tdSql.query("select timediff(1,0,1s) from db.ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,1) + tdSql.query("select timediff(1,0,1w) from ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff(1,0,1w) from db.ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff(1,0,1h) from ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff(1,0,1h) from db.ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff(1,0,1m) from ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff(1,0,1m) from db.ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff(1,0,1a) from ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,1000) + tdSql.query("select timediff(1,0,1a) from db.ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,1000) + tdSql.query("select timediff(1,0,1u) from ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,1000000) + tdSql.query("select timediff(1,0,1u) from db.ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,1000000) + + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00') from stb") + tdSql.checkRows(3) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00') from db.stb") + tdSql.checkRows(3) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1d) from stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,1) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1d) from db.stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,1) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1h) from stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,24) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1h) from db.stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,24) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1w) from stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1m) from stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,1440) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1m) from db.stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,1440) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1s) from stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,86400) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1s) from db.stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,86400) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1a) from stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,86400000) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1a) from db.stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,86400000) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1u) from stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,86400000000) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-2 00:00:00',1u) from db.stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,86400000000) + + + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00') from stb_1") + tdSql.checkRows(3) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00') from db.stb_1") + tdSql.checkRows(3) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1w) from stb_1 ") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1w) from db.stb_1 ") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1d) from stb_1 ") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1d) from db.stb_1 ") + tdSql.checkRows(3) + tdSql.checkData(0,0,0) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1h) from stb_1 ") + tdSql.checkRows(3) + tdSql.checkData(0,0,12) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1h) from db.stb_1 ") + tdSql.checkRows(3) + tdSql.checkData(0,0,12) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1m) from stb_1" ) + tdSql.checkRows(3) + tdSql.checkData(0,0,720) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1m) from db.stb_1" ) + tdSql.checkRows(3) + tdSql.checkData(0,0,720) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1s) from stb_1") + tdSql.checkRows(3) + tdSql.checkData(0,0,43200) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1s) from db.stb_1") + tdSql.checkRows(3) + tdSql.checkData(0,0,43200) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1a) from stb_1") + tdSql.checkRows(3) + tdSql.checkData(0,0,43200000) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1a) from db.stb_1") + tdSql.checkRows(3) + tdSql.checkData(0,0,43200000) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1u) from stb_1") + tdSql.checkRows(3) + tdSql.checkData(0,0,43200000000) + tdSql.query("select timediff('2020-1-1 00:00:00','2020-1-1 12:00:00',1u) from db.stb_1") + tdSql.checkRows(3) + tdSql.checkData(0,0,43200000000) + + tdSql.query("select timediff('a','b') from stb") + tdSql.checkRows(3) + tdSql.checkData(0,0,None) + tdSql.checkData(1,0,None) + tdSql.checkData(2,0,None) + tdSql.error("select timediff(1.5,1.5) from stb") + tdSql.error("select timediff(1) from stb") + tdSql.error("select timediff(10,1,1.5) from stb") + # tdSql.error("select timediff(10,1,2s) from stb") + # tdSql.error("select timedifff(10,1,c1) from stb") + + + + + + + + def stop(self): + tdSql.close() + tdLog.success(f"{__file__} successfully executed") + +tdCases.addLinux(__file__, TDTestCase()) +tdCases.addWindows(__file__, TDTestCase()) \ No newline at end of file diff --git a/tests/system-test/2-query/To_unixtimestamp.py b/tests/system-test/2-query/To_unixtimestamp.py index 7088142f9a..43315ff0d8 100644 --- a/tests/system-test/2-query/To_unixtimestamp.py +++ b/tests/system-test/2-query/To_unixtimestamp.py @@ -31,19 +31,40 @@ class TDTestCase: ''' ) tdLog.printNoPrefix("==========step2:insert data into ntb==========") + + # RFC3339:2020-01-01T00:00:00+8:00 + # ISO8601:2020-01-01T00:00:00.000+0800 + tdSql.execute( + 'insert into ntb values(now,1,1.55,100.555555,today())("2020-1-1 00:00:00",10,11.11,99.999999,now())(today(),3,3.333,333.333333,now())') + tdSql.execute( + 'insert into stb_1 values(now,1,1.55,100.555555,today())("2020-1-1 00:00:00",10,11.11,99.999999,now())(today(),3,3.333,333.333333,now())') + tdSql.query("select to_unixtimestamp('1970-01-01T08:00:00+0800') from ntb") + tdSql.checkData(0,0,0) + tdSql.checkData(1,0,0) + tdSql.checkData(2,0,0) + tdSql.checkRows(3) + tdSql.query("select to_unixtimestamp('1970-01-01T08:00:00+08:00') from ntb") + tdSql.checkData(0,0,0) + tdSql.checkRows(3) + tdSql.query("select to_unixtimestamp('1900-01-01T08:00:00+08:00') from ntb") + tdSql.checkRows(3) + tdSql.query("select to_unixtimestamp('2020-01-32T08:00:00') from ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,None) + tdSql.query("select to_unixtimestamp('2020-13-32T08:00:00') from ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,None) + tdSql.query("select to_unixtimestamp('acd') from ntb") + tdSql.checkRows(3) + tdSql.checkData(0,0,None) + tdSql.error("select to_unixtimestamp(1) from ntb") + tdSql.error("select to_unixtimestamp(1.5) from ntb") + tdSql.error("select to_unixtimestamp(ts) from ntb") + + tdSql.query("select ts from ntb where to_unixtimestamp('1970-01-01T08:00:00+08:00')=0") + tdSql.checkRows(3) - - - - - - - - - - - def stop(self): tdSql.close() tdLog.success(f"{__file__} successfully executed") diff --git a/tests/system-test/2-query/count.py b/tests/system-test/2-query/count.py new file mode 100644 index 0000000000..8888a182b8 --- /dev/null +++ b/tests/system-test/2-query/count.py @@ -0,0 +1,182 @@ +from util.log import * +from util.sql import * +from util.cases import * + +class TDTestCase: + def init(self, conn, logSql): + tdLog.debug("start to execute %s" % __file__) + tdSql.init(conn.cursor()) + + self.rowNum = 10 + self.ts = 1537146000000 + + def run(self): + tdSql.prepare() + + tdSql.execute('''create table stb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, + col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned) tags(loc nchar(20))''') + tdSql.execute("create table stb_1 using stb tags('beijing')") + + tdSql.execute('''create table ntb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, + col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned)''') + + for i in range(self.rowNum): + tdSql.execute("insert into stb_1 values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" + % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) + + for i in range(self.rowNum): + tdSql.execute("insert into ntb values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" + % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) + + tdSql.query("select count(*) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(*) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(ts) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(ts) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col1) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col1) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col2) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col2) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col3) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col3) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col4) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col4) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col5) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col5) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col6) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col6) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col7) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col7) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col8) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col8) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col9) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col9) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col11) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col11) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col12) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col12) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col13) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col13) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col14) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col14) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(ts) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(ts) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col1) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col1) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col2) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col2) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col3) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col3) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col4) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col4) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col5) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col5) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col6) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col6) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col7) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col7) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col8) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col8) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col9) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col9) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col11) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col11) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col12) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col12) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col13) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col13) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col14) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col14) from db.stb_1") + tdSql.checkData(0,0,10) + + tdSql.execute("insert into stb_1 values(now,null,null,null,null,null,null,null,null,null,null,null,null,null)") + tdSql.query("select count(col1) from stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col1),count(ts) from stb_1") + tdSql.checkData(0,0,10) + tdSql.checkData(0,1,11) + + tdSql.query("select count(col1) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.query("select count(col1),count(ts) from db.stb_1") + tdSql.checkData(0,0,10) + tdSql.checkData(0,1,11) + + tdSql.query("select count(col1) from stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col1),count(ts) from stb") + tdSql.checkData(0,0,10) + tdSql.checkData(0,1,11) + + tdSql.query("select count(col1) from db.stb") + tdSql.checkData(0,0,10) + tdSql.query("select count(col1),count(ts) from db.stb") + tdSql.checkData(0,0,10) + tdSql.checkData(0,1,11) + + + + + def stop(self): + tdSql.close() + tdLog.success("%s successfully executed" % __file__) + + +tdCases.addWindows(__file__, TDTestCase()) +tdCases.addLinux(__file__, TDTestCase()) \ No newline at end of file From 1854f065c9dbaa3a9b1c7060756a43f154f9e782 Mon Sep 17 00:00:00 2001 From: jiacy-jcy <714897623@qq.com> Date: Thu, 5 May 2022 20:22:53 +0800 Subject: [PATCH 05/27] add case for max and min --- tests/system-test/2-query/max.py | 205 +++++++++++++++++++++ tests/system-test/2-query/min.py | 215 ++++++++++++++++++++++ tests/system-test/2-query/timetruncate.py | 104 +++++++++++ 3 files changed, 524 insertions(+) create mode 100644 tests/system-test/2-query/max.py create mode 100644 tests/system-test/2-query/min.py create mode 100644 tests/system-test/2-query/timetruncate.py diff --git a/tests/system-test/2-query/max.py b/tests/system-test/2-query/max.py new file mode 100644 index 0000000000..5342c7d449 --- /dev/null +++ b/tests/system-test/2-query/max.py @@ -0,0 +1,205 @@ +from util.log import * +from util.cases import * +from util.sql import * +import numpy as np + + +class TDTestCase: + def init(self, conn, logSql): + tdLog.debug("start to execute %s" % __file__) + tdSql.init(conn.cursor()) + + self.rowNum = 10 + self.ts = 1537146000000 + + def prepare_data(self): + + pass + def run(self): + tdSql.prepare() + + intData = [] + floatData = [] + + tdSql.execute('''create table stb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, + col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned) tags(loc nchar(20))''') + tdSql.execute("create table stb_1 using stb tags('beijing')") + tdSql.execute('''create table ntb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, + col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned)''') + for i in range(self.rowNum): + tdSql.execute("insert into ntb values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" + % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) + intData.append(i + 1) + floatData.append(i + 0.1) + for i in range(self.rowNum): + tdSql.execute("insert into stb_1 values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" + % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) + intData.append(i + 1) + floatData.append(i + 0.1) + + # max verifacation + tdSql.error("select max(ts) from stb_1") + tdSql.error("select max(ts) from db.stb_1") + tdSql.error("select max(col7) from stb_1") + tdSql.error("select max(col7) from db.stb_1") + tdSql.error("select max(col8) from stb_1") + tdSql.error("select max(col8) from db.stb_1") + tdSql.error("select max(col9) from stb_1") + tdSql.error("select max(col9) from db.stb_1") + + tdSql.query("select max(col1) from stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col1) from db.stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col2) from stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col2) from db.stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col3) from stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col3) from db.stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col4) from stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col4) from db.stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col11) from stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col11) from db.stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col12) from stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col12) from db.stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col13) from stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col13) from db.stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col14) from stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col14) from db.stb_1") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col5) from stb_1") + tdSql.checkData(0, 0, np.max(floatData)) + tdSql.query("select max(col5) from db.stb_1") + tdSql.checkData(0, 0, np.max(floatData)) + tdSql.query("select max(col6) from stb_1") + tdSql.checkData(0, 0, np.max(floatData)) + tdSql.query("select max(col6) from db.stb_1") + tdSql.checkData(0, 0, np.max(floatData)) + tdSql.query("select max(col1) from stb_1 where col2<=5") + tdSql.checkData(0,0,5) + + + + tdSql.error("select max(ts) from stb") + tdSql.error("select max(ts) from db.stb") + tdSql.error("select max(col7) from stb") + tdSql.error("select max(col7) from db.stb") + tdSql.error("select max(col8) from stb") + tdSql.error("select max(col8) from db.stb") + tdSql.error("select max(col9) from stb") + tdSql.error("select max(col9) from db.stb") + + tdSql.query("select max(col1) from stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col1) from db.stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col2) from stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col2) from db.stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col3) from stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col3) from db.stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col4) from stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col4) from db.stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col11) from stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col11) from db.stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col12) from stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col12) from db.stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col13) from stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col13) from db.stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col14) from stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col14) from db.stb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col5) from stb") + tdSql.checkData(0, 0, np.max(floatData)) + tdSql.query("select max(col5) from db.stb") + tdSql.checkData(0, 0, np.max(floatData)) + tdSql.query("select max(col6) from stb") + tdSql.checkData(0, 0, np.max(floatData)) + tdSql.query("select max(col6) from db.stb") + tdSql.checkData(0, 0, np.max(floatData)) + tdSql.query("select max(col1) from stb where col2<=5") + tdSql.checkData(0,0,5) + + + + tdSql.error("select max(ts) from ntb") + tdSql.error("select max(ts) from db.ntb") + tdSql.error("select max(col7) from ntb") + tdSql.error("select max(col7) from db.ntb") + tdSql.error("select max(col8) from ntb") + tdSql.error("select max(col8) from db.ntb") + tdSql.error("select max(col9) from ntb") + tdSql.error("select max(col9) from db.ntb") + + tdSql.query("select max(col1) from ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col1) from db.ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col2) from ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col2) from db.ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col3) from ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col3) from db.ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col4) from ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col4) from db.ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col11) from ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col11) from db.ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col12) from ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col12) from db.ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col13) from ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col13) from db.ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col14) from ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col14) from db.ntb") + tdSql.checkData(0, 0, np.max(intData)) + tdSql.query("select max(col5) from ntb") + tdSql.checkData(0, 0, np.max(floatData)) + tdSql.query("select max(col5) from db.ntb") + tdSql.checkData(0, 0, np.max(floatData)) + tdSql.query("select max(col6) from ntb") + tdSql.checkData(0, 0, np.max(floatData)) + tdSql.query("select max(col6) from db.ntb") + tdSql.checkData(0, 0, np.max(floatData)) + tdSql.query("select max(col1) from stb_1 where col2<=5") + tdSql.checkData(0,0,5) + def stop(self): + tdSql.close() + tdLog.success("%s successfully executed" % __file__) + +tdCases.addWindows(__file__, TDTestCase()) +tdCases.addLinux(__file__, TDTestCase()) diff --git a/tests/system-test/2-query/min.py b/tests/system-test/2-query/min.py new file mode 100644 index 0000000000..60a8c105f1 --- /dev/null +++ b/tests/system-test/2-query/min.py @@ -0,0 +1,215 @@ +from wsgiref.headers import tspecials +from util.log import * +from util.cases import * +from util.sql import * +import numpy as np + + +class TDTestCase: + def init(self, conn, logSql): + tdLog.debug("start to execute %s" % __file__) + tdSql.init(conn.cursor()) + + self.rowNum = 10 + self.ts = 1537146000000 + + def run(self): + tdSql.prepare() + + intData = [] + floatData = [] + + tdSql.execute('''create table stb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, + col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned) tags(loc nchar(20))''') + tdSql.execute("create table stb_1 using stb tags('beijing')") + tdSql.execute('''create table ntb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, + col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned)''') + for i in range(self.rowNum): + tdSql.execute("insert into ntb values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" + % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) + intData.append(i + 1) + floatData.append(i + 0.1) + for i in range(self.rowNum): + tdSql.execute("insert into stb_1 values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" + % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) + intData.append(i + 1) + floatData.append(i + 0.1) + + # max verifacation + tdSql.error("select min(ts) from stb_1") + tdSql.error("select min(ts) from db.stb_1") + tdSql.error("select min(col7) from stb_1") + tdSql.error("select min(col7) from db.stb_1") + tdSql.error("select min(col8) from stb_1") + tdSql.error("select min(col8) from db.stb_1") + tdSql.error("select min(col9) from stb_1") + tdSql.error("select min(col9) from db.stb_1") + # tdSql.error("select min(a) from stb_1") + # tdSql.error("select min(1) from stb_1") + tdSql.error("select min(now()) from stb_1") + tdSql.error("select min(count(c1),count(c2)) from stb_1") + + tdSql.query("select min(col1) from stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col1) from db.stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col2) from stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col2) from db.stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col3) from stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col3) from db.stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col4) from stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col4) from db.stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col11) from stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col11) from db.stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col12) from stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col12) from db.stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col13) from stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col13) from db.stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col14) from stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col14) from db.stb_1") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col5) from stb_1") + tdSql.checkData(0, 0, np.min(floatData)) + tdSql.query("select min(col5) from db.stb_1") + tdSql.checkData(0, 0, np.min(floatData)) + tdSql.query("select min(col6) from stb_1") + tdSql.checkData(0, 0, np.min(floatData)) + tdSql.query("select min(col6) from db.stb_1") + tdSql.checkData(0, 0, np.min(floatData)) + tdSql.query("select min(col1) from stb_1 where col2>=5") + tdSql.checkData(0,0,5) + + + tdSql.error("select min(ts) from stb_1") + tdSql.error("select min(ts) from db.stb_1") + tdSql.error("select min(col7) from stb_1") + tdSql.error("select min(col7) from db.stb_1") + tdSql.error("select min(col8) from stb_1") + tdSql.error("select min(col8) from db.stb_1") + tdSql.error("select min(col9) from stb_1") + tdSql.error("select min(col9) from db.stb_1") + # tdSql.error("select min(a) from stb_1") + # tdSql.error("select min(1) from stb_1") + tdSql.error("select min(now()) from stb_1") + tdSql.error("select min(count(c1),count(c2)) from stb_1") + + tdSql.query("select min(col1) from stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col1) from db.stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col2) from stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col2) from db.stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col3) from stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col3) from db.stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col4) from stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col4) from db.stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col11) from stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col11) from db.stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col12) from stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col12) from db.stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col13) from stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col13) from db.stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col14) from stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col14) from db.stb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col5) from stb") + tdSql.checkData(0, 0, np.min(floatData)) + tdSql.query("select min(col5) from db.stb") + tdSql.checkData(0, 0, np.min(floatData)) + tdSql.query("select min(col6) from stb") + tdSql.checkData(0, 0, np.min(floatData)) + tdSql.query("select min(col6) from db.stb") + tdSql.checkData(0, 0, np.min(floatData)) + tdSql.query("select min(col1) from stb where col2>=5") + tdSql.checkData(0,0,5) + + + tdSql.error("select min(ts) from ntb") + tdSql.error("select min(ts) from db.ntb") + tdSql.error("select min(col7) from ntb") + tdSql.error("select min(col7) from db.ntb") + tdSql.error("select min(col8) from ntb") + tdSql.error("select min(col8) from db.ntb") + tdSql.error("select min(col9) from ntb") + tdSql.error("select min(col9) from db.ntb") + # tdSql.error("select min(a) from stb_1") + # tdSql.error("select min(1) from stb_1") + tdSql.error("select min(now()) from ntb") + tdSql.error("select min(count(c1),count(c2)) from ntb") + + tdSql.query("select min(col1) from ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col1) from db.ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col2) from ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col2) from db.ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col3) from ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col3) from db.ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col4) from ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col4) from db.ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col11) from ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col11) from db.ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col12) from ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col12) from db.ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col13) from ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col13) from db.ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col14) from ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col14) from db.ntb") + tdSql.checkData(0, 0, np.min(intData)) + tdSql.query("select min(col5) from ntb") + tdSql.checkData(0, 0, np.min(floatData)) + tdSql.query("select min(col5) from db.ntb") + tdSql.checkData(0, 0, np.min(floatData)) + tdSql.query("select min(col6) from ntb") + tdSql.checkData(0, 0, np.min(floatData)) + tdSql.query("select min(col6) from db.ntb") + tdSql.checkData(0, 0, np.min(floatData)) + tdSql.query("select min(col1) from ntb where col2>=5") + tdSql.checkData(0,0,5) + + + def stop(self): + tdSql.close() + tdLog.success("%s successfully executed" % __file__) + +tdCases.addWindows(__file__, TDTestCase()) +tdCases.addLinux(__file__, TDTestCase()) diff --git a/tests/system-test/2-query/timetruncate.py b/tests/system-test/2-query/timetruncate.py new file mode 100644 index 0000000000..4c4ebb09df --- /dev/null +++ b/tests/system-test/2-query/timetruncate.py @@ -0,0 +1,104 @@ +import taos +from util.log import * +from util.cases import * +from util.sql import * +import numpy as np + + +class TDTestCase: + def init(self, conn, logSql): + tdLog.debug("start to execute %s" % __file__) + tdSql.init(conn.cursor()) + + self.rowNum = 10 + self.ts = 1537146000000 # 2018-9-17 09:00:00.000 + + def run(self): + tdSql.prepare() + + intData = [] + floatData = [] + + tdSql.execute('''create table stb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, + col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned) tags(loc nchar(20))''') + tdSql.execute("create table stb_1 using stb tags('beijing')") + tdSql.execute('''create table ntb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, + col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned)''') + for i in range(self.rowNum): + tdSql.execute("insert into ntb values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" + % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) + intData.append(i + 1) + floatData.append(i + 0.1) + for i in range(self.rowNum): + tdSql.execute("insert into stb_1 values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" + % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) + intData.append(i + 1) + floatData.append(i + 0.1) + + tdSql.query("select timetruncate(1,1d) from ntb") + tdSql.checkRows(10) + tdSql.query("select timetruncate(1,1u) from ntb") + tdSql.checkRows(10) + tdSql.query("select timetruncate(1,1a) from ntb") + tdSql.checkRows(10) + tdSql.query("select timetruncate(1,1m) from ntb") + tdSql.checkRows(10) + tdSql.query("select timetruncate(1,1h) from ntb") + tdSql.checkRows(10) + tdSql.query("select timetruncate(ts,1d) from ntb") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 08:00:00.000") + tdSql.query("select timetruncate(ts,1h) from ntb") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 09:00:00.000") + tdSql.query("select timetruncate(ts,1m) from ntb") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 09:00:00.000") + tdSql.query("select timetruncate(ts,1s) from ntb") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 09:00:00.000") + tdSql.query("select timetruncate(ts,1a) from ntb") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 09:00:00.000") + tdSql.checkData(1,0,"2018-09-17 09:00:00.001") + tdSql.checkData(2,0,"2018-09-17 09:00:00.002") + tdSql.checkData(3,0,"2018-09-17 09:00:00.003") + tdSql.checkData(4,0,"2018-09-17 09:00:00.004") + tdSql.checkData(5,0,"2018-09-17 09:00:00.005") + tdSql.checkData(6,0,"2018-09-17 09:00:00.006") + tdSql.checkData(7,0,"2018-09-17 09:00:00.007") + tdSql.checkData(8,0,"2018-09-17 09:00:00.008") + tdSql.checkData(9,0,"2018-09-17 09:00:00.009") + # tdSql.query("select timetruncate(ts,1u) from ntb") + # tdSql.checkRows(10) + # tdSql.checkData(0,0,"2018-09-17 09:00:00.000000") + # tdSql.checkData(1,0,"2018-09-17 09:00:00.001000") + # tdSql.checkData(2,0,"2018-09-17 09:00:00.002000") + # tdSql.checkData(3,0,"2018-09-17 09:00:00.003000") + # tdSql.checkData(4,0,"2018-09-17 09:00:00.004000") + # tdSql.checkData(5,0,"2018-09-17 09:00:00.005000") + # tdSql.checkData(6,0,"2018-09-17 09:00:00.006000") + # tdSql.checkData(7,0,"2018-09-17 09:00:00.007000") + # tdSql.checkData(8,0,"2018-09-17 09:00:00.008000") + # tdSql.checkData(9,0,"2018-09-17 09:00:00.009000") + # tdSql.query("select timetruncate(ts,1b) from ntb") + # tdSql.checkRows(10) + # tdSql.checkData(0,0,"2018-09-17 09:00:00.000000000") + # tdSql.checkData(1,0,"2018-09-17 09:00:00.001000000") + # tdSql.checkData(2,0,"2018-09-17 09:00:00.002000000") + # tdSql.checkData(3,0,"2018-09-17 09:00:00.003000000") + # tdSql.checkData(4,0,"2018-09-17 09:00:00.004000000") + # tdSql.checkData(5,0,"2018-09-17 09:00:00.005000000") + # tdSql.checkData(6,0,"2018-09-17 09:00:00.006000000") + # tdSql.checkData(7,0,"2018-09-17 09:00:00.007000000") + # tdSql.checkData(8,0,"2018-09-17 09:00:00.008000000") + # tdSql.checkData(9,0,"2018-09-17 09:00:00.009000000") + + + + def stop(self): + tdSql.close() + tdLog.success("%s successfully executed" % __file__) + +tdCases.addWindows(__file__, TDTestCase()) +tdCases.addLinux(__file__, TDTestCase()) \ No newline at end of file From 3bcea212e4d8a38d3f0e1ad0ac32e3695caaf15f Mon Sep 17 00:00:00 2001 From: jiacy-jcy <714897623@qq.com> Date: Fri, 6 May 2022 10:33:37 +0800 Subject: [PATCH 06/27] add case for last --- tests/system-test/2-query/last.py | 132 ++++++++++++++++++++++++++++++ 1 file changed, 132 insertions(+) create mode 100644 tests/system-test/2-query/last.py diff --git a/tests/system-test/2-query/last.py b/tests/system-test/2-query/last.py new file mode 100644 index 0000000000..82e648f60a --- /dev/null +++ b/tests/system-test/2-query/last.py @@ -0,0 +1,132 @@ +from util.log import * +from util.cases import * +from util.sql import * +import numpy as np + + +class TDTestCase: + def init(self, conn, logSql): + tdLog.debug("start to execute %s" % __file__) + tdSql.init(conn.cursor()) + + self.rowNum = 10 + self.ts = 1537146000000 + + def run(self): + tdSql.prepare() + + tdSql.execute('''create table test(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, + col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned) tags(loc nchar(20))''') + tdSql.execute("create table test1 using test tags('beijing')") + tdSql.execute("insert into test1(ts) values(%d)" % (self.ts - 1)) + + # last verifacation + tdSql.query("select last(*) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 1, None) + + tdSql.query("select last(col1) from test1") + tdSql.checkRows(0) + + tdSql.query("select last(col2) from test1") + tdSql.checkRows(0) + + tdSql.query("select last(col3) from test1") + tdSql.checkRows(0) + + tdSql.query("select last(col4) from test1") + tdSql.checkRows(0) + + tdSql.query("select last(col11) from test1") + tdSql.checkRows(0) + + tdSql.query("select last(col12) from test1") + tdSql.checkRows(0) + + tdSql.query("select last(col13) from test1") + tdSql.checkRows(0) + + tdSql.query("select last(col14) from test1") + tdSql.checkRows(0) + + tdSql.query("select last(col5) from test1") + tdSql.checkRows(0) + + tdSql.query("select last(col6) from test1") + tdSql.checkRows(0) + + tdSql.query("select last(col7) from test1") + tdSql.checkRows(0) + + tdSql.query("select last(col8) from test1") + tdSql.checkRows(0) + + tdSql.query("select last(col9) from test1") + tdSql.checkRows(0) + + for i in range(self.rowNum): + tdSql.execute("insert into test1 values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" + % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) + + tdSql.query("select last(*) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 1, 10) + + tdSql.query("select last(col1) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + + tdSql.query("select last(col2) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + + tdSql.query("select last(col3) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + + tdSql.query("select last(col4) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + + tdSql.query("select last(col11) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + + tdSql.query("select last(col12) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + + tdSql.query("select last(col13) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + + tdSql.query("select last(col14) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + + tdSql.query("select last(col5) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 9.1) + + tdSql.query("select last(col6) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 9.1) + + tdSql.query("select last(col7) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, True) + + tdSql.query("select last(col8) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 'taosdata10') + + tdSql.query("select last(col9) from test1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, '涛思数据10') + + def stop(self): + tdSql.close() + tdLog.success("%s successfully executed" % __file__) + +tdCases.addWindows(__file__, TDTestCase()) +tdCases.addLinux(__file__, TDTestCase()) \ No newline at end of file From 971c13719e759fcf3dd2aa5e0a8bf0383293fcf9 Mon Sep 17 00:00:00 2001 From: jiacy-jcy <714897623@qq.com> Date: Fri, 6 May 2022 15:19:33 +0800 Subject: [PATCH 07/27] update last.py --- tests/system-test/2-query/last.py | 322 ++++++++++++++++++++++++------ 1 file changed, 260 insertions(+), 62 deletions(-) diff --git a/tests/system-test/2-query/last.py b/tests/system-test/2-query/last.py index 82e648f60a..fe5beefea5 100644 --- a/tests/system-test/2-query/last.py +++ b/tests/system-test/2-query/last.py @@ -15,115 +15,313 @@ class TDTestCase: def run(self): tdSql.prepare() - tdSql.execute('''create table test(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, + tdSql.execute('''create table stb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned) tags(loc nchar(20))''') - tdSql.execute("create table test1 using test tags('beijing')") - tdSql.execute("insert into test1(ts) values(%d)" % (self.ts - 1)) + tdSql.execute("create table stb_1 using stb tags('beijing')") + tdSql.execute("insert into stb_1(ts) values(%d)" % (self.ts - 1)) # last verifacation - tdSql.query("select last(*) from test1") + tdSql.query("select last(*) from stb_1") tdSql.checkRows(1) tdSql.checkData(0, 1, None) - - tdSql.query("select last(col1) from test1") - tdSql.checkRows(0) - - tdSql.query("select last(col2) from test1") + tdSql.query("select last(*) from db.stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 1, None) + tdSql.query("select last(col1) from stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col1) from db.stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col2) from stb_1") tdSql.checkRows(0) - - tdSql.query("select last(col3) from test1") + tdSql.query("select last(col2) from db.stb_1") tdSql.checkRows(0) - - tdSql.query("select last(col4) from test1") + tdSql.query("select last(col3) from stb_1") tdSql.checkRows(0) - - tdSql.query("select last(col11) from test1") - tdSql.checkRows(0) - - tdSql.query("select last(col12) from test1") + tdSql.query("select last(col3) from db.stb_1") tdSql.checkRows(0) - - tdSql.query("select last(col13) from test1") + tdSql.query("select last(col4) from stb_1") tdSql.checkRows(0) - - tdSql.query("select last(col14) from test1") + tdSql.query("select last(col4) from db.stb_1") tdSql.checkRows(0) - - tdSql.query("select last(col5) from test1") + tdSql.query("select last(col11) from stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col11) from db.stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col12) from stb_1") tdSql.checkRows(0) - - tdSql.query("select last(col6) from test1") + tdSql.query("select last(col12) from db.stb_1") tdSql.checkRows(0) - - tdSql.query("select last(col7) from test1") + tdSql.query("select last(col13) from stb_1") tdSql.checkRows(0) - - tdSql.query("select last(col8) from test1") + tdSql.query("select last(col13) from db.stb_1") tdSql.checkRows(0) - - tdSql.query("select last(col9) from test1") + tdSql.query("select last(col14) from stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col14) from db.stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col5) from stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col5) from db.stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col6) from stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col6) from db.stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col7) from stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col7) from db.stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col8) from stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col8) from db.stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col9) from stb_1") + tdSql.checkRows(0) + tdSql.query("select last(col9) from db.stb_1") tdSql.checkRows(0) for i in range(self.rowNum): - tdSql.execute("insert into test1 values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" + tdSql.execute("insert into stb_1 values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) - tdSql.query("select last(*) from test1") + tdSql.query("select last(*) from stb_1") tdSql.checkRows(1) tdSql.checkData(0, 1, 10) - - tdSql.query("select last(col1) from test1") + tdSql.query("select last(*) from db.stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 1, 10) + tdSql.query("select last(col1) from stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 10) - - tdSql.query("select last(col2) from test1") + tdSql.query("select last(col1) from db.stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 10) - - tdSql.query("select last(col3) from test1") + tdSql.query("select last(col2) from stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 10) - - tdSql.query("select last(col4) from test1") + tdSql.query("select last(col2) from db.stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 10) - - tdSql.query("select last(col11) from test1") + tdSql.query("select last(col3) from stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 10) - - tdSql.query("select last(col12) from test1") + tdSql.query("select last(col3) from db.stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 10) - - tdSql.query("select last(col13) from test1") + tdSql.query("select last(col4) from stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 10) - - tdSql.query("select last(col14) from test1") + tdSql.query("select last(col4) from db.stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 10) - - tdSql.query("select last(col5) from test1") + tdSql.query("select last(col11) from stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col11) from db.stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col12) from stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col12) from db.stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col13) from stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col13) from db.stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col14) from stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col14) from db.stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col5) from stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 9.1) - - tdSql.query("select last(col6) from test1") + tdSql.query("select last(col5) from db.stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 9.1) - - tdSql.query("select last(col7) from test1") + tdSql.query("select last(col6) from stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 9.1) + tdSql.query("select last(col6) from db.stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 9.1) + tdSql.query("select last(col7) from stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, True) - - tdSql.query("select last(col8) from test1") + tdSql.query("select last(col7) from db.stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, True) + tdSql.query("select last(col8) from stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 'taosdata10') - - tdSql.query("select last(col9) from test1") + tdSql.query("select last(col8) from db.stb_1") tdSql.checkRows(1) - tdSql.checkData(0, 0, '涛思数据10') - + tdSql.checkData(0, 0, 'taosdata10') + # tdSql.query("select last(col9) from stb_1") + # tdSql.checkRows(1) + # tdSql.checkData(0, 0, '涛思数据10') + # tdSql.query("select last(col9) from db.stb_1") + # tdSql.checkRows(1) + # tdSql.checkData(0, 0, '涛思数据10') + + + + tdSql.execute('''create table ntb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, + col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned)''') + tdSql.execute("insert into ntb(ts) values(%d)" % (self.ts - 1)) + tdSql.query("select last(*) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 1, None) + tdSql.query("select last(*) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 1, None) + tdSql.query("select last(col1) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col1) from db.ntb") + tdSql.checkRows(0) + tdSql.query("select last(col2) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col2) from db.ntb") + tdSql.checkRows(0) + tdSql.query("select last(col3) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col3) from db.ntb") + tdSql.checkRows(0) + tdSql.query("select last(col4) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col4) from db.ntb") + tdSql.checkRows(0) + tdSql.query("select last(col11) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col11) from db.ntb") + tdSql.checkRows(0) + tdSql.query("select last(col12) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col12) from db.ntb") + tdSql.checkRows(0) + tdSql.query("select last(col13) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col13) from db.ntb") + tdSql.checkRows(0) + tdSql.query("select last(col14) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col14) from db.ntb") + tdSql.checkRows(0) + tdSql.query("select last(col5) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col5) from db.ntb") + tdSql.checkRows(0) + tdSql.query("select last(col6) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col6) from db.ntb") + tdSql.checkRows(0) + tdSql.query("select last(col7) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col7) from db.ntb") + tdSql.checkRows(0) + tdSql.query("select last(col8) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col8) from db.ntb") + tdSql.checkRows(0) + tdSql.query("select last(col9) from ntb") + tdSql.checkRows(0) + tdSql.query("select last(col9) from db.ntb") + tdSql.checkRows(0) + + for i in range(self.rowNum): + tdSql.execute("insert into ntb values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" + % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) + + tdSql.query("select last(*) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 1, 10) + tdSql.query("select last(*) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 1, 10) + tdSql.query("select last(col1) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col1) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col2) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col2) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col3) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col3) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col4) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col4) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col11) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col11) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col12) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col12) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col13) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col13) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col14) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col14) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 10) + tdSql.query("select last(col5) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 9.1) + tdSql.query("select last(col5) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 9.1) + tdSql.query("select last(col6) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 9.1) + tdSql.query("select last(col6) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 9.1) + tdSql.query("select last(col7) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, True) + tdSql.query("select last(col7) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, True) + tdSql.query("select last(col8) from ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 'taosdata10') + tdSql.query("select last(col8) from db.ntb") + tdSql.checkRows(1) + tdSql.checkData(0, 0, 'taosdata10') + # tdSql.query("select last(col9) from ntb") + # tdSql.checkRows(1) + # tdSql.checkData(0, 0, '涛思数据10') + # tdSql.query("select last(col9) from db.ntb") + # tdSql.checkRows(1) + # tdSql.checkData(0, 0, '涛思数据10') + def stop(self): tdSql.close() tdLog.success("%s successfully executed" % __file__) From d12532b4d24527fef9a57f1a20518265e15a1a59 Mon Sep 17 00:00:00 2001 From: dapan1121 Date: Fri, 6 May 2022 16:35:24 +0800 Subject: [PATCH 08/27] feature/qnodew --- source/dnode/mgmt/mgmt_qnode/src/qmHandle.c | 1 + source/dnode/qnode/src/qnode.c | 2 + tests/script/api/batchprepare.c | 131 +++++++++++++++++++- 3 files changed, 130 insertions(+), 4 deletions(-) diff --git a/source/dnode/mgmt/mgmt_qnode/src/qmHandle.c b/source/dnode/mgmt/mgmt_qnode/src/qmHandle.c index 805b4c7264..11b91f0568 100644 --- a/source/dnode/mgmt/mgmt_qnode/src/qmHandle.c +++ b/source/dnode/mgmt/mgmt_qnode/src/qmHandle.c @@ -101,6 +101,7 @@ void qmInitMsgHandle(SMgmtWrapper *pWrapper) { dmSetMsgHandle(pWrapper, TDMT_VND_QUERY_CONTINUE, qmProcessQueryMsg, QNODE_HANDLE); dmSetMsgHandle(pWrapper, TDMT_VND_FETCH, qmProcessFetchMsg, QNODE_HANDLE); dmSetMsgHandle(pWrapper, TDMT_VND_FETCH_RSP, qmProcessFetchMsg, QNODE_HANDLE); + dmSetMsgHandle(pWrapper, TDMT_VND_QUERY_HEARTBEAT, qmProcessFetchMsg, QNODE_HANDLE); dmSetMsgHandle(pWrapper, TDMT_VND_RES_READY, qmProcessFetchMsg, QNODE_HANDLE); dmSetMsgHandle(pWrapper, TDMT_VND_TASKS_STATUS, qmProcessFetchMsg, QNODE_HANDLE); diff --git a/source/dnode/qnode/src/qnode.c b/source/dnode/qnode/src/qnode.c index 7a226a4c6b..b21141001a 100644 --- a/source/dnode/qnode/src/qnode.c +++ b/source/dnode/qnode/src/qnode.c @@ -84,6 +84,8 @@ int32_t qndProcessFetchMsg(SQnode *pQnode, SRpcMsg *pMsg) { // return vnodeGetTableMeta(pQnode, pMsg); case TDMT_VND_CONSUME: // return tqProcessConsumeReq(pQnode->pTq, pMsg); + case TDMT_VND_QUERY_HEARTBEAT: + return qWorkerProcessHbMsg(pQnode, pQnode->pQuery, pMsg); default: qError("unknown msg type:%d in fetch queue", pMsg->msgType); return TSDB_CODE_VND_APP_ERROR; diff --git a/tests/script/api/batchprepare.c b/tests/script/api/batchprepare.c index 828216d1d3..99e546fcd0 100644 --- a/tests/script/api/batchprepare.c +++ b/tests/script/api/batchprepare.c @@ -398,7 +398,7 @@ void bpAppendOperatorParam(BindData *data, int32_t *len, int32_t dataType, int32 } } -void generateQuerySQL(BindData *data, int32_t tblIdx) { +void generateQueryCondSQL(BindData *data, int32_t tblIdx) { int32_t len = sprintf(data->sql, "select * from %s%d where ", bpTbPrefix, tblIdx); if (!gCurCase->fullCol) { for (int c = 0; c < gCurCase->bindColNum; ++c) { @@ -462,6 +462,72 @@ void generateQuerySQL(BindData *data, int32_t tblIdx) { } } + +void generateQueryMiscSQL(BindData *data, int32_t tblIdx) { + int32_t len = sprintf(data->sql, "select * from %s%d where ", bpTbPrefix, tblIdx); + if (!gCurCase->fullCol) { + for (int c = 0; c < gCurCase->bindColNum; ++c) { + if (c) { + len += sprintf(data->sql + len, " and "); + } + switch (data->pBind[c].buffer_type) { + case TSDB_DATA_TYPE_BOOL: + len += sprintf(data->sql + len, "booldata"); + break; + case TSDB_DATA_TYPE_TINYINT: + len += sprintf(data->sql + len, "tinydata"); + break; + case TSDB_DATA_TYPE_SMALLINT: + len += sprintf(data->sql + len, "smalldata"); + break; + case TSDB_DATA_TYPE_INT: + len += sprintf(data->sql + len, "intdata"); + break; + case TSDB_DATA_TYPE_BIGINT: + len += sprintf(data->sql + len, "bigdata"); + break; + case TSDB_DATA_TYPE_FLOAT: + len += sprintf(data->sql + len, "floatdata"); + break; + case TSDB_DATA_TYPE_DOUBLE: + len += sprintf(data->sql + len, "doubledata"); + break; + case TSDB_DATA_TYPE_VARCHAR: + len += sprintf(data->sql + len, "binarydata"); + break; + case TSDB_DATA_TYPE_TIMESTAMP: + len += sprintf(data->sql + len, "ts"); + break; + case TSDB_DATA_TYPE_NCHAR: + len += sprintf(data->sql + len, "nchardata"); + break; + case TSDB_DATA_TYPE_UTINYINT: + len += sprintf(data->sql + len, "utinydata"); + break; + case TSDB_DATA_TYPE_USMALLINT: + len += sprintf(data->sql + len, "usmalldata"); + break; + case TSDB_DATA_TYPE_UINT: + len += sprintf(data->sql + len, "uintdata"); + break; + case TSDB_DATA_TYPE_UBIGINT: + len += sprintf(data->sql + len, "ubigdata"); + break; + default: + printf("!!!invalid col type:%d", data->pBind[c].buffer_type); + exit(1); + } + + bpAppendOperatorParam(data, &len, data->pBind[c].buffer_type, c); + } + } + + if (gCaseCtrl.printStmtSql) { + printf("\tSTMT SQL: %s\n", data->sql); + } +} + + void generateErrorSQL(BindData *data, int32_t tblIdx) { int32_t len = 0; data->sql = taosMemoryCalloc(1, 1024); @@ -677,7 +743,7 @@ int32_t prepareInsertData(BindData *data) { return 0; } -int32_t prepareQueryData(BindData *data, int32_t tblIdx) { +int32_t prepareQueryCondData(BindData *data, int32_t tblIdx) { static int64_t tsData = 1591060628000; uint64_t bindNum = gCurCase->rowNum / gCurCase->bindRowNum; @@ -735,6 +801,63 @@ int32_t prepareQueryData(BindData *data, int32_t tblIdx) { } +int32_t prepareQueryMiscData(BindData *data, int32_t tblIdx) { + static int64_t tsData = 1591060628000; + uint64_t bindNum = gCurCase->rowNum / gCurCase->bindRowNum; + + data->colNum = 0; + data->colTypes = taosMemoryCalloc(30, sizeof(int32_t)); + data->sql = taosMemoryCalloc(1, 1024); + data->pBind = taosMemoryCalloc(bindNum*gCurCase->bindColNum, sizeof(TAOS_MULTI_BIND)); + data->tsData = taosMemoryMalloc(bindNum * sizeof(int64_t)); + data->boolData = taosMemoryMalloc(bindNum * sizeof(bool)); + data->tinyData = taosMemoryMalloc(bindNum * sizeof(int8_t)); + data->utinyData = taosMemoryMalloc(bindNum * sizeof(uint8_t)); + data->smallData = taosMemoryMalloc(bindNum * sizeof(int16_t)); + data->usmallData = taosMemoryMalloc(bindNum * sizeof(uint16_t)); + data->intData = taosMemoryMalloc(bindNum * sizeof(int32_t)); + data->uintData = taosMemoryMalloc(bindNum * sizeof(uint32_t)); + data->bigData = taosMemoryMalloc(bindNum * sizeof(int64_t)); + data->ubigData = taosMemoryMalloc(bindNum * sizeof(uint64_t)); + data->floatData = taosMemoryMalloc(bindNum * sizeof(float)); + data->doubleData = taosMemoryMalloc(bindNum * sizeof(double)); + data->binaryData = taosMemoryMalloc(bindNum * gVarCharSize); + data->binaryLen = taosMemoryMalloc(bindNum * sizeof(int32_t)); + if (gCurCase->bindNullNum) { + data->isNull = taosMemoryCalloc(bindNum, sizeof(char)); + } + + for (int32_t i = 0; i < bindNum; ++i) { + data->tsData[i] = tsData + tblIdx*gCurCase->rowNum + rand()%gCurCase->rowNum; + data->boolData[i] = (bool)(tblIdx*gCurCase->rowNum + rand() % gCurCase->rowNum); + data->tinyData[i] = (int8_t)(tblIdx*gCurCase->rowNum + rand() % gCurCase->rowNum); + data->utinyData[i] = (uint8_t)(tblIdx*gCurCase->rowNum + rand() % gCurCase->rowNum); + data->smallData[i] = (int16_t)(tblIdx*gCurCase->rowNum + rand() % gCurCase->rowNum); + data->usmallData[i] = (uint16_t)(tblIdx*gCurCase->rowNum + rand() % gCurCase->rowNum); + data->intData[i] = (int32_t)(tblIdx*gCurCase->rowNum + rand() % gCurCase->rowNum); + data->uintData[i] = (uint32_t)(tblIdx*gCurCase->rowNum + rand() % gCurCase->rowNum); + data->bigData[i] = (int64_t)(tblIdx*gCurCase->rowNum + rand() % gCurCase->rowNum); + data->ubigData[i] = (uint64_t)(tblIdx*gCurCase->rowNum + rand() % gCurCase->rowNum); + data->floatData[i] = (float)(tblIdx*gCurCase->rowNum + rand() % gCurCase->rowNum); + data->doubleData[i] = (double)(tblIdx*gCurCase->rowNum + rand() % gCurCase->rowNum); + memset(data->binaryData + gVarCharSize * i, 'a'+i%26, gVarCharLen); + if (gCurCase->bindNullNum) { + data->isNull[i] = i % 2; + } + data->binaryLen[i] = gVarCharLen; + } + + for (int b = 0; b < bindNum; b++) { + for (int c = 0; c < gCurCase->bindColNum; ++c) { + prepareColData(data, b*gCurCase->bindColNum+c, b*gCurCase->bindRowNum, c); + } + } + + generateQueryMiscSQL(data, tblIdx); + + return 0; +} + void destroyData(BindData *data) { @@ -1385,7 +1508,7 @@ int querySUBTTest1(TAOS_STMT *stmt, TAOS *taos) { for (int32_t t = 0; t< gCurCase->tblNum; ++t) { memset(&data, 0, sizeof(data)); - prepareQueryData(&data, t); + prepareQueryCondData(&data, t); int code = taos_stmt_prepare(stmt, data.sql, 0); if (code != 0){ @@ -1431,7 +1554,7 @@ int querySUBTTest2(TAOS_STMT *stmt, TAOS *taos) { for (int32_t t = 0; t< gCurCase->tblNum; ++t) { memset(&data, 0, sizeof(data)); - prepareQueryData(&data, t); + prepareQueryMiscData(&data, t); int code = taos_stmt_prepare(stmt, data.sql, 0); if (code != 0){ From c9e6bd4afd45c4ee188b1d42828ba23980ccbf91 Mon Sep 17 00:00:00 2001 From: jiacy-jcy <714897623@qq.com> Date: Fri, 6 May 2022 18:07:51 +0800 Subject: [PATCH 09/27] update testcase --- tests/system-test/2-query/timetruncate.py | 118 +++++++++++++++++++++- tests/system-test/fulltest.sh | 5 +- 2 files changed, 121 insertions(+), 2 deletions(-) diff --git a/tests/system-test/2-query/timetruncate.py b/tests/system-test/2-query/timetruncate.py index 4c4ebb09df..a48851b251 100644 --- a/tests/system-test/2-query/timetruncate.py +++ b/tests/system-test/2-query/timetruncate.py @@ -95,7 +95,123 @@ class TDTestCase: # tdSql.checkData(9,0,"2018-09-17 09:00:00.009000000") - + tdSql.query("select timetruncate(1,1d) from stb") + tdSql.checkRows(10) + tdSql.query("select timetruncate(1,1u) from stb") + tdSql.checkRows(10) + tdSql.query("select timetruncate(1,1a) from stb") + tdSql.checkRows(10) + tdSql.query("select timetruncate(1,1m) from stb") + tdSql.checkRows(10) + tdSql.query("select timetruncate(1,1h) from stb") + tdSql.checkRows(10) + tdSql.query("select timetruncate(ts,1d) from stb") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 08:00:00.000") + tdSql.query("select timetruncate(ts,1h) from stb") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 09:00:00.000") + tdSql.query("select timetruncate(ts,1m) from stb") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 09:00:00.000") + tdSql.query("select timetruncate(ts,1s) from stb") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 09:00:00.000") + tdSql.query("select timetruncate(ts,1a) from stb") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 09:00:00.000") + tdSql.checkData(1,0,"2018-09-17 09:00:00.001") + tdSql.checkData(2,0,"2018-09-17 09:00:00.002") + tdSql.checkData(3,0,"2018-09-17 09:00:00.003") + tdSql.checkData(4,0,"2018-09-17 09:00:00.004") + tdSql.checkData(5,0,"2018-09-17 09:00:00.005") + tdSql.checkData(6,0,"2018-09-17 09:00:00.006") + tdSql.checkData(7,0,"2018-09-17 09:00:00.007") + tdSql.checkData(8,0,"2018-09-17 09:00:00.008") + tdSql.checkData(9,0,"2018-09-17 09:00:00.009") + # tdSql.query("select timetruncate(ts,1u) from stb") + # tdSql.checkRows(10) + # tdSql.checkData(0,0,"2018-09-17 09:00:00.000000") + # tdSql.checkData(1,0,"2018-09-17 09:00:00.001000") + # tdSql.checkData(2,0,"2018-09-17 09:00:00.002000") + # tdSql.checkData(3,0,"2018-09-17 09:00:00.003000") + # tdSql.checkData(4,0,"2018-09-17 09:00:00.004000") + # tdSql.checkData(5,0,"2018-09-17 09:00:00.005000") + # tdSql.checkData(6,0,"2018-09-17 09:00:00.006000") + # tdSql.checkData(7,0,"2018-09-17 09:00:00.007000") + # tdSql.checkData(8,0,"2018-09-17 09:00:00.008000") + # tdSql.checkData(9,0,"2018-09-17 09:00:00.009000") + # tdSql.query("select timetruncate(ts,1b) from stb") + # tdSql.checkRows(10) + # tdSql.checkData(0,0,"2018-09-17 09:00:00.000000000") + # tdSql.checkData(1,0,"2018-09-17 09:00:00.001000000") + # tdSql.checkData(2,0,"2018-09-17 09:00:00.002000000") + # tdSql.checkData(3,0,"2018-09-17 09:00:00.003000000") + # tdSql.checkData(4,0,"2018-09-17 09:00:00.004000000") + # tdSql.checkData(5,0,"2018-09-17 09:00:00.005000000") + # tdSql.checkData(6,0,"2018-09-17 09:00:00.006000000") + # tdSql.checkData(7,0,"2018-09-17 09:00:00.007000000") + # tdSql.checkData(8,0,"2018-09-17 09:00:00.008000000") + # tdSql.checkData(9,0,"2018-09-17 09:00:00.009000000") + + tdSql.query("select timetruncate(1,1d) from stb_1") + tdSql.checkRows(10) + tdSql.query("select timetruncate(1,1u) from stb_1") + tdSql.checkRows(10) + tdSql.query("select timetruncate(1,1a) from stb_1") + tdSql.checkRows(10) + tdSql.query("select timetruncate(1,1m) from stb_1") + tdSql.checkRows(10) + tdSql.query("select timetruncate(1,1h) from stb_1") + tdSql.checkRows(10) + tdSql.query("select timetruncate(ts,1d) from stb_1") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 08:00:00.000") + tdSql.query("select timetruncate(ts,1h) from stb_1") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 09:00:00.000") + tdSql.query("select timetruncate(ts,1m) from stb_1") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 09:00:00.000") + tdSql.query("select timetruncate(ts,1s) from stb_1") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 09:00:00.000") + tdSql.query("select timetruncate(ts,1a) from stb_1") + tdSql.checkRows(10) + tdSql.checkData(0,0,"2018-09-17 09:00:00.000") + tdSql.checkData(1,0,"2018-09-17 09:00:00.001") + tdSql.checkData(2,0,"2018-09-17 09:00:00.002") + tdSql.checkData(3,0,"2018-09-17 09:00:00.003") + tdSql.checkData(4,0,"2018-09-17 09:00:00.004") + tdSql.checkData(5,0,"2018-09-17 09:00:00.005") + tdSql.checkData(6,0,"2018-09-17 09:00:00.006") + tdSql.checkData(7,0,"2018-09-17 09:00:00.007") + tdSql.checkData(8,0,"2018-09-17 09:00:00.008") + tdSql.checkData(9,0,"2018-09-17 09:00:00.009") + # tdSql.query("select timetruncate(ts,1u) from stb_1") + # tdSql.checkRows(10) + # tdSql.checkData(0,0,"2018-09-17 09:00:00.000000") + # tdSql.checkData(1,0,"2018-09-17 09:00:00.001000") + # tdSql.checkData(2,0,"2018-09-17 09:00:00.002000") + # tdSql.checkData(3,0,"2018-09-17 09:00:00.003000") + # tdSql.checkData(4,0,"2018-09-17 09:00:00.004000") + # tdSql.checkData(5,0,"2018-09-17 09:00:00.005000") + # tdSql.checkData(6,0,"2018-09-17 09:00:00.006000") + # tdSql.checkData(7,0,"2018-09-17 09:00:00.007000") + # tdSql.checkData(8,0,"2018-09-17 09:00:00.008000") + # tdSql.checkData(9,0,"2018-09-17 09:00:00.009000") + # tdSql.query("select timetruncate(ts,1b) from stb_1") + # tdSql.checkRows(10) + # tdSql.checkData(0,0,"2018-09-17 09:00:00.000000000") + # tdSql.checkData(1,0,"2018-09-17 09:00:00.001000000") + # tdSql.checkData(2,0,"2018-09-17 09:00:00.002000000") + # tdSql.checkData(3,0,"2018-09-17 09:00:00.003000000") + # tdSql.checkData(4,0,"2018-09-17 09:00:00.004000000") + # tdSql.checkData(5,0,"2018-09-17 09:00:00.005000000") + # tdSql.checkData(6,0,"2018-09-17 09:00:00.006000000") + # tdSql.checkData(7,0,"2018-09-17 09:00:00.007000000") + # tdSql.checkData(8,0,"2018-09-17 09:00:00.008000000") + # tdSql.checkData(9,0,"2018-09-17 09:00:00.009000000") def stop(self): tdSql.close() tdLog.success("%s successfully executed" % __file__) diff --git a/tests/system-test/fulltest.sh b/tests/system-test/fulltest.sh index 0b7a71bbe4..026dba6490 100755 --- a/tests/system-test/fulltest.sh +++ b/tests/system-test/fulltest.sh @@ -14,7 +14,10 @@ python3 ./test.py -f 2-query/varchar.py #python3 ./test.py -f 2-query/timezone.py python3 ./test.py -f 2-query/Now.py python3 ./test.py -f 2-query/Today.py - +python3 ./test.py -f 2-query/max.py +python3 ./test.py -f 2-query/min.py +python3 ./test.py -f 2-query/count.py +# python3 ./test.py -f 2-query/Timediff.py #python3 ./test.py -f 2-query/cast.py From 679726ebb7cee70f0eb88e9efbb737fb96fbec51 Mon Sep 17 00:00:00 2001 From: jiacy-jcy <714897623@qq.com> Date: Fri, 6 May 2022 20:47:27 +0800 Subject: [PATCH 10/27] update test case --- tests/system-test/2-query/count.py | 86 +++++++++++++++++------------- tests/system-test/2-query/last.py | 3 +- 2 files changed, 52 insertions(+), 37 deletions(-) diff --git a/tests/system-test/2-query/count.py b/tests/system-test/2-query/count.py index 8888a182b8..a70a2e72f2 100644 --- a/tests/system-test/2-query/count.py +++ b/tests/system-test/2-query/count.py @@ -16,6 +16,7 @@ class TDTestCase: tdSql.execute('''create table stb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned) tags(loc nchar(20))''') tdSql.execute("create table stb_1 using stb tags('beijing')") + tdSql.execute("create table stb_2 using stb tags('shanghai')") tdSql.execute('''create table ntb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned)''') @@ -23,71 +24,76 @@ class TDTestCase: for i in range(self.rowNum): tdSql.execute("insert into stb_1 values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) + tdSql.execute("insert into stb_2 values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" + % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) for i in range(self.rowNum): tdSql.execute("insert into ntb values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) tdSql.query("select count(*) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(*) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(ts) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(ts) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col1) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col1) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col2) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col2) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col3) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col3) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col4) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col4) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col5) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col5) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col6) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col6) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col7) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col7) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col8) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col8) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col9) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col9) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col11) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col11) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col12) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col12) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col13) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col13) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col14) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col14) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) + + + tdSql.query("select count(ts) from db.stb_1") tdSql.checkData(0,0,10) tdSql.query("select count(ts) from db.stb_1") @@ -144,6 +150,8 @@ class TDTestCase: tdSql.checkData(0,0,10) tdSql.query("select count(col14) from db.stb_1") tdSql.checkData(0,0,10) + tdSql.query("select count(col1) from stb_1 group by col7") + tdSql.checkRows(2) tdSql.execute("insert into stb_1 values(now,null,null,null,null,null,null,null,null,null,null,null,null,null)") tdSql.query("select count(col1) from stb_1") @@ -159,16 +167,22 @@ class TDTestCase: tdSql.checkData(0,1,11) tdSql.query("select count(col1) from stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col1),count(ts) from stb") - tdSql.checkData(0,0,10) - tdSql.checkData(0,1,11) + tdSql.checkData(0,0,20) + tdSql.checkData(0,1,21) tdSql.query("select count(col1) from db.stb") - tdSql.checkData(0,0,10) + tdSql.checkData(0,0,20) tdSql.query("select count(col1),count(ts) from db.stb") - tdSql.checkData(0,0,10) - tdSql.checkData(0,1,11) + tdSql.checkData(0,0,20) + tdSql.checkData(0,1,21) + tdSql.query("select count(col1) from stb_1 group by col7") + tdSql.checkRows(3) + tdSql.query("select count(col1) from stb_2 group by col7") + tdSql.checkRows(2) + tdSql.query("select count(col1) from stb group by col7") + tdSql.checkRows(3) diff --git a/tests/system-test/2-query/last.py b/tests/system-test/2-query/last.py index fe5beefea5..a5407c7b12 100644 --- a/tests/system-test/2-query/last.py +++ b/tests/system-test/2-query/last.py @@ -79,6 +79,8 @@ class TDTestCase: tdSql.checkRows(0) tdSql.query("select last(col9) from db.stb_1") tdSql.checkRows(0) + tdSql.query("select count(col1) from stb_1 group by col7") + tdSql.checkRows(1) for i in range(self.rowNum): tdSql.execute("insert into stb_1 values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" @@ -170,7 +172,6 @@ class TDTestCase: # tdSql.checkData(0, 0, '涛思数据10') - tdSql.execute('''create table ntb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned)''') tdSql.execute("insert into ntb(ts) values(%d)" % (self.ts - 1)) From bc5e63b0a78bfd62dce88e3b5e81c0cd889f8a6b Mon Sep 17 00:00:00 2001 From: Ganlin Zhao Date: Fri, 6 May 2022 21:44:25 +0800 Subject: [PATCH 11/27] feat(query): add histogram param parsing --- source/libs/function/src/builtinsimpl.c | 169 +++++++++++++++++++++++- 1 file changed, 163 insertions(+), 6 deletions(-) diff --git a/source/libs/function/src/builtinsimpl.c b/source/libs/function/src/builtinsimpl.c index 3c9eca85dd..64cca47da0 100644 --- a/source/libs/function/src/builtinsimpl.c +++ b/source/libs/function/src/builtinsimpl.c @@ -14,13 +14,14 @@ */ #include "builtinsimpl.h" +#include "cJSON.h" #include "function.h" #include "querynodes.h" #include "taggfunction.h" #include "tdatablock.h" #include "tpercentile.h" -#define HISTOGRAM_MAX_BINS_NUM 100 +#define HISTOGRAM_MAX_BINS_NUM 100 typedef struct SSumRes { union { @@ -106,6 +107,13 @@ typedef struct SHistoFuncInfo { SHistoFuncBin bins[]; } SHistoFuncInfo; +typedef enum { + UNKNOWN_BIN = 0, + USER_INPUT_BIN, + LINEAR_BIN, + LOG_BIN +} EHistoBinType; + #define SET_VAL(_info, numOfElem, res) \ do { \ @@ -1801,16 +1809,165 @@ bool getHistogramFuncEnv(SFunctionNode* UNUSED_PARAM(pFunc), SFuncExecEnv* pEnv) return true; } -bool histogramFunctionSetup(SqlFunctionCtx *pCtx, SResultRowEntryInfo* pResultInfo) { +static int8_t getHistogramBinType(char *binTypeStr) { + int8_t binType; + if (strcasecmp(binTypeStr, "user_input") == 0) { + binType = USER_INPUT_BIN; + } else if (strcasecmp(binTypeStr, "linear_bin") == 0) { + binType = LINEAR_BIN; + } else if (strcasecmp(binTypeStr, "log_bin") == 0) { + binType = LOG_BIN; + } else { + binType = UNKNOWN_BIN; + } + + return binType; +} + +static bool getHistogramBinDesc(SHistoFuncInfo *pInfo, char *binDescStr, int8_t binType, bool normalized) { + cJSON* binDesc = cJSON_Parse(binDescStr); + int32_t counter; + int32_t numOfBins; + double* intervals; + if (cJSON_IsObject(binDesc)) { /* linaer/log bins */ + int32_t numOfParams = cJSON_GetArraySize(binDesc); + int32_t startIndex; + if (numOfParams != 4) { + return false; + } + + cJSON* start = cJSON_GetObjectItem(binDesc, "start"); + cJSON* factor = cJSON_GetObjectItem(binDesc, "factor"); + cJSON* width = cJSON_GetObjectItem(binDesc, "width"); + cJSON* count = cJSON_GetObjectItem(binDesc, "count"); + cJSON* infinity = cJSON_GetObjectItem(binDesc, "infinity"); + + if (!cJSON_IsNumber(start) || !cJSON_IsNumber(count) || !cJSON_IsBool(infinity)) { + return false; + } + + if (count->valueint <= 0 || count->valueint > 1000) { // limit count to 1000 + return false; + } + + if (isinf(start->valuedouble) || (width != NULL && isinf(width->valuedouble)) || + (factor != NULL && isinf(factor->valuedouble)) || (count != NULL && isinf(count->valuedouble))) { + return false; + } + + counter = (int32_t)count->valueint; + if (infinity->valueint == false) { + startIndex = 0; + numOfBins = counter + 1; + } else { + startIndex = 1; + numOfBins = counter + 3; + } + + intervals = taosMemoryCalloc(numOfBins, sizeof(double)); + if (cJSON_IsNumber(width) && factor == NULL && binType == LINEAR_BIN) { + // linear bin process + if (width->valuedouble == 0) { + taosMemoryFree(intervals); + return false; + } + for (int i = 0; i < counter + 1; ++i) { + intervals[startIndex] = start->valuedouble + i * width->valuedouble; + if (isinf(intervals[startIndex])) { + taosMemoryFree(intervals); + return false; + } + startIndex++; + } + } else if (cJSON_IsNumber(factor) && width == NULL && binType == LOG_BIN) { + // log bin process + if (start->valuedouble == 0) { + taosMemoryFree(intervals); + return false; + } + if (factor->valuedouble < 0 || factor->valuedouble == 0 || factor->valuedouble == 1) { + taosMemoryFree(intervals); + return false; + } + for (int i = 0; i < counter + 1; ++i) { + intervals[startIndex] = start->valuedouble * pow(factor->valuedouble, i * 1.0); + if (isinf(intervals[startIndex])) { + taosMemoryFree(intervals); + return false; + } + startIndex++; + } + } else { + taosMemoryFree(intervals); + return false; + } + + if (infinity->valueint == true) { + intervals[0] = -INFINITY; + intervals[numOfBins - 1] = INFINITY; + // in case of desc bin orders, -inf/inf should be swapped + ASSERT(numOfBins >= 4); + if (intervals[1] > intervals[numOfBins - 2]) { + TSWAP(intervals[0], intervals[numOfBins - 1]); + } + } + } else if (cJSON_IsArray(binDesc)) { /* user input bins */ + if (binType != USER_INPUT_BIN) { + return false; + } + counter = numOfBins = cJSON_GetArraySize(binDesc); + intervals = taosMemoryCalloc(numOfBins, sizeof(double)); + cJSON* bin = binDesc->child; + if (bin == NULL) { + taosMemoryFree(intervals); + return false; + } + int i = 0; + while (bin) { + intervals[i] = bin->valuedouble; + if (!cJSON_IsNumber(bin)) { + taosMemoryFree(intervals); + return false; + } + if (i != 0 && intervals[i] <= intervals[i - 1]) { + taosMemoryFree(intervals); + return false; + } + bin = bin->next; + i++; + } + } else { + return false; + } + + pInfo->numOfBins = numOfBins; + pInfo->normalized = normalized; + for (int32_t i = 0; i < numOfBins; ++i) { + pInfo->bins[i].lower = intervals[i] < intervals[i + 1] ? intervals[i] : intervals[i + 1]; + pInfo->bins[i].upper = intervals[i + 1] > intervals[i] ? intervals[i + 1] : intervals[i]; + pInfo->bins[i].count = 0; + } + + taosMemoryFree(intervals); + return true; +} + +bool histogramFunctionSetup(SqlFunctionCtx *pCtx, SResultRowEntryInfo *pResultInfo) { if (!functionSetup(pCtx, pResultInfo)) { return false; } - SHistoFuncInfo* pInfo = GET_ROWCELL_INTERBUF(pResultInfo); - char* binType = pCtx->param[1].param.pz; - char* binDesc = pCtx->param[2].param.pz; - int64_t nornalized = pCtx->param[3].param.i; + SHistoFuncInfo *pInfo = GET_ROWCELL_INTERBUF(pResultInfo); + int8_t binType = getHistogramBinType(varDataVal(pCtx->param[1].param.pz)); + if (binType == UNKNOWN_BIN) { + return false; + } + char* binDesc = varDataVal(pCtx->param[2].param.pz); + int64_t normalized = pCtx->param[3].param.i; + if (!getHistogramBinDesc(pInfo, binDesc, binType, (bool)normalized)) { + return false; + } return true; } From 1f9b6f14dd70c968bf8391afd5d0a74fe24ec527 Mon Sep 17 00:00:00 2001 From: dapan1121 Date: Fri, 6 May 2022 22:11:33 +0800 Subject: [PATCH 12/27] fix mem leak --- source/libs/executor/src/dataSinkMgt.c | 1 + source/libs/executor/src/executorimpl.c | 4 +++- source/libs/qworker/inc/qworkerInt.h | 6 ++++-- source/libs/qworker/src/qworker.c | 21 ++++++++++++++++++++- source/libs/transport/src/transSrv.c | 4 ++++ 5 files changed, 32 insertions(+), 4 deletions(-) diff --git a/source/libs/executor/src/dataSinkMgt.c b/source/libs/executor/src/dataSinkMgt.c index 997397314f..64206fc10a 100644 --- a/source/libs/executor/src/dataSinkMgt.c +++ b/source/libs/executor/src/dataSinkMgt.c @@ -60,4 +60,5 @@ void dsScheduleProcess(void* ahandle, void* pItem) { void dsDestroyDataSinker(DataSinkHandle handle) { SDataSinkHandle* pHandleImpl = (SDataSinkHandle*)handle; pHandleImpl->fDestroy(pHandleImpl); + taosMemoryFree(pHandleImpl); } diff --git a/source/libs/executor/src/executorimpl.c b/source/libs/executor/src/executorimpl.c index a237eb0e7d..10d5e717bd 100644 --- a/source/libs/executor/src/executorimpl.c +++ b/source/libs/executor/src/executorimpl.c @@ -2826,6 +2826,7 @@ void relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray } if (p->info.colId == pmInfo->colId) { + colDataDestroy(taosArrayGet(pBlock->pDataBlock, pmInfo->targetSlotId)); taosArraySet(pBlock->pDataBlock, pmInfo->targetSlotId, p); i++; j++; @@ -4687,7 +4688,7 @@ static SExecTaskInfo* createExecTaskInfo(uint64_t queryId, uint64_t taskId, EOPT char* p = taosMemoryCalloc(1, 128); snprintf(p, 128, "TID:0x%" PRIx64 " QID:0x%" PRIx64, taskId, queryId); - pTaskInfo->id.str = strdup(p); + pTaskInfo->id.str = p; return pTaskInfo; } @@ -5300,6 +5301,7 @@ void doDestroyTask(SExecTaskInfo* pTaskInfo) { // taosArrayDestroy(pTaskInfo->summary.queryProfEvents); // taosHashCleanup(pTaskInfo->summary.operatorProfResults); + destroyOperatorInfo(pTaskInfo->pRoot); taosMemoryFreeClear(pTaskInfo->sql); taosMemoryFreeClear(pTaskInfo->id.str); taosMemoryFreeClear(pTaskInfo); diff --git a/source/libs/qworker/inc/qworkerInt.h b/source/libs/qworker/inc/qworkerInt.h index ca471262ff..3ecf987811 100644 --- a/source/libs/qworker/inc/qworkerInt.h +++ b/source/libs/qworker/inc/qworkerInt.h @@ -25,6 +25,7 @@ extern "C" { #include "tlockfree.h" #include "ttimer.h" #include "tref.h" +#include "plannodes.h" #define QW_DEFAULT_SCHEDULER_NUMBER 10000 #define QW_DEFAULT_TASK_NUMBER 10000 @@ -131,8 +132,9 @@ typedef struct SQWTaskCtx { int8_t events[QW_EVENT_MAX]; - void *taskHandle; - void *sinkHandle; + void *taskHandle; + void *sinkHandle; + SSubplan *plan; } SQWTaskCtx; typedef struct SQWSchStatus { diff --git a/source/libs/qworker/src/qworker.c b/source/libs/qworker/src/qworker.c index bd7ee6321a..717958c033 100644 --- a/source/libs/qworker/src/qworker.c +++ b/source/libs/qworker/src/qworker.c @@ -424,6 +424,11 @@ void qwFreeTask(QW_FPARAMS_DEF, SQWTaskCtx *ctx) { dsDestroyDataSinker(ctx->sinkHandle); ctx->sinkHandle = NULL; } + + if (ctx->plan) { + nodesDestroyNode(ctx->plan); + ctx->plan = NULL; + } } int32_t qwDropTaskCtx(QW_FPARAMS_DEF) { @@ -440,6 +445,7 @@ int32_t qwDropTaskCtx(QW_FPARAMS_DEF) { atomic_store_ptr(&ctx->taskHandle, NULL); atomic_store_ptr(&ctx->sinkHandle, NULL); + atomic_store_ptr(&ctx->plan, NULL); QW_SET_EVENT_PROCESSED(ctx, QW_EVENT_DROP); @@ -922,7 +928,7 @@ _return: int32_t qwProcessQuery(QW_FPARAMS_DEF, SQWMsg *qwMsg, int8_t taskType, int8_t explain) { int32_t code = 0; bool queryRsped = false; - struct SSubplan *plan = NULL; + SSubplan* plan = NULL; SQWPhaseInput input = {0}; qTaskInfo_t pTaskInfo = NULL; DataSinkHandle sinkHandle = NULL; @@ -950,6 +956,8 @@ int32_t qwProcessQuery(QW_FPARAMS_DEF, SQWMsg *qwMsg, int8_t taskType, int8_t ex QW_ERR_JRET(code); } + ctx->plan = plan; + code = qCreateExecTask(qwMsg->node, mgmt->nodeId, tId, plan, &pTaskInfo, &sinkHandle, OPTR_EXEC_MODEL_BATCH); if (code) { QW_TASK_ELOG("qCreateExecTask failed, code:%x - %s", code, tstrerror(code)); @@ -1428,6 +1436,10 @@ void qwCloseRef(void) { taosWUnLockLatch(&gQwMgmt.lock); } +void qwDestroySchStatus(SQWSchStatus *pStatus) { + taosHashCleanup(pStatus->tasksHash); +} + void qwDestroyImpl(void *pMgmt) { SQWorker *mgmt = (SQWorker *)pMgmt; @@ -1439,6 +1451,13 @@ void qwDestroyImpl(void *pMgmt) { // TODO FREE ALL taosHashCleanup(mgmt->ctxHash); + + void *pIter = taosHashIterate(mgmt->schHash, NULL); + while (pIter) { + SQWSchStatus *sch = (SQWSchStatus *)pIter; + qwDestroySchStatus(sch); + pIter = taosHashIterate(mgmt->schHash, pIter); + } taosHashCleanup(mgmt->schHash); taosMemoryFree(mgmt); diff --git a/source/libs/transport/src/transSrv.c b/source/libs/transport/src/transSrv.c index ad3f520210..c0d8ca44ee 100644 --- a/source/libs/transport/src/transSrv.c +++ b/source/libs/transport/src/transSrv.c @@ -786,6 +786,10 @@ static void uvDestroyConn(uv_handle_t* handle) { transQueueDestroy(&conn->srvMsgs); QUEUE_REMOVE(&conn->queue); taosMemoryFree(conn->pTcp); + if (conn->regArg.init == 1) { + transFreeMsg(conn->regArg.msg.pCont); + conn->regArg.init = 0; + } taosMemoryFree(conn); if (thrd->quit && QUEUE_IS_EMPTY(&thrd->conn)) { From 20e0d336c7d1e28f40d663752e73ce4778d7c491 Mon Sep 17 00:00:00 2001 From: dapan1121 Date: Sat, 7 May 2022 08:47:52 +0800 Subject: [PATCH 13/27] fix memory leak issue --- source/libs/executor/src/executorimpl.c | 1 - 1 file changed, 1 deletion(-) diff --git a/source/libs/executor/src/executorimpl.c b/source/libs/executor/src/executorimpl.c index 10d5e717bd..dc0070cbee 100644 --- a/source/libs/executor/src/executorimpl.c +++ b/source/libs/executor/src/executorimpl.c @@ -2826,7 +2826,6 @@ void relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray } if (p->info.colId == pmInfo->colId) { - colDataDestroy(taosArrayGet(pBlock->pDataBlock, pmInfo->targetSlotId)); taosArraySet(pBlock->pDataBlock, pmInfo->targetSlotId, p); i++; j++; From cb4dd0f948c738935499f4a6eb7edf393b9fa7b8 Mon Sep 17 00:00:00 2001 From: jiacy-jcy <714897623@qq.com> Date: Sat, 7 May 2022 09:03:46 +0800 Subject: [PATCH 14/27] update last.py --- tests/system-test/2-query/last.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/system-test/2-query/last.py b/tests/system-test/2-query/last.py index a5407c7b12..531afd0117 100644 --- a/tests/system-test/2-query/last.py +++ b/tests/system-test/2-query/last.py @@ -164,12 +164,12 @@ class TDTestCase: tdSql.query("select last(col8) from db.stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 'taosdata10') - # tdSql.query("select last(col9) from stb_1") - # tdSql.checkRows(1) - # tdSql.checkData(0, 0, '涛思数据10') - # tdSql.query("select last(col9) from db.stb_1") - # tdSql.checkRows(1) - # tdSql.checkData(0, 0, '涛思数据10') + tdSql.query("select last(col9) from stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, '涛思数据10') + tdSql.query("select last(col9) from db.stb_1") + tdSql.checkRows(1) + tdSql.checkData(0, 0, '涛思数据10') tdSql.execute('''create table ntb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, From 285c396b0bcc82923dfaf6afb25117c9ba6463f3 Mon Sep 17 00:00:00 2001 From: Shengliang Guan Date: Sat, 7 May 2022 09:23:06 +0800 Subject: [PATCH 15/27] refactor: adjust cluster mgmt code --- source/dnode/mnode/impl/src/mndCluster.c | 68 ++++++++++++------------ 1 file changed, 35 insertions(+), 33 deletions(-) diff --git a/source/dnode/mnode/impl/src/mndCluster.c b/source/dnode/mnode/impl/src/mndCluster.c index 257db0dcd4..96845fcd42 100644 --- a/source/dnode/mnode/impl/src/mndCluster.c +++ b/source/dnode/mnode/impl/src/mndCluster.c @@ -17,8 +17,8 @@ #include "mndCluster.h" #include "mndShow.h" -#define TSDB_CLUSTER_VER_NUMBE 1 -#define TSDB_CLUSTER_RESERVE_SIZE 64 +#define CLUSTER_VER_NUMBE 1 +#define CLUSTER_RESERVE_SIZE 64 static SSdbRaw *mndClusterActionEncode(SClusterObj *pCluster); static SSdbRow *mndClusterActionDecode(SSdbRaw *pRaw); @@ -30,14 +30,16 @@ static int32_t mndRetrieveClusters(SNodeMsg *pMsg, SShowObj *pShow, SSDataBlock static void mndCancelGetNextCluster(SMnode *pMnode, void *pIter); int32_t mndInitCluster(SMnode *pMnode) { - SSdbTable table = {.sdbType = SDB_CLUSTER, - .keyType = SDB_KEY_INT64, - .deployFp = (SdbDeployFp)mndCreateDefaultCluster, - .encodeFp = (SdbEncodeFp)mndClusterActionEncode, - .decodeFp = (SdbDecodeFp)mndClusterActionDecode, - .insertFp = (SdbInsertFp)mndClusterActionInsert, - .updateFp = (SdbUpdateFp)mndClusterActionUpdate, - .deleteFp = (SdbDeleteFp)mndClusterActionDelete}; + SSdbTable table = { + .sdbType = SDB_CLUSTER, + .keyType = SDB_KEY_INT64, + .deployFp = (SdbDeployFp)mndCreateDefaultCluster, + .encodeFp = (SdbEncodeFp)mndClusterActionEncode, + .decodeFp = (SdbDecodeFp)mndClusterActionDecode, + .insertFp = (SdbInsertFp)mndClusterActionInsert, + .updateFp = (SdbUpdateFp)mndClusterActionUpdate, + .deleteFp = (SdbDeleteFp)mndClusterActionDelete, + }; mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_CLUSTER, mndRetrieveClusters); mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_CLUSTER, mndCancelGetNextCluster); @@ -79,19 +81,19 @@ int64_t mndGetClusterId(SMnode *pMnode) { static SSdbRaw *mndClusterActionEncode(SClusterObj *pCluster) { terrno = TSDB_CODE_OUT_OF_MEMORY; - SSdbRaw *pRaw = sdbAllocRaw(SDB_CLUSTER, TSDB_CLUSTER_VER_NUMBE, sizeof(SClusterObj) + TSDB_CLUSTER_RESERVE_SIZE); - if (pRaw == NULL) goto CLUSTER_ENCODE_OVER; + SSdbRaw *pRaw = sdbAllocRaw(SDB_CLUSTER, CLUSTER_VER_NUMBE, sizeof(SClusterObj) + CLUSTER_RESERVE_SIZE); + if (pRaw == NULL) goto _OVER; int32_t dataPos = 0; - SDB_SET_INT64(pRaw, dataPos, pCluster->id, CLUSTER_ENCODE_OVER) - SDB_SET_INT64(pRaw, dataPos, pCluster->createdTime, CLUSTER_ENCODE_OVER) - SDB_SET_INT64(pRaw, dataPos, pCluster->updateTime, CLUSTER_ENCODE_OVER) - SDB_SET_BINARY(pRaw, dataPos, pCluster->name, TSDB_CLUSTER_ID_LEN, CLUSTER_ENCODE_OVER) - SDB_SET_RESERVE(pRaw, dataPos, TSDB_CLUSTER_RESERVE_SIZE, CLUSTER_ENCODE_OVER) + SDB_SET_INT64(pRaw, dataPos, pCluster->id, _OVER) + SDB_SET_INT64(pRaw, dataPos, pCluster->createdTime, _OVER) + SDB_SET_INT64(pRaw, dataPos, pCluster->updateTime, _OVER) + SDB_SET_BINARY(pRaw, dataPos, pCluster->name, TSDB_CLUSTER_ID_LEN, _OVER) + SDB_SET_RESERVE(pRaw, dataPos, CLUSTER_RESERVE_SIZE, _OVER) terrno = 0; -CLUSTER_ENCODE_OVER: +_OVER: if (terrno != 0) { mError("cluster:%" PRId64 ", failed to encode to raw:%p since %s", pCluster->id, pRaw, terrstr()); sdbFreeRaw(pRaw); @@ -106,29 +108,29 @@ static SSdbRow *mndClusterActionDecode(SSdbRaw *pRaw) { terrno = TSDB_CODE_OUT_OF_MEMORY; int8_t sver = 0; - if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto CLUSTER_DECODE_OVER; + if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER; - if (sver != TSDB_CLUSTER_VER_NUMBE) { + if (sver != CLUSTER_VER_NUMBE) { terrno = TSDB_CODE_SDB_INVALID_DATA_VER; - goto CLUSTER_DECODE_OVER; + goto _OVER; } SSdbRow *pRow = sdbAllocRow(sizeof(SClusterObj)); - if (pRow == NULL) goto CLUSTER_DECODE_OVER; + if (pRow == NULL) goto _OVER; SClusterObj *pCluster = sdbGetRowObj(pRow); - if (pCluster == NULL) goto CLUSTER_DECODE_OVER; + if (pCluster == NULL) goto _OVER; int32_t dataPos = 0; - SDB_GET_INT64(pRaw, dataPos, &pCluster->id, CLUSTER_DECODE_OVER) - SDB_GET_INT64(pRaw, dataPos, &pCluster->createdTime, CLUSTER_DECODE_OVER) - SDB_GET_INT64(pRaw, dataPos, &pCluster->updateTime, CLUSTER_DECODE_OVER) - SDB_GET_BINARY(pRaw, dataPos, pCluster->name, TSDB_CLUSTER_ID_LEN, CLUSTER_DECODE_OVER) - SDB_GET_RESERVE(pRaw, dataPos, TSDB_CLUSTER_RESERVE_SIZE, CLUSTER_DECODE_OVER) + SDB_GET_INT64(pRaw, dataPos, &pCluster->id, _OVER) + SDB_GET_INT64(pRaw, dataPos, &pCluster->createdTime, _OVER) + SDB_GET_INT64(pRaw, dataPos, &pCluster->updateTime, _OVER) + SDB_GET_BINARY(pRaw, dataPos, pCluster->name, TSDB_CLUSTER_ID_LEN, _OVER) + SDB_GET_RESERVE(pRaw, dataPos, CLUSTER_RESERVE_SIZE, _OVER) terrno = 0; -CLUSTER_DECODE_OVER: +_OVER: if (terrno != 0) { mError("cluster:%" PRId64 ", failed to decode from raw:%p since %s", pCluster->id, pRaw, terrstr()); taosMemoryFreeClear(pRow); @@ -161,7 +163,7 @@ static int32_t mndCreateDefaultCluster(SMnode *pMnode) { int32_t code = taosGetSystemUUID(clusterObj.name, TSDB_CLUSTER_ID_LEN); if (code != 0) { - strcpy(clusterObj.name, "tdengine2.0"); + strcpy(clusterObj.name, "tdengine3.0"); mError("failed to get name from system, set to default val %s", clusterObj.name); } @@ -190,8 +192,8 @@ static int32_t mndRetrieveClusters(SNodeMsg *pMsg, SShowObj *pShow, SSDataBlock if (pShow->pIter == NULL) break; cols = 0; - SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, cols++); - colDataAppend(pColInfo, numOfRows, (const char*) &pCluster->id, false); + SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++); + colDataAppend(pColInfo, numOfRows, (const char *)&pCluster->id, false); char buf[tListLen(pCluster->name) + VARSTR_HEADER_SIZE] = {0}; STR_WITH_MAXSIZE_TO_VARSTR(buf, pCluster->name, pShow->pMeta->pSchemas[cols].bytes); @@ -200,7 +202,7 @@ static int32_t mndRetrieveClusters(SNodeMsg *pMsg, SShowObj *pShow, SSDataBlock colDataAppend(pColInfo, numOfRows, buf, false); pColInfo = taosArrayGet(pBlock->pDataBlock, cols++); - colDataAppend(pColInfo, numOfRows, (const char*) &pCluster->createdTime, false); + colDataAppend(pColInfo, numOfRows, (const char *)&pCluster->createdTime, false); sdbRelease(pSdb, pCluster); numOfRows++; From f018ccc28e91ac619b8fdf0267f1a56114c1781d Mon Sep 17 00:00:00 2001 From: Shengliang Guan Date: Sat, 7 May 2022 09:36:36 +0800 Subject: [PATCH 16/27] refactor: adjust mnode def --- source/dnode/mnode/impl/inc/mndDef.h | 84 ++++++++++++---------------- source/dnode/mnode/impl/src/mndDef.c | 15 +++++ 2 files changed, 51 insertions(+), 48 deletions(-) diff --git a/source/dnode/mnode/impl/inc/mndDef.h b/source/dnode/mnode/impl/inc/mndDef.h index 89638524ae..c113cd0bbe 100644 --- a/source/dnode/mnode/impl/inc/mndDef.h +++ b/source/dnode/mnode/impl/inc/mndDef.h @@ -72,7 +72,7 @@ typedef enum { TRN_TYPE_DROP_USER = 1003, TRN_TYPE_CREATE_FUNC = 1004, TRN_TYPE_DROP_FUNC = 1005, - + TRN_TYPE_CREATE_SNODE = 1006, TRN_TYPE_DROP_SNODE = 1007, TRN_TYPE_CREATE_QNODE = 1008, @@ -115,7 +115,10 @@ typedef enum { TRN_TYPE_STB_SCOPE_END, } ETrnType; -typedef enum { TRN_POLICY_ROLLBACK = 0, TRN_POLICY_RETRY = 1 } ETrnPolicy; +typedef enum { + TRN_POLICY_ROLLBACK = 0, + TRN_POLICY_RETRY = 1, +} ETrnPolicy; typedef enum { DND_REASON_ONLINE = 0, @@ -131,6 +134,15 @@ typedef enum { DND_REASON_OTHERS } EDndReason; +typedef enum { + CONSUMER_UPDATE__TOUCH = 1, + CONSUMER_UPDATE__ADD, + CONSUMER_UPDATE__REMOVE, + CONSUMER_UPDATE__LOST, + CONSUMER_UPDATE__RECOVER, + CONSUMER_UPDATE__MODIFY, +} ECsmUpdateType; + typedef struct { int32_t id; ETrnStage stage; @@ -386,7 +398,6 @@ typedef struct { int32_t codeSize; char* pComment; char* pCode; - char pData[]; } SFuncObj; typedef struct { @@ -425,18 +436,8 @@ typedef struct { int64_t offset; } SMqOffsetObj; -static FORCE_INLINE int32_t tEncodeSMqOffsetObj(void** buf, const SMqOffsetObj* pOffset) { - int32_t tlen = 0; - tlen += taosEncodeString(buf, pOffset->key); - tlen += taosEncodeFixedI64(buf, pOffset->offset); - return tlen; -} - -static FORCE_INLINE void* tDecodeSMqOffsetObj(void* buf, SMqOffsetObj* pOffset) { - buf = taosDecodeStringTo(buf, pOffset->key); - buf = taosDecodeFixedI64(buf, &pOffset->offset); - return buf; -} +int32_t tEncodeSMqOffsetObj(void** buf, const SMqOffsetObj* pOffset); +void* tDecodeSMqOffsetObj(void* buf, SMqOffsetObj* pOffset); typedef struct { char name[TSDB_TOPIC_FNAME_LEN]; @@ -459,26 +460,15 @@ typedef struct { SSchemaWrapper schema; } SMqTopicObj; -enum { - CONSUMER_UPDATE__TOUCH = 1, - CONSUMER_UPDATE__ADD, - CONSUMER_UPDATE__REMOVE, - CONSUMER_UPDATE__LOST, - CONSUMER_UPDATE__RECOVER, - CONSUMER_UPDATE__MODIFY, -}; - typedef struct { - int64_t consumerId; - char cgroup[TSDB_CGROUP_LEN]; - char appId[TSDB_CGROUP_LEN]; - int8_t updateType; // used only for update - int32_t epoch; - int32_t status; - // hbStatus is not applicable to serialization - int32_t hbStatus; - // lock is used for topics update - SRWLatch lock; + int64_t consumerId; + char cgroup[TSDB_CGROUP_LEN]; + char appId[TSDB_CGROUP_LEN]; + int8_t updateType; // used only for update + int32_t epoch; + int32_t status; + int32_t hbStatus; // hbStatus is not applicable to serialization + SRWLatch lock; // lock is used for topics update SArray* currentTopics; // SArray SArray* rebNewTopics; // SArray SArray* rebRemovedTopics; // SArray @@ -492,7 +482,6 @@ typedef struct { int64_t upTime; int64_t subscribeTime; int64_t rebalanceTime; - } SMqConsumerObj; SMqConsumerObj* tNewSMqConsumerObj(int64_t consumerId, char cgroup[TSDB_CGROUP_LEN]); @@ -581,19 +570,18 @@ typedef struct { } SMqRebOutputObj; typedef struct { - char name[TSDB_TOPIC_FNAME_LEN]; - char sourceDb[TSDB_DB_FNAME_LEN]; - char targetDb[TSDB_DB_FNAME_LEN]; - char targetSTbName[TSDB_TABLE_FNAME_LEN]; - int64_t createTime; - int64_t updateTime; - int64_t uid; - int64_t dbUid; - int32_t version; - int32_t vgNum; - SRWLatch lock; - int8_t status; - // int32_t sqlLen; + char name[TSDB_TOPIC_FNAME_LEN]; + char sourceDb[TSDB_DB_FNAME_LEN]; + char targetDb[TSDB_DB_FNAME_LEN]; + char targetSTbName[TSDB_TABLE_FNAME_LEN]; + int64_t createTime; + int64_t updateTime; + int64_t uid; + int64_t dbUid; + int32_t version; + int32_t vgNum; + SRWLatch lock; + int8_t status; int8_t createdBy; // STREAM_CREATED_BY__USER or SMA int32_t fixedSinkVgId; // 0 for shuffle int64_t smaId; // 0 for unused diff --git a/source/dnode/mnode/impl/src/mndDef.c b/source/dnode/mnode/impl/src/mndDef.c index 5bce6ac218..b8c120905c 100644 --- a/source/dnode/mnode/impl/src/mndDef.c +++ b/source/dnode/mnode/impl/src/mndDef.c @@ -13,12 +13,14 @@ * along with this program. If not, see . */ +#define _DEFAULT_SOURCE #include "mndDef.h" #include "mndConsumer.h" SMqConsumerObj *tNewSMqConsumerObj(int64_t consumerId, char cgroup[TSDB_CGROUP_LEN]) { SMqConsumerObj *pConsumer = taosMemoryCalloc(1, sizeof(SMqConsumerObj)); if (pConsumer == NULL) { + terrno = TSDB_CODE_OUT_OF_MEMORY; return NULL; } @@ -515,3 +517,16 @@ int32_t tDecodeSStreamObj(SCoder *pDecoder, SStreamObj *pObj) { #endif return 0; } + +int32_t tEncodeSMqOffsetObj(void **buf, const SMqOffsetObj *pOffset) { + int32_t tlen = 0; + tlen += taosEncodeString(buf, pOffset->key); + tlen += taosEncodeFixedI64(buf, pOffset->offset); + return tlen; +} + +void *tDecodeSMqOffsetObj(void *buf, SMqOffsetObj *pOffset) { + buf = taosDecodeStringTo(buf, pOffset->key); + buf = taosDecodeFixedI64(buf, &pOffset->offset); + return buf; +} \ No newline at end of file From 59fe535581a86f35f647d4f6f180e1688196def3 Mon Sep 17 00:00:00 2001 From: jiacy-jcy <714897623@qq.com> Date: Sat, 7 May 2022 10:19:26 +0800 Subject: [PATCH 17/27] update --- tests/system-test/fulltest.sh | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/system-test/fulltest.sh b/tests/system-test/fulltest.sh index 884d18f7cd..4149c2cd9d 100755 --- a/tests/system-test/fulltest.sh +++ b/tests/system-test/fulltest.sh @@ -11,12 +11,17 @@ python3 ./test.py -f 0-others/taosShellNetChk.py python3 ./test.py -f 2-query/distinct.py python3 ./test.py -f 2-query/varchar.py -#python3 ./test.py -f 2-query/timezone.py +python3 ./test.py -f 2-query/timezone.py python3 ./test.py -f 2-query/Now.py python3 ./test.py -f 2-query/Today.py python3 ./test.py -f 2-query/max.py python3 ./test.py -f 2-query/min.py python3 ./test.py -f 2-query/count.py +python3 ./test.py -f 2-query/last.py +#python3 ./test.py -f 2-query/To_iso8601.py +python3 ./test.py -f 2-query/To_unixtimestamp.py +python3 ./test.py -f 2-query/timetruncate.py + # python3 ./test.py -f 2-query/Timediff.py #python3 ./test.py -f 2-query/cast.py From bb3681cd713577f475746fc49d28c828501d9e33 Mon Sep 17 00:00:00 2001 From: Ganlin Zhao Date: Fri, 6 May 2022 21:44:25 +0800 Subject: [PATCH 18/27] feat(query): add histogram param parsing --- source/libs/function/src/builtinsimpl.c | 52 +++++++++++++++++++++++-- 1 file changed, 49 insertions(+), 3 deletions(-) diff --git a/source/libs/function/src/builtinsimpl.c b/source/libs/function/src/builtinsimpl.c index 64cca47da0..69d3e079c6 100644 --- a/source/libs/function/src/builtinsimpl.c +++ b/source/libs/function/src/builtinsimpl.c @@ -1973,13 +1973,59 @@ bool histogramFunctionSetup(SqlFunctionCtx *pCtx, SResultRowEntryInfo *pResultIn } int32_t histogramFunction(SqlFunctionCtx *pCtx) { + int32_t numOfElems = 0; + SHistoFuncInfo* pInfo = GET_ROWCELL_INTERBUF(GET_RES_INFO(pCtx)); + + SInputColumnInfoData* pInput = &pCtx->input; + SColumnInfoData* pCol = pInput->pData[0]; + + int32_t type = pInput->pData[0]->info.type; + + int32_t start = pInput->startRowIndex; + int32_t numOfRows = pInput->numOfRows; + + for (int32_t i = start; i < numOfRows + start; ++i) { + if (pCol->hasNull && colDataIsNull_f(pCol->nullbitmap, i)) { + continue; + } + + char* data = colDataGetData(pCol, i); + double v; + GET_TYPED_DATA(v, double, type, data); + + for (int32_t k = 0; k < pInfo->numOfBins; ++k) { + if (v > pInfo->bins[k].lower && v <= pInfo->bins[k].upper) { + pInfo->bins[k].count++; + numOfElems++; + break; + } + } + } + return TSDB_CODE_SUCCESS; } int32_t histogramFinalize(SqlFunctionCtx* pCtx, SSDataBlock* pBlock) { SHistoFuncInfo* pInfo = GET_ROWCELL_INTERBUF(GET_RES_INFO(pCtx)); - //if (pInfo->hasResult == true) { - // SET_DOUBLE_VAL(&pInfo->result, pInfo->max - pInfo->min); - //} + int32_t slotId = pCtx->pExpr->base.resSchema.slotId; + SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, slotId); + + int32_t currentRow = pBlock->info.rows; + + for (int32_t i = 0; i < pInfo->numOfBins; ++i) { + int32_t len; + char buf[400] = {0}; + if (!pInfo->normalized) { + len = sprintf(buf + VARSTR_HEADER_SIZE, "{\"lower_bin\":%g, \"upper_bin\":%g, \"count\":%"PRId64"}", + pInfo->bins[i].lower, pInfo->bins[i].upper, pInfo->bins[i].count); + } else { + len = sprintf(buf + VARSTR_HEADER_SIZE, "{\"lower_bin\":%g, \"upper_bin\":%g, \"count\":%lf}", + pInfo->bins[i].lower, pInfo->bins[i].upper, pInfo->bins[i].percentage); + } + varDataSetLen(buf, len); + colDataAppend(pCol, currentRow, buf, false); + currentRow++; + } + return functionFinalize(pCtx, pBlock); } From 6115d3dbe89e5f0c1f76ee44c0f743ddb9b0d5d7 Mon Sep 17 00:00:00 2001 From: Haojun Liao Date: Sat, 7 May 2022 11:04:38 +0800 Subject: [PATCH 19/27] fix(query): update the column match strategy. --- source/libs/executor/inc/executorimpl.h | 5 ++ source/libs/executor/src/executorimpl.c | 63 ++++++++++-------------- source/libs/executor/src/groupoperator.c | 1 + source/libs/executor/src/sortoperator.c | 14 ++---- 4 files changed, 37 insertions(+), 46 deletions(-) diff --git a/source/libs/executor/inc/executorimpl.h b/source/libs/executor/inc/executorimpl.h index 3b5d0c209f..9eed76a0fe 100644 --- a/source/libs/executor/inc/executorimpl.h +++ b/source/libs/executor/inc/executorimpl.h @@ -325,10 +325,15 @@ typedef struct SExchangeInfo { SLoadRemoteDataInfo loadInfo; } SExchangeInfo; +#define COL_MATCH_FROM_COL_ID 0x1 +#define COL_MATCH_FROM_SLOT_ID 0x2 + typedef struct SColMatchInfo { + int32_t srcSlotId; // source slot id int32_t colId; int32_t targetSlotId; bool output; + int32_t matchType; // determinate the source according to col id or slot id } SColMatchInfo; typedef struct SScanInfo { diff --git a/source/libs/executor/src/executorimpl.c b/source/libs/executor/src/executorimpl.c index 943d4b2783..46c30c43c7 100644 --- a/source/libs/executor/src/executorimpl.c +++ b/source/libs/executor/src/executorimpl.c @@ -4699,10 +4699,9 @@ static int32_t doCreateTableGroup(void* metaHandle, int32_t tableType, uint64_t uint64_t queryId, uint64_t taskId); static SArray* extractTableIdList(const STableGroupInfo* pTableGroupInfo); static SArray* extractColumnInfo(SNodeList* pNodeList); -static SArray* extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols); +static SArray* extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols, int32_t type); static SArray* createSortInfo(SNodeList* pNodeList); -static SArray* createIndexMap(SNodeList* pNodeList); static SArray* extractPartitionColInfo(SNodeList* pNodeList); static int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, const STableScanPhysiNode* pTableScanNode); static void setJoinColumnInfo(SColumnInfo* pInfo, const SColumnNode* pLeftNode); @@ -4734,9 +4733,10 @@ SOperatorInfo* createOperatorTree(SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo return NULL; } - SArray* pColList = - extractColMatchInfo(pScanPhyNode->pScanCols, pScanPhyNode->node.pOutputDataBlockDesc, &numOfCols); - SSDataBlock* pResBlock = createResDataBlock(pScanPhyNode->node.pOutputDataBlockDesc); + SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc; + + SArray* pColList = extractColMatchInfo(pScanPhyNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID); + SSDataBlock* pResBlock = createResDataBlock(pDescNode); SQueryTableDataCond cond = {0}; int32_t code = initQueryTableDataCond(&cond, pTableScanNode); @@ -4761,10 +4761,11 @@ SOperatorInfo* createOperatorTree(SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo int32_t code = doCreateTableGroup(pHandle->meta, pScanPhyNode->tableType, pScanPhyNode->uid, pTableGroupInfo, queryId, taskId); SArray* tableIdList = extractTableIdList(pTableGroupInfo); - SSDataBlock* pResBlock = createResDataBlock(pScanPhyNode->node.pOutputDataBlockDesc); + SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc; + SSDataBlock* pResBlock = createResDataBlock(pDescNode); int32_t numOfCols = 0; - SArray* pCols = extractColMatchInfo(pScanPhyNode->pScanCols, pScanPhyNode->node.pOutputDataBlockDesc, &numOfCols); + SArray* pCols = extractColMatchInfo(pScanPhyNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID); SOperatorInfo* pOperator = createStreamScanOperatorInfo(pHandle->reader, pResBlock, pCols, tableIdList, pTaskInfo, pScanPhyNode->node.pConditions); taosArrayDestroy(tableIdList); @@ -4773,18 +4774,22 @@ SOperatorInfo* createOperatorTree(SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo SSystemTableScanPhysiNode* pSysScanPhyNode = (SSystemTableScanPhysiNode*)pPhyNode; SScanPhysiNode* pScanNode = &pSysScanPhyNode->scan; - SSDataBlock* pResBlock = createResDataBlock(pScanNode->node.pOutputDataBlockDesc); + SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc; + + SSDataBlock* pResBlock = createResDataBlock(pDescNode); int32_t numOfOutputCols = 0; - SArray* colList = - extractColMatchInfo(pScanNode->pScanCols, pScanNode->node.pOutputDataBlockDesc, &numOfOutputCols); + SArray* colList = extractColMatchInfo(pScanNode->pScanCols, pDescNode, &numOfOutputCols, COL_MATCH_FROM_COL_ID); SOperatorInfo* pOperator = createSysTableScanOperatorInfo( pHandle, pResBlock, &pScanNode->tableName, pScanNode->node.pConditions, pSysScanPhyNode->mgmtEpSet, colList, pTaskInfo, pSysScanPhyNode->showRewrite, pSysScanPhyNode->accountId); return pOperator; } else if (QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN == type) { STagScanPhysiNode* pScanPhyNode = (STagScanPhysiNode*) pPhyNode; - SSDataBlock* pResBlock = createResDataBlock(pScanPhyNode->node.pOutputDataBlockDesc); + + SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc; + + SSDataBlock* pResBlock = createResDataBlock(pDescNode); int32_t code = doCreateTableGroup(pHandle->meta, pScanPhyNode->tableType, pScanPhyNode->uid, pTableGroupInfo, queryId, taskId); @@ -4796,8 +4801,7 @@ SOperatorInfo* createOperatorTree(SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo SExprInfo* pExprInfo = createExprInfo(pScanPhyNode->pScanPseudoCols, NULL, &num); int32_t numOfOutputCols = 0; - SArray* colList = - extractColMatchInfo(pScanPhyNode->pScanPseudoCols, pScanPhyNode->node.pOutputDataBlockDesc, &numOfOutputCols); + SArray* colList = extractColMatchInfo(pScanPhyNode->pScanPseudoCols, pDescNode, &numOfOutputCols, COL_MATCH_FROM_COL_ID); SOperatorInfo* pOperator = createTagScanOperatorInfo(pHandle, pExprInfo, num, pResBlock, colList, pTableGroupInfo, pTaskInfo); return pOperator; @@ -4869,15 +4873,16 @@ SOperatorInfo* createOperatorTree(SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo } else if (QUERY_NODE_PHYSICAL_PLAN_SORT == type) { SSortPhysiNode* pSortPhyNode = (SSortPhysiNode*)pPhyNode; - SSDataBlock* pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc); + SDataBlockDescNode* pDescNode = pPhyNode->pOutputDataBlockDesc; + + SSDataBlock* pResBlock = createResDataBlock(pDescNode); SArray* info = createSortInfo(pSortPhyNode->pSortKeys); int32_t numOfCols = 0; SExprInfo* pExprInfo = createExprInfo(pSortPhyNode->pExprs, NULL, &numOfCols); int32_t numOfOutputCols = 0; - SArray* pColList = - extractColMatchInfo(pSortPhyNode->pTargets, pSortPhyNode->node.pOutputDataBlockDesc, &numOfOutputCols); + SArray* pColList = extractColMatchInfo(pSortPhyNode->pTargets, pDescNode, &numOfOutputCols, COL_MATCH_FROM_SLOT_ID); pOptr = createSortOperatorInfo(ops[0], pResBlock, info, pExprInfo, numOfCols, pColList, pTaskInfo); } else if (QUERY_NODE_PHYSICAL_PLAN_SESSION_WINDOW == type) { @@ -5059,25 +5064,7 @@ SArray* createSortInfo(SNodeList* pNodeList) { return pList; } -SArray* createIndexMap(SNodeList* pNodeList) { - size_t numOfCols = LIST_LENGTH(pNodeList); - SArray* pList = taosArrayInit(numOfCols, sizeof(int32_t)); - if (pList == NULL) { - terrno = TSDB_CODE_OUT_OF_MEMORY; - return pList; - } - - for (int32_t i = 0; i < numOfCols; ++i) { - STargetNode* pTarget = (STargetNode*)nodesListGetNode(pNodeList, i); - - SColumnNode* pColNode = (SColumnNode*)pTarget->pExpr; - taosArrayPush(pList, &pColNode->slotId); - } - - return pList; -} - -SArray* extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols) { +SArray* extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols, int32_t type) { size_t numOfCols = LIST_LENGTH(pNodeList); SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchInfo)); if (pList == NULL) { @@ -5090,8 +5077,10 @@ SArray* extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNod SColumnNode* pColNode = (SColumnNode*)pNode->pExpr; SColMatchInfo c = {0}; - c.output = true; - c.colId = pColNode->colId; + c.output = true; + c.colId = pColNode->colId; + c.srcSlotId = pColNode->slotId; + c.matchType = type; c.targetSlotId = pNode->slotId; taosArrayPush(pList, &c); } diff --git a/source/libs/executor/src/groupoperator.c b/source/libs/executor/src/groupoperator.c index 58efd75a0b..183cb9dbe6 100644 --- a/source/libs/executor/src/groupoperator.c +++ b/source/libs/executor/src/groupoperator.c @@ -129,6 +129,7 @@ static void recordNewGroupKeys(SArray* pGroupCols, SArray* pGroupColVals, SSData char* val = colDataGetData(pColInfoData, rowIndex); if (IS_VAR_DATA_TYPE(pkey->type)) { memcpy(pkey->pData, val, varDataTLen(val)); + ASSERT(varDataTLen(val) <= pkey->bytes); } else { memcpy(pkey->pData, val, pkey->bytes); } diff --git a/source/libs/executor/src/sortoperator.c b/source/libs/executor/src/sortoperator.c index b34889ec46..a654876513 100644 --- a/source/libs/executor/src/sortoperator.c +++ b/source/libs/executor/src/sortoperator.c @@ -87,17 +87,13 @@ SSDataBlock* getSortedBlockData(SSortHandle* pHandle, SSDataBlock* pDataBlock, i if (p->info.rows > 0) { int32_t numOfCols = taosArrayGetSize(pColMatchInfo); - for(int32_t i = 0; i < numOfCols; ++i) { + for (int32_t i = 0; i < numOfCols; ++i) { SColMatchInfo* pmInfo = taosArrayGet(pColMatchInfo, i); + ASSERT(pmInfo->matchType == COL_MATCH_FROM_SLOT_ID); - for(int32_t j = 0; j < p->info.numOfCols; ++j) { - SColumnInfoData* pSrc = taosArrayGet(p->pDataBlock, j); - if (pSrc->info.colId == pmInfo->colId) { - SColumnInfoData* pDst = taosArrayGet(pDataBlock->pDataBlock, pmInfo->targetSlotId); - colDataAssign(pDst, pSrc, p->info.rows); - break; - } - } + SColumnInfoData* pSrc = taosArrayGet(p->pDataBlock, pmInfo->srcSlotId); + SColumnInfoData* pDst = taosArrayGet(pDataBlock->pDataBlock, pmInfo->targetSlotId); + colDataAssign(pDst, pSrc, p->info.rows); } pDataBlock->info.rows = p->info.rows; From 660b610f0411cb6e817aca4680c16578e791ebe6 Mon Sep 17 00:00:00 2001 From: Ganlin Zhao Date: Sat, 7 May 2022 11:35:03 +0800 Subject: [PATCH 20/27] fix(query): fix histogram function output --- source/libs/function/src/builtins.c | 2 +- source/libs/function/src/builtinsimpl.c | 20 +++++++++++--------- 2 files changed, 12 insertions(+), 10 deletions(-) diff --git a/source/libs/function/src/builtins.c b/source/libs/function/src/builtins.c index e0d9fb60f5..cb58a7a1d0 100644 --- a/source/libs/function/src/builtins.c +++ b/source/libs/function/src/builtins.c @@ -241,7 +241,7 @@ static int32_t translateHistogram(SFunctionNode* pFunc, char* pErrBuf, int32_t l return invaildFuncParaTypeErrMsg(pErrBuf, len, pFunc->functionName); } - pFunc->node.resType = (SDataType) { .bytes = tDataTypes[TSDB_DATA_TYPE_DOUBLE].bytes, .type = TSDB_DATA_TYPE_DOUBLE }; + pFunc->node.resType = (SDataType) { .bytes = 512, .type = TSDB_DATA_TYPE_BINARY }; return TSDB_CODE_SUCCESS; } diff --git a/source/libs/function/src/builtinsimpl.c b/source/libs/function/src/builtinsimpl.c index 69d3e079c6..909659701c 100644 --- a/source/libs/function/src/builtinsimpl.c +++ b/source/libs/function/src/builtinsimpl.c @@ -1826,7 +1826,6 @@ static int8_t getHistogramBinType(char *binTypeStr) { static bool getHistogramBinDesc(SHistoFuncInfo *pInfo, char *binDescStr, int8_t binType, bool normalized) { cJSON* binDesc = cJSON_Parse(binDescStr); - int32_t counter; int32_t numOfBins; double* intervals; if (cJSON_IsObject(binDesc)) { /* linaer/log bins */ @@ -1855,7 +1854,7 @@ static bool getHistogramBinDesc(SHistoFuncInfo *pInfo, char *binDescStr, int8_t return false; } - counter = (int32_t)count->valueint; + int32_t counter = (int32_t)count->valueint; if (infinity->valueint == false) { startIndex = 0; numOfBins = counter + 1; @@ -1915,7 +1914,7 @@ static bool getHistogramBinDesc(SHistoFuncInfo *pInfo, char *binDescStr, int8_t if (binType != USER_INPUT_BIN) { return false; } - counter = numOfBins = cJSON_GetArraySize(binDesc); + numOfBins = cJSON_GetArraySize(binDesc); intervals = taosMemoryCalloc(numOfBins, sizeof(double)); cJSON* bin = binDesc->child; if (bin == NULL) { @@ -1940,9 +1939,9 @@ static bool getHistogramBinDesc(SHistoFuncInfo *pInfo, char *binDescStr, int8_t return false; } - pInfo->numOfBins = numOfBins; + pInfo->numOfBins = numOfBins - 1; pInfo->normalized = normalized; - for (int32_t i = 0; i < numOfBins; ++i) { + for (int32_t i = 0; i < pInfo->numOfBins; ++i) { pInfo->bins[i].lower = intervals[i] < intervals[i + 1] ? intervals[i] : intervals[i + 1]; pInfo->bins[i].upper = intervals[i + 1] > intervals[i] ? intervals[i + 1] : intervals[i]; pInfo->bins[i].count = 0; @@ -1989,6 +1988,8 @@ int32_t histogramFunction(SqlFunctionCtx *pCtx) { continue; } + numOfElems++; + char* data = colDataGetData(pCol, i); double v; GET_TYPED_DATA(v, double, type, data); @@ -1996,25 +1997,26 @@ int32_t histogramFunction(SqlFunctionCtx *pCtx) { for (int32_t k = 0; k < pInfo->numOfBins; ++k) { if (v > pInfo->bins[k].lower && v <= pInfo->bins[k].upper) { pInfo->bins[k].count++; - numOfElems++; break; } } } + SET_VAL(GET_RES_INFO(pCtx), numOfElems, pInfo->numOfBins); return TSDB_CODE_SUCCESS; } int32_t histogramFinalize(SqlFunctionCtx* pCtx, SSDataBlock* pBlock) { + SResultRowEntryInfo* pResInfo = GET_RES_INFO(pCtx); SHistoFuncInfo* pInfo = GET_ROWCELL_INTERBUF(GET_RES_INFO(pCtx)); int32_t slotId = pCtx->pExpr->base.resSchema.slotId; SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, slotId); int32_t currentRow = pBlock->info.rows; - for (int32_t i = 0; i < pInfo->numOfBins; ++i) { + for (int32_t i = 0; i < pResInfo->numOfRes; ++i) { int32_t len; - char buf[400] = {0}; + char buf[512] = {0}; if (!pInfo->normalized) { len = sprintf(buf + VARSTR_HEADER_SIZE, "{\"lower_bin\":%g, \"upper_bin\":%g, \"count\":%"PRId64"}", pInfo->bins[i].lower, pInfo->bins[i].upper, pInfo->bins[i].count); @@ -2027,5 +2029,5 @@ int32_t histogramFinalize(SqlFunctionCtx* pCtx, SSDataBlock* pBlock) { currentRow++; } - return functionFinalize(pCtx, pBlock); + return pResInfo->numOfRes; } From cdb7c37d16a0ac95a7a92d12816d8568081d755d Mon Sep 17 00:00:00 2001 From: Shengliang Guan Date: Sat, 7 May 2022 12:07:45 +0800 Subject: [PATCH 21/27] refactor: adjust stb codes --- include/util/taoserror.h | 49 ++++++++++++----------- source/dnode/mnode/impl/inc/mndUser.h | 3 +- source/dnode/mnode/impl/src/mndStb.c | 44 ++++++++++++-------- source/dnode/mnode/impl/src/mndStream.c | 5 ++- source/util/src/terror.c | 3 ++ tests/script/general/stable/testSuite.sim | 5 --- 6 files changed, 63 insertions(+), 46 deletions(-) delete mode 100644 tests/script/general/stable/testSuite.sim diff --git a/include/util/taoserror.h b/include/util/taoserror.h index 441e87eff7..53781c0f0a 100644 --- a/include/util/taoserror.h +++ b/include/util/taoserror.h @@ -327,29 +327,32 @@ int32_t* taosGetErrno(); #define TSDB_CODE_TDB_INVALID_TABLE_TYPE TAOS_DEF_ERROR_CODE(0, 0x0601) #define TSDB_CODE_TDB_IVD_TB_SCHEMA_VERSION TAOS_DEF_ERROR_CODE(0, 0x0602) #define TSDB_CODE_TDB_TABLE_ALREADY_EXIST TAOS_DEF_ERROR_CODE(0, 0x0603) -#define TSDB_CODE_TDB_INVALID_CONFIG TAOS_DEF_ERROR_CODE(0, 0x0604) -#define TSDB_CODE_TDB_INIT_FAILED TAOS_DEF_ERROR_CODE(0, 0x0605) -#define TSDB_CODE_TDB_NO_DISKSPACE TAOS_DEF_ERROR_CODE(0, 0x0606) -#define TSDB_CODE_TDB_NO_DISK_PERMISSIONS TAOS_DEF_ERROR_CODE(0, 0x0607) -#define TSDB_CODE_TDB_FILE_CORRUPTED TAOS_DEF_ERROR_CODE(0, 0x0608) -#define TSDB_CODE_TDB_OUT_OF_MEMORY TAOS_DEF_ERROR_CODE(0, 0x0609) -#define TSDB_CODE_TDB_TAG_VER_OUT_OF_DATE TAOS_DEF_ERROR_CODE(0, 0x060A) -#define TSDB_CODE_TDB_TIMESTAMP_OUT_OF_RANGE TAOS_DEF_ERROR_CODE(0, 0x060B) -#define TSDB_CODE_TDB_SUBMIT_MSG_MSSED_UP TAOS_DEF_ERROR_CODE(0, 0x060C) -#define TSDB_CODE_TDB_INVALID_ACTION TAOS_DEF_ERROR_CODE(0, 0x060D) -#define TSDB_CODE_TDB_INVALID_CREATE_TB_MSG TAOS_DEF_ERROR_CODE(0, 0x060E) -#define TSDB_CODE_TDB_NO_TABLE_DATA_IN_MEM TAOS_DEF_ERROR_CODE(0, 0x060F) -#define TSDB_CODE_TDB_FILE_ALREADY_EXISTS TAOS_DEF_ERROR_CODE(0, 0x0610) -#define TSDB_CODE_TDB_TABLE_RECONFIGURE TAOS_DEF_ERROR_CODE(0, 0x0611) -#define TSDB_CODE_TDB_IVD_CREATE_TABLE_INFO TAOS_DEF_ERROR_CODE(0, 0x0612) -#define TSDB_CODE_TDB_NO_AVAIL_DISK TAOS_DEF_ERROR_CODE(0, 0x0613) -#define TSDB_CODE_TDB_MESSED_MSG TAOS_DEF_ERROR_CODE(0, 0x0614) -#define TSDB_CODE_TDB_IVLD_TAG_VAL TAOS_DEF_ERROR_CODE(0, 0x0615) -#define TSDB_CODE_TDB_NO_CACHE_LAST_ROW TAOS_DEF_ERROR_CODE(0, 0x0616) -#define TSDB_CODE_TDB_TABLE_RECREATED TAOS_DEF_ERROR_CODE(0, 0x0617) -#define TSDB_CODE_TDB_TDB_ENV_OPEN_ERROR TAOS_DEF_ERROR_CODE(0, 0x0618) -#define TSDB_CODE_TDB_NO_SMA_INDEX_IN_META TAOS_DEF_ERROR_CODE(0, 0x0619) -#define TSDB_CODE_TDB_INVALID_SMA_STAT TAOS_DEF_ERROR_CODE(0, 0x0620) +#define TSDB_CODE_TDB_TABLE_NOT_EXIST TAOS_DEF_ERROR_CODE(0, 0x0604) +#define TSDB_CODE_TDB_STB_ALREADY_EXIST TAOS_DEF_ERROR_CODE(0, 0x0605) +#define TSDB_CODE_TDB_STB_NOT_EXIST TAOS_DEF_ERROR_CODE(0, 0x0606) +#define TSDB_CODE_TDB_INVALID_CONFIG TAOS_DEF_ERROR_CODE(0, 0x0607) +#define TSDB_CODE_TDB_INIT_FAILED TAOS_DEF_ERROR_CODE(0, 0x0608) +#define TSDB_CODE_TDB_NO_DISKSPACE TAOS_DEF_ERROR_CODE(0, 0x0609) +#define TSDB_CODE_TDB_NO_DISK_PERMISSIONS TAOS_DEF_ERROR_CODE(0, 0x060A) +#define TSDB_CODE_TDB_FILE_CORRUPTED TAOS_DEF_ERROR_CODE(0, 0x060B) +#define TSDB_CODE_TDB_OUT_OF_MEMORY TAOS_DEF_ERROR_CODE(0, 0x060C) +#define TSDB_CODE_TDB_TAG_VER_OUT_OF_DATE TAOS_DEF_ERROR_CODE(0, 0x060D) +#define TSDB_CODE_TDB_TIMESTAMP_OUT_OF_RANGE TAOS_DEF_ERROR_CODE(0, 0x060E) +#define TSDB_CODE_TDB_SUBMIT_MSG_MSSED_UP TAOS_DEF_ERROR_CODE(0, 0x060F) +#define TSDB_CODE_TDB_INVALID_ACTION TAOS_DEF_ERROR_CODE(0, 0x0600) +#define TSDB_CODE_TDB_INVALID_CREATE_TB_MSG TAOS_DEF_ERROR_CODE(0, 0x0601) +#define TSDB_CODE_TDB_NO_TABLE_DATA_IN_MEM TAOS_DEF_ERROR_CODE(0, 0x0602) +#define TSDB_CODE_TDB_FILE_ALREADY_EXISTS TAOS_DEF_ERROR_CODE(0, 0x0613) +#define TSDB_CODE_TDB_TABLE_RECONFIGURE TAOS_DEF_ERROR_CODE(0, 0x0614) +#define TSDB_CODE_TDB_IVD_CREATE_TABLE_INFO TAOS_DEF_ERROR_CODE(0, 0x0615) +#define TSDB_CODE_TDB_NO_AVAIL_DISK TAOS_DEF_ERROR_CODE(0, 0x0616) +#define TSDB_CODE_TDB_MESSED_MSG TAOS_DEF_ERROR_CODE(0, 0x0617) +#define TSDB_CODE_TDB_IVLD_TAG_VAL TAOS_DEF_ERROR_CODE(0, 0x0618) +#define TSDB_CODE_TDB_NO_CACHE_LAST_ROW TAOS_DEF_ERROR_CODE(0, 0x0619) +#define TSDB_CODE_TDB_TABLE_RECREATED TAOS_DEF_ERROR_CODE(0, 0x061A) +#define TSDB_CODE_TDB_TDB_ENV_OPEN_ERROR TAOS_DEF_ERROR_CODE(0, 0x061B) +#define TSDB_CODE_TDB_NO_SMA_INDEX_IN_META TAOS_DEF_ERROR_CODE(0, 0x061C) +#define TSDB_CODE_TDB_INVALID_SMA_STAT TAOS_DEF_ERROR_CODE(0, 0x062D) // query #define TSDB_CODE_QRY_INVALID_QHANDLE TAOS_DEF_ERROR_CODE(0, 0x0700) diff --git a/source/dnode/mnode/impl/inc/mndUser.h b/source/dnode/mnode/impl/inc/mndUser.h index 2140d0fa67..016ec3e6e9 100644 --- a/source/dnode/mnode/impl/inc/mndUser.h +++ b/source/dnode/mnode/impl/inc/mndUser.h @@ -29,7 +29,8 @@ void mndReleaseUser(SMnode *pMnode, SUserObj *pUser); // for trans test SSdbRaw *mndUserActionEncode(SUserObj *pUser); -int32_t mndValidateUserAuthInfo(SMnode *pMnode, SUserAuthVersion *pUsers, int32_t numOfUses, void **ppRsp, int32_t *pRspLen); +int32_t mndValidateUserAuthInfo(SMnode *pMnode, SUserAuthVersion *pUsers, int32_t numOfUses, void **ppRsp, + int32_t *pRspLen); #ifdef __cplusplus } diff --git a/source/dnode/mnode/impl/src/mndStb.c b/source/dnode/mnode/impl/src/mndStb.c index 0fe78f4dcc..726631adea 100644 --- a/source/dnode/mnode/impl/src/mndStb.c +++ b/source/dnode/mnode/impl/src/mndStb.c @@ -28,8 +28,8 @@ #include "mndVgroup.h" #include "tname.h" -#define TSDB_STB_VER_NUMBER 1 -#define TSDB_STB_RESERVE_SIZE 64 +#define STB_VER_NUMBER 1 +#define STB_RESERVE_SIZE 64 static SSdbRow *mndStbActionDecode(SSdbRaw *pRaw); static int32_t mndStbActionInsert(SSdb *pSdb, SStbObj *pStb); @@ -46,13 +46,15 @@ static int32_t mndRetrieveStb(SNodeMsg *pReq, SShowObj *pShow, SSDataBlock *pBl static void mndCancelGetNextStb(SMnode *pMnode, void *pIter); int32_t mndInitStb(SMnode *pMnode) { - SSdbTable table = {.sdbType = SDB_STB, - .keyType = SDB_KEY_BINARY, - .encodeFp = (SdbEncodeFp)mndStbActionEncode, - .decodeFp = (SdbDecodeFp)mndStbActionDecode, - .insertFp = (SdbInsertFp)mndStbActionInsert, - .updateFp = (SdbUpdateFp)mndStbActionUpdate, - .deleteFp = (SdbDeleteFp)mndStbActionDelete}; + SSdbTable table = { + .sdbType = SDB_STB, + .keyType = SDB_KEY_BINARY, + .encodeFp = (SdbEncodeFp)mndStbActionEncode, + .decodeFp = (SdbDecodeFp)mndStbActionDecode, + .insertFp = (SdbInsertFp)mndStbActionInsert, + .updateFp = (SdbUpdateFp)mndStbActionUpdate, + .deleteFp = (SdbDeleteFp)mndStbActionDelete, + }; mndSetMsgHandle(pMnode, TDMT_MND_CREATE_STB, mndProcessMCreateStbReq); mndSetMsgHandle(pMnode, TDMT_MND_ALTER_STB, mndProcessMAlterStbReq); @@ -74,8 +76,8 @@ SSdbRaw *mndStbActionEncode(SStbObj *pStb) { terrno = TSDB_CODE_OUT_OF_MEMORY; int32_t size = sizeof(SStbObj) + (pStb->numOfColumns + pStb->numOfTags) * sizeof(SSchema) + +pStb->commentLen + - pStb->ast1Len + pStb->ast2Len + TSDB_STB_RESERVE_SIZE; - SSdbRaw *pRaw = sdbAllocRaw(SDB_STB, TSDB_STB_VER_NUMBER, size); + pStb->ast1Len + pStb->ast2Len + STB_RESERVE_SIZE; + SSdbRaw *pRaw = sdbAllocRaw(SDB_STB, STB_VER_NUMBER, size); if (pRaw == NULL) goto _OVER; int32_t dataPos = 0; @@ -99,6 +101,7 @@ SSdbRaw *mndStbActionEncode(SStbObj *pStb) { for (int32_t i = 0; i < pStb->numOfColumns; ++i) { SSchema *pSchema = &pStb->pColumns[i]; SDB_SET_INT8(pRaw, dataPos, pSchema->type, _OVER) + SDB_SET_INT8(pRaw, dataPos, pSchema->flags, _OVER) SDB_SET_INT16(pRaw, dataPos, pSchema->colId, _OVER) SDB_SET_INT32(pRaw, dataPos, pSchema->bytes, _OVER) SDB_SET_BINARY(pRaw, dataPos, pSchema->name, TSDB_COL_NAME_LEN, _OVER) @@ -107,6 +110,7 @@ SSdbRaw *mndStbActionEncode(SStbObj *pStb) { for (int32_t i = 0; i < pStb->numOfTags; ++i) { SSchema *pSchema = &pStb->pTags[i]; SDB_SET_INT8(pRaw, dataPos, pSchema->type, _OVER) + SDB_SET_INT8(pRaw, dataPos, pSchema->flags, _OVER) SDB_SET_INT16(pRaw, dataPos, pSchema->colId, _OVER) SDB_SET_INT32(pRaw, dataPos, pSchema->bytes, _OVER) SDB_SET_BINARY(pRaw, dataPos, pSchema->name, TSDB_COL_NAME_LEN, _OVER) @@ -121,7 +125,7 @@ SSdbRaw *mndStbActionEncode(SStbObj *pStb) { if (pStb->ast2Len > 0) { SDB_SET_BINARY(pRaw, dataPos, pStb->pAst2, pStb->ast2Len, _OVER) } - SDB_SET_RESERVE(pRaw, dataPos, TSDB_STB_RESERVE_SIZE, _OVER) + SDB_SET_RESERVE(pRaw, dataPos, STB_RESERVE_SIZE, _OVER) SDB_SET_DATALEN(pRaw, dataPos, _OVER) terrno = 0; @@ -143,7 +147,7 @@ static SSdbRow *mndStbActionDecode(SSdbRaw *pRaw) { int8_t sver = 0; if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER; - if (sver != TSDB_STB_VER_NUMBER) { + if (sver != STB_VER_NUMBER) { terrno = TSDB_CODE_SDB_INVALID_DATA_VER; goto _OVER; } @@ -183,6 +187,7 @@ static SSdbRow *mndStbActionDecode(SSdbRaw *pRaw) { for (int32_t i = 0; i < pStb->numOfColumns; ++i) { SSchema *pSchema = &pStb->pColumns[i]; SDB_GET_INT8(pRaw, dataPos, &pSchema->type, _OVER) + SDB_GET_INT8(pRaw, dataPos, &pSchema->flags, _OVER) SDB_GET_INT16(pRaw, dataPos, &pSchema->colId, _OVER) SDB_GET_INT32(pRaw, dataPos, &pSchema->bytes, _OVER) SDB_GET_BINARY(pRaw, dataPos, pSchema->name, TSDB_COL_NAME_LEN, _OVER) @@ -191,6 +196,7 @@ static SSdbRow *mndStbActionDecode(SSdbRaw *pRaw) { for (int32_t i = 0; i < pStb->numOfTags; ++i) { SSchema *pSchema = &pStb->pTags[i]; SDB_GET_INT8(pRaw, dataPos, &pSchema->type, _OVER) + SDB_GET_INT8(pRaw, dataPos, &pSchema->flags, _OVER) SDB_GET_INT16(pRaw, dataPos, &pSchema->colId, _OVER) SDB_GET_INT32(pRaw, dataPos, &pSchema->bytes, _OVER) SDB_GET_BINARY(pRaw, dataPos, pSchema->name, TSDB_COL_NAME_LEN, _OVER) @@ -211,7 +217,7 @@ static SSdbRow *mndStbActionDecode(SSdbRaw *pRaw) { if (pStb->pAst2 == NULL) goto _OVER; SDB_GET_BINARY(pRaw, dataPos, pStb->pAst2, pStb->ast2Len, _OVER) } - SDB_GET_RESERVE(pRaw, dataPos, TSDB_STB_RESERVE_SIZE, _OVER) + SDB_GET_RESERVE(pRaw, dataPos, STB_RESERVE_SIZE, _OVER) terrno = 0; @@ -488,7 +494,7 @@ int32_t mndCheckCreateStbReq(SMCreateStbReq *pCreate) { for (int32_t i = 0; i < pCreate->numOfColumns; ++i) { SField *pField1 = taosArrayGet(pCreate->pColumns, i); - if (pField->type < 0) { + if (pField1->type < 0) { terrno = TSDB_CODE_MND_INVALID_STB_OPTION; return -1; } @@ -574,6 +580,7 @@ static int32_t mndSetCreateStbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj action.pCont = pReq; action.contLen = contLen; action.msgType = TDMT_VND_CREATE_STB; + action.acceptableCode = TSDB_CODE_TDB_STB_ALREADY_EXIST; if (mndTransAppendRedoAction(pTrans, &action) != 0) { taosMemoryFree(pReq); sdbCancelFetch(pSdb, pIter); @@ -613,6 +620,7 @@ static int32_t mndSetCreateStbUndoActions(SMnode *pMnode, STrans *pTrans, SDbObj action.pCont = pReq; action.contLen = contLen; action.msgType = TDMT_VND_DROP_STB; + action.acceptableCode = TSDB_CODE_TDB_STB_NOT_EXIST; if (mndTransAppendUndoAction(pTrans, &action) != 0) { taosMemoryFree(pReq); sdbCancelFetch(pSdb, pIter); @@ -733,6 +741,7 @@ _OVER: mndTransDrop(pTrans); return code; } + int32_t mndAddStbToTrans(SMnode *pMnode, STrans *pTrans, SDbObj *pDb, SStbObj *pStb) { mndTransSetDbInfo(pTrans, pDb); if (mndSetCreateStbRedoLogs(pMnode, pTrans, pDb, pStb) != 0) return -1; @@ -792,7 +801,10 @@ static int32_t mndProcessMCreateStbReq(SNodeMsg *pReq) { } int32_t numOfStbs = -1; - mndGetNumOfStbs(pMnode, pDb->name, &numOfStbs); + if (mndGetNumOfStbs(pMnode, pDb->name, &numOfStbs) != 0) { + goto _OVER; + } + if (pDb->cfg.numOfStables == 1 && numOfStbs != 0) { terrno = TSDB_CODE_MND_SINGLE_STB_MODE_DB; goto _OVER; diff --git a/source/dnode/mnode/impl/src/mndStream.c b/source/dnode/mnode/impl/src/mndStream.c index 1404b1cd94..2260b178c8 100644 --- a/source/dnode/mnode/impl/src/mndStream.c +++ b/source/dnode/mnode/impl/src/mndStream.c @@ -330,7 +330,10 @@ static SStbObj *mndCreateStbForStream(SMnode *pMnode, STrans *pTrans, const SStr } int32_t numOfStbs = -1; - mndGetNumOfStbs(pMnode, pDb->name, &numOfStbs); + if (mndGetNumOfStbs(pMnode, pDb->name, &numOfStbs) != 0) { + goto _OVER; + } + if (pDb->cfg.numOfStables == 1 && numOfStbs != 0) { terrno = TSDB_CODE_MND_SINGLE_STB_MODE_DB; goto _OVER; diff --git a/source/util/src/terror.c b/source/util/src/terror.c index 1470496c68..8ebcaede8c 100644 --- a/source/util/src/terror.c +++ b/source/util/src/terror.c @@ -326,6 +326,9 @@ TAOS_DEFINE_ERROR(TSDB_CODE_TDB_INVALID_TABLE_ID, "Invalid table ID") TAOS_DEFINE_ERROR(TSDB_CODE_TDB_INVALID_TABLE_TYPE, "Invalid table type") TAOS_DEFINE_ERROR(TSDB_CODE_TDB_IVD_TB_SCHEMA_VERSION, "Invalid table schema version") TAOS_DEFINE_ERROR(TSDB_CODE_TDB_TABLE_ALREADY_EXIST, "Table already exists") +TAOS_DEFINE_ERROR(TSDB_CODE_TDB_TABLE_NOT_EXIST, "Table not exists") +TAOS_DEFINE_ERROR(TSDB_CODE_TDB_STB_ALREADY_EXIST, "Stable already exists") +TAOS_DEFINE_ERROR(TSDB_CODE_TDB_STB_NOT_EXIST, "Stable not exists") TAOS_DEFINE_ERROR(TSDB_CODE_TDB_INVALID_CONFIG, "Invalid configuration") TAOS_DEFINE_ERROR(TSDB_CODE_TDB_INIT_FAILED, "Tsdb init failed") TAOS_DEFINE_ERROR(TSDB_CODE_TDB_NO_DISKSPACE, "No diskspace for tsdb") diff --git a/tests/script/general/stable/testSuite.sim b/tests/script/general/stable/testSuite.sim deleted file mode 100644 index e786ac9ca4..0000000000 --- a/tests/script/general/stable/testSuite.sim +++ /dev/null @@ -1,5 +0,0 @@ -run general/stable/disk.sim -run general/stable/dnode3.sim -run general/stable/metrics.sim -run general/stable/values.sim -run general/stable/vnode3.sim From 073745c8937775107cf1260f31a11f77d4f13f2b Mon Sep 17 00:00:00 2001 From: Shengliang Guan Date: Sat, 7 May 2022 13:06:23 +0800 Subject: [PATCH 22/27] refactor: alter stb --- include/common/tmsg.h | 8 ++++---- source/common/src/tmsg.c | 6 +++--- source/dnode/mnode/impl/src/mndStb.c | 16 ++++++++-------- source/dnode/mnode/impl/test/stb/stb.cpp | 14 +++++++------- source/libs/parser/src/parTranslater.c | 6 +++--- 5 files changed, 25 insertions(+), 25 deletions(-) diff --git a/include/common/tmsg.h b/include/common/tmsg.h index 3889784f2f..f22e8fbb97 100644 --- a/include/common/tmsg.h +++ b/include/common/tmsg.h @@ -326,11 +326,11 @@ typedef struct { int8_t alterType; int32_t numOfFields; SArray* pFields; -} SMAltertbReq; +} SMAlterStbReq; -int32_t tSerializeSMAlterStbReq(void* buf, int32_t bufLen, SMAltertbReq* pReq); -int32_t tDeserializeSMAlterStbReq(void* buf, int32_t bufLen, SMAltertbReq* pReq); -void tFreeSMAltertbReq(SMAltertbReq* pReq); +int32_t tSerializeSMAlterStbReq(void* buf, int32_t bufLen, SMAlterStbReq* pReq); +int32_t tDeserializeSMAlterStbReq(void* buf, int32_t bufLen, SMAlterStbReq* pReq); +void tFreeSMAltertbReq(SMAlterStbReq* pReq); typedef struct SEpSet { int8_t inUse; diff --git a/source/common/src/tmsg.c b/source/common/src/tmsg.c index 6278b52a04..24e6fc9e63 100644 --- a/source/common/src/tmsg.c +++ b/source/common/src/tmsg.c @@ -594,7 +594,7 @@ int32_t tDeserializeSMDropStbReq(void *buf, int32_t bufLen, SMDropStbReq *pReq) return 0; } -int32_t tSerializeSMAlterStbReq(void *buf, int32_t bufLen, SMAltertbReq *pReq) { +int32_t tSerializeSMAlterStbReq(void *buf, int32_t bufLen, SMAlterStbReq *pReq) { SCoder encoder = {0}; tCoderInit(&encoder, TD_LITTLE_ENDIAN, buf, bufLen, TD_ENCODER); @@ -615,7 +615,7 @@ int32_t tSerializeSMAlterStbReq(void *buf, int32_t bufLen, SMAltertbReq *pReq) { return tlen; } -int32_t tDeserializeSMAlterStbReq(void *buf, int32_t bufLen, SMAltertbReq *pReq) { +int32_t tDeserializeSMAlterStbReq(void *buf, int32_t bufLen, SMAlterStbReq *pReq) { SCoder decoder = {0}; tCoderInit(&decoder, TD_LITTLE_ENDIAN, buf, bufLen, TD_DECODER); @@ -645,7 +645,7 @@ int32_t tDeserializeSMAlterStbReq(void *buf, int32_t bufLen, SMAltertbReq *pReq) return 0; } -void tFreeSMAltertbReq(SMAltertbReq *pReq) { +void tFreeSMAltertbReq(SMAlterStbReq *pReq) { taosArrayDestroy(pReq->pFields); pReq->pFields = NULL; } diff --git a/source/dnode/mnode/impl/src/mndStb.c b/source/dnode/mnode/impl/src/mndStb.c index 726631adea..b6912d0777 100644 --- a/source/dnode/mnode/impl/src/mndStb.c +++ b/source/dnode/mnode/impl/src/mndStb.c @@ -831,7 +831,7 @@ static int32_t mndProcessVCreateStbRsp(SNodeMsg *pRsp) { return 0; } -static int32_t mndCheckAlterStbReq(SMAltertbReq *pAlter) { +static int32_t mndCheckAlterStbReq(SMAlterStbReq *pAlter) { if (pAlter->numOfFields < 1 || pAlter->numOfFields != (int32_t)taosArrayGetSize(pAlter->pFields)) { terrno = TSDB_CODE_MND_INVALID_STB_OPTION; return -1; @@ -1182,7 +1182,7 @@ static int32_t mndSetAlterStbRedoActions(SMnode *pMnode, STrans *pTrans, SDbObj return 0; } -static int32_t mndAlterStb(SMnode *pMnode, SNodeMsg *pReq, const SMAltertbReq *pAlter, SDbObj *pDb, SStbObj *pOld) { +static int32_t mndAlterStb(SMnode *pMnode, SNodeMsg *pReq, const SMAlterStbReq *pAlter, SDbObj *pDb, SStbObj *pOld) { SStbObj stbObj = {0}; taosRLockLatch(&pOld->lock); memcpy(&stbObj, pOld, sizeof(SStbObj)); @@ -1246,12 +1246,12 @@ _OVER: } static int32_t mndProcessMAlterStbReq(SNodeMsg *pReq) { - SMnode *pMnode = pReq->pNode; - int32_t code = -1; - SDbObj *pDb = NULL; - SStbObj *pStb = NULL; - SUserObj *pUser = NULL; - SMAltertbReq alterReq = {0}; + SMnode *pMnode = pReq->pNode; + int32_t code = -1; + SDbObj *pDb = NULL; + SStbObj *pStb = NULL; + SUserObj *pUser = NULL; + SMAlterStbReq alterReq = {0}; if (tDeserializeSMAlterStbReq(pReq->rpcMsg.pCont, pReq->rpcMsg.contLen, &alterReq) != 0) { terrno = TSDB_CODE_INVALID_MSG; diff --git a/source/dnode/mnode/impl/test/stb/stb.cpp b/source/dnode/mnode/impl/test/stb/stb.cpp index 220a73def6..16974ad541 100644 --- a/source/dnode/mnode/impl/test/stb/stb.cpp +++ b/source/dnode/mnode/impl/test/stb/stb.cpp @@ -137,7 +137,7 @@ void* MndTestStb::BuildCreateStbReq(const char* stbname, int32_t* pContLen) { } void* MndTestStb::BuildAlterStbAddTagReq(const char* stbname, const char* tagname, int32_t* pContLen) { - SMAltertbReq req = {0}; + SMAlterStbReq req = {0}; strcpy(req.name, stbname); req.numOfFields = 1; req.pFields = taosArrayInit(1, sizeof(SField)); @@ -158,7 +158,7 @@ void* MndTestStb::BuildAlterStbAddTagReq(const char* stbname, const char* tagnam } void* MndTestStb::BuildAlterStbDropTagReq(const char* stbname, const char* tagname, int32_t* pContLen) { - SMAltertbReq req = {0}; + SMAlterStbReq req = {0}; strcpy(req.name, stbname); req.numOfFields = 1; req.pFields = taosArrayInit(1, sizeof(SField)); @@ -180,7 +180,7 @@ void* MndTestStb::BuildAlterStbDropTagReq(const char* stbname, const char* tagna void* MndTestStb::BuildAlterStbUpdateTagNameReq(const char* stbname, const char* tagname, const char* newtagname, int32_t* pContLen) { - SMAltertbReq req = {0}; + SMAlterStbReq req = {0}; strcpy(req.name, stbname); req.numOfFields = 2; req.pFields = taosArrayInit(2, sizeof(SField)); @@ -208,7 +208,7 @@ void* MndTestStb::BuildAlterStbUpdateTagNameReq(const char* stbname, const char* void* MndTestStb::BuildAlterStbUpdateTagBytesReq(const char* stbname, const char* tagname, int32_t bytes, int32_t* pContLen) { - SMAltertbReq req = {0}; + SMAlterStbReq req = {0}; strcpy(req.name, stbname); req.numOfFields = 1; req.pFields = taosArrayInit(1, sizeof(SField)); @@ -229,7 +229,7 @@ void* MndTestStb::BuildAlterStbUpdateTagBytesReq(const char* stbname, const char } void* MndTestStb::BuildAlterStbAddColumnReq(const char* stbname, const char* colname, int32_t* pContLen) { - SMAltertbReq req = {0}; + SMAlterStbReq req = {0}; strcpy(req.name, stbname); req.numOfFields = 1; req.pFields = taosArrayInit(1, sizeof(SField)); @@ -250,7 +250,7 @@ void* MndTestStb::BuildAlterStbAddColumnReq(const char* stbname, const char* col } void* MndTestStb::BuildAlterStbDropColumnReq(const char* stbname, const char* colname, int32_t* pContLen) { - SMAltertbReq req = {0}; + SMAlterStbReq req = {0}; strcpy(req.name, stbname); req.numOfFields = 1; req.pFields = taosArrayInit(1, sizeof(SField)); @@ -272,7 +272,7 @@ void* MndTestStb::BuildAlterStbDropColumnReq(const char* stbname, const char* co void* MndTestStb::BuildAlterStbUpdateColumnBytesReq(const char* stbname, const char* colname, int32_t bytes, int32_t* pContLen) { - SMAltertbReq req = {0}; + SMAlterStbReq req = {0}; strcpy(req.name, stbname); req.numOfFields = 1; req.pFields = taosArrayInit(1, sizeof(SField)); diff --git a/source/libs/parser/src/parTranslater.c b/source/libs/parser/src/parTranslater.c index 6874b5b7d4..bd614d0165 100644 --- a/source/libs/parser/src/parTranslater.c +++ b/source/libs/parser/src/parTranslater.c @@ -2596,7 +2596,7 @@ static int32_t translateDropSuperTable(STranslateContext* pCxt, SDropSuperTableS pStmt->ignoreNotExists); } -static int32_t setAlterTableField(SAlterTableStmt* pStmt, SMAltertbReq* pAlterReq) { +static int32_t setAlterTableField(SAlterTableStmt* pStmt, SMAlterStbReq* pAlterReq) { pAlterReq->pFields = taosArrayInit(2, sizeof(TAOS_FIELD)); if (NULL == pAlterReq->pFields) { return TSDB_CODE_OUT_OF_MEMORY; @@ -2632,8 +2632,8 @@ static int32_t setAlterTableField(SAlterTableStmt* pStmt, SMAltertbReq* pAlterRe } static int32_t translateAlterTable(STranslateContext* pCxt, SAlterTableStmt* pStmt) { - SMAltertbReq alterReq = {0}; - SName tableName; + SMAlterStbReq alterReq = {0}; + SName tableName; tNameExtractFullName(toName(pCxt->pParseCxt->acctId, pStmt->dbName, pStmt->tableName, &tableName), alterReq.name); alterReq.alterType = pStmt->alterType; alterReq.numOfFields = 1; From 7375cf4921ec0438be0af588e7a7774e4cbcb369 Mon Sep 17 00:00:00 2001 From: dapan1121 Date: Sat, 7 May 2022 13:20:48 +0800 Subject: [PATCH 23/27] fix mem issue --- source/libs/nodes/src/nodesUtilFuncs.c | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/source/libs/nodes/src/nodesUtilFuncs.c b/source/libs/nodes/src/nodesUtilFuncs.c index 0bceb84f4a..319b680161 100644 --- a/source/libs/nodes/src/nodesUtilFuncs.c +++ b/source/libs/nodes/src/nodesUtilFuncs.c @@ -1337,7 +1337,9 @@ void valueNodeToVariant(const SValueNode* pNode, SVariant* pVal) { case TSDB_DATA_TYPE_NCHAR: case TSDB_DATA_TYPE_VARCHAR: case TSDB_DATA_TYPE_VARBINARY: - pVal->pz = pNode->datum.p; + pVal->pz = taosMemoryMalloc(pVal->nLen + VARSTR_HEADER_SIZE + 1); + memcpy(pVal->pz, pNode->datum.p, pVal->nLen + VARSTR_HEADER_SIZE); + pVal->pz[pVal->nLen + VARSTR_HEADER_SIZE] = 0; break; case TSDB_DATA_TYPE_JSON: case TSDB_DATA_TYPE_DECIMAL: From 174bc6a5680e17fefe0800aade5ed607998ad416 Mon Sep 17 00:00:00 2001 From: wangmm0220 Date: Sat, 7 May 2022 13:58:10 +0800 Subject: [PATCH 24/27] refactor:modify schemaless interface --- cmake/cmake.define | 2 +- include/client/taos.h | 2 +- source/client/src/clientSml.c | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/cmake/cmake.define b/cmake/cmake.define index 2b4fe5c42f..aeab39cab4 100644 --- a/cmake/cmake.define +++ b/cmake/cmake.define @@ -46,7 +46,7 @@ ENDIF () IF (TD_WINDOWS) MESSAGE("${Yellow} set compiler flag for Windows! ${ColourReset}") - SET(COMMON_FLAGS "/W3 /D_WIN32 /vmg") + SET(COMMON_FLAGS "/W3 /D_WIN32") # IF (MSVC AND (MSVC_VERSION GREATER_EQUAL 1900)) # SET(COMMON_FLAGS "${COMMON_FLAGS} /Wv:18") diff --git a/include/client/taos.h b/include/client/taos.h index 9d8b5107bb..6e20900668 100644 --- a/include/client/taos.h +++ b/include/client/taos.h @@ -195,7 +195,7 @@ DLL_EXPORT void taos_unsubscribe(TAOS_SUB *tsub, int keepProgress); #endif DLL_EXPORT int taos_load_table_info(TAOS *taos, const char *tableNameList); -DLL_EXPORT TAOS_RES *taos_schemaless_insert(TAOS *taos, char *lines[], int numLines, int protocol, int precision, bool dataFormat); +DLL_EXPORT TAOS_RES *taos_schemaless_insert(TAOS *taos, char *lines[], int numLines, int protocol, int precision); /* --------------------------TMQ INTERFACE------------------------------- */ diff --git a/source/client/src/clientSml.c b/source/client/src/clientSml.c index 85cd0661e3..5f9138bb43 100644 --- a/source/client/src/clientSml.c +++ b/source/client/src/clientSml.c @@ -1724,13 +1724,13 @@ cleanup: * */ -TAOS_RES* taos_schemaless_insert(TAOS* taos, char* lines[], int numLines, int protocol, int precision, bool dataFormat) { +TAOS_RES* taos_schemaless_insert(TAOS* taos, char* lines[], int numLines, int protocol, int precision) { SRequestObj* request = createRequest(taos, NULL, NULL, TSDB_SQL_INSERT); if(!request){ return NULL; } - SSmlHandle* info = smlBuildSmlInfo(taos, request, protocol, precision, dataFormat); + SSmlHandle* info = smlBuildSmlInfo(taos, request, protocol, precision, false); if(!info){ return (TAOS_RES*)request; } From 67e8dccfec0f0c302d6d190ac440e590a36d51c7 Mon Sep 17 00:00:00 2001 From: jiacy-jcy <714897623@qq.com> Date: Sat, 7 May 2022 14:15:32 +0800 Subject: [PATCH 25/27] test case --- tests/system-test/2-query/last.py | 36 +++++++++++++++---------------- 1 file changed, 18 insertions(+), 18 deletions(-) diff --git a/tests/system-test/2-query/last.py b/tests/system-test/2-query/last.py index 531afd0117..b5c0498e40 100644 --- a/tests/system-test/2-query/last.py +++ b/tests/system-test/2-query/last.py @@ -21,12 +21,12 @@ class TDTestCase: tdSql.execute("insert into stb_1(ts) values(%d)" % (self.ts - 1)) # last verifacation - tdSql.query("select last(*) from stb_1") - tdSql.checkRows(1) - tdSql.checkData(0, 1, None) - tdSql.query("select last(*) from db.stb_1") - tdSql.checkRows(1) - tdSql.checkData(0, 1, None) + # tdSql.query("select last(*) from stb_1") + # tdSql.checkRows(1) + # tdSql.checkData(0, 1, None) + # tdSql.query("select last(*) from db.stb_1") + # tdSql.checkRows(1) + # tdSql.checkData(0, 1, None) tdSql.query("select last(col1) from stb_1") tdSql.checkRows(0) tdSql.query("select last(col1) from db.stb_1") @@ -86,12 +86,12 @@ class TDTestCase: tdSql.execute("insert into stb_1 values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1)) - tdSql.query("select last(*) from stb_1") - tdSql.checkRows(1) - tdSql.checkData(0, 1, 10) - tdSql.query("select last(*) from db.stb_1") - tdSql.checkRows(1) - tdSql.checkData(0, 1, 10) + # tdSql.query("select last(*) from stb_1") + # tdSql.checkRows(1) + # tdSql.checkData(0, 1, 10) + # tdSql.query("select last(*) from db.stb_1") + # tdSql.checkRows(1) + # tdSql.checkData(0, 1, 10) tdSql.query("select last(col1) from stb_1") tdSql.checkRows(1) tdSql.checkData(0, 0, 10) @@ -175,12 +175,12 @@ class TDTestCase: tdSql.execute('''create table ntb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, col7 bool, col8 binary(20), col9 nchar(20), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned)''') tdSql.execute("insert into ntb(ts) values(%d)" % (self.ts - 1)) - tdSql.query("select last(*) from ntb") - tdSql.checkRows(1) - tdSql.checkData(0, 1, None) - tdSql.query("select last(*) from db.ntb") - tdSql.checkRows(1) - tdSql.checkData(0, 1, None) + # tdSql.query("select last(*) from ntb") + # tdSql.checkRows(1) + # tdSql.checkData(0, 1, None) + # tdSql.query("select last(*) from db.ntb") + # tdSql.checkRows(1) + # tdSql.checkData(0, 1, None) tdSql.query("select last(col1) from ntb") tdSql.checkRows(0) tdSql.query("select last(col1) from db.ntb") From 5bf762d50d20c6a99f4cd6e6eebd4092cd1e9829 Mon Sep 17 00:00:00 2001 From: plum-lihui Date: Sat, 7 May 2022 15:15:31 +0800 Subject: [PATCH 26/27] [test: add test cases for telemetry] --- tests/system-test/0-others/telemetry.py | 197 ++++++++++++++++++++++++ 1 file changed, 197 insertions(+) create mode 100644 tests/system-test/0-others/telemetry.py diff --git a/tests/system-test/0-others/telemetry.py b/tests/system-test/0-others/telemetry.py new file mode 100644 index 0000000000..3ab39f9e7b --- /dev/null +++ b/tests/system-test/0-others/telemetry.py @@ -0,0 +1,197 @@ +import taos +import sys +import time +import socket +import pexpect +import os +import http.server +import gzip +import threading +import json + +from util.log import * +from util.sql import * +from util.cases import * +from util.dnodes import * + +telemetryPort = '80' + +#{ +# "instanceId": "5cf4cd7a-acd4-43ba-8b0d-e84395b76a65", +# "reportVersion": 1, +# "os": "Ubuntu 20.04.3 LTS", +# "cpuModel": "Intel(R) Xeon(R) CPU E5-2620 v3 @ 2.40GHz", +# "numOfCpu": 6, +# "memory": "65860292 kB", +# "version": "3.0.0.0", +# "buildInfo": "Built at 2022-05-07 14:09:02", +# "gitInfo": "2139ccceb0946cde86b6b553b11e338f1ba437e5", +# "email": "user@taosdata.com", +# "numOfDnode": 1, +# "numOfMnode": 1, +# "numOfVgroup": 32, +# "numOfDatabase": 2, +# "numOfSuperTable": 0, +# "numOfChildTable": 100, +# "numOfColumn": 200, +# "numOfPoint": 300, +# "totalStorage": 400, +# "compStorage": 500 +#} + +def telemetryInfoCheck(infoDict=''): + if "instanceId" not in infoDict or len(infoDict["instanceId"]) == 0: + tdLog.exit("instanceId is null!") + + if "reportVersion" not in infoDict or infoDict["reportVersion"] != 1: + tdLog.exit("reportVersion is null!") + + if "os" not in infoDict: + tdLog.exit("os is null!") + + if "cpuModel" not in infoDict: + tdLog.exit("cpuModel is null!") + + if "numOfCpu" not in infoDict or infoDict["numOfCpu"] == 0: + tdLog.exit("numOfCpu is null!") + + if "memory" not in infoDict: + tdLog.exit("memory is null!") + + if "version" not in infoDict: + tdLog.exit("version is null!") + + if "buildInfo" not in infoDict: + tdLog.exit("buildInfo is null!") + + if "gitInfo" not in infoDict: + tdLog.exit("gitInfo is null!") + + if "email" not in infoDict: + tdLog.exit("email is not exists!") + + if "numOfDnode" not in infoDict or infoDict["numOfDnode"] < 1: + tdLog.exit("numOfDnode is null!") + + if "numOfMnode" not in infoDict or infoDict["numOfMnode"] < 1: + tdLog.exit("numOfMnode is null!") + + if "numOfVgroup" not in infoDict or infoDict["numOfVgroup"] <= 0: + tdLog.exit("numOfVgroup is null!") + + if "numOfDatabase" not in infoDict or infoDict["numOfDatabase"] <= 0: + tdLog.exit("numOfDatabase is null!") + + if "numOfSuperTable" not in infoDict or infoDict["numOfSuperTable"] < 0: + tdLog.exit("numOfSuperTable is null!") + + if "numOfChildTable" not in infoDict or infoDict["numOfChildTable"] < 0: + tdLog.exit("numOfChildTable is null!") + + if "numOfColumn" not in infoDict or infoDict["numOfColumn"] < 0: + tdLog.exit("numOfColumn is null!") + + if "numOfPoint" not in infoDict or infoDict["numOfPoint"] < 0: + tdLog.exit("numOfPoint is null!") + + if "totalStorage" not in infoDict or infoDict["totalStorage"] < 0: + tdLog.exit("totalStorage is null!") + + if "compStorage" not in infoDict or infoDict["compStorage"] < 0: + tdLog.exit("compStorage is null!") + + +class RequestHandlerImpl(http.server.BaseHTTPRequestHandler): + def do_GET(self): + """ + process GET request + """ + + def do_POST(self): + """ + process POST request + """ + contentEncoding = self.headers["Content-Encoding"] + + if contentEncoding == 'gzip': + req_body = self.rfile.read(int(self.headers["Content-Length"])) + plainText = gzip.decompress(req_body).decode() + else: + plainText = self.rfile.read(int(self.headers["Content-Length"])).decode() + + print("monitor info:\n%s"%plainText) + + # 1. send response code and header + self.send_response(200) + self.send_header("Content-Type", "text/html; charset=utf-8") + self.end_headers() + + # 2. send response content + #self.wfile.write(("Hello World: " + req_body + "\n").encode("utf-8")) + + # 3. check request body info + infoDict = json.loads(plainText) + #print("================") + #print(infoDict) + telemetryInfoCheck(infoDict) + + # 4. shutdown the server and exit case + assassin = threading.Thread(target=httpServer.shutdown) + assassin.daemon = True + assassin.start() + print ("==== shutdown http server ====") + +class TDTestCase: + hostname = socket.gethostname() + serverPort = '7080' + rpcDebugFlagVal = '143' + clientCfgDict = {'serverPort': '', 'firstEp': '', 'secondEp':'', 'rpcDebugFlag':'135', 'fqdn':''} + clientCfgDict["serverPort"] = serverPort + clientCfgDict["firstEp"] = hostname + ':' + serverPort + clientCfgDict["secondEp"] = hostname + ':' + serverPort + clientCfgDict["rpcDebugFlag"] = rpcDebugFlagVal + clientCfgDict["fqdn"] = hostname + + updatecfgDict = {'clientCfg': {}, 'serverPort': '', 'firstEp': '', 'secondEp':'', 'rpcDebugFlag':'135', 'fqdn':''} + updatecfgDict["clientCfg"] = clientCfgDict + updatecfgDict["serverPort"] = serverPort + updatecfgDict["firstEp"] = hostname + ':' + serverPort + updatecfgDict["secondEp"] = hostname + ':' + serverPort + updatecfgDict["fqdn"] = hostname + + updatecfgDict["telemetryReporting"] = '1' + updatecfgDict["telemetryServer"] = hostname + updatecfgDict["telemetryPort"] = telemetryPort + updatecfgDict["telemetryInterval"] = "3" + + print ("===================: ", updatecfgDict) + + def init(self, conn, logSql): + tdLog.debug(f"start to excute {__file__}") + tdSql.init(conn.cursor()) + + def run(self): # sourcery skip: extract-duplicate-method, remove-redundant-fstring + tdSql.prepare() + # time.sleep(2) + vgroups = "30" + sql = "create database db3 vgroups " + vgroups + tdSql.query(sql) + + # loop to wait request + httpServer.serve_forever() + + def stop(self): + tdSql.close() + tdLog.success(f"{__file__} successfully executed") + +# create http server: bing ip/port , and request processor +serverAddress = ("", int(telemetryPort)) +httpServer = http.server.HTTPServer(serverAddress, RequestHandlerImpl) + +tdCases.addLinux(__file__, TDTestCase()) +tdCases.addWindows(__file__, TDTestCase()) + + + + + From 3a7b65b038a54d295c5df65a8499485d29b660de Mon Sep 17 00:00:00 2001 From: plum-lihui Date: Sat, 7 May 2022 15:18:42 +0800 Subject: [PATCH 27/27] [test: add test cases for telemetry] --- tests/system-test/fulltest.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/system-test/fulltest.sh b/tests/system-test/fulltest.sh index 4149c2cd9d..dfb6fdbefb 100755 --- a/tests/system-test/fulltest.sh +++ b/tests/system-test/fulltest.sh @@ -5,6 +5,7 @@ set -x python3 ./test.py -f 0-others/taosShell.py python3 ./test.py -f 0-others/taosShellError.py python3 ./test.py -f 0-others/taosShellNetChk.py +python3 ./test.py -f 0-others/telemetry.py #python3 ./test.py -f 2-query/between.py