join_test.go 54 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199
  1. // Copyright 2021 EMQ Technologies Co., Ltd.
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package operator
  15. import (
  16. "encoding/json"
  17. "errors"
  18. "fmt"
  19. "github.com/lf-edge/ekuiper/internal/conf"
  20. "github.com/lf-edge/ekuiper/internal/topo/context"
  21. "github.com/lf-edge/ekuiper/internal/xsql"
  22. "github.com/lf-edge/ekuiper/pkg/ast"
  23. "github.com/lf-edge/ekuiper/pkg/cast"
  24. "reflect"
  25. "strings"
  26. "testing"
  27. )
  28. func TestLeftJoinPlan_Apply(t *testing.T) {
  29. var tests = []struct {
  30. sql string
  31. data xsql.WindowTuplesSet
  32. result interface{}
  33. }{
  34. { //0
  35. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  36. data: xsql.WindowTuplesSet{
  37. Content: []xsql.WindowTuples{
  38. {
  39. Emitter: "src1",
  40. Tuples: []xsql.Tuple{
  41. {
  42. Emitter: "src1",
  43. Message: xsql.Message{"id1": 1, "f1": "v1"},
  44. }, {
  45. Emitter: "src1",
  46. Message: xsql.Message{"id1": 2, "f1": "v2"},
  47. }, {
  48. Emitter: "src1",
  49. Message: xsql.Message{"id1": 3, "f1": "v3"},
  50. },
  51. },
  52. },
  53. {
  54. Emitter: "src2",
  55. Tuples: []xsql.Tuple{
  56. {
  57. Emitter: "src2",
  58. Message: xsql.Message{"id2": 1, "f2": "w1"},
  59. }, {
  60. Emitter: "src2",
  61. Message: xsql.Message{"id2": 2, "f2": "w2"},
  62. }, {
  63. Emitter: "src2",
  64. Message: xsql.Message{"id2": 4, "f2": "w3"},
  65. },
  66. },
  67. },
  68. },
  69. WindowRange: &xsql.WindowRange{
  70. WindowStart: 1541152486013,
  71. WindowEnd: 1541152487013,
  72. },
  73. },
  74. result: &xsql.JoinTupleSets{
  75. Content: []xsql.JoinTuple{
  76. {
  77. Tuples: []xsql.Tuple{
  78. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  79. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  80. },
  81. },
  82. {
  83. Tuples: []xsql.Tuple{
  84. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  85. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  86. },
  87. },
  88. {
  89. Tuples: []xsql.Tuple{
  90. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  91. },
  92. },
  93. },
  94. WindowRange: &xsql.WindowRange{
  95. WindowStart: 1541152486013,
  96. WindowEnd: 1541152487013,
  97. },
  98. },
  99. },
  100. { // 1
  101. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  102. data: xsql.WindowTuplesSet{
  103. Content: []xsql.WindowTuples{
  104. {
  105. Emitter: "src1",
  106. Tuples: []xsql.Tuple{
  107. {
  108. Emitter: "src1",
  109. Message: xsql.Message{"id1": 1, "f1": "v1"},
  110. }, {
  111. Emitter: "src1",
  112. Message: xsql.Message{"id1": 2, "f1": "v2"},
  113. }, {
  114. Emitter: "src1",
  115. Message: xsql.Message{"id1": 3, "f1": "v3"},
  116. },
  117. },
  118. },
  119. {
  120. Emitter: "src2",
  121. Tuples: []xsql.Tuple{
  122. {
  123. Emitter: "src2",
  124. Message: xsql.Message{"id2": 1, "f2": "w1"},
  125. }, {
  126. Emitter: "src2",
  127. Message: xsql.Message{"f2": "w2"},
  128. }, {
  129. Emitter: "src2",
  130. Message: xsql.Message{"id2": 4, "f2": "w3"},
  131. },
  132. },
  133. },
  134. },
  135. },
  136. result: &xsql.JoinTupleSets{
  137. Content: []xsql.JoinTuple{
  138. {
  139. Tuples: []xsql.Tuple{
  140. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  141. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  142. },
  143. },
  144. {
  145. Tuples: []xsql.Tuple{
  146. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  147. },
  148. },
  149. {
  150. Tuples: []xsql.Tuple{
  151. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  152. },
  153. },
  154. },
  155. },
  156. },
  157. { // 2
  158. sql: "SELECT id1 FROM src1 left join src2 on src1.ts = src2.ts",
  159. data: xsql.WindowTuplesSet{
  160. Content: []xsql.WindowTuples{
  161. {
  162. Emitter: "src1",
  163. Tuples: []xsql.Tuple{
  164. {
  165. Emitter: "src1",
  166. Message: xsql.Message{"id1": 1, "f1": "v1", "ts": cast.TimeFromUnixMilli(1568854515000)},
  167. }, {
  168. Emitter: "src1",
  169. Message: xsql.Message{"id1": 2, "f1": "v2", "ts": cast.TimeFromUnixMilli(1568854525000)},
  170. }, {
  171. Emitter: "src1",
  172. Message: xsql.Message{"id1": 3, "f1": "v3", "ts": cast.TimeFromUnixMilli(1568854535000)},
  173. },
  174. },
  175. },
  176. {
  177. Emitter: "src2",
  178. Tuples: []xsql.Tuple{
  179. {
  180. Emitter: "src2",
  181. Message: xsql.Message{"id2": 1, "f2": "w1", "ts": cast.TimeFromUnixMilli(1568854515000)},
  182. }, {
  183. Emitter: "src2",
  184. Message: xsql.Message{"id2": 2, "f2": "w2", "ts": cast.TimeFromUnixMilli(1568854525000)},
  185. }, {
  186. Emitter: "src2",
  187. Message: xsql.Message{"id2": 4, "f2": "w3", "ts": cast.TimeFromUnixMilli(1568854545000)},
  188. },
  189. },
  190. },
  191. },
  192. },
  193. result: &xsql.JoinTupleSets{
  194. Content: []xsql.JoinTuple{
  195. {
  196. Tuples: []xsql.Tuple{
  197. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1", "ts": cast.TimeFromUnixMilli(1568854515000)}},
  198. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1", "ts": cast.TimeFromUnixMilli(1568854515000)}},
  199. },
  200. },
  201. {
  202. Tuples: []xsql.Tuple{
  203. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2", "ts": cast.TimeFromUnixMilli(1568854525000)}},
  204. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2", "ts": cast.TimeFromUnixMilli(1568854525000)}},
  205. },
  206. },
  207. {
  208. Tuples: []xsql.Tuple{
  209. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3", "ts": cast.TimeFromUnixMilli(1568854535000)}},
  210. },
  211. },
  212. },
  213. },
  214. },
  215. { // 3
  216. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  217. data: xsql.WindowTuplesSet{
  218. Content: []xsql.WindowTuples{
  219. {
  220. Emitter: "src1",
  221. Tuples: []xsql.Tuple{
  222. {
  223. Emitter: "src1",
  224. Message: xsql.Message{"id1": 1, "f1": "v1"},
  225. }, {
  226. Emitter: "src1",
  227. Message: xsql.Message{"id1": 2, "f1": "v2"},
  228. }, {
  229. Emitter: "src1",
  230. Message: xsql.Message{"id1": 3, "f1": "v3"},
  231. },
  232. },
  233. },
  234. {
  235. Emitter: "src2",
  236. Tuples: []xsql.Tuple{
  237. {
  238. Emitter: "src2",
  239. Message: xsql.Message{"id2": 4, "f2": "w1"},
  240. }, {
  241. Emitter: "src2",
  242. Message: xsql.Message{"id2": 5, "f2": "w2"},
  243. }, {
  244. Emitter: "src2",
  245. Message: xsql.Message{"id2": 6, "f2": "w3"},
  246. },
  247. },
  248. },
  249. },
  250. },
  251. result: nil,
  252. },
  253. { // 4
  254. sql: "SELECT id1 FROM src1 As s1 left join src2 as s2 on s1.id1 = s2.id2",
  255. data: xsql.WindowTuplesSet{
  256. Content: []xsql.WindowTuples{
  257. {
  258. Emitter: "s1",
  259. Tuples: []xsql.Tuple{
  260. {
  261. Emitter: "s1",
  262. Message: xsql.Message{"id1": 1, "f1": "v1"},
  263. }, {
  264. Emitter: "s1",
  265. Message: xsql.Message{"id1": 2, "f1": "v2"},
  266. }, {
  267. Emitter: "s1",
  268. Message: xsql.Message{"id1": 3, "f1": "v3"},
  269. },
  270. },
  271. },
  272. {
  273. Emitter: "s2",
  274. Tuples: []xsql.Tuple{
  275. {
  276. Emitter: "s2",
  277. Message: xsql.Message{"id2": 1, "f2": "w1"},
  278. }, {
  279. Emitter: "s2",
  280. Message: xsql.Message{"id2": 2, "f2": "w2"},
  281. }, {
  282. Emitter: "s2",
  283. Message: xsql.Message{"id2": 4, "f2": "w3"},
  284. },
  285. },
  286. },
  287. },
  288. },
  289. result: &xsql.JoinTupleSets{
  290. Content: []xsql.JoinTuple{
  291. {
  292. Tuples: []xsql.Tuple{
  293. {Emitter: "s1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  294. {Emitter: "s2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  295. },
  296. },
  297. {
  298. Tuples: []xsql.Tuple{
  299. {Emitter: "s1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  300. {Emitter: "s2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  301. },
  302. },
  303. {
  304. Tuples: []xsql.Tuple{
  305. {Emitter: "s1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  306. },
  307. },
  308. },
  309. },
  310. },
  311. { // 5
  312. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  313. data: xsql.WindowTuplesSet{
  314. Content: []xsql.WindowTuples{
  315. {
  316. Emitter: "src1",
  317. Tuples: []xsql.Tuple{
  318. {
  319. Emitter: "src1",
  320. Message: xsql.Message{"id1": 1, "f1": "v1"},
  321. },
  322. },
  323. },
  324. {
  325. Emitter: "src2",
  326. Tuples: []xsql.Tuple{
  327. {
  328. Emitter: "src2",
  329. Message: xsql.Message{"id2": 1, "f2": "w1"},
  330. }, {
  331. Emitter: "src2",
  332. Message: xsql.Message{"id2": 1, "f2": "w2"},
  333. },
  334. },
  335. },
  336. },
  337. },
  338. result: &xsql.JoinTupleSets{
  339. Content: []xsql.JoinTuple{
  340. {
  341. Tuples: []xsql.Tuple{
  342. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  343. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  344. },
  345. },
  346. {
  347. Tuples: []xsql.Tuple{
  348. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}}, {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w2"}},
  349. },
  350. },
  351. },
  352. },
  353. },
  354. { // 6
  355. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  356. data: xsql.WindowTuplesSet{
  357. Content: []xsql.WindowTuples{
  358. {
  359. Emitter: "src1",
  360. Tuples: []xsql.Tuple{
  361. {
  362. Emitter: "src1",
  363. Message: xsql.Message{"id1": 1, "f1": "v1"},
  364. }, {
  365. Emitter: "src1",
  366. Message: xsql.Message{"id1": 2, "f1": "v2"},
  367. }, {
  368. Emitter: "src1",
  369. Message: xsql.Message{"id1": 3, "f1": "v3"},
  370. },
  371. },
  372. },
  373. {
  374. Emitter: "src2",
  375. Tuples: []xsql.Tuple{},
  376. },
  377. },
  378. },
  379. result: &xsql.JoinTupleSets{
  380. Content: []xsql.JoinTuple{
  381. {
  382. Tuples: []xsql.Tuple{
  383. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  384. },
  385. },
  386. {
  387. Tuples: []xsql.Tuple{
  388. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  389. },
  390. },
  391. {
  392. Tuples: []xsql.Tuple{
  393. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  394. },
  395. },
  396. },
  397. },
  398. },
  399. {
  400. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  401. data: xsql.WindowTuplesSet{
  402. Content: []xsql.WindowTuples{
  403. {
  404. Emitter: "src1",
  405. Tuples: []xsql.Tuple{
  406. {
  407. Emitter: "src1",
  408. Message: xsql.Message{"id1": 1, "f1": "v1"},
  409. }, {
  410. Emitter: "src1",
  411. Message: xsql.Message{"id1": 2, "f1": "v2"},
  412. }, {
  413. Emitter: "src1",
  414. Message: xsql.Message{"id1": 3, "f1": "v3"},
  415. },
  416. },
  417. },
  418. {
  419. Emitter: "src2",
  420. Tuples: nil,
  421. },
  422. },
  423. },
  424. result: &xsql.JoinTupleSets{
  425. Content: []xsql.JoinTuple{
  426. {
  427. Tuples: []xsql.Tuple{
  428. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  429. },
  430. },
  431. {
  432. Tuples: []xsql.Tuple{
  433. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  434. },
  435. },
  436. {
  437. Tuples: []xsql.Tuple{
  438. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  439. },
  440. },
  441. },
  442. },
  443. },
  444. {
  445. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  446. data: xsql.WindowTuplesSet{
  447. Content: []xsql.WindowTuples{
  448. {
  449. Emitter: "src1",
  450. Tuples: []xsql.Tuple{},
  451. },
  452. {
  453. Emitter: "src2",
  454. Tuples: []xsql.Tuple{
  455. {
  456. Emitter: "src2",
  457. Message: xsql.Message{"id2": 1, "f2": "w1"},
  458. }, {
  459. Emitter: "src2",
  460. Message: xsql.Message{"id2": 1, "f2": "w2"},
  461. },
  462. },
  463. },
  464. },
  465. },
  466. result: nil,
  467. },
  468. {
  469. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  470. data: xsql.WindowTuplesSet{
  471. Content: []xsql.WindowTuples{
  472. {
  473. Emitter: "src1",
  474. Tuples: nil,
  475. },
  476. {
  477. Emitter: "src2",
  478. Tuples: []xsql.Tuple{
  479. {
  480. Emitter: "src2",
  481. Message: xsql.Message{"id2": 1, "f2": "w1"},
  482. }, {
  483. Emitter: "src2",
  484. Message: xsql.Message{"id2": 1, "f2": "w2"},
  485. },
  486. },
  487. },
  488. },
  489. },
  490. result: nil,
  491. },
  492. {
  493. sql: "SELECT id1 FROM src1 left join src2 on src1.id1*2 = src2.id2",
  494. data: xsql.WindowTuplesSet{
  495. Content: []xsql.WindowTuples{
  496. {
  497. Emitter: "src1",
  498. Tuples: []xsql.Tuple{
  499. {
  500. Emitter: "src1",
  501. Message: xsql.Message{"id1": 1, "f1": "v1"},
  502. }, {
  503. Emitter: "src1",
  504. Message: xsql.Message{"id1": 2, "f1": "v2"},
  505. }, {
  506. Emitter: "src1",
  507. Message: xsql.Message{"id1": 3, "f1": "v3"},
  508. },
  509. },
  510. },
  511. {
  512. Emitter: "src2",
  513. Tuples: []xsql.Tuple{
  514. {
  515. Emitter: "src2",
  516. Message: xsql.Message{"id2": 1, "f2": "w1"},
  517. }, {
  518. Emitter: "src2",
  519. Message: xsql.Message{"id2": 2, "f2": "w2"},
  520. }, {
  521. Emitter: "src2",
  522. Message: xsql.Message{"id2": 4, "f2": "w3"},
  523. },
  524. },
  525. },
  526. },
  527. },
  528. result: &xsql.JoinTupleSets{
  529. Content: []xsql.JoinTuple{
  530. {
  531. Tuples: []xsql.Tuple{
  532. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  533. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  534. },
  535. },
  536. {
  537. Tuples: []xsql.Tuple{
  538. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  539. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  540. },
  541. },
  542. {
  543. Tuples: []xsql.Tuple{
  544. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  545. },
  546. },
  547. },
  548. },
  549. },
  550. {
  551. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2*2",
  552. data: xsql.WindowTuplesSet{
  553. Content: []xsql.WindowTuples{
  554. {
  555. Emitter: "src1",
  556. Tuples: []xsql.Tuple{
  557. {
  558. Emitter: "src1",
  559. Message: xsql.Message{"id1": 1, "f1": "v1"},
  560. }, {
  561. Emitter: "src1",
  562. Message: xsql.Message{"id1": 2, "f1": "v2"},
  563. }, {
  564. Emitter: "src1",
  565. Message: xsql.Message{"id1": 3, "f1": "v3"},
  566. },
  567. },
  568. },
  569. {
  570. Emitter: "src2",
  571. Tuples: []xsql.Tuple{
  572. {
  573. Emitter: "src2",
  574. Message: xsql.Message{"id2": 1, "f2": "w1"},
  575. }, {
  576. Emitter: "src2",
  577. Message: xsql.Message{"id2": 2, "f2": "w2"},
  578. }, {
  579. Emitter: "src2",
  580. Message: xsql.Message{"id2": 4, "f2": "w3"},
  581. },
  582. },
  583. },
  584. },
  585. },
  586. result: &xsql.JoinTupleSets{
  587. Content: []xsql.JoinTuple{
  588. {
  589. Tuples: []xsql.Tuple{
  590. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  591. },
  592. },
  593. {
  594. Tuples: []xsql.Tuple{
  595. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  596. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  597. },
  598. },
  599. {
  600. Tuples: []xsql.Tuple{
  601. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  602. },
  603. },
  604. },
  605. },
  606. },
  607. {
  608. sql: "SELECT id1 FROM src1 left join src2 on src1.f1->cid = src2.f2->cid",
  609. data: xsql.WindowTuplesSet{
  610. Content: []xsql.WindowTuples{
  611. {
  612. Emitter: "src1",
  613. Tuples: []xsql.Tuple{
  614. {
  615. Emitter: "src1",
  616. Message: xsql.Message{"id1": 1, "f1": str2Map(`{"cid" : 1, "name" : "tom1"}`)},
  617. }, {
  618. Emitter: "src1",
  619. Message: xsql.Message{"id1": 2, "f1": str2Map(`{"cid" : 2, "name" : "mike1"}`)},
  620. }, {
  621. Emitter: "src1",
  622. Message: xsql.Message{"id1": 3, "f1": str2Map(`{"cid" : 3, "name" : "alice1"}`)},
  623. },
  624. },
  625. },
  626. {
  627. Emitter: "src2",
  628. Tuples: []xsql.Tuple{
  629. {
  630. Emitter: "src2",
  631. Message: xsql.Message{"id2": 1, "f2": str2Map(`{"cid" : 1, "name" : "tom2"}`)},
  632. }, {
  633. Emitter: "src2",
  634. Message: xsql.Message{"id2": 2, "f2": str2Map(`{"cid" : 2, "name" : "mike2"}`)},
  635. }, {
  636. Emitter: "src2",
  637. Message: xsql.Message{"id2": 4, "f2": str2Map(`{"cid" : 4, "name" : "alice2"}`)},
  638. },
  639. },
  640. },
  641. },
  642. },
  643. result: &xsql.JoinTupleSets{
  644. Content: []xsql.JoinTuple{
  645. {
  646. Tuples: []xsql.Tuple{
  647. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": str2Map(`{"cid" : 1, "name" : "tom1"}`)}},
  648. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": str2Map(`{"cid" : 1, "name" : "tom2"}`)}},
  649. },
  650. },
  651. {
  652. Tuples: []xsql.Tuple{
  653. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": str2Map(`{"cid" : 2, "name" : "mike1"}`)}},
  654. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": str2Map(`{"cid" : 2, "name" : "mike2"}`)}},
  655. },
  656. },
  657. {
  658. Tuples: []xsql.Tuple{
  659. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": str2Map(`{"cid" : 3, "name" : "alice1"}`)}},
  660. },
  661. },
  662. },
  663. },
  664. },
  665. {
  666. sql: "SELECT id1, mqtt(src1.topic) AS a, mqtt(src2.topic) as b FROM src1 left join src2 on src1.id1 = src2.id2",
  667. data: xsql.WindowTuplesSet{
  668. Content: []xsql.WindowTuples{
  669. {
  670. Emitter: "src1",
  671. Tuples: []xsql.Tuple{
  672. {
  673. Emitter: "src1",
  674. Message: xsql.Message{"id1": 1, "f1": "v1"},
  675. Metadata: xsql.Metadata{"topic": "devices/type1/device001"},
  676. },
  677. },
  678. },
  679. {
  680. Emitter: "src2",
  681. Tuples: []xsql.Tuple{
  682. {
  683. Emitter: "src2",
  684. Message: xsql.Message{"id2": 1, "f2": "w1"},
  685. Metadata: xsql.Metadata{"topic": "devices/type2/device001"},
  686. },
  687. },
  688. },
  689. },
  690. },
  691. result: &xsql.JoinTupleSets{
  692. Content: []xsql.JoinTuple{
  693. {
  694. Tuples: []xsql.Tuple{
  695. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}, Metadata: xsql.Metadata{"topic": "devices/type1/device001"}},
  696. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}, Metadata: xsql.Metadata{"topic": "devices/type2/device001"}},
  697. },
  698. },
  699. },
  700. },
  701. },
  702. {
  703. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  704. data: xsql.WindowTuplesSet{
  705. Content: []xsql.WindowTuples{
  706. {
  707. Emitter: "src1",
  708. Tuples: []xsql.Tuple{
  709. {
  710. Emitter: "src1",
  711. Message: xsql.Message{"id1": 1, "f1": "v1"},
  712. }, {
  713. Emitter: "src1",
  714. Message: xsql.Message{"id1": 1, "f1": "v2"},
  715. }, {
  716. Emitter: "src1",
  717. Message: xsql.Message{"id1": 3, "f1": "v3"},
  718. }, {
  719. Emitter: "src1",
  720. Message: xsql.Message{"id1": 3, "f1": "v4"},
  721. }, {
  722. Emitter: "src1",
  723. Message: xsql.Message{"id1": 4, "f1": "v5"},
  724. },
  725. },
  726. },
  727. {
  728. Emitter: "src2",
  729. Tuples: []xsql.Tuple{
  730. {
  731. Emitter: "src2",
  732. Message: xsql.Message{"id2": 1, "f2": "w1"},
  733. }, {
  734. Emitter: "src2",
  735. Message: xsql.Message{"id2": 3, "f2": "w2"},
  736. }, {
  737. Emitter: "src2",
  738. Message: xsql.Message{"id2": 3, "f2": "w3"},
  739. },
  740. },
  741. },
  742. },
  743. },
  744. result: &xsql.JoinTupleSets{
  745. Content: []xsql.JoinTuple{
  746. {
  747. Tuples: []xsql.Tuple{
  748. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  749. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  750. },
  751. },
  752. {
  753. Tuples: []xsql.Tuple{
  754. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v2"}},
  755. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  756. },
  757. },
  758. {
  759. Tuples: []xsql.Tuple{
  760. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  761. {Emitter: "src2", Message: xsql.Message{"id2": 3, "f2": "w2"}},
  762. },
  763. },
  764. {
  765. Tuples: []xsql.Tuple{
  766. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  767. {Emitter: "src2", Message: xsql.Message{"id2": 3, "f2": "w3"}},
  768. },
  769. },
  770. {
  771. Tuples: []xsql.Tuple{
  772. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v4"}},
  773. {Emitter: "src2", Message: xsql.Message{"id2": 3, "f2": "w2"}},
  774. },
  775. },
  776. {
  777. Tuples: []xsql.Tuple{
  778. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v4"}},
  779. {Emitter: "src2", Message: xsql.Message{"id2": 3, "f2": "w3"}},
  780. },
  781. },
  782. {
  783. Tuples: []xsql.Tuple{
  784. {Emitter: "src1", Message: xsql.Message{"id1": 4, "f1": "v5"}},
  785. },
  786. },
  787. },
  788. },
  789. },
  790. {
  791. sql: "SELECT id1 FROM src1 full join src2 on src1.id1 = src2.id2",
  792. data: xsql.WindowTuplesSet{
  793. Content: []xsql.WindowTuples{
  794. {
  795. Emitter: "src1",
  796. Tuples: []xsql.Tuple{
  797. {
  798. Emitter: "src1",
  799. Message: xsql.Message{"id1": 1, "f1": "v1"},
  800. }, {
  801. Emitter: "src1",
  802. Message: xsql.Message{"id1": 2, "f1": "v2"},
  803. }, {
  804. Emitter: "src1",
  805. Message: xsql.Message{"id1": 3, "f1": "v3"},
  806. },
  807. },
  808. },
  809. {
  810. Emitter: "src2",
  811. Tuples: []xsql.Tuple{
  812. {
  813. Emitter: "src2",
  814. Message: xsql.Message{"id2": 1, "f2": "w1"},
  815. }, {
  816. Emitter: "src2",
  817. Message: xsql.Message{"id2": 2, "f2": "w2"},
  818. }, {
  819. Emitter: "src2",
  820. Message: xsql.Message{"id2": 2, "f2": "w3"},
  821. },
  822. },
  823. },
  824. },
  825. },
  826. result: &xsql.JoinTupleSets{
  827. Content: []xsql.JoinTuple{
  828. {
  829. Tuples: []xsql.Tuple{
  830. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  831. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  832. },
  833. },
  834. {
  835. Tuples: []xsql.Tuple{
  836. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  837. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  838. },
  839. },
  840. {
  841. Tuples: []xsql.Tuple{
  842. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  843. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w3"}},
  844. },
  845. },
  846. {
  847. Tuples: []xsql.Tuple{
  848. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  849. },
  850. },
  851. },
  852. },
  853. },
  854. }
  855. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  856. contextLogger := conf.Log.WithField("rule", "TestLeftJoinPlan_Apply")
  857. ctx := context.WithValue(context.Background(), context.LoggerKey, contextLogger)
  858. for i, tt := range tests {
  859. stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
  860. if err != nil {
  861. t.Errorf("statement parse error %s", err)
  862. break
  863. }
  864. if table, ok := stmt.Sources[0].(*ast.Table); !ok {
  865. t.Errorf("statement source is not a table")
  866. } else {
  867. fv, afv := xsql.NewFunctionValuersForOp(nil, xsql.FuncRegisters)
  868. pp := &JoinOp{Joins: stmt.Joins, From: table}
  869. result := pp.Apply(ctx, tt.data, fv, afv)
  870. if !reflect.DeepEqual(tt.result, result) {
  871. t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
  872. }
  873. }
  874. }
  875. }
  876. func TestInnerJoinPlan_Apply(t *testing.T) {
  877. var tests = []struct {
  878. sql string
  879. data xsql.WindowTuplesSet
  880. result interface{}
  881. }{
  882. {
  883. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  884. data: xsql.WindowTuplesSet{
  885. Content: []xsql.WindowTuples{
  886. {
  887. Emitter: "src1",
  888. Tuples: []xsql.Tuple{
  889. {
  890. Emitter: "src1",
  891. Message: xsql.Message{"id1": 1, "f1": "v1"},
  892. }, {
  893. Emitter: "src1",
  894. Message: xsql.Message{"id1": 2, "f1": "v2"},
  895. }, {
  896. Emitter: "src1",
  897. Message: xsql.Message{"id1": 3, "f1": "v3"},
  898. },
  899. },
  900. },
  901. {
  902. Emitter: "src2",
  903. Tuples: []xsql.Tuple{
  904. {
  905. Emitter: "src2",
  906. Message: xsql.Message{"id2": 1, "f2": "w1"},
  907. }, {
  908. Emitter: "src2",
  909. Message: xsql.Message{"id2": 2, "f2": "w2"},
  910. }, {
  911. Emitter: "src2",
  912. Message: xsql.Message{"id2": 4, "f2": "w3"},
  913. },
  914. },
  915. },
  916. },
  917. },
  918. result: &xsql.JoinTupleSets{
  919. Content: []xsql.JoinTuple{
  920. {
  921. Tuples: []xsql.Tuple{
  922. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  923. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  924. },
  925. },
  926. {
  927. Tuples: []xsql.Tuple{
  928. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  929. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  930. },
  931. },
  932. },
  933. },
  934. },
  935. {
  936. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  937. data: xsql.WindowTuplesSet{
  938. Content: []xsql.WindowTuples{
  939. {
  940. Emitter: "src1",
  941. Tuples: []xsql.Tuple{
  942. {
  943. Emitter: "src1",
  944. Message: xsql.Message{"id1": 1, "f1": "v1"},
  945. }, {
  946. Emitter: "src1",
  947. Message: xsql.Message{"id1": 2, "f1": "v2"},
  948. }, {
  949. Emitter: "src1",
  950. Message: xsql.Message{"id1": 3, "f1": "v3"},
  951. },
  952. },
  953. },
  954. {
  955. Emitter: "src2",
  956. Tuples: []xsql.Tuple{
  957. {
  958. Emitter: "src2",
  959. Message: xsql.Message{"id2": 1, "f2": "w1"},
  960. }, {
  961. Emitter: "src2",
  962. Message: xsql.Message{"f2": "w2"},
  963. }, {
  964. Emitter: "src2",
  965. Message: xsql.Message{"id2": 4, "f2": "w3"},
  966. },
  967. },
  968. },
  969. },
  970. },
  971. result: &xsql.JoinTupleSets{
  972. Content: []xsql.JoinTuple{
  973. {
  974. Tuples: []xsql.Tuple{
  975. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  976. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  977. },
  978. },
  979. },
  980. },
  981. },
  982. {
  983. sql: "SELECT id1 FROM src1 As s1 inner join src2 as s2 on s1.id1 = s2.id2",
  984. data: xsql.WindowTuplesSet{
  985. Content: []xsql.WindowTuples{
  986. {
  987. Emitter: "s1",
  988. Tuples: []xsql.Tuple{
  989. {
  990. Emitter: "s1",
  991. Message: xsql.Message{"id1": 1, "f1": "v1"},
  992. }, {
  993. Emitter: "s1",
  994. Message: xsql.Message{"id1": 2, "f1": "v2"},
  995. }, {
  996. Emitter: "s1",
  997. Message: xsql.Message{"id1": 3, "f1": "v3"},
  998. },
  999. },
  1000. },
  1001. {
  1002. Emitter: "s2",
  1003. Tuples: []xsql.Tuple{
  1004. {
  1005. Emitter: "s2",
  1006. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1007. }, {
  1008. Emitter: "s2",
  1009. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1010. }, {
  1011. Emitter: "s2",
  1012. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1013. },
  1014. },
  1015. },
  1016. },
  1017. },
  1018. result: &xsql.JoinTupleSets{
  1019. Content: []xsql.JoinTuple{
  1020. {
  1021. Tuples: []xsql.Tuple{
  1022. {Emitter: "s1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1023. {Emitter: "s2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1024. },
  1025. },
  1026. {
  1027. Tuples: []xsql.Tuple{
  1028. {Emitter: "s1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1029. {Emitter: "s2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1030. },
  1031. },
  1032. },
  1033. },
  1034. },
  1035. {
  1036. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  1037. data: xsql.WindowTuplesSet{
  1038. Content: []xsql.WindowTuples{
  1039. {
  1040. Emitter: "src1",
  1041. Tuples: []xsql.Tuple{
  1042. {
  1043. Emitter: "src1",
  1044. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1045. },
  1046. },
  1047. },
  1048. {
  1049. Emitter: "src2",
  1050. Tuples: []xsql.Tuple{
  1051. {
  1052. Emitter: "src2",
  1053. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1054. }, {
  1055. Emitter: "src2",
  1056. Message: xsql.Message{"id2": 1, "f2": "w2"},
  1057. },
  1058. },
  1059. },
  1060. },
  1061. },
  1062. result: &xsql.JoinTupleSets{
  1063. Content: []xsql.JoinTuple{
  1064. {
  1065. Tuples: []xsql.Tuple{
  1066. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1067. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1068. },
  1069. },
  1070. {
  1071. Tuples: []xsql.Tuple{
  1072. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1073. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w2"}},
  1074. },
  1075. },
  1076. },
  1077. },
  1078. },
  1079. {
  1080. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  1081. data: xsql.WindowTuplesSet{
  1082. Content: []xsql.WindowTuples{
  1083. {
  1084. Emitter: "src1",
  1085. Tuples: []xsql.Tuple{
  1086. {
  1087. Emitter: "src1",
  1088. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1089. }, {
  1090. Emitter: "src1",
  1091. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1092. }, {
  1093. Emitter: "src1",
  1094. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1095. },
  1096. },
  1097. },
  1098. {
  1099. Emitter: "src2",
  1100. Tuples: []xsql.Tuple{},
  1101. },
  1102. },
  1103. },
  1104. result: nil,
  1105. },
  1106. {
  1107. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  1108. data: xsql.WindowTuplesSet{
  1109. Content: []xsql.WindowTuples{
  1110. {
  1111. Emitter: "src1",
  1112. Tuples: []xsql.Tuple{
  1113. {
  1114. Emitter: "src1",
  1115. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1116. }, {
  1117. Emitter: "src1",
  1118. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1119. }, {
  1120. Emitter: "src1",
  1121. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1122. },
  1123. },
  1124. },
  1125. {
  1126. Emitter: "src2",
  1127. Tuples: nil,
  1128. },
  1129. },
  1130. },
  1131. result: nil,
  1132. },
  1133. {
  1134. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  1135. data: xsql.WindowTuplesSet{
  1136. Content: []xsql.WindowTuples{
  1137. {
  1138. Emitter: "src1",
  1139. Tuples: []xsql.Tuple{},
  1140. },
  1141. {
  1142. Emitter: "src2",
  1143. Tuples: []xsql.Tuple{
  1144. {
  1145. Emitter: "src2",
  1146. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1147. }, {
  1148. Emitter: "src2",
  1149. Message: xsql.Message{"id2": 1, "f2": "w2"},
  1150. },
  1151. },
  1152. },
  1153. },
  1154. },
  1155. result: nil,
  1156. },
  1157. {
  1158. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  1159. data: xsql.WindowTuplesSet{
  1160. Content: []xsql.WindowTuples{
  1161. {
  1162. Emitter: "src1",
  1163. Tuples: nil,
  1164. },
  1165. {
  1166. Emitter: "src2",
  1167. Tuples: []xsql.Tuple{
  1168. {
  1169. Emitter: "src2",
  1170. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1171. }, {
  1172. Emitter: "src2",
  1173. Message: xsql.Message{"id2": 1, "f2": "w2"},
  1174. },
  1175. },
  1176. },
  1177. },
  1178. },
  1179. result: nil,
  1180. },
  1181. {
  1182. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1*2 = src2.id2",
  1183. data: xsql.WindowTuplesSet{
  1184. Content: []xsql.WindowTuples{
  1185. {
  1186. Emitter: "src1",
  1187. Tuples: []xsql.Tuple{
  1188. {
  1189. Emitter: "src1",
  1190. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1191. }, {
  1192. Emitter: "src1",
  1193. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1194. }, {
  1195. Emitter: "src1",
  1196. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1197. },
  1198. },
  1199. },
  1200. {
  1201. Emitter: "src2",
  1202. Tuples: []xsql.Tuple{
  1203. {
  1204. Emitter: "src2",
  1205. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1206. }, {
  1207. Emitter: "src2",
  1208. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1209. }, {
  1210. Emitter: "src2",
  1211. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1212. },
  1213. },
  1214. },
  1215. },
  1216. },
  1217. result: &xsql.JoinTupleSets{
  1218. Content: []xsql.JoinTuple{
  1219. {
  1220. Tuples: []xsql.Tuple{
  1221. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1222. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1223. },
  1224. },
  1225. {
  1226. Tuples: []xsql.Tuple{
  1227. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1228. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1229. },
  1230. },
  1231. },
  1232. },
  1233. },
  1234. {
  1235. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2*2",
  1236. data: xsql.WindowTuplesSet{
  1237. Content: []xsql.WindowTuples{
  1238. {
  1239. Emitter: "src1",
  1240. Tuples: []xsql.Tuple{
  1241. {
  1242. Emitter: "src1",
  1243. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1244. }, {
  1245. Emitter: "src1",
  1246. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1247. }, {
  1248. Emitter: "src1",
  1249. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1250. },
  1251. },
  1252. },
  1253. {
  1254. Emitter: "src2",
  1255. Tuples: []xsql.Tuple{
  1256. {
  1257. Emitter: "src2",
  1258. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1259. }, {
  1260. Emitter: "src2",
  1261. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1262. }, {
  1263. Emitter: "src2",
  1264. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1265. },
  1266. },
  1267. },
  1268. },
  1269. },
  1270. result: &xsql.JoinTupleSets{
  1271. Content: []xsql.JoinTuple{
  1272. {
  1273. Tuples: []xsql.Tuple{
  1274. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1275. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1276. },
  1277. },
  1278. },
  1279. },
  1280. },
  1281. {
  1282. sql: "SELECT id1 FROM src1 inner join src2 on src1.f1->cid = src2.f2->cid",
  1283. data: xsql.WindowTuplesSet{
  1284. Content: []xsql.WindowTuples{
  1285. {
  1286. Emitter: "src1",
  1287. Tuples: []xsql.Tuple{
  1288. {
  1289. Emitter: "src1",
  1290. Message: xsql.Message{"id1": 1, "f1": str2Map(`{"cid" : 1, "name" : "tom1"}`)},
  1291. }, {
  1292. Emitter: "src1",
  1293. Message: xsql.Message{"id1": 2, "f1": str2Map(`{"cid" : 2, "name" : "mike1"}`)},
  1294. }, {
  1295. Emitter: "src1",
  1296. Message: xsql.Message{"id1": 3, "f1": str2Map(`{"cid" : 3, "name" : "alice1"}`)},
  1297. },
  1298. },
  1299. },
  1300. {
  1301. Emitter: "src2",
  1302. Tuples: []xsql.Tuple{
  1303. {
  1304. Emitter: "src2",
  1305. Message: xsql.Message{"id2": 1, "f2": str2Map(`{"cid" : 1, "name" : "tom2"}`)},
  1306. }, {
  1307. Emitter: "src2",
  1308. Message: xsql.Message{"id2": 2, "f2": str2Map(`{"cid" : 2, "name" : "mike2"}`)},
  1309. }, {
  1310. Emitter: "src2",
  1311. Message: xsql.Message{"id2": 4, "f2": str2Map(`{"cid" : 4, "name" : "alice2"}`)},
  1312. },
  1313. },
  1314. },
  1315. },
  1316. },
  1317. result: &xsql.JoinTupleSets{
  1318. Content: []xsql.JoinTuple{
  1319. {
  1320. Tuples: []xsql.Tuple{
  1321. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": str2Map(`{"cid" : 1, "name" : "tom1"}`)}},
  1322. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": str2Map(`{"cid" : 1, "name" : "tom2"}`)}},
  1323. },
  1324. },
  1325. {
  1326. Tuples: []xsql.Tuple{
  1327. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": str2Map(`{"cid" : 2, "name" : "mike1"}`)}},
  1328. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": str2Map(`{"cid" : 2, "name" : "mike2"}`)}},
  1329. },
  1330. },
  1331. },
  1332. },
  1333. },
  1334. {
  1335. sql: "SELECT id1 FROM src1 As s1 inner join src2 as s2 on s1.id1 = s2.id2",
  1336. data: xsql.WindowTuplesSet{
  1337. Content: []xsql.WindowTuples{
  1338. {
  1339. Emitter: "s1",
  1340. Tuples: []xsql.Tuple{
  1341. {
  1342. Emitter: "s1",
  1343. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1344. }, {
  1345. Emitter: "s1",
  1346. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1347. }, {
  1348. Emitter: "s1",
  1349. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1350. },
  1351. },
  1352. },
  1353. {
  1354. Emitter: "s2",
  1355. Tuples: []xsql.Tuple{
  1356. {
  1357. Emitter: "s2",
  1358. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1359. }, {
  1360. Emitter: "s2",
  1361. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1362. }, {
  1363. Emitter: "s2",
  1364. Message: xsql.Message{"id2": 2, "f2": "w3"},
  1365. },
  1366. },
  1367. },
  1368. },
  1369. },
  1370. result: &xsql.JoinTupleSets{
  1371. Content: []xsql.JoinTuple{
  1372. {
  1373. Tuples: []xsql.Tuple{
  1374. {Emitter: "s1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1375. {Emitter: "s2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1376. },
  1377. },
  1378. {
  1379. Tuples: []xsql.Tuple{
  1380. {Emitter: "s1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1381. {Emitter: "s2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1382. },
  1383. },
  1384. {
  1385. Tuples: []xsql.Tuple{
  1386. {Emitter: "s1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1387. {Emitter: "s2", Message: xsql.Message{"id2": 2, "f2": "w3"}},
  1388. },
  1389. },
  1390. },
  1391. },
  1392. },
  1393. }
  1394. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  1395. contextLogger := conf.Log.WithField("rule", "TestInnerJoinPlan_Apply")
  1396. ctx := context.WithValue(context.Background(), context.LoggerKey, contextLogger)
  1397. for i, tt := range tests {
  1398. stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
  1399. if err != nil {
  1400. t.Errorf("statement parse error %s", err)
  1401. break
  1402. }
  1403. if table, ok := stmt.Sources[0].(*ast.Table); !ok {
  1404. t.Errorf("statement source is not a table")
  1405. } else {
  1406. fv, afv := xsql.NewFunctionValuersForOp(nil, xsql.FuncRegisters)
  1407. pp := &JoinOp{Joins: stmt.Joins, From: table}
  1408. result := pp.Apply(ctx, tt.data, fv, afv)
  1409. if !reflect.DeepEqual(tt.result, result) {
  1410. t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
  1411. }
  1412. }
  1413. }
  1414. }
  1415. func TestRightJoinPlan_Apply(t *testing.T) {
  1416. var tests = []struct {
  1417. sql string
  1418. data xsql.WindowTuplesSet
  1419. result interface{}
  1420. }{
  1421. {
  1422. sql: "SELECT id1 FROM src1 right join src2 on src1.id1 = src2.id2",
  1423. data: xsql.WindowTuplesSet{
  1424. Content: []xsql.WindowTuples{
  1425. {
  1426. Emitter: "src1",
  1427. Tuples: []xsql.Tuple{
  1428. {
  1429. Emitter: "src1",
  1430. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1431. }, {
  1432. Emitter: "src1",
  1433. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1434. }, {
  1435. Emitter: "src1",
  1436. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1437. },
  1438. },
  1439. },
  1440. {
  1441. Emitter: "src2",
  1442. Tuples: []xsql.Tuple{
  1443. {
  1444. Emitter: "src2",
  1445. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1446. }, {
  1447. Emitter: "src2",
  1448. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1449. }, {
  1450. Emitter: "src2",
  1451. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1452. },
  1453. },
  1454. },
  1455. },
  1456. },
  1457. result: &xsql.JoinTupleSets{
  1458. Content: []xsql.JoinTuple{
  1459. {
  1460. Tuples: []xsql.Tuple{
  1461. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1462. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1463. },
  1464. },
  1465. {
  1466. Tuples: []xsql.Tuple{
  1467. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1468. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1469. },
  1470. },
  1471. {
  1472. Tuples: []xsql.Tuple{
  1473. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1474. },
  1475. },
  1476. },
  1477. },
  1478. },
  1479. {
  1480. sql: "SELECT id1 FROM src1 right join src2 on src1.id1 = src2.id2",
  1481. data: xsql.WindowTuplesSet{
  1482. Content: []xsql.WindowTuples{
  1483. {
  1484. Emitter: "src1",
  1485. Tuples: []xsql.Tuple{
  1486. {
  1487. Emitter: "src1",
  1488. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1489. }, {
  1490. Emitter: "src1",
  1491. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1492. }, {
  1493. Emitter: "src1",
  1494. Message: xsql.Message{"id1": 1, "f1": "v3"},
  1495. },
  1496. },
  1497. },
  1498. {
  1499. Emitter: "src2",
  1500. Tuples: []xsql.Tuple{
  1501. {
  1502. Emitter: "src2",
  1503. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1504. }, {
  1505. Emitter: "src2",
  1506. Message: xsql.Message{"f2": "w2"},
  1507. }, {
  1508. Emitter: "src2",
  1509. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1510. },
  1511. },
  1512. },
  1513. },
  1514. },
  1515. result: &xsql.JoinTupleSets{
  1516. Content: []xsql.JoinTuple{
  1517. {
  1518. Tuples: []xsql.Tuple{
  1519. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1520. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1521. },
  1522. },
  1523. {
  1524. Tuples: []xsql.Tuple{
  1525. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1526. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v3"}},
  1527. },
  1528. },
  1529. {
  1530. Tuples: []xsql.Tuple{
  1531. {Emitter: "src2", Message: xsql.Message{"f2": "w2"}},
  1532. },
  1533. },
  1534. {
  1535. Tuples: []xsql.Tuple{
  1536. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1537. },
  1538. },
  1539. },
  1540. },
  1541. },
  1542. {
  1543. sql: "SELECT id1 FROM src1 right join src2 on src1.id1 = src2.id2",
  1544. data: xsql.WindowTuplesSet{
  1545. Content: []xsql.WindowTuples{
  1546. {
  1547. Emitter: "src1",
  1548. Tuples: []xsql.Tuple{
  1549. {
  1550. Emitter: "src1",
  1551. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1552. },
  1553. },
  1554. },
  1555. {
  1556. Emitter: "src2",
  1557. Tuples: []xsql.Tuple{
  1558. {
  1559. Emitter: "src2",
  1560. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1561. }, {
  1562. Emitter: "src2",
  1563. Message: xsql.Message{"id2": 1, "f2": "w2"},
  1564. },
  1565. },
  1566. },
  1567. },
  1568. },
  1569. result: &xsql.JoinTupleSets{
  1570. Content: []xsql.JoinTuple{
  1571. {
  1572. Tuples: []xsql.Tuple{
  1573. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1574. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1575. },
  1576. },
  1577. {
  1578. Tuples: []xsql.Tuple{
  1579. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w2"}},
  1580. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1581. },
  1582. },
  1583. },
  1584. },
  1585. },
  1586. }
  1587. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  1588. contextLogger := conf.Log.WithField("rule", "TestRightJoinPlan_Apply")
  1589. ctx := context.WithValue(context.Background(), context.LoggerKey, contextLogger)
  1590. for i, tt := range tests {
  1591. stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
  1592. if err != nil {
  1593. t.Errorf("statement parse error %s", err)
  1594. break
  1595. }
  1596. if table, ok := stmt.Sources[0].(*ast.Table); !ok {
  1597. t.Errorf("statement source is not a table")
  1598. } else {
  1599. fv, afv := xsql.NewFunctionValuersForOp(nil, xsql.FuncRegisters)
  1600. pp := &JoinOp{Joins: stmt.Joins, From: table}
  1601. result := pp.Apply(ctx, tt.data, fv, afv)
  1602. if !reflect.DeepEqual(tt.result, result) {
  1603. t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
  1604. }
  1605. }
  1606. }
  1607. }
  1608. func TestFullJoinPlan_Apply(t *testing.T) {
  1609. var tests = []struct {
  1610. sql string
  1611. data xsql.WindowTuplesSet
  1612. result interface{}
  1613. }{
  1614. {
  1615. sql: "SELECT id1 FROM src1 full join src2 on src1.id1 = src2.id2",
  1616. data: xsql.WindowTuplesSet{
  1617. Content: []xsql.WindowTuples{
  1618. {
  1619. Emitter: "src1",
  1620. Tuples: []xsql.Tuple{
  1621. {
  1622. Emitter: "src1",
  1623. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1624. }, {
  1625. Emitter: "src1",
  1626. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1627. }, {
  1628. Emitter: "src1",
  1629. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1630. },
  1631. },
  1632. },
  1633. {
  1634. Emitter: "src2",
  1635. Tuples: []xsql.Tuple{
  1636. {
  1637. Emitter: "src2",
  1638. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1639. }, {
  1640. Emitter: "src2",
  1641. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1642. }, {
  1643. Emitter: "src2",
  1644. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1645. }, {
  1646. Emitter: "src2",
  1647. Message: xsql.Message{"id2": 2, "f2": "w4"},
  1648. },
  1649. },
  1650. },
  1651. },
  1652. },
  1653. result: &xsql.JoinTupleSets{
  1654. Content: []xsql.JoinTuple{
  1655. {
  1656. Tuples: []xsql.Tuple{
  1657. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1658. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1659. },
  1660. },
  1661. {
  1662. Tuples: []xsql.Tuple{
  1663. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1664. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1665. },
  1666. },
  1667. {
  1668. Tuples: []xsql.Tuple{
  1669. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}}, {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w4"}},
  1670. },
  1671. },
  1672. {
  1673. Tuples: []xsql.Tuple{
  1674. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  1675. },
  1676. },
  1677. {
  1678. Tuples: []xsql.Tuple{
  1679. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1680. },
  1681. },
  1682. },
  1683. },
  1684. },
  1685. {
  1686. sql: "SELECT id1 FROM src1 full join src2 on src1.id1 = src2.id2",
  1687. data: xsql.WindowTuplesSet{
  1688. Content: []xsql.WindowTuples{
  1689. {
  1690. Emitter: "src1",
  1691. Tuples: []xsql.Tuple{
  1692. {
  1693. Emitter: "src1",
  1694. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1695. }, {
  1696. Emitter: "src1",
  1697. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1698. }, {
  1699. Emitter: "src1",
  1700. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1701. },
  1702. },
  1703. },
  1704. {
  1705. Emitter: "src2",
  1706. Tuples: []xsql.Tuple{
  1707. {
  1708. Emitter: "src2",
  1709. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1710. }, {
  1711. Emitter: "src2",
  1712. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1713. }, {
  1714. Emitter: "src2",
  1715. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1716. },
  1717. },
  1718. },
  1719. },
  1720. },
  1721. result: &xsql.JoinTupleSets{
  1722. Content: []xsql.JoinTuple{
  1723. {
  1724. Tuples: []xsql.Tuple{
  1725. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1726. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1727. },
  1728. },
  1729. {
  1730. Tuples: []xsql.Tuple{
  1731. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1732. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1733. },
  1734. },
  1735. {
  1736. Tuples: []xsql.Tuple{
  1737. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  1738. },
  1739. },
  1740. {
  1741. Tuples: []xsql.Tuple{
  1742. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1743. },
  1744. },
  1745. },
  1746. },
  1747. },
  1748. {
  1749. sql: "SELECT id1 FROM src1 full join src2 on src1.id1 = src2.id2",
  1750. data: xsql.WindowTuplesSet{
  1751. Content: []xsql.WindowTuples{
  1752. {
  1753. Emitter: "src1",
  1754. Tuples: []xsql.Tuple{
  1755. {
  1756. Emitter: "src1",
  1757. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1758. }, {
  1759. Emitter: "src1",
  1760. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1761. }, {
  1762. Emitter: "src1",
  1763. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1764. },
  1765. },
  1766. },
  1767. {
  1768. Emitter: "src2",
  1769. Tuples: []xsql.Tuple{},
  1770. },
  1771. },
  1772. },
  1773. result: &xsql.JoinTupleSets{
  1774. Content: []xsql.JoinTuple{
  1775. {
  1776. Tuples: []xsql.Tuple{
  1777. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1778. },
  1779. },
  1780. {
  1781. Tuples: []xsql.Tuple{
  1782. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1783. },
  1784. },
  1785. {
  1786. Tuples: []xsql.Tuple{
  1787. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  1788. },
  1789. },
  1790. },
  1791. },
  1792. },
  1793. {
  1794. sql: "SELECT id1 FROM src1 full join src2 on src1.id1 = src2.id2",
  1795. data: xsql.WindowTuplesSet{
  1796. Content: []xsql.WindowTuples{
  1797. {
  1798. Emitter: "src1",
  1799. Tuples: []xsql.Tuple{},
  1800. },
  1801. {
  1802. Emitter: "src2",
  1803. Tuples: []xsql.Tuple{
  1804. {
  1805. Emitter: "src2",
  1806. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1807. }, {
  1808. Emitter: "src2",
  1809. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1810. }, {
  1811. Emitter: "src2",
  1812. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1813. },
  1814. },
  1815. },
  1816. },
  1817. },
  1818. result: &xsql.JoinTupleSets{
  1819. Content: []xsql.JoinTuple{
  1820. {
  1821. Tuples: []xsql.Tuple{
  1822. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1823. },
  1824. },
  1825. {
  1826. Tuples: []xsql.Tuple{
  1827. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1828. },
  1829. },
  1830. {
  1831. Tuples: []xsql.Tuple{
  1832. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1833. },
  1834. },
  1835. },
  1836. },
  1837. },
  1838. }
  1839. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  1840. contextLogger := conf.Log.WithField("rule", "TestFullJoinPlan_Apply")
  1841. ctx := context.WithValue(context.Background(), context.LoggerKey, contextLogger)
  1842. for i, tt := range tests {
  1843. stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
  1844. if err != nil {
  1845. t.Errorf("statement parse error %s", err)
  1846. break
  1847. }
  1848. if table, ok := stmt.Sources[0].(*ast.Table); !ok {
  1849. t.Errorf("statement source is not a table")
  1850. } else {
  1851. fv, afv := xsql.NewFunctionValuersForOp(nil, xsql.FuncRegisters)
  1852. pp := &JoinOp{Joins: stmt.Joins, From: table}
  1853. result := pp.Apply(ctx, tt.data, fv, afv)
  1854. if !reflect.DeepEqual(tt.result, result) {
  1855. t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
  1856. }
  1857. }
  1858. }
  1859. }
  1860. func TestCrossJoinPlan_Apply(t *testing.T) {
  1861. var tests = []struct {
  1862. sql string
  1863. data xsql.WindowTuplesSet
  1864. result interface{}
  1865. }{
  1866. {
  1867. sql: "SELECT id1 FROM src1 cross join src2",
  1868. data: xsql.WindowTuplesSet{
  1869. Content: []xsql.WindowTuples{
  1870. {
  1871. Emitter: "src1",
  1872. Tuples: []xsql.Tuple{
  1873. {
  1874. Emitter: "src1",
  1875. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1876. }, {
  1877. Emitter: "src1",
  1878. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1879. }, {
  1880. Emitter: "src1",
  1881. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1882. },
  1883. },
  1884. },
  1885. {
  1886. Emitter: "src2",
  1887. Tuples: []xsql.Tuple{
  1888. {
  1889. Emitter: "src2",
  1890. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1891. }, {
  1892. Emitter: "src2",
  1893. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1894. }, {
  1895. Emitter: "src2",
  1896. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1897. },
  1898. },
  1899. },
  1900. },
  1901. },
  1902. result: &xsql.JoinTupleSets{
  1903. Content: []xsql.JoinTuple{
  1904. {
  1905. Tuples: []xsql.Tuple{
  1906. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1907. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1908. },
  1909. },
  1910. {
  1911. Tuples: []xsql.Tuple{
  1912. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1913. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1914. },
  1915. },
  1916. {
  1917. Tuples: []xsql.Tuple{
  1918. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1919. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1920. },
  1921. },
  1922. {
  1923. Tuples: []xsql.Tuple{
  1924. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1925. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1926. },
  1927. },
  1928. {
  1929. Tuples: []xsql.Tuple{
  1930. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1931. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1932. },
  1933. },
  1934. {
  1935. Tuples: []xsql.Tuple{
  1936. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1937. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1938. },
  1939. },
  1940. {
  1941. Tuples: []xsql.Tuple{
  1942. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  1943. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1944. },
  1945. },
  1946. {
  1947. Tuples: []xsql.Tuple{
  1948. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  1949. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1950. },
  1951. },
  1952. {
  1953. Tuples: []xsql.Tuple{
  1954. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  1955. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1956. },
  1957. },
  1958. },
  1959. },
  1960. },
  1961. {
  1962. sql: "SELECT id1 FROM src1 cross join src2",
  1963. data: xsql.WindowTuplesSet{
  1964. Content: []xsql.WindowTuples{
  1965. {
  1966. Emitter: "src1",
  1967. Tuples: []xsql.Tuple{
  1968. {
  1969. Emitter: "src1",
  1970. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1971. },
  1972. },
  1973. },
  1974. {
  1975. Emitter: "src2",
  1976. Tuples: []xsql.Tuple{
  1977. {
  1978. Emitter: "src2",
  1979. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1980. }, {
  1981. Emitter: "src2",
  1982. Message: xsql.Message{"id2": 1, "f2": "w2"},
  1983. },
  1984. },
  1985. },
  1986. },
  1987. },
  1988. result: &xsql.JoinTupleSets{
  1989. Content: []xsql.JoinTuple{
  1990. {
  1991. Tuples: []xsql.Tuple{
  1992. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1993. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1994. },
  1995. },
  1996. {
  1997. Tuples: []xsql.Tuple{
  1998. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}}, {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w2"}},
  1999. },
  2000. },
  2001. },
  2002. },
  2003. },
  2004. }
  2005. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  2006. contextLogger := conf.Log.WithField("rule", "TestCrossJoinPlan_Apply")
  2007. ctx := context.WithValue(context.Background(), context.LoggerKey, contextLogger)
  2008. for i, tt := range tests {
  2009. stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
  2010. if err != nil {
  2011. t.Errorf("statement parse error %s", err)
  2012. break
  2013. }
  2014. if table, ok := stmt.Sources[0].(*ast.Table); !ok {
  2015. t.Errorf("statement source is not a table")
  2016. } else {
  2017. fv, afv := xsql.NewFunctionValuersForOp(nil, xsql.FuncRegisters)
  2018. pp := &JoinOp{Joins: stmt.Joins, From: table}
  2019. result := pp.Apply(ctx, tt.data, fv, afv)
  2020. if !reflect.DeepEqual(tt.result, result) {
  2021. t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
  2022. }
  2023. }
  2024. }
  2025. }
  2026. func TestCrossJoinPlanError(t *testing.T) {
  2027. var tests = []struct {
  2028. sql string
  2029. data interface{}
  2030. result interface{}
  2031. }{
  2032. {
  2033. sql: "SELECT id1 FROM src1 cross join src2",
  2034. data: errors.New("an error from upstream"),
  2035. result: errors.New("an error from upstream"),
  2036. }, {
  2037. sql: "SELECT id1 FROM src1 full join src2 on src1.id1 = src2.id2",
  2038. data: xsql.WindowTuplesSet{
  2039. Content: []xsql.WindowTuples{
  2040. {
  2041. Emitter: "src1",
  2042. Tuples: []xsql.Tuple{
  2043. {
  2044. Emitter: "src1",
  2045. Message: xsql.Message{"id1": 1, "f1": "v1"},
  2046. }, {
  2047. Emitter: "src1",
  2048. Message: xsql.Message{"id1": 2, "f1": "v2"},
  2049. }, {
  2050. Emitter: "src1",
  2051. Message: xsql.Message{"id1": 3, "f1": "v3"},
  2052. },
  2053. },
  2054. },
  2055. {
  2056. Emitter: "src2",
  2057. Tuples: []xsql.Tuple{
  2058. {
  2059. Emitter: "src2",
  2060. Message: xsql.Message{"id2": 1, "f2": "w1"},
  2061. }, {
  2062. Emitter: "src2",
  2063. Message: xsql.Message{"id2": "3", "f2": "w2"},
  2064. }, {
  2065. Emitter: "src2",
  2066. Message: xsql.Message{"id2": 4, "f2": "w3"},
  2067. }, {
  2068. Emitter: "src2",
  2069. Message: xsql.Message{"id2": 2, "f2": "w4"},
  2070. },
  2071. },
  2072. },
  2073. },
  2074. },
  2075. result: errors.New("run Join error: invalid operation int64(1) = string(3)"),
  2076. },
  2077. }
  2078. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  2079. contextLogger := conf.Log.WithField("rule", "TestCrossJoinPlan_Apply")
  2080. ctx := context.WithValue(context.Background(), context.LoggerKey, contextLogger)
  2081. for i, tt := range tests {
  2082. stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
  2083. if err != nil {
  2084. t.Errorf("statement parse error %s", err)
  2085. break
  2086. }
  2087. if table, ok := stmt.Sources[0].(*ast.Table); !ok {
  2088. t.Errorf("statement source is not a table")
  2089. } else {
  2090. fv, afv := xsql.NewFunctionValuersForOp(nil, xsql.FuncRegisters)
  2091. pp := &JoinOp{Joins: stmt.Joins, From: table}
  2092. result := pp.Apply(ctx, tt.data, fv, afv)
  2093. if !reflect.DeepEqual(tt.result, result) {
  2094. t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
  2095. }
  2096. }
  2097. }
  2098. }
  2099. func str2Map(s string) map[string]interface{} {
  2100. var input map[string]interface{}
  2101. if err := json.Unmarshal([]byte(s), &input); err != nil {
  2102. fmt.Printf("Failed to parse the JSON data.\n")
  2103. return nil
  2104. }
  2105. return input
  2106. }