parser.go 36 KB

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