parser.go 23 KB

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