Implement CreateDependencyReader

Signed-off-by: Mahad Zaryab <[email protected]>
jaegertracing · Nov 7, 2024 · 9be5d3e · 9be5d3e
1 parent 4b97ffd
commit 9be5d3e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/cmd/jaeger/internal/extension/jaegerstorage/factoryadapter/factory.go b/cmd/jaeger/internal/extension/jaegerstorage/factoryadapter/factory.go
@@ -8,6 +8,7 @@ import (
 	"io"
 
 	storage_v1 "github.com/jaegertracing/jaeger/storage"
+	"github.com/jaegertracing/jaeger/storage/dependencystore"
 	"github.com/jaegertracing/jaeger/storage_v2/spanstore"
 )
 
@@ -51,3 +52,8 @@ func (f *Factory) CreateTraceWriter() (spanstore.Writer, error) {
 	}
 	return NewTraceWriter(spanWriter), nil
 }
+
+// CreateDependencyReader creates a dependencystore.Reader.
+func (f *Factory) CreateDependencyReader() (dependencystore.Reader, error) {
+	return f.ss.CreateDependencyReader()
+}
diff --git a/storage_v2/spanstore/factory.go b/storage_v2/spanstore/factory.go
@@ -4,6 +4,7 @@
 package spanstore
 
 import (
+	"github.com/jaegertracing/jaeger/storage/dependencystore"
 	"github.com/jaegertracing/jaeger/storage_v2"
 )
 
@@ -17,4 +18,7 @@ type Factory interface {
 
 	// CreateTraceWriter creates a spanstore.Writer.
 	CreateTraceWriter() (Writer, error)
+
+	// CreateDependencyReader creates a dependencystore.Reader.
+	CreateDependencyReader() (dependencystore.Reader, error)
 }