parser.go 36 KB

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