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

65 lines
1.3 KiB
Go
Raw Normal View History

package collections
import (
2018-10-28 07:45:26 +02:00
"context"
"github.com/MontFerret/ferret/pkg/runtime/core"
"github.com/MontFerret/ferret/pkg/runtime/values"
)
type KeyedIterator struct {
valVar string
keyVar string
values KeyedCollection
keys []values.String
pos int
}
func NewKeyedIterator(
valVar,
keyVar string,
2018-10-28 07:45:26 +02:00
values KeyedCollection,
) (Iterator, error) {
if valVar == "" {
return nil, core.Error(core.ErrMissedArgument, "value variable")
}
if values == nil {
return nil, core.Error(core.ErrMissedArgument, "result")
}
return &KeyedIterator{valVar, keyVar, values, nil, 0}, nil
}
2018-10-28 07:45:26 +02:00
func NewDefaultKeyedIterator(input KeyedCollection) (Iterator, error) {
return NewKeyedIterator(DefaultValueVar, DefaultKeyVar, input)
}
2018-10-28 07:45:26 +02:00
func (iterator *KeyedIterator) Next(_ context.Context, scope *core.Scope) (*core.Scope, error) {
if iterator.keys == nil {
iterator.keys = iterator.values.Keys()
}
if len(iterator.keys) > iterator.pos {
key := iterator.keys[iterator.pos]
val, _ := iterator.values.Get(key)
2018-10-28 07:45:26 +02:00
iterator.pos++
2018-10-28 07:45:26 +02:00
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
}
2018-10-28 07:45:26 +02:00
return nil, nil
}