Intermediate changes

2 files changed, 8 insertions, 3 deletions

diff --git a/library/go/core/metrics/prometheus/registry.go b/library/go/core/metrics/prometheus/registry.go
index e10aef935e1..d43e9267170 100644
--- a/library/go/core/metrics/prometheus/registry.go
+++ b/library/go/core/metrics/prometheus/registry.go
@@ -16,7 +16,7 @@ var _ metrics.Registry = (*Registry)(nil)
 var _ metrics.MetricsStreamer = (*Registry)(nil)
 
 type Registry struct {
-	rg *prometheus.Registry
+	rg PrometheusRegistry
 
 	m             *sync.Mutex
 	subregistries map[string]*Registry
diff --git a/library/go/core/metrics/prometheus/registry_opts.go b/library/go/core/metrics/prometheus/registry_opts.go
index 1dab30d40d9..2ced795b0cc 100644
--- a/library/go/core/metrics/prometheus/registry_opts.go
+++ b/library/go/core/metrics/prometheus/registry_opts.go
@@ -10,10 +10,15 @@ import (
 	"github.com/ydb-platform/ydb/library/go/core/metrics/internal/pkg/registryutil"
 )
 
+type PrometheusRegistry interface {
+	prometheus.Registerer
+	prometheus.Gatherer
+}
+
 type RegistryOpts struct {
 	Prefix        string
 	Tags          map[string]string
-	rg            *prometheus.Registry
+	rg            PrometheusRegistry
 	Collectors    []func(metrics.Registry)
 	NameSanitizer func(string) string
 	StreamFormat  expfmt.Format
@@ -58,7 +63,7 @@ func (o *RegistryOpts) AppendPrefix(prefix string) *RegistryOpts {
 //
 // This is primarily used to unite externally defined metrics with metrics kept
 // in the core registry.
-func (o *RegistryOpts) SetRegistry(rg *prometheus.Registry) *RegistryOpts {
+func (o *RegistryOpts) SetRegistry(rg PrometheusRegistry) *RegistryOpts {
 	o.rg = rg
 	return o
 }