join_test.go 39 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554
  1. package plans
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "github.com/emqx/kuiper/common"
  6. "github.com/emqx/kuiper/xsql"
  7. "github.com/emqx/kuiper/xstream/contexts"
  8. "reflect"
  9. "strings"
  10. "testing"
  11. )
  12. func TestLeftJoinPlan_Apply(t *testing.T) {
  13. var tests = []struct {
  14. sql string
  15. data xsql.WindowTuplesSet
  16. result interface{}
  17. }{
  18. {
  19. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  20. data: xsql.WindowTuplesSet{
  21. xsql.WindowTuples{
  22. Emitter: "src1",
  23. Tuples: []xsql.Tuple{
  24. {
  25. Emitter: "src1",
  26. Message: xsql.Message{"id1": 1, "f1": "v1"},
  27. }, {
  28. Emitter: "src1",
  29. Message: xsql.Message{"id1": 2, "f1": "v2"},
  30. }, {
  31. Emitter: "src1",
  32. Message: xsql.Message{"id1": 3, "f1": "v3"},
  33. },
  34. },
  35. },
  36. xsql.WindowTuples{
  37. Emitter: "src2",
  38. Tuples: []xsql.Tuple{
  39. {
  40. Emitter: "src2",
  41. Message: xsql.Message{"id2": 1, "f2": "w1"},
  42. }, {
  43. Emitter: "src2",
  44. Message: xsql.Message{"id2": 2, "f2": "w2"},
  45. }, {
  46. Emitter: "src2",
  47. Message: xsql.Message{"id2": 4, "f2": "w3"},
  48. },
  49. },
  50. },
  51. },
  52. result: xsql.JoinTupleSets{
  53. xsql.JoinTuple{
  54. Tuples: []xsql.Tuple{
  55. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  56. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  57. },
  58. },
  59. xsql.JoinTuple{
  60. Tuples: []xsql.Tuple{
  61. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  62. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  63. },
  64. },
  65. xsql.JoinTuple{
  66. Tuples: []xsql.Tuple{
  67. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  68. },
  69. },
  70. },
  71. },
  72. {
  73. sql: "SELECT id1 FROM src1 left join src2 on src1.ts = src2.ts",
  74. data: xsql.WindowTuplesSet{
  75. xsql.WindowTuples{
  76. Emitter: "src1",
  77. Tuples: []xsql.Tuple{
  78. {
  79. Emitter: "src1",
  80. Message: xsql.Message{"id1": 1, "f1": "v1", "ts": common.TimeFromUnixMilli(1568854515000)},
  81. }, {
  82. Emitter: "src1",
  83. Message: xsql.Message{"id1": 2, "f1": "v2", "ts": common.TimeFromUnixMilli(1568854525000)},
  84. }, {
  85. Emitter: "src1",
  86. Message: xsql.Message{"id1": 3, "f1": "v3", "ts": common.TimeFromUnixMilli(1568854535000)},
  87. },
  88. },
  89. },
  90. xsql.WindowTuples{
  91. Emitter: "src2",
  92. Tuples: []xsql.Tuple{
  93. {
  94. Emitter: "src2",
  95. Message: xsql.Message{"id2": 1, "f2": "w1", "ts": common.TimeFromUnixMilli(1568854515000)},
  96. }, {
  97. Emitter: "src2",
  98. Message: xsql.Message{"id2": 2, "f2": "w2", "ts": common.TimeFromUnixMilli(1568854525000)},
  99. }, {
  100. Emitter: "src2",
  101. Message: xsql.Message{"id2": 4, "f2": "w3", "ts": common.TimeFromUnixMilli(1568854545000)},
  102. },
  103. },
  104. },
  105. },
  106. result: xsql.JoinTupleSets{
  107. xsql.JoinTuple{
  108. Tuples: []xsql.Tuple{
  109. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1", "ts": common.TimeFromUnixMilli(1568854515000)}},
  110. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1", "ts": common.TimeFromUnixMilli(1568854515000)}},
  111. },
  112. },
  113. xsql.JoinTuple{
  114. Tuples: []xsql.Tuple{
  115. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2", "ts": common.TimeFromUnixMilli(1568854525000)}},
  116. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2", "ts": common.TimeFromUnixMilli(1568854525000)}},
  117. },
  118. },
  119. xsql.JoinTuple{
  120. Tuples: []xsql.Tuple{
  121. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3", "ts": common.TimeFromUnixMilli(1568854535000)}},
  122. },
  123. },
  124. },
  125. },
  126. {
  127. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  128. data: xsql.WindowTuplesSet{
  129. xsql.WindowTuples{
  130. Emitter: "src1",
  131. Tuples: []xsql.Tuple{
  132. {
  133. Emitter: "src1",
  134. Message: xsql.Message{"id1": 1, "f1": "v1"},
  135. }, {
  136. Emitter: "src1",
  137. Message: xsql.Message{"id1": 2, "f1": "v2"},
  138. }, {
  139. Emitter: "src1",
  140. Message: xsql.Message{"id1": 3, "f1": "v3"},
  141. },
  142. },
  143. },
  144. xsql.WindowTuples{
  145. Emitter: "src2",
  146. Tuples: []xsql.Tuple{
  147. {
  148. Emitter: "src2",
  149. Message: xsql.Message{"id2": 4, "f2": "w1"},
  150. }, {
  151. Emitter: "src2",
  152. Message: xsql.Message{"id2": 5, "f2": "w2"},
  153. }, {
  154. Emitter: "src2",
  155. Message: xsql.Message{"id2": 6, "f2": "w3"},
  156. },
  157. },
  158. },
  159. },
  160. result: nil,
  161. },
  162. {
  163. sql: "SELECT id1 FROM src1 As s1 left join src2 as s2 on s1.id1 = s2.id2",
  164. data: xsql.WindowTuplesSet{
  165. xsql.WindowTuples{
  166. Emitter: "s1",
  167. Tuples: []xsql.Tuple{
  168. {
  169. Emitter: "s1",
  170. Message: xsql.Message{"id1": 1, "f1": "v1"},
  171. }, {
  172. Emitter: "s1",
  173. Message: xsql.Message{"id1": 2, "f1": "v2"},
  174. }, {
  175. Emitter: "s1",
  176. Message: xsql.Message{"id1": 3, "f1": "v3"},
  177. },
  178. },
  179. },
  180. xsql.WindowTuples{
  181. Emitter: "s2",
  182. Tuples: []xsql.Tuple{
  183. {
  184. Emitter: "s2",
  185. Message: xsql.Message{"id2": 1, "f2": "w1"},
  186. }, {
  187. Emitter: "s2",
  188. Message: xsql.Message{"id2": 2, "f2": "w2"},
  189. }, {
  190. Emitter: "s2",
  191. Message: xsql.Message{"id2": 4, "f2": "w3"},
  192. },
  193. },
  194. },
  195. },
  196. result: xsql.JoinTupleSets{
  197. xsql.JoinTuple{
  198. Tuples: []xsql.Tuple{
  199. {Emitter: "s1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  200. {Emitter: "s2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  201. },
  202. },
  203. xsql.JoinTuple{
  204. Tuples: []xsql.Tuple{
  205. {Emitter: "s1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  206. {Emitter: "s2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  207. },
  208. },
  209. xsql.JoinTuple{
  210. Tuples: []xsql.Tuple{
  211. {Emitter: "s1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  212. },
  213. },
  214. },
  215. },
  216. {
  217. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  218. data: xsql.WindowTuplesSet{
  219. xsql.WindowTuples{
  220. Emitter: "src1",
  221. Tuples: []xsql.Tuple{
  222. {
  223. Emitter: "src1",
  224. Message: xsql.Message{"id1": 1, "f1": "v1"},
  225. },
  226. },
  227. },
  228. xsql.WindowTuples{
  229. Emitter: "src2",
  230. Tuples: []xsql.Tuple{
  231. {
  232. Emitter: "src2",
  233. Message: xsql.Message{"id2": 1, "f2": "w1"},
  234. }, {
  235. Emitter: "src2",
  236. Message: xsql.Message{"id2": 1, "f2": "w2"},
  237. },
  238. },
  239. },
  240. },
  241. result: xsql.JoinTupleSets{
  242. xsql.JoinTuple{
  243. Tuples: []xsql.Tuple{
  244. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  245. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  246. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w2"}},
  247. },
  248. },
  249. },
  250. },
  251. {
  252. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  253. data: xsql.WindowTuplesSet{
  254. xsql.WindowTuples{
  255. Emitter: "src1",
  256. Tuples: []xsql.Tuple{
  257. {
  258. Emitter: "src1",
  259. Message: xsql.Message{"id1": 1, "f1": "v1"},
  260. }, {
  261. Emitter: "src1",
  262. Message: xsql.Message{"id1": 2, "f1": "v2"},
  263. }, {
  264. Emitter: "src1",
  265. Message: xsql.Message{"id1": 3, "f1": "v3"},
  266. },
  267. },
  268. },
  269. xsql.WindowTuples{
  270. Emitter: "src2",
  271. Tuples: []xsql.Tuple{},
  272. },
  273. },
  274. result: xsql.JoinTupleSets{
  275. xsql.JoinTuple{
  276. Tuples: []xsql.Tuple{
  277. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  278. },
  279. },
  280. xsql.JoinTuple{
  281. Tuples: []xsql.Tuple{
  282. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  283. },
  284. },
  285. xsql.JoinTuple{
  286. Tuples: []xsql.Tuple{
  287. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  288. },
  289. },
  290. },
  291. },
  292. {
  293. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  294. data: xsql.WindowTuplesSet{
  295. xsql.WindowTuples{
  296. Emitter: "src1",
  297. Tuples: []xsql.Tuple{
  298. {
  299. Emitter: "src1",
  300. Message: xsql.Message{"id1": 1, "f1": "v1"},
  301. }, {
  302. Emitter: "src1",
  303. Message: xsql.Message{"id1": 2, "f1": "v2"},
  304. }, {
  305. Emitter: "src1",
  306. Message: xsql.Message{"id1": 3, "f1": "v3"},
  307. },
  308. },
  309. },
  310. xsql.WindowTuples{
  311. Emitter: "src2",
  312. Tuples: nil,
  313. },
  314. },
  315. result: xsql.JoinTupleSets{
  316. xsql.JoinTuple{
  317. Tuples: []xsql.Tuple{
  318. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  319. },
  320. },
  321. xsql.JoinTuple{
  322. Tuples: []xsql.Tuple{
  323. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  324. },
  325. },
  326. xsql.JoinTuple{
  327. Tuples: []xsql.Tuple{
  328. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  329. },
  330. },
  331. },
  332. },
  333. {
  334. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  335. data: xsql.WindowTuplesSet{
  336. xsql.WindowTuples{
  337. Emitter: "src1",
  338. Tuples: []xsql.Tuple{},
  339. },
  340. xsql.WindowTuples{
  341. Emitter: "src2",
  342. Tuples: []xsql.Tuple{
  343. {
  344. Emitter: "src2",
  345. Message: xsql.Message{"id2": 1, "f2": "w1"},
  346. }, {
  347. Emitter: "src2",
  348. Message: xsql.Message{"id2": 1, "f2": "w2"},
  349. },
  350. },
  351. },
  352. },
  353. result: nil,
  354. },
  355. {
  356. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2",
  357. data: xsql.WindowTuplesSet{
  358. xsql.WindowTuples{
  359. Emitter: "src1",
  360. Tuples: nil,
  361. },
  362. xsql.WindowTuples{
  363. Emitter: "src2",
  364. Tuples: []xsql.Tuple{
  365. {
  366. Emitter: "src2",
  367. Message: xsql.Message{"id2": 1, "f2": "w1"},
  368. }, {
  369. Emitter: "src2",
  370. Message: xsql.Message{"id2": 1, "f2": "w2"},
  371. },
  372. },
  373. },
  374. },
  375. result: nil,
  376. },
  377. {
  378. sql: "SELECT id1 FROM src1 left join src2 on src1.id1*2 = src2.id2",
  379. data: xsql.WindowTuplesSet{
  380. xsql.WindowTuples{
  381. Emitter: "src1",
  382. Tuples: []xsql.Tuple{
  383. {
  384. Emitter: "src1",
  385. Message: xsql.Message{"id1": 1, "f1": "v1"},
  386. }, {
  387. Emitter: "src1",
  388. Message: xsql.Message{"id1": 2, "f1": "v2"},
  389. }, {
  390. Emitter: "src1",
  391. Message: xsql.Message{"id1": 3, "f1": "v3"},
  392. },
  393. },
  394. },
  395. xsql.WindowTuples{
  396. Emitter: "src2",
  397. Tuples: []xsql.Tuple{
  398. {
  399. Emitter: "src2",
  400. Message: xsql.Message{"id2": 1, "f2": "w1"},
  401. }, {
  402. Emitter: "src2",
  403. Message: xsql.Message{"id2": 2, "f2": "w2"},
  404. }, {
  405. Emitter: "src2",
  406. Message: xsql.Message{"id2": 4, "f2": "w3"},
  407. },
  408. },
  409. },
  410. },
  411. result: xsql.JoinTupleSets{
  412. xsql.JoinTuple{
  413. Tuples: []xsql.Tuple{
  414. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  415. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  416. },
  417. },
  418. xsql.JoinTuple{
  419. Tuples: []xsql.Tuple{
  420. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  421. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  422. },
  423. },
  424. xsql.JoinTuple{
  425. Tuples: []xsql.Tuple{
  426. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  427. },
  428. },
  429. },
  430. },
  431. {
  432. sql: "SELECT id1 FROM src1 left join src2 on src1.id1 = src2.id2*2",
  433. data: xsql.WindowTuplesSet{
  434. xsql.WindowTuples{
  435. Emitter: "src1",
  436. Tuples: []xsql.Tuple{
  437. {
  438. Emitter: "src1",
  439. Message: xsql.Message{"id1": 1, "f1": "v1"},
  440. }, {
  441. Emitter: "src1",
  442. Message: xsql.Message{"id1": 2, "f1": "v2"},
  443. }, {
  444. Emitter: "src1",
  445. Message: xsql.Message{"id1": 3, "f1": "v3"},
  446. },
  447. },
  448. },
  449. xsql.WindowTuples{
  450. Emitter: "src2",
  451. Tuples: []xsql.Tuple{
  452. {
  453. Emitter: "src2",
  454. Message: xsql.Message{"id2": 1, "f2": "w1"},
  455. }, {
  456. Emitter: "src2",
  457. Message: xsql.Message{"id2": 2, "f2": "w2"},
  458. }, {
  459. Emitter: "src2",
  460. Message: xsql.Message{"id2": 4, "f2": "w3"},
  461. },
  462. },
  463. },
  464. },
  465. result: xsql.JoinTupleSets{
  466. xsql.JoinTuple{
  467. Tuples: []xsql.Tuple{
  468. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  469. },
  470. },
  471. xsql.JoinTuple{
  472. Tuples: []xsql.Tuple{
  473. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  474. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  475. },
  476. },
  477. xsql.JoinTuple{
  478. Tuples: []xsql.Tuple{
  479. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  480. },
  481. },
  482. },
  483. },
  484. {
  485. sql: "SELECT id1 FROM src1 left join src2 on src1.f1->cid = src2.f2->cid",
  486. data: xsql.WindowTuplesSet{
  487. xsql.WindowTuples{
  488. Emitter: "src1",
  489. Tuples: []xsql.Tuple{
  490. {
  491. Emitter: "src1",
  492. Message: xsql.Message{"id1": 1, "f1": str2Map(`{"cid" : 1, "name" : "tom1"}`)},
  493. }, {
  494. Emitter: "src1",
  495. Message: xsql.Message{"id1": 2, "f1": str2Map(`{"cid" : 2, "name" : "mike1"}`)},
  496. }, {
  497. Emitter: "src1",
  498. Message: xsql.Message{"id1": 3, "f1": str2Map(`{"cid" : 3, "name" : "alice1"}`)},
  499. },
  500. },
  501. },
  502. xsql.WindowTuples{
  503. Emitter: "src2",
  504. Tuples: []xsql.Tuple{
  505. {
  506. Emitter: "src2",
  507. Message: xsql.Message{"id2": 1, "f2": str2Map(`{"cid" : 1, "name" : "tom2"}`)},
  508. }, {
  509. Emitter: "src2",
  510. Message: xsql.Message{"id2": 2, "f2": str2Map(`{"cid" : 2, "name" : "mike2"}`)},
  511. }, {
  512. Emitter: "src2",
  513. Message: xsql.Message{"id2": 4, "f2": str2Map(`{"cid" : 4, "name" : "alice2"}`)},
  514. },
  515. },
  516. },
  517. },
  518. result: xsql.JoinTupleSets{
  519. xsql.JoinTuple{
  520. Tuples: []xsql.Tuple{
  521. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": str2Map(`{"cid" : 1, "name" : "tom1"}`)}},
  522. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": str2Map(`{"cid" : 1, "name" : "tom2"}`)}},
  523. },
  524. },
  525. xsql.JoinTuple{
  526. Tuples: []xsql.Tuple{
  527. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": str2Map(`{"cid" : 2, "name" : "mike1"}`)}},
  528. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": str2Map(`{"cid" : 2, "name" : "mike2"}`)}},
  529. },
  530. },
  531. xsql.JoinTuple{
  532. Tuples: []xsql.Tuple{
  533. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": str2Map(`{"cid" : 3, "name" : "alice1"}`)}},
  534. },
  535. },
  536. },
  537. },
  538. {
  539. sql: "SELECT id1, mqtt(src1.topic) AS a, mqtt(src2.topic) as b FROM src1 left join src2 on src1.id1 = src2.id2",
  540. data: xsql.WindowTuplesSet{
  541. xsql.WindowTuples{
  542. Emitter: "src1",
  543. Tuples: []xsql.Tuple{
  544. {
  545. Emitter: "src1",
  546. Message: xsql.Message{"id1": 1, "f1": "v1", xsql.INTERNAL_MQTT_TOPIC_KEY: "devices/type1/device001"},
  547. },
  548. },
  549. },
  550. xsql.WindowTuples{
  551. Emitter: "src2",
  552. Tuples: []xsql.Tuple{
  553. {
  554. Emitter: "src2",
  555. Message: xsql.Message{"id2": 1, "f2": "w1", xsql.INTERNAL_MQTT_TOPIC_KEY: "devices/type2/device001"},
  556. },
  557. },
  558. },
  559. },
  560. result: xsql.JoinTupleSets{
  561. xsql.JoinTuple{
  562. Tuples: []xsql.Tuple{
  563. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1", xsql.INTERNAL_MQTT_TOPIC_KEY: "devices/type1/device001"}},
  564. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1", xsql.INTERNAL_MQTT_TOPIC_KEY: "devices/type2/device001"}},
  565. },
  566. },
  567. },
  568. },
  569. }
  570. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  571. contextLogger := common.Log.WithField("rule", "TestLeftJoinPlan_Apply")
  572. ctx := contexts.WithValue(contexts.Background(), contexts.LoggerKey, contextLogger)
  573. for i, tt := range tests {
  574. stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
  575. if err != nil {
  576. t.Errorf("statement parse error %s", err)
  577. break
  578. }
  579. if table, ok := stmt.Sources[0].(*xsql.Table); !ok {
  580. t.Errorf("statement source is not a table")
  581. } else {
  582. pp := &JoinPlan{Joins: stmt.Joins, From: table}
  583. result := pp.Apply(ctx, tt.data)
  584. if !reflect.DeepEqual(tt.result, result) {
  585. t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
  586. }
  587. }
  588. }
  589. }
  590. func TestInnerJoinPlan_Apply(t *testing.T) {
  591. var tests = []struct {
  592. sql string
  593. data xsql.WindowTuplesSet
  594. result interface{}
  595. }{
  596. {
  597. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  598. data: xsql.WindowTuplesSet{
  599. xsql.WindowTuples{
  600. Emitter: "src1",
  601. Tuples: []xsql.Tuple{
  602. {
  603. Emitter: "src1",
  604. Message: xsql.Message{"id1": 1, "f1": "v1"},
  605. }, {
  606. Emitter: "src1",
  607. Message: xsql.Message{"id1": 2, "f1": "v2"},
  608. }, {
  609. Emitter: "src1",
  610. Message: xsql.Message{"id1": 3, "f1": "v3"},
  611. },
  612. },
  613. },
  614. xsql.WindowTuples{
  615. Emitter: "src2",
  616. Tuples: []xsql.Tuple{
  617. {
  618. Emitter: "src2",
  619. Message: xsql.Message{"id2": 1, "f2": "w1"},
  620. }, {
  621. Emitter: "src2",
  622. Message: xsql.Message{"id2": 2, "f2": "w2"},
  623. }, {
  624. Emitter: "src2",
  625. Message: xsql.Message{"id2": 4, "f2": "w3"},
  626. },
  627. },
  628. },
  629. },
  630. result: xsql.JoinTupleSets{
  631. xsql.JoinTuple{
  632. Tuples: []xsql.Tuple{
  633. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  634. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  635. },
  636. },
  637. xsql.JoinTuple{
  638. Tuples: []xsql.Tuple{
  639. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  640. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  641. },
  642. },
  643. },
  644. },
  645. {
  646. sql: "SELECT id1 FROM src1 As s1 inner join src2 as s2 on s1.id1 = s2.id2",
  647. data: xsql.WindowTuplesSet{
  648. xsql.WindowTuples{
  649. Emitter: "s1",
  650. Tuples: []xsql.Tuple{
  651. {
  652. Emitter: "s1",
  653. Message: xsql.Message{"id1": 1, "f1": "v1"},
  654. }, {
  655. Emitter: "s1",
  656. Message: xsql.Message{"id1": 2, "f1": "v2"},
  657. }, {
  658. Emitter: "s1",
  659. Message: xsql.Message{"id1": 3, "f1": "v3"},
  660. },
  661. },
  662. },
  663. xsql.WindowTuples{
  664. Emitter: "s2",
  665. Tuples: []xsql.Tuple{
  666. {
  667. Emitter: "s2",
  668. Message: xsql.Message{"id2": 1, "f2": "w1"},
  669. }, {
  670. Emitter: "s2",
  671. Message: xsql.Message{"id2": 2, "f2": "w2"},
  672. }, {
  673. Emitter: "s2",
  674. Message: xsql.Message{"id2": 4, "f2": "w3"},
  675. },
  676. },
  677. },
  678. },
  679. result: xsql.JoinTupleSets{
  680. xsql.JoinTuple{
  681. Tuples: []xsql.Tuple{
  682. {Emitter: "s1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  683. {Emitter: "s2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  684. },
  685. },
  686. xsql.JoinTuple{
  687. Tuples: []xsql.Tuple{
  688. {Emitter: "s1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  689. {Emitter: "s2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  690. },
  691. },
  692. },
  693. },
  694. {
  695. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  696. data: xsql.WindowTuplesSet{
  697. xsql.WindowTuples{
  698. Emitter: "src1",
  699. Tuples: []xsql.Tuple{
  700. {
  701. Emitter: "src1",
  702. Message: xsql.Message{"id1": 1, "f1": "v1"},
  703. },
  704. },
  705. },
  706. xsql.WindowTuples{
  707. Emitter: "src2",
  708. Tuples: []xsql.Tuple{
  709. {
  710. Emitter: "src2",
  711. Message: xsql.Message{"id2": 1, "f2": "w1"},
  712. }, {
  713. Emitter: "src2",
  714. Message: xsql.Message{"id2": 1, "f2": "w2"},
  715. },
  716. },
  717. },
  718. },
  719. result: xsql.JoinTupleSets{
  720. xsql.JoinTuple{
  721. Tuples: []xsql.Tuple{
  722. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  723. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  724. },
  725. },
  726. xsql.JoinTuple{
  727. Tuples: []xsql.Tuple{
  728. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  729. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w2"}},
  730. },
  731. },
  732. },
  733. },
  734. {
  735. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  736. data: xsql.WindowTuplesSet{
  737. xsql.WindowTuples{
  738. Emitter: "src1",
  739. Tuples: []xsql.Tuple{
  740. {
  741. Emitter: "src1",
  742. Message: xsql.Message{"id1": 1, "f1": "v1"},
  743. }, {
  744. Emitter: "src1",
  745. Message: xsql.Message{"id1": 2, "f1": "v2"},
  746. }, {
  747. Emitter: "src1",
  748. Message: xsql.Message{"id1": 3, "f1": "v3"},
  749. },
  750. },
  751. },
  752. xsql.WindowTuples{
  753. Emitter: "src2",
  754. Tuples: []xsql.Tuple{},
  755. },
  756. },
  757. result: nil,
  758. },
  759. {
  760. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  761. data: xsql.WindowTuplesSet{
  762. xsql.WindowTuples{
  763. Emitter: "src1",
  764. Tuples: []xsql.Tuple{
  765. {
  766. Emitter: "src1",
  767. Message: xsql.Message{"id1": 1, "f1": "v1"},
  768. }, {
  769. Emitter: "src1",
  770. Message: xsql.Message{"id1": 2, "f1": "v2"},
  771. }, {
  772. Emitter: "src1",
  773. Message: xsql.Message{"id1": 3, "f1": "v3"},
  774. },
  775. },
  776. },
  777. xsql.WindowTuples{
  778. Emitter: "src2",
  779. Tuples: nil,
  780. },
  781. },
  782. result: nil,
  783. },
  784. {
  785. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  786. data: xsql.WindowTuplesSet{
  787. xsql.WindowTuples{
  788. Emitter: "src1",
  789. Tuples: []xsql.Tuple{},
  790. },
  791. xsql.WindowTuples{
  792. Emitter: "src2",
  793. Tuples: []xsql.Tuple{
  794. {
  795. Emitter: "src2",
  796. Message: xsql.Message{"id2": 1, "f2": "w1"},
  797. }, {
  798. Emitter: "src2",
  799. Message: xsql.Message{"id2": 1, "f2": "w2"},
  800. },
  801. },
  802. },
  803. },
  804. result: nil,
  805. },
  806. {
  807. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2",
  808. data: xsql.WindowTuplesSet{
  809. xsql.WindowTuples{
  810. Emitter: "src1",
  811. Tuples: nil,
  812. },
  813. xsql.WindowTuples{
  814. Emitter: "src2",
  815. Tuples: []xsql.Tuple{
  816. {
  817. Emitter: "src2",
  818. Message: xsql.Message{"id2": 1, "f2": "w1"},
  819. }, {
  820. Emitter: "src2",
  821. Message: xsql.Message{"id2": 1, "f2": "w2"},
  822. },
  823. },
  824. },
  825. },
  826. result: nil,
  827. },
  828. {
  829. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1*2 = src2.id2",
  830. data: xsql.WindowTuplesSet{
  831. xsql.WindowTuples{
  832. Emitter: "src1",
  833. Tuples: []xsql.Tuple{
  834. {
  835. Emitter: "src1",
  836. Message: xsql.Message{"id1": 1, "f1": "v1"},
  837. }, {
  838. Emitter: "src1",
  839. Message: xsql.Message{"id1": 2, "f1": "v2"},
  840. }, {
  841. Emitter: "src1",
  842. Message: xsql.Message{"id1": 3, "f1": "v3"},
  843. },
  844. },
  845. },
  846. xsql.WindowTuples{
  847. Emitter: "src2",
  848. Tuples: []xsql.Tuple{
  849. {
  850. Emitter: "src2",
  851. Message: xsql.Message{"id2": 1, "f2": "w1"},
  852. }, {
  853. Emitter: "src2",
  854. Message: xsql.Message{"id2": 2, "f2": "w2"},
  855. }, {
  856. Emitter: "src2",
  857. Message: xsql.Message{"id2": 4, "f2": "w3"},
  858. },
  859. },
  860. },
  861. },
  862. result: xsql.JoinTupleSets{
  863. xsql.JoinTuple{
  864. Tuples: []xsql.Tuple{
  865. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  866. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  867. },
  868. },
  869. xsql.JoinTuple{
  870. Tuples: []xsql.Tuple{
  871. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  872. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  873. },
  874. },
  875. },
  876. },
  877. {
  878. sql: "SELECT id1 FROM src1 inner join src2 on src1.id1 = src2.id2*2",
  879. data: xsql.WindowTuplesSet{
  880. xsql.WindowTuples{
  881. Emitter: "src1",
  882. Tuples: []xsql.Tuple{
  883. {
  884. Emitter: "src1",
  885. Message: xsql.Message{"id1": 1, "f1": "v1"},
  886. }, {
  887. Emitter: "src1",
  888. Message: xsql.Message{"id1": 2, "f1": "v2"},
  889. }, {
  890. Emitter: "src1",
  891. Message: xsql.Message{"id1": 3, "f1": "v3"},
  892. },
  893. },
  894. },
  895. xsql.WindowTuples{
  896. Emitter: "src2",
  897. Tuples: []xsql.Tuple{
  898. {
  899. Emitter: "src2",
  900. Message: xsql.Message{"id2": 1, "f2": "w1"},
  901. }, {
  902. Emitter: "src2",
  903. Message: xsql.Message{"id2": 2, "f2": "w2"},
  904. }, {
  905. Emitter: "src2",
  906. Message: xsql.Message{"id2": 4, "f2": "w3"},
  907. },
  908. },
  909. },
  910. },
  911. result: xsql.JoinTupleSets{
  912. xsql.JoinTuple{
  913. Tuples: []xsql.Tuple{
  914. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  915. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  916. },
  917. },
  918. },
  919. },
  920. {
  921. sql: "SELECT id1 FROM src1 inner join src2 on src1.f1->cid = src2.f2->cid",
  922. data: xsql.WindowTuplesSet{
  923. xsql.WindowTuples{
  924. Emitter: "src1",
  925. Tuples: []xsql.Tuple{
  926. {
  927. Emitter: "src1",
  928. Message: xsql.Message{"id1": 1, "f1": str2Map(`{"cid" : 1, "name" : "tom1"}`)},
  929. }, {
  930. Emitter: "src1",
  931. Message: xsql.Message{"id1": 2, "f1": str2Map(`{"cid" : 2, "name" : "mike1"}`)},
  932. }, {
  933. Emitter: "src1",
  934. Message: xsql.Message{"id1": 3, "f1": str2Map(`{"cid" : 3, "name" : "alice1"}`)},
  935. },
  936. },
  937. },
  938. xsql.WindowTuples{
  939. Emitter: "src2",
  940. Tuples: []xsql.Tuple{
  941. {
  942. Emitter: "src2",
  943. Message: xsql.Message{"id2": 1, "f2": str2Map(`{"cid" : 1, "name" : "tom2"}`)},
  944. }, {
  945. Emitter: "src2",
  946. Message: xsql.Message{"id2": 2, "f2": str2Map(`{"cid" : 2, "name" : "mike2"}`)},
  947. }, {
  948. Emitter: "src2",
  949. Message: xsql.Message{"id2": 4, "f2": str2Map(`{"cid" : 4, "name" : "alice2"}`)},
  950. },
  951. },
  952. },
  953. },
  954. result: xsql.JoinTupleSets{
  955. xsql.JoinTuple{
  956. Tuples: []xsql.Tuple{
  957. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": str2Map(`{"cid" : 1, "name" : "tom1"}`)}},
  958. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": str2Map(`{"cid" : 1, "name" : "tom2"}`)}},
  959. },
  960. },
  961. xsql.JoinTuple{
  962. Tuples: []xsql.Tuple{
  963. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": str2Map(`{"cid" : 2, "name" : "mike1"}`)}},
  964. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": str2Map(`{"cid" : 2, "name" : "mike2"}`)}},
  965. },
  966. },
  967. },
  968. },
  969. }
  970. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  971. contextLogger := common.Log.WithField("rule", "TestInnerJoinPlan_Apply")
  972. ctx := contexts.WithValue(contexts.Background(), contexts.LoggerKey, contextLogger)
  973. for i, tt := range tests {
  974. stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
  975. if err != nil {
  976. t.Errorf("statement parse error %s", err)
  977. break
  978. }
  979. if table, ok := stmt.Sources[0].(*xsql.Table); !ok {
  980. t.Errorf("statement source is not a table")
  981. } else {
  982. pp := &JoinPlan{Joins: stmt.Joins, From: table}
  983. result := pp.Apply(ctx, tt.data)
  984. if !reflect.DeepEqual(tt.result, result) {
  985. t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
  986. }
  987. }
  988. }
  989. }
  990. func TestRightJoinPlan_Apply(t *testing.T) {
  991. var tests = []struct {
  992. sql string
  993. data xsql.WindowTuplesSet
  994. result interface{}
  995. }{
  996. {
  997. sql: "SELECT id1 FROM src1 right join src2 on src1.id1 = src2.id2",
  998. data: xsql.WindowTuplesSet{
  999. xsql.WindowTuples{
  1000. Emitter: "src1",
  1001. Tuples: []xsql.Tuple{
  1002. {
  1003. Emitter: "src1",
  1004. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1005. }, {
  1006. Emitter: "src1",
  1007. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1008. }, {
  1009. Emitter: "src1",
  1010. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1011. },
  1012. },
  1013. },
  1014. xsql.WindowTuples{
  1015. Emitter: "src2",
  1016. Tuples: []xsql.Tuple{
  1017. {
  1018. Emitter: "src2",
  1019. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1020. }, {
  1021. Emitter: "src2",
  1022. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1023. }, {
  1024. Emitter: "src2",
  1025. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1026. },
  1027. },
  1028. },
  1029. },
  1030. result: xsql.JoinTupleSets{
  1031. xsql.JoinTuple{
  1032. Tuples: []xsql.Tuple{
  1033. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1034. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1035. },
  1036. },
  1037. xsql.JoinTuple{
  1038. Tuples: []xsql.Tuple{
  1039. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1040. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1041. },
  1042. },
  1043. xsql.JoinTuple{
  1044. Tuples: []xsql.Tuple{
  1045. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1046. },
  1047. },
  1048. },
  1049. },
  1050. {
  1051. sql: "SELECT id1 FROM src1 right join src2 on src1.id1 = src2.id2",
  1052. data: xsql.WindowTuplesSet{
  1053. xsql.WindowTuples{
  1054. Emitter: "src1",
  1055. Tuples: []xsql.Tuple{
  1056. {
  1057. Emitter: "src1",
  1058. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1059. },
  1060. },
  1061. },
  1062. xsql.WindowTuples{
  1063. Emitter: "src2",
  1064. Tuples: []xsql.Tuple{
  1065. {
  1066. Emitter: "src2",
  1067. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1068. }, {
  1069. Emitter: "src2",
  1070. Message: xsql.Message{"id2": 1, "f2": "w2"},
  1071. },
  1072. },
  1073. },
  1074. },
  1075. result: xsql.JoinTupleSets{
  1076. xsql.JoinTuple{
  1077. Tuples: []xsql.Tuple{
  1078. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1079. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1080. },
  1081. },
  1082. xsql.JoinTuple{
  1083. Tuples: []xsql.Tuple{
  1084. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w2"}},
  1085. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1086. },
  1087. },
  1088. },
  1089. },
  1090. }
  1091. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  1092. contextLogger := common.Log.WithField("rule", "TestRightJoinPlan_Apply")
  1093. ctx := contexts.WithValue(contexts.Background(), contexts.LoggerKey, contextLogger)
  1094. for i, tt := range tests {
  1095. stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
  1096. if err != nil {
  1097. t.Errorf("statement parse error %s", err)
  1098. break
  1099. }
  1100. if table, ok := stmt.Sources[0].(*xsql.Table); !ok {
  1101. t.Errorf("statement source is not a table")
  1102. } else {
  1103. pp := &JoinPlan{Joins: stmt.Joins, From: table}
  1104. result := pp.Apply(ctx, tt.data)
  1105. if !reflect.DeepEqual(tt.result, result) {
  1106. t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
  1107. }
  1108. }
  1109. }
  1110. }
  1111. func TestFullJoinPlan_Apply(t *testing.T) {
  1112. var tests = []struct {
  1113. sql string
  1114. data xsql.WindowTuplesSet
  1115. result interface{}
  1116. }{
  1117. {
  1118. sql: "SELECT id1 FROM src1 full join src2 on src1.id1 = src2.id2",
  1119. data: xsql.WindowTuplesSet{
  1120. xsql.WindowTuples{
  1121. Emitter: "src1",
  1122. Tuples: []xsql.Tuple{
  1123. {
  1124. Emitter: "src1",
  1125. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1126. }, {
  1127. Emitter: "src1",
  1128. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1129. }, {
  1130. Emitter: "src1",
  1131. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1132. },
  1133. },
  1134. },
  1135. xsql.WindowTuples{
  1136. Emitter: "src2",
  1137. Tuples: []xsql.Tuple{
  1138. {
  1139. Emitter: "src2",
  1140. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1141. }, {
  1142. Emitter: "src2",
  1143. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1144. }, {
  1145. Emitter: "src2",
  1146. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1147. }, {
  1148. Emitter: "src2",
  1149. Message: xsql.Message{"id2": 2, "f2": "w4"},
  1150. },
  1151. },
  1152. },
  1153. },
  1154. result: xsql.JoinTupleSets{
  1155. xsql.JoinTuple{
  1156. Tuples: []xsql.Tuple{
  1157. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1158. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1159. },
  1160. },
  1161. xsql.JoinTuple{
  1162. Tuples: []xsql.Tuple{
  1163. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1164. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1165. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w4"}},
  1166. },
  1167. },
  1168. xsql.JoinTuple{
  1169. Tuples: []xsql.Tuple{
  1170. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  1171. },
  1172. },
  1173. xsql.JoinTuple{
  1174. Tuples: []xsql.Tuple{
  1175. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1176. },
  1177. },
  1178. },
  1179. },
  1180. {
  1181. sql: "SELECT id1 FROM src1 full join src2 on src1.id1 = src2.id2",
  1182. data: xsql.WindowTuplesSet{
  1183. xsql.WindowTuples{
  1184. Emitter: "src1",
  1185. Tuples: []xsql.Tuple{
  1186. {
  1187. Emitter: "src1",
  1188. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1189. }, {
  1190. Emitter: "src1",
  1191. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1192. }, {
  1193. Emitter: "src1",
  1194. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1195. },
  1196. },
  1197. },
  1198. xsql.WindowTuples{
  1199. Emitter: "src2",
  1200. Tuples: []xsql.Tuple{
  1201. {
  1202. Emitter: "src2",
  1203. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1204. }, {
  1205. Emitter: "src2",
  1206. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1207. }, {
  1208. Emitter: "src2",
  1209. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1210. },
  1211. },
  1212. },
  1213. },
  1214. result: xsql.JoinTupleSets{
  1215. xsql.JoinTuple{
  1216. Tuples: []xsql.Tuple{
  1217. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1218. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1219. },
  1220. },
  1221. xsql.JoinTuple{
  1222. Tuples: []xsql.Tuple{
  1223. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1224. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1225. },
  1226. },
  1227. xsql.JoinTuple{
  1228. Tuples: []xsql.Tuple{
  1229. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  1230. },
  1231. },
  1232. xsql.JoinTuple{
  1233. Tuples: []xsql.Tuple{
  1234. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1235. },
  1236. },
  1237. },
  1238. },
  1239. {
  1240. sql: "SELECT id1 FROM src1 full join src2 on src1.id1 = src2.id2",
  1241. data: xsql.WindowTuplesSet{
  1242. xsql.WindowTuples{
  1243. Emitter: "src1",
  1244. Tuples: []xsql.Tuple{
  1245. {
  1246. Emitter: "src1",
  1247. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1248. }, {
  1249. Emitter: "src1",
  1250. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1251. }, {
  1252. Emitter: "src1",
  1253. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1254. },
  1255. },
  1256. },
  1257. xsql.WindowTuples{
  1258. Emitter: "src2",
  1259. Tuples: []xsql.Tuple{},
  1260. },
  1261. },
  1262. result: xsql.JoinTupleSets{
  1263. xsql.JoinTuple{
  1264. Tuples: []xsql.Tuple{
  1265. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1266. },
  1267. },
  1268. xsql.JoinTuple{
  1269. Tuples: []xsql.Tuple{
  1270. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1271. },
  1272. },
  1273. xsql.JoinTuple{
  1274. Tuples: []xsql.Tuple{
  1275. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  1276. },
  1277. },
  1278. },
  1279. },
  1280. {
  1281. sql: "SELECT id1 FROM src1 full join src2 on src1.id1 = src2.id2",
  1282. data: xsql.WindowTuplesSet{
  1283. xsql.WindowTuples{
  1284. Emitter: "src1",
  1285. Tuples: []xsql.Tuple{},
  1286. },
  1287. xsql.WindowTuples{
  1288. Emitter: "src2",
  1289. Tuples: []xsql.Tuple{
  1290. {
  1291. Emitter: "src2",
  1292. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1293. }, {
  1294. Emitter: "src2",
  1295. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1296. }, {
  1297. Emitter: "src2",
  1298. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1299. },
  1300. },
  1301. },
  1302. },
  1303. result: xsql.JoinTupleSets{
  1304. xsql.JoinTuple{
  1305. Tuples: []xsql.Tuple{
  1306. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1307. },
  1308. },
  1309. xsql.JoinTuple{
  1310. Tuples: []xsql.Tuple{
  1311. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1312. },
  1313. },
  1314. xsql.JoinTuple{
  1315. Tuples: []xsql.Tuple{
  1316. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1317. },
  1318. },
  1319. },
  1320. },
  1321. }
  1322. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  1323. contextLogger := common.Log.WithField("rule", "TestFullJoinPlan_Apply")
  1324. ctx := contexts.WithValue(contexts.Background(), contexts.LoggerKey, contextLogger)
  1325. for i, tt := range tests {
  1326. stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
  1327. if err != nil {
  1328. t.Errorf("statement parse error %s", err)
  1329. break
  1330. }
  1331. if table, ok := stmt.Sources[0].(*xsql.Table); !ok {
  1332. t.Errorf("statement source is not a table")
  1333. } else {
  1334. pp := &JoinPlan{Joins: stmt.Joins, From: table}
  1335. result := pp.Apply(ctx, tt.data)
  1336. if !reflect.DeepEqual(tt.result, result) {
  1337. t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
  1338. }
  1339. }
  1340. }
  1341. }
  1342. func TestCrossJoinPlan_Apply(t *testing.T) {
  1343. var tests = []struct {
  1344. sql string
  1345. data xsql.WindowTuplesSet
  1346. result interface{}
  1347. }{
  1348. {
  1349. sql: "SELECT id1 FROM src1 cross join src2",
  1350. data: xsql.WindowTuplesSet{
  1351. xsql.WindowTuples{
  1352. Emitter: "src1",
  1353. Tuples: []xsql.Tuple{
  1354. {
  1355. Emitter: "src1",
  1356. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1357. }, {
  1358. Emitter: "src1",
  1359. Message: xsql.Message{"id1": 2, "f1": "v2"},
  1360. }, {
  1361. Emitter: "src1",
  1362. Message: xsql.Message{"id1": 3, "f1": "v3"},
  1363. },
  1364. },
  1365. },
  1366. xsql.WindowTuples{
  1367. Emitter: "src2",
  1368. Tuples: []xsql.Tuple{
  1369. {
  1370. Emitter: "src2",
  1371. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1372. }, {
  1373. Emitter: "src2",
  1374. Message: xsql.Message{"id2": 2, "f2": "w2"},
  1375. }, {
  1376. Emitter: "src2",
  1377. Message: xsql.Message{"id2": 4, "f2": "w3"},
  1378. },
  1379. },
  1380. },
  1381. },
  1382. result: xsql.JoinTupleSets{
  1383. xsql.JoinTuple{
  1384. Tuples: []xsql.Tuple{
  1385. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1386. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1387. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1388. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1389. },
  1390. },
  1391. xsql.JoinTuple{
  1392. Tuples: []xsql.Tuple{
  1393. {Emitter: "src1", Message: xsql.Message{"id1": 2, "f1": "v2"}},
  1394. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1395. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1396. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1397. },
  1398. },
  1399. xsql.JoinTuple{
  1400. Tuples: []xsql.Tuple{
  1401. {Emitter: "src1", Message: xsql.Message{"id1": 3, "f1": "v3"}},
  1402. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1403. {Emitter: "src2", Message: xsql.Message{"id2": 2, "f2": "w2"}},
  1404. {Emitter: "src2", Message: xsql.Message{"id2": 4, "f2": "w3"}},
  1405. },
  1406. },
  1407. },
  1408. },
  1409. {
  1410. sql: "SELECT id1 FROM src1 cross join src2",
  1411. data: xsql.WindowTuplesSet{
  1412. xsql.WindowTuples{
  1413. Emitter: "src1",
  1414. Tuples: []xsql.Tuple{
  1415. {
  1416. Emitter: "src1",
  1417. Message: xsql.Message{"id1": 1, "f1": "v1"},
  1418. },
  1419. },
  1420. },
  1421. xsql.WindowTuples{
  1422. Emitter: "src2",
  1423. Tuples: []xsql.Tuple{
  1424. {
  1425. Emitter: "src2",
  1426. Message: xsql.Message{"id2": 1, "f2": "w1"},
  1427. }, {
  1428. Emitter: "src2",
  1429. Message: xsql.Message{"id2": 1, "f2": "w2"},
  1430. },
  1431. },
  1432. },
  1433. },
  1434. result: xsql.JoinTupleSets{
  1435. xsql.JoinTuple{
  1436. Tuples: []xsql.Tuple{
  1437. {Emitter: "src1", Message: xsql.Message{"id1": 1, "f1": "v1"}},
  1438. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w1"}},
  1439. {Emitter: "src2", Message: xsql.Message{"id2": 1, "f2": "w2"}},
  1440. },
  1441. },
  1442. },
  1443. },
  1444. }
  1445. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  1446. contextLogger := common.Log.WithField("rule", "TestCrossJoinPlan_Apply")
  1447. ctx := contexts.WithValue(contexts.Background(), contexts.LoggerKey, contextLogger)
  1448. for i, tt := range tests {
  1449. stmt, err := xsql.NewParser(strings.NewReader(tt.sql)).Parse()
  1450. if err != nil {
  1451. t.Errorf("statement parse error %s", err)
  1452. break
  1453. }
  1454. if table, ok := stmt.Sources[0].(*xsql.Table); !ok {
  1455. t.Errorf("statement source is not a table")
  1456. } else {
  1457. pp := &JoinPlan{Joins: stmt.Joins, From: table}
  1458. result := pp.Apply(ctx, tt.data)
  1459. if !reflect.DeepEqual(tt.result, result) {
  1460. t.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.sql, tt.result, result)
  1461. }
  1462. }
  1463. }
  1464. }
  1465. func str2Map(s string) map[string]interface{} {
  1466. var input map[string]interface{}
  1467. if err := json.Unmarshal([]byte(s), &input); err != nil {
  1468. fmt.Printf("Failed to parse the JSON data.\n")
  1469. return nil
  1470. }
  1471. return input
  1472. }