diff --git a/pkg/compiler/func_scope.go b/pkg/compiler/func_scope.go index e0df8f6a0..41f3db76c 100644 --- a/pkg/compiler/func_scope.go +++ b/pkg/compiler/func_scope.go @@ -136,6 +136,18 @@ func (c *funcScope) analyzeVoidCalls(node ast.Node) bool { c.voidCalls[n] = true } return false + case *ast.CompositeLit: + for _, e := range n.Elts { + switch val := e.(type) { + case *ast.CallExpr: // slice + c.voidCalls[val] = false + case *ast.KeyValueExpr: // struct and map + ce, ok := val.Value.(*ast.CallExpr) + if ok { + c.voidCalls[ce] = false + } + } + } } return true } diff --git a/pkg/compiler/slice_test.go b/pkg/compiler/slice_test.go index 858666b70..95e667235 100644 --- a/pkg/compiler/slice_test.go +++ b/pkg/compiler/slice_test.go @@ -321,6 +321,15 @@ var sliceTestCases = []testCase{ }`, []byte("string"), }, + { + "literal slice with function call", + `package foo + func fn() byte { return 't' } + func Main() []byte { + return []byte{'s', fn(), 'r'} + }`, + []byte("str"), + }, } func TestSliceOperations(t *testing.T) { diff --git a/pkg/compiler/struct_test.go b/pkg/compiler/struct_test.go index 2048d612c..fe48e9429 100644 --- a/pkg/compiler/struct_test.go +++ b/pkg/compiler/struct_test.go @@ -29,6 +29,19 @@ var structTestCases = []testCase{ `, big.NewInt(2), }, + { + "struct field from func result", + ` + package foo + type S struct { x int } + func fn() int { return 2 } + func Main() int { + t := S{x: fn()} + return t.x + } + `, + big.NewInt(2), + }, { "struct field return", `