refactor: improve topology access to indexing

graph-gateway/src/client_query.rs

@@ -232,8 +232,8 @@ async fn handle_client_query_inner(
         .iter()
         .flat_map(move |deployment| {
             let id = deployment.id;
-            deployment.indexers.iter().map(move |indexer| Indexing {
-                indexer: indexer.id,
+            deployment.indexers.keys().map(move |indexer| Indexing {
+                indexer: *indexer,
                 deployment: id,
             })
         })

graph-gateway/src/indexers/indexing.rs

@@ -85,7 +85,7 @@ async fn update_statuses(
     // There can only be one URL per indexer entity in the network subgraph
     let mut indexers: HashMap<Address, (Url, Vec<DeploymentId>)> = Default::default();
     for deployment in deployments.values() {
-        for indexer in &deployment.indexers {
+        for indexer in deployment.indexers.values() {
             let (_, deployments) = indexers
                 .entry(indexer.id)
                 .or_insert_with(|| (indexer.url.clone(), vec![]));

graph-gateway/src/indexings_blocklist.rs

@@ -120,13 +120,7 @@ pub fn deployment_indexer_addresses(
 ) -> Vec<Address> {
     deployments
         .get(deployment_id)
-        .map(|deployment| {
-            deployment
-                .indexers
-                .iter()
-                .map(|indexer| indexer.id)
-                .collect::<Vec<_>>()
-        })
+        .map(|deployment| deployment.indexers.keys().cloned().collect())
         .unwrap_or_default()
 }
 

graph-gateway/src/main.rs

@@ -485,8 +485,8 @@ async fn write_indexer_inputs(
     for (deployment, indexer) in deployments.values().flat_map(|deployment| {
         deployment
             .indexers
-            .iter()
-            .map(move |indexer| (deployment.as_ref(), indexer.as_ref()))
+            .values()
+            .map(|indexer| (deployment.as_ref(), indexer.as_ref()))
     }) {
         let indexing = Indexing {
             indexer: indexer.id,

graph-gateway/src/topology.rs

@@ -11,7 +11,7 @@ use thegraph::types::{DeploymentId, SubgraphId};
 use tokio::sync::RwLock;
 use toolshed::url::Url;
 
-use gateway_common::types::GRT;
+use gateway_common::types::{Indexing, GRT};
 use gateway_framework::{ipfs, network::network_subgraph};
 
 /// Representation of the graph network being used to serve queries
@@ -40,7 +40,7 @@ pub struct Deployment {
     pub manifest: Arc<Manifest>,
     /// An indexer may have multiple active allocations on a deployment. We collapse them into a
     /// single logical allocation using the largest allocation ID and sum of the allocated tokens.
-    pub indexers: Vec<Arc<Indexer>>,
+    pub indexers: HashMap<Address, Arc<Indexer>>,
     /// A deployment may be associated with multiple subgraphs.
     pub subgraphs: BTreeSet<SubgraphId>,
     /// Indicates that the deployment should not be served directly by this gateway. This will
@@ -113,7 +113,7 @@ impl GraphNetwork {
                 .values()
                 .flat_map(|subgraph| &subgraph.deployments)
                 .flat_map(|deployment| &deployment.indexers)
-                .map(|indexer| (indexer.id, indexer.clone()))
+                .map(|(id, indexer)| (*id, indexer.clone()))
                 .collect::<HashMap<Address, Arc<Indexer>>>()
                 .into()
         });
@@ -207,7 +207,7 @@ impl GraphNetwork {
                 indexer.allocated_tokens = total_allocation;
                 Some(indexer)
             })
-            .map(Arc::new)
+            .map(|indexer| (indexer.id, indexer.into()))
             .collect();
 
         // abf62a6d-c071-4507-b528-ddc8e250127a
@@ -232,6 +232,16 @@ impl GraphNetwork {
     pub fn deployment_by_id(&self, id: &DeploymentId) -> Option<Arc<Deployment>> {
         self.deployments.value_immediate()?.get(id).cloned()
     }
+
+    // Get then indexer data for some deployment.
+    pub fn indexing(&self, indexing: &Indexing) -> Option<Arc<Indexer>> {
+        self.deployments
+            .value_immediate()?
+            .get(&indexing.deployment)?
+            .indexers
+            .get(&indexing.indexer)
+            .cloned()
+    }
 }
 
 struct IpfsCache {