Method parsePartitionByClause

pkg/sql/parser/ddl.go:432–477

parsePartitionByClause parses PARTITION BY RANGE/LIST/HASH (columns)

()

// parsePartitionByClause parses PARTITION BY RANGE/LIST/HASH (columns)
func (p *Parser) parsePartitionByClause() (*ast.PartitionBy, error) {
	partitionBy := &ast.PartitionBy{}

	// Parse partition type
	if p.isType(models.TokenTypeRange) {
		partitionBy.Type = "RANGE"
		p.advance()
	} else if p.isTokenMatch("LIST") {
		partitionBy.Type = "LIST"
		p.advance()
	} else if p.isTokenMatch("HASH") {
		partitionBy.Type = "HASH"
		p.advance()
	} else {
		return nil, p.expectedError("RANGE, LIST, or HASH")
	}

	// Expect opening parenthesis
	if !p.isType(models.TokenTypeLParen) {
		return nil, p.expectedError("(")
	}
	p.advance() // Consume (

	// Parse column list
	for {
		if !p.isIdentifier() {
			return nil, p.expectedError("column name")
		}
		partitionBy.Columns = append(partitionBy.Columns, p.currentToken.Token.Value)
		p.advance()

		if p.isType(models.TokenTypeComma) {
			p.advance() // Consume comma
			continue
		}
		break
	}

	// Expect closing parenthesis
	if !p.isType(models.TokenTypeRParen) {
		return nil, p.expectedError(")")
	}
	p.advance() // Consume )

	return partitionBy, nil
}

// parsePartitionDefinition parses a single partition definition
func (p *Parser) parsePartitionDefinition() (*ast.PartitionDefinition, error) {