Expose data source list method to querier (#5400)

mindersec · Feb 3, 2025 · f5f00ed · f5f00ed
1 parent f9b49bf
commit f5f00ed
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/pkg/querier/datasource.go b/pkg/querier/datasource.go
@@ -15,6 +15,10 @@ import (
 
 // DataSourceHandlers interface provides functions to interact with data sources
 type DataSourceHandlers interface {
+	ListDataSourcesByProject(
+		ctx context.Context,
+		projectID uuid.UUID,
+	) ([]*pb.DataSource, error)
 	CreateDataSource(
 		ctx context.Context,
 		projectID uuid.UUID,
@@ -39,6 +43,17 @@ type DataSourceHandlers interface {
 	) error
 }
 
+// ListDataSourcesByProject returns a list of data sources by project ID
+func (q *querierType) ListDataSourcesByProject(ctx context.Context, projectID uuid.UUID) ([]*pb.DataSource, error) {
+	if q.querier == nil {
+		return nil, ErrQuerierMissing
+	}
+	if q.dataSourceSvc == nil {
+		return nil, ErrDataSourceSvcMissing
+	}
+	return q.dataSourceSvc.List(ctx, projectID, dsservice.ReadBuilder().WithTransaction(q.querier))
+}
+
 // CreateDataSource creates a data source
 func (q *querierType) CreateDataSource(
 	ctx context.Context,