1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374 |
- // Copyright 2022 EMQ Technologies Co., Ltd.
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- package operator
- import (
- "errors"
- "fmt"
- "github.com/lf-edge/ekuiper/internal/conf"
- "github.com/lf-edge/ekuiper/internal/topo/context"
- "github.com/lf-edge/ekuiper/internal/xsql"
- "github.com/lf-edge/ekuiper/pkg/cast"
- "reflect"
- "strings"
- "testing"
- )
- func TestProjectPlan_Apply1(t *testing.T) {
- var tests = []struct {
- sql string
- data *xsql.Tuple
- result []map[string]interface{}
- }{
- { //0
- sql: "SELECT a FROM test",
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": "val_a",
- },
- Metadata: xsql.Metadata{
- "id": 45,
- "other": "mock",
- },
- },
- result: []map[string]interface{}{{
- "a": "val_a",
- "__meta": xsql.Metadata{
- "id": 45,
- "other": "mock",
- },
- }},
- },
- { //1
- sql: "SELECT b FROM test",
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": "val_a",
- },
- },
- result: []map[string]interface{}{{}},
- },
- { //2
- sql: "SELECT ts FROM test",
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": "val_a",
- "ts": cast.TimeFromUnixMilli(1568854573431),
- },
- },
- result: []map[string]interface{}{{
- "ts": cast.TimeFromUnixMilli(1568854573431),
- }},
- },
- //Schemaless may return a message without selecting column
- { //3
- sql: "SELECT ts FROM test",
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": "val_a",
- "ts2": cast.TimeFromUnixMilli(1568854573431),
- },
- },
- result: []map[string]interface{}{{}},
- },
- { //4
- sql: "SELECT A FROM test",
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": "val_a",
- },
- },
- result: []map[string]interface{}{{
- "A": "val_a",
- }},
- },
- //5
- {
- sql: `SELECT "value" FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{},
- },
- result: []map[string]interface{}{{
- "kuiper_field_0": "value",
- }},
- },
- //6
- {
- sql: `SELECT 3.4 FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{},
- },
- result: []map[string]interface{}{{
- "kuiper_field_0": 3.4,
- }},
- },
- //7
- {
- sql: `SELECT 5 FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{},
- },
- result: []map[string]interface{}{{
- "kuiper_field_0": 5,
- }},
- },
- //8
- {
- sql: `SELECT a, "value" AS b FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": "val_a",
- },
- },
- result: []map[string]interface{}{{
- "a": "val_a",
- "b": "value",
- }},
- },
- //9
- {
- sql: `SELECT a, "value" AS b, 3.14 as Pi, 0 as Zero FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": "val_a",
- },
- },
- result: []map[string]interface{}{{
- "a": "val_a",
- "b": "value",
- "Pi": 3.14,
- "Zero": 0,
- }},
- },
- //10
- {
- sql: `SELECT a->b AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": map[string]interface{}{"b": "hello"},
- },
- },
- result: []map[string]interface{}{{
- "ab": "hello",
- }},
- },
- //11
- {
- sql: `SELECT a->b AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": map[string]interface{}(nil),
- },
- },
- result: []map[string]interface{}{{}},
- },
- //12
- {
- sql: `SELECT a->b AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "name": "name",
- },
- },
- result: []map[string]interface{}{{}},
- },
- //13
- {
- sql: `SELECT a->b AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": "commonstring",
- },
- },
- result: []map[string]interface{}{{}},
- },
- //14
- {
- sql: `SELECT a[0]->b AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": []interface{}{
- map[string]interface{}{"b": "hello1"},
- map[string]interface{}{"b": "hello2"},
- },
- },
- },
- result: []map[string]interface{}{{
- "ab": "hello1",
- }},
- },
- //15
- {
- sql: `SELECT a[0]->b AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": []map[string]interface{}{
- {"b": "hello1"},
- {"b": "hello2"},
- },
- },
- },
- result: []map[string]interface{}{{
- "ab": "hello1",
- }},
- },
- //16
- {
- sql: `SELECT a[2:4] AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": []map[string]interface{}{
- {"b": "hello1"},
- {"b": "hello2"},
- {"b": "hello3"},
- {"b": "hello4"},
- {"b": "hello5"},
- },
- },
- },
- result: []map[string]interface{}{{
- "ab": []map[string]interface{}{
- {"b": "hello3"},
- {"b": "hello4"},
- },
- }},
- },
- //17
- {
- sql: `SELECT a[2:] AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": []map[string]interface{}{
- {"b": "hello1"},
- {"b": "hello2"},
- {"b": "hello3"},
- {"b": "hello4"},
- {"b": "hello5"},
- },
- },
- },
- result: []map[string]interface{}{{
- "ab": []map[string]interface{}{
- {"b": "hello3"},
- {"b": "hello4"},
- {"b": "hello5"},
- },
- }},
- },
- //18
- {
- sql: `SELECT a[2:] AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": []interface{}{
- true, false, true, false, true, true,
- },
- },
- },
- result: []map[string]interface{}{{
- "ab": []interface{}{
- true, false, true, true,
- },
- }},
- },
- //19
- {
- sql: `SELECT a[:4] AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": []interface{}{
- true, false, true, false, true, true,
- },
- },
- },
- result: []map[string]interface{}{{
- "ab": []interface{}{
- true, false, true, false,
- },
- }},
- },
- //20
- {
- sql: `SELECT a[:4] AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": []interface{}{
- 3.14, 3.141, 3.1415, 3.14159, 3.141592, 3.1415926,
- },
- },
- },
- result: []map[string]interface{}{{
- "ab": []interface{}{
- 3.14, 3.141, 3.1415, 3.14159,
- },
- }},
- },
- //21
- {
- sql: `SELECT a->b[:4] AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": map[string]interface{}{
- "b": []float64{3.14, 3.141, 3.1415, 3.14159, 3.141592, 3.1415926},
- },
- },
- },
- result: []map[string]interface{}{{
- "ab": []float64{
- 3.14, 3.141, 3.1415, 3.14159,
- },
- }},
- },
- //22
- {
- sql: `SELECT a->b[0:1] AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": map[string]interface{}{
- "b": []float64{3.14, 3.141, 3.1415, 3.14159, 3.141592, 3.1415926},
- },
- },
- },
- result: []map[string]interface{}{{
- "ab": []float64{
- 3.14,
- },
- }},
- },
- //23
- {
- sql: `SELECT a->c->d AS f1 FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": map[string]interface{}{
- "b": "hello",
- "c": map[string]interface{}{
- "d": 35.2,
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "f1": 35.2,
- }},
- },
- //24
- {
- sql: `SELECT a->c->d AS f1 FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": map[string]interface{}{
- "b": "hello",
- "c": map[string]interface{}{
- "e": 35.2,
- },
- },
- },
- },
- result: []map[string]interface{}{{}},
- },
- //25
- {
- sql: `SELECT a->c->d AS f1 FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": map[string]interface{}{
- "b": "hello",
- },
- },
- },
- result: []map[string]interface{}{{}},
- },
- //26
- //The int type is not supported yet, the json parser returns float64 for int values
- {
- sql: `SELECT a->c->d AS f1 FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": map[string]interface{}{
- "b": "hello",
- "c": map[string]interface{}{
- "d": float64(35),
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "f1": float64(35),
- }},
- },
- //27
- {
- sql: "SELECT a FROM test",
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{},
- },
- result: []map[string]interface{}{
- {},
- },
- },
- //28
- {
- sql: "SELECT * FROM test",
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{},
- },
- result: []map[string]interface{}{
- {},
- },
- },
- //29
- {
- sql: `SELECT * FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": map[string]interface{}{
- "b": "hello",
- "c": map[string]interface{}{
- "d": 35.2,
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "a": map[string]interface{}{
- "b": "hello",
- "c": map[string]interface{}{
- "d": 35.2,
- },
- },
- }},
- },
- //30
- {
- sql: `SELECT * FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": "val1",
- "b": 3.14,
- },
- },
- result: []map[string]interface{}{{
- "a": "val1",
- "b": 3.14,
- }},
- },
- //31
- {
- sql: `SELECT 3*4 AS f1 FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{},
- },
- result: []map[string]interface{}{{
- "f1": int64(12),
- }},
- },
- //32
- {
- sql: `SELECT 4.5*2 AS f1 FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{},
- },
- result: []map[string]interface{}{{
- "f1": float64(9),
- }},
- },
- //33
- {
- sql: "SELECT `a.b.c` FROM test",
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a.b.c": "val_a",
- },
- },
- result: []map[string]interface{}{{
- "a.b.c": "val_a",
- }},
- },
- //34
- {
- sql: `SELECT CASE a WHEN 10 THEN "true" END AS b FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": int64(10),
- },
- },
- result: []map[string]interface{}{{
- "b": "true",
- }},
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- contextLogger := conf.Log.WithField("rule", "TestProjectPlan_Apply1")
- ctx := context.WithValue(context.Background(), context.LoggerKey, contextLogger)
- for i, tt := range tests {
- stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
- if err != nil {
- t.Errorf("parse sql error: %s", err)
- continue
- }
- pp := &ProjectOp{Fields: stmt.Fields, SendMeta: true}
- fv, afv := xsql.NewFunctionValuersForOp(nil)
- result := pp.Apply(ctx, tt.data, fv, afv)
- if !reflect.DeepEqual(tt.result, result) {
- t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
- }
- }
- }
- func TestProjectPlan_MultiInput(t *testing.T) {
- var tests = []struct {
- sql string
- data interface{}
- result []map[string]interface{}
- }{ //0
- {
- sql: "SELECT * FROM tbl WHERE abc*2+3 > 12 AND abc < 20",
- data: &xsql.Tuple{
- Emitter: "tbl",
- Message: xsql.Message{
- "abc": int64(6),
- },
- },
- result: []map[string]interface{}{{
- "abc": int64(6),
- }},
- },
- //1
- {
- sql: "SELECT abc FROM tbl WHERE abc*2+3 > 12 OR def = \"hello\"",
- data: &xsql.Tuple{
- Emitter: "tbl",
- Message: xsql.Message{
- "abc": int64(34),
- "def": "hello",
- },
- },
- result: []map[string]interface{}{{
- "abc": int64(34),
- }},
- },
- //2
- {
- sql: "SELECT id1 FROM src1 WHERE f1 = \"v1\" GROUP BY TUMBLINGWINDOW(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{
- {
- Emitter: "src1",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"id1": 1, "f1": "v1"},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"id1": 2, "f1": "v2"},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"id1": 3, "f1": "v1"},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id1": 1,
- }, {
- "id1": 2,
- }, {
- "id1": 3,
- }},
- },
- //3
- {
- sql: "SELECT id1 FROM src1 WHERE f1 = \"v1\" GROUP BY TUMBLINGWINDOW(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{
- {
- Emitter: "src1",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"id1": 1, "f1": "v1"},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"id2": 2, "f1": "v2"},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"id1": 3, "f1": "v1"},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id1": 1,
- }, {}, {
- "id1": 3,
- }},
- },
- //4
- {
- sql: "SELECT * FROM src1 WHERE f1 = \"v1\" GROUP BY TUMBLINGWINDOW(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{
- {
- Emitter: "src1",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"id1": 1, "f1": "v1"},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"id1": 2, "f1": "v2"},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"id1": 3, "f1": "v1"},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id1": 1,
- "f1": "v1",
- }, {
- "id1": 2,
- "f1": "v2",
- }, {
- "id1": 3,
- "f1": "v1",
- }},
- },
- //5
- {
- sql: "SELECT * FROM src1 WHERE f1 = \"v1\" GROUP BY TUMBLINGWINDOW(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{
- {
- Emitter: "src1",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"id1": 1, "f1": "v1"},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"id2": 2, "f2": "v2"},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"id1": 3, "f1": "v1"},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id1": 1,
- "f1": "v1",
- }, {
- "id2": 2,
- "f2": "v2",
- }, {
- "id1": 3,
- "f1": "v1",
- }},
- },
- //6
- {
- sql: "SELECT src1.* FROM src1 WHERE f1 = \"v1\" GROUP BY TUMBLINGWINDOW(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{
- {
- Emitter: "src1",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"id1": 1, "f1": "v1"},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"id1": 2, "f1": "v2"},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"id1": 3, "f1": "v1"},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id1": 1,
- "f1": "v1",
- }, {
- "id1": 2,
- "f1": "v2",
- }, {
- "id1": 3,
- "f1": "v1",
- }},
- },
- //7
- {
- sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2 WHERE src1.f1 = \"v1\" GROUP BY TUMBLINGWINDOW(ss, 10)",
- data: &xsql.JoinTupleSets{
- Content: []xsql.JoinTuple{
- {
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
- {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
- {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v1"}},
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id1": 1,
- }, {
- "id1": 2,
- }, {
- "id1": 3,
- }},
- },
- //8
- {
- sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2 WHERE src1.f1 = \"v1\" GROUP BY TUMBLINGWINDOW(ss, 10)",
- data: &xsql.JoinTupleSets{
- Content: []xsql.JoinTuple{
- {
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
- {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
- {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id2": 3, "f1": "v1"}},
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id1": 1,
- }, {
- "id1": 2,
- }, {}},
- },
- //9
- {
- sql: "SELECT abc FROM tbl group by abc",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.Tuple{
- Emitter: "tbl",
- Message: xsql.Message{
- "abc": int64(6),
- "def": "hello",
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "abc": int64(6),
- }},
- },
- //10
- {
- sql: "SELECT abc FROM tbl group by abc",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.Tuple{
- Emitter: "tbl",
- Message: xsql.Message{
- "def": "hello",
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{}},
- },
- //11
- {
- sql: "SELECT id1 FROM src1 GROUP BY TUMBLINGWINDOW(ss, 10), f1",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.Tuple{
- Emitter: "src1",
- Message: xsql.Message{"id1": 1, "f1": "v1"},
- },
- &xsql.Tuple{
- Emitter: "src1",
- Message: xsql.Message{"id1": 3, "f1": "v1"},
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.Tuple{
- Emitter: "src1",
- Message: xsql.Message{"id1": 2, "f1": "v2"},
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id1": 1,
- }, {
- "id1": 2,
- }},
- },
- //12
- {
- sql: "SELECT id1 FROM src1 GROUP BY TUMBLINGWINDOW(ss, 10), f1",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.Tuple{
- Emitter: "src1",
- Message: xsql.Message{"id1": 1, "f1": "v1"},
- },
- &xsql.Tuple{
- Emitter: "src1",
- Message: xsql.Message{"id1": 3, "f1": "v1"},
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.Tuple{
- Emitter: "src1",
- Message: xsql.Message{"id2": 2, "f1": "v2"},
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id1": 1,
- }, {}},
- },
- //13
- {
- sql: "SELECT src2.id2 FROM src1 left join src2 on src1.id1 = src2.id2 GROUP BY src2.f2, TUMBLINGWINDOW(ss, 10)",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
- {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
- {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v1"}},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id2": 2,
- }, {
- "id2": 4,
- }, {}},
- },
- //14
- {
- sql: "SELECT src1.*, f2 FROM src1 left join src2 GROUP BY TUMBLINGWINDOW(ss, 10)",
- data: &xsql.JoinTupleSets{
- Content: []xsql.JoinTuple{
- {
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
- {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
- {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v1"}},
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id1": 1,
- "f1": "v1",
- "f2": "w2",
- }, {
- "id1": 2,
- "f1": "v2",
- "f2": "w3",
- }, {
- "id1": 3,
- "f1": "v1",
- }},
- },
- //15
- {
- sql: "SELECT * FROM src1 left join src2 GROUP BY TUMBLINGWINDOW(ss, 10)",
- data: &xsql.JoinTupleSets{
- Content: []xsql.JoinTuple{
- {
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id": 1, "f1": "v1"}},
- {Emitter: "src2", Message: xsql.Message{"id": 2, "f2": "w2"}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id": 2, "f1": "v2"}},
- {Emitter: "src2", Message: xsql.Message{"id": 4, "f2": "w3"}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id": 3, "f1": "v1"}},
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id": 1,
- "f1": "v1",
- "f2": "w2",
- }, {
- "id": 2,
- "f1": "v2",
- "f2": "w3",
- }, {
- "id": 3,
- "f1": "v1",
- }},
- },
- //16
- {
- sql: "SELECT src1.* FROM src1 GROUP BY TUMBLINGWINDOW(ss, 10), f1",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.Tuple{
- Emitter: "src1",
- Message: xsql.Message{"id1": 1, "f1": "v1"},
- },
- &xsql.Tuple{
- Emitter: "src1",
- Message: xsql.Message{"id1": 3, "f1": "v1"},
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.Tuple{
- Emitter: "src1",
- Message: xsql.Message{"id1": 2, "f1": "v2"},
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id1": 1,
- "f1": "v1",
- }, {
- "id1": 2,
- "f1": "v2",
- }},
- },
- //17
- {
- sql: "SELECT src2.id2, src1.* FROM src1 left join src2 on src1.id1 = src2.id2 GROUP BY src2.f2, TUMBLINGWINDOW(ss, 10)",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
- {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
- {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v1"}},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id2": 2,
- "id1": 1,
- "f1": "v1",
- }, {
- "id2": 4,
- "id1": 2,
- "f1": "v2",
- }, {
- "id1": 3,
- "f1": "v1",
- }},
- },
- //18
- {
- sql: "SELECT src2.id2, src1.* FROM src1 left join src2 on src1.id1 = src2.id2 GROUP BY src2.f2, TUMBLINGWINDOW(ss, 10)",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
- {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
- {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v1"}},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "id2": 2,
- "id1": 1,
- "f1": "v1",
- }, {
- "id2": 4,
- "id1": 2,
- "f1": "v2",
- }, {
- "id1": 3,
- "f1": "v1",
- }},
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- contextLogger := conf.Log.WithField("rule", "TestProjectPlan_MultiInput")
- ctx := context.WithValue(context.Background(), context.LoggerKey, contextLogger)
- for i, tt := range tests {
- stmt, _ := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
- pp := &ProjectOp{Fields: stmt.Fields}
- fv, afv := xsql.NewFunctionValuersForOp(nil)
- result := pp.Apply(ctx, tt.data, fv, afv)
- if !reflect.DeepEqual(tt.result, result) {
- t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
- }
- }
- }
- func TestProjectPlan_Funcs(t *testing.T) {
- var tests = []struct {
- sql string
- data interface{}
- result []map[string]interface{}
- }{
- //0
- {
- sql: "SELECT round(a) as r FROM test",
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": 47.5,
- },
- },
- result: []map[string]interface{}{{
- "r": float64(48),
- }},
- },
- //1
- {
- sql: "SELECT round(a) as r FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{
- {
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53.1},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 27.4},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 123123.7},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "r": float64(53),
- }, {
- "r": float64(27),
- }, {
- "r": float64(123124),
- }},
- },
- //2
- {
- sql: "SELECT round(a) as r FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{
- {
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53.1},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 27.4},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 123123.7},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "r": float64(53),
- }, {
- "r": float64(27),
- }, {
- "r": float64(123124),
- }},
- },
- //3
- {
- sql: "SELECT round(a) as r FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10)",
- data: &xsql.JoinTupleSets{
- Content: []xsql.JoinTuple{
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 65.55}},
- {Emitter: "test1", Message: xsql.Message{"id": 1, "b": 12}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 73.499}},
- {Emitter: "test1", Message: xsql.Message{"id": 2, "b": 34}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 3, "a": 88.88}},
- {Emitter: "test1", Message: xsql.Message{"id": 3, "b": 6}},
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "r": float64(66),
- }, {
- "r": float64(73),
- }, {
- "r": float64(89),
- }},
- },
- //4
- {
- sql: "SELECT CONCAT(test.id, test.a, test1.b) as concat FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10)",
- data: &xsql.JoinTupleSets{
- Content: []xsql.JoinTuple{
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 65.55}},
- {Emitter: "test1", Message: xsql.Message{"id": 1, "b": 12}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 73.499}},
- {Emitter: "test1", Message: xsql.Message{"id": 2, "b": 34}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 3, "a": 88.88}},
- {Emitter: "test1", Message: xsql.Message{"id": 3, "b": 6}},
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "concat": "165.5512",
- }, {
- "concat": "273.49934",
- }, {
- "concat": "388.886",
- }},
- },
- //5
- {
- sql: "SELECT count(a) as r FROM test",
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": 47.5,
- },
- },
- result: []map[string]interface{}{{
- "r": 1,
- }},
- },
- //6
- {
- sql: "SELECT meta(test.device) as d FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10)",
- data: &xsql.JoinTupleSets{
- Content: []xsql.JoinTuple{
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 65.55}, Metadata: xsql.Metadata{"device": "devicea"}},
- {Emitter: "test1", Message: xsql.Message{"id": 1, "b": 12}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 73.499}, Metadata: xsql.Metadata{"device": "deviceb"}},
- {Emitter: "test1", Message: xsql.Message{"id": 2, "b": 34}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 3, "a": 88.88}, Metadata: xsql.Metadata{"device": "devicec"}},
- {Emitter: "test1", Message: xsql.Message{"id": 3, "b": 6}},
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "d": "devicea",
- }, {
- "d": "deviceb",
- }, {
- "d": "devicec",
- }},
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- contextLogger := conf.Log.WithField("rule", "TestProjectPlan_Funcs")
- ctx := context.WithValue(context.Background(), context.LoggerKey, contextLogger)
- for i, tt := range tests {
- stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
- if err != nil {
- t.Error(err)
- }
- pp := &ProjectOp{Fields: stmt.Fields, IsAggregate: xsql.IsAggStatement(stmt)}
- fv, afv := xsql.NewFunctionValuersForOp(nil)
- result := pp.Apply(ctx, tt.data, fv, afv)
- if !reflect.DeepEqual(tt.result, result) {
- t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
- }
- }
- }
- func TestProjectPlan_AggFuncs(t *testing.T) {
- var tests = []struct {
- sql string
- data interface{}
- result []map[string]interface{}
- }{
- //0
- {
- sql: "SELECT count(*) as c, round(a) as r, window_start() as ws, window_end() as we FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10), test1.color",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 122.33, "c": 2, "r": 122}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 5, "a": 177.51}},
- {Emitter: "src2", Message: xsql.Message{"id": 5, "color": "w2"}},
- },
- },
- },
- WindowRange: &xsql.WindowRange{
- WindowStart: 1541152486013,
- WindowEnd: 1541152487013,
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 89.03, "c": 2, "r": 89}},
- {Emitter: "src2", Message: xsql.Message{"id": 2, "color": "w1"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 4, "a": 14.6}},
- {Emitter: "src2", Message: xsql.Message{"id": 4, "color": "w1"}},
- },
- },
- },
- WindowRange: &xsql.WindowRange{
- WindowStart: 1541152486013,
- WindowEnd: 1541152487013,
- },
- },
- },
- result: []map[string]interface{}{{
- "c": 2,
- "r": float64(122),
- "ws": int64(1541152486013),
- "we": int64(1541152487013),
- }, {
- "c": 2,
- "r": float64(89),
- "ws": int64(1541152486013),
- "we": int64(1541152487013),
- }},
- },
- //1
- {
- sql: "SELECT count(a) as c, avg(a) as a, sum(a) as s, min(a) as min, max(a) as max FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10), test1.color",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 122.33, "c": 2, "r": 122}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 5}},
- {Emitter: "src2", Message: xsql.Message{"id": 5, "color": "w2"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 89.03, "c": 2, "r": 89}},
- {Emitter: "src2", Message: xsql.Message{"id": 2, "color": "w1"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 4, "a": 14.6}},
- {Emitter: "src2", Message: xsql.Message{"id": 4, "color": "w1"}},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "c": 1,
- "a": 122.33,
- "s": 122.33,
- "min": 122.33,
- "max": 122.33,
- }, {
- "c": 2,
- "s": 103.63,
- "a": 51.815,
- "min": 14.6,
- "max": 89.03,
- }},
- },
- //2
- {
- sql: "SELECT avg(a) FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10), test1.color",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 122.33}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 68.54}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 4, "a": 98.31}},
- {Emitter: "src2", Message: xsql.Message{"id": 4, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 5, "a": 177.54}},
- {Emitter: "src2", Message: xsql.Message{"id": 5, "color": "w2"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 89.03}},
- {Emitter: "src2", Message: xsql.Message{"id": 2, "color": "w1"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 4, "a": 14.6}},
- {Emitter: "src2", Message: xsql.Message{"id": 4, "color": "w1"}},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "avg": 116.68,
- }, {
- "avg": 51.815,
- }},
- },
- //3
- {
- sql: "SELECT max(a) FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10), test1.color",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 122.33}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 68.55}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 5, "a": 177.51}},
- {Emitter: "src2", Message: xsql.Message{"id": 5, "color": "w2"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 89.03}},
- {Emitter: "src2", Message: xsql.Message{"id": 2, "color": "w1"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 4, "a": 14.6}},
- {Emitter: "src2", Message: xsql.Message{"id": 4, "color": "w1"}},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "max": 177.51,
- }, {
- "max": 89.03,
- }},
- },
- //4
- {
- sql: "SELECT min(a), window_start(), window_end() FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10)",
- data: &xsql.JoinTupleSets{
- Content: []xsql.JoinTuple{
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 122.33}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 68.55}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 5, "a": 177.51}},
- {Emitter: "src2", Message: xsql.Message{"id": 5, "color": "w2"}},
- },
- },
- },
- WindowRange: &xsql.WindowRange{
- WindowStart: 1541152486013,
- WindowEnd: 1541152487013,
- },
- },
- result: []map[string]interface{}{{
- "min": 68.55,
- "window_start": int64(1541152486013),
- "window_end": int64(1541152487013),
- }},
- },
- //5
- {
- sql: "SELECT count(*) as all, count(a) as c, avg(a) as a, sum(a) as s, min(a) as min, max(a) as max FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10)",
- data: &xsql.JoinTupleSets{
- Content: []xsql.JoinTuple{
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 68.55}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- {
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 5, "a": 177.51}},
- {Emitter: "src2", Message: xsql.Message{"id": 5, "color": "w2"}},
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "all": 3,
- "c": 2,
- "a": 123.03,
- "s": 246.06,
- "min": 68.55,
- "max": 177.51,
- }},
- },
- //6
- {
- sql: "SELECT sum(a), window_start() as ws, window_end() FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{{
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 27},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 123123},
- },
- },
- },
- },
- WindowRange: &xsql.WindowRange{
- WindowStart: 1541152486013,
- WindowEnd: 1541152487013,
- },
- },
- result: []map[string]interface{}{{
- "sum": 123203,
- "ws": int64(1541152486013),
- "window_end": int64(1541152487013),
- }},
- },
- //7
- {
- sql: "SELECT sum(a) as s FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{{
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53, "s": 123203},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 27},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 123123},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "s": 123203,
- }},
- },
- //8
- {
- sql: "SELECT sum(a) FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{{
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 27},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 123123},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "sum": 123203,
- }},
- },
- //9
- {
- sql: "SELECT count(*) as all, count(a) as c, avg(a) as a, sum(a) as s, min(a) as min, max(a) as max FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{{
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 27},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"s": 123123},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "all": 3,
- "c": 2,
- "a": 40,
- "s": 80,
- "min": 27,
- "max": 53,
- }},
- },
- //10
- {
- sql: "SELECT count(*), meta(test1.device) FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10), test1.color",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 122.33}},
- {Emitter: "test1", Message: xsql.Message{"id": 1, "color": "w2"}, Metadata: xsql.Metadata{"device": "devicea"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 5, "a": 177.51}},
- {Emitter: "test1", Message: xsql.Message{"id": 5, "color": "w2"}, Metadata: xsql.Metadata{"device": "deviceb"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 89.03}},
- {Emitter: "test1", Message: xsql.Message{"id": 2, "color": "w1"}, Metadata: xsql.Metadata{"device": "devicec"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 4, "a": 14.6}},
- {Emitter: "test1", Message: xsql.Message{"id": 4, "color": "w1"}, Metadata: xsql.Metadata{"device": "deviced"}},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "count": 2,
- "meta": "devicea",
- }, {
- "count": 2,
- "meta": "devicec",
- }},
- },
- //11
- {
- sql: "SELECT count(*) as c, meta(test1.device) as d FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10), test1.color",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 122.33, "c": 2, "d": "devicea"}},
- {Emitter: "test1", Message: xsql.Message{"id": 1, "color": "w2"}, Metadata: xsql.Metadata{"device": "devicea"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 5, "a": 177.51}},
- {Emitter: "test1", Message: xsql.Message{"id": 5, "color": "w2"}, Metadata: xsql.Metadata{"device": "deviceb"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 89.03, "c": 2, "d": "devicec"}},
- {Emitter: "test1", Message: xsql.Message{"id": 2, "color": "w1"}, Metadata: xsql.Metadata{"device": "devicec"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 4, "a": 14.6}},
- {Emitter: "test1", Message: xsql.Message{"id": 4, "color": "w1"}, Metadata: xsql.Metadata{"device": "deviced"}},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "c": 2,
- "d": "devicea",
- }, {
- "c": 2,
- "d": "devicec",
- }},
- },
- //12
- {
- sql: "SELECT * FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10), test1.color",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 122.33, "c": 2, "r": 122}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 5, "a": 177.51}},
- {Emitter: "src2", Message: xsql.Message{"id": 5, "color": "w2"}},
- },
- },
- },
- WindowRange: &xsql.WindowRange{
- WindowStart: 1541152486013,
- WindowEnd: 1541152487013,
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 89.03, "c": 2, "r": 89}},
- {Emitter: "src2", Message: xsql.Message{"id": 2, "color": "w1"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 4, "a": 14.6}},
- {Emitter: "src2", Message: xsql.Message{"id": 4, "color": "w1"}},
- },
- },
- },
- WindowRange: &xsql.WindowRange{
- WindowStart: 1541152486013,
- WindowEnd: 1541152487013,
- },
- },
- },
- result: []map[string]interface{}{{
- "a": 122.33,
- "c": 2,
- "color": "w2",
- "id": 1,
- "r": 122,
- }, {
- "a": 89.03,
- "c": 2,
- "color": "w1",
- "id": 2,
- "r": 89,
- }},
- },
- //13
- {
- sql: "SELECT collect(a) as r1 FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10), test1.color",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 122.33, "c": 2, "r": 122}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 5, "a": 177.51}},
- {Emitter: "src2", Message: xsql.Message{"id": 5, "color": "w2"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 89.03, "c": 2, "r": 89}},
- {Emitter: "src2", Message: xsql.Message{"id": 2, "color": "w1"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 4, "a": 14.6}},
- {Emitter: "src2", Message: xsql.Message{"id": 4, "color": "w1"}},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "r1": []interface{}{122.33, 177.51},
- }, {"r1": []interface{}{89.03, 14.6}}},
- },
- //14
- {
- sql: "SELECT collect(*)[1] as c1 FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{
- {
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53, "s": 123203},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 27},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 123123},
- },
- },
- },
- },
- WindowRange: &xsql.WindowRange{
- WindowStart: 1541152486013,
- WindowEnd: 1541152487013,
- },
- },
- result: []map[string]interface{}{{
- "c1": xsql.Message{
- "a": 27,
- },
- }},
- },
- //15
- {
- sql: "SELECT collect(*)[1]->a as c1 FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{{
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53, "s": 123203},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 27},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 123123},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "c1": 27,
- }},
- },
- //16
- {
- sql: "SELECT collect(*)[1]->sl[0] as c1 FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{{
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53, "sl": []string{"hello", "world"}},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 27, "sl": []string{"new", "horizon"}},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 123123, "sl": []string{"south", "africa"}},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "c1": "new",
- }},
- },
- //17
- {
- sql: "SELECT deduplicate(id, true) as r1 FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10), test1.color",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 122.33, "c": 2, "r": 122}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 5, "a": 177.51}},
- {Emitter: "src2", Message: xsql.Message{"id": 5, "color": "w2"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 89.03, "c": 2, "r": 89}},
- {Emitter: "src2", Message: xsql.Message{"id": 2, "color": "w1"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 4, "a": 14.6}},
- {Emitter: "src2", Message: xsql.Message{"id": 4, "color": "w1"}},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{
- {
- "r1": []interface{}{
- xsql.Message{"a": 122.33, "c": 2, "color": "w2", "id": 1, "r": 122},
- xsql.Message{"a": 177.51, "color": "w2", "id": 5}},
- }, {
- "r1": []interface{}{
- xsql.Message{"a": 89.03, "c": 2, "color": "w1", "id": 2, "r": 89},
- xsql.Message{"a": 14.6, "color": "w1", "id": 4}},
- },
- },
- },
- //18
- {
- sql: "SELECT deduplicate(a, false)->a as c1 FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{{
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53, "s": 123203},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 27},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 123123},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "c1": 123123,
- }},
- },
- //19
- {
- sql: "SELECT deduplicate(a, false) as c1 FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{{
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53, "s": 123203},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 27},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 53},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{}},
- },
- //20
- {
- sql: "SELECT deduplicate(a, false) as c1 FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{{
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53, "s": 123203},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 27},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 53},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{}},
- },
- //21 when got column after group by operation, return the first tuple's column
- {
- sql: "SELECT A.module, A.topic , max(A.value), B.topic as var2, max(B.value) as max2, C.topic as var3, max(C.value) as max3 FROM A FULL JOIN B on A.module=B.module FULL JOIN C on A.module=C.module GROUP BY A.module, TUMBLINGWINDOW(ss, 10)",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "B", Message: xsql.Message{"module": 1, "topic": "moduleB topic", "value": 1}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "C", Message: xsql.Message{"module": 1, "topic": "moduleC topic", "value": 100}},
- },
- },
- },
- },
- },
- result: []map[string]interface{}{{
- "var2": "moduleB topic",
- "max2": 1,
- "max3": 100,
- }},
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- contextLogger := conf.Log.WithField("rule", "TestProjectPlan_AggFuncs")
- ctx := context.WithValue(context.Background(), context.LoggerKey, contextLogger)
- for i, tt := range tests {
- stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
- if err != nil {
- t.Error(err)
- }
- pp := &ProjectOp{Fields: stmt.Fields, IsAggregate: true}
- fv, afv := xsql.NewFunctionValuersForOp(nil)
- result := pp.Apply(ctx, tt.data, fv, afv)
- if !reflect.DeepEqual(tt.result, result) {
- t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
- }
- }
- }
- func TestProjectPlanError(t *testing.T) {
- var tests = []struct {
- sql string
- data interface{}
- result interface{}
- }{
- //0
- {
- sql: "SELECT a FROM test",
- data: errors.New("an error from upstream"),
- result: errors.New("an error from upstream"),
- },
- //1
- {
- sql: "SELECT a * 5 FROM test",
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": "val_a",
- },
- },
- result: errors.New("run Select error: invalid operation string(val_a) * int64(5)"),
- },
- //2
- {
- sql: `SELECT a[0]->b AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": "common string",
- },
- },
- result: errors.New("run Select error: invalid operation string(common string) [] *xsql.BracketEvalResult(&{0 0})"),
- },
- //3
- {
- sql: `SELECT round(a) as r FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": "common string",
- },
- },
- result: errors.New("run Select error: call func round error: only float64 & int type are supported"),
- },
- //4
- {
- sql: `SELECT round(a) as r FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "abc": "common string",
- },
- },
- result: errors.New("run Select error: call func round error: only float64 & int type are supported"),
- },
- //5
- {
- sql: "SELECT avg(a) as avg FROM test Inner Join test1 on test.id = test1.id GROUP BY TumblingWindow(ss, 10), test1.color",
- data: xsql.GroupedTuplesSet{
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 122.33}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 1, "a": 68.54}},
- {Emitter: "src2", Message: xsql.Message{"id": 1, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 4, "a": "dde"}},
- {Emitter: "src2", Message: xsql.Message{"id": 4, "color": "w2"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 5, "a": 177.54}},
- {Emitter: "src2", Message: xsql.Message{"id": 5, "color": "w2"}},
- },
- },
- },
- },
- {
- Content: []xsql.DataValuer{
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 2, "a": 89.03}},
- {Emitter: "src2", Message: xsql.Message{"id": 2, "color": "w1"}},
- },
- },
- &xsql.JoinTuple{
- Tuples: []xsql.Tuple{
- {Emitter: "test", Message: xsql.Message{"id": 4, "a": 14.6}},
- {Emitter: "src2", Message: xsql.Message{"id": 4, "color": "w1"}},
- },
- },
- },
- },
- },
- result: errors.New("run Select error: call func avg error: requires float64 but found string(dde)"),
- },
- //6
- {
- sql: "SELECT sum(a) as sum FROM test GROUP BY TumblingWindow(ss, 10)",
- data: xsql.WindowTuplesSet{
- Content: []xsql.WindowTuples{{
- Emitter: "test",
- Tuples: []xsql.Tuple{
- {
- Emitter: "src1",
- Message: xsql.Message{"a": 53},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": "ddd"},
- }, {
- Emitter: "src1",
- Message: xsql.Message{"a": 123123},
- },
- },
- },
- },
- },
- result: errors.New("run Select error: call func sum error: requires int but found string(ddd)"),
- },
- //7
- {
- sql: `SELECT a[0]->b AS ab FROM test`,
- data: &xsql.Tuple{
- Emitter: "test",
- Message: xsql.Message{
- "a": []map[string]interface{}(nil),
- },
- },
- result: errors.New("run Select error: out of index: 0 of 0"),
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- contextLogger := conf.Log.WithField("rule", "TestProjectPlanError")
- ctx := context.WithValue(context.Background(), context.LoggerKey, contextLogger)
- for i, tt := range tests {
- stmt, _ := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
- pp := &ProjectOp{Fields: stmt.Fields, IsAggregate: xsql.IsAggStatement(stmt)}
- fv, afv := xsql.NewFunctionValuersForOp(nil)
- result := pp.Apply(ctx, tt.data, fv, afv)
- if !reflect.DeepEqual(tt.result, result) {
- t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
- }
- }
- }
|