'Short Arrow' Function now support Block

haifenghuang · Dec 20, 2017 · ffb181b · ffb181b
1 parent adfa986
commit ffb181b
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/monkey/parser/parser.go b/src/monkey/parser/parser.go
@@ -1692,10 +1692,14 @@ func (p *Parser) parseThinArrowFunction(left ast.Expression) ast.Expression {
 	}
 
 	p.nextToken()
-	fn.Body = &ast.BlockStatement{
-		Statements: []ast.Statement{
-			p.parseExpressionStatement(),
-		},
+	if p.curTokenIs(token.LBRACE) { //if it's block, we use parseBlockStatement
+		fn.Body = p.parseBlockStatement().(*ast.BlockStatement)
+	} else { //not block, we use parseExpressionStatement
+		fn.Body = &ast.BlockStatement{
+			Statements: []ast.Statement{
+				p.parseExpressionStatement(),
+			},
+		}
 	}
 
 	return fn