parser.go 34 KB

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