parser.go 35 KB

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