TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

compiler: support named returns

Browse source
This commit is contained in:
Evgenii Stratonikov 2020-05-06 17:24:32 +03:00
parent 156a2eddc5
commit b0a89e8a1a
3 changed files with 52 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

19
pkg/compiler/codegen.go
View file

@ -421,9 +421,22 @@ func (c *codegen) Visit(node ast.Node) ast.Visitor {
}
c.dropItems(cnt)
// first result should be on top of the stack
for i := len(n.Results) - 1; i >= 0; i-- {
ast.Walk(c, n.Results[i])
if len(n.Results) == 0 {
results := c.scope.decl.Type.Results
if results.NumFields() != 0 {
// function with named returns
for i := len(results.List) - 1; i >= 0; i-- {
names := results.List[i].Names
for j := len(names) - 1; j >= 0; j-- {
c.emitLoadLocal(names[j].Name)
}
}
}
} else {
// first result should be on top of the stack
for i := len(n.Results) - 1; i >= 0; i-- {
ast.Walk(c, n.Results[i])
}
}
c.saveSequencePoint(n)