72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package provider
|
|
|
|
import "encoding/json"
|
|
|
|
// TimeoutHandler is a wrapper for a Getter that times out after a given duration
|
|
type TimeoutHandler struct {
|
|
ticker func() (string, error)
|
|
}
|
|
|
|
func NewTimeoutHandler(ticker func() (string, error)) *TimeoutHandler {
|
|
return &TimeoutHandler{ticker}
|
|
}
|
|
|
|
func (h *TimeoutHandler) BoolGetter(p BoolProvider) (func() (bool, error), error) {
|
|
g, err := p.BoolGetter()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return func() (val bool, err error) {
|
|
if val, err = g(); err == nil {
|
|
_, err = h.ticker()
|
|
}
|
|
return val, err
|
|
}, nil
|
|
}
|
|
|
|
func (h *TimeoutHandler) FloatGetter(p FloatProvider) (func() (float64, error), error) {
|
|
g, err := p.FloatGetter()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return func() (val float64, err error) {
|
|
if val, err = g(); err == nil {
|
|
_, err = h.ticker()
|
|
}
|
|
return val, err
|
|
}, nil
|
|
}
|
|
|
|
func (h *TimeoutHandler) StringGetter(p StringProvider) (func() (string, error), error) {
|
|
g, err := p.StringGetter()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return func() (val string, err error) {
|
|
if val, err = g(); err == nil {
|
|
_, err = h.ticker()
|
|
}
|
|
return val, err
|
|
}, nil
|
|
}
|
|
|
|
func (h *TimeoutHandler) JsonGetter(p StringProvider) (func(any) error, error) {
|
|
g, err := p.StringGetter()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return func(res any) error {
|
|
val, err := g()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if _, err := h.ticker(); err != nil {
|
|
return err
|
|
}
|
|
return json.Unmarshal([]byte(val), res)
|
|
}, nil
|
|
}
|