TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-11-22 09:29:38 +00:00
Code Issues Projects Releases Packages Wiki Activity

compiler: support iota

Browse source
This commit is contained in:
Evgenii Stratonikov 2020-08-25 10:22:58 +03:00
parent d8badd9a8d
commit 05ef951055
2 changed files with 20 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
pkg/compiler/codegen.go
View file

@ -408,7 +408,12 @@ func (c *codegen) Visit(node ast.Node) ast.Visitor {
for _, spec := range n.Specs {
vs := spec.(*ast.ValueSpec)
for i := range vs.Names {
c.constMap[c.getIdentName("", vs.Names[i].Name)] = c.typeAndValueOf(vs.Values[i])
info := c.buildInfo.program.Package(c.currPkg.Path())
obj := info.Defs[vs.Names[i]]
c.constMap[c.getIdentName("", vs.Names[i].Name)] = types.TypeAndValue{
Type: obj.Type(),
Value: obj.(*types.Const).Val(),
}
}
}
return nil

14
pkg/compiler/constant_test.go
View file

@ -72,3 +72,17 @@ func TestGlobalsWithFunctionParams(t *testing.T) {
`
eval(t, src, []byte("FOO"))
}
func TestIota(t *testing.T) {
src := `package foo
const (
a = 2 << iota
b
c = 11
)
func Main() int {
return a + b + c
}`
eval(t, src, big.NewInt(17))
}