Skip to content

Commit 54714ab

Browse files
committed
Fix more literal strings
1 parent 0cc62d6 commit 54714ab

File tree

6 files changed

+56
-52
lines changed

6 files changed

+56
-52
lines changed

parser/lexer.go

+19-15
Original file line numberDiff line numberDiff line change
@@ -9,21 +9,21 @@ import (
99
)
1010

1111
const (
12-
TokenKindEOF TokenKind = "<eof>"
13-
TokenKindIdent TokenKind = "<ident>"
14-
TokenKindKeyword TokenKind = "<keyword>"
15-
TokenKindInt TokenKind = "<int>"
16-
TokenKindFloat TokenKind = "<float>"
17-
TokenKindString TokenKind = "<string>"
18-
TokenKindDot = "."
19-
TokenKindSingleEQ TokenKind = "="
20-
TokenKindDoubleEQ TokenKind = "=="
21-
TokenKindNE TokenKind = "!="
22-
TokenKindLT TokenKind = "<"
23-
TokenKindLE TokenKind = "<="
24-
TokenKindGT TokenKind = ">"
25-
TokenKindGE TokenKind = ">="
26-
TokenKindQuery = "?"
12+
TokenKindEOF TokenKind = "<eof>"
13+
TokenKindIdent TokenKind = "<ident>"
14+
TokenKindKeyword TokenKind = "<keyword>"
15+
TokenKindInt TokenKind = "<int>"
16+
TokenKindFloat TokenKind = "<float>"
17+
TokenKindString TokenKind = "<string>"
18+
TokenKindDot = "."
19+
TokenKindSingleEQ TokenKind = "="
20+
TokenKindDoubleEQ TokenKind = "=="
21+
TokenKindNE TokenKind = "!="
22+
TokenKindLT TokenKind = "<"
23+
TokenKindLE TokenKind = "<="
24+
TokenKindGT TokenKind = ">"
25+
TokenKindGE TokenKind = ">="
26+
TokenKindQuestionMark TokenKind = "?"
2727

2828
TokenKindPlus TokenKind = "+"
2929
TokenKindMinus TokenKind = "-"
@@ -40,6 +40,10 @@ const (
4040
TokenKindRBrace TokenKind = "}"
4141
TokenKindLBracket TokenKind = "["
4242
TokenKindRBracket TokenKind = "]"
43+
44+
TokenKindComma TokenKind = ","
45+
TokenKindColon TokenKind = ":"
46+
TokenKindAtSign TokenKind = "@"
4347
)
4448

4549
const (

parser/parse_system.go

+7-7
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ func (p *Parser) parseRoleName(_ Pos) (*RoleName, error) {
349349
return nil, err
350350
}
351351
var scope *StringLiteral
352-
if p.tryConsumeTokenKind("@") != nil {
352+
if p.tryConsumeTokenKind(TokenKindAtSign) != nil {
353353
scope, err = p.parseString(p.Pos())
354354
if err != nil {
355355
return nil, err
@@ -447,7 +447,7 @@ func (p *Parser) parseRoleSettings(_ Pos) ([]*RoleSetting, error) {
447447
return nil, err
448448
}
449449
settings = append(settings, setting)
450-
if p.tryConsumeTokenKind(",") == nil {
450+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
451451
break
452452
}
453453
}
@@ -485,7 +485,7 @@ func (p *Parser) parseCreateRole(pos Pos) (*CreateRole, error) {
485485
return nil, err
486486
}
487487
roleNames = append(roleNames, roleName)
488-
for p.tryConsumeTokenKind(",") != nil {
488+
for p.tryConsumeTokenKind(TokenKindComma) != nil {
489489
roleName, err := p.parseRoleName(p.Pos())
490490
if err != nil {
491491
return nil, err
@@ -543,7 +543,7 @@ func (p *Parser) parserDropUserOrRole(pos Pos) (*DropUserOrRole, error) {
543543
return nil, err
544544
}
545545
names = append(names, name)
546-
for p.tryConsumeTokenKind(",") != nil {
546+
for p.tryConsumeTokenKind(TokenKindComma) != nil {
547547
name, err := p.parseRoleName(p.Pos())
548548
if err != nil {
549549
return nil, err
@@ -879,7 +879,7 @@ func (p *Parser) parsePrivilegeRoles(_ Pos) ([]*Ident, error) {
879879
return nil, err
880880
}
881881
roles = append(roles, role)
882-
for p.tryConsumeTokenKind(",") != nil {
882+
for p.tryConsumeTokenKind(TokenKindComma) != nil {
883883
role, err := p.parseIdent()
884884
if err != nil {
885885
return nil, err
@@ -950,7 +950,7 @@ func (p *Parser) parseGrantPrivilegeStmt(pos Pos) (*GrantPrivilegeStmt, error) {
950950
return nil, err
951951
}
952952
privileges = append(privileges, privilege)
953-
for p.tryConsumeTokenKind(",") != nil {
953+
for p.tryConsumeTokenKind(TokenKindComma) != nil {
954954
privilege, err := p.parsePrivilegeClause(p.Pos())
955955
if err != nil {
956956
return nil, err
@@ -1012,7 +1012,7 @@ func (p *Parser) parseAlterRole(pos Pos) (*AlterRole, error) {
10121012
return nil, err
10131013
}
10141014
roleRenamePairs = append(roleRenamePairs, roleRenamePair)
1015-
for p.tryConsumeTokenKind(",") != nil {
1015+
for p.tryConsumeTokenKind(TokenKindComma) != nil {
10161016
roleRenamePair, err := p.parseRoleRenamePair(p.Pos())
10171017
if err != nil {
10181018
return nil, err

parser/parser_alter.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ func (p *Parser) parseAlterTable(pos Pos) (*AlterTable, error) {
5858
return nil, err
5959
}
6060
alterTable.AlterExprs = append(alterTable.AlterExprs, alter)
61-
if p.tryConsumeTokenKind(",") == nil {
61+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
6262
break
6363
}
6464
}

parser/parser_column.go

+16-16
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ func (p *Parser) getNextPrecedence() int {
6565
return precedenceIn
6666
case p.matchKeyword(KeywordGlobal):
6767
return PrecedenceGlobal
68-
case p.matchTokenKind(TokenKindQuery):
68+
case p.matchTokenKind(TokenKindQuestionMark):
6969
return PrecedenceQuery
7070
default:
7171
return PrecedenceUnknown
@@ -191,7 +191,7 @@ func (p *Parser) parseInfix(expr Expr, precedence int) (Expr, error) {
191191
Object: expr,
192192
Params: params,
193193
}, nil
194-
case p.matchTokenKind(TokenKindQuery):
194+
case p.matchTokenKind(TokenKindQuestionMark):
195195
return p.parseTernaryExpr(expr)
196196
case p.matchKeyword(KeywordIs):
197197
_ = p.lexer.consumeToken()
@@ -238,14 +238,14 @@ func (p *Parser) parseSubExpr(pos Pos, precedence int) (Expr, error) {
238238
}
239239

240240
func (p *Parser) parseTernaryExpr(condition Expr) (*TernaryOperation, error) {
241-
if _, err := p.consumeTokenKind(TokenKindQuery); err != nil {
241+
if _, err := p.consumeTokenKind(TokenKindQuestionMark); err != nil {
242242
return nil, err
243243
}
244244
trueExpr, err := p.parseExpr(p.Pos())
245245
if err != nil {
246246
return nil, err
247247
}
248-
if _, err := p.consumeTokenKind(":"); err != nil {
248+
if _, err := p.consumeTokenKind(TokenKindColon); err != nil {
249249
return nil, err
250250
}
251251
falseExpr, err := p.parseExpr(p.Pos())
@@ -377,13 +377,13 @@ func (p *Parser) parseColumnExpr(pos Pos) (Expr, error) { //nolint:funlen
377377
return p.parseMapLiteral(p.Pos())
378378
case p.matchTokenKind(TokenKindDot):
379379
return p.parseNumber(p.Pos())
380-
case p.matchTokenKind(TokenKindQuery):
380+
case p.matchTokenKind(TokenKindQuestionMark):
381381
// Placeholder `?`
382382
_ = p.lexer.consumeToken()
383383
return &PlaceHolder{
384384
PlaceholderPos: pos,
385385
PlaceHolderEnd: pos,
386-
Type: TokenKindQuery,
386+
Type: string(TokenKindQuestionMark),
387387
}, nil
388388
default:
389389
return nil, fmt.Errorf("unexpected token kind: %s", p.lastTokenKind())
@@ -470,7 +470,7 @@ func (p *Parser) parseColumnExprListWithTerm(term TokenKind, pos Pos) (*ColumnEx
470470
break
471471
}
472472
columnList = append(columnList, columnExpr)
473-
if p.tryConsumeTokenKind(",") == nil {
473+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
474474
break
475475
}
476476
}
@@ -492,7 +492,7 @@ func (p *Parser) parseSelectItems() ([]*SelectItem, error) {
492492
break
493493
}
494494
selectItems = append(selectItems, selectItem)
495-
if p.tryConsumeTokenKind(",") == nil {
495+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
496496
break
497497
}
498498
}
@@ -558,7 +558,7 @@ func (p *Parser) parseColumnArgList(pos Pos) (*ColumnArgList, error) {
558558
return nil, err
559559
}
560560
items = append(items, item)
561-
if p.tryConsumeTokenKind(",") == nil {
561+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
562562
break
563563
}
564564
}
@@ -616,7 +616,7 @@ func (p *Parser) parseMapLiteral(pos Pos) (*MapLiteral, error) {
616616
if err != nil {
617617
return nil, err
618618
}
619-
if _, err := p.consumeTokenKind(":"); err != nil {
619+
if _, err := p.consumeTokenKind(TokenKindColon); err != nil {
620620
return nil, err
621621
}
622622
value, err := p.parseExpr(p.Pos())
@@ -627,7 +627,7 @@ func (p *Parser) parseMapLiteral(pos Pos) (*MapLiteral, error) {
627627
Key: *key,
628628
Value: value,
629629
})
630-
if p.tryConsumeTokenKind(",") == nil {
630+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
631631
break
632632
}
633633
}
@@ -651,7 +651,7 @@ func (p *Parser) parseQueryParam(pos Pos) (*QueryParam, error) {
651651
if err != nil {
652652
return nil, err
653653
}
654-
if _, err := p.consumeTokenKind(":"); err != nil {
654+
if _, err := p.consumeTokenKind(TokenKindColon); err != nil {
655655
return nil, err
656656
}
657657
columnType, err := p.parseColumnType(p.Pos())
@@ -849,7 +849,7 @@ func (p *Parser) parseComplexType(name *Ident, pos Pos) (*ComplexType, error) {
849849
return nil, err
850850
}
851851
subTypes = append(subTypes, subExpr)
852-
if p.tryConsumeTokenKind(",") == nil {
852+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
853853
break
854854
}
855855
}
@@ -880,7 +880,7 @@ func (p *Parser) parseEnumType(name *Ident, pos Pos) (*EnumType, error) {
880880
break
881881
}
882882
enumType.Values = append(enumType.Values, *enumValue)
883-
if p.tryConsumeTokenKind(",") == nil {
883+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
884884
break
885885
}
886886
}
@@ -900,7 +900,7 @@ func (p *Parser) parseColumnTypeWithParams(name *Ident, pos Pos) (*TypeWithParam
900900
return nil, err
901901
}
902902
params = append(params, param)
903-
for !p.lexer.isEOF() && p.tryConsumeTokenKind(",") != nil {
903+
for !p.lexer.isEOF() && p.tryConsumeTokenKind(TokenKindComma) != nil {
904904
size, err := p.parseLiteral(p.Pos())
905905
if err != nil {
906906
return nil, err
@@ -964,7 +964,7 @@ func (p *Parser) tryParseCompressionCodecs(pos Pos) (*CompressionCodec, error) {
964964
return nil, err
965965
}
966966
// consume comma
967-
if _, err := p.consumeTokenKind(","); err != nil {
967+
if _, err := p.consumeTokenKind(TokenKindComma); err != nil {
968968
return nil, err
969969
}
970970
name, err = p.parseIdent()

parser/parser_query.go

+4-4
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ func (p *Parser) parseWithClause(pos Pos) (*WithClause, error) {
2222
return nil, err
2323
}
2424
ctes := []*CTEStmt{cteExpr}
25-
for p.tryConsumeTokenKind(",") != nil {
25+
for p.tryConsumeTokenKind(TokenKindComma) != nil {
2626
cteExpr, err := p.parseCTEStmt(p.Pos())
2727
if err != nil {
2828
return nil, err
@@ -224,7 +224,7 @@ func (p *Parser) parseJoinRightExpr(pos Pos) (expr Expr, err error) {
224224
switch {
225225
case p.tryConsumeKeyword(KeywordGlobal) != nil:
226226
case p.tryConsumeKeyword(KeywordLocal) != nil:
227-
case p.tryConsumeTokenKind(",") != nil:
227+
case p.tryConsumeTokenKind(TokenKindComma) != nil:
228228
return p.parseJoinExpr(p.Pos())
229229
default:
230230
modifiers = p.parseJoinOp(p.Pos())
@@ -473,7 +473,7 @@ func (p *Parser) parseLimitClause(pos Pos) (*LimitClause, error) {
473473
var offset Expr
474474
if p.tryConsumeKeyword(KeywordOffset) != nil {
475475
offset, err = p.parseExpr(p.Pos())
476-
} else if p.tryConsumeTokenKind(",") != nil {
476+
} else if p.tryConsumeTokenKind(TokenKindComma) != nil {
477477
offset = limit
478478
limit, err = p.parseExpr(p.Pos())
479479
}
@@ -1022,7 +1022,7 @@ func (p *Parser) tryParseColumnAliases() ([]*Ident, error) {
10221022
if p.matchTokenKind(TokenKindRParen) {
10231023
break
10241024
}
1025-
if _, err := p.consumeTokenKind(","); err != nil {
1025+
if _, err := p.consumeTokenKind(TokenKindComma); err != nil {
10261026
return nil, err
10271027
}
10281028
}

parser/parser_table.go

+9-9
Original file line numberDiff line numberDiff line change
@@ -394,7 +394,7 @@ func (p *Parser) parseTableColumns() ([]Expr, error) {
394394
}
395395
columns = append(columns, column)
396396
}
397-
if p.tryConsumeTokenKind(",") == nil {
397+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
398398
break
399399
}
400400
}
@@ -529,7 +529,7 @@ func (p *Parser) parseTableArgList(pos Pos) (*TableArgListExpr, error) {
529529
return nil, err
530530
}
531531
args = append(args, arg)
532-
if p.tryConsumeTokenKind(",") == nil {
532+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
533533
break
534534
}
535535
}
@@ -637,7 +637,7 @@ func (p *Parser) parseOrderByClause(pos Pos) (*OrderByClause, error) {
637637
}
638638
items = append(items, expr)
639639

640-
if p.lexer.isEOF() || p.tryConsumeTokenKind(",") == nil {
640+
if p.lexer.isEOF() || p.tryConsumeTokenKind(TokenKindComma) == nil {
641641
break
642642
}
643643
}
@@ -704,7 +704,7 @@ func (p *Parser) parseTTLClause(pos Pos) ([]*TTLExpr, error) {
704704
return nil, err
705705
}
706706
items = append(items, expr)
707-
for !p.lexer.isEOF() && p.tryConsumeTokenKind(",") != nil {
707+
for !p.lexer.isEOF() && p.tryConsumeTokenKind(TokenKindComma) != nil {
708708
expr, err = p.parseTTLExpr(pos)
709709
if err != nil {
710710
return nil, err
@@ -774,7 +774,7 @@ func (p *Parser) parseSettingsClause(pos Pos) (*SettingsClause, error) {
774774
return nil, err
775775
}
776776
items = append(items, expr)
777-
for p.tryConsumeTokenKind(",") != nil {
777+
for p.tryConsumeTokenKind(TokenKindComma) != nil {
778778
expr, err = p.parseSettingsExprList(p.Pos())
779779
if err != nil {
780780
return nil, err
@@ -1106,7 +1106,7 @@ func (p *Parser) parseColumnNamesExpr(pos Pos) (*ColumnNamesExpr, error) {
11061106
}
11071107

11081108
columnNames = append(columnNames, *name)
1109-
if p.tryConsumeTokenKind(",") == nil {
1109+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
11101110
break
11111111
}
11121112
}
@@ -1140,7 +1140,7 @@ func (p *Parser) parseAssignmentValues(pos Pos) (*AssignmentValues, error) {
11401140
return nil, err
11411141
}
11421142
values = append(values, value)
1143-
if p.tryConsumeTokenKind(",") == nil {
1143+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
11441144
break
11451145
}
11461146
}
@@ -1215,7 +1215,7 @@ func (p *Parser) parseInsertStmt(pos Pos) (*InsertStmt, error) {
12151215
return nil, err
12161216
}
12171217
values = append(values, value)
1218-
if p.tryConsumeTokenKind(",") == nil {
1218+
if p.tryConsumeTokenKind(TokenKindComma) == nil {
12191219
break
12201220
}
12211221
}
@@ -1246,7 +1246,7 @@ func (p *Parser) parseRenameStmt(pos Pos) (*RenameStmt, error) {
12461246
return nil, err
12471247
}
12481248
tablePairList := []*TargetPair{targetPair}
1249-
for p.tryConsumeTokenKind(",") != nil {
1249+
for p.tryConsumeTokenKind(TokenKindComma) != nil {
12501250
tablePair, err := p.parseTargetPair(p.Pos())
12511251
if err != nil {
12521252
return nil, err

0 commit comments

Comments
 (0)