from random import randrange import time import threading import secrets from util.log import * from util.sql import * from util.cases import * from util.dnodes import * from util.common import * # from tmqCommon import * class TDTestCase: updatecfgDict = {'asynclog': 0, 'ttlUnit': 1, 'ttlPushInterval': 5, 'ratioOfVnodeStreamThrea': 4} def __init__(self): self.vgroups = 4 self.ctbNum = 10 self.rowsPerTbl = 10000 self.duraion = '1h' def init(self, conn, logSql, replicaVar=1): self.replicaVar = int(replicaVar) tdLog.debug(f"start to excute {__file__}") tdSql.init(conn.cursor(), False) def create_database(self, tsql, dbName, dropFlag=1, vgroups=2, replica=1, duration: str = '1d'): if dropFlag == 1: tsql.execute("drop database if exists %s" % (dbName)) tsql.execute("create database if not exists %s vgroups %d replica %d duration %s" % ( dbName, vgroups, replica, duration)) tdLog.debug("complete to create database %s" % (dbName)) return def create_stable(self, tsql, paraDict): colString = tdCom.gen_column_type_str( colname_prefix=paraDict["colPrefix"], column_elm_list=paraDict["colSchema"]) tagString = tdCom.gen_tag_type_str( tagname_prefix=paraDict["tagPrefix"], tag_elm_list=paraDict["tagSchema"]) sqlString = f"create table if not exists %s.%s (%s) tags (%s)" % ( paraDict["dbName"], paraDict["stbName"], colString, tagString) tdLog.debug("%s" % (sqlString)) tsql.execute(sqlString) return def create_ctable(self, tsql=None, dbName='dbx', stbName='stb', ctbPrefix='ctb', ctbNum=1, ctbStartIdx=0): for i in range(ctbNum): sqlString = "create table %s.%s%d using %s.%s tags(%d, 'tb%d', 'tb%d', %d, %d, %d)" % (dbName, ctbPrefix, i+ctbStartIdx, dbName, stbName, (i+ctbStartIdx) % 5, i+ctbStartIdx + random.randint( 1, 100), i+ctbStartIdx + random.randint(1, 100), i+ctbStartIdx + random.randint(1, 100), i+ctbStartIdx + random.randint(1, 100), i+ctbStartIdx + random.randint(1, 100)) tsql.execute(sqlString) tdLog.debug("complete to create %d child tables by %s.%s" % (ctbNum, dbName, stbName)) return def init_normal_tb(self, tsql, db_name: str, tb_name: str, rows: int, start_ts: int, ts_step: int): sql = 'CREATE TABLE %s.%s (ts timestamp, c1 INT, c2 INT, c3 INT, c4 double, c5 VARCHAR(255))' % ( db_name, tb_name) tsql.execute(sql) sql = 'INSERT INTO %s.%s values' % (db_name, tb_name) for j in range(rows): sql += f'(%d, %d,%d,%d,{random.random()},"varchar_%d"),' % (start_ts + j * ts_step + randrange(500), j % 10 + randrange(200), j % 10, j % 10, j % 10 + randrange(100)) tsql.execute(sql) def insert_data(self, tsql, dbName, ctbPrefix, ctbNum, rowsPerTbl, batchNum, startTs, tsStep): tdLog.debug("start to insert data ............") tsql.execute("use %s" % dbName) pre_insert = "insert into " sql = pre_insert for i in range(ctbNum): rowsBatched = 0 sql += " %s.%s%d values " % (dbName, ctbPrefix, i) for j in range(rowsPerTbl): if (i < ctbNum/2): sql += "(%d, %d, %d, %d,%d,%d,%d,true,'binary%d', 'nchar%d') " % (startTs + j*tsStep + randrange( 500), j % 10 + randrange(100), j % 10 + randrange(200), j % 10, j % 10, j % 10, j % 10, j % 10, j % 10) else: sql += "(%d, %d, NULL, %d,NULL,%d,%d,true,'binary%d', 'nchar%d') " % ( startTs + j*tsStep + randrange(500), j % 10, j % 10, j % 10, j % 10, j % 10, j % 10) rowsBatched += 1 if ((rowsBatched == batchNum) or (j == rowsPerTbl - 1)): tsql.execute(sql) rowsBatched = 0 if j < rowsPerTbl - 1: sql = "insert into %s.%s%d values " % (dbName, ctbPrefix, i) else: sql = "insert into " if sql != pre_insert: tsql.execute(sql) tdLog.debug("insert data ............ [OK]") return def init_data(self, db: str = 'test', ctb_num: int = 10, rows_per_ctb: int = 10000, start_ts: int = 1537146000000, ts_step: int = 500): tdLog.printNoPrefix( "======== prepare test env include database, stable, ctables, and insert data: ") paraDict = {'dbName': db, 'dropFlag': 1, 'vgroups': 2, 'stbName': 'meters', 'colPrefix': 'c', 'tagPrefix': 't', 'colSchema': [{'type': 'INT', 'count': 1}, {'type': 'BIGINT', 'count': 1}, {'type': 'FLOAT', 'count': 1}, {'type': 'DOUBLE', 'count': 1}, {'type': 'smallint', 'count': 1}, {'type': 'tinyint', 'count': 1}, {'type': 'bool', 'count': 1}, {'type': 'binary', 'len': 10, 'count': 1}, {'type': 'nchar', 'len': 10, 'count': 1}], 'tagSchema': [{'type': 'INT', 'count': 1}, {'type': 'nchar', 'len': 20, 'count': 1}, {'type': 'binary', 'len': 20, 'count': 1}, {'type': 'BIGINT', 'count': 1}, {'type': 'smallint', 'count': 1}, {'type': 'DOUBLE', 'count': 1}], 'ctbPrefix': 't', 'ctbStartIdx': 0, 'ctbNum': ctb_num, 'rowsPerTbl': rows_per_ctb, 'batchNum': 3000, 'startTs': start_ts, 'tsStep': ts_step} paraDict['vgroups'] = self.vgroups paraDict['ctbNum'] = ctb_num paraDict['rowsPerTbl'] = rows_per_ctb tdLog.info("create database") self.create_database(tsql=tdSql, dbName=paraDict["dbName"], dropFlag=paraDict["dropFlag"], vgroups=paraDict["vgroups"], replica=self.replicaVar, duration=self.duraion) tdLog.info("create stb") self.create_stable(tsql=tdSql, paraDict=paraDict) tdLog.info("create child tables") self.create_ctable(tsql=tdSql, dbName=paraDict["dbName"], stbName=paraDict["stbName"], ctbPrefix=paraDict["ctbPrefix"], ctbNum=paraDict["ctbNum"], ctbStartIdx=paraDict["ctbStartIdx"]) self.insert_data(tsql=tdSql, dbName=paraDict["dbName"], ctbPrefix=paraDict["ctbPrefix"], ctbNum=paraDict["ctbNum"], rowsPerTbl=paraDict["rowsPerTbl"], batchNum=paraDict["batchNum"], startTs=paraDict["startTs"], tsStep=paraDict["tsStep"]) self.init_normal_tb(tdSql, paraDict['dbName'], 'norm_tb', paraDict['rowsPerTbl'], paraDict['startTs'], paraDict['tsStep']) def test_select_asterisk_from_subquery_with_duplicate_aliasname(self): sql = "select * from (select c8 as a, c9 as a from t1 order by ts desc limit 10)t;" tdSql.query(sql, queryTimes=1) tdSql.checkData(0, 0, "binary9") tdSql.checkData(0, 1, "nchar9") sql = "select * from (select c8 as a, c9 as a, ts from t1 order by ts desc limit 10)t order by ts desc;" tdSql.query(sql, queryTimes=1) tdSql.checkData(0, 0, "binary9") tdSql.checkData(0, 1, "nchar9") sql = "select * from (select c8 as a, c9 as a, ts, t1 from t1 order by ts desc limit 10)t partition by t1 order by ts desc;" tdSql.query(sql, queryTimes=1) tdSql.checkData(0, 0, "binary9") tdSql.checkData(0, 1, "nchar9") sql = " select * from (select a.c8, b.c8, a.ts, a.t1,b.t1 from t1 a, t3 b where a.ts = b.ts order by a.ts)ttt" tdSql.query(sql, queryTimes=1) tdSql.checkData(0, 3, 1) tdSql.checkData(0, 4, 3) def run(self): self.init_data() self.test_select_asterisk_from_subquery_with_duplicate_aliasname() def stop(self): tdSql.close() tdLog.success(f"{__file__} successfully executed") event = threading.Event() tdCases.addLinux(__file__, TDTestCase()) tdCases.addWindows(__file__, TDTestCase())