Method parseAlterPolicyStatement

pkg/sql/parser/alter.go:215–270

parseAlterPolicyStatement parses ALTER POLICY statements

(stmt *ast.AlterStatement)

// parseAlterPolicyStatement parses ALTER POLICY statements
func (p *Parser) parseAlterPolicyStatement(stmt *ast.AlterStatement) (*ast.AlterStatement, error) {
	stmt.Name = p.parseIdentAsString()
	if !p.matchType(models.TokenTypeOn) {
		return nil, p.expectedError("ON")
	}
	p.parseIdentAsString() // table name

	op := &ast.AlterPolicyOperation{}

	if p.matchType(models.TokenTypeRename) {
		if !p.matchType(models.TokenTypeTo) {
			return nil, p.expectedError("TO")
		}
		op.Type = ast.RenamePolicy
		op.NewName = p.parseIdentAsString()
	} else {
		op.Type = ast.ModifyPolicy
		if p.matchType(models.TokenTypeTo) {
			for {
				op.To = append(op.To, p.parseIdentAsString())
				if !p.matchType(models.TokenTypeComma) {
					break
				}
			}
		}
		if p.matchType(models.TokenTypeUsing) {
			if !p.matchType(models.TokenTypeLParen) {
				return nil, p.expectedError("(")
			}
			expr, err := p.parseExpression()
			if err != nil {
				return nil, err
			}
			op.Using = expr
			if !p.matchType(models.TokenTypeRParen) {
				return nil, p.expectedError(")")
			}
		}
		if p.matchType(models.TokenTypeWith) && p.matchType(models.TokenTypeCheck) {
			if !p.matchType(models.TokenTypeLParen) {
				return nil, p.expectedError("(")
			}
			expr, err := p.parseExpression()
			if err != nil {
				return nil, err
			}
			op.WithCheck = expr
			if !p.matchType(models.TokenTypeRParen) {
				return nil, p.expectedError(")")
			}
		}
	}

	stmt.Operation = op
	return stmt, nil
}

// parseAlterConnectorStatement parses ALTER CONNECTOR statements