Allow to use the column type for Tuple

Author: git-hulk

parser/parser_column.go

@@ -81,7 +81,7 @@ func (p *Parser) parseInfix(expr Expr, precedence int) (Expr, error) {
 		p.matchTokenKind(opTypeDiv), p.matchTokenKind(opTypeMod),
 		p.matchKeyword(KeywordIn), p.matchKeyword(KeywordLike),
 		p.matchKeyword(KeywordIlike), p.matchKeyword(KeywordAnd), p.matchKeyword(KeywordOr),
-		p.matchTokenKind(opTypeCast), p.matchTokenKind(opTypeArrow), p.matchTokenKind(opTypeDoubleEQ):
+		p.matchTokenKind(opTypeArrow), p.matchTokenKind(opTypeDoubleEQ):
 		op := p.last().ToString()
 		_ = p.lexer.consumeToken()
 		rightExpr, err := p.parseSubExpr(p.Pos(), precedence)
@@ -93,6 +93,36 @@ func (p *Parser) parseInfix(expr Expr, precedence int) (Expr, error) {
 			Operation: TokenKind(op),
 			RightExpr: rightExpr,
 		}, nil
+	case p.matchTokenKind(opTypeCast):
+		_ = p.lexer.consumeToken()
+
+		if p.matchTokenKind(TokenIdent) && p.last().String == "Tuple" {
+			name, err := p.parseIdent()
+			if err != nil {
+				return nil, err
+			}
+			p.consumeTokenKind("(")
+			// it's a tuple type definition after "::" operator
+			rightExpr, err := p.parseNestedType(name, p.Pos())
+			if err != nil {
+				return nil, err
+			}
+			return &BinaryOperation{
+				LeftExpr:  expr,
+				Operation: opTypeCast,
+				RightExpr: rightExpr,
+			}, nil
+		}
+
+		rightExpr, err := p.parseSubExpr(p.Pos(), precedence)
+		if err != nil {
+			return nil, err
+		}
+		return &BinaryOperation{
+			LeftExpr:  expr,
+			Operation: opTypeCast,
+			RightExpr: rightExpr,
+		}, nil
 	case p.matchKeyword(KeywordBetween):
 		return p.parseBetweenClause(expr)
 	case p.matchKeyword(KeywordGlobal):

parser/testdata/query/access_tuple_with_dot.sql

@@ -1 +1,6 @@
-SELECT tuple('a','b','c').3, .1234
+SELECT tuple('a','b','c').3, .1234;
+
+SELECT toTypeName( tuple('a' as first,'b' as second ,'c' as third)::Tuple(first String,second String,third String)),
+       (tuple('a' as first,'b' as second ,'c' as third)::Tuple(first String,second String,third String)).second,
+       tuple('a','b','c').3,
+       tupleElement(tuple('a','b','c'),1)

parser/testdata/query/format/access_tuple_with_dot.sql

@@ -1,5 +1,11 @@
 -- Origin SQL:
-SELECT tuple('a','b','c').3, .1234
+SELECT tuple('a','b','c').3, .1234;
+
+SELECT toTypeName( tuple('a' as first,'b' as second ,'c' as third)::Tuple(first String,second String,third String)),
+       (tuple('a' as first,'b' as second ,'c' as third)::Tuple(first String,second String,third String)).second,
+       tuple('a','b','c').3,
+       tupleElement(tuple('a','b','c'),1)
 
 -- Format SQL:
 SELECT tuple('a', 'b', 'c').3, .1234;
+SELECT toTypeName(tuple('a' AS first, 'b' AS second, 'c' AS third)::Tuple(first String, second String, third String)), (tuple('a' AS first, 'b' AS second, 'c' AS third)::Tuple(first String, second String, third String)).second, tuple('a', 'b', 'c').3, tupleElement(tuple('a', 'b', 'c'), 1);