parser.go 28 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130
  1. package xsql
  2. import (
  3. "fmt"
  4. "github.com/emqx/kuiper/common"
  5. "github.com/golang-collections/collections/stack"
  6. "io"
  7. "strconv"
  8. "strings"
  9. )
  10. type Parser struct {
  11. s *Scanner
  12. i int // buffer index
  13. n int // buffer char count
  14. buf [3]struct {
  15. tok Token
  16. lit string
  17. }
  18. inmeta bool
  19. }
  20. func (p *Parser) parseCondition() (Expr, error) {
  21. if tok, _ := p.scanIgnoreWhitespace(); tok != WHERE {
  22. p.unscan()
  23. return nil, nil
  24. }
  25. expr, err := p.ParseExpr()
  26. if err != nil {
  27. return nil, err
  28. }
  29. return expr, nil
  30. }
  31. func (p *Parser) scan() (tok Token, lit string) {
  32. if p.n > 0 {
  33. p.n--
  34. return p.curr()
  35. }
  36. tok, lit = p.s.Scan()
  37. if tok != WS && tok != COMMENT {
  38. p.i = (p.i + 1) % len(p.buf)
  39. buf := &p.buf[p.i]
  40. buf.tok, buf.lit = tok, lit
  41. }
  42. return
  43. }
  44. func (p *Parser) curr() (Token, string) {
  45. i := (p.i - p.n + len(p.buf)) % len(p.buf)
  46. buf := &p.buf[i]
  47. return buf.tok, buf.lit
  48. }
  49. func (p *Parser) scanIgnoreWhitespace() (tok Token, lit string) {
  50. tok, lit = p.scan()
  51. for {
  52. if tok == WS || tok == COMMENT {
  53. tok, lit = p.scan()
  54. } else {
  55. break
  56. }
  57. }
  58. return tok, lit
  59. }
  60. func (p *Parser) unscan() { p.n++ }
  61. func NewParser(r io.Reader) *Parser {
  62. return &Parser{s: NewScanner(r)}
  63. }
  64. func (p *Parser) ParseQueries() (SelectStatements, error) {
  65. var stmts SelectStatements
  66. if stmt, err := p.Parse(); err != nil {
  67. return nil, err
  68. } else {
  69. stmts = append(stmts, *stmt)
  70. }
  71. for {
  72. if tok, _ := p.scanIgnoreWhitespace(); tok == SEMICOLON {
  73. if stmt, err := p.Parse(); err != nil {
  74. return nil, err
  75. } else {
  76. if stmt != nil {
  77. stmts = append(stmts, *stmt)
  78. }
  79. }
  80. } else if tok == EOF {
  81. break
  82. }
  83. }
  84. return stmts, nil
  85. }
  86. func (p *Parser) Parse() (*SelectStatement, error) {
  87. selects := &SelectStatement{}
  88. if tok, lit := p.scanIgnoreWhitespace(); tok == EOF {
  89. return nil, nil
  90. } else if tok != SELECT {
  91. return nil, fmt.Errorf("Found %q, Expected SELECT.\n", lit)
  92. }
  93. if fields, err := p.parseFields(); err != nil {
  94. return nil, err
  95. } else {
  96. selects.Fields = fields
  97. }
  98. if src, err := p.parseSource(); err != nil {
  99. return nil, err
  100. } else {
  101. selects.Sources = src
  102. }
  103. if joins, err := p.parseJoins(); err != nil {
  104. return nil, err
  105. } else {
  106. selects.Joins = joins
  107. }
  108. if exp, err := p.parseCondition(); err != nil {
  109. return nil, err
  110. } else {
  111. if exp != nil {
  112. selects.Condition = exp
  113. }
  114. }
  115. if dims, err := p.parseDimensions(); err != nil {
  116. return nil, err
  117. } else {
  118. selects.Dimensions = dims
  119. }
  120. if having, err := p.parseHaving(); err != nil {
  121. return nil, err
  122. } else {
  123. selects.Having = having
  124. }
  125. if sorts, err := p.parseSorts(); err != nil {
  126. return nil, err
  127. } else {
  128. selects.SortFields = sorts
  129. }
  130. if tok, lit := p.scanIgnoreWhitespace(); tok == SEMICOLON {
  131. p.unscan()
  132. return selects, nil
  133. } else if tok != EOF {
  134. return nil, fmt.Errorf("found %q, expected EOF.", lit)
  135. }
  136. if err := Validate(selects); err != nil {
  137. return nil, err
  138. }
  139. return selects, nil
  140. }
  141. func (p *Parser) parseSource() (Sources, error) {
  142. var sources Sources
  143. if tok, lit := p.scanIgnoreWhitespace(); tok != FROM {
  144. return nil, fmt.Errorf("found %q, expected FROM.", lit)
  145. }
  146. if src, alias, err := p.parseSourceLiteral(); err != nil {
  147. return nil, err
  148. } else {
  149. sources = append(sources, &Table{Name: src, Alias: alias})
  150. }
  151. return sources, nil
  152. }
  153. //TODO Current func has problems when the source includes white space.
  154. func (p *Parser) parseSourceLiteral() (string, string, error) {
  155. var sourceSeg []string
  156. var alias string
  157. for {
  158. //HASH, DIV & ADD token is specially support for MQTT topic name patterns.
  159. if tok, lit := p.scanIgnoreWhitespace(); tok.allowedSourceToken() {
  160. sourceSeg = append(sourceSeg, lit)
  161. if tok1, lit1 := p.scanIgnoreWhitespace(); tok1 == AS {
  162. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == IDENT {
  163. alias = lit2
  164. } else {
  165. return "", "", fmt.Errorf("found %q, expected JOIN key word.", lit)
  166. }
  167. } else if tok1.allowedSourceToken() {
  168. sourceSeg = append(sourceSeg, lit1)
  169. } else {
  170. p.unscan()
  171. break
  172. }
  173. } else {
  174. p.unscan()
  175. break
  176. }
  177. }
  178. return strings.Join(sourceSeg, ""), alias, nil
  179. }
  180. func (p *Parser) parseFieldNameSections() ([]string, error) {
  181. var fieldNameSects []string
  182. for {
  183. if tok, lit := p.scanIgnoreWhitespace(); tok == IDENT || tok == ASTERISK {
  184. fieldNameSects = append(fieldNameSects, lit)
  185. if tok1, _ := p.scanIgnoreWhitespace(); !tok1.allowedSFNToken() {
  186. p.unscan()
  187. break
  188. }
  189. } else {
  190. p.unscan()
  191. break
  192. }
  193. }
  194. if len(fieldNameSects) == 0 {
  195. return nil, fmt.Errorf("Cannot find any field name.\n")
  196. } else if len(fieldNameSects) > 2 {
  197. return nil, fmt.Errorf("Too many field names. Please use -> to reference keys in struct.\n")
  198. }
  199. return fieldNameSects, nil
  200. }
  201. func (p *Parser) parseJoins() (Joins, error) {
  202. var joins Joins
  203. for {
  204. if tok, lit := p.scanIgnoreWhitespace(); tok == INNER || tok == LEFT || tok == RIGHT || tok == FULL || tok == CROSS {
  205. if tok1, _ := p.scanIgnoreWhitespace(); tok1 == JOIN {
  206. var jt = INNER_JOIN
  207. switch tok {
  208. case INNER:
  209. jt = INNER_JOIN
  210. case LEFT:
  211. jt = LEFT_JOIN
  212. case RIGHT:
  213. jt = RIGHT_JOIN
  214. case FULL:
  215. jt = FULL_JOIN
  216. case CROSS:
  217. jt = CROSS_JOIN
  218. }
  219. if j, err := p.ParseJoin(jt); err != nil {
  220. return nil, err
  221. } else {
  222. joins = append(joins, *j)
  223. }
  224. } else {
  225. return nil, fmt.Errorf("found %q, expected JOIN key word.", lit)
  226. }
  227. } else {
  228. p.unscan()
  229. if len(joins) > 0 {
  230. return joins, nil
  231. }
  232. return nil, nil
  233. }
  234. }
  235. return joins, nil
  236. }
  237. func (p *Parser) ParseJoin(joinType JoinType) (*Join, error) {
  238. var j = &Join{JoinType: joinType}
  239. if src, alias, err := p.parseSourceLiteral(); err != nil {
  240. return nil, err
  241. } else {
  242. j.Name = src
  243. j.Alias = alias
  244. if tok1, _ := p.scanIgnoreWhitespace(); tok1 == ON {
  245. if CROSS_JOIN == joinType {
  246. return nil, fmt.Errorf("On expression is not required for cross join type.\n")
  247. }
  248. if exp, err := p.ParseExpr(); err != nil {
  249. return nil, err
  250. } else {
  251. j.Expr = exp
  252. }
  253. } else {
  254. p.unscan()
  255. }
  256. }
  257. return j, nil
  258. }
  259. func (p *Parser) parseDimensions() (Dimensions, error) {
  260. var ds Dimensions
  261. if t, _ := p.scanIgnoreWhitespace(); t == GROUP {
  262. if t1, l1 := p.scanIgnoreWhitespace(); t1 == BY {
  263. for {
  264. if exp, err := p.ParseExpr(); err != nil {
  265. return nil, err
  266. } else {
  267. d := Dimension{Expr: exp}
  268. ds = append(ds, d)
  269. }
  270. if tok, _ := p.scanIgnoreWhitespace(); tok == COMMA {
  271. continue
  272. } else {
  273. p.unscan()
  274. break
  275. }
  276. }
  277. } else {
  278. return nil, fmt.Errorf("found %q, expected BY statement.", l1)
  279. }
  280. } else {
  281. p.unscan()
  282. }
  283. return ds, nil
  284. }
  285. func (p *Parser) parseHaving() (Expr, error) {
  286. if tok, _ := p.scanIgnoreWhitespace(); tok != HAVING {
  287. p.unscan()
  288. return nil, nil
  289. }
  290. expr, err := p.ParseExpr()
  291. if err != nil {
  292. return nil, err
  293. }
  294. return expr, nil
  295. }
  296. func (p *Parser) parseSorts() (SortFields, error) {
  297. var ss SortFields
  298. if t, _ := p.scanIgnoreWhitespace(); t == ORDER {
  299. if t1, l1 := p.scanIgnoreWhitespace(); t1 == BY {
  300. for {
  301. if t1, l1 = p.scanIgnoreWhitespace(); t1 == IDENT {
  302. s := SortField{Ascending: true}
  303. p.unscan()
  304. if name, err := p.parseFieldNameSections(); err == nil {
  305. s.Name = strings.Join(name, tokens[COLSEP])
  306. } else {
  307. return nil, err
  308. }
  309. if t2, _ := p.scanIgnoreWhitespace(); t2 == DESC {
  310. s.Ascending = false
  311. ss = append(ss, s)
  312. } else if t2 == ASC {
  313. ss = append(ss, s)
  314. } else {
  315. ss = append(ss, s)
  316. p.unscan()
  317. continue
  318. }
  319. } else if t1 == COMMA {
  320. continue
  321. } else {
  322. p.unscan()
  323. break
  324. }
  325. }
  326. } else {
  327. return nil, fmt.Errorf("found %q, expected BY keyword.", l1)
  328. }
  329. } else {
  330. p.unscan()
  331. }
  332. return ss, nil
  333. }
  334. func (p *Parser) parseFields() (Fields, error) {
  335. var fields Fields
  336. tok, _ := p.scanIgnoreWhitespace()
  337. if tok == ASTERISK {
  338. fields = append(fields, Field{AName: "", Expr: &Wildcard{Token: tok}})
  339. return fields, nil
  340. }
  341. p.unscan()
  342. for {
  343. field, err := p.parseField()
  344. if err != nil {
  345. return nil, err
  346. } else {
  347. fields = append(fields, *field)
  348. }
  349. tok, _ = p.scanIgnoreWhitespace()
  350. if tok != COMMA {
  351. p.unscan()
  352. break
  353. }
  354. }
  355. return fields, nil
  356. }
  357. func (p *Parser) parseField() (*Field, error) {
  358. field := &Field{}
  359. if exp, err := p.ParseExpr(); err != nil {
  360. return nil, err
  361. } else {
  362. if e, ok := exp.(*FieldRef); ok {
  363. field.Name = e.Name
  364. } else if e, ok := exp.(*Call); ok {
  365. field.Name = e.Name
  366. }
  367. field.Expr = exp
  368. }
  369. if alias, err := p.parseAlias(); err != nil {
  370. return nil, err
  371. } else {
  372. if alias != "" {
  373. field.AName = alias
  374. }
  375. }
  376. return field, nil
  377. }
  378. func (p *Parser) parseAlias() (string, error) {
  379. tok, lit := p.scanIgnoreWhitespace()
  380. if tok == AS {
  381. if tok, lit = p.scanIgnoreWhitespace(); tok != IDENT {
  382. return "", fmt.Errorf("found %q, expected as alias.", lit)
  383. } else {
  384. return lit, nil
  385. }
  386. }
  387. p.unscan()
  388. return "", nil
  389. }
  390. func (p *Parser) ParseExpr() (Expr, error) {
  391. var err error
  392. root := &BinaryExpr{}
  393. root.RHS, err = p.parseUnaryExpr()
  394. if err != nil {
  395. return nil, err
  396. }
  397. for {
  398. op, _ := p.scanIgnoreWhitespace()
  399. if !op.isOperator() {
  400. p.unscan()
  401. return root.RHS, nil
  402. } else if op == ASTERISK { //Change the asterisk to Mul token.
  403. op = MUL
  404. } else if op == LBRACKET { //LBRACKET is a special token, need to unscan
  405. op = SUBSET
  406. p.unscan()
  407. }
  408. var rhs Expr
  409. if rhs, err = p.parseUnaryExpr(); err != nil {
  410. return nil, err
  411. }
  412. for node := root; ; {
  413. r, ok := node.RHS.(*BinaryExpr)
  414. if !ok || r.OP.Precedence() >= op.Precedence() {
  415. node.RHS = &BinaryExpr{LHS: node.RHS, RHS: rhs, OP: op}
  416. break
  417. }
  418. node = r
  419. }
  420. }
  421. return nil, nil
  422. }
  423. func (p *Parser) parseUnaryExpr() (Expr, error) {
  424. if tok1, _ := p.scanIgnoreWhitespace(); tok1 == LPAREN {
  425. expr, err := p.ParseExpr()
  426. if err != nil {
  427. return nil, err
  428. }
  429. // Expect an RPAREN at the end.
  430. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 != RPAREN {
  431. return nil, fmt.Errorf("found %q, expected right paren.", lit2)
  432. }
  433. return &ParenExpr{Expr: expr}, nil
  434. } else if tok1 == LBRACKET {
  435. return p.parseBracketExpr()
  436. }
  437. p.unscan()
  438. tok, lit := p.scanIgnoreWhitespace()
  439. if tok == IDENT {
  440. if tok1, _ := p.scanIgnoreWhitespace(); tok1 == LPAREN {
  441. return p.parseCall(lit)
  442. }
  443. p.unscan() //Back the Lparen token
  444. p.unscan() //Back the ident token
  445. if n, err := p.parseFieldNameSections(); err != nil {
  446. return nil, err
  447. } else {
  448. if p.inmeta {
  449. if len(n) == 2 {
  450. return &MetaRef{StreamName: StreamName(n[0]), Name: n[1]}, nil
  451. }
  452. return &MetaRef{StreamName: "", Name: n[0]}, nil
  453. } else {
  454. if len(n) == 2 {
  455. return &FieldRef{StreamName: StreamName(n[0]), Name: n[1]}, nil
  456. }
  457. return &FieldRef{StreamName: "", Name: n[0]}, nil
  458. }
  459. }
  460. } else if tok == STRING {
  461. return &StringLiteral{Val: lit}, nil
  462. } else if tok == INTEGER {
  463. val, _ := strconv.Atoi(lit)
  464. return &IntegerLiteral{Val: val}, nil
  465. } else if tok == NUMBER {
  466. if v, err := strconv.ParseFloat(lit, 64); err != nil {
  467. return nil, fmt.Errorf("found %q, invalid number value.", lit)
  468. } else {
  469. return &NumberLiteral{Val: v}, nil
  470. }
  471. } else if tok == TRUE || tok == FALSE {
  472. if v, err := strconv.ParseBool(lit); err != nil {
  473. return nil, fmt.Errorf("found %q, invalid boolean value.", lit)
  474. } else {
  475. return &BooleanLiteral{Val: v}, nil
  476. }
  477. } else if tok.isTimeLiteral() {
  478. return &TimeLiteral{Val: tok}, nil
  479. }
  480. return nil, fmt.Errorf("found %q, expected expression.", lit)
  481. }
  482. func (p *Parser) parseBracketExpr() (Expr, error) {
  483. tok2, lit2 := p.scanIgnoreWhitespace()
  484. if tok2 == RBRACKET {
  485. //field[]
  486. return &ColonExpr{Start: 0, End: -1}, nil
  487. } else if tok2 == INTEGER {
  488. start, err := strconv.Atoi(lit2)
  489. if err != nil {
  490. return nil, fmt.Errorf("The start index %s is not an int value in bracket expression.", lit2)
  491. }
  492. if tok3, _ := p.scanIgnoreWhitespace(); tok3 == RBRACKET {
  493. //Such as field[2]
  494. return &IndexExpr{Index: start}, nil
  495. } else if tok3 == COLON {
  496. //Such as field[2:] or field[2:4]
  497. return p.parseColonExpr(start)
  498. }
  499. } else if tok2 == COLON {
  500. //Such as field[:3] or [:]
  501. return p.parseColonExpr(0)
  502. }
  503. return nil, fmt.Errorf("Unexpected token %q. when parsing bracket expressions.", lit2)
  504. }
  505. func (p *Parser) parseColonExpr(start int) (Expr, error) {
  506. tok, lit := p.scanIgnoreWhitespace()
  507. if tok == INTEGER {
  508. end, err := strconv.Atoi(lit)
  509. if err != nil {
  510. return nil, fmt.Errorf("The end index %s is not an int value in bracket expression.", lit)
  511. }
  512. if tok1, lit1 := p.scanIgnoreWhitespace(); tok1 == RBRACKET {
  513. return &ColonExpr{Start: start, End: end}, nil
  514. } else {
  515. return nil, fmt.Errorf("Found %q, expected right bracket.", lit1)
  516. }
  517. } else if tok == RBRACKET {
  518. return &ColonExpr{Start: start, End: -1}, nil
  519. }
  520. return nil, fmt.Errorf("Found %q, expected right bracket.", lit)
  521. }
  522. func (p *Parser) parseAs(f *Field) (*Field, error) {
  523. tok, lit := p.scanIgnoreWhitespace()
  524. if tok != IDENT {
  525. return nil, fmt.Errorf("found %q, expected as alias.", lit)
  526. }
  527. f.AName = lit
  528. return f, nil
  529. }
  530. func (p *Parser) parseCall(name string) (Expr, error) {
  531. if strings.ToLower(name) == "meta" || strings.ToLower(name) == "mqtt" {
  532. p.inmeta = true
  533. defer func() {
  534. p.inmeta = false
  535. }()
  536. }
  537. var args []Expr
  538. for {
  539. if tok, _ := p.scanIgnoreWhitespace(); tok == RPAREN {
  540. if valErr := validateFuncs(name, nil); valErr != nil {
  541. return nil, valErr
  542. }
  543. return &Call{Name: name, Args: args}, nil
  544. } else if tok == ASTERISK {
  545. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 != RPAREN {
  546. return nil, fmt.Errorf("found %q, expected right paren.", lit2)
  547. } else {
  548. if p.inmeta {
  549. args = append(args, &MetaRef{StreamName: "", Name: "*"})
  550. } else if _, ok := funcWithAsteriskSupportMap[name]; ok {
  551. args = append(args, &Wildcard{Token: ASTERISK})
  552. } else {
  553. args = append(args, &StringLiteral{Val: "*"})
  554. }
  555. return &Call{Name: name, Args: args}, nil
  556. }
  557. } else {
  558. p.unscan()
  559. }
  560. if exp, err := p.ParseExpr(); err != nil {
  561. return nil, err
  562. } else {
  563. args = append(args, exp)
  564. }
  565. if tok, _ := p.scanIgnoreWhitespace(); tok != COMMA {
  566. p.unscan()
  567. break
  568. }
  569. }
  570. if tok, lit := p.scanIgnoreWhitespace(); tok != RPAREN {
  571. return nil, fmt.Errorf("found function call %q, expected ), but with %q.", name, lit)
  572. }
  573. if wt, error := validateWindows(name, args); wt == NOT_WINDOW {
  574. if valErr := validateFuncs(name, args); valErr != nil {
  575. return nil, valErr
  576. }
  577. // Add context for some aggregate func
  578. if name == "deduplicate" {
  579. args = append([]Expr{&Wildcard{Token: ASTERISK}}, args...)
  580. }
  581. return &Call{Name: name, Args: args}, nil
  582. } else {
  583. if error != nil {
  584. return nil, error
  585. }
  586. win, err := p.ConvertToWindows(wt, args)
  587. if err != nil {
  588. return nil, error
  589. }
  590. // parse filter clause
  591. f, err := p.parseFilter()
  592. if err != nil {
  593. return nil, err
  594. } else if f != nil {
  595. win.Filter = f
  596. }
  597. return win, nil
  598. }
  599. }
  600. func validateWindows(name string, args []Expr) (WindowType, error) {
  601. fname := strings.ToLower(name)
  602. switch fname {
  603. case "tumblingwindow":
  604. if err := validateWindow(fname, 2, args); err != nil {
  605. return TUMBLING_WINDOW, err
  606. }
  607. return TUMBLING_WINDOW, nil
  608. case "hoppingwindow":
  609. if err := validateWindow(fname, 3, args); err != nil {
  610. return HOPPING_WINDOW, err
  611. }
  612. return HOPPING_WINDOW, nil
  613. case "sessionwindow":
  614. if err := validateWindow(fname, 3, args); err != nil {
  615. return SESSION_WINDOW, err
  616. }
  617. return SESSION_WINDOW, nil
  618. case "slidingwindow":
  619. if err := validateWindow(fname, 2, args); err != nil {
  620. return SLIDING_WINDOW, err
  621. }
  622. return SLIDING_WINDOW, nil
  623. case "countwindow":
  624. if len(args) == 1 {
  625. if para1, ok := args[0].(*IntegerLiteral); ok && para1.Val > 0 {
  626. return COUNT_WINDOW, nil
  627. } else {
  628. return COUNT_WINDOW, fmt.Errorf("Invalid parameter value %s.", args[0])
  629. }
  630. } else if len(args) == 2 {
  631. if para1, ok1 := args[0].(*IntegerLiteral); ok1 {
  632. if para2, ok2 := args[1].(*IntegerLiteral); ok2 {
  633. if para1.Val < para2.Val {
  634. return COUNT_WINDOW, fmt.Errorf("The second parameter value %d should be less than the first parameter %d.", para2.Val, para1.Val)
  635. } else {
  636. return COUNT_WINDOW, nil
  637. }
  638. }
  639. }
  640. return COUNT_WINDOW, fmt.Errorf("Invalid parameter value %s, %s.", args[0], args[1])
  641. } else {
  642. return COUNT_WINDOW, fmt.Errorf("Invalid parameter count.")
  643. }
  644. }
  645. return NOT_WINDOW, nil
  646. }
  647. func validateWindow(funcName string, expectLen int, args []Expr) error {
  648. if len(args) != expectLen {
  649. return fmt.Errorf("The arguments for %s should be %d.\n", funcName, expectLen)
  650. }
  651. if _, ok := args[0].(*TimeLiteral); !ok {
  652. return fmt.Errorf("The 1st argument for %s is expecting timer literal expression. One value of [dd|hh|mi|ss|ms].\n", funcName)
  653. }
  654. for i := 1; i < len(args); i++ {
  655. if _, ok := args[i].(*IntegerLiteral); !ok {
  656. return fmt.Errorf("The %d argument for %s is expecting interger literal expression. \n", i, funcName)
  657. }
  658. }
  659. return nil
  660. }
  661. func (p *Parser) ConvertToWindows(wtype WindowType, args []Expr) (*Window, error) {
  662. win := &Window{WindowType: wtype}
  663. if wtype == COUNT_WINDOW {
  664. win.Length = &IntegerLiteral{Val: args[0].(*IntegerLiteral).Val}
  665. if len(args) == 2 {
  666. win.Interval = &IntegerLiteral{Val: args[1].(*IntegerLiteral).Val}
  667. }
  668. return win, nil
  669. }
  670. var unit = 1
  671. v := args[0].(*TimeLiteral).Val
  672. switch v {
  673. case DD:
  674. unit = 24 * 3600 * 1000
  675. case HH:
  676. unit = 3600 * 1000
  677. case MI:
  678. unit = 60 * 1000
  679. case SS:
  680. unit = 1000
  681. case MS:
  682. unit = 1
  683. default:
  684. return nil, fmt.Errorf("Invalid timeliteral %s", v)
  685. }
  686. win.Length = &IntegerLiteral{Val: args[1].(*IntegerLiteral).Val * unit}
  687. if len(args) > 2 {
  688. win.Interval = &IntegerLiteral{Val: args[2].(*IntegerLiteral).Val * unit}
  689. } else {
  690. win.Interval = &IntegerLiteral{Val: 0}
  691. }
  692. return win, nil
  693. }
  694. func (p *Parser) ParseCreateStreamStmt() (*StreamStmt, error) {
  695. stmt := &StreamStmt{}
  696. if tok, _ := p.scanIgnoreWhitespace(); tok == CREATE {
  697. if tok1, lit1 := p.scanIgnoreWhitespace(); tok1 == STREAM {
  698. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == IDENT {
  699. stmt.Name = StreamName(lit2)
  700. if fields, err := p.parseStreamFields(); err != nil {
  701. return nil, err
  702. } else {
  703. stmt.StreamFields = fields
  704. }
  705. if opts, err := p.parseStreamOptions(); err != nil {
  706. return nil, err
  707. } else {
  708. stmt.Options = opts
  709. }
  710. if tok3, lit3 := p.scanIgnoreWhitespace(); tok3 == SEMICOLON {
  711. p.unscan()
  712. } else if tok3 == EOF {
  713. //Finish parsing create stream statement. Jump to validate
  714. } else {
  715. return nil, fmt.Errorf("found %q, expected semicolon or EOF.", lit3)
  716. }
  717. } else {
  718. return nil, fmt.Errorf("found %q, expected stream name.", lit2)
  719. }
  720. } else {
  721. return nil, fmt.Errorf("found %q, expected keyword stream.", lit1)
  722. }
  723. } else {
  724. p.unscan()
  725. return nil, nil
  726. }
  727. if valErr := validateStream(stmt); valErr != nil {
  728. return nil, valErr
  729. }
  730. return stmt, nil
  731. }
  732. func validateStream(stmt *StreamStmt) error {
  733. f, ok := stmt.Options["FORMAT"]
  734. if !ok {
  735. f = common.FORMAT_JSON
  736. }
  737. switch strings.ToLower(f) {
  738. case common.FORMAT_JSON:
  739. //do nothing
  740. case common.FORMAT_BINARY:
  741. switch len(stmt.StreamFields) {
  742. case 0:
  743. // do nothing for schemaless
  744. case 1:
  745. f := stmt.StreamFields[0]
  746. if bt, ok := f.FieldType.(*BasicType); ok {
  747. if bt.Type == BYTEA {
  748. break
  749. }
  750. }
  751. return fmt.Errorf("'binary' format stream can have only 'bytea' type field")
  752. default:
  753. return fmt.Errorf("'binary' format stream can have only one field")
  754. }
  755. default:
  756. return fmt.Errorf("option 'format=%s' is invalid", f)
  757. }
  758. return nil
  759. }
  760. func (p *Parser) parseShowStreamsStmt() (*ShowStreamsStatement, error) {
  761. ss := &ShowStreamsStatement{}
  762. if tok, _ := p.scanIgnoreWhitespace(); tok == SHOW {
  763. if tok1, lit1 := p.scanIgnoreWhitespace(); tok1 == STREAMS {
  764. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == EOF || tok2 == SEMICOLON {
  765. return ss, nil
  766. } else {
  767. return nil, fmt.Errorf("found %q, expected semecolon or EOF.", lit2)
  768. }
  769. } else {
  770. return nil, fmt.Errorf("found %q, expected keyword streams.", lit1)
  771. }
  772. } else {
  773. p.unscan()
  774. return nil, nil
  775. }
  776. return ss, nil
  777. }
  778. func (p *Parser) parseDescribeStreamStmt() (*DescribeStreamStatement, error) {
  779. dss := &DescribeStreamStatement{}
  780. if tok, _ := p.scanIgnoreWhitespace(); tok == DESCRIBE {
  781. if tok1, lit1 := p.scanIgnoreWhitespace(); tok1 == STREAM {
  782. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == IDENT {
  783. dss.Name = lit2
  784. return dss, nil
  785. } else {
  786. return nil, fmt.Errorf("found %q, expected stream name.", lit2)
  787. }
  788. } else {
  789. return nil, fmt.Errorf("found %q, expected keyword stream.", lit1)
  790. }
  791. } else {
  792. p.unscan()
  793. return nil, nil
  794. }
  795. }
  796. func (p *Parser) parseExplainStreamsStmt() (*ExplainStreamStatement, error) {
  797. ess := &ExplainStreamStatement{}
  798. if tok, _ := p.scanIgnoreWhitespace(); tok == EXPLAIN {
  799. if tok1, lit1 := p.scanIgnoreWhitespace(); tok1 == STREAM {
  800. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == IDENT {
  801. ess.Name = lit2
  802. return ess, nil
  803. } else {
  804. return nil, fmt.Errorf("found %q, expected stream name.", lit2)
  805. }
  806. } else {
  807. return nil, fmt.Errorf("found %q, expected keyword stream.", lit1)
  808. }
  809. } else {
  810. p.unscan()
  811. return nil, nil
  812. }
  813. }
  814. func (p *Parser) parseDropStreamsStmt() (*DropStreamStatement, error) {
  815. ess := &DropStreamStatement{}
  816. if tok, _ := p.scanIgnoreWhitespace(); tok == DROP {
  817. if tok1, lit1 := p.scanIgnoreWhitespace(); tok1 == STREAM {
  818. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == IDENT {
  819. ess.Name = lit2
  820. return ess, nil
  821. } else {
  822. return nil, fmt.Errorf("found %q, expected stream name.", lit2)
  823. }
  824. } else {
  825. return nil, fmt.Errorf("found %q, expected keyword stream.", lit1)
  826. }
  827. } else {
  828. p.unscan()
  829. return nil, nil
  830. }
  831. }
  832. func (p *Parser) parseStreamFields() (StreamFields, error) {
  833. lStack := &stack.Stack{}
  834. var fields StreamFields
  835. if tok, lit := p.scanIgnoreWhitespace(); tok == LPAREN {
  836. lStack.Push(lit)
  837. for {
  838. //For the schemaless streams
  839. //create stream demo () WITH (FORMAT="JSON", DATASOURCE="demo" TYPE="edgex")
  840. if tok1, _ := p.scanIgnoreWhitespace(); tok1 == RPAREN {
  841. lStack.Pop()
  842. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 != WITH {
  843. return nil, fmt.Errorf("found %q, expected is with.", lit2)
  844. }
  845. return fields, nil
  846. } else {
  847. p.unscan()
  848. }
  849. if f, err := p.parseStreamField(); err != nil {
  850. return nil, err
  851. } else {
  852. fields = append(fields, *f)
  853. }
  854. if tok1, _ := p.scanIgnoreWhitespace(); tok1 == RPAREN {
  855. lStack.Pop()
  856. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == WITH {
  857. //Check the stack for LPAREN; If the stack for LPAREN is not zero, then it's not correct.
  858. if lStack.Len() > 0 {
  859. return nil, fmt.Errorf("Parenthesis is not matched.")
  860. }
  861. break
  862. } else if tok2 == COMMA {
  863. if lStack.Len() > 0 {
  864. return nil, fmt.Errorf("Parenthesis is in create record type not matched.")
  865. }
  866. p.unscan()
  867. break
  868. } else if tok2 == RPAREN { //The nested type definition of ARRAY and Struct, such as "field ARRAY(STRUCT(f BIGINT))"
  869. if lStack.Len() > 0 {
  870. return nil, fmt.Errorf("Parenthesis is not matched.")
  871. }
  872. p.unscan()
  873. break
  874. } else {
  875. if lStack.Len() == 0 {
  876. return nil, fmt.Errorf("found %q, expected is with.", lit2)
  877. }
  878. p.unscan()
  879. }
  880. } else {
  881. p.unscan()
  882. }
  883. }
  884. } else {
  885. return nil, fmt.Errorf("found %q, expected lparen after stream name.", lit)
  886. }
  887. return fields, nil
  888. }
  889. func (p *Parser) parseStreamField() (*StreamField, error) {
  890. field := &StreamField{}
  891. if tok, lit := p.scanIgnoreWhitespace(); tok == IDENT {
  892. field.Name = lit
  893. tok1, lit1 := p.scanIgnoreWhitespace()
  894. if t := getDataType(tok1); t != UNKNOWN && t.isSimpleType() {
  895. field.FieldType = &BasicType{Type: t}
  896. } else if t == ARRAY {
  897. if f, e := p.parseStreamArrayType(); e != nil {
  898. return nil, e
  899. } else {
  900. field.FieldType = f
  901. }
  902. } else if t == STRUCT {
  903. if f, e := p.parseStreamStructType(); e != nil {
  904. return nil, e
  905. } else {
  906. field.FieldType = f
  907. }
  908. } else if t == UNKNOWN {
  909. return nil, fmt.Errorf("found %q, expect valid stream field types(BIGINT | FLOAT | STRINGS | DATETIME | BOOLEAN | ARRAY | STRUCT).", lit1)
  910. }
  911. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == COMMA {
  912. //Just consume the comma.
  913. } else if tok2 == RPAREN {
  914. p.unscan()
  915. } else {
  916. return nil, fmt.Errorf("found %q, expect comma or rparen.", lit2)
  917. }
  918. } else {
  919. return nil, fmt.Errorf("found %q, expect stream field name.", lit)
  920. }
  921. return field, nil
  922. }
  923. func (p *Parser) parseStreamArrayType() (FieldType, error) {
  924. lStack := &stack.Stack{}
  925. if tok, _ := p.scanIgnoreWhitespace(); tok == LPAREN {
  926. lStack.Push(LPAREN)
  927. tok1, lit1 := p.scanIgnoreWhitespace()
  928. if t := getDataType(tok1); t != UNKNOWN && t.isSimpleType() {
  929. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == RPAREN {
  930. lStack.Pop()
  931. if lStack.Len() > 0 {
  932. return nil, fmt.Errorf("Parenthesis is in array type not matched.")
  933. }
  934. return &ArrayType{Type: t}, nil
  935. } else {
  936. return nil, fmt.Errorf("found %q, expect rparen in array type definition.", lit2)
  937. }
  938. } else if tok1 == XSTRUCT {
  939. if f, err := p.parseStreamStructType(); err != nil {
  940. return nil, err
  941. } else {
  942. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == RPAREN {
  943. lStack.Pop()
  944. if lStack.Len() > 0 {
  945. return nil, fmt.Errorf("Parenthesis is in struct of array type %q not matched.", tok1)
  946. }
  947. return &ArrayType{Type: STRUCT, FieldType: f}, nil
  948. } else {
  949. return nil, fmt.Errorf("found %q, expect rparen in struct of array type definition.", lit2)
  950. }
  951. }
  952. } else if tok1 == COMMA {
  953. p.unscan()
  954. } else {
  955. return nil, fmt.Errorf("found %q, expect stream data types.", lit1)
  956. }
  957. } else {
  958. }
  959. return nil, nil
  960. }
  961. func (p *Parser) parseStreamStructType() (FieldType, error) {
  962. rf := &RecType{}
  963. if sfs, err := p.parseStreamFields(); err != nil {
  964. return nil, err
  965. } else {
  966. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == COMMA {
  967. rf.StreamFields = sfs
  968. p.unscan()
  969. } else if tok2 == RPAREN {
  970. rf.StreamFields = sfs
  971. p.unscan()
  972. } else {
  973. return nil, fmt.Errorf("found %q, expect comma in create stream record statement.", lit2)
  974. }
  975. }
  976. return rf, nil
  977. }
  978. func (p *Parser) parseStreamOptions() (map[string]string, error) {
  979. var opts = make(map[string]string)
  980. lStack := &stack.Stack{}
  981. if tok, lit := p.scanIgnoreWhitespace(); tok == LPAREN {
  982. lStack.Push(LPAREN)
  983. for {
  984. if tok1, lit1 := p.scanIgnoreWhitespace(); tok1 == DATASOURCE || tok1 == FORMAT || tok1 == KEY || tok1 == CONF_KEY || tok1 == STRICT_VALIDATION || tok1 == TYPE || tok1 == TIMESTAMP || tok1 == TIMESTAMP_FORMAT {
  985. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == EQ {
  986. if tok3, lit3 := p.scanIgnoreWhitespace(); tok3 == STRING {
  987. if tok1 == STRICT_VALIDATION {
  988. if val := strings.ToUpper(lit3); (val != "TRUE") && (val != "FALSE") {
  989. return nil, fmt.Errorf("found %q, expect TRUE/FALSE value in %s option.", lit3, tok1)
  990. }
  991. }
  992. opts[lit1] = lit3
  993. } else {
  994. return nil, fmt.Errorf("found %q, expect string value in option.", lit3)
  995. }
  996. } else {
  997. return nil, fmt.Errorf("found %q, expect equals(=) in options.", lit2)
  998. }
  999. } else if tok1 == COMMA {
  1000. continue
  1001. } else if tok1 == RPAREN {
  1002. if lStack.Pop(); lStack.Len() == 0 {
  1003. break
  1004. } else {
  1005. return nil, fmt.Errorf("Parenthesis is not matched in options definition.")
  1006. }
  1007. } else {
  1008. return nil, fmt.Errorf("found %q, unknown option keys(DATASOURCE|FORMAT|KEY|CONF_KEY|STRICT_VALIDATION|TYPE).", lit1)
  1009. }
  1010. }
  1011. } else {
  1012. return nil, fmt.Errorf("found %q, expect stream options.", lit)
  1013. }
  1014. return opts, nil
  1015. }
  1016. // Only support filter on window now
  1017. func (p *Parser) parseFilter() (Expr, error) {
  1018. if tok, _ := p.scanIgnoreWhitespace(); tok != FILTER {
  1019. p.unscan()
  1020. return nil, nil
  1021. }
  1022. if tok, lit := p.scanIgnoreWhitespace(); tok != LPAREN {
  1023. return nil, fmt.Errorf("Found %q after FILTER, expect parentheses.", lit)
  1024. }
  1025. if tok, lit := p.scanIgnoreWhitespace(); tok != WHERE {
  1026. return nil, fmt.Errorf("Found %q after FILTER(, expect WHERE.", lit)
  1027. }
  1028. expr, err := p.ParseExpr()
  1029. if err != nil {
  1030. return nil, err
  1031. }
  1032. if tok, lit := p.scanIgnoreWhitespace(); tok != RPAREN {
  1033. return nil, fmt.Errorf("Found %q after FILTER, expect right parentheses.", lit)
  1034. }
  1035. return expr, nil
  1036. }