1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463 |
- package processors
- import (
- "encoding/json"
- "fmt"
- "github.com/emqx/kuiper/common"
- "github.com/emqx/kuiper/xsql"
- "github.com/emqx/kuiper/xstream"
- "github.com/emqx/kuiper/xstream/api"
- "github.com/emqx/kuiper/xstream/nodes"
- "github.com/emqx/kuiper/xstream/test"
- "os"
- "path"
- "reflect"
- "strings"
- "testing"
- "time"
- )
- var DbDir = getDbDir()
- func getDbDir() string {
- common.InitConf()
- dbDir, err := common.GetDataLoc()
- if err != nil {
- log.Panic(err)
- }
- log.Infof("db location is %s", dbDir)
- return dbDir
- }
- func cleanStateData() {
- dbDir, err := common.GetDataLoc()
- if err != nil {
- log.Panic(err)
- }
- c := path.Join(dbDir, "checkpoints")
- err = os.RemoveAll(c)
- if err != nil {
- log.Errorf("%s", err)
- }
- s := path.Join(dbDir, "sink", "cache")
- err = os.RemoveAll(s)
- if err != nil {
- log.Errorf("%s", err)
- }
- }
- func TestStreamCreateProcessor(t *testing.T) {
- var tests = []struct {
- s string
- r []string
- err string
- }{
- {
- s: `SHOW STREAMS;`,
- r: []string{"No stream definitions are found."},
- },
- {
- s: `EXPLAIN STREAM topic1;`,
- err: "Stream topic1 is not found.",
- },
- {
- s: `CREATE STREAM topic1 (
- USERID BIGINT,
- FIRST_NAME STRING,
- LAST_NAME STRING,
- NICKNAMES ARRAY(STRING),
- Gender BOOLEAN,
- ADDRESS STRUCT(STREET_NAME STRING, NUMBER BIGINT),
- ) WITH (DATASOURCE="users", FORMAT="AVRO", KEY="USERID");`,
- r: []string{"Stream topic1 is created."},
- },
- {
- s: `CREATE STREAM ` + "`stream`" + ` (
- USERID BIGINT,
- FIRST_NAME STRING,
- LAST_NAME STRING,
- NICKNAMES ARRAY(STRING),
- Gender BOOLEAN,
- ` + "`地址`" + ` STRUCT(STREET_NAME STRING, NUMBER BIGINT),
- ) WITH (DATASOURCE="users", FORMAT="AVRO", KEY="USERID");`,
- r: []string{"Stream stream is created."},
- },
- {
- s: `CREATE STREAM topic1 (
- USERID BIGINT,
- ) WITH (DATASOURCE="users", FORMAT="AVRO", KEY="USERID");`,
- err: "Create stream fails: Item topic1 already exists.",
- },
- {
- s: `EXPLAIN STREAM topic1;`,
- r: []string{"TO BE SUPPORTED"},
- },
- {
- s: `DESCRIBE STREAM topic1;`,
- r: []string{"Fields\n--------------------------------------------------------------------------------\nUSERID\tbigint\nFIRST_NAME\tstring\nLAST_NAME\tstring\nNICKNAMES\t" +
- "array(string)\nGender\tboolean\nADDRESS\tstruct(STREET_NAME string, NUMBER bigint)\n\n" +
- "DATASOURCE: users\nFORMAT: AVRO\nKEY: USERID\n"},
- },
- {
- s: `DROP STREAM topic1;`,
- r: []string{"Stream topic1 is dropped."},
- },
- {
- s: `SHOW STREAMS;`,
- r: []string{"stream"},
- },
- {
- s: `DESCRIBE STREAM topic1;`,
- err: "Stream topic1 is not found.",
- },
- {
- s: `DROP STREAM topic1;`,
- err: "Drop stream fails: topic1 is not found.",
- },
- {
- s: "DROP STREAM `stream`;",
- r: []string{"Stream stream is dropped."},
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- streamDB := path.Join(getDbDir(), "streamTest")
- for i, tt := range tests {
- results, err := NewStreamProcessor(streamDB).ExecStmt(tt.s)
- if !reflect.DeepEqual(tt.err, errstring(err)) {
- t.Errorf("%d. %q: error mismatch:\n exp=%s\n got=%s\n\n", i, tt.s, tt.err, err)
- } else if tt.err == "" {
- if !reflect.DeepEqual(tt.r, results) {
- t.Errorf("%d. %q\n\nstmt mismatch:\nexp=%s\ngot=%#v\n\n", i, tt.s, tt.r, results)
- }
- }
- }
- }
- func createStreams(t *testing.T) {
- p := NewStreamProcessor(path.Join(DbDir, "stream"))
- demo := `CREATE STREAM demo (
- color STRING,
- size BIGINT,
- ts BIGINT
- ) WITH (DATASOURCE="demo", FORMAT="json", KEY="ts");`
- _, err := p.ExecStmt(demo)
- if err != nil {
- t.Log(err)
- }
- demoE := `CREATE STREAM demoE (
- color STRING,
- size BIGINT,
- ts BIGINT
- ) WITH (DATASOURCE="demoE", FORMAT="json", KEY="ts");`
- _, err = p.ExecStmt(demoE)
- if err != nil {
- t.Log(err)
- }
- demo1 := `CREATE STREAM demo1 (
- temp FLOAT,
- hum BIGINT,` +
- "`from`" + ` STRING,
- ts BIGINT
- ) WITH (DATASOURCE="demo1", FORMAT="json", KEY="ts");`
- _, err = p.ExecStmt(demo1)
- if err != nil {
- t.Log(err)
- }
- sessionDemo := `CREATE STREAM sessionDemo (
- temp FLOAT,
- hum BIGINT,
- ts BIGINT
- ) WITH (DATASOURCE="sessionDemo", FORMAT="json", KEY="ts");`
- _, err = p.ExecStmt(sessionDemo)
- if err != nil {
- t.Log(err)
- }
- }
- func dropStreams(t *testing.T) {
- p := NewStreamProcessor(path.Join(DbDir, "stream"))
- demo := `DROP STREAM demo`
- _, err := p.ExecStmt(demo)
- if err != nil {
- t.Log(err)
- }
- demoE := `DROP STREAM demoE`
- _, err = p.ExecStmt(demoE)
- if err != nil {
- t.Log(err)
- }
- demo1 := `DROP STREAM demo1`
- _, err = p.ExecStmt(demo1)
- if err != nil {
- t.Log(err)
- }
- sessionDemo := `DROP STREAM sessionDemo`
- _, err = p.ExecStmt(sessionDemo)
- if err != nil {
- t.Log(err)
- }
- }
- func createSchemalessStreams(t *testing.T) {
- p := NewStreamProcessor(path.Join(DbDir, "stream"))
- demo := `CREATE STREAM ldemo (
- ) WITH (DATASOURCE="ldemo", FORMAT="json");`
- _, err := p.ExecStmt(demo)
- if err != nil {
- t.Log(err)
- }
- demo1 := `CREATE STREAM ldemo1 (
- ) WITH (DATASOURCE="ldemo1", FORMAT="json");`
- _, err = p.ExecStmt(demo1)
- if err != nil {
- t.Log(err)
- }
- sessionDemo := `CREATE STREAM lsessionDemo (
- ) WITH (DATASOURCE="lsessionDemo", FORMAT="json");`
- _, err = p.ExecStmt(sessionDemo)
- if err != nil {
- t.Log(err)
- }
- }
- func dropSchemalessStreams(t *testing.T) {
- p := NewStreamProcessor(path.Join(DbDir, "stream"))
- demo := `DROP STREAM ldemo`
- _, err := p.ExecStmt(demo)
- if err != nil {
- t.Log(err)
- }
- demo1 := `DROP STREAM ldemo1`
- _, err = p.ExecStmt(demo1)
- if err != nil {
- t.Log(err)
- }
- sessionDemo := `DROP STREAM lsessionDemo`
- _, err = p.ExecStmt(sessionDemo)
- if err != nil {
- t.Log(err)
- }
- }
- func getMockSource(name string, done <-chan int, size int) *nodes.SourceNode {
- var data []*xsql.Tuple
- switch name {
- case "demo":
- data = []*xsql.Tuple{
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "red",
- "size": 3,
- "ts": 1541152486013,
- },
- Timestamp: 1541152486013,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "blue",
- "size": 6,
- "ts": 1541152486822,
- },
- Timestamp: 1541152486822,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "blue",
- "size": 2,
- "ts": 1541152487632,
- },
- Timestamp: 1541152487632,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "yellow",
- "size": 4,
- "ts": 1541152488442,
- },
- Timestamp: 1541152488442,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "red",
- "size": 1,
- "ts": 1541152489252,
- },
- Timestamp: 1541152489252,
- },
- }
- case "demoE":
- data = []*xsql.Tuple{
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": 3,
- "size": "red",
- "ts": 1541152486013,
- },
- Timestamp: 1541152486013,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "blue",
- "size": 6,
- "ts": "1541152486822",
- },
- Timestamp: 1541152486822,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "blue",
- "size": 2,
- "ts": 1541152487632,
- },
- Timestamp: 1541152487632,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": 7,
- "size": 4,
- "ts": 1541152488442,
- },
- Timestamp: 1541152488442,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "red",
- "size": "blue",
- "ts": 1541152489252,
- },
- Timestamp: 1541152489252,
- },
- }
- case "demo1":
- data = []*xsql.Tuple{
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 65,
- "from": "device1",
- "ts": 1541152486013,
- },
- Timestamp: 1541152486013,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 27.5,
- "hum": 59,
- "from": "device2",
- "ts": 1541152486823,
- },
- Timestamp: 1541152486823,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 28.1,
- "hum": 75,
- "from": "device3",
- "ts": 1541152487632,
- },
- Timestamp: 1541152487632,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 27.4,
- "hum": 80,
- "from": "device1",
- "ts": 1541152488442,
- },
- Timestamp: 1541152488442,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 62,
- "from": "device3",
- "ts": 1541152489252,
- },
- Timestamp: 1541152489252,
- },
- }
- case "sessionDemo":
- data = []*xsql.Tuple{
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 65,
- "ts": 1541152486013,
- },
- Timestamp: 1541152486013,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 27.5,
- "hum": 59,
- "ts": 1541152486823,
- },
- Timestamp: 1541152486823,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 28.1,
- "hum": 75,
- "ts": 1541152487932,
- },
- Timestamp: 1541152487932,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 27.4,
- "hum": 80,
- "ts": 1541152488442,
- },
- Timestamp: 1541152488442,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 62,
- "ts": 1541152489252,
- },
- Timestamp: 1541152489252,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 26.2,
- "hum": 63,
- "ts": 1541152490062,
- },
- Timestamp: 1541152490062,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 26.8,
- "hum": 71,
- "ts": 1541152490872,
- },
- Timestamp: 1541152490872,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 28.9,
- "hum": 85,
- "ts": 1541152491682,
- },
- Timestamp: 1541152491682,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 29.1,
- "hum": 92,
- "ts": 1541152492492,
- },
- Timestamp: 1541152492492,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 32.2,
- "hum": 99,
- "ts": 1541152493202,
- },
- Timestamp: 1541152493202,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 30.9,
- "hum": 87,
- "ts": 1541152494112,
- },
- Timestamp: 1541152494112,
- },
- }
- }
- return nodes.NewSourceNodeWithSource(name, test.NewMockSource(data[:size], done, false), map[string]string{
- "DATASOURCE": name,
- })
- }
- func TestSingleSQL(t *testing.T) {
- var tests = []struct {
- name string
- sql string
- r [][]map[string]interface{}
- s string
- m map[string]interface{}
- }{
- {
- name: `rule1`,
- sql: `SELECT * FROM demo`,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- "size": float64(3),
- "ts": float64(1541152486013),
- }},
- {{
- "color": "blue",
- "size": float64(6),
- "ts": float64(1541152486822),
- }},
- {{
- "color": "blue",
- "size": float64(2),
- "ts": float64(1541152487632),
- }},
- {{
- "color": "yellow",
- "size": float64(4),
- "ts": float64(1541152488442),
- }},
- {{
- "color": "red",
- "size": float64(1),
- "ts": float64(1541152489252),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(5),
- "op_project_0_records_out_total": int64(5),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(5),
- "sink_mockSink_0_records_out_total": int64(5),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- },
- s: "sink_mockSink_0_records_out_total",
- }, {
- name: `rule2`,
- sql: `SELECT color, ts FROM demo where size > 3`,
- r: [][]map[string]interface{}{
- {{
- "color": "blue",
- "ts": float64(1541152486822),
- }},
- {{
- "color": "yellow",
- "ts": float64(1541152488442),
- }},
- },
- s: "op_filter_0_records_in_total",
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(2),
- "op_project_0_records_out_total": int64(2),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(2),
- "sink_mockSink_0_records_out_total": int64(2),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- "op_filter_0_exceptions_total": int64(0),
- "op_filter_0_process_latency_ms": int64(0),
- "op_filter_0_records_in_total": int64(5),
- "op_filter_0_records_out_total": int64(2),
- },
- }, {
- name: `rule3`,
- sql: `SELECT size as Int8, ts FROM demo where size > 3`,
- r: [][]map[string]interface{}{
- {{
- "Int8": float64(6),
- "ts": float64(1541152486822),
- }},
- {{
- "Int8": float64(4),
- "ts": float64(1541152488442),
- }},
- },
- s: "op_filter_0_records_in_total",
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(2),
- "op_project_0_records_out_total": int64(2),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(2),
- "sink_mockSink_0_records_out_total": int64(2),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- "op_filter_0_exceptions_total": int64(0),
- "op_filter_0_process_latency_ms": int64(0),
- "op_filter_0_records_in_total": int64(5),
- "op_filter_0_records_out_total": int64(2),
- },
- }, {
- name: `rule4`,
- sql: `SELECT size as Int8, ts FROM demoE where size > 3`,
- r: [][]map[string]interface{}{
- {{
- "error": "error in preprocessor: invalid data type for color, expect string but found int(3)",
- }},
- {{
- "Int8": float64(6),
- "ts": float64(1541152486822),
- }},
- {{
- "error": "error in preprocessor: invalid data type for color, expect string but found int(7)",
- }},
- {{
- "error": "error in preprocessor: invalid data type for size, expect bigint but found string(blue)",
- }},
- },
- s: "op_filter_0_records_in_total",
- m: map[string]interface{}{
- "op_preprocessor_demoE_0_exceptions_total": int64(3),
- "op_preprocessor_demoE_0_process_latency_ms": int64(0),
- "op_preprocessor_demoE_0_records_in_total": int64(5),
- "op_preprocessor_demoE_0_records_out_total": int64(2),
- "op_project_0_exceptions_total": int64(3),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(4),
- "op_project_0_records_out_total": int64(1),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(4),
- "sink_mockSink_0_records_out_total": int64(4),
- "source_demoE_0_exceptions_total": int64(0),
- "source_demoE_0_records_in_total": int64(5),
- "source_demoE_0_records_out_total": int64(5),
- "op_filter_0_exceptions_total": int64(3),
- "op_filter_0_process_latency_ms": int64(0),
- "op_filter_0_records_in_total": int64(5),
- "op_filter_0_records_out_total": int64(1),
- },
- }, {
- name: `rule4`,
- sql: `SELECT size as Int8, ts FROM demoE where size > 3`,
- r: [][]map[string]interface{}{
- {{
- "error": "error in preprocessor: invalid data type for color, expect string but found int(3)",
- }},
- {{
- "Int8": float64(6),
- "ts": float64(1541152486822),
- }},
- {{
- "error": "error in preprocessor: invalid data type for color, expect string but found int(7)",
- }},
- {{
- "error": "error in preprocessor: invalid data type for size, expect bigint but found string(blue)",
- }},
- },
- s: "op_filter_0_records_in_total",
- m: map[string]interface{}{
- "op_preprocessor_demoE_0_exceptions_total": int64(3),
- "op_preprocessor_demoE_0_process_latency_ms": int64(0),
- "op_preprocessor_demoE_0_records_in_total": int64(5),
- "op_preprocessor_demoE_0_records_out_total": int64(2),
- "op_project_0_exceptions_total": int64(3),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(4),
- "op_project_0_records_out_total": int64(1),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(4),
- "sink_mockSink_0_records_out_total": int64(4),
- "source_demoE_0_exceptions_total": int64(0),
- "source_demoE_0_records_in_total": int64(5),
- "source_demoE_0_records_out_total": int64(5),
- "op_filter_0_exceptions_total": int64(3),
- "op_filter_0_process_latency_ms": int64(0),
- "op_filter_0_records_in_total": int64(5),
- "op_filter_0_records_out_total": int64(1),
- },
- }, {
- name: `rule5`,
- sql: `SELECT meta(topic) as m, ts FROM demo`,
- r: [][]map[string]interface{}{
- {{
- "m": "mock",
- "ts": float64(1541152486013),
- }},
- {{
- "m": "mock",
- "ts": float64(1541152486822),
- }},
- {{
- "m": "mock",
- "ts": float64(1541152487632),
- }},
- {{
- "m": "mock",
- "ts": float64(1541152488442),
- }},
- {{
- "m": "mock",
- "ts": float64(1541152489252),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(5),
- "op_project_0_records_out_total": int64(5),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(5),
- "sink_mockSink_0_records_out_total": int64(5),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- },
- s: "sink_mockSink_0_records_out_total",
- }, {
- name: `rule6`,
- sql: `SELECT color, ts FROM demo where size > 3 and meta(topic)="mock"`,
- r: [][]map[string]interface{}{
- {{
- "color": "blue",
- "ts": float64(1541152486822),
- }},
- {{
- "color": "yellow",
- "ts": float64(1541152488442),
- }},
- },
- s: "op_filter_0_records_in_total",
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(2),
- "op_project_0_records_out_total": int64(2),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(2),
- "sink_mockSink_0_records_out_total": int64(2),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- "op_filter_0_exceptions_total": int64(0),
- "op_filter_0_process_latency_ms": int64(0),
- "op_filter_0_records_in_total": int64(5),
- "op_filter_0_records_out_total": int64(2),
- },
- }, {
- name: `rule1`,
- sql: "SELECT `from` FROM demo1",
- r: [][]map[string]interface{}{
- {{
- "from": "device1",
- }},
- {{
- "from": "device2",
- }},
- {{
- "from": "device3",
- }},
- {{
- "from": "device1",
- }},
- {{
- "from": "device3",
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demo1_0_exceptions_total": int64(0),
- "op_preprocessor_demo1_0_process_latency_ms": int64(0),
- "op_preprocessor_demo1_0_records_in_total": int64(5),
- "op_preprocessor_demo1_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(5),
- "op_project_0_records_out_total": int64(5),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(5),
- "sink_mockSink_0_records_out_total": int64(5),
- "source_demo1_0_exceptions_total": int64(0),
- "source_demo1_0_records_in_total": int64(5),
- "source_demo1_0_records_out_total": int64(5),
- },
- s: "sink_mockSink_0_records_out_total",
- }, {
- name: `rule1`,
- sql: "SELECT * FROM demo1 where `from`=\"device1\"",
- r: [][]map[string]interface{}{
- {{
- "temp": float64(25.5),
- "hum": float64(65),
- "from": "device1",
- "ts": float64(1541152486013),
- }},
- {{
- "temp": float64(27.4),
- "hum": float64(80),
- "from": "device1",
- "ts": float64(1541152488442),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demo1_0_exceptions_total": int64(0),
- "op_preprocessor_demo1_0_process_latency_ms": int64(0),
- "op_preprocessor_demo1_0_records_in_total": int64(5),
- "op_preprocessor_demo1_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(2),
- "op_project_0_records_out_total": int64(2),
- "op_filter_0_exceptions_total": int64(0),
- "op_filter_0_process_latency_ms": int64(0),
- "op_filter_0_records_in_total": int64(5),
- "op_filter_0_records_out_total": int64(2),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(2),
- "sink_mockSink_0_records_out_total": int64(2),
- "source_demo1_0_exceptions_total": int64(0),
- "source_demo1_0_records_in_total": int64(5),
- "source_demo1_0_records_out_total": int64(5),
- },
- s: "sink_mockSink_0_records_out_total",
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- createStreams(t)
- defer dropStreams(t)
- //defer close(done)
- options := []*api.RuleOption{
- {
- BufferLength: 100,
- }, {
- BufferLength: 100,
- Qos: api.AtLeastOnce,
- CheckpointInterval: 5000,
- }, {
- BufferLength: 100,
- Qos: api.ExactlyOnce,
- CheckpointInterval: 5000,
- },
- }
- for j, opt := range options {
- for i, tt := range tests {
- test.ResetClock(1541152486000)
- p := NewRuleProcessor(DbDir)
- parser := xsql.NewParser(strings.NewReader(tt.sql))
- var (
- sources []*nodes.SourceNode
- syncs []chan int
- )
- if stmt, err := xsql.Language.Parse(parser); err != nil {
- t.Errorf("parse sql %s error: %s", tt.sql, err)
- } else {
- if selectStmt, ok := stmt.(*xsql.SelectStatement); !ok {
- t.Errorf("sql %s is not a select statement", tt.sql)
- } else {
- streams := xsql.GetStreams(selectStmt)
- for _, stream := range streams {
- next := make(chan int)
- syncs = append(syncs, next)
- source := getMockSource(stream, next, 5)
- sources = append(sources, source)
- }
- }
- }
- tp, inputs, err := p.createTopoWithSources(&api.Rule{Id: fmt.Sprintf("%s_%d", tt.name, j), Sql: tt.sql, Options: opt}, sources)
- if err != nil {
- t.Error(err)
- }
- mockSink := test.NewMockSink()
- sink := nodes.NewSinkNodeWithSink("mockSink", mockSink, nil)
- tp.AddSink(inputs, sink)
- errCh := tp.Open()
- func() {
- for i := 0; i < 5; i++ {
- syncs[i%len(syncs)] <- i
- select {
- case err = <-errCh:
- t.Log(err)
- tp.Cancel()
- return
- default:
- }
- }
- for retry := 100; retry > 0; retry-- {
- if err := compareMetrics(tp, tt.m, tt.sql); err == nil {
- break
- }
- time.Sleep(time.Duration(retry) * time.Millisecond)
- }
- }()
- results := mockSink.GetResults()
- var maps [][]map[string]interface{}
- for _, v := range results {
- var mapRes []map[string]interface{}
- err := json.Unmarshal(v, &mapRes)
- if err != nil {
- t.Errorf("Failed to parse the input into map")
- continue
- }
- maps = append(maps, mapRes)
- }
- if !reflect.DeepEqual(tt.r, maps) {
- t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.r, maps)
- continue
- }
- if err := compareMetrics(tp, tt.m, tt.sql); err != nil {
- t.Errorf("%d. %q\n\n%v", i, tt.sql, err)
- }
- tp.Cancel()
- }
- cleanStateData()
- }
- }
- func TestSingleSQLTemplate(t *testing.T) {
- var tests = []struct {
- name string
- sql string
- r []map[string]interface{}
- s string
- m map[string]interface{}
- }{
- {
- name: `rule1`,
- sql: `SELECT * FROM demo`,
- r: []map[string]interface{}{
- {
- "c": "red",
- "wrapper": "w1",
- },
- {
- "c": "blue",
- "wrapper": "w1",
- },
- {
- "c": "blue",
- "wrapper": "w1",
- },
- {
- "c": "yellow",
- "wrapper": "w1",
- },
- {
- "c": "red",
- "wrapper": "w1",
- },
- },
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(5),
- "op_project_0_records_out_total": int64(5),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(5),
- "sink_mockSink_0_records_out_total": int64(5),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- },
- s: "sink_mockSink_0_records_out_total",
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- createStreams(t)
- defer dropStreams(t)
- //defer close(done)
- for i, tt := range tests {
- test.ResetClock(1541152486000)
- p := NewRuleProcessor(DbDir)
- parser := xsql.NewParser(strings.NewReader(tt.sql))
- var (
- sources []*nodes.SourceNode
- syncs []chan int
- )
- if stmt, err := xsql.Language.Parse(parser); err != nil {
- t.Errorf("parse sql %s error: %s", tt.sql, err)
- } else {
- if selectStmt, ok := stmt.(*xsql.SelectStatement); !ok {
- t.Errorf("sql %s is not a select statement", tt.sql)
- } else {
- streams := xsql.GetStreams(selectStmt)
- for _, stream := range streams {
- next := make(chan int)
- syncs = append(syncs, next)
- source := getMockSource(stream, next, 5)
- sources = append(sources, source)
- }
- }
- }
- tp, inputs, err := p.createTopoWithSources(&api.Rule{Id: tt.name, Sql: tt.sql, Options: &api.RuleOption{
- BufferLength: 100,
- }}, sources)
- if err != nil {
- t.Error(err)
- }
- mockSink := test.NewMockSink()
- sink := nodes.NewSinkNodeWithSink("mockSink", mockSink, map[string]interface{}{
- "dataTemplate": `{"wrapper":"w1", "c":"{{.color}}"}`,
- "sendSingle": true,
- })
- tp.AddSink(inputs, sink)
- errCh := tp.Open()
- func() {
- for i := 0; i < 5; i++ {
- syncs[i%len(syncs)] <- i
- select {
- case err = <-errCh:
- t.Log(err)
- tp.Cancel()
- return
- default:
- }
- }
- for retry := 100; retry > 0; retry-- {
- if err := compareMetrics(tp, tt.m, tt.sql); err == nil {
- break
- }
- time.Sleep(time.Duration(retry) * time.Millisecond)
- }
- }()
- results := mockSink.GetResults()
- var maps []map[string]interface{}
- for _, v := range results {
- var mapRes map[string]interface{}
- err := json.Unmarshal(v, &mapRes)
- if err != nil {
- t.Errorf("Failed to parse the input into map")
- continue
- }
- maps = append(maps, mapRes)
- }
- if !reflect.DeepEqual(tt.r, maps) {
- t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.r, maps)
- continue
- }
- if err := compareMetrics(tp, tt.m, tt.sql); err != nil {
- t.Errorf("%d. %q\n\n%v", i, tt.sql, err)
- }
- tp.Cancel()
- }
- cleanStateData()
- }
- func TestNoneSingleSQLTemplate(t *testing.T) {
- var tests = []struct {
- name string
- sql string
- r [][]byte
- s string
- m map[string]interface{}
- }{
- {
- name: `rule1`,
- sql: `SELECT * FROM demo`,
- r: [][]byte{
- []byte("<div>results</div><ul><li>red - 3</li></ul>"),
- []byte("<div>results</div><ul><li>blue - 6</li></ul>"),
- []byte("<div>results</div><ul><li>blue - 2</li></ul>"),
- []byte("<div>results</div><ul><li>yellow - 4</li></ul>"),
- []byte("<div>results</div><ul><li>red - 1</li></ul>"),
- },
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(5),
- "op_project_0_records_out_total": int64(5),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(5),
- "sink_mockSink_0_records_out_total": int64(5),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- },
- s: "sink_mockSink_0_records_out_total",
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- createStreams(t)
- defer dropStreams(t)
- //defer close(done)
- for i, tt := range tests {
- test.ResetClock(1541152486000)
- p := NewRuleProcessor(DbDir)
- parser := xsql.NewParser(strings.NewReader(tt.sql))
- var (
- sources []*nodes.SourceNode
- syncs []chan int
- )
- if stmt, err := xsql.Language.Parse(parser); err != nil {
- t.Errorf("parse sql %s error: %s", tt.sql, err)
- } else {
- if selectStmt, ok := stmt.(*xsql.SelectStatement); !ok {
- t.Errorf("sql %s is not a select statement", tt.sql)
- } else {
- streams := xsql.GetStreams(selectStmt)
- for _, stream := range streams {
- next := make(chan int)
- syncs = append(syncs, next)
- source := getMockSource(stream, next, 5)
- sources = append(sources, source)
- }
- }
- }
- tp, inputs, err := p.createTopoWithSources(&api.Rule{Id: tt.name, Sql: tt.sql, Options: &api.RuleOption{
- BufferLength: 100,
- }}, sources)
- if err != nil {
- t.Error(err)
- }
- mockSink := test.NewMockSink()
- sink := nodes.NewSinkNodeWithSink("mockSink", mockSink, map[string]interface{}{
- "dataTemplate": `<div>results</div><ul>{{range .}}<li>{{.color}} - {{.size}}</li>{{end}}</ul>`,
- })
- tp.AddSink(inputs, sink)
- errCh := tp.Open()
- func() {
- for i := 0; i < 5; i++ {
- syncs[i%len(syncs)] <- i
- select {
- case err = <-errCh:
- t.Log(err)
- tp.Cancel()
- return
- default:
- }
- }
- for retry := 100; retry > 0; retry-- {
- if err := compareMetrics(tp, tt.m, tt.sql); err == nil {
- break
- }
- time.Sleep(time.Duration(retry) * time.Millisecond)
- }
- }()
- results := mockSink.GetResults()
- if !reflect.DeepEqual(tt.r, results) {
- t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.r, results)
- continue
- }
- if err := compareMetrics(tp, tt.m, tt.sql); err != nil {
- t.Errorf("%d. %q\n\n%v", i, tt.sql, err)
- }
- tp.Cancel()
- }
- cleanStateData()
- }
- func getMockSourceL(name string, done <-chan int, size int) *nodes.SourceNode {
- var data []*xsql.Tuple
- switch name {
- case "ldemo":
- data = []*xsql.Tuple{
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "red",
- "size": 3,
- "ts": 1541152486013,
- },
- Timestamp: 1541152486013,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "blue",
- "size": "string",
- "ts": 1541152486822,
- },
- Timestamp: 1541152486822,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "size": 3,
- "ts": 1541152487632,
- },
- Timestamp: 1541152487632,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": 49,
- "size": 2,
- "ts": 1541152488442,
- },
- Timestamp: 1541152488442,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "red",
- "ts": 1541152489252,
- },
- Timestamp: 1541152489252,
- },
- }
- case "ldemo1":
- data = []*xsql.Tuple{
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 65,
- "ts": 1541152486013,
- },
- Timestamp: 1541152486013,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 27.5,
- "hum": 59,
- "ts": 1541152486823,
- },
- Timestamp: 1541152486823,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 28.1,
- "hum": 75,
- "ts": 1541152487632,
- },
- Timestamp: 1541152487632,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 27.4,
- "hum": 80,
- "ts": "1541152488442",
- },
- Timestamp: 1541152488442,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 62,
- "ts": 1541152489252,
- },
- Timestamp: 1541152489252,
- },
- }
- case "lsessionDemo":
- data = []*xsql.Tuple{
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 65,
- "ts": 1541152486013,
- },
- Timestamp: 1541152486013,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 27.5,
- "hum": 59,
- "ts": 1541152486823,
- },
- Timestamp: 1541152486823,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 28.1,
- "hum": 75,
- "ts": 1541152487932,
- },
- Timestamp: 1541152487932,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 27.4,
- "hum": 80,
- "ts": 1541152488442,
- },
- Timestamp: 1541152488442,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 62,
- "ts": 1541152489252,
- },
- Timestamp: 1541152489252,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 26.2,
- "hum": 63,
- "ts": 1541152490062,
- },
- Timestamp: 1541152490062,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 26.8,
- "hum": 71,
- "ts": 1541152490872,
- },
- Timestamp: 1541152490872,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 28.9,
- "hum": 85,
- "ts": 1541152491682,
- },
- Timestamp: 1541152491682,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 29.1,
- "hum": 92,
- "ts": 1541152492492,
- },
- Timestamp: 1541152492492,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 2.2,
- "hum": 99,
- "ts": 1541152493202,
- },
- Timestamp: 1541152493202,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 30.9,
- "hum": 87,
- "ts": 1541152494112,
- },
- Timestamp: 1541152494112,
- },
- }
- }
- return nodes.NewSourceNodeWithSource(name, test.NewMockSource(data[:size], done, false), map[string]string{
- "DATASOURCE": name,
- })
- }
- func TestSingleSQLError(t *testing.T) {
- var tests = []struct {
- name string
- sql string
- r [][]map[string]interface{}
- s string
- m map[string]interface{}
- }{
- {
- name: `rule1`,
- sql: `SELECT color, ts FROM ldemo where size >= 3`,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- "ts": float64(1541152486013),
- }},
- {{
- "error": "run Where error: invalid operation string(string) >= int64(3)",
- }},
- {{
- "ts": float64(1541152487632),
- }},
- },
- s: "op_filter_0_records_in_total",
- m: map[string]interface{}{
- "op_preprocessor_ldemo_0_exceptions_total": int64(0),
- "op_preprocessor_ldemo_0_process_latency_ms": int64(0),
- "op_preprocessor_ldemo_0_records_in_total": int64(5),
- "op_preprocessor_ldemo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(1),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(3),
- "op_project_0_records_out_total": int64(2),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(3),
- "sink_mockSink_0_records_out_total": int64(3),
- "source_ldemo_0_exceptions_total": int64(0),
- "source_ldemo_0_records_in_total": int64(5),
- "source_ldemo_0_records_out_total": int64(5),
- "op_filter_0_exceptions_total": int64(1),
- "op_filter_0_process_latency_ms": int64(0),
- "op_filter_0_records_in_total": int64(5),
- "op_filter_0_records_out_total": int64(2),
- },
- }, {
- name: `rule2`,
- sql: `SELECT size * 5 FROM ldemo`,
- r: [][]map[string]interface{}{
- {{
- "rengine_field_0": float64(15),
- }},
- {{
- "error": "run Select error: invalid operation string(string) * int64(5)",
- }},
- {{
- "rengine_field_0": float64(15),
- }},
- {{
- "rengine_field_0": float64(10),
- }},
- {{}},
- },
- s: "op_filter_0_records_in_total",
- m: map[string]interface{}{
- "op_preprocessor_ldemo_0_exceptions_total": int64(0),
- "op_preprocessor_ldemo_0_process_latency_ms": int64(0),
- "op_preprocessor_ldemo_0_records_in_total": int64(5),
- "op_preprocessor_ldemo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(1),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(5),
- "op_project_0_records_out_total": int64(4),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(5),
- "sink_mockSink_0_records_out_total": int64(5),
- "source_ldemo_0_exceptions_total": int64(0),
- "source_ldemo_0_records_in_total": int64(5),
- "source_ldemo_0_records_out_total": int64(5),
- },
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- createSchemalessStreams(t)
- defer dropSchemalessStreams(t)
- //defer close(done)
- for i, tt := range tests {
- test.ResetClock(1541152486000)
- p := NewRuleProcessor(DbDir)
- parser := xsql.NewParser(strings.NewReader(tt.sql))
- var (
- sources []*nodes.SourceNode
- syncs []chan int
- )
- if stmt, err := xsql.Language.Parse(parser); err != nil {
- t.Errorf("parse sql %s error: %s", tt.sql, err)
- } else {
- if selectStmt, ok := stmt.(*xsql.SelectStatement); !ok {
- t.Errorf("sql %s is not a select statement", tt.sql)
- } else {
- streams := xsql.GetStreams(selectStmt)
- for _, stream := range streams {
- next := make(chan int)
- syncs = append(syncs, next)
- source := getMockSourceL(stream, next, 5)
- sources = append(sources, source)
- }
- }
- }
- tp, inputs, err := p.createTopoWithSources(&api.Rule{Id: tt.name, Sql: tt.sql, Options: &api.RuleOption{
- BufferLength: 100,
- }}, sources)
- if err != nil {
- t.Error(err)
- }
- mockSink := test.NewMockSink()
- sink := nodes.NewSinkNodeWithSink("mockSink", mockSink, nil)
- tp.AddSink(inputs, sink)
- errCh := tp.Open()
- func() {
- for i := 0; i < 5; i++ {
- syncs[i%len(syncs)] <- i
- select {
- case err = <-errCh:
- t.Log(err)
- tp.Cancel()
- return
- default:
- }
- }
- for retry := 100; retry > 0; retry-- {
- if err := compareMetrics(tp, tt.m, tt.sql); err == nil {
- break
- }
- time.Sleep(time.Duration(retry) * time.Millisecond)
- }
- }()
- results := mockSink.GetResults()
- var maps [][]map[string]interface{}
- for _, v := range results {
- var mapRes []map[string]interface{}
- err := json.Unmarshal(v, &mapRes)
- if err != nil {
- t.Errorf("Failed to parse the input into map")
- continue
- }
- maps = append(maps, mapRes)
- }
- if !reflect.DeepEqual(tt.r, maps) {
- t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.r, maps)
- continue
- }
- if err := compareMetrics(tp, tt.m, tt.sql); err != nil {
- t.Errorf("%d. %q\n\n%v", i, tt.sql, err)
- }
- tp.Cancel()
- }
- cleanStateData()
- }
- func TestWindow(t *testing.T) {
- var tests = []struct {
- name string
- sql string
- size int
- r [][]map[string]interface{}
- m map[string]interface{}
- }{
- {
- name: `rule1`,
- sql: `SELECT * FROM demo GROUP BY HOPPINGWINDOW(ss, 2, 1)`,
- size: 5,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- "size": float64(3),
- "ts": float64(1541152486013),
- }, {
- "color": "blue",
- "size": float64(6),
- "ts": float64(1541152486822),
- }},
- {{
- "color": "red",
- "size": float64(3),
- "ts": float64(1541152486013),
- }, {
- "color": "blue",
- "size": float64(6),
- "ts": float64(1541152486822),
- }, {
- "color": "blue",
- "size": float64(2),
- "ts": float64(1541152487632),
- }},
- {{
- "color": "blue",
- "size": float64(2),
- "ts": float64(1541152487632),
- }, {
- "color": "yellow",
- "size": float64(4),
- "ts": float64(1541152488442),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(3),
- "op_project_0_records_out_total": int64(3),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(3),
- "sink_mockSink_0_records_out_total": int64(3),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(5),
- "op_window_0_records_out_total": int64(3),
- },
- }, {
- name: `rule2`,
- sql: `SELECT color, ts FROM demo where size > 2 GROUP BY tumblingwindow(ss, 1)`,
- size: 5,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- "ts": float64(1541152486013),
- }, {
- "color": "blue",
- "ts": float64(1541152486822),
- }},
- {{
- "color": "yellow",
- "ts": float64(1541152488442),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(2),
- "op_project_0_records_out_total": int64(2),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(2),
- "sink_mockSink_0_records_out_total": int64(2),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(5),
- "op_window_0_records_out_total": int64(3),
- "op_filter_0_exceptions_total": int64(0),
- "op_filter_0_process_latency_ms": int64(0),
- "op_filter_0_records_in_total": int64(3),
- "op_filter_0_records_out_total": int64(2),
- },
- }, {
- name: `rule3`,
- sql: `SELECT color, temp, ts FROM demo INNER JOIN demo1 ON demo.ts = demo1.ts GROUP BY SlidingWindow(ss, 1)`,
- size: 5,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- "temp": 25.5,
- "ts": float64(1541152486013),
- }}, {{
- "color": "red",
- "temp": 25.5,
- "ts": float64(1541152486013),
- }}, {{
- "color": "red",
- "temp": 25.5,
- "ts": float64(1541152486013),
- }}, {{
- "color": "blue",
- "temp": 28.1,
- "ts": float64(1541152487632),
- }}, {{
- "color": "blue",
- "temp": 28.1,
- "ts": float64(1541152487632),
- }}, {{
- "color": "blue",
- "temp": 28.1,
- "ts": float64(1541152487632),
- }, {
- "color": "yellow",
- "temp": 27.4,
- "ts": float64(1541152488442),
- }}, {{
- "color": "yellow",
- "temp": 27.4,
- "ts": float64(1541152488442),
- }}, {{
- "color": "yellow",
- "temp": 27.4,
- "ts": float64(1541152488442),
- }, {
- "color": "red",
- "temp": 25.5,
- "ts": float64(1541152489252),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_preprocessor_demo1_0_exceptions_total": int64(0),
- "op_preprocessor_demo1_0_process_latency_ms": int64(0),
- "op_preprocessor_demo1_0_records_in_total": int64(5),
- "op_preprocessor_demo1_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(8),
- "op_project_0_records_out_total": int64(8),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(8),
- "sink_mockSink_0_records_out_total": int64(8),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- "source_demo1_0_exceptions_total": int64(0),
- "source_demo1_0_records_in_total": int64(5),
- "source_demo1_0_records_out_total": int64(5),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(10),
- "op_window_0_records_out_total": int64(10),
- "op_join_0_exceptions_total": int64(0),
- "op_join_0_process_latency_ms": int64(0),
- "op_join_0_records_in_total": int64(10),
- "op_join_0_records_out_total": int64(8),
- },
- }, {
- name: `rule4`,
- sql: `SELECT color FROM demo GROUP BY SlidingWindow(ss, 2), color ORDER BY color`,
- size: 5,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- }}, {{
- "color": "blue",
- }, {
- "color": "red",
- }}, {{
- "color": "blue",
- }, {
- "color": "red",
- }}, {{
- "color": "blue",
- }, {
- "color": "yellow",
- }}, {{
- "color": "blue",
- }, {
- "color": "red",
- }, {
- "color": "yellow",
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(5),
- "op_project_0_records_out_total": int64(5),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(5),
- "sink_mockSink_0_records_out_total": int64(5),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(5),
- "op_window_0_records_out_total": int64(5),
- "op_aggregate_0_exceptions_total": int64(0),
- "op_aggregate_0_process_latency_ms": int64(0),
- "op_aggregate_0_records_in_total": int64(5),
- "op_aggregate_0_records_out_total": int64(5),
- "op_order_0_exceptions_total": int64(0),
- "op_order_0_process_latency_ms": int64(0),
- "op_order_0_records_in_total": int64(5),
- "op_order_0_records_out_total": int64(5),
- },
- }, {
- name: `rule5`,
- sql: `SELECT temp FROM sessionDemo GROUP BY SessionWindow(ss, 2, 1) `,
- size: 11,
- r: [][]map[string]interface{}{
- {{
- "temp": 25.5,
- }, {
- "temp": 27.5,
- }}, {{
- "temp": 28.1,
- }, {
- "temp": 27.4,
- }, {
- "temp": 25.5,
- }}, {{
- "temp": 26.2,
- }, {
- "temp": 26.8,
- }, {
- "temp": 28.9,
- }, {
- "temp": 29.1,
- }, {
- "temp": 32.2,
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_sessionDemo_0_exceptions_total": int64(0),
- "op_preprocessor_sessionDemo_0_process_latency_ms": int64(0),
- "op_preprocessor_sessionDemo_0_records_in_total": int64(11),
- "op_preprocessor_sessionDemo_0_records_out_total": int64(11),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(3),
- "op_project_0_records_out_total": int64(3),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(3),
- "sink_mockSink_0_records_out_total": int64(3),
- "source_sessionDemo_0_exceptions_total": int64(0),
- "source_sessionDemo_0_records_in_total": int64(11),
- "source_sessionDemo_0_records_out_total": int64(11),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(11),
- "op_window_0_records_out_total": int64(3),
- },
- }, {
- name: `rule6`,
- sql: `SELECT max(temp) as m, count(color) as c FROM demo INNER JOIN demo1 ON demo.ts = demo1.ts GROUP BY SlidingWindow(ss, 1)`,
- size: 5,
- r: [][]map[string]interface{}{
- {{
- "m": 25.5,
- "c": float64(1),
- }}, {{
- "m": 25.5,
- "c": float64(1),
- }}, {{
- "m": 25.5,
- "c": float64(1),
- }}, {{
- "m": 28.1,
- "c": float64(1),
- }}, {{
- "m": 28.1,
- "c": float64(1),
- }}, {{
- "m": 28.1,
- "c": float64(2),
- }}, {{
- "m": 27.4,
- "c": float64(1),
- }}, {{
- "m": 27.4,
- "c": float64(2),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_preprocessor_demo1_0_exceptions_total": int64(0),
- "op_preprocessor_demo1_0_process_latency_ms": int64(0),
- "op_preprocessor_demo1_0_records_in_total": int64(5),
- "op_preprocessor_demo1_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(8),
- "op_project_0_records_out_total": int64(8),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(8),
- "sink_mockSink_0_records_out_total": int64(8),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- "source_demo1_0_exceptions_total": int64(0),
- "source_demo1_0_records_in_total": int64(5),
- "source_demo1_0_records_out_total": int64(5),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(10),
- "op_window_0_records_out_total": int64(10),
- "op_join_0_exceptions_total": int64(0),
- "op_join_0_process_latency_ms": int64(0),
- "op_join_0_records_in_total": int64(10),
- "op_join_0_records_out_total": int64(8),
- },
- }, {
- name: `rule7`,
- sql: `SELECT * FROM demoE GROUP BY HOPPINGWINDOW(ss, 2, 1)`,
- size: 5,
- r: [][]map[string]interface{}{
- {{
- "error": "error in preprocessor: invalid data type for color, expect string but found int(3)",
- }},
- {{
- "color": "blue",
- "size": float64(6),
- "ts": float64(1541152486822),
- }},
- {{
- "color": "blue",
- "size": float64(6),
- "ts": float64(1541152486822),
- }, {
- "color": "blue",
- "size": float64(2),
- "ts": float64(1541152487632),
- }},
- {{
- "error": "error in preprocessor: invalid data type for color, expect string but found int(7)",
- }},
- {{
- "color": "blue",
- "size": float64(2),
- "ts": float64(1541152487632),
- }},
- {{
- "error": "error in preprocessor: invalid data type for size, expect bigint but found string(blue)",
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demoE_0_exceptions_total": int64(3),
- "op_preprocessor_demoE_0_process_latency_ms": int64(0),
- "op_preprocessor_demoE_0_records_in_total": int64(5),
- "op_preprocessor_demoE_0_records_out_total": int64(2),
- "op_project_0_exceptions_total": int64(3),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(6),
- "op_project_0_records_out_total": int64(3),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(6),
- "sink_mockSink_0_records_out_total": int64(6),
- "source_demoE_0_exceptions_total": int64(0),
- "source_demoE_0_records_in_total": int64(5),
- "source_demoE_0_records_out_total": int64(5),
- "op_window_0_exceptions_total": int64(3),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(5),
- "op_window_0_records_out_total": int64(3),
- },
- }, {
- name: `rule8`,
- sql: `SELECT color, ts, count(*) as c FROM demo where size > 2 GROUP BY tumblingwindow(ss, 1) having c > 1`,
- size: 5,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- "ts": float64(1541152486013),
- "c": float64(2),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demo_0_exceptions_total": int64(0),
- "op_preprocessor_demo_0_process_latency_ms": int64(0),
- "op_preprocessor_demo_0_records_in_total": int64(5),
- "op_preprocessor_demo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(1),
- "op_project_0_records_out_total": int64(1),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(1),
- "sink_mockSink_0_records_out_total": int64(1),
- "source_demo_0_exceptions_total": int64(0),
- "source_demo_0_records_in_total": int64(5),
- "source_demo_0_records_out_total": int64(5),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(5),
- "op_window_0_records_out_total": int64(3),
- "op_filter_0_exceptions_total": int64(0),
- "op_filter_0_process_latency_ms": int64(0),
- "op_filter_0_records_in_total": int64(3),
- "op_filter_0_records_out_total": int64(2),
- "op_aggregate_0_exceptions_total": int64(0),
- "op_aggregate_0_process_latency_ms": int64(0),
- "op_aggregate_0_records_in_total": int64(2),
- "op_aggregate_0_records_out_total": int64(2),
- "op_having_0_exceptions_total": int64(0),
- "op_having_0_process_latency_ms": int64(0),
- "op_having_0_records_in_total": int64(2),
- "op_having_0_records_out_total": int64(1),
- },
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- createStreams(t)
- defer dropStreams(t)
- options := []*api.RuleOption{
- {
- BufferLength: 100,
- }, {
- BufferLength: 100,
- Qos: api.AtLeastOnce,
- CheckpointInterval: 5000,
- }, {
- BufferLength: 100,
- Qos: api.ExactlyOnce,
- CheckpointInterval: 5000,
- },
- }
- for j, opt := range options {
- for i, tt := range tests {
- test.ResetClock(1541152486000)
- p := NewRuleProcessor(DbDir)
- parser := xsql.NewParser(strings.NewReader(tt.sql))
- var (
- sources []*nodes.SourceNode
- syncs []chan int
- )
- if stmt, err := xsql.Language.Parse(parser); err != nil {
- t.Errorf("parse sql %s error: %s", tt.sql, err)
- } else {
- if selectStmt, ok := stmt.(*xsql.SelectStatement); !ok {
- t.Errorf("sql %s is not a select statement", tt.sql)
- } else {
- streams := xsql.GetStreams(selectStmt)
- for _, stream := range streams {
- next := make(chan int)
- syncs = append(syncs, next)
- source := getMockSource(stream, next, tt.size)
- sources = append(sources, source)
- }
- }
- }
- tp, inputs, err := p.createTopoWithSources(&api.Rule{Id: fmt.Sprintf("%s_%d", tt.name, j), Sql: tt.sql, Options: opt}, sources)
- if err != nil {
- t.Error(err)
- }
- mockSink := test.NewMockSink()
- sink := nodes.NewSinkNodeWithSink("mockSink", mockSink, nil)
- tp.AddSink(inputs, sink)
- errCh := tp.Open()
- func() {
- for i := 0; i < tt.size*len(syncs); i++ {
- syncs[i%len(syncs)] <- i
- for {
- time.Sleep(1)
- if getMetric(tp, "op_window_0_records_in_total") == (i + 1) {
- break
- }
- }
- select {
- case err = <-errCh:
- t.Log(err)
- tp.Cancel()
- return
- default:
- }
- }
- retry := 100
- for ; retry > 0; retry-- {
- if err := compareMetrics(tp, tt.m, tt.sql); err == nil {
- break
- }
- t.Logf("wait to try another %d times", retry)
- time.Sleep(time.Duration(retry) * time.Millisecond)
- }
- if retry == 0 {
- err := compareMetrics(tp, tt.m, tt.sql)
- t.Errorf("could not get correct metrics: %v", err)
- }
- }()
- results := mockSink.GetResults()
- var maps [][]map[string]interface{}
- for _, v := range results {
- var mapRes []map[string]interface{}
- err := json.Unmarshal(v, &mapRes)
- if err != nil {
- t.Errorf("Failed to parse the input into map")
- continue
- }
- maps = append(maps, mapRes)
- }
- if !reflect.DeepEqual(tt.r, maps) {
- t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.r, maps)
- }
- if err := compareMetrics(tp, tt.m, tt.sql); err != nil {
- t.Errorf("%d. %q\n\n%v", i, tt.sql, err)
- }
- tp.Cancel()
- }
- cleanStateData()
- }
- }
- func TestWindowError(t *testing.T) {
- var tests = []struct {
- name string
- sql string
- size int
- r [][]map[string]interface{}
- m map[string]interface{}
- }{
- {
- name: `rule1`,
- sql: `SELECT size * 3 FROM ldemo GROUP BY TUMBLINGWINDOW(ss, 2)`,
- size: 5,
- r: [][]map[string]interface{}{
- {{
- "error": "run Select error: invalid operation string(string) * int64(3)",
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_ldemo_0_exceptions_total": int64(0),
- "op_preprocessor_ldemo_0_process_latency_ms": int64(0),
- "op_preprocessor_ldemo_0_records_in_total": int64(5),
- "op_preprocessor_ldemo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(1),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(1),
- "op_project_0_records_out_total": int64(0),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(1),
- "sink_mockSink_0_records_out_total": int64(1),
- "source_ldemo_0_exceptions_total": int64(0),
- "source_ldemo_0_records_in_total": int64(5),
- "source_ldemo_0_records_out_total": int64(5),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(5),
- "op_window_0_records_out_total": int64(1),
- },
- }, {
- name: `rule2`,
- sql: `SELECT color, ts FROM ldemo where size > 2 GROUP BY tumblingwindow(ss, 1)`,
- size: 5,
- r: [][]map[string]interface{}{
- {{
- "error": "run Where error: invalid operation string(string) > int64(2)",
- }}, {{
- "ts": float64(1541152487632),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_ldemo_0_exceptions_total": int64(0),
- "op_preprocessor_ldemo_0_process_latency_ms": int64(0),
- "op_preprocessor_ldemo_0_records_in_total": int64(5),
- "op_preprocessor_ldemo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(1),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(2),
- "op_project_0_records_out_total": int64(1),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(2),
- "sink_mockSink_0_records_out_total": int64(2),
- "source_ldemo_0_exceptions_total": int64(0),
- "source_ldemo_0_records_in_total": int64(5),
- "source_ldemo_0_records_out_total": int64(5),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(5),
- "op_window_0_records_out_total": int64(3),
- "op_filter_0_exceptions_total": int64(1),
- "op_filter_0_process_latency_ms": int64(0),
- "op_filter_0_records_in_total": int64(3),
- "op_filter_0_records_out_total": int64(1),
- },
- }, {
- name: `rule3`,
- sql: `SELECT color, temp, ts FROM ldemo INNER JOIN ldemo1 ON ldemo.ts = ldemo1.ts GROUP BY SlidingWindow(ss, 1)`,
- size: 5,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- "temp": 25.5,
- "ts": float64(1541152486013),
- }}, {{
- "color": "red",
- "temp": 25.5,
- "ts": float64(1541152486013),
- }}, {{
- "color": "red",
- "temp": 25.5,
- "ts": float64(1541152486013),
- }}, {{
- "temp": 28.1,
- "ts": float64(1541152487632),
- }}, {{
- "temp": 28.1,
- "ts": float64(1541152487632),
- }}, {{
- "error": "run Join error: invalid operation int64(1541152487632) = string(1541152488442)",
- }}, {{
- "error": "run Join error: invalid operation int64(1541152488442) = string(1541152488442)",
- }}, {{
- "error": "run Join error: invalid operation int64(1541152488442) = string(1541152488442)",
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_ldemo_0_exceptions_total": int64(0),
- "op_preprocessor_ldemo_0_process_latency_ms": int64(0),
- "op_preprocessor_ldemo_0_records_in_total": int64(5),
- "op_preprocessor_ldemo_0_records_out_total": int64(5),
- "op_preprocessor_ldemo1_0_exceptions_total": int64(0),
- "op_preprocessor_ldemo1_0_process_latency_ms": int64(0),
- "op_preprocessor_ldemo1_0_records_in_total": int64(5),
- "op_preprocessor_ldemo1_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(3),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(8),
- "op_project_0_records_out_total": int64(5),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(8),
- "sink_mockSink_0_records_out_total": int64(8),
- "source_ldemo_0_exceptions_total": int64(0),
- "source_ldemo_0_records_in_total": int64(5),
- "source_ldemo_0_records_out_total": int64(5),
- "source_ldemo1_0_exceptions_total": int64(0),
- "source_ldemo1_0_records_in_total": int64(5),
- "source_ldemo1_0_records_out_total": int64(5),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(10),
- "op_window_0_records_out_total": int64(10),
- "op_join_0_exceptions_total": int64(3),
- "op_join_0_process_latency_ms": int64(0),
- "op_join_0_records_in_total": int64(10),
- "op_join_0_records_out_total": int64(5),
- },
- }, {
- name: `rule4`,
- sql: `SELECT color FROM ldemo GROUP BY SlidingWindow(ss, 2), color having size >= 2 order by color`,
- size: 5,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- }}, {{
- "error": "run Having error: invalid operation string(string) >= int64(2)",
- }}, {{
- "error": "run Having error: invalid operation string(string) >= int64(2)",
- }}, {{
- "error": "run Having error: invalid operation string(string) >= int64(2)",
- }}, {{
- "color": float64(49),
- }, {}},
- },
- m: map[string]interface{}{
- "op_preprocessor_ldemo_0_exceptions_total": int64(0),
- "op_preprocessor_ldemo_0_process_latency_ms": int64(0),
- "op_preprocessor_ldemo_0_records_in_total": int64(5),
- "op_preprocessor_ldemo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(3),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(5),
- "op_project_0_records_out_total": int64(2),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(5),
- "sink_mockSink_0_records_out_total": int64(5),
- "source_ldemo_0_exceptions_total": int64(0),
- "source_ldemo_0_records_in_total": int64(5),
- "source_ldemo_0_records_out_total": int64(5),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(5),
- "op_window_0_records_out_total": int64(5),
- "op_aggregate_0_exceptions_total": int64(0),
- "op_aggregate_0_process_latency_ms": int64(0),
- "op_aggregate_0_records_in_total": int64(5),
- "op_aggregate_0_records_out_total": int64(5),
- "op_having_0_exceptions_total": int64(3),
- "op_having_0_process_latency_ms": int64(0),
- "op_having_0_records_in_total": int64(5),
- "op_having_0_records_out_total": int64(2),
- },
- }, {
- name: `rule5`,
- sql: `SELECT color, size FROM ldemo GROUP BY tumblingwindow(ss, 1) ORDER BY size`,
- size: 5,
- r: [][]map[string]interface{}{
- {{
- "error": "run Order By error: incompatible types for comparison: int and string",
- }}, {{
- "size": float64(3),
- }}, {{
- "color": float64(49),
- "size": float64(2),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_ldemo_0_exceptions_total": int64(0),
- "op_preprocessor_ldemo_0_process_latency_ms": int64(0),
- "op_preprocessor_ldemo_0_records_in_total": int64(5),
- "op_preprocessor_ldemo_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(1),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(3),
- "op_project_0_records_out_total": int64(2),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(3),
- "sink_mockSink_0_records_out_total": int64(3),
- "source_ldemo_0_exceptions_total": int64(0),
- "source_ldemo_0_records_in_total": int64(5),
- "source_ldemo_0_records_out_total": int64(5),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(5),
- "op_window_0_records_out_total": int64(3),
- "op_order_0_exceptions_total": int64(1),
- "op_order_0_process_latency_ms": int64(0),
- "op_order_0_records_in_total": int64(3),
- "op_order_0_records_out_total": int64(2),
- },
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- createSchemalessStreams(t)
- defer dropSchemalessStreams(t)
- for i, tt := range tests {
- test.ResetClock(1541152486000)
- p := NewRuleProcessor(DbDir)
- parser := xsql.NewParser(strings.NewReader(tt.sql))
- var (
- sources []*nodes.SourceNode
- syncs []chan int
- )
- if stmt, err := xsql.Language.Parse(parser); err != nil {
- t.Errorf("parse sql %s error: %s", tt.sql, err)
- } else {
- if selectStmt, ok := stmt.(*xsql.SelectStatement); !ok {
- t.Errorf("sql %s is not a select statement", tt.sql)
- } else {
- streams := xsql.GetStreams(selectStmt)
- for _, stream := range streams {
- next := make(chan int)
- syncs = append(syncs, next)
- source := getMockSourceL(stream, next, tt.size)
- sources = append(sources, source)
- }
- }
- }
- tp, inputs, err := p.createTopoWithSources(&api.Rule{Id: tt.name, Sql: tt.sql, Options: &api.RuleOption{
- BufferLength: 100,
- }}, sources)
- if err != nil {
- t.Error(err)
- }
- mockSink := test.NewMockSink()
- sink := nodes.NewSinkNodeWithSink("mockSink", mockSink, nil)
- tp.AddSink(inputs, sink)
- errCh := tp.Open()
- func() {
- for i := 0; i < tt.size*len(syncs); i++ {
- syncs[i%len(syncs)] <- i
- for {
- time.Sleep(1)
- if getMetric(tp, "op_window_0_records_in_total") == (i + 1) {
- break
- }
- }
- select {
- case err = <-errCh:
- t.Log(err)
- tp.Cancel()
- return
- default:
- }
- }
- retry := 100
- for ; retry > 0; retry-- {
- if err := compareMetrics(tp, tt.m, tt.sql); err == nil {
- break
- }
- t.Logf("wait to try another %d times", retry)
- time.Sleep(time.Duration(retry) * time.Millisecond)
- }
- if retry == 0 {
- err := compareMetrics(tp, tt.m, tt.sql)
- t.Errorf("could not get correct metrics: %v", err)
- }
- }()
- results := mockSink.GetResults()
- var maps [][]map[string]interface{}
- for _, v := range results {
- var mapRes []map[string]interface{}
- err := json.Unmarshal(v, &mapRes)
- if err != nil {
- t.Errorf("Failed to parse the input into map")
- continue
- }
- maps = append(maps, mapRes)
- }
- if !reflect.DeepEqual(tt.r, maps) {
- t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.r, maps)
- }
- if err := compareMetrics(tp, tt.m, tt.sql); err != nil {
- t.Errorf("%d. %q\n\n%v", i, tt.sql, err)
- }
- tp.Cancel()
- }
- cleanStateData()
- }
- func createEventStreams(t *testing.T) {
- p := NewStreamProcessor(path.Join(DbDir, "stream"))
- demo := `CREATE STREAM demoE (
- color STRING,
- size BIGINT,
- ts BIGINT
- ) WITH (DATASOURCE="demoE", FORMAT="json", KEY="ts", TIMESTAMP="ts");`
- _, err := p.ExecStmt(demo)
- if err != nil {
- t.Log(err)
- }
- demo1 := `CREATE STREAM demo1E (
- temp FLOAT,
- hum BIGINT,
- ts BIGINT
- ) WITH (DATASOURCE="demo1E", FORMAT="json", KEY="ts", TIMESTAMP="ts");`
- _, err = p.ExecStmt(demo1)
- if err != nil {
- t.Log(err)
- }
- sessionDemo := `CREATE STREAM sessionDemoE (
- temp FLOAT,
- hum BIGINT,
- ts BIGINT
- ) WITH (DATASOURCE="sessionDemoE", FORMAT="json", KEY="ts", TIMESTAMP="ts");`
- _, err = p.ExecStmt(sessionDemo)
- if err != nil {
- t.Log(err)
- }
- demoErr := `CREATE STREAM demoErr (
- color STRING,
- size BIGINT,
- ts BIGINT
- ) WITH (DATASOURCE="demoErr", FORMAT="json", KEY="ts", TIMESTAMP="ts");`
- _, err = p.ExecStmt(demoErr)
- if err != nil {
- t.Log(err)
- }
- }
- func dropEventStreams(t *testing.T) {
- p := NewStreamProcessor(path.Join(DbDir, "stream"))
- demo := `DROP STREAM demoE`
- _, err := p.ExecStmt(demo)
- if err != nil {
- t.Log(err)
- }
- demo1 := `DROP STREAM demo1E`
- _, err = p.ExecStmt(demo1)
- if err != nil {
- t.Log(err)
- }
- sessionDemo := `DROP STREAM sessionDemoE`
- _, err = p.ExecStmt(sessionDemo)
- if err != nil {
- t.Log(err)
- }
- demoErr := `DROP STREAM demoErr`
- _, err = p.ExecStmt(demoErr)
- if err != nil {
- t.Log(err)
- }
- }
- func getEventMockSource(name string, done <-chan int, size int) *nodes.SourceNode {
- var data []*xsql.Tuple
- switch name {
- case "demoE":
- data = []*xsql.Tuple{
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "red",
- "size": 3,
- "ts": 1541152486013,
- },
- Timestamp: 1541152486013,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "blue",
- "size": 2,
- "ts": 1541152487632,
- },
- Timestamp: 1541152487632,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "red",
- "size": 1,
- "ts": 1541152489252,
- },
- Timestamp: 1541152489252,
- },
- { //dropped item
- Emitter: name,
- Message: map[string]interface{}{
- "color": "blue",
- "size": 6,
- "ts": 1541152486822,
- },
- Timestamp: 1541152486822,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "yellow",
- "size": 4,
- "ts": 1541152488442,
- },
- Timestamp: 1541152488442,
- },
- { //To lift the watermark and issue all windows
- Emitter: name,
- Message: map[string]interface{}{
- "color": "yellow",
- "size": 4,
- "ts": 1541152492342,
- },
- Timestamp: 1541152488442,
- },
- }
- case "demo1E":
- data = []*xsql.Tuple{
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 27.5,
- "hum": 59,
- "ts": 1541152486823,
- },
- Timestamp: 1541152486823,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 65,
- "ts": 1541152486013,
- },
- Timestamp: 1541152486013,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 27.4,
- "hum": 80,
- "ts": 1541152488442,
- },
- Timestamp: 1541152488442,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 28.1,
- "hum": 75,
- "ts": 1541152487632,
- },
- Timestamp: 1541152487632,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 62,
- "ts": 1541152489252,
- },
- Timestamp: 1541152489252,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 62,
- "ts": 1541152499252,
- },
- Timestamp: 1541152499252,
- },
- }
- case "sessionDemoE":
- data = []*xsql.Tuple{
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 65,
- "ts": 1541152486013,
- },
- Timestamp: 1541152486013,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 28.1,
- "hum": 75,
- "ts": 1541152487932,
- },
- Timestamp: 1541152487932,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 27.5,
- "hum": 59,
- "ts": 1541152486823,
- },
- Timestamp: 1541152486823,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 25.5,
- "hum": 62,
- "ts": 1541152489252,
- },
- Timestamp: 1541152489252,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 27.4,
- "hum": 80,
- "ts": 1541152488442,
- },
- Timestamp: 1541152488442,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 26.2,
- "hum": 63,
- "ts": 1541152490062,
- },
- Timestamp: 1541152490062,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 28.9,
- "hum": 85,
- "ts": 1541152491682,
- },
- Timestamp: 1541152491682,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 26.8,
- "hum": 71,
- "ts": 1541152490872,
- },
- Timestamp: 1541152490872,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 29.1,
- "hum": 92,
- "ts": 1541152492492,
- },
- Timestamp: 1541152492492,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 30.9,
- "hum": 87,
- "ts": 1541152494112,
- },
- Timestamp: 1541152494112,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 32.2,
- "hum": 99,
- "ts": 1541152493202,
- },
- Timestamp: 1541152493202,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "temp": 32.2,
- "hum": 99,
- "ts": 1541152499202,
- },
- Timestamp: 1541152499202,
- },
- }
- case "demoErr":
- data = []*xsql.Tuple{
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "red",
- "size": 3,
- "ts": 1541152486013,
- },
- Timestamp: 1541152486013,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": 2,
- "size": "blue",
- "ts": 1541152487632,
- },
- Timestamp: 1541152487632,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "red",
- "size": 1,
- "ts": 1541152489252,
- },
- Timestamp: 1541152489252,
- },
- { //dropped item
- Emitter: name,
- Message: map[string]interface{}{
- "color": "blue",
- "size": 6,
- "ts": 1541152486822,
- },
- Timestamp: 1541152486822,
- },
- {
- Emitter: name,
- Message: map[string]interface{}{
- "color": "yellow",
- "size": 4,
- "ts": 1541152488442,
- },
- Timestamp: 1541152488442,
- },
- { //To lift the watermark and issue all windows
- Emitter: name,
- Message: map[string]interface{}{
- "color": "yellow",
- "size": 4,
- "ts": 1541152492342,
- },
- Timestamp: 1541152488442,
- },
- }
- }
- return nodes.NewSourceNodeWithSource(name, test.NewMockSource(data[:size], done, true), map[string]string{
- "DATASOURCE": name,
- })
- }
- func TestEventWindow(t *testing.T) {
- var tests = []struct {
- name string
- sql string
- size int
- r [][]map[string]interface{}
- m map[string]interface{}
- }{
- {
- name: `rule1`,
- sql: `SELECT * FROM demoE GROUP BY HOPPINGWINDOW(ss, 2, 1)`,
- size: 6,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- "size": float64(3),
- "ts": float64(1541152486013),
- }},
- {{
- "color": "red",
- "size": float64(3),
- "ts": float64(1541152486013),
- }, {
- "color": "blue",
- "size": float64(2),
- "ts": float64(1541152487632),
- }},
- {{
- "color": "blue",
- "size": float64(2),
- "ts": float64(1541152487632),
- }, {
- "color": "yellow",
- "size": float64(4),
- "ts": float64(1541152488442),
- }}, {{
- "color": "yellow",
- "size": float64(4),
- "ts": float64(1541152488442),
- }, {
- "color": "red",
- "size": float64(1),
- "ts": float64(1541152489252),
- }}, {{
- "color": "red",
- "size": float64(1),
- "ts": float64(1541152489252),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demoE_0_exceptions_total": int64(0),
- "op_preprocessor_demoE_0_process_latency_ms": int64(0),
- "op_preprocessor_demoE_0_records_in_total": int64(6),
- "op_preprocessor_demoE_0_records_out_total": int64(6),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(5),
- "op_project_0_records_out_total": int64(5),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(5),
- "sink_mockSink_0_records_out_total": int64(5),
- "source_demoE_0_exceptions_total": int64(0),
- "source_demoE_0_records_in_total": int64(6),
- "source_demoE_0_records_out_total": int64(6),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(6),
- "op_window_0_records_out_total": int64(5),
- },
- }, {
- name: `rule2`,
- sql: `SELECT color, ts FROM demoE where size > 2 GROUP BY tumblingwindow(ss, 1)`,
- size: 6,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- "ts": float64(1541152486013),
- }},
- {{
- "color": "yellow",
- "ts": float64(1541152488442),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demoE_0_exceptions_total": int64(0),
- "op_preprocessor_demoE_0_process_latency_ms": int64(0),
- "op_preprocessor_demoE_0_records_in_total": int64(6),
- "op_preprocessor_demoE_0_records_out_total": int64(6),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(2),
- "op_project_0_records_out_total": int64(2),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(2),
- "sink_mockSink_0_records_out_total": int64(2),
- "source_demoE_0_exceptions_total": int64(0),
- "source_demoE_0_records_in_total": int64(6),
- "source_demoE_0_records_out_total": int64(6),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(6),
- "op_window_0_records_out_total": int64(4),
- "op_filter_0_exceptions_total": int64(0),
- "op_filter_0_process_latency_ms": int64(0),
- "op_filter_0_records_in_total": int64(4),
- "op_filter_0_records_out_total": int64(2),
- },
- }, {
- name: `rule3`,
- sql: `SELECT color, temp, ts FROM demoE INNER JOIN demo1E ON demoE.ts = demo1E.ts GROUP BY SlidingWindow(ss, 1)`,
- size: 6,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- "temp": 25.5,
- "ts": float64(1541152486013),
- }}, {{
- "color": "red",
- "temp": 25.5,
- "ts": float64(1541152486013),
- }}, {{
- "color": "blue",
- "temp": 28.1,
- "ts": float64(1541152487632),
- }}, {{
- "color": "blue",
- "temp": 28.1,
- "ts": float64(1541152487632),
- }, {
- "color": "yellow",
- "temp": 27.4,
- "ts": float64(1541152488442),
- }}, {{
- "color": "yellow",
- "temp": 27.4,
- "ts": float64(1541152488442),
- }, {
- "color": "red",
- "temp": 25.5,
- "ts": float64(1541152489252),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demoE_0_exceptions_total": int64(0),
- "op_preprocessor_demoE_0_process_latency_ms": int64(0),
- "op_preprocessor_demoE_0_records_in_total": int64(6),
- "op_preprocessor_demoE_0_records_out_total": int64(6),
- "op_preprocessor_demo1E_0_exceptions_total": int64(0),
- "op_preprocessor_demo1E_0_process_latency_ms": int64(0),
- "op_preprocessor_demo1E_0_records_in_total": int64(6),
- "op_preprocessor_demo1E_0_records_out_total": int64(6),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(5),
- "op_project_0_records_out_total": int64(5),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(5),
- "sink_mockSink_0_records_out_total": int64(5),
- "source_demoE_0_exceptions_total": int64(0),
- "source_demoE_0_records_in_total": int64(6),
- "source_demoE_0_records_out_total": int64(6),
- "source_demo1E_0_exceptions_total": int64(0),
- "source_demo1E_0_records_in_total": int64(6),
- "source_demo1E_0_records_out_total": int64(6),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(12),
- "op_window_0_records_out_total": int64(5),
- "op_join_0_exceptions_total": int64(0),
- "op_join_0_process_latency_ms": int64(0),
- "op_join_0_records_in_total": int64(5),
- "op_join_0_records_out_total": int64(5),
- },
- }, {
- name: `rule4`,
- sql: `SELECT color FROM demoE GROUP BY SlidingWindow(ss, 2), color ORDER BY color`,
- size: 6,
- r: [][]map[string]interface{}{
- {{
- "color": "red",
- }}, {{
- "color": "blue",
- }, {
- "color": "red",
- }}, {{
- "color": "blue",
- }, {
- "color": "yellow",
- }}, {{
- "color": "blue",
- }, {
- "color": "red",
- }, {
- "color": "yellow",
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demoE_0_exceptions_total": int64(0),
- "op_preprocessor_demoE_0_process_latency_ms": int64(0),
- "op_preprocessor_demoE_0_records_in_total": int64(6),
- "op_preprocessor_demoE_0_records_out_total": int64(6),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(4),
- "op_project_0_records_out_total": int64(4),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(4),
- "sink_mockSink_0_records_out_total": int64(4),
- "source_demoE_0_exceptions_total": int64(0),
- "source_demoE_0_records_in_total": int64(6),
- "source_demoE_0_records_out_total": int64(6),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(6),
- "op_window_0_records_out_total": int64(4),
- "op_aggregate_0_exceptions_total": int64(0),
- "op_aggregate_0_process_latency_ms": int64(0),
- "op_aggregate_0_records_in_total": int64(4),
- "op_aggregate_0_records_out_total": int64(4),
- "op_order_0_exceptions_total": int64(0),
- "op_order_0_process_latency_ms": int64(0),
- "op_order_0_records_in_total": int64(4),
- "op_order_0_records_out_total": int64(4),
- },
- }, {
- name: `rule5`,
- sql: `SELECT temp FROM sessionDemoE GROUP BY SessionWindow(ss, 2, 1) `,
- size: 12,
- r: [][]map[string]interface{}{
- {{
- "temp": 25.5,
- }}, {{
- "temp": 28.1,
- }, {
- "temp": 27.4,
- }, {
- "temp": 25.5,
- }}, {{
- "temp": 26.2,
- }, {
- "temp": 26.8,
- }, {
- "temp": 28.9,
- }, {
- "temp": 29.1,
- }, {
- "temp": 32.2,
- }}, {{
- "temp": 30.9,
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_sessionDemoE_0_exceptions_total": int64(0),
- "op_preprocessor_sessionDemoE_0_process_latency_ms": int64(0),
- "op_preprocessor_sessionDemoE_0_records_in_total": int64(12),
- "op_preprocessor_sessionDemoE_0_records_out_total": int64(12),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(4),
- "op_project_0_records_out_total": int64(4),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(4),
- "sink_mockSink_0_records_out_total": int64(4),
- "source_sessionDemoE_0_exceptions_total": int64(0),
- "source_sessionDemoE_0_records_in_total": int64(12),
- "source_sessionDemoE_0_records_out_total": int64(12),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(12),
- "op_window_0_records_out_total": int64(4),
- },
- }, {
- name: `rule6`,
- sql: `SELECT max(temp) as m, count(color) as c FROM demoE INNER JOIN demo1E ON demoE.ts = demo1E.ts GROUP BY SlidingWindow(ss, 1)`,
- size: 6,
- r: [][]map[string]interface{}{
- {{
- "m": 25.5,
- "c": float64(1),
- }}, {{
- "m": 25.5,
- "c": float64(1),
- }}, {{
- "m": 28.1,
- "c": float64(1),
- }}, {{
- "m": 28.1,
- "c": float64(2),
- }}, {{
- "m": 27.4,
- "c": float64(2),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demoE_0_exceptions_total": int64(0),
- "op_preprocessor_demoE_0_process_latency_ms": int64(0),
- "op_preprocessor_demoE_0_records_in_total": int64(6),
- "op_preprocessor_demoE_0_records_out_total": int64(6),
- "op_preprocessor_demo1E_0_exceptions_total": int64(0),
- "op_preprocessor_demo1E_0_process_latency_ms": int64(0),
- "op_preprocessor_demo1E_0_records_in_total": int64(6),
- "op_preprocessor_demo1E_0_records_out_total": int64(6),
- "op_project_0_exceptions_total": int64(0),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(5),
- "op_project_0_records_out_total": int64(5),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(5),
- "sink_mockSink_0_records_out_total": int64(5),
- "source_demoE_0_exceptions_total": int64(0),
- "source_demoE_0_records_in_total": int64(6),
- "source_demoE_0_records_out_total": int64(6),
- "source_demo1E_0_exceptions_total": int64(0),
- "source_demo1E_0_records_in_total": int64(6),
- "source_demo1E_0_records_out_total": int64(6),
- "op_window_0_exceptions_total": int64(0),
- "op_window_0_records_in_total": int64(12),
- "op_window_0_records_out_total": int64(5),
- "op_join_0_exceptions_total": int64(0),
- "op_join_0_process_latency_ms": int64(0),
- "op_join_0_records_in_total": int64(5),
- "op_join_0_records_out_total": int64(5),
- },
- }, {
- name: `rule7`,
- sql: `SELECT * FROM demoErr GROUP BY HOPPINGWINDOW(ss, 2, 1)`,
- size: 6,
- r: [][]map[string]interface{}{
- {{
- "error": "error in preprocessor: invalid data type for color, expect string but found int(2)",
- }},
- {{
- "color": "red",
- "size": float64(3),
- "ts": float64(1541152486013),
- }},
- {{
- "color": "red",
- "size": float64(3),
- "ts": float64(1541152486013),
- }},
- {{
- "color": "yellow",
- "size": float64(4),
- "ts": float64(1541152488442),
- }}, {{
- "color": "yellow",
- "size": float64(4),
- "ts": float64(1541152488442),
- }, {
- "color": "red",
- "size": float64(1),
- "ts": float64(1541152489252),
- }}, {{
- "color": "red",
- "size": float64(1),
- "ts": float64(1541152489252),
- }},
- },
- m: map[string]interface{}{
- "op_preprocessor_demoErr_0_exceptions_total": int64(1),
- "op_preprocessor_demoErr_0_process_latency_ms": int64(0),
- "op_preprocessor_demoErr_0_records_in_total": int64(6),
- "op_preprocessor_demoErr_0_records_out_total": int64(5),
- "op_project_0_exceptions_total": int64(1),
- "op_project_0_process_latency_ms": int64(0),
- "op_project_0_records_in_total": int64(6),
- "op_project_0_records_out_total": int64(5),
- "sink_mockSink_0_exceptions_total": int64(0),
- "sink_mockSink_0_records_in_total": int64(6),
- "sink_mockSink_0_records_out_total": int64(6),
- "source_demoErr_0_exceptions_total": int64(0),
- "source_demoErr_0_records_in_total": int64(6),
- "source_demoErr_0_records_out_total": int64(6),
- "op_window_0_exceptions_total": int64(1),
- "op_window_0_process_latency_ms": int64(0),
- "op_window_0_records_in_total": int64(6),
- "op_window_0_records_out_total": int64(5),
- },
- },
- }
- fmt.Printf("The test bucket size is %d.\n\n", len(tests))
- createEventStreams(t)
- defer dropEventStreams(t)
- options := []*api.RuleOption{
- {
- BufferLength: 100,
- IsEventTime: true,
- LateTol: 1000,
- }, {
- BufferLength: 100,
- Qos: api.AtLeastOnce,
- CheckpointInterval: 5000,
- IsEventTime: true,
- LateTol: 1000,
- }, {
- BufferLength: 100,
- Qos: api.ExactlyOnce,
- CheckpointInterval: 5000,
- IsEventTime: true,
- LateTol: 1000,
- },
- }
- for j, opt := range options {
- for i, tt := range tests {
- test.ResetClock(1541152486000)
- p := NewRuleProcessor(DbDir)
- parser := xsql.NewParser(strings.NewReader(tt.sql))
- var (
- sources []*nodes.SourceNode
- syncs []chan int
- )
- if stmt, err := xsql.Language.Parse(parser); err != nil {
- t.Errorf("parse sql %s error: %s", tt.sql, err)
- } else {
- if selectStmt, ok := stmt.(*xsql.SelectStatement); !ok {
- t.Errorf("sql %s is not a select statement", tt.sql)
- } else {
- streams := xsql.GetStreams(selectStmt)
- for _, stream := range streams {
- next := make(chan int)
- syncs = append(syncs, next)
- source := getEventMockSource(stream, next, tt.size)
- sources = append(sources, source)
- }
- }
- }
- tp, inputs, err := p.createTopoWithSources(&api.Rule{Id: fmt.Sprintf("%s_%d", tt.name, j), Sql: tt.sql, Options: opt}, sources)
- if err != nil {
- t.Error(err)
- }
- mockSink := test.NewMockSink()
- sink := nodes.NewSinkNodeWithSink("mockSink", mockSink, nil)
- tp.AddSink(inputs, sink)
- errCh := tp.Open()
- func() {
- for i := 0; i < tt.size*len(syncs); i++ {
- syncs[i%len(syncs)] <- i
- for {
- time.Sleep(1)
- if getMetric(tp, "op_window_0_records_in_total") == (i + 1) {
- break
- }
- }
- select {
- case err = <-errCh:
- t.Log(err)
- tp.Cancel()
- return
- default:
- }
- }
- mockClock := test.GetMockClock()
- mockClock.Add(1000 * time.Millisecond)
- retry := 100
- for ; retry > 0; retry-- {
- if err := compareMetrics(tp, tt.m, tt.sql); err == nil {
- break
- }
- t.Logf("wait to try another %d times", retry)
- time.Sleep(time.Duration(retry) * time.Millisecond)
- }
- if retry == 0 {
- err := compareMetrics(tp, tt.m, tt.sql)
- t.Errorf("could not get correct metrics: %v", err)
- }
- }()
- results := mockSink.GetResults()
- var maps [][]map[string]interface{}
- for _, v := range results {
- var mapRes []map[string]interface{}
- err := json.Unmarshal(v, &mapRes)
- if err != nil {
- t.Errorf("Failed to parse the input into map")
- continue
- }
- maps = append(maps, mapRes)
- }
- if !reflect.DeepEqual(tt.r, maps) {
- t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.r, maps)
- }
- if err := compareMetrics(tp, tt.m, tt.sql); err != nil {
- t.Errorf("%d. %q\n\n%v", i, tt.sql, err)
- }
- tp.Cancel()
- }
- cleanStateData()
- }
- }
- func getMetric(tp *xstream.TopologyNew, name string) int {
- keys, values := tp.GetMetrics()
- for index, key := range keys {
- if key == name {
- return int(values[index].(int64))
- }
- }
- fmt.Println("can't find " + name)
- return 0
- }
- func compareMetrics(tp *xstream.TopologyNew, m map[string]interface{}, sql string) (err error) {
- keys, values := tp.GetMetrics()
- //for i, k := range keys {
- // log.Printf("%s:%v", k, values[i])
- //}
- for k, v := range m {
- var (
- index int
- key string
- matched bool
- )
- for index, key = range keys {
- if k == key {
- if strings.HasSuffix(k, "process_latency_ms") {
- if values[index].(int64) >= v.(int64) {
- matched = true
- continue
- } else {
- break
- }
- }
- if values[index] == v {
- matched = true
- }
- break
- }
- }
- if matched {
- continue
- }
- //do not find
- if index < len(values) {
- return fmt.Errorf("metrics mismatch for %s:\n\nexp=%#v(%t)\n\ngot=%#v(%t)\n\n", k, v, v, values[index], values[index])
- } else {
- return fmt.Errorf("metrics mismatch for %s:\n\nexp=%#v\n\ngot=nil\n\n", k, v)
- }
- }
- return nil
- }
- func errstring(err error) string {
- if err != nil {
- return err.Error()
- }
- return ""
- }
|