diff --git a/pkg/zones/log.go b/pkg/zones/log.go new file mode 100644 index 0000000..ceaf5ee --- /dev/null +++ b/pkg/zones/log.go @@ -0,0 +1,49 @@ +package zones + +import "darvaza.org/slog" + +type logger interface { + withDebug() (slog.Logger, bool) + withInfo() (slog.Logger, bool) + + debug() slog.Logger + info() slog.Logger + warn(error) slog.Logger + error(error) slog.Logger +} + +var ( + _ logger = (*Zones)(nil) +) + +func (z *Zones) withDebug() (slog.Logger, bool) { + return z.debug().WithEnabled() +} + +func (z *Zones) withInfo() (slog.Logger, bool) { + return z.debug().WithEnabled() +} + +func (z *Zones) debug() slog.Logger { + return z.log.Debug() +} + +func (z *Zones) info() slog.Logger { + return z.log.Info() +} + +func (z *Zones) warn(err error) slog.Logger { + l := z.log.Warn() + if err != nil { + l = l.WithField(slog.ErrorFieldName, err) + } + return l +} + +func (z *Zones) error(err error) slog.Logger { + l := z.log.Error() + if err != nil { + l = l.WithField(slog.ErrorFieldName, err) + } + return l +} diff --git a/pkg/zones/options.go b/pkg/zones/options.go index faefb9f..e7a66f2 100644 --- a/pkg/zones/options.go +++ b/pkg/zones/options.go @@ -69,6 +69,7 @@ func WithLogger(log slog.Logger) ScanOption { } opt.Logger = log + m.log = log return nil } } diff --git a/pkg/zones/zones.go b/pkg/zones/zones.go index fad4c30..89d2509 100644 --- a/pkg/zones/zones.go +++ b/pkg/zones/zones.go @@ -6,6 +6,7 @@ import ( "sort" "darvaza.org/resolver" + "darvaza.org/slog" "github.com/gofrs/uuid/v5" ) @@ -141,6 +142,7 @@ func (z *Zone) GatewayIDs() ([]int, int) { // Zones represents all zones in a cluster type Zones struct { dir fs.FS + log slog.Logger resolver resolver.Resolver domain string