You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

75 lines
1.3 KiB

4 years ago
package watcher
import (
"context"
"fmt"
9 months ago
"log/slog"
4 years ago
"time"
"gitoa.ru/go-4devs/config"
)
var (
_ config.Provider = (*Provider)(nil)
_ config.WatchProvider = (*Provider)(nil)
)
9 months ago
func New(duration time.Duration, provider config.Provider, opts ...Option) *Provider {
prov := &Provider{
9 months ago
Provider: provider,
duration: duration,
logger: slog.ErrorContext,
4 years ago
}
for _, opt := range opts {
opt(prov)
4 years ago
}
return prov
4 years ago
}
9 months ago
func WithLogger(l func(context.Context, string, ...any)) Option {
4 years ago
return func(p *Provider) {
p.logger = l
}
}
type Option func(*Provider)
type Provider struct {
9 months ago
config.Provider
duration time.Duration
logger func(context.Context, string, ...any)
4 years ago
}
func (p *Provider) Watch(ctx context.Context, callback config.WatchCallback, key ...string) error {
9 months ago
old, err := p.Provider.Value(ctx, key...)
4 years ago
if err != nil {
return fmt.Errorf("failed watch variable: %w", err)
4 years ago
}
9 months ago
go func(oldVar config.Value) {
ticker := time.NewTicker(p.duration)
defer func() {
ticker.Stop()
}()
4 years ago
for {
select {
9 months ago
case <-ticker.C:
newVar, err := p.Provider.Value(ctx, key...)
4 years ago
if err != nil {
9 months ago
p.logger(ctx, "get value%v:%v", key, err.Error())
4 years ago
} else if !newVar.IsEquals(oldVar) {
callback(ctx, oldVar, newVar)
oldVar = newVar
}
case <-ctx.Done():
return
}
}
9 months ago
}(old)
4 years ago
return nil
}