yq/pkg/yqlib/operator_shuffle.go

35 lines
899 B
Go
Raw Normal View History

2023-02-10 18:08:20 +00:00
package yqlib
import (
"container/list"
"fmt"
"math/rand"
)
func shuffleOperator(d *dataTreeNavigator, context Context, expressionNode *ExpressionNode) (Context, error) {
// ignore CWE-338 gosec issue of not using crypto/rand
// this is just to shuffle an array rather generating a
// secret or something that needs proper rand.
myRand := rand.New(rand.NewSource(Now().UnixNano())) // #nosec
results := list.New()
for el := context.MatchingNodes.Front(); el != nil; el = el.Next() {
candidate := el.Value.(*CandidateNode)
2023-06-07 17:45:42 +00:00
if candidate.Kind != SequenceNode {
return context, fmt.Errorf("node at path [%v] is not an array (it's a %v)", candidate.GetNicePath(), candidate.Tag)
2023-02-10 18:08:20 +00:00
}
2023-06-07 17:45:42 +00:00
result := candidate.Copy()
2023-02-10 18:08:20 +00:00
a := result.Content
myRand.Shuffle(len(a), func(i, j int) { a[i], a[j] = a[j], a[i] })
2023-04-08 11:27:47 +00:00
results.PushBack(result)
2023-02-10 18:08:20 +00:00
}
return context.ChildContext(results), nil
}