1
0
mirror of https://github.com/MontFerret/ferret.git synced 2024-12-23 01:24:30 +02:00
ferret/pkg/runtime/collections/collection.go

68 lines
1.3 KiB
Go
Raw Normal View History

2018-09-18 22:42:38 +02:00
package collections
import (
"context"
"github.com/MontFerret/ferret/pkg/runtime/core"
"github.com/MontFerret/ferret/pkg/runtime/values"
)
2018-09-18 22:42:38 +02:00
type (
// Measurable represents an interface of a value that can has length.
Measurable interface {
Length() values.Int
}
IndexedCollection interface {
core.Value
Measurable
Get(idx values.Int) core.Value
Set(idx values.Int, value core.Value) error
}
KeyedCollection interface {
core.Value
Measurable
Keys() []values.String
Get(key values.String) (core.Value, values.Boolean)
Set(key values.String, value core.Value)
}
coreIterator struct {
valVar string
keyVar string
values core.Iterator
}
)
func NewCoreIterator(valVar, keyVar string, values core.Iterator) (Iterator, error) {
return &coreIterator{valVar, keyVar, values}, nil
}
func (iterator *coreIterator) Next(ctx context.Context, scope *core.Scope) (*core.Scope, error) {
val, key, err := iterator.values.Next(ctx)
if err != nil {
return nil, err
}
// end of iteration
if val == values.None {
return nil, nil
}
nextScope := scope.Fork()
if err := nextScope.SetVariable(iterator.valVar, val); err != nil {
return nil, err
}
if iterator.keyVar != "" {
if err := nextScope.SetVariable(iterator.keyVar, key); err != nil {
return nil, err
}
}
return nextScope, nil
}