project_test.go 61 KB

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