parser.go 39 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505
  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. p.unscan()
  343. if exp, err := p.ParseExpr(); err != nil {
  344. return nil, err
  345. } else {
  346. s.FieldExpr = exp
  347. }
  348. if t2, _ := p.scanIgnoreWhitespace(); t2 == ast.DESC {
  349. s.Ascending = false
  350. ss = append(ss, s)
  351. } else if t2 == ast.ASC {
  352. ss = append(ss, s)
  353. } else {
  354. ss = append(ss, s)
  355. p.unscan()
  356. continue
  357. }
  358. } else if t1 == ast.COMMA {
  359. continue
  360. } else {
  361. p.unscan()
  362. break
  363. }
  364. }
  365. } else {
  366. return nil, fmt.Errorf("found %q, expected BY keyword.", l1)
  367. }
  368. } else {
  369. p.unscan()
  370. }
  371. return ss, nil
  372. }
  373. func (p *Parser) parseFields() (ast.Fields, error) {
  374. var fields ast.Fields
  375. tok, _ := p.scanIgnoreWhitespace()
  376. if tok == ast.ASTERISK {
  377. fields = append(fields, ast.Field{AName: "", Expr: &ast.Wildcard{Token: tok}})
  378. return fields, nil
  379. }
  380. p.unscan()
  381. for {
  382. field, err := p.parseField()
  383. if err != nil {
  384. return nil, err
  385. } else {
  386. fields = append(fields, *field)
  387. }
  388. tok, _ = p.scanIgnoreWhitespace()
  389. if tok != ast.COMMA {
  390. p.unscan()
  391. break
  392. }
  393. }
  394. return fields, nil
  395. }
  396. func (p *Parser) parseField() (*ast.Field, error) {
  397. field := &ast.Field{}
  398. if exp, err := p.ParseExpr(); err != nil {
  399. return nil, err
  400. } else {
  401. field.Name = nameExpr(exp)
  402. field.Expr = exp
  403. }
  404. if alias, err := p.parseAlias(); err != nil {
  405. return nil, err
  406. } else {
  407. if alias != "" {
  408. field.AName = alias
  409. }
  410. }
  411. if field.Name == "" && field.AName == "" {
  412. field.Name = DEFAULT_FIELD_NAME_PREFIX + strconv.Itoa(p.f)
  413. p.f += 1
  414. }
  415. return field, nil
  416. }
  417. func nameExpr(exp ast.Expr) string {
  418. switch e := exp.(type) {
  419. case *ast.FieldRef:
  420. return e.Name
  421. case *ast.Call:
  422. return e.Name
  423. default:
  424. return ""
  425. }
  426. }
  427. func (p *Parser) parseAlias() (string, error) {
  428. tok, lit := p.scanIgnoreWhitespace()
  429. if tok == ast.AS {
  430. if tok, lit = p.scanIgnoreWhitespace(); tok != ast.IDENT {
  431. return "", fmt.Errorf("found %q, expected as alias.", lit)
  432. } else {
  433. return lit, nil
  434. }
  435. }
  436. p.unscan()
  437. return "", nil
  438. }
  439. func (p *Parser) ParseExpr() (ast.Expr, error) {
  440. var err error
  441. root := &ast.BinaryExpr{}
  442. root.RHS, err = p.parseUnaryExpr(false)
  443. if err != nil {
  444. return nil, err
  445. }
  446. for {
  447. op, _ := p.scanIgnoreWhitespace()
  448. if !op.IsOperator() {
  449. p.unscan()
  450. return root.RHS, nil
  451. } else if op == ast.ASTERISK { //Change the asterisk to Mul token.
  452. op = ast.MUL
  453. } else if op == ast.LBRACKET { //LBRACKET is a special token, need to unscan
  454. op = ast.SUBSET
  455. p.unscan()
  456. } else if op == ast.IN { //IN is a special token, need to unscan
  457. p.unscan()
  458. } else if op == ast.NOT {
  459. afterNot, tk1 := p.scanIgnoreWhitespace()
  460. switch afterNot {
  461. case ast.IN: //IN is a special token, need to unscan
  462. op = ast.NOTIN
  463. p.unscan()
  464. break
  465. case ast.BETWEEN:
  466. op = ast.NOTBETWEEN
  467. node := root
  468. var lhs ast.Expr
  469. for {
  470. r, ok := node.RHS.(*ast.BinaryExpr)
  471. if !ok || r.OP.Precedence() >= op.Precedence() {
  472. lhs = node.RHS
  473. break
  474. }
  475. node = r
  476. }
  477. expr, err := p.parseBetween(lhs, ast.NOTBETWEEN)
  478. if err != nil {
  479. return nil, err
  480. }
  481. node.RHS = expr
  482. continue
  483. case ast.LIKE:
  484. op = ast.NOTLIKE
  485. default:
  486. return nil, fmt.Errorf("found %q, expected expression", tk1)
  487. }
  488. } else if op == ast.BETWEEN {
  489. node := root
  490. var lhs ast.Expr
  491. for {
  492. r, ok := node.RHS.(*ast.BinaryExpr)
  493. if !ok || r.OP.Precedence() >= op.Precedence() {
  494. lhs = node.RHS
  495. break
  496. }
  497. node = r
  498. }
  499. expr, err := p.parseBetween(lhs, op)
  500. if err != nil {
  501. return nil, err
  502. }
  503. node.RHS = expr
  504. continue
  505. }
  506. var rhs ast.Expr
  507. if rhs, err = p.parseUnaryExpr(op == ast.ARROW); err != nil {
  508. return nil, err
  509. }
  510. if op == ast.LIKE || op == ast.NOTLIKE {
  511. lp := &ast.LikePattern{
  512. Expr: rhs,
  513. }
  514. if l, ok := lp.Expr.(*ast.StringLiteral); ok {
  515. lp.Pattern, err = lp.Compile(l.Val)
  516. if err != nil {
  517. return nil, fmt.Errorf("invalid LIKE pattern: %s", err)
  518. }
  519. }
  520. rhs = lp
  521. }
  522. for node := root; ; {
  523. r, ok := node.RHS.(*ast.BinaryExpr)
  524. if !ok || r.OP.Precedence() >= op.Precedence() {
  525. node.RHS = &ast.BinaryExpr{LHS: node.RHS, RHS: rhs, OP: op}
  526. break
  527. }
  528. node = r
  529. }
  530. }
  531. }
  532. func (p *Parser) parseBetween(lhs ast.Expr, op ast.Token) (ast.Expr, error) {
  533. alhs, err := p.parseUnaryExpr(false)
  534. if err != nil {
  535. return nil, err
  536. }
  537. opp, _ := p.scanIgnoreWhitespace()
  538. if opp != ast.AND {
  539. return nil, fmt.Errorf("expect AND expression after between but found %s", opp)
  540. }
  541. arhs, err := p.parseUnaryExpr(false)
  542. if err != nil {
  543. return nil, err
  544. }
  545. return &ast.BinaryExpr{
  546. LHS: lhs,
  547. OP: op,
  548. RHS: &ast.BetweenExpr{
  549. Lower: alhs,
  550. Higher: arhs,
  551. },
  552. }, nil
  553. }
  554. func (p *Parser) parseUnaryExpr(isSubField bool) (ast.Expr, error) {
  555. if tok1, _ := p.scanIgnoreWhitespace(); tok1 == ast.LPAREN {
  556. expr, err := p.ParseExpr()
  557. if err != nil {
  558. return nil, err
  559. }
  560. // Expect an RPAREN at the end.
  561. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 != ast.RPAREN {
  562. return nil, fmt.Errorf("found %q, expected right paren.", lit2)
  563. }
  564. return &ast.ParenExpr{Expr: expr}, nil
  565. } else if tok1 == ast.LBRACKET {
  566. return p.parseBracketExpr()
  567. } else if tok1 == ast.IN {
  568. return p.parseValueSetExpr()
  569. }
  570. p.unscan()
  571. tok, lit := p.scanIgnoreWhiteSpaceWithNegativeNum()
  572. if tok == ast.CASE {
  573. return p.parseCaseExpr()
  574. } else if tok == ast.IDENT {
  575. if tok1, _ := p.scanIgnoreWhitespace(); tok1 == ast.LPAREN {
  576. return p.parseCall(lit)
  577. }
  578. p.unscan() //Back the Lparen token
  579. p.unscan() //Back the ident token
  580. if n, err := p.parseFieldNameSections(); err != nil {
  581. return nil, err
  582. } else {
  583. if p.inmeta() {
  584. if len(n) == 2 {
  585. return &ast.MetaRef{StreamName: ast.StreamName(n[0]), Name: n[1]}, nil
  586. }
  587. if isSubField {
  588. return &ast.JsonFieldRef{Name: n[0]}, nil
  589. }
  590. return &ast.MetaRef{StreamName: ast.DefaultStream, Name: n[0]}, nil
  591. } else {
  592. if len(n) == 2 {
  593. return &ast.FieldRef{StreamName: ast.StreamName(n[0]), Name: n[1]}, nil
  594. }
  595. if isSubField {
  596. return &ast.JsonFieldRef{Name: n[0]}, nil
  597. }
  598. return &ast.FieldRef{StreamName: ast.DefaultStream, Name: n[0]}, nil
  599. }
  600. }
  601. } else if tok == ast.STRING {
  602. return &ast.StringLiteral{Val: lit}, nil
  603. } else if tok == ast.INTEGER {
  604. val, _ := strconv.Atoi(lit)
  605. return &ast.IntegerLiteral{Val: val}, nil
  606. } else if tok == ast.NUMBER {
  607. if v, err := strconv.ParseFloat(lit, 64); err != nil {
  608. return nil, fmt.Errorf("found %q, invalid number value.", lit)
  609. } else {
  610. return &ast.NumberLiteral{Val: v}, nil
  611. }
  612. } else if tok == ast.TRUE || tok == ast.FALSE {
  613. if v, err := strconv.ParseBool(lit); err != nil {
  614. return nil, fmt.Errorf("found %q, invalid boolean value.", lit)
  615. } else {
  616. return &ast.BooleanLiteral{Val: v}, nil
  617. }
  618. } else if tok.IsTimeLiteral() {
  619. return &ast.TimeLiteral{Val: tok}, nil
  620. } else if tok == ast.ASTERISK {
  621. return p.parseAsterisk()
  622. }
  623. return nil, fmt.Errorf("found %q, expected expression.", lit)
  624. }
  625. func (p *Parser) parseValueSetExpr() (ast.Expr, error) {
  626. valsetExpr := &ast.ValueSetExpr{
  627. LiteralExprs: nil,
  628. ArrayExpr: nil,
  629. }
  630. // IN ("A", "B") or IN expression
  631. tk, _ := p.scanIgnoreWhitespace()
  632. if tk == ast.LPAREN {
  633. for {
  634. element, err := p.ParseExpr()
  635. if err != nil {
  636. return nil, fmt.Errorf("expect elements for IN expression, but %v", err)
  637. }
  638. valsetExpr.LiteralExprs = append(valsetExpr.LiteralExprs, element)
  639. if tok2, _ := p.scanIgnoreWhitespace(); tok2 != ast.COMMA {
  640. p.unscan()
  641. break
  642. }
  643. }
  644. if tok, lit := p.scanIgnoreWhitespace(); tok != ast.RPAREN {
  645. return nil, fmt.Errorf("expect ) for IN expression, but got %q", lit)
  646. }
  647. return valsetExpr, nil
  648. } else {
  649. //back to IN
  650. p.unscan()
  651. }
  652. if exp, err := p.parseUnaryExpr(false); err != nil {
  653. return nil, fmt.Errorf("expect expression after IN, but got error %v", err)
  654. } else {
  655. return exp, nil
  656. }
  657. }
  658. func (p *Parser) parseBracketExpr() (ast.Expr, error) {
  659. tok2, lit2 := p.scanIgnoreWhiteSpaceWithNegativeNum()
  660. if tok2 == ast.RBRACKET {
  661. //field[]
  662. return &ast.ColonExpr{Start: &ast.IntegerLiteral{Val: 0}, End: &ast.IntegerLiteral{Val: math.MinInt32}}, nil
  663. } else if tok2 == ast.INTEGER {
  664. start, err := strconv.Atoi(lit2)
  665. if err != nil {
  666. return nil, fmt.Errorf("The start index %s is not an int value in bracket expression.", lit2)
  667. }
  668. if tok3, _ := p.scanIgnoreWhitespace(); tok3 == ast.RBRACKET {
  669. //Such as field[2]
  670. return &ast.IndexExpr{Index: &ast.IntegerLiteral{Val: start}}, nil
  671. } else if tok3 == ast.COLON {
  672. //Such as field[2:] or field[2:4]
  673. return p.parseColonExpr(&ast.IntegerLiteral{Val: start})
  674. }
  675. } else if tok2 == ast.COLON {
  676. //Such as field[:3] or [:]
  677. return p.parseColonExpr(&ast.IntegerLiteral{Val: 0})
  678. } else {
  679. p.unscan()
  680. start, err := p.ParseExpr()
  681. if err != nil {
  682. return nil, fmt.Errorf("The start index %s is invalid in bracket expression.", lit2)
  683. }
  684. if tok3, _ := p.scanIgnoreWhitespace(); tok3 == ast.RBRACKET {
  685. //Such as field[2]
  686. return &ast.IndexExpr{Index: start}, nil
  687. } else if tok3 == ast.COLON {
  688. //Such as field[2:] or field[2:4]
  689. return p.parseColonExpr(start)
  690. }
  691. }
  692. return nil, fmt.Errorf("Unexpected token %q. when parsing bracket expressions.", lit2)
  693. }
  694. func (p *Parser) parseColonExpr(start ast.Expr) (ast.Expr, error) {
  695. tok, lit := p.scanIgnoreWhiteSpaceWithNegativeNum()
  696. if tok == ast.INTEGER {
  697. end, err := strconv.Atoi(lit)
  698. if err != nil {
  699. return nil, fmt.Errorf("The end index %s is not an int value in bracket expression.", lit)
  700. }
  701. if tok1, lit1 := p.scanIgnoreWhitespace(); tok1 == ast.RBRACKET {
  702. return &ast.ColonExpr{Start: start, End: &ast.IntegerLiteral{Val: end}}, nil
  703. } else {
  704. return nil, fmt.Errorf("Found %q, expected right bracket.", lit1)
  705. }
  706. } else if tok == ast.RBRACKET {
  707. return &ast.ColonExpr{Start: start, End: &ast.IntegerLiteral{Val: math.MinInt32}}, nil
  708. }
  709. return nil, fmt.Errorf("Found %q, expected right bracket.", lit)
  710. }
  711. func (p *Parser) scanIgnoreWhiteSpaceWithNegativeNum() (ast.Token, string) {
  712. tok, lit := p.scanIgnoreWhitespace()
  713. if tok == ast.SUB {
  714. _, _ = p.s.ScanWhiteSpace()
  715. r := p.s.read()
  716. if isDigit(r) {
  717. p.s.unread()
  718. tok, lit = p.s.ScanNumber(false, true)
  719. }
  720. }
  721. return tok, lit
  722. }
  723. func (p *Parser) parseAs(f *ast.Field) (*ast.Field, error) {
  724. tok, lit := p.scanIgnoreWhitespace()
  725. if tok != ast.IDENT {
  726. return nil, fmt.Errorf("found %q, expected as alias.", lit)
  727. }
  728. f.AName = lit
  729. return f, nil
  730. }
  731. var WindowFuncs = map[string]struct{}{
  732. "tumblingwindow": {},
  733. "hoppingwindow": {},
  734. "sessionwindow": {},
  735. "slidingwindow": {},
  736. "countwindow": {},
  737. }
  738. func convFuncName(n string) (string, bool) {
  739. lname := strings.ToLower(n)
  740. if _, ok := WindowFuncs[lname]; ok {
  741. return lname, ok
  742. } else {
  743. return function.ConvName(n)
  744. }
  745. }
  746. func (p *Parser) parseCall(n string) (ast.Expr, error) {
  747. // Check if n function exists and convert it to lowercase for built-in func
  748. name, ok := convFuncName(n)
  749. if !ok {
  750. return nil, fmt.Errorf("function %s not found", n)
  751. }
  752. p.inFunc = name
  753. defer func() { p.inFunc = "" }()
  754. ft := function.GetFuncType(name)
  755. if ft == ast.FuncTypeCols && p.clause != "select" {
  756. return nil, fmt.Errorf("function %s can only be used inside the select clause", n)
  757. }
  758. var args []ast.Expr
  759. for {
  760. if tok, _ := p.scanIgnoreWhitespace(); tok == ast.RPAREN {
  761. if valErr := validateFuncs(name, nil); valErr != nil {
  762. return nil, valErr
  763. }
  764. c := &ast.Call{Name: name, Args: args, FuncId: p.fn, FuncType: ft}
  765. p.fn += 1
  766. return c, nil
  767. } else {
  768. p.unscan()
  769. }
  770. if exp, err := p.ParseExpr(); err != nil {
  771. return nil, err
  772. } else {
  773. if ft == ast.FuncTypeCols {
  774. field := &ast.ColFuncField{Expr: exp, Name: nameExpr(exp)}
  775. args = append(args, field)
  776. } else {
  777. args = append(args, exp)
  778. }
  779. }
  780. if tok, _ := p.scanIgnoreWhitespace(); tok != ast.COMMA {
  781. p.unscan()
  782. break
  783. }
  784. }
  785. if tok, lit := p.scanIgnoreWhitespace(); tok != ast.RPAREN {
  786. return nil, fmt.Errorf("found function call %q, expected ), but with %q.", name, lit)
  787. }
  788. if wt, err := validateWindows(name, args); wt == ast.NOT_WINDOW {
  789. if valErr := validateFuncs(name, args); valErr != nil {
  790. return nil, valErr
  791. }
  792. // Add context for some aggregate func
  793. if name == "deduplicate" {
  794. args = append([]ast.Expr{&ast.Wildcard{Token: ast.ASTERISK}}, args...)
  795. }
  796. c := &ast.Call{Name: name, Args: args, FuncId: p.fn, FuncType: ft}
  797. p.fn += 1
  798. return c, nil
  799. } else {
  800. if err != nil {
  801. return nil, err
  802. }
  803. win, err := p.ConvertToWindows(wt, args)
  804. if err != nil {
  805. return nil, err
  806. }
  807. // parse filter clause
  808. f, err := p.parseFilter()
  809. if err != nil {
  810. return nil, err
  811. } else if f != nil {
  812. win.Filter = f
  813. }
  814. return win, nil
  815. }
  816. }
  817. func (p *Parser) parseCaseExpr() (*ast.CaseExpr, error) {
  818. c := &ast.CaseExpr{}
  819. tok, _ := p.scanIgnoreWhitespace()
  820. p.unscan()
  821. if tok != ast.WHEN { // no condition value for case, additional validation needed
  822. if exp, err := p.ParseExpr(); err != nil {
  823. return nil, err
  824. } else {
  825. c.Value = exp
  826. }
  827. }
  828. loop:
  829. for {
  830. tok, _ := p.scanIgnoreWhitespace()
  831. switch tok {
  832. case ast.WHEN:
  833. if exp, err := p.ParseExpr(); err != nil {
  834. return nil, err
  835. } else {
  836. if c.WhenClauses == nil {
  837. c.WhenClauses = make([]*ast.WhenClause, 0)
  838. }
  839. if c.Value == nil && !ast.IsBooleanArg(exp) {
  840. return nil, fmt.Errorf("invalid CASE expression, WHEN expression must be a bool condition")
  841. }
  842. w := &ast.WhenClause{
  843. Expr: exp,
  844. }
  845. tokThen, _ := p.scanIgnoreWhitespace()
  846. if tokThen != ast.THEN {
  847. return nil, fmt.Errorf("invalid CASE expression, THEN expected after WHEN")
  848. } else {
  849. if expThen, err := p.ParseExpr(); err != nil {
  850. return nil, err
  851. } else {
  852. w.Result = expThen
  853. c.WhenClauses = append(c.WhenClauses, w)
  854. }
  855. }
  856. }
  857. case ast.ELSE:
  858. if c.WhenClauses != nil {
  859. if exp, err := p.ParseExpr(); err != nil {
  860. return nil, err
  861. } else {
  862. c.ElseClause = exp
  863. }
  864. } else {
  865. return nil, fmt.Errorf("invalid CASE expression, WHEN expected before ELSE")
  866. }
  867. case ast.END:
  868. if c.WhenClauses != nil {
  869. break loop
  870. } else {
  871. return nil, fmt.Errorf("invalid CASE expression, WHEN expected before END")
  872. }
  873. default:
  874. return nil, fmt.Errorf("invalid CASE expression, END expected")
  875. }
  876. }
  877. return c, nil
  878. }
  879. func validateWindows(fname string, args []ast.Expr) (ast.WindowType, error) {
  880. switch fname {
  881. case "tumblingwindow":
  882. if err := validateWindow(fname, 2, args); err != nil {
  883. return ast.TUMBLING_WINDOW, err
  884. }
  885. return ast.TUMBLING_WINDOW, nil
  886. case "hoppingwindow":
  887. if err := validateWindow(fname, 3, args); err != nil {
  888. return ast.HOPPING_WINDOW, err
  889. }
  890. return ast.HOPPING_WINDOW, nil
  891. case "sessionwindow":
  892. if err := validateWindow(fname, 3, args); err != nil {
  893. return ast.SESSION_WINDOW, err
  894. }
  895. return ast.SESSION_WINDOW, nil
  896. case "slidingwindow":
  897. if err := validateWindow(fname, 2, args); err != nil {
  898. return ast.SLIDING_WINDOW, err
  899. }
  900. return ast.SLIDING_WINDOW, nil
  901. case "countwindow":
  902. if len(args) == 1 {
  903. if para1, ok := args[0].(*ast.IntegerLiteral); ok && para1.Val > 0 {
  904. return ast.COUNT_WINDOW, nil
  905. } else {
  906. return ast.COUNT_WINDOW, fmt.Errorf("Invalid parameter value %s.", args[0])
  907. }
  908. } else if len(args) == 2 {
  909. if para1, ok1 := args[0].(*ast.IntegerLiteral); ok1 {
  910. if para2, ok2 := args[1].(*ast.IntegerLiteral); ok2 {
  911. if para1.Val < para2.Val {
  912. return ast.COUNT_WINDOW, fmt.Errorf("The second parameter value %d should be less than the first parameter %d.", para2.Val, para1.Val)
  913. } else {
  914. return ast.COUNT_WINDOW, nil
  915. }
  916. }
  917. }
  918. return ast.COUNT_WINDOW, fmt.Errorf("Invalid parameter value %s, %s.", args[0], args[1])
  919. } else {
  920. return ast.COUNT_WINDOW, fmt.Errorf("Invalid parameter count.")
  921. }
  922. }
  923. return ast.NOT_WINDOW, nil
  924. }
  925. func validateWindow(funcName string, expectLen int, args []ast.Expr) error {
  926. if len(args) != expectLen {
  927. return fmt.Errorf("The arguments for %s should be %d.\n", funcName, expectLen)
  928. }
  929. if _, ok := args[0].(*ast.TimeLiteral); !ok {
  930. return fmt.Errorf("The 1st argument for %s is expecting timer literal expression. One value of [dd|hh|mi|ss|ms].\n", funcName)
  931. }
  932. for i := 1; i < len(args); i++ {
  933. if _, ok := args[i].(*ast.IntegerLiteral); !ok {
  934. return fmt.Errorf("The %d argument for %s is expecting interger literal expression. \n", i, funcName)
  935. }
  936. }
  937. return nil
  938. }
  939. func (p *Parser) ConvertToWindows(wtype ast.WindowType, args []ast.Expr) (*ast.Window, error) {
  940. win := &ast.Window{WindowType: wtype}
  941. if wtype == ast.COUNT_WINDOW {
  942. win.Length = &ast.IntegerLiteral{Val: args[0].(*ast.IntegerLiteral).Val}
  943. if len(args) == 2 {
  944. win.Interval = &ast.IntegerLiteral{Val: args[1].(*ast.IntegerLiteral).Val}
  945. }
  946. return win, nil
  947. }
  948. var unit = 1
  949. v := args[0].(*ast.TimeLiteral).Val
  950. switch v {
  951. case ast.DD:
  952. unit = 24 * 3600 * 1000
  953. case ast.HH:
  954. unit = 3600 * 1000
  955. case ast.MI:
  956. unit = 60 * 1000
  957. case ast.SS:
  958. unit = 1000
  959. case ast.MS:
  960. unit = 1
  961. default:
  962. return nil, fmt.Errorf("Invalid timeliteral %s", v)
  963. }
  964. win.Length = &ast.IntegerLiteral{Val: args[1].(*ast.IntegerLiteral).Val * unit}
  965. if len(args) > 2 {
  966. win.Interval = &ast.IntegerLiteral{Val: args[2].(*ast.IntegerLiteral).Val * unit}
  967. } else {
  968. win.Interval = &ast.IntegerLiteral{Val: 0}
  969. }
  970. return win, nil
  971. }
  972. func (p *Parser) ParseCreateStmt() (ast.Statement, error) {
  973. if tok, _ := p.scanIgnoreWhitespace(); tok == ast.CREATE {
  974. tok1, lit1 := p.scanIgnoreWhitespace()
  975. stmt := &ast.StreamStmt{}
  976. switch tok1 {
  977. case ast.STREAM:
  978. stmt.StreamType = ast.TypeStream
  979. case ast.TABLE:
  980. stmt.StreamType = ast.TypeTable
  981. default:
  982. return nil, fmt.Errorf("found %q, expected keyword stream or table.", lit1)
  983. }
  984. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.IDENT {
  985. stmt.Name = ast.StreamName(lit2)
  986. if fields, err := p.parseStreamFields(); err != nil {
  987. return nil, err
  988. } else {
  989. stmt.StreamFields = fields
  990. }
  991. if opts, err := p.parseStreamOptions(); err != nil {
  992. return nil, err
  993. } else {
  994. stmt.Options = opts
  995. }
  996. if tok3, lit3 := p.scanIgnoreWhitespace(); tok3 == ast.SEMICOLON {
  997. p.unscan()
  998. } else if tok3 == ast.EOF {
  999. //Finish parsing create stream statement. Jump to validate
  1000. } else {
  1001. return nil, fmt.Errorf("found %q, expected semicolon or EOF.", lit3)
  1002. }
  1003. } else {
  1004. return nil, fmt.Errorf("found %q, expected stream name.", lit2)
  1005. }
  1006. if valErr := validateStream(stmt); valErr != nil {
  1007. return nil, valErr
  1008. }
  1009. return stmt, nil
  1010. } else {
  1011. p.unscan()
  1012. return nil, nil
  1013. }
  1014. }
  1015. // TODO more accurate validation for table
  1016. func validateStream(stmt *ast.StreamStmt) error {
  1017. f := stmt.Options.FORMAT
  1018. if f == "" {
  1019. f = message.FormatJson
  1020. }
  1021. lf := strings.ToLower(f)
  1022. switch lf {
  1023. case message.FormatBinary:
  1024. if stmt.StreamType == ast.TypeTable {
  1025. return fmt.Errorf("'binary' format is not supported for table")
  1026. }
  1027. switch len(stmt.StreamFields) {
  1028. case 0:
  1029. // do nothing for schemaless
  1030. case 1:
  1031. f := stmt.StreamFields[0]
  1032. if bt, ok := f.FieldType.(*ast.BasicType); ok {
  1033. if bt.Type == ast.BYTEA {
  1034. break
  1035. }
  1036. }
  1037. return fmt.Errorf("'binary' format stream can have only 'bytea' type field")
  1038. default:
  1039. return fmt.Errorf("'binary' format stream can have only one field")
  1040. }
  1041. default:
  1042. if !message.IsFormatSupported(lf) {
  1043. return fmt.Errorf("option 'format=%s' is invalid", f)
  1044. }
  1045. }
  1046. return nil
  1047. }
  1048. func (p *Parser) parseShowStmt() (ast.Statement, error) {
  1049. if tok, _ := p.scanIgnoreWhitespace(); tok == ast.SHOW {
  1050. tok1, lit1 := p.scanIgnoreWhitespace()
  1051. switch tok1 {
  1052. case ast.STREAMS:
  1053. ss := &ast.ShowStreamsStatement{}
  1054. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.EOF || tok2 == ast.SEMICOLON {
  1055. return ss, nil
  1056. } else {
  1057. return nil, fmt.Errorf("found %q, expected semecolon or EOF.", lit2)
  1058. }
  1059. case ast.TABLES:
  1060. ss := &ast.ShowTablesStatement{}
  1061. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.EOF || tok2 == ast.SEMICOLON {
  1062. return ss, nil
  1063. } else {
  1064. return nil, fmt.Errorf("found %q, expected semecolon or EOF.", lit2)
  1065. }
  1066. default:
  1067. return nil, fmt.Errorf("found %q, expected keyword streams or tables.", lit1)
  1068. }
  1069. } else {
  1070. p.unscan()
  1071. return nil, nil
  1072. }
  1073. }
  1074. func (p *Parser) parseDescribeStmt() (ast.Statement, error) {
  1075. if tok, _ := p.scanIgnoreWhitespace(); tok == ast.DESCRIBE {
  1076. tok1, lit1 := p.scanIgnoreWhitespace()
  1077. switch tok1 {
  1078. case ast.STREAM:
  1079. dss := &ast.DescribeStreamStatement{}
  1080. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.IDENT {
  1081. dss.Name = lit2
  1082. return dss, nil
  1083. } else {
  1084. return nil, fmt.Errorf("found %q, expected stream name.", lit2)
  1085. }
  1086. case ast.TABLE:
  1087. dss := &ast.DescribeTableStatement{}
  1088. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.IDENT {
  1089. dss.Name = lit2
  1090. return dss, nil
  1091. } else {
  1092. return nil, fmt.Errorf("found %q, expected table name.", lit2)
  1093. }
  1094. default:
  1095. return nil, fmt.Errorf("found %q, expected keyword stream or table.", lit1)
  1096. }
  1097. } else {
  1098. p.unscan()
  1099. return nil, nil
  1100. }
  1101. }
  1102. func (p *Parser) parseExplainStmt() (ast.Statement, error) {
  1103. if tok, _ := p.scanIgnoreWhitespace(); tok == ast.EXPLAIN {
  1104. tok1, lit1 := p.scanIgnoreWhitespace()
  1105. switch tok1 {
  1106. case ast.STREAM:
  1107. ess := &ast.ExplainStreamStatement{}
  1108. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.IDENT {
  1109. ess.Name = lit2
  1110. return ess, nil
  1111. } else {
  1112. return nil, fmt.Errorf("found %q, expected stream name.", lit2)
  1113. }
  1114. case ast.TABLE:
  1115. ess := &ast.ExplainTableStatement{}
  1116. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.IDENT {
  1117. ess.Name = lit2
  1118. return ess, nil
  1119. } else {
  1120. return nil, fmt.Errorf("found %q, expected table name.", lit2)
  1121. }
  1122. default:
  1123. return nil, fmt.Errorf("found %q, expected keyword stream or table.", lit1)
  1124. }
  1125. } else {
  1126. p.unscan()
  1127. return nil, nil
  1128. }
  1129. }
  1130. func (p *Parser) parseDropStmt() (ast.Statement, error) {
  1131. if tok, _ := p.scanIgnoreWhitespace(); tok == ast.DROP {
  1132. tok1, lit1 := p.scanIgnoreWhitespace()
  1133. switch tok1 {
  1134. case ast.STREAM:
  1135. ess := &ast.DropStreamStatement{}
  1136. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.IDENT {
  1137. ess.Name = lit2
  1138. return ess, nil
  1139. } else {
  1140. return nil, fmt.Errorf("found %q, expected stream name.", lit2)
  1141. }
  1142. case ast.TABLE:
  1143. ess := &ast.DropTableStatement{}
  1144. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.IDENT {
  1145. ess.Name = lit2
  1146. return ess, nil
  1147. } else {
  1148. return nil, fmt.Errorf("found %q, expected table name.", lit2)
  1149. }
  1150. default:
  1151. return nil, fmt.Errorf("found %q, expected keyword stream or table.", lit1)
  1152. }
  1153. } else {
  1154. p.unscan()
  1155. return nil, nil
  1156. }
  1157. }
  1158. func (p *Parser) parseStreamFields() (ast.StreamFields, error) {
  1159. lStack := &stack.Stack{}
  1160. var fields ast.StreamFields
  1161. if tok, lit := p.scanIgnoreWhitespace(); tok == ast.LPAREN {
  1162. lStack.Push(lit)
  1163. for {
  1164. //For the schemaless streams
  1165. //create stream demo () WITH (FORMAT="JSON", DATASOURCE="demo" TYPE="edgex")
  1166. if tok1, _ := p.scanIgnoreWhitespace(); tok1 == ast.RPAREN {
  1167. lStack.Pop()
  1168. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 != ast.WITH {
  1169. return nil, fmt.Errorf("found %q, expected is with.", lit2)
  1170. }
  1171. return fields, nil
  1172. } else {
  1173. p.unscan()
  1174. }
  1175. if f, err := p.parseStreamField(); err != nil {
  1176. return nil, err
  1177. } else {
  1178. fields = append(fields, *f)
  1179. }
  1180. if tok1, _ := p.scanIgnoreWhitespace(); tok1 == ast.RPAREN {
  1181. lStack.Pop()
  1182. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.WITH {
  1183. //Check the stack for LPAREN; If the stack for LPAREN is not zero, then it's not correct.
  1184. if lStack.Len() > 0 {
  1185. return nil, fmt.Errorf("Parenthesis is not matched.")
  1186. }
  1187. break
  1188. } else if tok2 == ast.COMMA {
  1189. if lStack.Len() > 0 {
  1190. return nil, fmt.Errorf("Parenthesis is in create record type not matched.")
  1191. }
  1192. p.unscan()
  1193. break
  1194. } else if tok2 == ast.RPAREN { //The nested type definition of ARRAY and Struct, such as "field ARRAY(STRUCT(f BIGINT))"
  1195. if lStack.Len() > 0 {
  1196. return nil, fmt.Errorf("Parenthesis is not matched.")
  1197. }
  1198. p.unscan()
  1199. break
  1200. } else {
  1201. if lStack.Len() == 0 {
  1202. return nil, fmt.Errorf("found %q, expected is with.", lit2)
  1203. }
  1204. p.unscan()
  1205. }
  1206. } else {
  1207. p.unscan()
  1208. }
  1209. }
  1210. } else {
  1211. return nil, fmt.Errorf("found %q, expected lparen after stream name.", lit)
  1212. }
  1213. return fields, nil
  1214. }
  1215. func (p *Parser) parseStreamField() (*ast.StreamField, error) {
  1216. field := &ast.StreamField{}
  1217. if tok, lit := p.scanIgnoreWhitespace(); tok == ast.IDENT {
  1218. field.Name = lit
  1219. tok1, lit1 := p.scanIgnoreWhitespace()
  1220. if t := ast.GetDataType(tok1); t != ast.UNKNOWN && t.IsSimpleType() {
  1221. field.FieldType = &ast.BasicType{Type: t}
  1222. } else if t == ast.ARRAY {
  1223. if f, e := p.parseStreamArrayType(); e != nil {
  1224. return nil, e
  1225. } else {
  1226. field.FieldType = f
  1227. }
  1228. } else if t == ast.STRUCT {
  1229. if f, e := p.parseStreamStructType(); e != nil {
  1230. return nil, e
  1231. } else {
  1232. field.FieldType = f
  1233. }
  1234. } else if t == ast.UNKNOWN {
  1235. return nil, fmt.Errorf("found %q, expect valid stream field types(BIGINT | FLOAT | STRINGS | DATETIME | BOOLEAN | BYTEA | ARRAY | STRUCT).", lit1)
  1236. }
  1237. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.COMMA {
  1238. //Just consume the comma.
  1239. } else if tok2 == ast.RPAREN {
  1240. p.unscan()
  1241. } else {
  1242. return nil, fmt.Errorf("found %q, expect comma or rparen.", lit2)
  1243. }
  1244. } else {
  1245. return nil, fmt.Errorf("found %q, expect stream field name.", lit)
  1246. }
  1247. return field, nil
  1248. }
  1249. func (p *Parser) parseStreamArrayType() (ast.FieldType, error) {
  1250. lStack := &stack.Stack{}
  1251. if tok, _ := p.scanIgnoreWhitespace(); tok == ast.LPAREN {
  1252. lStack.Push(ast.LPAREN)
  1253. tok1, lit1 := p.scanIgnoreWhitespace()
  1254. if t := ast.GetDataType(tok1); t != ast.UNKNOWN && t.IsSimpleType() {
  1255. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.RPAREN {
  1256. lStack.Pop()
  1257. if lStack.Len() > 0 {
  1258. return nil, fmt.Errorf("Parenthesis is in array type not matched.")
  1259. }
  1260. return &ast.ArrayType{Type: t}, nil
  1261. } else {
  1262. return nil, fmt.Errorf("found %q, expect rparen in array type definition.", lit2)
  1263. }
  1264. } else if tok1 == ast.XSTRUCT {
  1265. if f, err := p.parseStreamStructType(); err != nil {
  1266. return nil, err
  1267. } else {
  1268. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.RPAREN {
  1269. lStack.Pop()
  1270. if lStack.Len() > 0 {
  1271. return nil, fmt.Errorf("Parenthesis is in struct of array type %q not matched.", tok1)
  1272. }
  1273. return &ast.ArrayType{Type: ast.STRUCT, FieldType: f}, nil
  1274. } else {
  1275. return nil, fmt.Errorf("found %q, expect rparen in struct of array type definition.", lit2)
  1276. }
  1277. }
  1278. } else if tok1 == ast.COMMA {
  1279. p.unscan()
  1280. } else {
  1281. return nil, fmt.Errorf("found %q, expect stream data types.", lit1)
  1282. }
  1283. } else {
  1284. }
  1285. return nil, nil
  1286. }
  1287. func (p *Parser) parseStreamStructType() (ast.FieldType, error) {
  1288. rf := &ast.RecType{}
  1289. if sfs, err := p.parseStreamFields(); err != nil {
  1290. return nil, err
  1291. } else {
  1292. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.COMMA {
  1293. rf.StreamFields = sfs
  1294. p.unscan()
  1295. } else if tok2 == ast.RPAREN {
  1296. rf.StreamFields = sfs
  1297. p.unscan()
  1298. } else {
  1299. return nil, fmt.Errorf("found %q, expect comma in create stream record statement.", lit2)
  1300. }
  1301. }
  1302. return rf, nil
  1303. }
  1304. func (p *Parser) parseStreamOptions() (*ast.Options, error) {
  1305. opts := &ast.Options{STRICT_VALIDATION: true}
  1306. v := reflect.ValueOf(opts)
  1307. lStack := &stack.Stack{}
  1308. if tok, lit := p.scanIgnoreWhitespace(); tok == ast.LPAREN {
  1309. lStack.Push(ast.LPAREN)
  1310. for {
  1311. 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 || tok1 == ast.SCHEMAID || tok1 == ast.KIND {
  1312. if tok2, lit2 := p.scanIgnoreWhitespace(); tok2 == ast.EQ {
  1313. if tok3, lit3 := p.scanIgnoreWhitespace(); tok3 == ast.STRING {
  1314. switch tok1 {
  1315. case ast.STRICT_VALIDATION:
  1316. if val := strings.ToUpper(lit3); (val != "TRUE") && (val != "FALSE") {
  1317. return nil, fmt.Errorf("found %q, expect TRUE/FALSE value in %s option.", lit3, tok1)
  1318. } else {
  1319. opts.STRICT_VALIDATION = val == "TRUE"
  1320. }
  1321. case ast.RETAIN_SIZE:
  1322. if val, err := strconv.Atoi(lit3); err != nil {
  1323. return nil, fmt.Errorf("found %q, expect number value in %s option.", lit3, tok1)
  1324. } else {
  1325. opts.RETAIN_SIZE = val
  1326. }
  1327. case ast.SHARED:
  1328. if val := strings.ToUpper(lit3); (val != "TRUE") && (val != "FALSE") {
  1329. return nil, fmt.Errorf("found %q, expect TRUE/FALSE value in %s option.", lit3, tok1)
  1330. } else {
  1331. opts.SHARED = val == "TRUE"
  1332. }
  1333. case ast.KIND:
  1334. val := strings.ToLower(lit3)
  1335. opts.KIND = val
  1336. default:
  1337. f := v.Elem().FieldByName(lit1)
  1338. if f.IsValid() {
  1339. f.SetString(lit3)
  1340. } else { // should not happen
  1341. return nil, fmt.Errorf("invalid field %s.", lit1)
  1342. }
  1343. }
  1344. } else {
  1345. return nil, fmt.Errorf("found %q, expect string value in option.", lit3)
  1346. }
  1347. } else {
  1348. return nil, fmt.Errorf("found %q, expect equals(=) in options.", lit2)
  1349. }
  1350. } else if tok1 == ast.COMMA {
  1351. continue
  1352. } else if tok1 == ast.RPAREN {
  1353. if lStack.Pop(); lStack.Len() == 0 {
  1354. break
  1355. } else {
  1356. return nil, fmt.Errorf("Parenthesis is not matched in options definition.")
  1357. }
  1358. } else {
  1359. return nil, fmt.Errorf("found %q, unknown option keys(DATASOURCE|FORMAT|KEY|CONF_KEY|SHARED|STRICT_VALIDATION|TYPE|TIMESTAMP|TIMESTAMP_FORMAT|RETAIN_SIZE|SCHEMAID).", lit1)
  1360. }
  1361. }
  1362. } else {
  1363. return nil, fmt.Errorf("found %q, expect stream options.", lit)
  1364. }
  1365. if opts.KIND == ast.StreamKindLookup && opts.TYPE == "memory" && opts.KEY == "" {
  1366. return nil, fmt.Errorf("Option \"key\" is required for memory lookup table.")
  1367. }
  1368. return opts, nil
  1369. }
  1370. // Only support filter on window now
  1371. func (p *Parser) parseFilter() (ast.Expr, error) {
  1372. if tok, _ := p.scanIgnoreWhitespace(); tok != ast.FILTER {
  1373. p.unscan()
  1374. return nil, nil
  1375. }
  1376. if tok, lit := p.scanIgnoreWhitespace(); tok != ast.LPAREN {
  1377. return nil, fmt.Errorf("Found %q after FILTER, expect parentheses.", lit)
  1378. }
  1379. if tok, lit := p.scanIgnoreWhitespace(); tok != ast.WHERE {
  1380. return nil, fmt.Errorf("Found %q after FILTER(, expect WHERE.", lit)
  1381. }
  1382. expr, err := p.ParseExpr()
  1383. if err != nil {
  1384. return nil, err
  1385. }
  1386. if tok, lit := p.scanIgnoreWhitespace(); tok != ast.RPAREN {
  1387. return nil, fmt.Errorf("Found %q after FILTER, expect right parentheses.", lit)
  1388. }
  1389. return expr, nil
  1390. }
  1391. func (p *Parser) parseAsterisk() (ast.Expr, error) {
  1392. switch p.inFunc {
  1393. case "mqtt", "meta":
  1394. return &ast.MetaRef{StreamName: ast.DefaultStream, Name: "*"}, nil
  1395. case "":
  1396. return nil, fmt.Errorf("unsupported * expression, it must be used inside fields or function parameters.")
  1397. default:
  1398. return &ast.Wildcard{Token: ast.ASTERISK}, nil
  1399. }
  1400. }
  1401. func (p *Parser) inmeta() bool {
  1402. return p.inFunc == "meta" || p.inFunc == "mqtt"
  1403. }