parser.go 31 KB

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