parser_test.go 40 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740
  1. package xsql
  2. import (
  3. "fmt"
  4. "reflect"
  5. "strings"
  6. "testing"
  7. )
  8. // Ensure the parser can parse strings into Statement ASTs.
  9. func TestParser_ParseStatement(t *testing.T) {
  10. var tests = []struct {
  11. s string
  12. stmt *SelectStatement
  13. err string
  14. }{
  15. {
  16. s: `SELECT name FROM tbl`,
  17. stmt: &SelectStatement{
  18. Fields: []Field{
  19. Field{
  20. Expr:&FieldRef{Name: "name"},
  21. Name: "name",
  22. AName:""},
  23. },
  24. Sources: []Source{&Table{Name:"tbl"}},
  25. },
  26. },
  27. {
  28. s: `SELECT name FROM topic/sensor1`,
  29. stmt: &SelectStatement{
  30. Fields: []Field{
  31. Field{
  32. Expr:&FieldRef{Name: "name"},
  33. Name: "name",
  34. AName:""},
  35. },
  36. Sources: []Source{&Table{Name:"topic/sensor1"}},
  37. },
  38. },
  39. {
  40. s: `SELECT t1.name FROM topic/sensor1 AS t1`,
  41. stmt: &SelectStatement{
  42. Fields: []Field{
  43. Field{
  44. Expr:&FieldRef{StreamName: StreamName("t1"), Name: "name"},
  45. Name: "name",
  46. AName:""},
  47. },
  48. Sources: []Source{&Table{Name:"topic/sensor1", Alias:"t1"}},
  49. },
  50. },
  51. {
  52. s: `SELECT name FROM topic/sensor1 AS t1`,
  53. stmt: &SelectStatement{
  54. Fields: []Field{
  55. Field{
  56. Expr:&FieldRef{Name: "name"},
  57. Name: "name",
  58. AName:""},
  59. },
  60. Sources: []Source{&Table{Name:"topic/sensor1", Alias:"t1"}},
  61. },
  62. },
  63. {
  64. s: `SELECT name FROM topic/sensor1/#`,
  65. stmt: &SelectStatement{
  66. Fields: []Field{
  67. Field{
  68. Expr:&FieldRef{Name: "name"},
  69. Name: "name",
  70. AName:""},
  71. },
  72. Sources: []Source{&Table{Name:"topic/sensor1/#"}},
  73. },
  74. },
  75. {
  76. s: `SELECT name FROM topic/sensor1/# AS t2 `,
  77. stmt: &SelectStatement{
  78. Fields: []Field{
  79. Field{
  80. Expr:&FieldRef{Name: "name"},
  81. Name: "name",
  82. AName:""},
  83. },
  84. Sources: []Source{&Table{Name:"topic/sensor1/#", Alias:"t2"}},
  85. },
  86. },
  87. {
  88. s: `SELECT name FROM /topic/sensor1/#`,
  89. stmt: &SelectStatement{
  90. Fields: []Field{
  91. Field{
  92. Expr:&FieldRef{Name: "name"},
  93. Name: "name",
  94. AName:""},
  95. },
  96. Sources: []Source{&Table{Name:"/topic/sensor1/#"}},
  97. },
  98. },
  99. {
  100. s: `SELECT name FROM /topic/sensor1/#/`,
  101. stmt: &SelectStatement{
  102. Fields: []Field{
  103. Field{
  104. Expr:&FieldRef{Name: "name"},
  105. Name: "name",
  106. AName:""},
  107. },
  108. Sources: []Source{&Table{Name:"/topic/sensor1/#/"}},
  109. },
  110. },
  111. {
  112. s: `SELECT name FROM /topic/sensor1/+/temp1/`,
  113. stmt: &SelectStatement{
  114. Fields: []Field{
  115. Field{
  116. Expr:&FieldRef{Name: "name"},
  117. Name: "name",
  118. AName:""},
  119. },
  120. Sources: []Source{&Table{Name:"/topic/sensor1/+/temp1/"}},
  121. },
  122. },
  123. {
  124. s: `SELECT name FROM topic/sensor1/+/temp`,
  125. stmt: &SelectStatement{
  126. Fields: []Field{
  127. Field{
  128. Expr:&FieldRef{Name: "name"},
  129. Name: "name",
  130. AName:""},
  131. },
  132. Sources: []Source{&Table{Name:"topic/sensor1/+/temp"}},
  133. },
  134. },
  135. {
  136. s: `SELECT * FROM tbl`,
  137. stmt: &SelectStatement{
  138. Fields: []Field{
  139. Field{
  140. Expr: &Wildcard{Token: ASTERISK},
  141. Name: "",
  142. AName:""},
  143. },
  144. Sources: []Source{&Table{Name:"tbl"}},
  145. },
  146. },
  147. {
  148. s: `SELECT a,b FROM tbl`,
  149. stmt: &SelectStatement{
  150. Fields: []Field{
  151. Field{Expr:&FieldRef{Name: "a"}, Name: "a", AName:""},
  152. Field{Expr:&FieldRef{Name: "b"}, Name: "b", AName:""},
  153. },
  154. Sources: []Source{&Table{Name:"tbl"}},
  155. },
  156. },
  157. {
  158. s: `SELECT a, b,c FROM tbl`,
  159. stmt: &SelectStatement{
  160. Fields: []Field{
  161. Field{Expr:&FieldRef{Name: "a"}, Name: "a", AName:""},
  162. Field{Expr:&FieldRef{Name: "b"}, Name: "b", AName:""},
  163. Field{Expr:&FieldRef{Name: "c"}, Name: "c", AName:""},
  164. },
  165. Sources: []Source{&Table{Name:"tbl"}},
  166. },
  167. },
  168. {
  169. s: `SELECT a AS alias FROM tbl`,
  170. stmt: &SelectStatement{
  171. Fields: []Field{Field{Expr:&FieldRef{Name: "a"}, Name: "a", AName:"alias"},},
  172. Sources: []Source{&Table{Name:"tbl"}},
  173. },
  174. },
  175. {
  176. s: `SELECT a AS alias1, b as Alias2 FROM tbl`,
  177. stmt: &SelectStatement{
  178. Fields: []Field{
  179. Field{Expr: &FieldRef{Name:"a"}, Name: "a", AName:"alias1"},
  180. Field{Expr: &FieldRef{Name:"b"}, Name: "b", AName:"Alias2"},
  181. },
  182. Sources: []Source{&Table{Name:"tbl"}},
  183. },
  184. },
  185. {
  186. s: `SELECT sample0("test") FROM tbl`,
  187. stmt: &SelectStatement{
  188. Fields: []Field{
  189. Field{
  190. AName:"",
  191. Name: "sample0",
  192. Expr:&Call{
  193. Name:"sample0",
  194. Args: []Expr{&StringLiteral{Val:"test"}},
  195. },
  196. },
  197. },
  198. Sources: []Source{&Table{Name:"tbl"}},
  199. },
  200. },
  201. {
  202. s: `SELECT sample1(test) FROM tbl`,
  203. stmt: &SelectStatement{
  204. Fields: []Field{
  205. Field{
  206. AName:"",
  207. Name: "sample1",
  208. Expr:&Call{
  209. Name:"sample1",
  210. Args: []Expr{&FieldRef{Name:"test"}},
  211. },
  212. },
  213. },
  214. Sources: []Source{&Table{Name:"tbl"}},
  215. },
  216. },
  217. {
  218. s: `SELECT sample2(123) FROM tbl`,
  219. stmt: &SelectStatement{
  220. Fields: []Field{
  221. Field{
  222. AName:"",
  223. Name: "sample2",
  224. Expr:&Call{
  225. Name:"sample2",
  226. Args: []Expr{&IntegerLiteral{Val:123}},
  227. },
  228. },
  229. },
  230. Sources: []Source{&Table{Name:"tbl"}},
  231. },
  232. },
  233. {
  234. s: `SELECT sample3(123, "abc") FROM tbl`,
  235. stmt: &SelectStatement{
  236. Fields: []Field{
  237. Field{
  238. AName:"",
  239. Name: "sample3",
  240. Expr:&Call{
  241. Name:"sample3",
  242. Args: []Expr{&IntegerLiteral{Val:123}, &StringLiteral{Val:"abc"}},
  243. },
  244. },
  245. },
  246. Sources: []Source{&Table{Name:"tbl"}},
  247. },
  248. },
  249. {
  250. s: `SELECT sample3() FROM tbl`,
  251. stmt: &SelectStatement{
  252. Fields: []Field{
  253. Field{
  254. AName:"",
  255. Name: "sample3",
  256. Expr:&Call{
  257. Name:"sample3",
  258. Args: nil ,
  259. },
  260. },
  261. },
  262. Sources: []Source{&Table{Name:"tbl"}},
  263. },
  264. },
  265. {
  266. s: `SELECT sample4("abc", 1234, field1) FROM tbl`,
  267. stmt: &SelectStatement{
  268. Fields: []Field{
  269. Field{
  270. AName:"",
  271. Name: "sample4",
  272. Expr:&Call{
  273. Name:"sample4",
  274. Args: []Expr{
  275. &StringLiteral{Val:"abc"},
  276. &IntegerLiteral{Val:1234},
  277. &FieldRef{Name:"field1"},
  278. } ,
  279. },
  280. },
  281. },
  282. Sources: []Source{&Table{Name:"tbl"}},
  283. },
  284. },
  285. {
  286. s: `SELECT sample2(sample1(test),1) FROM tbl`,
  287. stmt: &SelectStatement{
  288. Fields: []Field{
  289. Field{
  290. AName:"",
  291. Name: "sample2",
  292. Expr:&Call{
  293. Name:"sample2",
  294. Args: []Expr{
  295. &Call{
  296. Name:"sample1",
  297. Args:[]Expr{
  298. &FieldRef{Name: "test"},
  299. },
  300. },
  301. &IntegerLiteral{Val:1},
  302. },
  303. },
  304. },
  305. },
  306. Sources: []Source{&Table{Name:"tbl"}},
  307. },
  308. },
  309. {
  310. s: `SELECT sample2(sample1(test),1) AS field1 FROM tbl`,
  311. stmt: &SelectStatement{
  312. Fields: []Field{
  313. Field{
  314. AName:"field1",
  315. Name: "sample2",
  316. Expr:&Call{
  317. Name:"sample2",
  318. Args: []Expr{
  319. &Call{
  320. Name:"sample1",
  321. Args:[]Expr{
  322. &FieldRef{Name: "test"},
  323. },
  324. },
  325. &IntegerLiteral{Val:1},
  326. },
  327. },
  328. },
  329. },
  330. Sources: []Source{&Table{Name:"tbl"}},
  331. },
  332. },
  333. {
  334. s: `SELECT sample2(sample1("test")) FROM tbl`,
  335. stmt: &SelectStatement{
  336. Fields: []Field{
  337. Field{
  338. AName:"",
  339. Name: "sample2",
  340. Expr:&Call{
  341. Name:"sample2",
  342. Args: []Expr{
  343. &Call{
  344. Name:"sample1",
  345. Args:[]Expr{
  346. &StringLiteral{Val: "test"},
  347. },
  348. },
  349. },
  350. },
  351. },
  352. },
  353. Sources: []Source{&Table{Name:"tbl"}},
  354. },
  355. },
  356. {
  357. s: `SELECT "abc" FROM tbl`,
  358. stmt: &SelectStatement{
  359. Fields: []Field{Field{AName:"", Name: "", Expr: &StringLiteral{Val:"abc"}}},
  360. Sources: []Source{&Table{Name:"tbl"}},
  361. },
  362. },
  363. {
  364. s: `SELECT "abc" AS field1 FROM tbl`,
  365. stmt: &SelectStatement{
  366. Fields: []Field{Field{AName:"field1", Name: "", Expr: &StringLiteral{Val:"abc"}}},
  367. Sources: []Source{&Table{Name:"tbl"}},
  368. },
  369. },
  370. {
  371. s: `SELECT field0, "abc" AS field1, field2 FROM tbl`,
  372. stmt: &SelectStatement{
  373. Fields: []Field{
  374. Field{AName:"", Name: "field0", Expr: &FieldRef{Name:"field0"}},
  375. Field{AName:"field1", Name: "", Expr: &StringLiteral{Val:"abc"}},
  376. Field{AName:"", Name: "field2", Expr: &FieldRef{Name:"field2"}},},
  377. Sources: []Source{&Table{Name:"tbl"}},
  378. },
  379. },
  380. {
  381. s: `SELECT * AS alias FROM tbl`,
  382. stmt: nil,
  383. err: `found "AS", expected FROM.`,
  384. },
  385. {
  386. s: `SELECT *, FROM tbl`,
  387. stmt: nil,
  388. err: `found ",", expected FROM.`,
  389. },
  390. {
  391. s: `SELECTname FROM tbl`,
  392. stmt: nil,
  393. err: `Found "SELECTname", Expected SELECT.` + "\n",
  394. },
  395. {
  396. s: `SELECT abc FROM tbl WHERE abc > 12 `,
  397. stmt: &SelectStatement{
  398. Fields: []Field{Field{AName:"", Name: "abc", Expr: &FieldRef{Name:"abc"}}},
  399. Sources: []Source{&Table{Name:"tbl"}},
  400. Condition: &BinaryExpr{
  401. LHS:&FieldRef{Name:"abc"},
  402. OP:GT,
  403. RHS:&IntegerLiteral{Val:12},
  404. },
  405. },
  406. },
  407. {
  408. s: `SELECT abc FROM tbl WHERE abc = "hello" `,
  409. stmt: &SelectStatement{
  410. Fields: []Field{Field{AName:"", Name: "abc", Expr: &FieldRef{Name:"abc"}}},
  411. Sources: []Source{&Table{Name:"tbl"}},
  412. Condition: &BinaryExpr{
  413. LHS:&FieldRef{Name:"abc"},
  414. OP:EQ,
  415. RHS:&StringLiteral{Val:"hello"},
  416. },
  417. },
  418. },
  419. {
  420. s: `SELECT t1.abc FROM tbl AS t1 WHERE t1.abc = "hello" `,
  421. stmt: &SelectStatement{
  422. Fields: []Field{Field{AName:"", Name: "abc", Expr: &FieldRef{StreamName:StreamName("t1"), Name:"abc"}}},
  423. Sources: []Source{&Table{Name:"tbl", Alias:"t1"}},
  424. Condition: &BinaryExpr{
  425. LHS:&FieldRef{StreamName:StreamName("t1"), Name:"abc"},
  426. OP:EQ,
  427. RHS:&StringLiteral{Val:"hello"},
  428. },
  429. },
  430. },
  431. {
  432. s: `SELECT abc, "fff" AS fa FROM tbl WHERE fa >= 5 `,
  433. stmt: &SelectStatement{
  434. Fields: []Field{Field{AName:"", Name: "abc", Expr: &FieldRef{Name:"abc"}}, Field{AName:"fa", Name: "", Expr: &StringLiteral{Val:"fff"}}},
  435. Sources: []Source{&Table{Name:"tbl"}},
  436. Condition: &BinaryExpr{
  437. LHS:&FieldRef{Name:"fa"},
  438. OP:GTE,
  439. RHS:&IntegerLiteral{Val:5},
  440. },
  441. },
  442. },
  443. {
  444. s: `SELECT field2 FROM tbl WHERE field2 != 5 `,
  445. stmt: &SelectStatement{
  446. Fields: []Field{Field{AName:"", Name: "field2", Expr: &FieldRef{Name:"field2"}}, },
  447. Sources: []Source{&Table{Name:"tbl"}},
  448. Condition: &BinaryExpr{
  449. LHS:&FieldRef{Name:"field2"},
  450. OP:NEQ,
  451. RHS:&IntegerLiteral{Val:5},
  452. },
  453. },
  454. },
  455. {
  456. s: `SELECT field2 FROM tbl WHERE field2 ! = 5 `, //Add space char in expression
  457. stmt: &SelectStatement{
  458. Fields: []Field{Field{AName:"", Name: "field2", Expr: &FieldRef{Name:"field2"}}, },
  459. Sources: []Source{&Table{Name:"tbl"}},
  460. Condition: &BinaryExpr{
  461. LHS:&FieldRef{Name:"field2"},
  462. OP:NEQ,
  463. RHS:&IntegerLiteral{Val:5},
  464. },
  465. },
  466. },
  467. {
  468. s: `SELECT *f FROM tbl`,
  469. stmt: nil,
  470. err: `found "f", expected FROM.`,
  471. },
  472. ////TODO
  473. //{
  474. // s: `SELECT *from FROM tbl`,
  475. // stmt: nil,
  476. // err: `found "f", expected FROM.`,
  477. //},
  478. {
  479. s: `SELECT abc+2 FROM tbl`,
  480. stmt: &SelectStatement{
  481. Fields: []Field{
  482. Field{
  483. AName:"",
  484. Name: "",
  485. Expr: &BinaryExpr{
  486. LHS:&FieldRef{Name:"abc"},
  487. OP:ADD,
  488. RHS:&IntegerLiteral{Val:2},
  489. },
  490. },
  491. },
  492. Sources: []Source{&Table{Name:"tbl"}},
  493. },
  494. },
  495. {
  496. s: `SELECT t1.abc+2 FROM tbl AS t1`,
  497. stmt: &SelectStatement{
  498. Fields: []Field{
  499. Field{
  500. AName:"",
  501. Name: "",
  502. Expr: &BinaryExpr{
  503. LHS:&FieldRef{StreamName:StreamName("t1"), Name:"abc"},
  504. OP:ADD,
  505. RHS:&IntegerLiteral{Val:2},
  506. },
  507. },
  508. },
  509. Sources: []Source{&Table{Name:"tbl", Alias:"t1"}},
  510. },
  511. },
  512. {
  513. s: `SELECT abc + "hello" FROM tbl`,
  514. stmt: &SelectStatement{
  515. Fields: []Field{
  516. Field{
  517. AName:"",
  518. Name: "",
  519. Expr: &BinaryExpr{
  520. LHS:&FieldRef{Name:"abc"},
  521. OP:ADD,
  522. RHS:&StringLiteral{Val:"hello"},
  523. },
  524. },
  525. },
  526. Sources: []Source{&Table{Name:"tbl"}},
  527. },
  528. },
  529. {
  530. s: `SELECT abc*2 + 3 FROM tbl`,
  531. stmt: &SelectStatement{
  532. Fields: []Field{
  533. Field{
  534. AName:"",
  535. Name: "",
  536. Expr: &BinaryExpr{
  537. LHS: &BinaryExpr{
  538. LHS:&FieldRef{Name:"abc"},
  539. OP:MUL,
  540. RHS:&IntegerLiteral{Val:2},
  541. },
  542. OP:ADD,
  543. RHS:&IntegerLiteral{Val:3},
  544. },
  545. },
  546. },
  547. Sources: []Source{&Table{Name:"tbl"}},
  548. },
  549. },
  550. {
  551. s: `SELECT sample(abc*2 + 3) FROM tbl`,
  552. stmt: &SelectStatement{
  553. Fields: []Field{
  554. Field{
  555. AName:"",
  556. Name: "sample",
  557. Expr: &Call{
  558. Name:"sample",
  559. Args: []Expr{
  560. &BinaryExpr{
  561. LHS: &BinaryExpr{
  562. LHS:&FieldRef{Name:"abc"},
  563. OP:MUL,
  564. RHS:&IntegerLiteral{Val:2},
  565. },
  566. OP:ADD,
  567. RHS:&IntegerLiteral{Val:3},
  568. },
  569. },
  570. },
  571. },
  572. },
  573. Sources: []Source{&Table{Name:"tbl"}},
  574. },
  575. },
  576. {
  577. s: `SELECT sample(t1.abc*2 + 3) FROM tbl AS t1`,
  578. stmt: &SelectStatement{
  579. Fields: []Field{
  580. Field{
  581. AName:"",
  582. Name: "sample",
  583. Expr: &Call{
  584. Name:"sample",
  585. Args: []Expr{
  586. &BinaryExpr{
  587. LHS: &BinaryExpr{
  588. LHS:&FieldRef{StreamName:StreamName("t1"), Name:"abc"},
  589. OP:MUL,
  590. RHS:&IntegerLiteral{Val:2},
  591. },
  592. OP:ADD,
  593. RHS:&IntegerLiteral{Val:3},
  594. },
  595. },
  596. },
  597. },
  598. },
  599. Sources: []Source{&Table{Name:"tbl", Alias:"t1"}},
  600. },
  601. },
  602. {
  603. s: `SELECT sample(abc*2 + 3, "param2") FROM tbl`,
  604. stmt: &SelectStatement{
  605. Fields: []Field{
  606. Field{
  607. AName:"",
  608. Name: "sample",
  609. Expr: &Call{
  610. Name:"sample",
  611. Args: []Expr{
  612. &BinaryExpr{
  613. LHS: &BinaryExpr{
  614. LHS:&FieldRef{Name:"abc"},
  615. OP:MUL,
  616. RHS:&IntegerLiteral{Val:2},
  617. },
  618. OP:ADD,
  619. RHS:&IntegerLiteral{Val:3},
  620. },
  621. &StringLiteral{Val:"param2"},
  622. },
  623. },
  624. },
  625. },
  626. Sources: []Source{&Table{Name:"tbl"}},
  627. },
  628. },
  629. {
  630. s: `SELECT 0.2 FROM tbl`,
  631. stmt: &SelectStatement{
  632. Fields: []Field{
  633. Field{
  634. AName:"",
  635. Name: "",
  636. Expr: &NumberLiteral{Val: 0.2},
  637. },
  638. },
  639. Sources: []Source{&Table{Name:"tbl"}},
  640. },
  641. },
  642. {
  643. s: `SELECT .2 FROM tbl`,
  644. stmt: &SelectStatement{
  645. Fields: []Field{
  646. Field{
  647. AName:"",
  648. Name: "",
  649. Expr: &NumberLiteral{Val: 0.2},
  650. },
  651. },
  652. Sources: []Source{&Table{Name:"tbl"}},
  653. },
  654. },
  655. {
  656. s: `SELECT sample(.2) FROM tbl`,
  657. stmt: &SelectStatement{
  658. Fields: []Field{
  659. Field{
  660. AName:"",
  661. Name: "sample",
  662. Expr:&Call{
  663. Name:"sample",
  664. Args: []Expr{&NumberLiteral{Val:0.2}},
  665. },
  666. },
  667. },
  668. Sources: []Source{&Table{Name:"tbl"}},
  669. },
  670. },
  671. {
  672. s: `SELECT sample(.2, "abc") FROM tbl`,
  673. stmt: &SelectStatement{
  674. Fields: []Field{
  675. Field{
  676. AName:"",
  677. Name: "sample",
  678. Expr:&Call{
  679. Name:"sample",
  680. Args: []Expr{&NumberLiteral{Val:0.2}, &StringLiteral{Val: "abc"}},
  681. },
  682. },
  683. },
  684. Sources: []Source{&Table{Name:"tbl"}},
  685. },
  686. },
  687. {
  688. s: `SELECT sample(.2, "abc") AS f1 FROM tbl WHERE f1 > 2.2`,
  689. stmt: &SelectStatement{
  690. Fields: []Field{
  691. Field{
  692. AName:"f1",
  693. Name: "sample",
  694. Expr:&Call{
  695. Name:"sample",
  696. Args: []Expr{&NumberLiteral{Val:0.2}, &StringLiteral{Val: "abc"}},
  697. },
  698. },
  699. },
  700. Sources: []Source{&Table{Name:"tbl"}},
  701. Condition: &BinaryExpr{
  702. LHS: &FieldRef{Name:"f1"},
  703. OP: GT,
  704. RHS: &NumberLiteral{Val:2.2},
  705. },
  706. },
  707. },
  708. {
  709. s: `SELECT deviceId, name FROM topic/sensor1 WHERE deviceId=1 AND name = "dname"`,
  710. stmt: &SelectStatement{
  711. Fields: []Field{
  712. Field{ Expr:&FieldRef{Name: "deviceId"}, Name: "deviceId", AName:""},
  713. Field{ Expr:&FieldRef{Name: "name"}, Name: "name", AName:""},
  714. },
  715. Sources: []Source{&Table{Name:"topic/sensor1"}},
  716. Condition: &BinaryExpr{
  717. LHS: &BinaryExpr{LHS: &FieldRef{Name: "deviceId"}, OP: EQ, RHS: &IntegerLiteral{Val: 1},},
  718. OP: AND,
  719. RHS: &BinaryExpr{LHS: &FieldRef{Name: "name"}, OP: EQ, RHS: &StringLiteral{Val: "dname"},},
  720. },
  721. },
  722. },
  723. {
  724. s: `SELECT deviceId, name FROM topic/sensor1 AS t1 WHERE t1.deviceId=1 AND t1.name = "dname"`,
  725. stmt: &SelectStatement{
  726. Fields: []Field{
  727. Field{ Expr:&FieldRef{Name: "deviceId"}, Name: "deviceId", AName:""},
  728. Field{ Expr:&FieldRef{Name: "name"}, Name: "name", AName:""},
  729. },
  730. Sources: []Source{&Table{Name:"topic/sensor1", Alias:"t1"}},
  731. Condition: &BinaryExpr{
  732. LHS: &BinaryExpr{LHS: &FieldRef{StreamName:StreamName("t1"), Name: "deviceId"}, OP: EQ, RHS: &IntegerLiteral{Val: 1},},
  733. OP: AND,
  734. RHS: &BinaryExpr{LHS: &FieldRef{StreamName:StreamName("t1"), Name: "name"}, OP: EQ, RHS: &StringLiteral{Val: "dname"},},
  735. },
  736. },
  737. },
  738. {
  739. s: `SELECT temp AS t, name FROM topic/sensor1 WHERE t> = 20.5 OR name = "dname"`,
  740. stmt: &SelectStatement{
  741. Fields: []Field{
  742. Field{ Expr:&FieldRef{Name: "temp"}, Name: "temp", AName:"t"},
  743. Field{ Expr:&FieldRef{Name: "name"}, Name: "name", AName:""},
  744. },
  745. Sources: []Source{&Table{Name:"topic/sensor1"}},
  746. Condition: &BinaryExpr{
  747. LHS: &BinaryExpr{LHS: &FieldRef{Name: "t"}, OP: GTE, RHS: &NumberLiteral{Val: 20.5},},
  748. OP: OR,
  749. RHS: &BinaryExpr{LHS: &FieldRef{Name: "name"}, OP: EQ, RHS: &StringLiteral{Val: "dname"},},
  750. },
  751. },
  752. },
  753. {
  754. s: `SELECT temp AS t, name FROM topic/sensor1 WHERE name = "dname" GROUP BY name`,
  755. stmt: &SelectStatement{
  756. Fields: []Field{
  757. Field{ Expr:&FieldRef{Name: "temp"}, Name: "temp", AName:"t"},
  758. Field{ Expr:&FieldRef{Name: "name"}, Name: "name", AName:""},
  759. },
  760. Sources: []Source{&Table{Name:"topic/sensor1"}},
  761. Condition: &BinaryExpr{LHS: &FieldRef{Name: "name"}, OP: EQ, RHS: &StringLiteral{Val: "dname"},},
  762. Dimensions:Dimensions{Dimension{Expr:&FieldRef{Name:"name"}}},
  763. },
  764. },
  765. {
  766. s: `SELECT s1.temp AS t, name FROM topic/sensor1 AS s1 WHERE t = "dname" GROUP BY s1.temp`,
  767. stmt: &SelectStatement{
  768. Fields: []Field{
  769. Field{ Expr:&FieldRef{StreamName:"s1", Name: "temp"}, Name: "temp", AName:"t"},
  770. Field{ Expr:&FieldRef{Name: "name"}, Name: "name", AName:""},
  771. },
  772. Sources: []Source{&Table{Name:"topic/sensor1", Alias:"s1"}},
  773. Condition: &BinaryExpr{LHS: &FieldRef{Name: "t"}, OP: EQ, RHS: &StringLiteral{Val: "dname"},},
  774. Dimensions:Dimensions{Dimension{Expr:&FieldRef{StreamName:"s1", Name: "temp"}}},
  775. },
  776. },
  777. {
  778. s: `SELECT temp AS t, name FROM topic/sensor1 WHERE name = "dname" GROUP BY sample(name,1)`,
  779. stmt: &SelectStatement{
  780. Fields: []Field{
  781. Field{ Expr:&FieldRef{Name: "temp"}, Name: "temp", AName:"t"},
  782. Field{ Expr:&FieldRef{Name: "name"}, Name: "name", AName:""},
  783. },
  784. Sources: []Source{&Table{Name:"topic/sensor1"}},
  785. Condition: &BinaryExpr{LHS: &FieldRef{Name: "name"}, OP: EQ, RHS: &StringLiteral{Val: "dname"},},
  786. Dimensions:Dimensions{Dimension{
  787. Expr:&Call{Name:"sample", Args:[]Expr{&FieldRef{Name:"name"}, &IntegerLiteral{Val:1}}},
  788. },
  789. },
  790. },
  791. },
  792. {
  793. s: `SELECT temp AS t, name FROM topic/sensor1 AS s1 WHERE name = "dname" GROUP BY sample(s1.name,1)`,
  794. stmt: &SelectStatement{
  795. Fields: []Field{
  796. Field{ Expr:&FieldRef{Name: "temp"}, Name: "temp", AName:"t"},
  797. Field{ Expr:&FieldRef{Name: "name"}, Name: "name", AName:""},
  798. },
  799. Sources: []Source{&Table{Name:"topic/sensor1", Alias:"s1"}},
  800. Condition: &BinaryExpr{LHS: &FieldRef{Name: "name"}, OP: EQ, RHS: &StringLiteral{Val: "dname"},},
  801. Dimensions:Dimensions{Dimension{
  802. Expr:&Call{Name:"sample", Args:[]Expr{&FieldRef{StreamName:StreamName("s1"), Name:"name"}, &IntegerLiteral{Val:1}}},
  803. },
  804. },
  805. },
  806. },
  807. {
  808. s: `SELECT temp AS t, name FROM topic/sensor1 WHERE name = "dname" GROUP BY sample(name,1) ORDER BY name`,
  809. stmt: &SelectStatement{
  810. Fields: []Field{
  811. Field{ Expr:&FieldRef{Name: "temp"}, Name: "temp", AName:"t"},
  812. Field{ Expr:&FieldRef{Name: "name"}, Name: "name", AName:""},
  813. },
  814. Sources: []Source{&Table{Name:"topic/sensor1"}},
  815. Condition: &BinaryExpr{LHS: &FieldRef{Name: "name"}, OP: EQ, RHS: &StringLiteral{Val: "dname"},},
  816. Dimensions:Dimensions{
  817. Dimension{
  818. Expr:&Call{Name:"sample", Args:[]Expr{
  819. &FieldRef{Name:"name"},
  820. &IntegerLiteral{Val:1}},
  821. },
  822. },
  823. },
  824. SortFields: []SortField {SortField{Name:"name", Ascending:true}, },
  825. },
  826. },
  827. {
  828. s: `SELECT temp AS t, name FROM topic/sensor1 AS s1 WHERE s1.name = "dname" GROUP BY sample(s1.name,1) ORDER BY s1.name`,
  829. stmt: &SelectStatement{
  830. Fields: []Field{
  831. Field{ Expr:&FieldRef{Name: "temp"}, Name: "temp", AName:"t"},
  832. Field{ Expr:&FieldRef{Name: "name"}, Name: "name", AName:""},
  833. },
  834. Sources: []Source{&Table{Name:"topic/sensor1", Alias:"s1"}},
  835. Condition: &BinaryExpr{LHS: &FieldRef{StreamName:StreamName("s1"), Name: "name"}, OP: EQ, RHS: &StringLiteral{Val: "dname"},},
  836. Dimensions:Dimensions{
  837. Dimension{
  838. Expr:&Call{Name:"sample", Args:[]Expr{
  839. &FieldRef{StreamName:StreamName("s1"), Name:"name"},
  840. &IntegerLiteral{Val:1}},
  841. },
  842. },
  843. },
  844. SortFields: []SortField {SortField{Name:"s1.name", Ascending:true}, },
  845. },
  846. },
  847. {
  848. s: `SELECT temp AS t, name FROM topic/sensor1 WHERE name = "dname" GROUP BY sample(name,1) ORDER BY name DESC`,
  849. stmt: &SelectStatement{
  850. Fields: []Field{
  851. Field{ Expr:&FieldRef{Name: "temp"}, Name: "temp", AName:"t"},
  852. Field{ Expr:&FieldRef{Name: "name"}, Name: "name", AName:""},
  853. },
  854. Sources: []Source{&Table{Name:"topic/sensor1"}},
  855. Condition: &BinaryExpr{LHS: &FieldRef{Name: "name"}, OP: EQ, RHS: &StringLiteral{Val: "dname"},},
  856. Dimensions:Dimensions{
  857. Dimension{
  858. Expr:&Call{Name:"sample", Args:[]Expr{
  859. &FieldRef{Name:"name"},
  860. &IntegerLiteral{Val:1}},
  861. },
  862. },
  863. },
  864. SortFields: []SortField {SortField{Name:"name", Ascending:false}, },
  865. },
  866. },
  867. {
  868. s: `SELECT * FROM topic/sensor1 ORDER BY name DESC`,
  869. stmt: &SelectStatement{
  870. Fields: []Field{
  871. Field{
  872. Expr: &Wildcard{Token: ASTERISK},
  873. Name: "",
  874. AName:""},
  875. },
  876. Sources: []Source{&Table{Name:"topic/sensor1"}},
  877. SortFields: []SortField {SortField{Name:"name", Ascending:false}, },
  878. },
  879. },
  880. {
  881. s: `SELECT * FROM topic/sensor1 ORDER BY name DESC, name2 ASC`,
  882. stmt: &SelectStatement{
  883. Fields: []Field{
  884. Field{
  885. Expr: &Wildcard{Token: ASTERISK},
  886. Name: "",
  887. AName:""},
  888. },
  889. Sources: []Source{&Table{Name:"topic/sensor1"}},
  890. SortFields: []SortField {SortField{Name:"name", Ascending:false}, SortField{Name:"name2", Ascending:true},},
  891. },
  892. },
  893. {
  894. s: `SELECT * FROM topic/sensor1 GROUP BY name, name2,sample(name3,1.8) ORDER BY name DESC, name2 ASC`,
  895. stmt: &SelectStatement{
  896. Fields: []Field{
  897. Field{
  898. Expr: &Wildcard{Token: ASTERISK},
  899. Name: "",
  900. AName:""},
  901. },
  902. Sources: []Source{&Table{Name:"topic/sensor1"}},
  903. Dimensions: Dimensions{
  904. Dimension{Expr:&FieldRef{Name:"name"}},
  905. Dimension{Expr:&FieldRef{Name:"name2"}},
  906. Dimension{
  907. Expr:&Call{Name:"sample", Args:[]Expr{
  908. &FieldRef{Name:"name3"},
  909. &NumberLiteral{Val:1.8}},
  910. },
  911. },
  912. },
  913. SortFields: []SortField {SortField{Name:"name", Ascending:false}, SortField{Name:"name2", Ascending:true},},
  914. },
  915. },
  916. //{
  917. // s: `SELECT .2sd FROM tbl`,
  918. // stmt: &SelectStatement{
  919. // Fields: []Field{
  920. // Field{
  921. // AName:"",
  922. // Expr: &NumberLiteral{Val: 0.2},
  923. // },
  924. // },
  925. // TableName: "tbl",
  926. // },
  927. //},
  928. {
  929. s: `SELECT name FROM tbl/*SELECT comment FROM testComments*/`,
  930. stmt: &SelectStatement{
  931. Fields: []Field{
  932. Field{
  933. Expr:&FieldRef{Name: "name"},
  934. Name: "name",
  935. AName:""},
  936. },
  937. Sources: []Source{&Table{Name:"tbl"}},
  938. },
  939. },
  940. {
  941. s: `/*SELECT comment FROM testComments*/SELECT name FROM tbl`,
  942. stmt: &SelectStatement{
  943. Fields: []Field{
  944. Field{
  945. Expr:&FieldRef{Name: "name"},
  946. Name: "name",
  947. AName:""},
  948. },
  949. Sources: []Source{&Table{Name:"tbl"}},
  950. },
  951. },
  952. {
  953. s: `SELECT name /*SELECT comment FROM testComments*/ FROM tbl`,
  954. stmt: &SelectStatement{
  955. Fields: []Field{
  956. Field{
  957. Expr:&FieldRef{Name: "name"},
  958. Name: "name",
  959. AName:""},
  960. },
  961. Sources: []Source{&Table{Name:"tbl"}},
  962. },
  963. },
  964. {
  965. s: `SELECT true AS f1, FALSE as f2 FROM tbl`,
  966. stmt: &SelectStatement{
  967. Fields: []Field{
  968. Field{AName:"f1", Name: "", Expr: &BooleanLiteral{Val:true}},
  969. Field{AName:"f2", Name: "", Expr: &BooleanLiteral{Val:false}},
  970. },
  971. Sources: []Source{&Table{Name:"tbl"}},
  972. },
  973. },
  974. {
  975. s: `SELECT true AS f1 FROM tbl WHERE f2 = true`,
  976. stmt: &SelectStatement{
  977. Fields: []Field{
  978. Field{AName:"f1", Name: "", Expr: &BooleanLiteral{Val:true}},
  979. },
  980. Sources: []Source{&Table{Name:"tbl"}},
  981. Condition: &BinaryExpr{LHS: &FieldRef{Name: "f2"}, OP: EQ, RHS: &BooleanLiteral{Val: true},},
  982. },
  983. },
  984. {
  985. s: `SELECT sample(true, "abc") FROM tbl`,
  986. stmt: &SelectStatement{
  987. Fields: []Field{
  988. Field{
  989. AName:"",
  990. Name: "sample",
  991. Expr:&Call{
  992. Name:"sample",
  993. Args: []Expr{&BooleanLiteral{Val:true}, &StringLiteral{Val: "abc"}},
  994. },
  995. },
  996. },
  997. Sources: []Source{&Table{Name:"tbl"}},
  998. },
  999. },
  1000. //The negative value expression support.
  1001. {
  1002. s: `SELECT -3 AS t1 FROM tbl`,
  1003. stmt: &SelectStatement{
  1004. Fields: []Field{
  1005. Field{
  1006. Expr:&IntegerLiteral{Val:-3},
  1007. Name: "",
  1008. AName:"t1"},
  1009. },
  1010. Sources: []Source{&Table{Name:"tbl"}},
  1011. },
  1012. },
  1013. {
  1014. s: `SELECT - 3 AS t1 FROM tbl`,
  1015. stmt: &SelectStatement{
  1016. Fields: []Field{
  1017. Field{
  1018. Expr:&IntegerLiteral{Val:-3},
  1019. Name: "",
  1020. AName:"t1"},
  1021. },
  1022. Sources: []Source{&Table{Name:"tbl"}},
  1023. },
  1024. },
  1025. {
  1026. s: `SELECT -. 3 AS t1 FROM tbl`,
  1027. stmt: &SelectStatement{
  1028. Fields: []Field{
  1029. Field{
  1030. Expr:&NumberLiteral{Val:-.3},
  1031. Name: "",
  1032. AName:"t1"},
  1033. },
  1034. Sources: []Source{&Table{Name:"tbl"}},
  1035. },
  1036. },
  1037. {
  1038. s: `SELECT -3.3 AS t1 FROM tbl`,
  1039. stmt: &SelectStatement{
  1040. Fields: []Field{
  1041. Field{
  1042. Expr:&NumberLiteral{Val:-3.3},
  1043. Name: "",
  1044. AName:"t1"},
  1045. },
  1046. Sources: []Source{&Table{Name:"tbl"}},
  1047. },
  1048. },
  1049. {
  1050. s: `SELECT sample(-.3,) FROM tbl`,
  1051. stmt: &SelectStatement{
  1052. Fields: []Field{
  1053. Field{
  1054. Expr:&Call{
  1055. Name:"sample",
  1056. Args: []Expr {
  1057. &NumberLiteral{Val:-0.3},
  1058. },
  1059. },
  1060. Name: "sample",
  1061. AName:""},
  1062. },
  1063. Sources: []Source{&Table{Name:"tbl"}},
  1064. },
  1065. },
  1066. }
  1067. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  1068. for i, tt := range tests {
  1069. //fmt.Printf("Parsing SQL %q.\n", tt.s)
  1070. stmt, err := NewParser(strings.NewReader(tt.s)).Parse()
  1071. if !reflect.DeepEqual(tt.err, errstring(err)) {
  1072. t.Errorf("%d. %q: error mismatch:\n exp=%s\n got=%s\n\n", i, tt.s, tt.err, err)
  1073. } else if tt.err == "" && !reflect.DeepEqual(tt.stmt, stmt) {
  1074. t.Errorf("%d. %q\n\nstmt mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.s, tt.stmt, stmt)
  1075. }
  1076. }
  1077. }
  1078. func TestParser_ParseWindowsExpr(t *testing.T) {
  1079. var tests = []struct {
  1080. s string
  1081. stmt *SelectStatement
  1082. err string
  1083. }{
  1084. {
  1085. s: `SELECT f1 FROM tbl GROUP BY TUMBLINGWINDOW(ss, 10)`,
  1086. stmt: &SelectStatement{
  1087. Fields: []Field{
  1088. Field{
  1089. Expr: &FieldRef{Name:"f1"},
  1090. Name: "f1",
  1091. AName:""},
  1092. },
  1093. Sources: []Source{&Table{Name:"tbl"}},
  1094. Dimensions: Dimensions{
  1095. Dimension{
  1096. Expr:&Windows{
  1097. WindowType: TUMBLING_WINDOW,
  1098. Args:[]Expr{&TimeLiteral{Val:SS}, &IntegerLiteral{Val:10}},
  1099. },
  1100. },
  1101. },
  1102. },
  1103. },
  1104. {
  1105. s: `SELECT f1 FROM tbl GROUP BY HOPPINGWINDOW(mi, 5, 1)`,
  1106. stmt: &SelectStatement{
  1107. Fields: []Field{
  1108. Field{
  1109. Expr: &FieldRef{Name:"f1"},
  1110. Name: "f1",
  1111. AName:""},
  1112. },
  1113. Sources: []Source{&Table{Name:"tbl"}},
  1114. Dimensions: Dimensions{
  1115. Dimension{
  1116. Expr:&Windows{
  1117. WindowType: HOPPING_WINDOW,
  1118. Args:[]Expr{ &TimeLiteral{Val:MI}, &IntegerLiteral{Val:5}, &IntegerLiteral{Val:1}},
  1119. },
  1120. },
  1121. },
  1122. },
  1123. },
  1124. {
  1125. s: `SELECT f1 FROM tbl GROUP BY SESSIONWINDOW(hh, 5, 1)`,
  1126. stmt: &SelectStatement{
  1127. Fields: []Field{
  1128. Field{
  1129. Expr: &FieldRef{Name:"f1"},
  1130. Name: "f1",
  1131. AName:""},
  1132. },
  1133. Sources: []Source{&Table{Name:"tbl"}},
  1134. Dimensions: Dimensions{
  1135. Dimension{
  1136. Expr:&Windows{
  1137. WindowType: SESSION_WINDOW,
  1138. Args:[]Expr{ &TimeLiteral{Val:HH}, &IntegerLiteral{Val:5}, &IntegerLiteral{Val:1}},
  1139. },
  1140. },
  1141. },
  1142. },
  1143. },
  1144. {
  1145. s: `SELECT f1 FROM tbl GROUP BY SLIDINGWINDOW(ms, 5)`,
  1146. stmt: &SelectStatement{
  1147. Fields: []Field{
  1148. Field{
  1149. Expr: &FieldRef{Name:"f1"},
  1150. Name: "f1",
  1151. AName:""},
  1152. },
  1153. Sources: []Source{&Table{Name:"tbl"}},
  1154. Dimensions: Dimensions{
  1155. Dimension{
  1156. Expr:&Windows{
  1157. WindowType: SLIDING_WINDOW,
  1158. Args:[]Expr{ &TimeLiteral{Val:MS}, &IntegerLiteral{Val:5}},
  1159. },
  1160. },
  1161. },
  1162. },
  1163. },
  1164. {
  1165. s: `SELECT f1 FROM tbl GROUP BY SLIDINGWINDOW(mi, 5, 1)`,
  1166. stmt: nil,
  1167. err: "The arguments for slidingwindow should be 2.\n",
  1168. },
  1169. {
  1170. s: `SELECT f1 FROM tbl GROUP BY SLIDINGWINDOW("mi", 5)`,
  1171. stmt: nil,
  1172. err: "The 1st argument for slidingwindow is expecting timer literal expression. One value of [dd|hh|mi|ss|ms].\n",
  1173. },
  1174. }
  1175. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  1176. for i, tt := range tests {
  1177. //fmt.Printf("Parsing SQL %q.\n", tt.s)
  1178. stmt, err := NewParser(strings.NewReader(tt.s)).Parse()
  1179. if !reflect.DeepEqual(tt.err, errstring(err)) {
  1180. t.Errorf("%d. %q: error mismatch:\n exp=%s\n got=%s\n\n", i, tt.s, tt.err, err)
  1181. } else if tt.err == "" && !reflect.DeepEqual(tt.stmt, stmt) {
  1182. t.Errorf("%d. %q\n\nstmt mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.s, tt.stmt, stmt)
  1183. }
  1184. }
  1185. }
  1186. func TestParser_ParseJsonExpr(t *testing.T) {
  1187. var tests = []struct {
  1188. s string
  1189. stmt *SelectStatement
  1190. err string
  1191. }{
  1192. {
  1193. s: `SELECT children[0] FROM demo`,
  1194. stmt: &SelectStatement{
  1195. Fields: []Field{
  1196. Field{
  1197. Expr: &BinaryExpr{
  1198. LHS: &FieldRef{Name:"children"},
  1199. OP: SUBSET,
  1200. RHS: &IndexExpr{Index:0},
  1201. },
  1202. Name: "",
  1203. AName:""},
  1204. },
  1205. Sources: []Source{&Table{Name:"demo"}},
  1206. },
  1207. },
  1208. {
  1209. s: `SELECT children[0]->first FROM demo`,
  1210. stmt: &SelectStatement{
  1211. Fields: []Field{
  1212. Field{
  1213. Expr: &BinaryExpr{
  1214. LHS:&BinaryExpr{
  1215. LHS: &FieldRef{Name:"children"},
  1216. OP: SUBSET,
  1217. RHS: &IndexExpr{Index:0},
  1218. },
  1219. OP: ARROW,
  1220. RHS: &FieldRef{Name:"first"},
  1221. },
  1222. Name: "",
  1223. AName:""},
  1224. },
  1225. Sources: []Source{&Table{Name:"demo"}},
  1226. },
  1227. },
  1228. {
  1229. s: `SELECT children->first[2] FROM demo`,
  1230. stmt: &SelectStatement{
  1231. Fields: []Field{
  1232. Field{
  1233. Expr: &BinaryExpr{
  1234. LHS:&BinaryExpr{
  1235. LHS: &FieldRef{Name:"children"},
  1236. OP: ARROW,
  1237. RHS: &FieldRef{Name:"first"},
  1238. },
  1239. OP: SUBSET,
  1240. RHS: &IndexExpr{Index:2},
  1241. },
  1242. Name: "",
  1243. AName:""},
  1244. },
  1245. Sources: []Source{&Table{Name:"demo"}},
  1246. },
  1247. },
  1248. {
  1249. s: `SELECT children->first[2]->test FROM demo`,
  1250. stmt: &SelectStatement{
  1251. Fields: []Field{
  1252. Field{
  1253. Expr: &BinaryExpr{
  1254. LHS:&BinaryExpr{
  1255. LHS:&BinaryExpr{
  1256. LHS: &FieldRef{Name:"children"},
  1257. OP: ARROW,
  1258. RHS: &FieldRef{Name:"first"},
  1259. },
  1260. OP: SUBSET,
  1261. RHS: &IndexExpr{Index:2},
  1262. },
  1263. OP: ARROW,
  1264. RHS:&FieldRef{Name:"test"},
  1265. },
  1266. Name: "",
  1267. AName:""},
  1268. },
  1269. Sources: []Source{&Table{Name:"demo"}},
  1270. },
  1271. },
  1272. {
  1273. s: `SELECT children[0:1] FROM demo`,
  1274. stmt: &SelectStatement{
  1275. Fields: []Field{
  1276. Field{
  1277. Expr: &BinaryExpr{
  1278. LHS: &FieldRef{Name:"children"},
  1279. OP: SUBSET,
  1280. RHS: &ColonExpr{Start:0, End:1},
  1281. },
  1282. Name: "",
  1283. AName:""},
  1284. },
  1285. Sources: []Source{&Table{Name:"demo"}},
  1286. },
  1287. },
  1288. {
  1289. s: `SELECT children[:1] FROM demo`,
  1290. stmt: &SelectStatement{
  1291. Fields: []Field{
  1292. Field{
  1293. Expr: &BinaryExpr{
  1294. LHS: &FieldRef{Name:"children"},
  1295. OP: SUBSET,
  1296. RHS: &ColonExpr{Start:0, End:1},
  1297. },
  1298. Name: "",
  1299. AName:""},
  1300. },
  1301. Sources: []Source{&Table{Name:"demo"}},
  1302. },
  1303. },
  1304. {
  1305. s: `SELECT children[:] FROM demo`,
  1306. stmt: &SelectStatement{
  1307. Fields: []Field{
  1308. Field{
  1309. Expr: &BinaryExpr{
  1310. LHS: &FieldRef{Name:"children"},
  1311. OP: SUBSET,
  1312. RHS: &ColonExpr{Start:0, End:-1},
  1313. },
  1314. Name: "",
  1315. AName:""},
  1316. },
  1317. Sources: []Source{&Table{Name:"demo"}},
  1318. },
  1319. },
  1320. {
  1321. s: `SELECT children[2:] AS c FROM demo`,
  1322. stmt: &SelectStatement{
  1323. Fields: []Field{
  1324. Field{
  1325. Expr: &BinaryExpr{
  1326. LHS: &FieldRef{Name: "children"},
  1327. OP: SUBSET,
  1328. RHS: &ColonExpr{Start: 2, End: -1},
  1329. },
  1330. Name: "",
  1331. AName: "c"},
  1332. },
  1333. Sources: []Source{&Table{Name: "demo"}},
  1334. },
  1335. },
  1336. {
  1337. s: `SELECT children[2:]->first AS c FROM demo`,
  1338. stmt: &SelectStatement{
  1339. Fields: []Field{
  1340. Field{
  1341. Expr: &BinaryExpr{
  1342. LHS: &BinaryExpr{LHS: &FieldRef{Name: "children"}, OP: SUBSET, RHS: &ColonExpr{Start: 2, End: -1},
  1343. },
  1344. OP: ARROW,
  1345. RHS: &FieldRef{Name: "first"},
  1346. },
  1347. Name: "",
  1348. AName: "c"},
  1349. },
  1350. Sources: []Source{&Table{Name: "demo"}},
  1351. },
  1352. },
  1353. {
  1354. s: `SELECT demo.children[2:]->first AS c FROM demo`,
  1355. stmt: &SelectStatement{
  1356. Fields: []Field{
  1357. Field{
  1358. Expr: &BinaryExpr{
  1359. LHS: &BinaryExpr{LHS: &FieldRef{StreamName:StreamName("demo"), Name: "children"}, OP: SUBSET, RHS: &ColonExpr{Start: 2, End: -1},
  1360. },
  1361. OP: ARROW,
  1362. RHS: &FieldRef{Name: "first"},
  1363. },
  1364. Name: "",
  1365. AName: "c"},
  1366. },
  1367. Sources: []Source{&Table{Name: "demo"}},
  1368. },
  1369. },
  1370. {
  1371. s: `SELECT sample(demo.children[2:]->first) AS c FROM demo`,
  1372. stmt: &SelectStatement{
  1373. Fields: []Field{
  1374. Field{
  1375. Expr:&Call{
  1376. Name:"sample",
  1377. Args: []Expr{
  1378. &BinaryExpr{
  1379. LHS: &BinaryExpr{LHS: &FieldRef{StreamName:StreamName("demo"), Name: "children"}, OP: SUBSET, RHS: &ColonExpr{Start: 2, End: -1},
  1380. },
  1381. OP: ARROW,
  1382. RHS: &FieldRef{Name: "first"},
  1383. },
  1384. },
  1385. },
  1386. Name: "sample",
  1387. AName: "c"},
  1388. },
  1389. Sources: []Source{&Table{Name: "demo"}},
  1390. },
  1391. },
  1392. {
  1393. s: `SELECT children[:1] FROM demo WHERE abc[0] > 12`,
  1394. stmt: &SelectStatement{
  1395. Fields: []Field{
  1396. Field{
  1397. Expr: &BinaryExpr{
  1398. LHS: &FieldRef{Name:"children"},
  1399. OP: SUBSET,
  1400. RHS: &ColonExpr{Start:0, End:1},
  1401. },
  1402. Name: "",
  1403. AName:""},
  1404. },
  1405. Sources: []Source{&Table{Name:"demo"}},
  1406. Condition: &BinaryExpr{
  1407. LHS: &BinaryExpr{
  1408. LHS: &FieldRef{Name:"abc"},
  1409. OP: SUBSET,
  1410. RHS: &IndexExpr{Index:0},
  1411. },
  1412. OP:GT,
  1413. RHS:&IntegerLiteral{Val:12},
  1414. },
  1415. },
  1416. },
  1417. {
  1418. s: `SELECT demo.children.first AS c FROM demo`,
  1419. stmt: nil,
  1420. err: "Too many field names. Please use -> to reference keys in struct.\n",
  1421. },
  1422. }
  1423. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  1424. for i, tt := range tests {
  1425. stmt, err := NewParser(strings.NewReader(tt.s)).Parse()
  1426. if !reflect.DeepEqual(tt.err, errstring(err)) {
  1427. t.Errorf("%d. %q: error mismatch:\n exp=%s\n got=%s\n\n", i, tt.s, tt.err, err)
  1428. } else if tt.err == "" && !reflect.DeepEqual(tt.stmt, stmt) {
  1429. t.Errorf("%d. %q\n\nstmt mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.s, tt.stmt, stmt)
  1430. }
  1431. }
  1432. }
  1433. func TestParser_ParseJoins(t *testing.T) {
  1434. var tests = []struct {
  1435. s string
  1436. stmt *SelectStatement
  1437. err string
  1438. }{
  1439. {
  1440. s: `SELECT * FROM topic/sensor1 LEFT JOIN topic1 ON f=k`,
  1441. stmt: &SelectStatement{
  1442. Fields: []Field{
  1443. Field{
  1444. Expr: &Wildcard{Token: ASTERISK},
  1445. Name: "",
  1446. AName:""},
  1447. },
  1448. Sources: []Source{&Table{Name:"topic/sensor1"}},
  1449. Joins: []Join{
  1450. Join{
  1451. Name:"topic1", Alias: "", JoinType: LEFT_JOIN, Expr: &BinaryExpr{
  1452. LHS: &FieldRef{Name:"f"},
  1453. OP: EQ,
  1454. RHS:&FieldRef{Name:"k"},
  1455. },
  1456. },
  1457. },
  1458. },
  1459. },
  1460. {
  1461. s: `SELECT * FROM topic/sensor1 AS t1 INNER JOIN topic1 AS t2 ON f=k`,
  1462. stmt: &SelectStatement{
  1463. Fields: []Field{
  1464. Field{
  1465. Expr: &Wildcard{Token: ASTERISK},
  1466. Name: "",
  1467. AName:""},
  1468. },
  1469. Sources: []Source{&Table{Name:"topic/sensor1", Alias:"t1"}},
  1470. Joins: []Join{
  1471. Join{
  1472. Name:"topic1", Alias: "t2", JoinType: INNER_JOIN, Expr: &BinaryExpr{
  1473. LHS: &FieldRef{Name:"f"},
  1474. OP: EQ,
  1475. RHS:&FieldRef{Name:"k"},
  1476. },
  1477. },
  1478. },
  1479. },
  1480. },
  1481. {
  1482. s: `SELECT * FROM topic/sensor1 AS t1 LEFT JOIN topic1/sensor2 AS t2 ON f=k`,
  1483. stmt: &SelectStatement{
  1484. Fields: []Field{
  1485. Field{
  1486. Expr: &Wildcard{Token: ASTERISK},
  1487. Name: "",
  1488. AName:""},
  1489. },
  1490. Sources: []Source{&Table{Name:"topic/sensor1", Alias:"t1"}},
  1491. Joins: []Join{
  1492. Join{
  1493. Name:"topic1/sensor2", Alias: "t2", JoinType: LEFT_JOIN, Expr: &BinaryExpr{
  1494. LHS: &FieldRef{Name:"f"},
  1495. OP: EQ,
  1496. RHS:&FieldRef{Name:"k"},
  1497. },
  1498. },
  1499. },
  1500. },
  1501. },
  1502. {
  1503. s: `SELECT t1.name FROM topic/sensor1 AS t1 LEFT JOIN topic1/sensor2 AS t2 ON f=k`,
  1504. stmt: &SelectStatement{
  1505. Fields: []Field{
  1506. Field{
  1507. Expr: &FieldRef{StreamName:StreamName("t1"), Name:"name"},
  1508. Name: "name",
  1509. AName:""},
  1510. },
  1511. Sources: []Source{&Table{Name:"topic/sensor1", Alias:"t1"}},
  1512. Joins: []Join{
  1513. Join{
  1514. Name:"topic1/sensor2", Alias: "t2", JoinType: LEFT_JOIN, Expr: &BinaryExpr{
  1515. LHS: &FieldRef{Name:"f"},
  1516. OP: EQ,
  1517. RHS:&FieldRef{Name:"k"},
  1518. },
  1519. },
  1520. },
  1521. },
  1522. },
  1523. {
  1524. s: `SELECT t1.name FROM topic/sensor1 AS t1 LEFT JOIN topic1/sensor2 AS t2 ON t1.f=t2.k`,
  1525. stmt: &SelectStatement{
  1526. Fields: []Field{
  1527. Field{
  1528. Expr: &FieldRef{StreamName:StreamName("t1"), Name:"name"},
  1529. Name: "name",
  1530. AName:""},
  1531. },
  1532. Sources: []Source{&Table{Name:"topic/sensor1", Alias:"t1"}},
  1533. Joins: []Join{
  1534. Join{
  1535. Name:"topic1/sensor2", Alias: "t2", JoinType: LEFT_JOIN, Expr: &BinaryExpr{
  1536. LHS: &FieldRef{StreamName:StreamName("t1"), Name:"f"},
  1537. OP: EQ,
  1538. RHS:&FieldRef{StreamName:StreamName("t2"), Name:"k"},
  1539. },
  1540. },
  1541. },
  1542. },
  1543. },
  1544. }
  1545. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  1546. for i, tt := range tests {
  1547. stmt, err := NewParser(strings.NewReader(tt.s)).Parse()
  1548. if !reflect.DeepEqual(tt.err, errstring(err)) {
  1549. t.Errorf("%d. %q: error mismatch:\n exp=%s\n got=%s\n\n", i, tt.s, tt.err, err)
  1550. } else if tt.err == "" && !reflect.DeepEqual(tt.stmt, stmt) {
  1551. t.Errorf("%d. %q\n\nstmt mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.s, tt.stmt, stmt)
  1552. }
  1553. }
  1554. }
  1555. func TestParser_ParseStatements(t *testing.T) {
  1556. var tests = []struct {
  1557. s string
  1558. stmts SelectStatements
  1559. err string
  1560. }{
  1561. {s: "SELECT name FROM tbl;\nSELECT name FROM topic/sensor1\n",
  1562. stmts: []SelectStatement{
  1563. SelectStatement{
  1564. Fields: []Field{
  1565. Field{
  1566. Expr: &FieldRef{Name: "name"},
  1567. Name: "name",
  1568. AName: ""},
  1569. },
  1570. Sources: []Source{&Table{Name:"tbl"}},
  1571. },
  1572. SelectStatement{
  1573. Fields: []Field{
  1574. Field{
  1575. Expr: &FieldRef{Name: "name"},
  1576. Name: "name",
  1577. AName: ""},
  1578. },
  1579. Sources: []Source{&Table{Name:"topic/sensor1"}},
  1580. },
  1581. },
  1582. },
  1583. {s: "SELECT name FROM tbl;\nSELECT name FROM topic/sensor1;\n--SELECT comment FROM topic/comment",
  1584. stmts: []SelectStatement{
  1585. SelectStatement{
  1586. Fields: []Field{
  1587. Field{
  1588. Expr: &FieldRef{Name: "name"},
  1589. Name: "name",
  1590. AName: ""},
  1591. },
  1592. Sources: []Source{&Table{Name:"tbl"}},
  1593. },
  1594. SelectStatement{
  1595. Fields: []Field{
  1596. Field{
  1597. Expr: &FieldRef{Name: "name"},
  1598. Name: "name",
  1599. AName: ""},
  1600. },
  1601. Sources: []Source{&Table{Name:"topic/sensor1"}},
  1602. },
  1603. },
  1604. },
  1605. }
  1606. fmt.Printf("The test bucket size is %d.\n\n", len(tests))
  1607. for i, tt := range tests {
  1608. stmts, err := NewParser(strings.NewReader(tt.s)).ParseQueries()
  1609. if !reflect.DeepEqual(tt.err, errstring(err)) {
  1610. t.Errorf("%d. %q: error mismatch:\n exp=%s\n got=%s\n\n", i, tt.s, tt.err, err)
  1611. } else if tt.err == "" && !reflect.DeepEqual(tt.stmts, stmts) {
  1612. t.Errorf("%d. %q\n\nstmt mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.s, tt.stmts, stmts)
  1613. }
  1614. }
  1615. }
  1616. // errstring returns the string representation of an error.
  1617. func errstring(err error) string {
  1618. if err != nil {
  1619. return err.Error()
  1620. }
  1621. return ""
  1622. }