규격

for ( var i = 0; i < 10; i = i + 1) { // 初期化; 継続条件; 増分処理
    puts(i); // 文
}

자구 분석

var keywords = map[string]TokenType{
    ...
    "for":    FOR,
}

문법 분석

type ForStatement struct {
    Token            token.Token
    InitialStatement Statement       // 初期化
    Condition        Expression      // 継続条件
    PostStatement    Statement       // 増分処理
    Block            *BlockStatement // 本体のブロック
}

// "for"のTokenを読み込むと呼ばれる関数
func (p *Parser) parseForStatement() *ast.ForStatement {
    stmt := &ast.ForStatement{Token: p.curToken}

    // forの次は"("
    if !p.expectPeek(token.LPAREN) {
        return nil
    }
    p.nextToken()

    // 初期化の文の解析
    stmt.InitialStatement = p.parseStatement()
    p.nextToken()

    // 継続条件の式の解析
    stmt.Condition = p.parseExpression(LOWEST)
    if !p.expectPeek(token.SEMICOLON) {
        return nil
    }
    p.nextToken()

    // 増分処理の文の解析
    stmt.PostStatement = p.parseStatement()

    // 増分処理の次は")"のはず。ただし、増分処理が空の場合はすでに")"が読まれている。
    if p.peekTokenIs(token.RPAREN) {
        p.nextToken()
    }
    p.nextToken()

    // 本体の文の解析
    stmt.Block = p.parseBlockStatement()

    return stmt
}

평가

// for文のNodeを評価するときに呼ばれる関数
func evalForStatement(stmt *ast.ForStatement, env *object.Environment) object.Object {
    // 初期化の文を評価
    initStmt := Eval(stmt.InitialStatement, env)
    if isError(initStmt) {
        return initStmt
    }

    var result object.Object

    for {
        // まず継続条件を評価
        condition := Eval(stmt.Condition, env)
        if isError(condition) {
            return condition
        }
        if !isTruthy(condition) {
            // 評価結果が偽だったらループを抜ける
            break
        }

        // 本体の文を評価
        result = Eval(stmt.Block, env)
        // エラーもしくはreturnで返り値が渡ってきたらそのまま返す
        if isError(result) || isReturn(result) { 
            return result
        }

        // 増分処理の文を評価
        postStmt := Eval(stmt.PostStatement, env)
        if isError(postStmt) {
            return postStmt
        }
    }

    return result
}

REPL로 해볼게요.

>> for (var i = 0; i < 10; i = i + 1) { puts(i); }       
0
1
2
3
4
5
6
7
8
9

마지막