TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-11-27 03:58:06 +00:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #1361 from nspcc-dev/feature/iota

Browse source 
compiler: support `iota`
This commit is contained in:
Roman Khimov 2020-08-25 15:50:20 +03:00 committed by GitHub
commit 962f45f35e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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 { for _, spec := range n.Specs {
vs := spec.(*ast.ValueSpec) vs := spec.(*ast.ValueSpec)
for i := range vs.Names { 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 return nil

14
pkg/compiler/constant_test.go
View file

@ -72,3 +72,17 @@ func TestGlobalsWithFunctionParams(t *testing.T) {
` `
eval(t, src, []byte("FOO")) 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))
}