49 lines
859 B
Go
49 lines
859 B
Go
package provider
|
|
|
|
import (
|
|
"context"
|
|
"math"
|
|
"math/rand/v2"
|
|
|
|
"github.com/evcc-io/evcc/util"
|
|
)
|
|
|
|
type randomProvider struct {
|
|
ctx context.Context
|
|
set Config
|
|
}
|
|
|
|
func init() {
|
|
registry.AddCtx("random", NewRandomFromConfig)
|
|
}
|
|
|
|
// NewRandomFromConfig creates random provider
|
|
func NewRandomFromConfig(ctx context.Context, other map[string]interface{}) (Provider, error) {
|
|
var cc struct {
|
|
Set Config
|
|
}
|
|
|
|
if err := util.DecodeOther(other, &cc); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
o := &randomProvider{
|
|
ctx: ctx,
|
|
set: cc.Set,
|
|
}
|
|
|
|
return o, nil
|
|
}
|
|
|
|
var _ SetIntProvider = (*randomProvider)(nil)
|
|
|
|
func (o *randomProvider) IntSetter(param string) (func(int64) error, error) {
|
|
set, err := NewIntSetterFromConfig(o.ctx, param, o.set)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return func(int64) error {
|
|
return set(rand.Int64N(math.MaxInt64-1) + 1)
|
|
}, nil
|
|
}
|