diff --git a/api/datastore/datastore.go b/api/datastore/datastore.go index e93236ca4..b50148ed2 100644 --- a/api/datastore/datastore.go +++ b/api/datastore/datastore.go @@ -18,7 +18,11 @@ func New(ctx context.Context, dbURL string) (models.Datastore, error) { return nil, err } - return datastoreutil.MetricDS(datastoreutil.NewValidator(ds)), nil + return Wrap(ds), nil +} + +func Wrap(ds models.Datastore) models.Datastore { + return datastoreutil.MetricDS(datastoreutil.NewValidator(ds)) } func newds(ctx context.Context, dbURL string) (models.Datastore, error) { diff --git a/api/server/server.go b/api/server/server.go index 4cfd9392b..e13c26a0e 100644 --- a/api/server/server.go +++ b/api/server/server.go @@ -325,7 +325,7 @@ func WithNodeCertAuthority(ca string) ServerOption { func WithDatastore(ds models.Datastore) ServerOption { return func(ctx context.Context, s *Server) error { - s.datastore = ds + s.datastore = datastore.Wrap(ds) return nil } }