TDengine/source/libs/executor/src/externalwindowoperator.c
Pan Wei 0b46238d75
enh: new stream (#32112)
* feat: [TS-6100] Fix virtual supertable plan.

* add case stream1 for vehicle

* test(stream): photovoltaic scene #TD-36783

* fix: errlog

* test: update test case.

* test: update test case.

* modify case (#32016)

* fix: check node type

* feat: [TS-6100] set proper curr clause when translating stream trigger.

* test asan error

* feat: [TD-36758] Fix MAX_DELAY wrong precision.

* case: restore bug9 case

* test asan error

* test: add cases

* test: add test for manual recalc.

* add case

* add case

* test: event manual pass.

* test: add case into ci

* feat: [TD-36761] Remove COMMENT in stream tag def

* test(stream): ci test replay #TD-36808

* test: make manual ok.

* test: reproduce bugs

* case: JIRA  TD-36815 set ignore this check

* JIRA TD-36815 ignore

* feat: [TD-36786] collect now and today to timeRangeExpr

* fix: stack overflow

* feat: [TD-36709] Optimize create vtable and table's error msg

* test: reproduce bugs

* test: add bug 11.

* add case

* test: add cases

* test: add vehicle bug1

* test: add cases

* test: add cases (#32041)

* test: cases

* test(stream): photovoltaic scene testing #TD-36783

* test: add test for manual.

* fix: force output

* add case

* add case

* remove wrong path case

* test(stream): pv scene #TD-36783

* case : add bug2 for vehicle

* test: reproduce cases

* test(stream): add case #TD-36808

* test: update test case.

* test: move case to new test frame.

* fix: output

* enh: support indef rows functions in extwindow (#32047)

* enh: add operator reset func

* fix: merge join reset issue

* fix: memory issues

* fix: add debug assert

* fix: memory issues

* fix: memory leak

* fix: memory issues

* fix taos log miss

* fix: case issue

* fix: case issue

* fix: case issues

* fix: drop dnode issue

* fix: memory issues

* fix: memory issues

* fix: memory leak issues

* fix: recalculate time range issue

* fix: add debug log

* fix: memory issues

* fix: enable case asan

* Update streamlist_for_ci.task

* fix: case asan issue

* fix: stream name issue

* fix: external window compile issues

* fix: deploy memory issue

* fix: ahandle issue

* fix: ahandle issue

* fix: ahandle issue

* fix: virtual table reader list issue

* fix: log info

* fix: msg error

* fix: virtual table addr list issue

* fix: memory issues

* fix: memory leak issue

* fix: memory issues

* fix: memory free issues

* fix: memory issues

* fix: snode deploy issue

* fix: mnode reader issue

* fix: memory issues

* fix: add debug test

* enh: add ignore nodata trigger

* fix: memory leaks

* fix: configuration issue

* fix: memory issue

* fix: external window issue

* fix: external window issues

* fix: external window placeholder issue

* fix: placeholder function init issues

* fix: memory leak issue

* fix: add debug log

* fix: compile issues

* fix: double free issue

* fix: runner addr update issue

* fix: msg rsp issue

* fix: external window reset issue

* fix: configuration issue

* fix: deploy msg issue

* fix: compile issue

* fix: external window idx issue

* fix: ci issues

* fix: ci case issues

* fix: drop dnode issue

* fix: add debug log

* fix: conflict

* fix: create stream if not exists issue

* fix: ahandle memory leak

* fix: case issue

* fix: exchange issues

* fix: crash issue

* fix: exchange prefetch issue

* fix: snode quit issue

* enh: support indef rows func

* fix: crash issues

* Fix external window collect vector function

* fix: external window indef rows issues

---------

Co-authored-by: huohong <sallyhuo@taosdata.com>
Co-authored-by: Jing Sima <simondominic9997@outlook.com>

* fix: log

* feat: [TD-36838] Fix trigger virtual super table

* test: vehicle add bug3

* test: vehicle add bug3 pass

* test: vehicle add bug3 2

* test: vehicle add bug3 write end

* fix(stream): fix filter of virtual tables in trigger task

* fix(stream): add virtual table pseudo column value request

* fix(stream): get tag for virtual child table

* fix(stream): fetch virtual table pseudo columns in trigger

* fix(stream): get tag value for vtable

* fix(stream): ignore trigger control msg at server close down

* fix(stream): retry pull/calc request in trigger task

* test(stream): modify and disable some cases

* fix: external window issue (#32052)

* enh: add operator reset func

* fix: merge join reset issue

* fix: memory issues

* fix: add debug assert

* fix: memory issues

* fix: memory leak

* fix: memory issues

* fix taos log miss

* fix: case issue

* fix: case issue

* fix: case issues

* fix: drop dnode issue

* fix: memory issues

* fix: memory issues

* fix: memory leak issues

* fix: recalculate time range issue

* fix: add debug log

* fix: memory issues

* fix: enable case asan

* Update streamlist_for_ci.task

* fix: case asan issue

* fix: stream name issue

* fix: external window compile issues

* fix: deploy memory issue

* fix: ahandle issue

* fix: ahandle issue

* fix: ahandle issue

* fix: virtual table reader list issue

* fix: log info

* fix: msg error

* fix: virtual table addr list issue

* fix: memory issues

* fix: memory leak issue

* fix: memory issues

* fix: memory free issues

* fix: memory issues

* fix: snode deploy issue

* fix: mnode reader issue

* fix: memory issues

* fix: add debug test

* enh: add ignore nodata trigger

* fix: memory leaks

* fix: configuration issue

* fix: memory issue

* fix: external window issue

* fix: external window issues

* fix: external window placeholder issue

* fix: placeholder function init issues

* fix: memory leak issue

* fix: add debug log

* fix: compile issues

* fix: double free issue

* fix: runner addr update issue

* fix: msg rsp issue

* fix: external window reset issue

* fix: configuration issue

* fix: deploy msg issue

* fix: compile issue

* fix: external window idx issue

* fix: ci issues

* fix: ci case issues

* fix: drop dnode issue

* fix: add debug log

* fix: conflict

* fix: create stream if not exists issue

* fix: ahandle memory leak

* fix: case issue

* fix: exchange issues

* fix: crash issue

* fix: exchange prefetch issue

* fix: snode quit issue

* enh: support indef rows func

* fix: crash issues

* Fix external window collect vector function

* fix: external window indef rows issues

* fix: external window issue

---------

Co-authored-by: huohong <sallyhuo@taosdata.com>
Co-authored-by: Jing Sima <simondominic9997@outlook.com>

* test: add nevados cases

* test(stream): fix tobacco case of stream 11 #TD-36514

* test(stream): photovoltaic scene #TD-36783

* fix: output for proj

* test:remove nevados.py

* add stream case

* fix(stream): fix interval window open with watermark

* test: add old cases

* add case

* test:add old case

* fix: cann't merge externalwindows

* fix(stream): batch get block when fetch data in stream reader

* test: add cases to ci (#32064)

* test: update test case.

* test: update test case.

* case: add case

* test:add old cases

* fix(stream): batch get block when fetch data in stream reader

* test: add case into ci

* test: add case into ci

* fix: datainserter no data

* fix: stream result row data inconsistency error (#32053)

* fix: set null type

* feat: [TS-6100] Fix compile error

* feat: [TD-36866] Fix state window expr

* test: add bug4 for vehicle

* case: bug4 add null update

* test(stream): migrate old py cases #TD-36887

* fix(stream): batch get block when fetch data in stream reader

* test: compatibility ok.

* fix(stream): batch get block when fetch data in stream reader

* test: add more versions for stream compatibility.

* test: test_empty_identifier.py ok.

* fix(stream): columnInfo is null if pData==NULL

* test(stream): migrate old py case #TD-36887

* test(stream): migrate old case #TD-36887

* test: add case stream4 ok

* test: add case into ci

* test: add old cases into ci

* test: add old cases into ci

* add case

* modify case

* test: migrate empty_identifier.py to new test framework.

* test: add bug5 for vehicle

* test: remove mistakenly added files

* case: bug5 run failed modify

* add stream common

* add stream case

* add test case

* test: bug5 add self.step variant

* test: add cases (#32094)

* test: migrate test_drop.py.

* test: update test case.

* add case

* add case

* test: add use case

* Feat/ts 6100 3.0.0722m (#32103)

* migrate system-test/2-query

* revert file

* update case.task

* resolve script migrate

* run new test framework on new_testcases

* migrate system-test/2-query

* format docstring

* fix test validation

* fix test validation

* fix error

* migrate army case

* migrate army case

* fix error

* migrate system-test/2-query

* migrate system-test/2-query

* migrate system-test/2-query

* migrate system-test/2-query

* test exe time

* fix ci error

* migrate system-test/1-insert

* new common function

* migrate system-test/1-insert

* fix ci error

* migrate system-test/1-insert

* feat: add configuration and script for memory allocator settings

* fix: correct HEAPPROFILE path and remove redundant metadata_thp setting in memory allocator script

* fix ci error

* migrate system-test/1-insert, 2-query

* feat:insert into subquery (#31401) (#31710)

* migrate system-test/99-TDcase

* feat(gpt): add grant check for gpt. (#31708)

* migrate system-test/99-TDcase

* migrate system-test/99-TDcase

* migrate system-test/99-TDcase

* fix/send-heartbeat-statis (#31680)

* migrate system-test/7-tmq

* chore: support cmake option TAOSWS_GIT_TAG like taosadapter [skip ci] (#31486)

* add system-test/6/cluster test

* chore: move default branch from main to 3.3.6 for adapter/taosws

* fix docstring validation

* migrate system-test/7-tmq

* migrate system-test/7-tmq

* feat: add set_taos_malloc_env configuration and update related scripts

* Update 03-kubernetes.md

* enh: add log for snapshot (#31681)

* simple test

* more

* migrate system-test/7-tmq

* migrate system-test 0-others cases

* migrate system-test/7-tmq

* fix docstring validation

* migrate system-test/7-tmq

* fix docstring validation

* fix: invalid queue

* fix docstring validation

* refactor: reorganize memory allocator script constants and improve mode descriptions

* enh: TD-36324-improve-sync-heartbeat-log (#31727)

* recover log level

* fix: taosd crush in query when insufficient memory (#31746)

Co-authored-by: Tony Zhang <tonyzhang@taosdata.com>

* fix: overflow check in snprintf (#31780)

* migrate system-test/7-tmq

* migrate some cases

* migrate system-test/7-tmq

* fix failed cases

* fix failed cases

* migrate some cases

* fix failed cases

* fix failed cases

* migrate testcases

* fix: update environment file path in taosd.service and adjust set_taos_malloc.sh configuration

* fix: set lcn before do s3 migrate. (#31782)

* migrate testcases

* fix failed cases

* fix: add condition to set default malloc config for taosd service

* fix: update usage message in set_taos_malloc.sh to include quiet mode option

* fix: add quiet mode option to set malloc config in install script

* fix: update set_taos_malloc.sh to improve output messages and adjust default malloc config invocation

* recover log level

* fix(tmq): [TS-6569]tdb error if write tmq meta data in multi thread (#31808)

* fix: the calculation of dnode uptime (#31832)

* migrate testcase

* fix: correct timediff function bug and redress docs (#31798)

* fix: enhance malloc configuration for taosd and taosadapter in install script

* fix: TD-36442 show full condition (#31796)

* enh/TD-36466-sync-heartbeat (#31805)

* fix(plan) virtual table support BI moudle when use in select (#31787)

* fix failed cases

* enh: TS-5926-force-repair-wal (#31828)

* migrate testcase

* rename taos & taosd (#31855)

* migrate testcase

* migrate testcase

* fix failed cases

* fix: update environment file paths in taosd.service and set_taos_malloc.sh

* feat: add performance tuning documentation for memory optimization and set_taos_malloc.sh usage

* fix: update file paths in performance tuning documentation for consistency

* feat: add performance tuning documentation for memory allocator configuration script

* fix: add missing line breaks for improved readability in performance tuning documentation

* fix: adjust heading levels for consistency in performance tuning documentation

* fix cases

* fix new_testcases

* fix failed cases

* fix(query): support show tags on virtual table (#31831)

* fix failed cases

* docs: replace mysql screenshot (#31888)

* feat: use the new TDengine product name (#31859)

* merge 3.0

* fix failed cases

* fix failed cases

* merge 3.0

* fix cases

* fix cases

* doc: Update 03-stream.md (#31675)

* chore(deps): bump requests from 2.27.1 to 2.32.4 in /test (#31326)

Bumps [requests](https://github.com/psf/requests) from 2.27.1 to 2.32.4.
- [Release notes](https://github.com/psf/requests/releases)
- [Changelog](https://github.com/psf/requests/blob/main/HISTORY.md)
- [Commits](https://github.com/psf/requests/compare/v2.27.1...v2.32.4)

---
updated-dependencies:
- dependency-name: requests
  dependency-version: 2.32.4
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* fix new cases

* fix:Convert line endings from LF to CRLF for ans file

* build(deps): bump golang.org/x/net in /tools/keeper (#30811)

Bumps [golang.org/x/net](https://github.com/golang/net) from 0.36.0 to 0.38.0.
- [Commits](https://github.com/golang/net/compare/v0.36.0...v0.38.0)

---
updated-dependencies:
- dependency-name: golang.org/x/net
  dependency-version: 0.38.0
  dependency-type: indirect
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* chore(deps): bump urllib3 from 1.26.20 to 2.5.0 in /test (#31414)

Bumps [urllib3](https://github.com/urllib3/urllib3) from 1.26.20 to 2.5.0.
- [Release notes](https://github.com/urllib3/urllib3/releases)
- [Changelog](https://github.com/urllib3/urllib3/blob/main/CHANGES.rst)
- [Commits](https://github.com/urllib3/urllib3/compare/1.26.20...2.5.0)

---
updated-dependencies:
- dependency-name: urllib3
  dependency-version: 2.5.0
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* chore(deps): bump org.apache.tomcat.embed:tomcat-embed-core (#31392)

Bumps org.apache.tomcat.embed:tomcat-embed-core from 9.0.104 to 9.0.106.

---
updated-dependencies:
- dependency-name: org.apache.tomcat.embed:tomcat-embed-core
  dependency-version: 9.0.106
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* chore(deps): bump org.apache.tomcat.embed:tomcat-embed-core (#31393)

Bumps org.apache.tomcat.embed:tomcat-embed-core from 9.0.104 to 9.0.106.

---
updated-dependencies:
- dependency-name: org.apache.tomcat.embed:tomcat-embed-core
  dependency-version: 9.0.106
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* chore(deps): bump org.apache.kafka:kafka-clients (#31341)

Bumps org.apache.kafka:kafka-clients from 3.9.0 to 3.9.1.

---
updated-dependencies:
- dependency-name: org.apache.kafka:kafka-clients
  dependency-version: 3.9.1
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* fix: add lock when calculating log buffer start/end (#31883)

* fix new cases

* fix new cases

* fix failed cases

* feat: new stream (#31678)

* fix: windows compile issue

* test: add vtable cases (#31829)

* fix: windows compile issues

* test:add test cases

* fix: windows compile issue

* case: em-4 stream case submit

* test: stream4_sub1 found bug2

* test: submit test_scene_meters_bug2.py

* add stream parameters example

* feat: [TS-6100] Do not translate const value as column.

* Feat/ts 6100 3.0 zlv (#31747)

* modify asan exampel

* modify asan exampel

* add example

* add example

* modify case example

---------

Co-authored-by: zelv01 <1101510017@qq.com>

* feat(stream): fix memory leak

* modify sliding example

* test: update test case.

* feat(stream): fix conflicts

* fix: add offset case 10a 10s 10m 10h 10d

* feat(stream): fix conflicts

* chore(stream): rename case name #TS-6100

* add case

* modify example

* fix: windows compile issues

* fix: data null check

* feat: [TS-6100] Forbid where when using %%trows (#31827)

* feat: [TS-6100] Forbid where when using %%trows

* test: update cases

* feat: [TS-6100] Fix leaks.

---------

Co-authored-by: Simon Guan <guanshengliang@qq.com>

* test: reproduce bugs

* test: update test case.

* test: update test case.

* feat: [TS-6100] Fix leaks.

* test: add cases

* Feat/ts 6100 3.0.pw10 (#31841)

* enh: add operator reset func

* fix: merge join reset issue

* fix: memory issues

* fix: add debug assert

* fix: memory issues

* fix: memory leak

* fix: memory issues

* fix taos log miss

* fix: case issue

* fix: case issue

* fix: case issues

* fix: drop dnode issue

* fix: memory issues

* fix: memory issues

* fix: memory leak issues

* fix: recalculate time range issue

* fix: add debug log

* fix: memory issues

* fix: enable case asan

* Update streamlist_for_ci.task

* fix: case asan issue

* fix: stream name issue

* fix: external window compile issues

* fix: deploy memory issue

* fix: ahandle issue

* fix: ahandle issue

* fix: ahandle issue

* fix: virtual table reader list issue

* fix: log info

* fix: msg error

* fix: virtual table addr list issue

* fix: memory issues

* fix: memory leak issue

* fix: memory issues

* fix: memory free issues

* fix: memory issues

* fix: snode deploy issue

* fix: mnode reader issue

* fix: memory issues

* fix: add debug test

* enh: add ignore nodata trigger

* fix: memory leaks

* fix: configuration issue

* fix: memory issue

* fix: external window issue

* fix: external window issues

* fix: external window placeholder issue

* fix: placeholder function init issues

* fix: memory leak issue

* fix: add debug log

* fix: compile issues

* fix: double free issue

* fix: runner addr update issue

* fix: msg rsp issue

* fix: external window reset issue

* fix: configuration issue

* fix: deploy msg issue

* fix: compile issue

---------

Co-authored-by: huohong <sallyhuo@taosdata.com>

* test: reproduce bugs

* fix: add sliding interval combine case

* test: add cases

* test: add recalc test.

* test: reproduce bugs

* case : add vt ts is null check

* modify case

* bug: submit test_idmp_meters_bug3.py

* test: add test for recalc.

* test: add cases

* fix: error code check

* test: add cases

* fix(stream): scan wal with schema in that version

* add case

* test: add cases

* test: update test case.

* fix: windows compile issues

* add case

* test: add cases (#31845)

* modify case

* fix: reset interpPrev

* test: add test_idmp_meters bug4 and bug3

* add case

* fix(stream): opti wal interface

* fix: remove test_idmp_meters_bug5.py

* test: add cases

* fix(stream): fix ts data fetch for virtual tables

* cancel asan case

* test: update test case.

* test: update test case.

* add case

* test: add cases

* test: add cases

* test: add case test_idmp_meters_bug5.py

* test: update test case.

* fix(stream): tmq error

* test: add cases

* feat: [TS-6100] Restore deleted code in mndSma.c since they are still in use.

* fix(stream): optimize val scan logic

* test: add test_recalc_expired_time.py  to ci.

* test: update test case.

* test: update test case.

* feat: [TS-6100] Fix fill range check

* fix(stream): optimize val scan logic

* add case

* test: modify for partition by %%1

* test: add fun case stream4_sub7

* fix(stream): optimize val scan logic

* add case

* feat: [TS-6100] Rename OPTIONS to STREAM_OPTIONS.

* test: add test for recalc.

* test: use stream_options.

* fix: some cases error.

* test: remove recalc from ci.

* fix: ci case issues (#31880)

* enh: add operator reset func

* fix: merge join reset issue

* fix: memory issues

* fix: add debug assert

* fix: memory issues

* fix: memory leak

* fix: memory issues

* fix taos log miss

* fix: case issue

* fix: case issue

* fix: case issues

* fix: drop dnode issue

* fix: memory issues

* fix: memory issues

* fix: memory leak issues

* fix: recalculate time range issue

* fix: add debug log

* fix: memory issues

* fix: enable case asan

* Update streamlist_for_ci.task

* fix: case asan issue

* fix: stream name issue

* fix: external window compile issues

* fix: deploy memory issue

* fix: ahandle issue

* fix: ahandle issue

* fix: ahandle issue

* fix: virtual table reader list issue

* fix: log info

* fix: msg error

* fix: virtual table addr list issue

* fix: memory issues

* fix: memory leak issue

* fix: memory issues

* fix: memory free issues

* fix: memory issues

* fix: snode deploy issue

* fix: mnode reader issue

* fix: memory issues

* fix: add debug test

* enh: add ignore nodata trigger

* fix: memory leaks

* fix: configuration issue

* fix: memory issue

* fix: external window issue

* fix: external window issues

* fix: external window placeholder issue

* fix: placeholder function init issues

* fix: memory leak issue

* fix: add debug log

* fix: compile issues

* fix: double free issue

* fix: runner addr update issue

* fix: msg rsp issue

* fix: external window reset issue

* fix: configuration issue

* fix: deploy msg issue

* fix: compile issue

* fix: external window idx issue

* fix: ci issues

---------

Co-authored-by: huohong <sallyhuo@taosdata.com>

* fix(stream): fix compilation error

* fix(stream): optimize val scan logic

* test:add test cases

* test: modify case

* fix: external agg error

* test(stream): tobacco scene testing #TD-36514

* test: add stream cases (#31885)

* fix: windows compile issue

* fix: calc timerange

* fix: windows compile issue

* modify case

* fix(stream): compile error

* test: remove one debug test case file

* test: modify

* test: add test cases

* test: reproduce bugs

* test: reproduce bugs

* feat: [TS-6100] Placeholder function should only appera in SELECT and… (#31868)

* feat: [TS-6100] Placeholder function should only appera in SELECT and WHERE and FROM.

* test: update case

---------

Co-authored-by: Simon Guan <guanshengliang@qq.com>

* add example

* add example

* modify case example

* modify case

* test:alter sql

* test: add stream5 case

* fix(stream): get schema error with version

* test: add delete recalc test py.

* test: remove bug cases

* test: stream5 case test passed

* test: add state cases (#31893)

* fix(stream): compile error

* test: modify case

* test: add cases

* test: add test.

* test: update test case.

* chore(test): fix case err

* test: update test case.

* fix: align data get

* fix(stream): fix row index of datablock written into data cache

* fix: put align data

* test: update test case.

* test: add test cases for virtual table

* chore(test): fix case err #TD-36514

* add case

* test: add test for water mark.

* test: add meters bug6 for stream5

* test: add cases (#31903)

* test: add test for recalc.

* feat: [TS-6100] %%trows can only be used when event type is window close.

* test: add precision of database for ms/us/ns

* modify case

* add case

* add case

* test: add test to ci.

* modify case

* fix: ci case issues (#31904)

* enh: add operator reset func

* fix: merge join reset issue

* fix: memory issues

* fix: add debug assert

* fix: memory issues

* fix: memory leak

* fix: memory issues

* fix taos log miss

* fix: case issue

* fix: case issue

* fix: case issues

* fix: drop dnode issue

* fix: memory issues

* fix: memory issues

* fix: memory leak issues

* fix: recalculate time range issue

* fix: add debug log

* fix: memory issues

* fix: enable case asan

* Update streamlist_for_ci.task

* fix: case asan issue

* fix: stream name issue

* fix: external window compile issues

* fix: deploy memory issue

* fix: ahandle issue

* fix: ahandle issue

* fix: ahandle issue

* fix: virtual table reader list issue

* fix: log info

* fix: msg error

* fix: virtual table addr list issue

* fix: memory issues

* fix: memory leak issue

* fix: memory issues

* fix: memory free issues

* fix: memory issues

* fix: snode deploy issue

* fix: mnode reader issue

* fix: memory issues

* fix: add debug test

* enh: add ignore nodata trigger

* fix: memory leaks

* fix: configuration issue

* fix: memory issue

* fix: external window issue

* fix: external window issues

* fix: external window placeholder issue

* fix: placeholder function init issues

* fix: memory leak issue

* fix: add debug log

* fix: compile issues

* fix: double free issue

* fix: runner addr update issue

* fix: msg rsp issue

* fix: external window reset issue

* fix: configuration issue

* fix: deploy msg issue

* fix: compile issue

* fix: external window idx issue

* fix: ci issues

* fix: ci case issues

* fix: drop dnode issue

---------

Co-authored-by: huohong <sallyhuo@taosdata.com>

* fix(stream): ci error

* test: update test case.

* feat: [TS-6100] Disable some failed UT.

* feat: [TS-6100] Fix virtual table

* test: add bug 5.

* test: add test delete recalc to ci.

* test: add bug 6.

* test(stream): tobacco scene #TD-36514

* fix: reqCids,reqCols memory leak in SSTriggerRealtimeContext

Co-authored-by: Tony Zhang <tonyzhang@taosdata.com>

* test: add case stream6

* fix(stream): implement some pending features in trigger task

* modify case

* modify case

* fix: case issues

* modify case

* test: add recalc for warter mark.

* fix(stream): fix count window trigger of virtual tables

* fix(stream): memory leak

* test: fix run err.

* test: add stream6 bug7

* fix: adjust format

* test(stream): tobacco scene testing #TD-36514

* test: change bug7 with update window1 and 2

* test: add test bug 7.

* case: restore write 3 window

* fix: windows compile issue

* fix: notify

* test: add cases

* modify case

* test: update test case.

* test(stream): toobacco scene testing #TD-36514

---------

Co-authored-by: Simon Guan <slguan@taosdata.com>
Co-authored-by: plum-lihui <huili@taosdata.com>
Co-authored-by: Alex Duan <417921451@qq.com>
Co-authored-by: zelv01 <1101510017@qq.com>
Co-authored-by: Jing Sima <simondominic9997@outlook.com>
Co-authored-by: xiangyang guo <66111494+happyguoxy@users.noreply.github.com>
Co-authored-by: wangmm0220 <wangmm0220@gmail.com>
Co-authored-by: Haojun Liao <hjliao@taosdata.com>
Co-authored-by: zyyang90 <zyyang@taosdata.com>
Co-authored-by: Alex Duan <51781608+DuanKuanJun@users.noreply.github.com>
Co-authored-by: facetosea <285808407@qq.com>
Co-authored-by: Simon Guan <guanshengliang@qq.com>
Co-authored-by: huohong <sallyhuo@taosdata.com>
Co-authored-by: Li Hui <52318143+plum-lihui@users.noreply.github.com>
Co-authored-by: Jinqing Kuang <kuangjinqingcn@gmail.com>
Co-authored-by: xiao-77 <berylbao@taosdata.com>
Co-authored-by: Zhixiao Bao <62235797+xiao-77@users.noreply.github.com>
Co-authored-by: happyguoxy <happy_guoxy@163.com>
Co-authored-by: Tony Zhang <34825804+Tony2h@users.noreply.github.com>
Co-authored-by: Tony Zhang <tonyzhang@taosdata.com>

* test: rename TSDB

* docs: fix rust examples (#31908)

* docs: modify rust native test case

* docs: modify rust ws test case

* docs: modify rust examples

* docs: update rust pool docs

* fix new cases

* migrate test case

* feat: support reading sub table names and tag values from CSV files to create sub tables (#31909)

* feat: add obtaining table names from tag files

* feat: add write data table control

* feat: add table params to write

* feat: delete log file

* feat: modify test case csv path

* feat: resolve memory leakage in the table building thread

* feat: resolve compilation errors

* feat: resolve table name copy len error

* feat: modify create table log level

* feat: modifying query configuration parameter array out of bounds

* feat: support custom primary key names

* feat: modify log level

* feat: add set primary key name case

* feat: add column  keywords case

* feat: add keywords case data

* feat: modify primaryKeyName value len

* feat: modify primaryKeyName value define

* feat: modify primaryKeyName value size

* fix: compile issue (#31943)

Co-authored-by: taos-support <it@taosdata.com>

* package: fix error

* package: fix error

* fix failed cases

* merge 3.0

* rename create_table_keywords.py to test_create_table_keywords.py

* fix failed cases

* fix new cases

* docs: update stream (#31957)

* docs: update jdbc out-dated descripiton (#31959)

* fix: TD-36560 refactor arbitrator group function name and log (#31852)

* feat: support BLOB data type (#31704)

* rename 0-others/mounts.py to 0-others/test_mounts.py

* fix failed cases

* docs: update gpt (#31975)

* fix failed cases

* fix failed cases

* package: fix error

* feat: add taosBenchmark command line parameters (#31967)

* feat: add command line parameters

* feat: add command line parameter test cases

* fix: tableName len error

* enh: set TD Release build in tdengine-build.yml

* Update tdengine-build.yml

* fix: add json file path log

* fix: streamline TD_CONFIG export in build steps

* fix: Restore the build configuration

---------

Co-authored-by: haoranchen <haoran920c@163.com>

* fix: tableName len error (#31977)

* fix: tableName len error

* fix: modify TD_CONFIG=Release

* fix: code format

* fix: Restore the build configuration

* enh: set TD Release build in tdengine-build.yml (#31980)

* enh: set TD Release build in tdengine-build.yml

* Update tdengine-build.yml

* fix: update cache key for externals to include debug build version

* fix: remove verbose flag from build commands in tdengine-build.yml

* skip memleak cases

* fix failed case

* fix failed cases

* package: fix error

* fix(stmt2):tbname error output (#31997)

* fix: possible memory leak (#31972)

* feat: create connect add dbname params (#32002)

* feat: create connect add dbname params

* fix: connect param error

* skip failed cases

* fix cases on windows

* fix cases

* support connect bi mode and fix log level

* unique sql connect username and password

* fix log level

* enh: mounted vnode may have no tq (#31916)

* fix: subquery memleak (#32024)

* fix failed case

* fix cases

* rename 2-query/test_insert_select.py to 2-query/test_system_insert_select.py

* skip memleak cases

* enh: rename data forecast/detect to forecasting/anomaly detection (#32021)

* package: unique product name

* package: update for main

* skip tsim cases

* chore: update jdbc connection pool validation query sql (#32056)

* refactor: quotes usage in bash scripts

Signed-off-by: WANG Xu <feici02@outlook.com>

* enhn(mqtt/rawblock): new format for msg payload (#31801)

* fix: fix broken link in 14-stream.md

* docs: 15-spark.md is missing end semicolon (#32068)

* chore: bump dev version to 3.3.7.0.alpha (#32066)

* fix: blob test (#32020)

* fix blob query error

* fix blob query error

* fix blob query error

* fix blob query error

* fix blob query error

* opt query

* opt write

* opt write

* opt write

* opt bse

* opt write

* opt write

* opt write

* opt write

* opt write

* opt write

* opt write

* opt write

* opt write

* opt write

* opt write

* opt write

* opt write

* add cache

* opt query

* opt query

* opt bse

* add data iter

* add data iter

* add more compress

* add more compress

* add more compress

* add more compress

* add more compress

* add more compress

* opt blob transfer

* opt blob transfer

* opt blob transfer

* opt write

* avoid unordered data write

* avoid unordered data write

* opt read

* refactor log

* fix invalid write

* refactor code

* fix merge error

* fix merge error

* add error code

* support blob type len

* support blob type len

* support blob type len

* support blob type len

* support blob type len

* support blob type len

* support blob type len

* support blob type len

* support blob type len

* support blob type len

* Merge remote-tracking branch 'origin/3.0' into enh/blob

* refactor code

* support blob type len

* refactor code

* refactor code

* benchmark support blob type

* benchmark support blob type

* add log

* handle sort and merge row

* change file set

* change file set

* change file set

* change file set

* change file set

* change file set

* change file set

* opt code

* opt read

* add restart error and add unit test

* add restart error and add unit test

* add restart error and add unit test

* add restart error and add unit test

* add restart error and add unit test

* refactor code

* add restart error and add unit test

* opt code

* refactor code

* fix invalid write

* blob test

* blob test

* blob test

* blob test

* blob test

* support blob

* add str trim

* add str trim

* Merge remote-tracking branch 'origin/3.0' into feat/blob

* Merge remote-tracking branch 'origin/3.0' into feat/blob

* update test case

* fix invalid read

* fix invalid read

* fix invalid read

* add stmt2

* add stmt2

* add stmt2

* update parameter

* refactor test case

* refactor test case

* support blob

* support stmt2

* add sub

* support blob

* support blob

* support sub

* fix tmq crash

* support windows/darwin

* fix stmt2 bind row

* fix blob crash

* fix blob crash

* fix blob query error

* fix blob crash

* fix merge error

* refactor bse

* add blob transfer

* add blob transfer

* add transfer snapshot

* refactor code

* change log level

* change log level

* add test case

* refactor code

* revert taosBenchmark

* revert taosbench

* fix: improve error handling and encoding for file reading in grep_asserts_in_file function

* rm assert

* fix mem leak

* fix compile error

* fix conflict

* fix conflict

* fix compile error

* fix compile error

* fix pre check error

* fix pre check error

* fix compile error on windows

* fix error on window

* fix error on windows

* fix error on windows

* opt no-blob sql

* fix compile error on dawain

* fix compile error on dawain

* fix invalid read

* fix mem leak

* fix invalid read

* fix invalid read

* fix error on windows

* remove unused code

* refactor code

* fix mem leak

* fix mem leak

* rm unused code

* rm unused code

* fix invalid copy

* fix invalid copy

* fix invalid copy

* fix invalid copy

* fix invalid copy

* refactor code

* make ci happy

* refactor code

* change make

* update test case

* update ignore code

* update bse snapshot and test case

* update bse snapshot and test case

* refactor code

* change unit test

* update bse snapshot

* update bse snapshot

* fix test case

* fix bse snapshot

* fix bse snapshot

* fix snapshot transfer

* remove unused log

* support func query

* add test case

* forbidden  unsupport code

* merge 3.0

* merge 3.0

* change test case

* add forbidden code

* add forbidden code

* add code

* support length func

* support length func

* taosBenchmark support  blob

* support blob raw block

* support write raw block

* support more query

* Merge branch 'feat/blob' into feat/blob_test

* SBlobRow2

* change bse commit change

* refactor code

* rm exe test

* refactor code

* checke return code

* rename blob name

* refactor code

* refactor code

* refactor code

* refactor code

* fix unordere write

* fix unordere write

* fix row merge error

* fix row merge error

* fix row merge error

* fix row merge error

* support ordered data

* support ordered data

* Merge remote-tracking branch 'origin/3.0' into feat/blob_test

* fix stmt2 blob crash

* add not support write type

* add not support write type

* add not support write type

* rm exe

* fix col-formate error

* fix col-formate error

* fix mem leak

* refactor code

* add error code for single blob column restriction and update error message

* add error code for single blob column restriction and update error message

* add error code for single blob column restriction and update error message

* refactor code

* change error code

* make ci happy

---------

Co-authored-by: yihaoDeng <yhdeng@taosdata.com>
Co-authored-by: chenhaoran <haoran920c@163.com>

* package: update for main (#32091)

* fix compile error on windows (#32089)

* fix: source code merge issues

---------

Signed-off-by: dependabot[bot] <support@github.com>
Signed-off-by: WANG Xu <feici02@outlook.com>
Co-authored-by: minhuinie <nminhui@163.com>
Co-authored-by: huohong <sallyhuo@taosdata.com>
Co-authored-by: chenhaoran <haoran920c@163.com>
Co-authored-by: Nie Minhui <143420805+minhuinie@users.noreply.github.com>
Co-authored-by: Mario Peng <48949600+Pengrongkun@users.noreply.github.com>
Co-authored-by: Haojun Liao <hjxilinx@users.noreply.github.com>
Co-authored-by: dongming chen <cademfly@hotmail.com>
Co-authored-by: Linhe Huo <linhehuo@gmail.com>
Co-authored-by: huohong <346479823@qq.com>
Co-authored-by: Joel Brass <joel@jbrass.com>
Co-authored-by: WANG Xu <feici02@outlook.com>
Co-authored-by: Hongze Cheng <hzcheng@taosdata.com>
Co-authored-by: Tony Zhang <34825804+Tony2h@users.noreply.github.com>
Co-authored-by: Tony Zhang <tonyzhang@taosdata.com>
Co-authored-by: Kaili Xu <klxu@taosdata.com>
Co-authored-by: Zhixiao Bao <62235797+xiao-77@users.noreply.github.com>
Co-authored-by: WANG MINGMING <wangmm0220@gmail.com>
Co-authored-by: hongzhenliu <wluckyjob@gmail.com>
Co-authored-by: Daniel Clow <106956386+danielclow@users.noreply.github.com>
Co-authored-by: She Yanjie <57549981+sheyanjie-qq@users.noreply.github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: Simon Guan <slguan@taosdata.com>
Co-authored-by: plum-lihui <huili@taosdata.com>
Co-authored-by: Alex Duan <417921451@qq.com>
Co-authored-by: zelv01 <1101510017@qq.com>
Co-authored-by: Jing Sima <simondominic9997@outlook.com>
Co-authored-by: xiangyang guo <66111494+happyguoxy@users.noreply.github.com>
Co-authored-by: Haojun Liao <hjliao@taosdata.com>
Co-authored-by: zyyang90 <zyyang@taosdata.com>
Co-authored-by: Alex Duan <51781608+DuanKuanJun@users.noreply.github.com>
Co-authored-by: facetosea <285808407@qq.com>
Co-authored-by: Simon Guan <guanshengliang@qq.com>
Co-authored-by: Li Hui <52318143+plum-lihui@users.noreply.github.com>
Co-authored-by: Jinqing Kuang <kuangjinqingcn@gmail.com>
Co-authored-by: xiao-77 <berylbao@taosdata.com>
Co-authored-by: happyguoxy <happy_guoxy@163.com>
Co-authored-by: guozhenwei <2227465945@qq.com>
Co-authored-by: kevin men <men_shi_bin@163.com>
Co-authored-by: taos-support <it@taosdata.com>
Co-authored-by: Yihao Deng <luomoxyz@126.com>
Co-authored-by: Minglei Jin <49711132+stephenkgu@users.noreply.github.com>
Co-authored-by: yihaoDeng <yhdeng@taosdata.com>

* test: add cases (#32102)

* test: add old cases (#32101)

* test: modify old case

* add case

* test: remove ts-5617.

* test: add case bug6 for vehicle

* add case

* add case

* test: modify old case

* fix: merge issue

* test: migrate test_interval.py.

* fix: row merge strategy

* test(stream): photovoltaic scene add ci #TD-36783

* test(stream): fix case #TD-36783

* test: add old cases (#32110)

* test: update test case.

* test: add use case

* test: update test case.

* test: update test case.

* test: update test case.

* fix: windows compile issue

* test: modify for format

* test(stream): fix ci warning #TD-36783

* test(stream): fix #TD-36514

* fix(stream): pass processOneBlock=true to vnodeProcessStreamFetchMsg when fetch data from virtual table to avoid heap use after free

* fix: output tbname contains a dot

* test(stream): fix ci warning #TD-36514

* test(stream): fix ci warning #TD-36887

* test: modify notes format

* add case

* test(stream): fix ci #TD-36514

* enh: support count like functions in ext window (#32082)

* enh: add operator reset func

* fix: merge join reset issue

* fix: memory issues

* fix: add debug assert

* fix: memory issues

* fix: memory leak

* fix: memory issues

* fix taos log miss

* fix: case issue

* fix: case issue

* fix: case issues

* fix: drop dnode issue

* fix: memory issues

* fix: memory issues

* fix: memory leak issues

* fix: recalculate time range issue

* fix: add debug log

* fix: memory issues

* fix: enable case asan

* Update streamlist_for_ci.task

* fix: case asan issue

* fix: stream name issue

* fix: external window compile issues

* fix: deploy memory issue

* fix: ahandle issue

* fix: ahandle issue

* fix: ahandle issue

* fix: virtual table reader list issue

* fix: log info

* fix: msg error

* fix: virtual table addr list issue

* fix: memory issues

* fix: memory leak issue

* fix: memory issues

* fix: memory free issues

* fix: memory issues

* fix: snode deploy issue

* fix: mnode reader issue

* fix: memory issues

* fix: add debug test

* enh: add ignore nodata trigger

* fix: memory leaks

* fix: configuration issue

* fix: memory issue

* fix: external window issue

* fix: external window issues

* fix: external window placeholder issue

* fix: placeholder function init issues

* fix: memory leak issue

* fix: add debug log

* fix: compile issues

* fix: double free issue

* fix: runner addr update issue

* fix: msg rsp issue

* fix: external window reset issue

* fix: configuration issue

* fix: deploy msg issue

* fix: compile issue

* fix: external window idx issue

* fix: ci issues

* fix: ci case issues

* fix: drop dnode issue

* fix: add debug log

* fix: conflict

* fix: create stream if not exists issue

* fix: ahandle memory leak

* fix: case issue

* fix: exchange issues

* fix: crash issue

* fix: exchange prefetch issue

* fix: snode quit issue

* enh: support indef rows func

* fix: crash issues

* Fix external window collect vector function

* fix: external window indef rows issues

* fix: external window issue

* enh: support count always return value in external window

* fix: force output when has more result block

* fix: runner block retrieve issue

* fix: crash issue

* fix: count cases issue

---------

Co-authored-by: huohong <sallyhuo@taosdata.com>
Co-authored-by: Jing Sima <simondominic9997@outlook.com>
Co-authored-by: facetosea <285808407@qq.com>

* test(stream): fix ci warning #TD-36514

* modify example

* test: update test case.

* add case to ci (#32113)

* add case to ci

* modify example

* fix: compile issues

* test: add more test for stream.

* feat: [TD-36624] Fix join table extract time range expr.

* fix(stream): add missing fields in stream notification

* test: add stream cases (#32129)

* fix: error code check

* feat: [TD-36624] Fix join table extract time range expr.

* fix(stream): add log

* modify example

* feat: [TS-6100] Modify alter vtable test case

* test: migrate test compatibility.

* fix: ci check.

* feat: [TS-6100] Fix leak

* modify example

* enh: external window support scalar (#32139)

* enh: add operator reset func

* fix: merge join reset issue

* fix: memory issues

* fix: add debug assert

* fix: memory issues

* fix: memory leak

* fix: memory issues

* fix taos log miss

* fix: case issue

* fix: case issue

* fix: case issues

* fix: drop dnode issue

* fix: memory issues

* fix: memory issues

* fix: memory leak issues

* fix: recalculate time range issue

* fix: add debug log

* fix: memory issues

* fix: enable case asan

* Update streamlist_for_ci.task

* fix: case asan issue

* fix: stream name issue

* fix: external window compile issues

* fix: deploy memory issue

* fix: ahandle issue

* fix: ahandle issue

* fix: ahandle issue

* fix: virtual table reader list issue

* fix: log info

* fix: msg error

* fix: virtual table addr list issue

* fix: memory issues

* fix: memory leak issue

* fix: memory issues

* fix: memory free issues

* fix: memory issues

* fix: snode deploy issue

* fix: mnode reader issue

* fix: memory issues

* fix: add debug test

* enh: add ignore nodata trigger

* fix: memory leaks

* fix: configuration issue

* fix: memory issue

* fix: external window issue

* fix: external window issues

* fix: external window placeholder issue

* fix: placeholder function init issues

* fix: memory leak issue

* fix: add debug log

* fix: compile issues

* fix: double free issue

* fix: runner addr update issue

* fix: msg rsp issue

* fix: external window reset issue

* fix: configuration issue

* fix: deploy msg issue

* fix: compile issue

* fix: external window idx issue

* fix: ci issues

* fix: ci case issues

* fix: drop dnode issue

* fix: add debug log

* fix: conflict

* fix: create stream if not exists issue

* fix: ahandle memory leak

* fix: case issue

* fix: exchange issues

* fix: crash issue

* fix: exchange prefetch issue

* fix: snode quit issue

* enh: support indef rows func

* fix: crash issues

* Fix external window collect vector function

* fix: external window indef rows issues

* fix: external window issue

* enh: support count always return value in external window

* fix: force output when has more result block

* fix: runner block retrieve issue

* fix: crash issue

* fix: count cases issue

* fix: reader deploy message issue

* fix: task deploy issue

* fix: external window scalar issue

* fix: compile issue

---------

Co-authored-by: huohong <sallyhuo@taosdata.com>
Co-authored-by: Jing Sima <simondominic9997@outlook.com>
Co-authored-by: facetosea <285808407@qq.com>

* feat: [TS-6100] Fix show create table/vtable

* fix: tagsName

---------

Signed-off-by: dependabot[bot] <support@github.com>
Signed-off-by: WANG Xu <feici02@outlook.com>
Co-authored-by: Jing Sima <simondominic9997@outlook.com>
Co-authored-by: Alex Duan <417921451@qq.com>
Co-authored-by: zyyang90 <zyyang@taosdata.com>
Co-authored-by: Zhixiao Bao <62235797+xiao-77@users.noreply.github.com>
Co-authored-by: facetosea <285808407@qq.com>
Co-authored-by: Li Hui <52318143+plum-lihui@users.noreply.github.com>
Co-authored-by: Haojun Liao <hjliao@taosdata.com>
Co-authored-by: Lv Ze <1101510017@qq.com>
Co-authored-by: root <happy_guoxy@163.com>
Co-authored-by: plum-lihui <huili@taosdata.com>
Co-authored-by: xiao-77 <berylbao@taosdata.com>
Co-authored-by: xiangyang guo <66111494+happyguoxy@users.noreply.github.com>
Co-authored-by: Simon Guan <guanshengliang@qq.com>
Co-authored-by: Alex Duan <51781608+DuanKuanJun@users.noreply.github.com>
Co-authored-by: Simon Guan <slguan@taosdata.com>
Co-authored-by: huohong <sallyhuo@taosdata.com>
Co-authored-by: Zhiyu Yang <69311263+zyyang90@users.noreply.github.com>
Co-authored-by: Jinqing Kuang <kuangjinqingcn@gmail.com>
Co-authored-by: wangmm0220 <wangmm0220@gmail.com>
Co-authored-by: Tony Zhang <34825804+Tony2h@users.noreply.github.com>
Co-authored-by: minhuinie <nminhui@163.com>
Co-authored-by: chenhaoran <haoran920c@163.com>
Co-authored-by: Nie Minhui <143420805+minhuinie@users.noreply.github.com>
Co-authored-by: Mario Peng <48949600+Pengrongkun@users.noreply.github.com>
Co-authored-by: Haojun Liao <hjxilinx@users.noreply.github.com>
Co-authored-by: dongming chen <cademfly@hotmail.com>
Co-authored-by: Linhe Huo <linhehuo@gmail.com>
Co-authored-by: huohong <346479823@qq.com>
Co-authored-by: Joel Brass <joel@jbrass.com>
Co-authored-by: WANG Xu <feici02@outlook.com>
Co-authored-by: Hongze Cheng <hzcheng@taosdata.com>
Co-authored-by: Tony Zhang <tonyzhang@taosdata.com>
Co-authored-by: Kaili Xu <klxu@taosdata.com>
Co-authored-by: hongzhenliu <wluckyjob@gmail.com>
Co-authored-by: Daniel Clow <106956386+danielclow@users.noreply.github.com>
Co-authored-by: She Yanjie <57549981+sheyanjie-qq@users.noreply.github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: guozhenwei <2227465945@qq.com>
Co-authored-by: kevin men <men_shi_bin@163.com>
Co-authored-by: taos-support <it@taosdata.com>
Co-authored-by: Yihao Deng <luomoxyz@126.com>
Co-authored-by: Minglei Jin <49711132+stephenkgu@users.noreply.github.com>
Co-authored-by: yihaoDeng <yhdeng@taosdata.com>
2025-07-23 17:11:16 +08:00

1333 lines
51 KiB
C

/*
* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
*
* This program is free software: you can use, redistribute, and/or modify
* it under the terms of the GNU Affero General Public License, version 3
* or later ("AGPL"), as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "executorInt.h"
#include "operator.h"
#include "querytask.h"
#include "tdatablock.h"
#include "stream.h"
#include "filter.h"
typedef struct SBlockList {
const SSDataBlock* pSrcBlock;
SList* pBlocks;
int32_t blockRowNumThreshold;
} SBlockList;
typedef struct SExternalWindowOperator {
SOptrBasicInfo binfo;
SAggSupporter aggSup;
SExprSupp scalarSupp;
STimeWindowAggSupp twAggSup;
SGroupResInfo groupResInfo;
int32_t primaryTsIndex;
EExtWinMode mode;
SArray* pWins;
SArray* pOutputBlocks; // for each window, we have a list of blocks
// SArray* pOffsetList; // for each window
int32_t lastWinId;
int32_t outputWinId;
SListNode* pOutputBlockListNode; // block index in block array used for output
SSDataBlock* pTmpBlock;
SSDataBlock* pEmptyInputBlock;
SArray* pPseudoColInfo;
bool hasCountFunc;
// SLimitInfo limitInfo; // limit info for each window
} SExternalWindowOperator;
static int32_t blockListInit(SBlockList* pBlockList, int32_t threshold) {
pBlockList->pBlocks = tdListNew(sizeof(SSDataBlock*));
if (!pBlockList->pBlocks) {
return terrno;
}
return 0;
}
static int32_t blockListAddBlock(SBlockList* pBlockList) {
SSDataBlock* pRes = NULL;
int32_t code = createOneDataBlock(pBlockList->pSrcBlock, false, &pRes);
if (code != 0) {
return code;
}
code = blockDataEnsureCapacity(pRes, pBlockList->blockRowNumThreshold);
if (code != 0) {
blockDataDestroy(pRes);
return code;
}
code = tdListAppend(pBlockList->pBlocks, &pRes);
if (code != 0) {
blockDataDestroy(pRes);
return code;
}
return 0;
}
static int32_t blockListGetLastBlock(SBlockList* pBlockList, SSDataBlock** ppBlock) {
SListNode* pNode = TD_DLIST_TAIL(pBlockList->pBlocks);
int32_t code = 0;
*ppBlock = NULL;
code = blockListAddBlock(pBlockList);
if (0 == code) {
pNode = TD_DLIST_TAIL(pBlockList->pBlocks);
*ppBlock = *(SSDataBlock**)pNode->data;
}
return code;
}
static void blockListDestroy(void* p) {
SBlockList* pBlockList = (SBlockList*)p;
if (TD_DLIST_NELES(pBlockList->pBlocks) > 0) {
SListNode* pNode = TD_DLIST_HEAD(pBlockList->pBlocks);
while (pNode) {
SSDataBlock* pBlock = *(SSDataBlock**)pNode->data;
blockDataDestroy(pBlock);
pNode = pNode->dl_next_;
}
}
taosMemoryFree(pBlockList->pBlocks);
}
void destroyExternalWindowOperatorInfo(void* param) {
if (NULL == param) {
return;
}
SExternalWindowOperator* pInfo = (SExternalWindowOperator*)param;
cleanupBasicInfo(&pInfo->binfo);
taosArrayDestroyEx(pInfo->pOutputBlocks, blockListDestroy);
taosArrayDestroy(pInfo->pWins);
colDataDestroy(&pInfo->twAggSup.timeWindowData);
cleanupGroupResInfo(&pInfo->groupResInfo);
taosArrayDestroy(pInfo->pPseudoColInfo);
blockDataDestroy(pInfo->pTmpBlock);
blockDataDestroy(pInfo->pEmptyInputBlock);
cleanupAggSup(&pInfo->aggSup);
cleanupExprSupp(&pInfo->scalarSupp);
pInfo->binfo.resultRowInfo.openWindow = tdListFree(pInfo->binfo.resultRowInfo.openWindow);
taosMemoryFreeClear(pInfo);
}
static int32_t doOpenExternalWindow(SOperatorInfo* pOperator);
static int32_t externalWindowNext(SOperatorInfo* pOperator, SSDataBlock** ppRes);
typedef struct SMergeAlignedExternalWindowOperator {
SExternalWindowOperator* pExtW;
int64_t curTs;
SSDataBlock* pPrefetchedBlock;
SResultRow* pResultRow;
} SMergeAlignedExternalWindowOperator;
void destroyMergeAlignedExternalWindowOperator(void* pOperator) {
SMergeAlignedExternalWindowOperator* pMlExtInfo = (SMergeAlignedExternalWindowOperator*)pOperator;
destroyExternalWindowOperatorInfo(pMlExtInfo->pExtW);
}
void doMergeAlignExternalWindow(SOperatorInfo* pOperator);
static int32_t mergeAlignedExternalWindowNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SMergeAlignedExternalWindowOperator* pMlExtInfo = pOperator->info;
SExternalWindowOperator* pExtW = pMlExtInfo->pExtW;
int32_t code = 0;
int32_t lino = 0;
if (pOperator->status == OP_EXEC_DONE) {
(*ppRes) = NULL;
return TSDB_CODE_SUCCESS;
}
SSDataBlock* pRes = pExtW->binfo.pRes;
blockDataCleanup(pRes);
if (!pExtW->pWins) {
size_t size = taosArrayGetSize(pTaskInfo->pStreamRuntimeInfo->funcInfo.pStreamPesudoFuncVals);
pExtW->pWins = taosArrayInit(size, sizeof(STimeWindow));
if (!pExtW->pWins) QUERY_CHECK_CODE(terrno, lino, _end);
// pExtW->pOffsetList = taosArrayInit(size, sizeof(SLimitInfo));
// if (!pExtW->pOffsetList) QUERY_CHECK_CODE(terrno, lino, _end);
for (int32_t i = 0; i < size; ++i) {
SSTriggerCalcParam* pParam = taosArrayGet(pTaskInfo->pStreamRuntimeInfo->funcInfo.pStreamPesudoFuncVals, i);
STimeWindow win = {.skey = pParam->wstart, .ekey = pParam->wend};
if (pTaskInfo->pStreamRuntimeInfo->funcInfo.triggerType != 1){ // 1 meams STREAM_TRIGGER_SLIDING
win.ekey++;
}
TSDB_CHECK_NULL(taosArrayPush(pExtW->pWins, &win), code, lino, _end, terrno);
}
pExtW->outputWinId = pTaskInfo->pStreamRuntimeInfo->funcInfo.curIdx;
}
doMergeAlignExternalWindow(pOperator);
size_t rows = pRes->info.rows;
pOperator->resultInfo.totalRows += rows;
(*ppRes) = (rows == 0) ? NULL : pRes;
_end:
if (code != 0) {
qError("%s failed at line %d since:%s", __func__, lino, tstrerror(code));
pTaskInfo->code = code;
T_LONG_JMP(pTaskInfo->env, code);
}
return code;
}
int32_t resetMergeAlignedExternalWindowOperator(SOperatorInfo* pOperator) {
SMergeAlignedExternalWindowOperator* pMlExtInfo = pOperator->info;
SExternalWindowOperator* pExtW = pMlExtInfo->pExtW;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SMergeAlignedIntervalPhysiNode * pPhynode = (SMergeAlignedIntervalPhysiNode*)pOperator->pPhyNode;
pOperator->status = OP_NOT_OPENED;
taosArrayDestroy(pExtW->pWins);
pExtW->pWins = NULL;
resetBasicOperatorState(&pExtW->binfo);
pMlExtInfo->pResultRow = NULL;
pMlExtInfo->curTs = INT64_MIN;
if (pMlExtInfo->pPrefetchedBlock) blockDataCleanup(pMlExtInfo->pPrefetchedBlock);
int32_t code = resetAggSup(&pOperator->exprSupp, &pExtW->aggSup, pTaskInfo, pPhynode->window.pFuncs, NULL,
sizeof(int64_t) * 2 + POINTER_BYTES, pTaskInfo->id.str, pTaskInfo->streamInfo.pState,
&pTaskInfo->storageAPI.functionStore);
if (code == 0) {
colDataDestroy(&pExtW->twAggSup.timeWindowData);
code = initExecTimeWindowInfo(&pExtW->twAggSup.timeWindowData, &pTaskInfo->window);
}
return code;
}
int32_t createMergeAlignedExternalWindowOperator(SOperatorInfo* pDownstream, SPhysiNode* pNode,
SExecTaskInfo* pTaskInfo, SOperatorInfo** ppOptrOut) {
SMergeAlignedIntervalPhysiNode* pPhynode = (SMergeAlignedIntervalPhysiNode*)pNode;
int32_t code = 0;
int32_t lino = 0;
SMergeAlignedExternalWindowOperator* pMlExtInfo = taosMemoryCalloc(1, sizeof(SMergeAlignedExternalWindowOperator));
SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
if (pTaskInfo->pStreamRuntimeInfo != NULL){
pTaskInfo->pStreamRuntimeInfo->funcInfo.withExternalWindow = true;
}
pOperator->pPhyNode = pNode;
if (!pMlExtInfo || !pOperator) {
code = terrno;
goto _error;
}
pMlExtInfo->pExtW = taosMemoryCalloc(1, sizeof(SExternalWindowOperator));
if (!pMlExtInfo->pExtW) {
code = terrno;
goto _error;
}
SExternalWindowOperator* pExtW = pMlExtInfo->pExtW;
SExprSupp* pSup = &pOperator->exprSupp;
pSup->hasWindowOrGroup = true;
pMlExtInfo->curTs = INT64_MIN;
pExtW->primaryTsIndex = ((SColumnNode*)pPhynode->window.pTspk)->slotId;
pExtW->mode = pPhynode->window.pProjs ? EEXT_MODE_SCALAR : EEXT_MODE_AGG;
pExtW->binfo.inputTsOrder = pPhynode->window.node.inputTsOrder = TSDB_ORDER_ASC;
pExtW->binfo.outputTsOrder = pExtW->binfo.inputTsOrder;
// pExtW->limitInfo = (SLimitInfo){0};
// initLimitInfo(pPhynode->window.node.pLimit, pPhynode->window.node.pSlimit, &pExtW->limitInfo);
size_t keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
initResultSizeInfo(&pOperator->resultInfo, 512);
int32_t num = 0;
SExprInfo* pExprInfo = NULL;
code = createExprInfo(pPhynode->window.pFuncs, NULL, &pExprInfo, &num);
QUERY_CHECK_CODE(code, lino, _error);
if (pExtW->mode == EEXT_MODE_AGG) {
code = initAggSup(pSup, &pExtW->aggSup, pExprInfo, num, keyBufSize, pTaskInfo->id.str, pTaskInfo->streamInfo.pState,
&pTaskInfo->storageAPI.functionStore);
QUERY_CHECK_CODE(code, lino, _error);
}
SSDataBlock* pResBlock = createDataBlockFromDescNode(pPhynode->window.node.pOutputDataBlockDesc);
QUERY_CHECK_NULL(pResBlock, code, lino, _error, terrno);
initBasicInfo(&pExtW->binfo, pResBlock);
initResultRowInfo(&pExtW->binfo.resultRowInfo);
code = blockDataEnsureCapacity(pExtW->binfo.pRes, pOperator->resultInfo.capacity);
QUERY_CHECK_CODE(code, lino, _error);
setOperatorInfo(pOperator, "MergeAlignedExternalWindowOperator", QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW, false, OP_NOT_OPENED, pMlExtInfo, pTaskInfo);
pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, mergeAlignedExternalWindowNext, NULL,
destroyMergeAlignedExternalWindowOperator, optrDefaultBufFn, NULL,
optrDefaultGetNextExtFn, NULL);
setOperatorResetStateFn(pOperator, resetMergeAlignedExternalWindowOperator);
code = appendDownstream(pOperator, &pDownstream, 1);
QUERY_CHECK_CODE(code, lino, _error);
*ppOptrOut = pOperator;
return code;
_error:
if (pMlExtInfo) destroyMergeAlignedExternalWindowOperator(pMlExtInfo);
destroyOperatorAndDownstreams(pOperator, &pDownstream, 1);
pTaskInfo->code = code;
return code;
}
static int32_t resetExternalWindowOperator(SOperatorInfo* pOperator) {
SExternalWindowOperator* pExtW = pOperator->info;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SExternalWindowPhysiNode* pPhynode = (SExternalWindowPhysiNode*)pOperator->pPhyNode;
pOperator->status = OP_NOT_OPENED;
resetBasicOperatorState(&pExtW->binfo);
pExtW->outputWinId = 0;
pExtW->lastWinId = -1;
taosArrayDestroyEx(pExtW->pOutputBlocks, blockListDestroy);
// taosArrayDestroy(pExtW->pOffsetList);
taosArrayDestroy(pExtW->pWins);
pExtW->pWins = NULL;
pExtW->pOutputBlockListNode = NULL;
pExtW->pOutputBlocks = NULL;
// pExtW->pOffsetList = NULL;
initResultSizeInfo(&pOperator->resultInfo, 512);
int32_t code = blockDataEnsureCapacity(pExtW->binfo.pRes, pOperator->resultInfo.capacity);
if (code == 0) {
code = resetAggSup(&pOperator->exprSupp, &pExtW->aggSup, pTaskInfo, pPhynode->window.pFuncs, NULL,
sizeof(int64_t) * 2 + POINTER_BYTES, pTaskInfo->id.str, pTaskInfo->streamInfo.pState,
&pTaskInfo->storageAPI.functionStore);
}
if (code == 0) {
code = resetExprSupp(&pExtW->scalarSupp, pTaskInfo, pPhynode->window.pProjs, NULL,
&pTaskInfo->storageAPI.functionStore);
}
if (code == 0) {
colDataDestroy(&pExtW->twAggSup.timeWindowData);
code = initExecTimeWindowInfo(&pExtW->twAggSup.timeWindowData, &pTaskInfo->window);
}
if (code == 0) {
cleanupGroupResInfo(&pExtW->groupResInfo);
}
blockDataDestroy(pExtW->pTmpBlock);
pExtW->pTmpBlock = NULL;
return code;
}
static EDealRes extWindowHasCountLikeFunc(SNode* pNode, void* res) {
if (QUERY_NODE_FUNCTION == nodeType(pNode)) {
SFunctionNode* pFunc = (SFunctionNode*)pNode;
if (fmIsCountLikeFunc(pFunc->funcId) || (pFunc->hasOriginalFunc && fmIsCountLikeFunc(pFunc->originalFuncId))) {
*(bool*)res = true;
return DEAL_RES_END;
}
}
return DEAL_RES_CONTINUE;
}
static int32_t extWindowCreateEmptyInputBlock(SOperatorInfo* pOperator, SSDataBlock** ppBlock) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SSDataBlock* pBlock = NULL;
if (!tsCountAlwaysReturnValue) {
return TSDB_CODE_SUCCESS;
}
SExternalWindowOperator* pExtW = pOperator->info;
if (!pExtW->hasCountFunc) {
return TSDB_CODE_SUCCESS;
}
code = createDataBlock(&pBlock);
if (code) {
return code;
}
pBlock->info.rows = 1;
pBlock->info.capacity = 0;
for (int32_t i = 0; i < pOperator->exprSupp.numOfExprs; ++i) {
SColumnInfoData colInfo = {0};
colInfo.hasNull = true;
colInfo.info.type = TSDB_DATA_TYPE_NULL;
colInfo.info.bytes = 1;
SExprInfo* pOneExpr = &pOperator->exprSupp.pExprInfo[i];
for (int32_t j = 0; j < pOneExpr->base.numOfParams; ++j) {
SFunctParam* pFuncParam = &pOneExpr->base.pParam[j];
if (pFuncParam->type == FUNC_PARAM_TYPE_COLUMN) {
int32_t slotId = pFuncParam->pCol->slotId;
int32_t numOfCols = taosArrayGetSize(pBlock->pDataBlock);
if (slotId >= numOfCols) {
code = taosArrayEnsureCap(pBlock->pDataBlock, slotId + 1);
QUERY_CHECK_CODE(code, lino, _end);
for (int32_t k = numOfCols; k < slotId + 1; ++k) {
void* tmp = taosArrayPush(pBlock->pDataBlock, &colInfo);
QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
}
}
} else if (pFuncParam->type == FUNC_PARAM_TYPE_VALUE) {
// do nothing
}
}
}
code = blockDataEnsureCapacity(pBlock, pBlock->info.rows);
QUERY_CHECK_CODE(code, lino, _end);
for (int32_t i = 0; i < blockDataGetNumOfCols(pBlock); ++i) {
SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);
QUERY_CHECK_NULL(pColInfoData, code, lino, _end, terrno);
colDataSetNULL(pColInfoData, 0);
}
*ppBlock = pBlock;
_end:
if (code != TSDB_CODE_SUCCESS) {
blockDataDestroy(pBlock);
qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
}
return code;
}
int32_t createExternalWindowOperator(SOperatorInfo* pDownstream, SPhysiNode* pNode, SExecTaskInfo* pTaskInfo,
SOperatorInfo** pOptrOut) {
SExternalWindowPhysiNode* pPhynode = (SExternalWindowPhysiNode*)pNode;
QRY_PARAM_CHECK(pOptrOut);
int32_t code = 0;
int32_t lino = 0;
SExternalWindowOperator* pExtW = taosMemoryCalloc(1, sizeof(SExternalWindowOperator));
SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
pOperator->pPhyNode = pNode;
if (!pExtW || !pOperator) {
code = terrno;
lino = __LINE__;
goto _error;
}
setOperatorInfo(pOperator, "ExternalWindowOperator", QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW, true, OP_NOT_OPENED,
pExtW, pTaskInfo);
if (pTaskInfo->pStreamRuntimeInfo != NULL){
pTaskInfo->pStreamRuntimeInfo->funcInfo.withExternalWindow = true;
}
SSDataBlock* pResBlock = createDataBlockFromDescNode(pPhynode->window.node.pOutputDataBlockDesc);
QUERY_CHECK_NULL(pResBlock, code, lino, _error, terrno);
initBasicInfo(&pExtW->binfo, pResBlock);
pExtW->primaryTsIndex = ((SColumnNode*)pPhynode->window.pTspk)->slotId;
pExtW->mode = pPhynode->window.pProjs ? EEXT_MODE_SCALAR : (pPhynode->window.indefRowsFunc ? EEXT_MODE_INDEFR_FUNC : EEXT_MODE_AGG);
pExtW->binfo.inputTsOrder = pPhynode->window.node.inputTsOrder = TSDB_ORDER_ASC;
pExtW->binfo.outputTsOrder = pExtW->binfo.inputTsOrder;
// pExtW->limitInfo = (SLimitInfo){0};
// initLimitInfo(pPhynode->window.node.pLimit, pPhynode->window.node.pSlimit, &pExtW->limitInfo);
if (pPhynode->window.pProjs) {
int32_t numOfScalarExpr = 0;
SExprInfo* pScalarExprInfo = NULL;
code = createExprInfo(pPhynode->window.pProjs, NULL, &pScalarExprInfo, &numOfScalarExpr);
QUERY_CHECK_CODE(code, lino, _error);
code = initExprSupp(&pExtW->scalarSupp, pScalarExprInfo, numOfScalarExpr, &pTaskInfo->storageAPI.functionStore);
QUERY_CHECK_CODE(code, lino, _error);
} else if (pExtW->mode == EEXT_MODE_AGG) {
if (pPhynode->window.pExprs != NULL) {
int32_t num = 0;
SExprInfo* pSExpr = NULL;
code = createExprInfo(pPhynode->window.pExprs, NULL, &pSExpr, &num);
QUERY_CHECK_CODE(code, lino, _error);
code = initExprSupp(&pExtW->scalarSupp, pSExpr, num, &pTaskInfo->storageAPI.functionStore);
if (code != TSDB_CODE_SUCCESS) {
goto _error;
}
}
size_t keyBufSize = sizeof(int64_t) * 2 + POINTER_BYTES;
initResultSizeInfo(&pOperator->resultInfo, 512);
code = blockDataEnsureCapacity(pExtW->binfo.pRes, pOperator->resultInfo.capacity);
QUERY_CHECK_CODE(code, lino, _error);
int32_t num = 0;
SExprInfo* pExprInfo = NULL;
code = createExprInfo(pPhynode->window.pFuncs, NULL, &pExprInfo, &num);
QUERY_CHECK_CODE(code, lino, _error);
code = initAggSup(&pOperator->exprSupp, &pExtW->aggSup, pExprInfo, num, keyBufSize, pTaskInfo->id.str, 0, 0);
QUERY_CHECK_CODE(code, lino, _error);
nodesWalkExprs(pPhynode->window.pFuncs, extWindowHasCountLikeFunc, &pExtW->hasCountFunc);
if (pExtW->hasCountFunc) {
code = extWindowCreateEmptyInputBlock(pOperator, &pExtW->pEmptyInputBlock);
QUERY_CHECK_CODE(code, lino, _error);
qDebug("%s ext window has CountLikeFunc", pOperator->pTaskInfo->id.str);
} else {
qDebug("%s ext window doesn't have CountLikeFunc", pOperator->pTaskInfo->id.str);
}
code = initExecTimeWindowInfo(&pExtW->twAggSup.timeWindowData, &pTaskInfo->window);
QUERY_CHECK_CODE(code, lino, _error);
} else {
size_t keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
if (pPhynode->window.pExprs != NULL) {
int32_t num = 0;
SExprInfo* pSExpr = NULL;
code = createExprInfo(pPhynode->window.pExprs, NULL, &pSExpr, &num);
QUERY_CHECK_CODE(code, lino, _error);
code = initExprSupp(&pExtW->scalarSupp, pSExpr, num, &pTaskInfo->storageAPI.functionStore);
if (code != TSDB_CODE_SUCCESS) {
goto _error;
}
}
initResultSizeInfo(&pOperator->resultInfo, 512);
code = blockDataEnsureCapacity(pResBlock, 512);
TSDB_CHECK_CODE(code, lino, _error);
int32_t numOfExpr = 0;
SExprInfo* pExprInfo = NULL;
code = createExprInfo(pPhynode->window.pFuncs, NULL, &pExprInfo, &numOfExpr);
TSDB_CHECK_CODE(code, lino, _error);
code = initAggSup(&pOperator->exprSupp, &pExtW->aggSup, pExprInfo, numOfExpr, keyBufSize, pTaskInfo->id.str,
pTaskInfo->streamInfo.pState, &pTaskInfo->storageAPI.functionStore);
TSDB_CHECK_CODE(code, lino, _error);
pOperator->exprSupp.hasWindowOrGroup = false;
code = setFunctionResultOutput(pOperator, &pExtW->binfo, &pExtW->aggSup, MAIN_SCAN, numOfExpr);
TSDB_CHECK_CODE(code, lino, _error);
code = filterInitFromNode((SNode*)pNode->pConditions, &pOperator->exprSupp.pFilterInfo, 0,
pTaskInfo->pStreamRuntimeInfo);
TSDB_CHECK_CODE(code, lino, _error);
pExtW->binfo.pRes = pResBlock;
pExtW->binfo.inputTsOrder = pNode->inputTsOrder;
pExtW->binfo.outputTsOrder = pNode->outputTsOrder;
code = setRowTsColumnOutputInfo(pOperator->exprSupp.pCtx, numOfExpr, &pExtW->pPseudoColInfo);
TSDB_CHECK_CODE(code, lino, _error);
}
initResultRowInfo(&pExtW->binfo.resultRowInfo);
pOperator->fpSet = createOperatorFpSet(doOpenExternalWindow, externalWindowNext, NULL, destroyExternalWindowOperatorInfo,
optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
setOperatorResetStateFn(pOperator, resetExternalWindowOperator);
code = appendDownstream(pOperator, &pDownstream, 1);
if (code != 0) {
goto _error;
}
*pOptrOut = pOperator;
return code;
_error:
if (pExtW != NULL) {
destroyExternalWindowOperatorInfo(pExtW);
}
destroyOperatorAndDownstreams(pOperator, &pDownstream, 1);
pTaskInfo->code = code;
qError("error happens at %s %d, code:%s", __func__, lino, tstrerror(code));
return code;
}
int64_t* extractTsCol(SSDataBlock* pBlock, int32_t primaryTsIndex, SExecTaskInfo* pTaskInfo) {
TSKEY* tsCols = NULL;
if (pBlock->pDataBlock != NULL && pBlock->info.dataLoad) {
SColumnInfoData* pColDataInfo = taosArrayGet(pBlock->pDataBlock, primaryTsIndex);
if (!pColDataInfo) {
pTaskInfo->code = terrno;
T_LONG_JMP(pTaskInfo->env, terrno);
}
tsCols = (int64_t*)pColDataInfo->pData;
if (tsCols[0] == 0) {
qWarn("%s at line %d.block start ts:%" PRId64 ",end ts:%" PRId64, __func__, __LINE__, tsCols[0],
tsCols[pBlock->info.rows - 1]);
}
if (tsCols[0] != 0 && (pBlock->info.window.skey == 0 && pBlock->info.window.ekey == 0)) {
int32_t code = blockDataUpdateTsWindow(pBlock, primaryTsIndex);
if (code != TSDB_CODE_SUCCESS) {
qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
pTaskInfo->code = code;
T_LONG_JMP(pTaskInfo->env, code);
}
}
}
return tsCols;
}
static int32_t getExtWinCurIdx(SExecTaskInfo* pTaskInfo) {
return pTaskInfo->pStreamRuntimeInfo->funcInfo.curIdx;
}
static void incExtWinCurIdx(SOperatorInfo* pOperator) {
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
pTaskInfo->pStreamRuntimeInfo->funcInfo.curIdx++;
}
static void setExtWinCurIdx(SOperatorInfo* pOperator, int32_t idx) {
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
pTaskInfo->pStreamRuntimeInfo->funcInfo.curIdx = idx;
}
static void incExtWinOutIdx(SOperatorInfo* pOperator) {
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
pTaskInfo->pStreamRuntimeInfo->funcInfo.curOutIdx++;
}
static const STimeWindow* getExtWindow(SOperatorInfo* pOperator, SExternalWindowOperator* pExtW, TSKEY ts) {
// TODO handle desc order
for (int32_t i = 0; i < pExtW->pWins->size; ++i) {
const STimeWindow* pWin = taosArrayGet(pExtW->pWins, i);
if (ts >= pWin->skey && ts < pWin->ekey) {
setExtWinCurIdx(pOperator, i);
return pWin;
}
}
return NULL;
}
static const STimeWindow* getExtNextWindow(SExternalWindowOperator* pExtW, SExecTaskInfo* pTaskInfo) {
int32_t curIdx = getExtWinCurIdx(pTaskInfo);
if (curIdx + 1 >= pExtW->pWins->size) return NULL;
return taosArrayGet(pExtW->pWins, curIdx + 1);
}
static int32_t getNextStartPos(STimeWindow win, const SDataBlockInfo* pBlockInfo, int32_t lastEndPos, int32_t order, int32_t* nextPos, int64_t* tsCol) {
bool ascQuery = order == TSDB_ORDER_ASC;
if (win.ekey <= pBlockInfo->window.skey && ascQuery) {
return -2;
}
//if (win.skey > pBlockInfo->window.ekey && !ascQuery) return -2;
if (win.skey > pBlockInfo->window.ekey && ascQuery) return -1;
//if (win.ekey < pBlockInfo->window.skey && !ascQuery) return -1;
while (true) {
if (win.ekey <= tsCol[lastEndPos + 1] && ascQuery) return -2;
if (win.skey <= tsCol[lastEndPos + 1] && ascQuery) break;
lastEndPos++;
}
*nextPos = lastEndPos + 1;
return 0;
}
static int32_t setExtWindowOutputBuf(SResultRowInfo* pResultRowInfo, const STimeWindow* win, SResultRow** pResult, int64_t tableGroupId, SqlFunctionCtx* pCtx,
int32_t numOfOutput, int32_t* rowEntryInfoOffset, SAggSupporter* pAggSup,
SExecTaskInfo* pTaskInfo) {
SResultRow* pResultRow = doSetResultOutBufByKey(pAggSup->pResultBuf, pResultRowInfo, (char*)&win->skey, TSDB_KEYSIZE,
true, tableGroupId, pTaskInfo, true, pAggSup, true);
if (pResultRow == NULL || pTaskInfo->code != 0) {
*pResult = NULL;
qError("failed to set result output buffer, error:%s", tstrerror(pTaskInfo->code));
return pTaskInfo->code;
}
qDebug("current result rows num:%d", tSimpleHashGetSize(pAggSup->pResultRowHashTable));
// set time window for current result
TAOS_SET_POBJ_ALIGNED(&pResultRow->win, win);
*pResult = pResultRow;
pTaskInfo->code = setResultRowInitCtx(pResultRow, pCtx, numOfOutput, rowEntryInfoOffset);
if (pTaskInfo->code) {
*pResult = NULL;
qError("failed to set result row ctx, error:%s", tstrerror(pTaskInfo->code));
return pTaskInfo->code;
}
return TSDB_CODE_SUCCESS;
}
static int32_t extWindowDoHashAgg(SOperatorInfo* pOperator, int32_t startPos, int32_t forwardRows,
SSDataBlock* pInputBlock) {
if (forwardRows == 0) return 0;
SExprSupp* pSup = &pOperator->exprSupp;
SExternalWindowOperator* pExtW = pOperator->info;
return applyAggFunctionOnPartialTuples(pOperator->pTaskInfo, pSup->pCtx, &pExtW->twAggSup.timeWindowData, startPos,
forwardRows, pInputBlock->info.rows, pSup->numOfExprs);
}
static SSDataBlock* extWindowGetOutputBlock(SOperatorInfo* pOperator, int32_t winIdx) {
SExternalWindowOperator* pExtW = pOperator->info;
SSDataBlock* pBlock = NULL;
SBlockList* pBlockList = taosArrayGet(pExtW->pOutputBlocks, winIdx);
int32_t code = blockListGetLastBlock(pBlockList, &pBlock);
if (code != 0) terrno = code;
return pBlock;
}
static int32_t extWindowCopyRows(SOperatorInfo* pOperator, SSDataBlock* pInputBlock, int32_t startPos,
int32_t forwardRows) {
if (forwardRows == 0) return 0;
SExternalWindowOperator* pExtW = pOperator->info;
SSDataBlock* pResBlock = extWindowGetOutputBlock(pOperator, getExtWinCurIdx(pOperator->pTaskInfo));
SExprSupp* pExprSup = &pExtW->scalarSupp;
if (!pResBlock) {
qError("%s failed to get output block for ext window:%s", GET_TASKID(pOperator->pTaskInfo), tstrerror(terrno));
return terrno;
}
int32_t rowsToCopy = forwardRows;
int32_t code = 0;
if (!pExtW->pTmpBlock)
code = createOneDataBlock(pInputBlock, false, &pExtW->pTmpBlock);
else
blockDataCleanup(pExtW->pTmpBlock);
if (code) {
qError("%s failed to create datablock:%s", GET_TASKID(pOperator->pTaskInfo), tstrerror(terrno));
return code;
}
code = blockDataEnsureCapacity(pExtW->pTmpBlock, TMAX(1, rowsToCopy));
if (code) {
qError("%s failed to ensure capacity:%s", GET_TASKID(pOperator->pTaskInfo), tstrerror(terrno));
return code;
}
if (rowsToCopy > 0) {
code = blockDataMergeNRows(pExtW->pTmpBlock, pInputBlock, startPos, rowsToCopy);
if (code == 0) {
code = projectApplyFunctionsWithSelect(pExprSup->pExprInfo, pResBlock, pExtW->pTmpBlock, pExprSup->pCtx, pExprSup->numOfExprs,
NULL, GET_STM_RTINFO(pOperator->pTaskInfo), true);
}
}
if (code != 0) return code;
return code;
}
static int32_t hashExternalWindowProject(SOperatorInfo* pOperator, SSDataBlock* pInputBlock) {
SExprSupp* pSup = &pOperator->exprSupp;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SExternalWindowOperator* pExtW = pOperator->info;
SqlFunctionCtx* pCtx = NULL;
int64_t* tsCol = extractTsCol(pInputBlock, pExtW->primaryTsIndex, pTaskInfo);
TSKEY ts = getStartTsKey(&pInputBlock->info.window, tsCol);
const STimeWindow* pWin = getExtWindow(pOperator, pExtW, ts);
bool ascScan = pExtW->binfo.inputTsOrder == TSDB_ORDER_ASC;
int32_t tsOrder = pExtW->binfo.inputTsOrder;
int32_t startPos = 0;
if (!pWin) return 0;
TSKEY ekey = ascScan ? pWin->ekey : pWin->skey;
int32_t forwardRows =
getNumOfRowsInTimeWindow(&pInputBlock->info, tsCol, startPos, ekey-1, binarySearchForKey, NULL, tsOrder);
int32_t code = extWindowCopyRows(pOperator, pInputBlock, startPos, forwardRows);
while (code == 0 && pInputBlock->info.rows > startPos + forwardRows) {
pWin = getExtNextWindow(pExtW, pTaskInfo);
if (!pWin) break;
incExtWinCurIdx(pOperator);
startPos = startPos + forwardRows;
ekey = ascScan ? pWin->ekey : pWin->skey;
forwardRows =
getNumOfRowsInTimeWindow(&pInputBlock->info, tsCol, startPos, ekey-1, binarySearchForKey, NULL, tsOrder);
code = extWindowCopyRows(pOperator, pInputBlock, startPos, forwardRows);
}
return code;
}
static int32_t extWindowDoIndefRows(SOperatorInfo* pOperator, SSDataBlock* pRes, SSDataBlock* pBlock) {
SExternalWindowOperator* pExtW = pOperator->info;
SOptrBasicInfo* pInfo = &pExtW->binfo;
SExprSupp* pSup = &pOperator->exprSupp;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
int32_t order = pInfo->inputTsOrder;
int32_t scanFlag = pBlock->info.scanFlag;
int32_t code = TSDB_CODE_SUCCESS, lino = 0;
SExprSupp* pScalarSup = &pExtW->scalarSupp;
if (pScalarSup->pExprInfo != NULL) {
TAOS_CHECK_EXIT(projectApplyFunctions(pScalarSup->pExprInfo, pBlock, pBlock, pScalarSup->pCtx, pScalarSup->numOfExprs,
pExtW->pPseudoColInfo, GET_STM_RTINFO(pOperator->pTaskInfo)));
}
TAOS_CHECK_EXIT(setInputDataBlock(pSup, pBlock, order, scanFlag, false));
TAOS_CHECK_EXIT(blockDataEnsureCapacity(pRes, pRes->info.rows + pBlock->info.rows));
TAOS_CHECK_EXIT(projectApplyFunctions(pSup->pExprInfo, pRes, pBlock, pSup->pCtx, pSup->numOfExprs,
pExtW->pPseudoColInfo, GET_STM_RTINFO(pOperator->pTaskInfo)));
_exit:
if (code) {
qError("%s %s failed at line %d since %s", pOperator->pTaskInfo->id.str, __FUNCTION__, lino, tstrerror(code));
}
return code;
}
static int32_t extWindowHandleIndefRows(SOperatorInfo* pOperator, SSDataBlock* pInputBlock, int32_t startPos,
int32_t forwardRows) {
if (forwardRows == 0) return 0;
SExternalWindowOperator* pExtW = pOperator->info;
SSDataBlock* pResBlock = extWindowGetOutputBlock(pOperator, getExtWinCurIdx(pOperator->pTaskInfo));
if (!pResBlock) {
qError("%s failed to get output block for ext window:%s", GET_TASKID(pOperator->pTaskInfo), tstrerror(terrno));
return terrno;
}
int32_t rowsToCopy = forwardRows;
int32_t code = 0;
if (!pExtW->pTmpBlock)
code = createOneDataBlock(pInputBlock, false, &pExtW->pTmpBlock);
else
blockDataCleanup(pExtW->pTmpBlock);
if (code) {
qError("%s failed to create datablock:%s", GET_TASKID(pOperator->pTaskInfo), tstrerror(terrno));
return code;
}
code = blockDataEnsureCapacity(pExtW->pTmpBlock, TMAX(1, rowsToCopy));
if (code) {
qError("%s failed to ensure capacity:%s", GET_TASKID(pOperator->pTaskInfo), tstrerror(terrno));
return code;
}
if (rowsToCopy > 0) {
code = blockDataMergeNRows(pExtW->pTmpBlock, pInputBlock, startPos, rowsToCopy);
if (code == 0) {
code = extWindowDoIndefRows(pOperator, pResBlock, pExtW->pTmpBlock);
}
}
return code;
}
static int32_t hashExternalWindowIndefRows(SOperatorInfo* pOperator, SSDataBlock* pInputBlock) {
SExprSupp* pSup = &pOperator->exprSupp;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SExternalWindowOperator* pExtW = pOperator->info;
SqlFunctionCtx* pCtx = NULL;
int64_t* tsCol = extractTsCol(pInputBlock, pExtW->primaryTsIndex, pTaskInfo);
TSKEY ts = getStartTsKey(&pInputBlock->info.window, tsCol);
const STimeWindow* pWin = getExtWindow(pOperator, pExtW, ts);
bool ascScan = pExtW->binfo.inputTsOrder == TSDB_ORDER_ASC;
int32_t tsOrder = pExtW->binfo.inputTsOrder;
int32_t startPos = 0;
if (!pWin) return 0;
TSKEY ekey = ascScan ? pWin->ekey : pWin->skey;
int32_t forwardRows =
getNumOfRowsInTimeWindow(&pInputBlock->info, tsCol, startPos, ekey-1, binarySearchForKey, NULL, tsOrder);
int32_t code = extWindowHandleIndefRows(pOperator, pInputBlock, startPos, forwardRows);
while (code == 0 && pInputBlock->info.rows > startPos + forwardRows) {
pWin = getExtNextWindow(pExtW, pTaskInfo);
if (!pWin) break;
incExtWinCurIdx(pOperator);
startPos = startPos + forwardRows;
ekey = ascScan ? pWin->ekey : pWin->skey;
forwardRows =
getNumOfRowsInTimeWindow(&pInputBlock->info, tsCol, startPos, ekey-1, binarySearchForKey, NULL, tsOrder);
code = extWindowHandleIndefRows(pOperator, pInputBlock, startPos, forwardRows);
}
return code;
}
static int32_t extWindowProcessEmptyWins(SOperatorInfo* pOperator, SSDataBlock* pBlock, bool allRemains) {
int32_t code = 0, lino = 0;
SExternalWindowOperator* pExtW = (SExternalWindowOperator*)pOperator->info;
bool ascScan = pExtW->binfo.inputTsOrder == TSDB_ORDER_ASC;
int32_t currIdx = getExtWinCurIdx(pOperator->pTaskInfo);
SExprSupp* pSup = &pOperator->exprSupp;
if (NULL == pExtW->pEmptyInputBlock) {
goto _exit;
}
int32_t endIdx = allRemains ? (pExtW->pWins->size - 1) : (currIdx - 1);
SResultRowInfo* pResultRowInfo = &pExtW->binfo.resultRowInfo;
SResultRow* pResult = NULL;
SSDataBlock* pInput = pExtW->pEmptyInputBlock;
if ((pExtW->lastWinId + 1) <= endIdx) {
TAOS_CHECK_EXIT(setInputDataBlock(pSup, pExtW->pEmptyInputBlock, pExtW->binfo.inputTsOrder, MAIN_SCAN, true));
}
for (int32_t i = pExtW->lastWinId + 1; i <= endIdx; ++i) {
const STimeWindow* pWin = taosArrayGet(pExtW->pWins, i);
setExtWinCurIdx(pOperator, i);
qDebug("%s %dth ext empty window start:%" PRId64 ", end:%" PRId64 ", ascScan:%d",
GET_TASKID(pOperator->pTaskInfo), i, pWin->skey, pWin->ekey, ascScan);
TAOS_CHECK_EXIT(setExtWindowOutputBuf(pResultRowInfo, pWin, &pResult, pInput->info.id.groupId, pSup->pCtx, pSup->numOfExprs,
pSup->rowEntryInfoOffset, &pExtW->aggSup, pOperator->pTaskInfo));
updateTimeWindowInfo(&pExtW->twAggSup.timeWindowData, pWin, 1);
code = extWindowDoHashAgg(pOperator, 0, 1, pInput);
pExtW->lastWinId = i;
TAOS_CHECK_EXIT(code);
}
_exit:
if (code) {
qError("%s %s failed at line %d since %s", pOperator->pTaskInfo->id.str, __FUNCTION__, lino, tstrerror(code));
} else {
if (pBlock) {
TAOS_CHECK_EXIT(setInputDataBlock(pSup, pBlock, pExtW->binfo.inputTsOrder, MAIN_SCAN, true));
}
if (!allRemains) {
setExtWinCurIdx(pOperator, currIdx);
}
}
return code;
}
static void hashExternalWindowAgg(SOperatorInfo* pOperator, SSDataBlock* pInputBlock) {
SExternalWindowOperator* pExtW = (SExternalWindowOperator*)pOperator->info;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SExprSupp* pSup = &pOperator->exprSupp;
SResultRowInfo* pResultRowInfo = &pExtW->binfo.resultRowInfo;
int32_t startPos = 0;
int32_t numOfOutput = pSup->numOfExprs;
int64_t* tsCols = extractTsCol(pInputBlock, pExtW->primaryTsIndex, pTaskInfo);
bool ascScan = pExtW->binfo.inputTsOrder == TSDB_ORDER_ASC;
TSKEY ts = getStartTsKey(&pInputBlock->info.window, tsCols);
SResultRow* pResult = NULL;
int32_t ret = 0;
const STimeWindow* pWin = getExtWindow(pOperator, pExtW, ts);
if (pWin == NULL) {
qError("%s failed to get time window for ts:%" PRId64 ", error:%s", GET_TASKID(pOperator->pTaskInfo), ts, tstrerror(terrno));
return;
}
ret = extWindowProcessEmptyWins(pOperator, pInputBlock, false);
if (ret != 0) {
T_LONG_JMP(pTaskInfo->env, ret);
}
qDebug("%s ext window1 start:%" PRId64 ", end:%" PRId64 ", ts:%" PRId64 ", ascScan:%d",
GET_TASKID(pOperator->pTaskInfo), pWin->skey, pWin->ekey, ts, ascScan);
SExprSupp* pScalarSup = &pExtW->scalarSupp;
if (pScalarSup->pExprInfo != NULL) {
ret = projectApplyFunctions(pScalarSup->pExprInfo, pInputBlock, pInputBlock, pScalarSup->pCtx, pScalarSup->numOfExprs,
pExtW->pPseudoColInfo, GET_STM_RTINFO(pOperator->pTaskInfo));
if (ret != 0) {
T_LONG_JMP(pTaskInfo->env, ret);
}
}
STimeWindow win = *pWin;
ret = setExtWindowOutputBuf(pResultRowInfo, &win, &pResult, pInputBlock->info.id.groupId, pSup->pCtx, numOfOutput,
pSup->rowEntryInfoOffset, &pExtW->aggSup, pTaskInfo);
if (ret != 0 || !pResult) {
T_LONG_JMP(pTaskInfo->env, ret);
}
TSKEY ekey = ascScan ? win.ekey : win.skey;
int32_t forwardRows = getNumOfRowsInTimeWindow(&pInputBlock->info, tsCols, startPos, ekey - 1, binarySearchForKey, NULL,
pExtW->binfo.inputTsOrder);
updateTimeWindowInfo(&pExtW->twAggSup.timeWindowData, &win, 1);
ret = extWindowDoHashAgg(pOperator, startPos, forwardRows, pInputBlock);
pExtW->lastWinId = getExtWinCurIdx(pTaskInfo);
if (ret != 0) {
T_LONG_JMP(pTaskInfo->env, ret);
}
int32_t nextPosGot = 0;
while (1) {
int32_t prevEndPos = forwardRows + startPos - 1;
if (prevEndPos >= pInputBlock->info.rows) {
break;
}
pWin = getExtNextWindow(pExtW, pTaskInfo);
if (!pWin)
break;
else
win = *pWin;
qDebug("%s ext window2 start:%" PRId64 ", end:%" PRId64 ", ts:%" PRId64 ", ascScan:%d",
GET_TASKID(pOperator->pTaskInfo), win.skey, win.ekey, ts, ascScan);
nextPosGot = getNextStartPos(win, &pInputBlock->info, prevEndPos, pExtW->binfo.inputTsOrder, &startPos, tsCols);
if (-1 == nextPosGot) {
qDebug("%s ignore current block", GET_TASKID(pOperator->pTaskInfo));
break;
}
if (-2 == nextPosGot) {
qDebug("%s skip current window", GET_TASKID(pOperator->pTaskInfo));
incExtWinCurIdx(pOperator);
continue;
}
incExtWinCurIdx(pOperator);
ret = extWindowProcessEmptyWins(pOperator, pInputBlock, false);
if (ret != 0) {
T_LONG_JMP(pTaskInfo->env, ret);
}
ekey = ascScan ? win.ekey : win.skey;
forwardRows = getNumOfRowsInTimeWindow(&pInputBlock->info, tsCols, startPos, ekey - 1, binarySearchForKey, NULL,
pExtW->binfo.inputTsOrder);
ret = setExtWindowOutputBuf(pResultRowInfo, &win, &pResult, pInputBlock->info.id.groupId, pSup->pCtx,
numOfOutput, pSup->rowEntryInfoOffset, &pExtW->aggSup, pTaskInfo);
if (ret != 0 || !pResult) T_LONG_JMP(pTaskInfo->env, ret);
updateTimeWindowInfo(&pExtW->twAggSup.timeWindowData, &win, 1);
ret = extWindowDoHashAgg(pOperator, startPos, forwardRows, pInputBlock);
pExtW->lastWinId = getExtWinCurIdx(pTaskInfo);
if (ret != 0) {
T_LONG_JMP(pTaskInfo->env, ret);
}
}
}
static int32_t doOpenExternalWindow(SOperatorInfo* pOperator) {
if (OPTR_IS_OPENED(pOperator)) return TSDB_CODE_SUCCESS;
int32_t code = 0;
int32_t lino = 0;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SOperatorInfo* pDownstream = pOperator->pDownstream[0];
SExternalWindowOperator* pExtW = pOperator->info;
SExprSupp* pSup = &pOperator->exprSupp;
pTaskInfo->pStreamRuntimeInfo->funcInfo.extWinProjMode = (pExtW->mode != EEXT_MODE_AGG);
int32_t scanFlag = MAIN_SCAN;
int64_t st = taosGetTimestampUs();
if (!pExtW->pWins) {
size_t size = taosArrayGetSize(pTaskInfo->pStreamRuntimeInfo->funcInfo.pStreamPesudoFuncVals);
pExtW->pWins = taosArrayInit(size, sizeof(STimeWindow));
if (!pExtW->pWins) QUERY_CHECK_CODE(terrno, lino, _exit);
pExtW->pOutputBlocks = taosArrayInit(size, sizeof(SBlockList));
if (!pExtW->pOutputBlocks) QUERY_CHECK_CODE(terrno, lino, _exit);
// pExtW->pOffsetList = taosArrayInit(size, sizeof(SLimitInfo));
// if (!pExtW->pOffsetList) QUERY_CHECK_CODE(terrno, lino, _end);
for (int32_t i = 0; i < size; ++i) {
SSTriggerCalcParam* pParam = taosArrayGet(pTaskInfo->pStreamRuntimeInfo->funcInfo.pStreamPesudoFuncVals, i);
STimeWindow win = {.skey = pParam->wstart, .ekey = pParam->wend};
if (pTaskInfo->pStreamRuntimeInfo->funcInfo.triggerType != STREAM_TRIGGER_SLIDING) {
win.ekey++;
}
TSDB_CHECK_NULL(taosArrayPush(pExtW->pWins, &win), code, lino, _exit, terrno);
SBlockList bl = {.pSrcBlock = pExtW->binfo.pRes, .pBlocks = 0, .blockRowNumThreshold = 4096};
code = blockListInit(&bl, 4096);
if (code != 0) QUERY_CHECK_CODE(code, lino, _exit);
TSDB_CHECK_NULL(taosArrayPush(pExtW->pOutputBlocks, &bl), code, lino, _exit, terrno);
}
pExtW->outputWinId = pTaskInfo->pStreamRuntimeInfo->funcInfo.curIdx;
pExtW->lastWinId = -1;
}
while (1) {
SSDataBlock* pBlock = getNextBlockFromDownstream(pOperator, 0);
if (pBlock == NULL) {
if (EEXT_MODE_AGG == pExtW->mode) {
TAOS_CHECK_EXIT(extWindowProcessEmptyWins(pOperator, pBlock, true));
}
break;
}
printDataBlock(pBlock, __func__, "externalwindow");
qDebug("ext windowpExtW->mode:%d", pExtW->mode);
switch (pExtW->mode) {
case EEXT_MODE_SCALAR:
TAOS_CHECK_EXIT(hashExternalWindowProject(pOperator, pBlock));
break;
case EEXT_MODE_AGG:
hashExternalWindowAgg(pOperator, pBlock);
break;
case EEXT_MODE_INDEFR_FUNC:
TAOS_CHECK_EXIT(hashExternalWindowIndefRows(pOperator, pBlock));
break;
default:
break;
}
}
OPTR_SET_OPENED(pOperator);
if (pExtW->mode == EEXT_MODE_AGG) {
qDebug("ext window before dump final rows num:%d", tSimpleHashGetSize(pExtW->aggSup.pResultRowHashTable));
code = initGroupedResultInfo(&pExtW->groupResInfo, pExtW->aggSup.pResultRowHashTable, pExtW->binfo.inputTsOrder);
QUERY_CHECK_CODE(code, lino, _exit);
qDebug("ext window after dump final rows num:%d", tSimpleHashGetSize(pExtW->aggSup.pResultRowHashTable));
}
_exit:
if (code != 0) {
qError("%s failed at line %d since:%s", __func__, lino, tstrerror(code));
pTaskInfo->code = code;
T_LONG_JMP(pTaskInfo->env, code);
}
return code;
}
static int32_t externalWindowNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
int32_t code = 0;
int32_t lino = 0;
SExternalWindowOperator* pExtW = pOperator->info;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SExprSupp* pSup = &pOperator->exprSupp;
if (pOperator->status == OP_EXEC_DONE) {
*ppRes = NULL;
return code;
}
SSDataBlock* pBlock = pExtW->binfo.pRes;
blockDataCleanup(pBlock);
code = pOperator->fpSet._openFn(pOperator);
QUERY_CHECK_CODE(code, lino, _end);
while (1) {
if (pExtW->mode == EEXT_MODE_SCALAR || pExtW->mode == EEXT_MODE_INDEFR_FUNC) {
if (pExtW->outputWinId >= taosArrayGetSize(pExtW->pOutputBlocks)) {
pBlock->info.rows = 0;
break;
}
SBlockList* pList = taosArrayGet(pExtW->pOutputBlocks, pExtW->outputWinId);
// SLimitInfo* pLimitInfo = taosArrayGet(pExtW->pOffsetList, pExtW->outputWinId);
if (pExtW->pOutputBlockListNode == NULL) {
pExtW->pOutputBlockListNode = tdListGetHead(pList->pBlocks);
} else {
if (!pExtW->pOutputBlockListNode->dl_next_) {
pExtW->pOutputBlockListNode = NULL;
pExtW->outputWinId++;
incExtWinOutIdx(pOperator);
continue;
}
pExtW->pOutputBlockListNode = pExtW->pOutputBlockListNode->dl_next_;
}
if (pExtW->pOutputBlockListNode) {
code = blockDataMerge(pBlock, *(SSDataBlock**)pExtW->pOutputBlockListNode->data);
if (code != 0) goto _end;
// int32_t status = handleLimitOffset(pOperator, pLimitInfo, pBlock);
// if (status == EXTERNAL_RETRIEVE_CONTINUE) {
// continue;
// } else if (status == EXTERNAL_RETRIEVE_NEXT_WINDOW) {
// pExtW->pOutputBlockListNode = NULL;
// pExtW->outputWinId++;
// incExtWinOutIdx(pOperator);
// continue;
// } else if (status == EXTERNAL_RETRIEVE_DONE) {
// // do nothing, just break
// }
} else {
pExtW->outputWinId++;
incExtWinOutIdx(pOperator);
continue;
}
break;
} else {
doBuildResultDatablock(pOperator, &pExtW->binfo, &pExtW->groupResInfo, pExtW->aggSup.pResultBuf);
bool hasRemain = hasRemainResults(&pExtW->groupResInfo);
if (!hasRemain) {
setOperatorCompleted(pOperator);
break;
}
if (pExtW->binfo.pRes->info.rows > 0) break;
}
}
pOperator->resultInfo.totalRows += pExtW->binfo.pRes->info.rows;
_end:
if (code != 0) {
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
pTaskInfo->code = code;
T_LONG_JMP(pTaskInfo->env, code);
}
(*ppRes) = (pExtW->binfo.pRes->info.rows == 0) ? NULL : pExtW->binfo.pRes;
if (pTaskInfo->execModel == OPTR_EXEC_MODEL_STREAM && (*ppRes)) {
printDataBlock(*ppRes, getStreamOpName(pOperator->operatorType), GET_TASKID(pTaskInfo));
}
return code;
}
static int32_t setSingleOutputTupleBuf(SResultRowInfo* pResultRowInfo, const STimeWindow* pWin, SResultRow** pResult,
SExprSupp* pExprSup, SAggSupporter* pAggSup) {
if (*pResult == NULL) {
*pResult = getNewResultRow(pAggSup->pResultBuf, &pAggSup->currentPageId, pAggSup->resultRowSize);
if (!*pResult) {
qError("get new resultRow failed, err:%s", tstrerror(terrno));
return terrno;
}
pResultRowInfo->cur = (SResultRowPosition){.pageId = (*pResult)->pageId, .offset = (*pResult)->offset};
}
(*pResult)->win = *pWin;
return setResultRowInitCtx((*pResult), pExprSup->pCtx, pExprSup->numOfExprs, pExprSup->rowEntryInfoOffset);
}
static int32_t doMergeAlignExtWindowAgg(SOperatorInfo* pOperator, SResultRowInfo* pResultRowInfo, SSDataBlock* pBlock, SSDataBlock* pResultBlock) {
SMergeAlignedExternalWindowOperator* pMlExtInfo = pOperator->info;
SExternalWindowOperator* pExtW = pMlExtInfo->pExtW;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SExprSupp* pSup = &pOperator->exprSupp;
int32_t code = 0;
int32_t startPos = 0;
int64_t* tsCols = extractTsCol(pBlock, pExtW->primaryTsIndex, pTaskInfo);
TSKEY ts = getStartTsKey(&pBlock->info.window, tsCols);
const STimeWindow *pWin = getExtWindow(pOperator, pExtW, ts);
if (pWin == NULL) {
qError("failed to get time window for ts:%" PRId64 ", index:%d, error:%s", ts, pExtW->primaryTsIndex, tstrerror(terrno));
T_LONG_JMP(pTaskInfo->env, TSDB_CODE_INVALID_PARA);
}
code = setSingleOutputTupleBuf(pResultRowInfo, pWin, &pMlExtInfo->pResultRow, pSup, &pExtW->aggSup);
if (code != 0 || pMlExtInfo->pResultRow == NULL) {
T_LONG_JMP(pTaskInfo->env, code);
}
int32_t currPos = startPos;
pMlExtInfo->curTs = pWin->skey;
while (++currPos < pBlock->info.rows) {
if (tsCols[currPos] == pMlExtInfo->curTs) continue;
qDebug("current ts:%" PRId64 ", startPos:%d, currPos:%d, tsCols[currPos]:%" PRId64,
pMlExtInfo->curTs, startPos, currPos, tsCols[currPos]);
code = applyAggFunctionOnPartialTuples(pTaskInfo, pSup->pCtx, &pExtW->twAggSup.timeWindowData, startPos,
currPos - startPos, pBlock->info.rows, pSup->numOfExprs);
if (code != 0) {
T_LONG_JMP(pTaskInfo->env, code);
}
finalizeResultRows(pExtW->aggSup.pResultBuf, &pResultRowInfo->cur, pSup, pResultBlock, pTaskInfo);
resetResultRow(pMlExtInfo->pResultRow, pExtW->aggSup.resultRowSize - sizeof(SResultRow));
pWin = getExtNextWindow(pExtW, pTaskInfo);
if (!pWin) break;
qDebug("ext window align2 start:%" PRId64 ", end:%" PRId64, pWin->skey, pWin->ekey);
incExtWinCurIdx(pOperator);
startPos = currPos;
code = setSingleOutputTupleBuf(pResultRowInfo, pWin, &pMlExtInfo->pResultRow, pSup, &pExtW->aggSup);
if (code != 0 || pMlExtInfo->pResultRow == NULL) {
T_LONG_JMP(pTaskInfo->env, code);
}
pMlExtInfo->curTs = pWin->skey;
}
code = applyAggFunctionOnPartialTuples(pTaskInfo, pSup->pCtx, &pExtW->twAggSup.timeWindowData, startPos,
currPos - startPos, pBlock->info.rows, pSup->numOfExprs);
if (code != 0) {
T_LONG_JMP(pTaskInfo->env, code);
}
return code;
}
static int32_t doMergeAlignExtWindowProject(SOperatorInfo* pOperator, SResultRowInfo* pResultRowInfo, SSDataBlock* pBlock,
SSDataBlock* pResultBlock) {
SExternalWindowOperator* pExtW = pOperator->info;
SExprSupp* pExprSup = &pExtW->scalarSupp;
int32_t code = projectApplyFunctions(pExprSup->pExprInfo, pResultBlock, pBlock, pExprSup->pCtx, pExprSup->numOfExprs, NULL,
GET_STM_RTINFO(pOperator->pTaskInfo));
return code;
}
void doMergeAlignExternalWindow(SOperatorInfo* pOperator) {
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SMergeAlignedExternalWindowOperator* pMlExtInfo = pOperator->info;
SExternalWindowOperator* pExtW = pMlExtInfo->pExtW;
SResultRow* pResultRow = NULL;
int32_t code = 0;
SSDataBlock* pRes = pExtW->binfo.pRes;
SExprSupp* pSup = &pOperator->exprSupp;
int32_t lino = 0;
while (1) {
SSDataBlock* pBlock = getNextBlockFromDownstream(pOperator, 0);
if (pBlock == NULL) {
// close last time window
if (pMlExtInfo->curTs != INT64_MIN && EEXT_MODE_AGG == pExtW->mode) {
finalizeResultRows(pMlExtInfo->pExtW->aggSup.pResultBuf, &pExtW->binfo.resultRowInfo.cur, pSup, pRes, pTaskInfo);
resetResultRow(pMlExtInfo->pResultRow,pExtW->aggSup.resultRowSize - sizeof(SResultRow));
}
setOperatorCompleted(pOperator);
break;
}
pRes->info.scanFlag = pBlock->info.scanFlag;
code = setInputDataBlock(pSup, pBlock, pExtW->binfo.inputTsOrder, pBlock->info.scanFlag, true);
QUERY_CHECK_CODE(code, lino, _end);
printDataBlock(pBlock, __func__, "externalwindowAlign");
qDebug("ext windowpExtWAlign->scalarMode:%d", pExtW->mode);
if (EEXT_MODE_SCALAR == pExtW->mode) {
code = doMergeAlignExtWindowProject(pOperator, &pExtW->binfo.resultRowInfo, pBlock, pRes);
} else {
code = doMergeAlignExtWindowAgg(pOperator, &pExtW->binfo.resultRowInfo, pBlock, pRes);
}
// int32_t status = handleLimitOffset(pOperator, pLimitInfo, pBlock);
// if (status == EXTERNAL_RETRIEVE_CONTINUE) {
// continue;
// } else if (status == EXTERNAL_RETRIEVE_NEXT_WINDOW) {
// pExtW->pOutputBlockListNode = NULL;
// pExtW->outputWinId++;
// incExtWinOutIdx(pOperator);
// continue;
// } else if (status == EXTERNAL_RETRIEVE_DONE) {
// // do nothing, just break
// }
}
_end:
if (code != 0) {
qError("%s failed at line %d since:%s", __func__, lino, tstrerror(code));
pTaskInfo->code = code;
T_LONG_JMP(pTaskInfo->env, code);
}
}