parser.go 24 KB

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