-
Notifications
You must be signed in to change notification settings - Fork 0
/
decorate.go
39 lines (34 loc) · 950 Bytes
/
decorate.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package timelimiter
import "github.com/CharLemAznable/gogo/lang"
func DecorateRunnable(limiter TimeLimiter, fn func() error) func() error {
return func() error {
_, err := limiter.Execute(func() (any, error) {
return nil, fn()
})
return err
}
}
func DecorateSupplier[T any](limiter TimeLimiter, fn func() (T, error)) func() (T, error) {
return func() (T, error) {
ret, err := limiter.Execute(func() (any, error) {
return fn()
})
return lang.CastQuietly[T](ret), err
}
}
func DecorateConsumer[T any](limiter TimeLimiter, fn func(T) error) func(T) error {
return func(t T) error {
_, err := limiter.Execute(func() (any, error) {
return nil, fn(t)
})
return err
}
}
func DecorateFunction[T any, R any](limiter TimeLimiter, fn func(T) (R, error)) func(T) (R, error) {
return func(t T) (R, error) {
ret, err := limiter.Execute(func() (any, error) {
return fn(t)
})
return lang.CastQuietly[R](ret), err
}
}