add twa ci test

This commit is contained in:
54liuyao 2024-11-06 10:43:13 +08:00
parent ba31bf2c52
commit 379df5f5b2
2 changed files with 213 additions and 585 deletions

View File

@ -754,9 +754,6 @@ int32_t streamMetaRegisterTask(SStreamMeta* pMeta, int64_t ver, SStreamTask* pTa
int32_t val = atomic_add_fetch_32(&pMeta->numOfStreamTasks, 1);
}
// enable the scheduler for stream tasks
streamSetupScheduleTrigger(pTask);
*pAdded = true;
return code;
}
@ -1159,9 +1156,6 @@ void streamMetaLoadAllTasks(SStreamMeta* pMeta) {
continue;
}
// enable the scheduler for stream tasks after acquire the task RefId.
streamSetupScheduleTrigger(pTask);
stInfo("s-task:0x%x vgId:%d set refId:%"PRId64, (int32_t) id.taskId, vgId, pTask->id.refId);
if (pTask->info.fillHistory == 0) {
int32_t val = atomic_add_fetch_32(&pMeta->numOfStreamTasks, 1);

View File

@ -14,289 +14,42 @@ class TDTestCase:
tdLog.debug("start to execute %s" % __file__)
tdSql.init(conn.cursor(), logSql)
self.tdCom = tdCom
def get_source_firt_ts(self, table_name1):
tdSql.query(
f'select cast(first(ts) as bigint) from {table_name1} order by 1'
)
# getData don't support negative index
res_ts = tdSql.getData(0, 0)
return res_ts
def force_window_close(
self,
interval,
partition="tbname",
funciton_name="",
funciton_name_alias="",
delete=False,
fill_value=None,
fill_history_value=None,
case_when=None,
ignore_expired=1,
ignore_update=1,
):
# partition must be tbname, and not NONE.
tdLog.info(
f"*** testing stream force_window_close+interp+every: every: {interval}, partition: {partition}, fill_history: {fill_history_value}, fill: {fill_value}, delete: {delete}, case_when: {case_when} ***"
def get_source_last_ts(self, table_name1):
tdSql.query(
f'select cast(last(ts) as bigint) from {table_name1} order by 1'
)
self.tdCom.subtable = False
col_value_type = "Incremental" if partition == "c1" else "random"
custom_col_index = 1 if partition == "c1" else None
self.tdCom.custom_col_val = 0
self.delete = delete
self.tdCom.case_name = sys._getframe().f_code.co_name
self.tdCom.prepare_data(
interval=interval,
fill_history_value=fill_history_value,
custom_col_index=custom_col_index,
col_value_type=col_value_type,
# getData don't support negative index
res_ts = tdSql.getData(0, 0)
return res_ts
def get_stream_first_win_ts(self, table_name1):
tdSql.query(
f'select _wstart, count(*) from {table_name1} interval({self.tdCom.dataDict["interval"]}s) order by 1'
)
self.stb_name = self.tdCom.stb_name.replace(f"{self.tdCom.dbname}.", "")
self.ctb_name = self.tdCom.ctb_name.replace(f"{self.tdCom.dbname}.", "")
self.tb_name = self.tdCom.tb_name.replace(f"{self.tdCom.dbname}.", "")
self.stb_stream_des_table = f"{self.stb_name}{self.tdCom.des_table_suffix}"
self.ctb_stream_des_table = f"{self.ctb_name}{self.tdCom.des_table_suffix}"
self.tb_stream_des_table = f"{self.tb_name}{self.tdCom.des_table_suffix}"
if partition == "tbname":
if case_when:
stream_case_when_partition = case_when
else:
stream_case_when_partition = self.tdCom.partition_tbname_alias
partition_elm_alias = self.tdCom.partition_tbname_alias
elif partition == "c1":
if case_when:
stream_case_when_partition = case_when
else:
stream_case_when_partition = self.tdCom.partition_col_alias
partition_elm_alias = self.tdCom.partition_col_alias
elif partition == "abs(c1)":
partition_elm_alias = self.tdCom.partition_expression_alias
elif partition is None:
partition_elm_alias = '"no_partition"'
else:
partition_elm_alias = self.tdCom.partition_tag_alias
if partition == "tbname" or partition is None:
if case_when:
stb_subtable_value = (
f'concat(concat("{self.stb_name}_{self.tdCom.subtable_prefix}", {stream_case_when_partition}), "{self.tdCom.subtable_suffix}")'
if self.tdCom.subtable
else None
)
ctb_subtable_value = (
f'concat(concat("{self.ctb_name}_{self.tdCom.subtable_prefix}", {stream_case_when_partition}), "{self.tdCom.subtable_suffix}")'
if self.tdCom.subtable
else None
)
tb_subtable_value = (
f'concat(concat("{self.tb_name}_{self.tdCom.subtable_prefix}", {stream_case_when_partition}), "{self.tdCom.subtable_suffix}")'
if self.tdCom.subtable
else None
)
else:
stb_subtable_value = (
f'concat(concat("{self.stb_name}_{self.tdCom.subtable_prefix}", {partition_elm_alias}), "{self.tdCom.subtable_suffix}")'
if self.tdCom.subtable
else None
)
ctb_subtable_value = (
f'concat(concat("{self.ctb_name}_{self.tdCom.subtable_prefix}", {partition_elm_alias}), "{self.tdCom.subtable_suffix}")'
if self.tdCom.subtable
else None
)
tb_subtable_value = (
f'concat(concat("{self.tb_name}_{self.tdCom.subtable_prefix}", {partition_elm_alias}), "{self.tdCom.subtable_suffix}")'
if self.tdCom.subtable
else None
)
else:
stb_subtable_value = (
f'concat(concat("{self.stb_name}_{self.tdCom.subtable_prefix}", cast(cast(abs(cast({partition_elm_alias} as int)) as bigint) as varchar(100))), "{self.tdCom.subtable_suffix}")'
if self.tdCom.subtable
else None
)
ctb_subtable_value = (
f'concat(concat("{self.ctb_name}_{self.tdCom.subtable_prefix}", cast(cast(abs(cast({partition_elm_alias} as int)) as bigint) as varchar(100))), "{self.tdCom.subtable_suffix}")'
if self.tdCom.subtable
else None
)
tb_subtable_value = (
f'concat(concat("{self.tb_name}_{self.tdCom.subtable_prefix}", cast(cast(abs(cast({partition_elm_alias} as int)) as bigint) as varchar(100))), "{self.tdCom.subtable_suffix}")'
if self.tdCom.subtable
else None
)
if partition:
partition_elm = f"partition by {partition} {partition_elm_alias}"
else:
partition_elm = ""
if fill_value:
if "value" in fill_value.lower():
fill_value = "VALUE,1"
# create error stream
tdLog.info("create error stream")
sleep(10)
tdSql.error(
f"create stream itp_force_error_1 trigger force_window_close IGNORE EXPIRED 1 IGNORE UPDATE 0 into itp_force_error_1 as select _irowts,tbname,_isfilled,interp(c1,1) from {self.stb_name} partition by tbname every(5s) fill(prev) ;"
)
tdSql.error(
f"create stream itp_force_error_1 trigger force_window_close IGNORE EXPIRED 0 IGNORE UPDATE 1 into itp_force_error_1 as select _irowts,tbname,_isfilled,interp(c1,1) from {self.stb_name} partition by tbname every(5s) fill(prev) ;"
)
tdSql.error(
f"create stream itp_force_error_1 trigger at_once IGNORE EXPIRED 0 IGNORE UPDATE 1 into itp_force_error_1 as select _irowts,tbname,_isfilled,interp(c1,1) from {self.stb_name} partition by tbname every(5s) fill(prev) ;"
)
tdSql.error(
f"create stream itp_force_error_1 trigger force_window_close IGNORE EXPIRED 1 IGNORE UPDATE 0 into itp_force_error_1 as select _irowts,tbname,_isfilled,interp(c11,1) from {self.stb_name} partition by tbname every(5s) fill(prev) ;"
)
# function name : interp
trigger_mode = "force_window_close"
# # subtable is true
# create stream add :subtable_value=stb_subtable_value or subtable_value=ctb_subtable_value
# no subtable
# create stream super table and child table
tdLog.info("create stream super table and child table")
self.tdCom.create_stream(
stream_name=f"{self.stb_name}{self.tdCom.stream_suffix}",
des_table=self.stb_stream_des_table,
source_sql=f'select _irowts as irowts,tbname as table_name, _isfilled as isfilled, {funciton_name} as {funciton_name_alias} from {self.stb_name} {partition_elm} every({self.tdCom.dataDict["interval"]}s)',
trigger_mode=trigger_mode,
fill_value=fill_value,
fill_history_value=fill_history_value,
ignore_expired=ignore_expired,
ignore_update=ignore_update,
)
self.tdCom.create_stream(
stream_name=f"{self.ctb_name}{self.tdCom.stream_suffix}",
des_table=self.ctb_stream_des_table,
source_sql=f'select _irowts as irowts, tbname as table_name, _isfilled as isfilled, {funciton_name} as {funciton_name_alias} from {self.ctb_name} {partition_elm} every({self.tdCom.dataDict["interval"]}s)',
trigger_mode=trigger_mode,
fill_value=fill_value,
fill_history_value=fill_history_value,
ignore_expired=ignore_expired,
ignore_update=ignore_update,
)
# creat stream set filter of tag and tbname
tdLog.info("create stream with tag and tbname filter")
tag_t1_value = self.tdCom.tag_value_list[0]
where_tag = f"where t1 = {tag_t1_value}"
where_tbname = f'where tbname="{self.ctb_name}"'
# print(f"tag: {tag_t1_value}")
self.stb_stream_des_where_tag_table = (
f"{self.stb_name}_where_tag{self.tdCom.des_table_suffix}"
)
self.tdCom.create_stream(
stream_name=f"{self.stb_name}_where_tag{self.tdCom.stream_suffix}",
des_table=self.stb_stream_des_where_tag_table,
source_sql=f'select _irowts as irowts,tbname as table_name, _isfilled as isfilled, {funciton_name} as {funciton_name_alias} from {self.stb_name} {where_tag} {partition_elm} every({self.tdCom.dataDict["interval"]}s)',
trigger_mode=trigger_mode,
fill_value=fill_value,
fill_history_value=fill_history_value,
ignore_expired=ignore_expired,
ignore_update=ignore_update,
)
self.stb_stream_des_where_tbname_table = (
f"{self.stb_name}_where_tbname{self.tdCom.des_table_suffix}"
)
self.tdCom.create_stream(
stream_name=f"{self.stb_name}_where_tbname{self.tdCom.stream_suffix}",
des_table=self.stb_stream_des_where_tbname_table,
source_sql=f'select _irowts as irowts,tbname as table_name, _isfilled as isfilled, {funciton_name} as {funciton_name_alias} from {self.stb_name} {where_tbname} {partition_elm} every({self.tdCom.dataDict["interval"]}s)',
trigger_mode=trigger_mode,
fill_value=fill_value,
fill_history_value=fill_history_value,
ignore_expired=ignore_expired,
ignore_update=ignore_update,
)
# set partition by tag and column
self.stb_stream_des_partition_tag_table = (
f"{self.stb_name}_partition_tag{self.tdCom.des_table_suffix}"
)
self.stb_stream_des_partition_column1_table = (
f"{self.stb_name}_partition_column1{self.tdCom.des_table_suffix}"
)
self.stb_stream_des_partition_column2_table = (
f"{self.stb_name}_partition_column2{self.tdCom.des_table_suffix}"
)
if partition:
tdLog.info("create stream with partition by tag and tbname ")
partition_elm_new = f"partition by {partition}, t1"
self.tdCom.create_stream(
stream_name=f"{self.stb_name}_partition_tag{self.tdCom.stream_suffix}",
des_table=self.stb_stream_des_partition_tag_table,
source_sql=f'select _irowts as irowts, tbname as table_name, t1 as t_t1, _isfilled as isfilled, {funciton_name} as {funciton_name_alias} from {self.stb_name} {partition_elm_new} every({self.tdCom.dataDict["interval"]}s)',
trigger_mode=trigger_mode,
fill_value=fill_value,
fill_history_value=fill_history_value,
ignore_expired=ignore_expired,
ignore_update=ignore_update,
)
partition_elm_new = f"partition by {partition}, c1"
self.tdCom.create_stream(
stream_name=f"{self.stb_name}_partition_column1{self.tdCom.stream_suffix}",
des_table=f"{self.stb_name}_partition_column1{self.tdCom.des_table_suffix}",
source_sql=f'select _irowts as irowts, tbname as table_name, c1 as c_c1, _isfilled as isfilled, {funciton_name} as {funciton_name_alias} from {self.stb_name} {partition_elm_new} every({self.tdCom.dataDict["interval"]}s)',
trigger_mode=trigger_mode,
fill_value=fill_value,
fill_history_value=fill_history_value,
ignore_expired=ignore_expired,
ignore_update=ignore_update,
)
partition_elm_new = f"partition by {partition}, c2"
self.tdCom.create_stream(
stream_name=f"{self.stb_name}_partition_column2{self.tdCom.stream_suffix}",
des_table=f"{self.stb_name}_partition_column2{self.tdCom.des_table_suffix}",
source_sql=f'select _irowts as irowts, tbname as table_name, c2 as c_c2, _isfilled as isfilled, {funciton_name} as {funciton_name_alias} from {self.stb_name} {partition_elm_new} every({self.tdCom.dataDict["interval"]}s)',
trigger_mode=trigger_mode,
fill_value=fill_value,
fill_history_value=fill_history_value,
ignore_expired=ignore_expired,
ignore_update=ignore_update,
)
if fill_value:
if "value" in fill_value.lower():
fill_value = "VALUE,1"
# create stream general table
tdLog.info("create stream general table")
self.tdCom.create_stream(
stream_name=f"{self.tb_name}{self.tdCom.stream_suffix}",
des_table=self.tb_stream_des_table,
source_sql=f'select _irowts as irowts,tbname as table_name, _isfilled as isfilled, {funciton_name} as {funciton_name_alias} from {self.tb_name} every({self.tdCom.dataDict["interval"]}s)',
trigger_mode=trigger_mode,
fill_value=fill_value,
fill_history_value=fill_history_value,
ignore_expired=ignore_expired,
ignore_update=ignore_update,
)
# wait and check stream_task status is ready
time.sleep(self.tdCom.dataDict["interval"])
tdSql.query("show streams")
tdLog.info(f"tdSql.queryResult:{tdSql.queryResult},tdSql.queryRows:{tdSql.queryRows}")
localQueryResult = tdSql.queryResult
for stream_number in range(tdSql.queryRows):
stream_name = localQueryResult[stream_number][0]
tdCom.check_stream_task_status(
stream_name=stream_name, vgroups=2, stream_timeout=20,check_wal_info=False
)
time.sleep(self.tdCom.dataDict["interval"])
time.sleep(30)
# insert data
res_ts = tdSql.getData(0, 0)
return res_ts
def insert_data(self, custom_col_index, col_value_type):
self.tdCom.date_time = self.tdCom.genTs(precision=self.tdCom.precision)[0]
start_time = self.tdCom.date_time
time.sleep(1)
start_force_ts = str(0)
min_new_ts = 0
for i in range(self.tdCom.range_count):
cur_time = str(self.tdCom.date_time + self.tdCom.dataDict["interval"])
ts_value = (
str(self.tdCom.date_time + self.tdCom.dataDict["interval"])
+ f"+{i*10}s"
cur_time + f"+{i * 5 + 30}s"
)
# print(ts_value)
if start_force_ts == "0":
start_force_ts = ts_value
if min_new_ts == 0:
min_new_ts = ts_value
ts_cast_delete_value = self.tdCom.time_cast(ts_value)
self.tdCom.sinsert_rows(
tbname=self.tdCom.ctb_name,
@ -304,7 +57,7 @@ class TDTestCase:
custom_col_index=custom_col_index,
col_value_type=col_value_type,
)
if i % 2 == 0:
if i % 2 == 0 and min_new_ts != ts_value:
self.tdCom.sinsert_rows(
tbname=self.tdCom.ctb_name,
ts_value=ts_value,
@ -322,7 +75,7 @@ class TDTestCase:
custom_col_index=custom_col_index,
col_value_type=col_value_type,
)
if i % 2 == 0:
if i % 2 == 0 and min_new_ts != ts_value:
self.tdCom.sinsert_rows(
tbname=self.tdCom.tb_name,
ts_value=ts_value,
@ -334,327 +87,208 @@ class TDTestCase:
tbname=self.tdCom.tb_name, start_ts=ts_cast_delete_value
)
self.tdCom.date_time += 1
if partition:
partition_elm = f"partition by {partition}"
else:
partition_elm = ""
cur_time = str(self.tdCom.date_time + self.tdCom.dataDict["interval"])
max_new_ts = (cur_time + f"+{self.tdCom.range_count * 10 + 30}s")
self.tdCom.sinsert_rows(
tbname=self.tdCom.ctb_name,
ts_value=max_new_ts,
custom_col_index=custom_col_index,
col_value_type=col_value_type,
)
self.tdCom.sinsert_rows(
tbname=self.tdCom.tb_name,
ts_value=max_new_ts,
custom_col_index=custom_col_index,
col_value_type=col_value_type,
)
return (min_new_ts, max_new_ts)
if not fill_value:
for tbname in [self.stb_name, self.ctb_name, self.tb_name]:
if tbname != self.tb_name:
self.tdCom.check_query_data(
f"select wstart, {self.tdCom.stb_output_select_str} from {tbname}{self.tdCom.des_table_suffix} order by wstart",
f'select _wstart AS wstart, {self.tdCom.stb_source_select_str} from {tbname} {partition_elm} interval({self.tdCom.dataDict["interval"]}s) order by wstart',
sorted=True,
)
else:
self.tdCom.check_query_data(
f"select wstart, {self.tdCom.tb_output_select_str} from {tbname}{self.tdCom.des_table_suffix} order by wstart",
f'select _wstart AS wstart, {self.tdCom.tb_source_select_str} from {tbname} {partition_elm} interval({self.tdCom.dataDict["interval"]}s) order by wstart',
sorted=True,
)
def insert_disorder_data(self, custom_col_index, col_value_type):
min_ts = self.get_source_firt_ts(self.tb_name)
max_ts = self.get_source_last_ts(self.tb_name)
min_ts_str = str(min_ts) + f"-10000s"
max_ts_str = str(max_ts) + f"+10000s"
self.tdCom.sinsert_rows(
tbname=self.tdCom.ctb_name,
ts_value=min_ts_str,
custom_col_index=custom_col_index,
col_value_type=col_value_type,
)
self.tdCom.sinsert_rows(
tbname=self.tdCom.tb_name,
ts_value=min_ts_str,
custom_col_index=custom_col_index,
col_value_type=col_value_type,
)
self.tdCom.sinsert_rows(
tbname=self.tdCom.ctb_name,
ts_value=max_ts_str,
custom_col_index=custom_col_index,
col_value_type=col_value_type,
)
self.tdCom.sinsert_rows(
tbname=self.tdCom.tb_name,
ts_value=max_ts_str,
custom_col_index=custom_col_index,
col_value_type=col_value_type,
)
if self.tdCom.subtable:
for tname in [self.stb_name, self.ctb_name]:
group_id = self.tdCom.get_group_id_from_stb(f"{tname}_output")
tdSql.query(f"select * from {self.ctb_name}")
ptn_counter = 0
for c1_value in tdSql.queryResult:
if partition == "c1":
tbname = self.tdCom.get_subtable_wait(
f"{tname}_{self.tdCom.subtable_prefix}{abs(c1_value[1])}{self.tdCom.subtable_suffix}"
)
tdSql.query(f"select count(*) from `{tbname}`")
elif partition is None:
tbname = self.tdCom.get_subtable_wait(
f"{tname}_{self.tdCom.subtable_prefix}no_partition{self.tdCom.subtable_suffix}"
)
tdSql.query(f"select count(*) from `{tbname}`")
elif partition == "abs(c1)":
abs_c1_value = abs(c1_value[1])
tbname = self.tdCom.get_subtable_wait(
f"{tname}_{self.tdCom.subtable_prefix}{abs_c1_value}{self.tdCom.subtable_suffix}"
)
tdSql.query(f"select count(*) from `{tbname}`")
elif partition == "tbname" and ptn_counter == 0:
tbname = self.tdCom.get_subtable_wait(
f"{tname}_{self.tdCom.subtable_prefix}{self.ctb_name}{self.tdCom.subtable_suffix}_{tname}_output_{group_id}"
)
tdSql.query(f"select count(*) from `{tbname}`")
ptn_counter += 1
tdSql.checkEqual(tdSql.queryResult[0][0] > 0, True)
group_id = self.tdCom.get_group_id_from_stb(f"{self.tb_name}_output")
tdSql.query(f"select * from {self.tb_name}")
ptn_counter = 0
for c1_value in tdSql.queryResult:
if partition == "c1":
tbname = self.tdCom.get_subtable_wait(
f"{self.tb_name}_{self.tdCom.subtable_prefix}{abs(c1_value[1])}{self.tdCom.subtable_suffix}"
)
tdSql.query(f"select count(*) from `{tbname}`")
elif partition is None:
tbname = self.tdCom.get_subtable_wait(
f"{self.tb_name}_{self.tdCom.subtable_prefix}no_partition{self.tdCom.subtable_suffix}"
)
tdSql.query(f"select count(*) from `{tbname}`")
elif partition == "abs(c1)":
abs_c1_value = abs(c1_value[1])
tbname = self.tdCom.get_subtable_wait(
f"{self.tb_name}_{self.tdCom.subtable_prefix}{abs_c1_value}{self.tdCom.subtable_suffix}"
)
tdSql.query(f"select count(*) from `{tbname}`")
elif partition == "tbname" and ptn_counter == 0:
tbname = self.tdCom.get_subtable_wait(
f"{self.tb_name}_{self.tdCom.subtable_prefix}{self.tb_name}{self.tdCom.subtable_suffix}_{self.tb_name}_output_{group_id}"
)
tdSql.query(f"select count(*) from `{tbname}`")
ptn_counter += 1
def force_window_close(
self,
interval,
partition="tbname",
delete=False,
fill_value=None,
filter=None
):
# partition must be tbname, and not NONE.
tdLog.info(
f"*** testing stream force_window_close + interval + fill. partition: {partition}, interval: {interval}, fill: {fill_value}, delete: {delete} ***"
)
fwc_downsampling_function_list = ["min(c1)", "max(c2)", "sum(c3)", "twa(c7)", "count(c8)", "elapsed(ts)", "timediff(1, 0, 1h)", "timezone()","min(t1)", "max(t2)", "sum(t3)",
"twa(t7)", "count(t8)"]
fwc_stb_output_select_str = ','.join(list(map(lambda x:f'`{x}`', fwc_downsampling_function_list)))
fwc_tb_output_select_str = ','.join(list(map(lambda x:f'`{x}`', fwc_downsampling_function_list[0:7])))
fwc_stb_source_select_str = ','.join(fwc_downsampling_function_list)
fwc_tb_source_select_str = ','.join(fwc_downsampling_function_list[0:7])
fill_history_value = 0
ignore_expired = 1
ignore_update = 1
self.tdCom.subtable = False
col_value_type = "Incremental" if partition == "c1" else "random"
custom_col_index = 1 if partition == "c1" else None
self.tdCom.custom_col_val = 0
self.delete = delete
self.tdCom.case_name = sys._getframe().f_code.co_name
self.tdCom.prepare_data(
interval=interval,
custom_col_index=custom_col_index,
col_value_type=col_value_type,
)
self.stb_name = self.tdCom.stb_name.replace(f"{self.tdCom.dbname}.", "")
self.ctb_name = self.tdCom.ctb_name.replace(f"{self.tdCom.dbname}.", "")
self.tb_name = self.tdCom.tb_name.replace(f"{self.tdCom.dbname}.", "")
self.stb_stream_des_table = f"{self.stb_name}{self.tdCom.des_table_suffix}"
self.ctb_stream_des_table = f"{self.ctb_name}{self.tdCom.des_table_suffix}"
self.tb_stream_des_table = f"{self.tb_name}{self.tdCom.des_table_suffix}"
if partition:
partition_elm = f"partition by {partition}"
else:
partition_elm = ""
query_partition_elm = partition_elm
tdSql.checkEqual(tdSql.queryResult[0][0] > 0, True)
if fill_value:
end_date_time = self.tdCom.date_time
final_range_count = self.tdCom.range_count
history_ts = (
str(start_time)
+ f'-{self.tdCom.dataDict["interval"]*(final_range_count+2)}s'
)
start_ts = self.tdCom.time_cast(history_ts, "-")
future_ts = (
str(end_date_time)
+ f'+{self.tdCom.dataDict["interval"]*(final_range_count+2)}s'
)
end_ts = self.tdCom.time_cast(future_ts)
self.tdCom.sinsert_rows(tbname=self.ctb_name, ts_value=history_ts)
self.tdCom.sinsert_rows(tbname=self.tb_name, ts_value=history_ts)
self.tdCom.sinsert_rows(tbname=self.ctb_name, ts_value=future_ts)
self.tdCom.sinsert_rows(tbname=self.tb_name, ts_value=future_ts)
self.tdCom.date_time = start_time
# update
history_ts = (
str(start_time)
+ f'-{self.tdCom.dataDict["interval"]*(final_range_count+2)}s'
)
start_ts = self.tdCom.time_cast(history_ts, "-")
future_ts = (
str(end_date_time)
+ f'+{self.tdCom.dataDict["interval"]*(final_range_count+2)}s'
)
end_ts = self.tdCom.time_cast(future_ts)
self.tdCom.sinsert_rows(tbname=self.ctb_name, ts_value=history_ts)
self.tdCom.sinsert_rows(tbname=self.tb_name, ts_value=history_ts)
self.tdCom.sinsert_rows(tbname=self.ctb_name, ts_value=future_ts)
self.tdCom.sinsert_rows(tbname=self.tb_name, ts_value=future_ts)
if "value" in fill_value.lower():
stb_fill_value='VALUE,1,2,3,4,5,6,1,2,3,4,5'
tb_fill_value='VALUE,1,2,3,4,5,6'
else:
stb_fill_value=fill_value
tb_fill_value=fill_value
query_stb_fill_elm = f"fill({stb_fill_value})"
query_tb_fill_elm = f"fill({tb_fill_value})"
else:
query_stb_fill_elm = ""
query_tb_fill_elm = ""
stb_fill_value = None
tb_fill_value=None
where_elm = "where 1=1"
if filter:
where_elm = f" and {filter}"
# get query time range using interval count windows
tdSql.query(
f'select _wstart, _wend ,last(ts) from {self.stb_name} where ts >= {start_force_ts} and ts <= {end_ts} partition by tbname interval({self.tdCom.dataDict["interval"]}s)fill ({fill_value}) '
)
# getData don't support negative index
end_new_ts = tdSql.getData(tdSql.queryRows - 1, 1)
end_last_but_one_ts = tdSql.getData(tdSql.queryRows - 2, 1)
# source data include that fill valuse is null and "_isfilled" column of the stream output is false
tdSql.execute(
f'insert into {self.ctb_name} (ts,c1) values("{end_new_ts}",-102) '
)
tdSql.execute(
f'insert into {self.tb_name} (ts,c1) values("{end_new_ts}",-51) '
)
tdSql.execute(
f'insert into {self.ctb_name} (ts,c1) values("{end_last_but_one_ts}",NULL) '
)
trigger_mode = "force_window_close"
for i in range(self.tdCom.range_count):
ts_value = (
str(self.tdCom.date_time + self.tdCom.dataDict["interval"])
+ f"+{i*10}s"
)
ts_cast_delete_value = self.tdCom.time_cast(ts_value)
self.tdCom.sinsert_rows(tbname=self.ctb_name, ts_value=ts_value)
self.tdCom.date_time += 1
self.tdCom.sinsert_rows(tbname=self.tb_name, ts_value=ts_value)
self.tdCom.date_time += 1
if self.delete:
self.tdCom.sdelete_rows(
tbname=self.ctb_name,
start_ts=self.tdCom.time_cast(start_time),
end_ts=ts_cast_delete_value,
)
self.tdCom.sdelete_rows(
tbname=self.tb_name,
start_ts=self.tdCom.time_cast(start_time),
end_ts=ts_cast_delete_value,
)
# no subtable
# create stream super table and child table
tdLog.info("create stream super table and child table")
self.tdCom.create_stream(
stream_name=f'{self.stb_name}{self.tdCom.stream_suffix}',
des_table=self.stb_stream_des_table,
source_sql=f'select _wstart AS wstart, {fwc_stb_source_select_str} from {self.stb_name} {where_elm} {partition_elm} interval({self.tdCom.dataDict["interval"]}s)',
trigger_mode=trigger_mode,
fill_value=stb_fill_value,
fill_history_value=fill_history_value,
ignore_expired=ignore_expired,
ignore_update=ignore_update,
)
# wait for the stream to process the data
# print(self.tdCom.dataDict["interval"]*(final_range_count+2))
time.sleep(self.tdCom.dataDict["interval"] * (final_range_count + 2))
self.tdCom.create_stream(
stream_name=f"{self.tb_name}{self.tdCom.stream_suffix}",
des_table=self.tb_stream_des_table,
source_sql=f'select _wstart AS wstart, {fwc_tb_source_select_str} from {self.tb_name} {where_elm} {partition_elm} interval({self.tdCom.dataDict["interval"]}s)',
trigger_mode=trigger_mode,
fill_value=tb_fill_value,
fill_history_value=fill_history_value,
ignore_expired=ignore_expired,
ignore_update=ignore_update,
)
# check the data
for tbname in [self.stb_name, self.ctb_name, self.tb_name]:
tdLog.info(f"tbname:{tbname}")
tdSql.query(
f'select _wstart, _wend ,last(ts) from {tbname} where ts >= {start_force_ts} and ts <= {end_ts} partition by tbname interval({self.tdCom.dataDict["interval"]}s)fill ({fill_value}) '
)
start_new_ts = tdSql.getData(0, 1)
if tbname == self.ctb_name:
if partition == "tbname":
# check data for child table
tdLog.info("check data for child table ")
self.tdCom.check_query_data(
f'select irowts, table_name, isfilled, {funciton_name_alias} from {tbname}{self.tdCom.des_table_suffix} where irowts >= {start_force_ts} and irowts <= "{end_new_ts}" order by irowts',
f'select _irowts as irowts ,tbname as table_name, _isfilled as isfilled , {funciton_name} as {funciton_name_alias} from {tbname} partition by {partition} range("{start_new_ts}","{end_new_ts}") every({self.tdCom.dataDict["interval"]}s) fill ({fill_value}) order by irowts',
fill_value=fill_value,
)
elif tbname == self.stb_name:
if partition == "tbname":
# check data for super table
tdLog.info("check data for super table")
self.tdCom.check_query_data(
f'select irowts, table_name, isfilled, {funciton_name_alias} from {tbname}{self.tdCom.des_table_suffix} where irowts >= {start_force_ts} and irowts <= "{end_new_ts}" order by irowts',
f'select _irowts as irowts ,tbname as table_name, _isfilled as isfilled , {funciton_name} as {funciton_name_alias} from {tbname} partition by {partition} range("{start_new_ts}","{end_new_ts}") every({self.tdCom.dataDict["interval"]}s) fill ({fill_value}) order by irowts',
fill_value=fill_value,
)
# tag and tbname filter
tdLog.info("check data for tag and tbname filter")
self.tdCom.check_query_data(
f'select irowts, table_name, isfilled, {funciton_name_alias} from {self.stb_stream_des_where_tag_table} where irowts >= {start_force_ts} and irowts <= "{end_new_ts}" order by irowts',
f'select _irowts as irowts ,tbname as table_name, _isfilled as isfilled , {funciton_name} as {funciton_name_alias} from {tbname} {where_tag} and ts >= {start_force_ts} partition by {partition} range("{start_new_ts}","{end_new_ts}") every({self.tdCom.dataDict["interval"]}s) fill ({fill_value}) order by irowts',
fill_value=fill_value,
)
self.tdCom.check_query_data(
f'select irowts, table_name, isfilled, {funciton_name_alias} from {self.stb_stream_des_where_tbname_table} where irowts >= {start_force_ts} and irowts <= "{end_new_ts}" order by irowts',
f'select _irowts as irowts ,tbname as table_name, _isfilled as isfilled , {funciton_name} as {funciton_name_alias} from {tbname} {where_tbname} partition by {partition} range("{start_new_ts}","{end_new_ts}") every({self.tdCom.dataDict["interval"]}s) fill ({fill_value}) order by irowts',
fill_value=fill_value,
)
# check partition by tag and column(c1 or c2)
tdLog.info("check data for partition by tag and column")
self.tdCom.check_query_data(
f'select irowts, table_name, t_t1, isfilled, {funciton_name_alias} from {self.stb_stream_des_partition_tag_table} where irowts >= {start_force_ts} and irowts <= "{end_new_ts}" order by t_t1, irowts',
f'select _irowts as irowts ,tbname as table_name, t1 as t_t1, _isfilled as isfilled , {funciton_name} as {funciton_name_alias} from {tbname} {where_tbname} partition by {partition},t1 range("{start_new_ts}","{end_new_ts}") every({self.tdCom.dataDict["interval"]}s) fill ({fill_value}) order by t_t1, irowts',
fill_value=fill_value,
)
if fill_value == "PREV":
self.tdCom.check_query_data(
f'select irowts, c_c1, isfilled, {funciton_name_alias} from {self.stb_stream_des_partition_column1_table} where irowts >= {start_force_ts} and irowts <= "{end_new_ts}" order by c_c1, irowts',
f'select _irowts as irowts , c1 as c_c1, _isfilled as isfilled , {funciton_name} as {funciton_name_alias} from {tbname} {where_tbname} and ts >= {start_force_ts} partition by {partition},c1 range("{start_new_ts}","{end_new_ts}") every({self.tdCom.dataDict["interval"]}s) fill ({fill_value}) order by c_c1, irowts',
fill_value=fill_value,
)
self.tdCom.check_query_data(
f'select irowts, c_c2, isfilled, {funciton_name_alias} from {self.stb_stream_des_partition_column2_table} where irowts >= {start_force_ts} and irowts <= "{end_new_ts}" order by c_c2, irowts',
f'select _irowts as irowts , c2 as c_c2, _isfilled as isfilled , {funciton_name} as {funciton_name_alias} from {tbname} {where_tbname} and ts >= {start_force_ts} partition by {partition},c2 range("{start_new_ts}","{end_new_ts}") every({self.tdCom.dataDict["interval"]}s) fill ({fill_value}) order by c_c2, irowts',
fill_value=fill_value,
)
else:
if partition == "tbname":
# check data for general table
self.tdCom.check_query_data(
f'select irowts, isfilled, {funciton_name_alias} from {tbname}{self.tdCom.des_table_suffix} where irowts >= {start_force_ts} and irowts <= "{end_new_ts}" order by irowts',
f'select _irowts as irowts , _isfilled as isfilled , {funciton_name} as {funciton_name_alias} from {tbname} partition by {partition} range("{start_new_ts}","{end_new_ts}") every({self.tdCom.dataDict["interval"]}s) fill ({fill_value}) order by irowts',
fill_value=fill_value,
)
# Recreate a sub-table that meets the filtering "where_tag" and check if the streaming results are automatically included within it."
where_tag_ctbname = f"{self.ctb_name}_where_tag"
where_tag_ctbname_other_tag = f"{self.ctb_name}_where_tag_1"
tag_t1_value_other = abs(tag_t1_value)-1
tdSql.execute(
f"create table {where_tag_ctbname} using {self.stb_name} (t1) tags({tag_t1_value}) "
# wait and check stream_task status is ready
tdSql.query("show streams")
tdLog.info(f"tdSql.queryResult:{tdSql.queryResult},tdSql.queryRows:{tdSql.queryRows}")
localQueryResult = tdSql.queryResult
for stream_number in range(tdSql.queryRows):
stream_name = localQueryResult[stream_number][0]
tdCom.check_stream_task_status(
stream_name=stream_name, vgroups=2, stream_timeout=20,check_wal_info=False
)
tdSql.execute(
f"create table {where_tag_ctbname_other_tag} using {self.stb_name} (t1) tags({tag_t1_value_other}) "
)
where_tag_timestamp = self.tdCom.genTs(precision=self.tdCom.precision)[0]
where_tag_ts_start_value = str(where_tag_timestamp) + "+2s"
self.tdCom.sinsert_rows(
tbname=where_tag_ctbname, ts_value=where_tag_ts_start_value
)
self.tdCom.sinsert_rows(
tbname=where_tag_ctbname_other_tag, ts_value=where_tag_ts_start_value
)
time.sleep(self.tdCom.dataDict["interval"])
for _ in range(self.tdCom.dataDict["interval"]):
tdSql.query(
f"select distinct(table_name) from {self.stb_stream_des_where_tag_table} where table_name=\"{where_tag_ctbname}\""
)
if tdSql.queryRows > 0:
if tdSql.checkDataNotExit(0,0, where_tag_ctbname):
break
else:
time.sleep(1)
time.sleep(self.tdCom.dataDict["interval"])
time.sleep(20)
if self.delete:
self.tdCom.sdelete_rows(
tbname=self.ctb_name, start_ts=start_ts, end_ts=ts_cast_delete_value
)
self.tdCom.sdelete_rows(
tbname=self.tb_name, start_ts=start_ts, end_ts=ts_cast_delete_value
)
for tbname in [self.stb_name, self.ctb_name, self.tb_name]:
if tbname != self.tb_name:
if "value" in fill_value.lower():
fill_value = (
"VALUE,1,2,3,6,7,8,9,10,11,1,2,3,4,5,6,7,8,9,10,11"
)
if partition == "tbname":
self.tdCom.check_query_data(
f"select wstart, {self.tdCom.fill_stb_output_select_str} from {tbname}{self.tdCom.des_table_suffix} order by wstart",
f'select _wstart AS wstart, {self.tdCom.fill_stb_source_select_str} from {tbname} where ts >= {start_ts.replace("-", "+")} and ts <= {end_ts} partition by {partition} interval({self.tdCom.dataDict["interval"]}s) fill ({fill_value}) order by wstart',
fill_value=fill_value,
)
else:
self.tdCom.check_query_data(
f"select wstart, {self.tdCom.fill_stb_output_select_str} from {tbname}{self.tdCom.des_table_suffix} order by wstart,`min(c1)`",
f'select * from (select _wstart AS wstart, {self.tdCom.fill_stb_source_select_str} from {tbname} where ts >= {start_ts} and ts <= {end_ts} partition by {partition} interval({self.tdCom.dataDict["interval"]}s) fill ({fill_value}) order by wstart) where `min(c1)` is not Null order by wstart,`min(c1)`',
fill_value=fill_value,
)
# insert data
tdLog.info("insert data")
start_new_ts, temp = self.insert_data(custom_col_index, col_value_type)
time.sleep(self.tdCom.dataDict["interval"] * 2)
tdLog.info("insert data")
temp, end_new_ts = self.insert_data(custom_col_index, col_value_type)
else:
if "value" in fill_value.lower():
fill_value = "VALUE,1,2,3,6,7,8,9,10,11"
if partition == "tbname":
self.tdCom.check_query_data(
f"select wstart, {self.tdCom.fill_tb_output_select_str} from {tbname}{self.tdCom.des_table_suffix} order by wstart",
f'select _wstart AS wstart, {self.tdCom.fill_tb_source_select_str} from {tbname} where ts >= {start_ts.replace("-", "+")} and ts <= {end_ts} partition by {partition} interval({self.tdCom.dataDict["interval"]}s) fill ({fill_value}) order by wstart',
fill_value=fill_value,
)
else:
self.tdCom.check_query_data(
f"select wstart, {self.tdCom.fill_tb_output_select_str} from {tbname}{self.tdCom.des_table_suffix} order by wstart,`min(c1)`",
f'select * from (select _wstart AS wstart, {self.tdCom.fill_tb_source_select_str} from {tbname} where ts >= {start_ts} and ts <= {end_ts} partition by {partition} interval({self.tdCom.dataDict["interval"]}s) fill ({fill_value}) order by wstart) where `min(c1)` is not Null order by wstart,`min(c1)`',
fill_value=fill_value,
)
#history and future
self.insert_disorder_data(custom_col_index, col_value_type)
time.sleep(self.tdCom.dataDict["interval"] * 6 * 2)
tdLog.info("check data")
# check the data
where_elm = f'{where_elm} and ts >= {start_new_ts} and ts <= {end_new_ts}'
for tbname in [self.stb_name, self.tb_name]:
if fill_value:
query_first_win_ts = self.get_stream_first_win_ts(tbname)
query_where_elm = f'where wstart >= "{query_first_win_ts}"'
stream_where_elm = f'where wstart <= {end_new_ts}'
else:
query_where_elm = ""
stream_where_elm = ""
# check data
tdLog.info(f"check data for table {tbname}")
if tbname == self.stb_name:
self.tdCom.check_query_data(
f'select wstart, {fwc_stb_output_select_str} from {tbname}{self.tdCom.des_table_suffix} {stream_where_elm} order by wstart',
f'select * from (select _wstart AS wstart, {fwc_stb_source_select_str} from {tbname} {where_elm} {query_partition_elm} interval({self.tdCom.dataDict["interval"]}s) {query_stb_fill_elm} order by wstart) {query_where_elm}',
sorted=True
)
else:
self.tdCom.check_query_data(
f'select wstart, {fwc_tb_output_select_str} from {tbname}{self.tdCom.des_table_suffix} {stream_where_elm} order by wstart',
f'select * from (select _wstart AS wstart, {fwc_tb_source_select_str} from {tbname} {where_elm} {query_partition_elm} interval({self.tdCom.dataDict["interval"]}s) {query_tb_fill_elm} order by wstart) {query_where_elm}',
sorted=True
)
def run(self):
for fill_value in ["PREV", "VALUE","NULL"]:
for fill_value in ["VALUE", "NULL", "PREV", None]:
self.force_window_close(
interval=10,
interval=5,
partition="tbname",
funciton_name="interp(c1)",
funciton_name_alias="intp_c1",
delete=False,
ignore_update=1,
delete=True,
fill_value=fill_value,
)
self.force_window_close(
interval=8,
partition="tbname",
funciton_name="interp(c1)",
funciton_name_alias="intp_c1",
delete=False,
ignore_update=1,
fill_value="PREV",
)
# self.force_window_close(interval=random.randint(10, 15), partition="c1", ignore_update=1)
# self.force_window_close(interval=random.randint(10, 15), partition="abs(c1)", ignore_update=1)
# self.force_window_close(interval=random.randint(10, 15), partition=None, delete=True)
# self.force_window_close(interval=random.randint(10, 15), partition=self.tdCom.stream_case_when_tbname, case_when=f'case when {self.tdCom.stream_case_when_tbname} = tbname then {self.tdCom.partition_tbname_alias} else tbname end')
# self.force_window_close(interval=random.randint(10, 15), partition="tbname", fill_history_value=1, fill_value="NULL")
# for fill_value in ["NULL", "PREV", "NEXT", "LINEAR", "VALUE,1,2,3,4,5,6,7,8,9,10,11,1,2,3,4,5,6,7,8,9,10,11"]:
# # for fill_value in ["PREV", "NEXT", "LINEAR", "VALUE,1,2,3,4,5,6,7,8,9,10,11,1,2,3,4,5,6,7,8,9,10,11"]:
# self.at_once_interval(interval=random.randint(10, 15), partition="tbname", fill_value=fill_value)
# self.at_once_interval(interval=random.randint(10, 15), partition="tbname", fill_value=fill_value, delete=True)
def stop(self):
tdLog.info("stop========================================")
tdSql.close()
tdLog.success(f"{__file__} successfully executed")