core, graph, store: Label deployment_(head|failed) with the shard

Author: lutter

core/src/subgraph/instance_manager.rs

@@ -362,6 +362,7 @@ where
             registry.clone(),
             &deployment.hash,
             manifest.network_name(),
+            store.shard().to_string(),
             stopwatch_metrics,
         ));
 

graph/src/blockchain/block_stream.rs

@@ -94,6 +94,7 @@ impl BlockStreamMetrics {
         registry: Arc<impl MetricsRegistry>,
         deployment_id: &DeploymentHash,
         network: String,
+        shard: String,
         stopwatch: StopwatchMetrics,
     ) -> Self {
         let reverted_blocks = registry
@@ -103,7 +104,11 @@ impl BlockStreamMetrics {
                 deployment_id.as_str(),
             )
             .expect("Failed to create `deployment_reverted_blocks` gauge");
-        let labels = labels! { String::from("deployment") => deployment_id.to_string(), String::from("network") => network };
+        let labels = labels! {
+            String::from("deployment") => deployment_id.to_string(),
+            String::from("network") => network,
+            String::from("shard") => shard
+        };
         let deployment_head = registry
             .new_gauge(
                 "deployment_head",

graph/src/components/store.rs

@@ -1039,6 +1039,10 @@ pub trait WritableStore: Send + Sync + 'static {
 
     /// Load the dynamic data sources for the given deployment
     async fn load_dynamic_data_sources(&self) -> Result<Vec<StoredDynamicDataSource>, StoreError>;
+
+    /// Report the name of the shard in which the subgraph is stored. This
+    /// should only be used for reporting and monitoring
+    fn shard(&self) -> &str;
 }
 
 #[async_trait]
@@ -1211,6 +1215,10 @@ impl WritableStore for MockStore {
     fn deployment_synced(&self) -> Result<(), Error> {
         unimplemented!()
     }
+
+    fn shard(&self) -> &str {
+        unimplemented!()
+    }
 }
 
 pub trait BlockStore: Send + Sync + 'static {

store/postgres/src/subgraph_store.rs

@@ -1143,6 +1143,10 @@ impl WritableStoreTrait for WritableStore {
 
         Ok(self.store.send_store_event(&event)?)
     }
+
+    fn shard(&self) -> &str {
+        self.site.shard.as_str()
+    }
 }
 
 fn same_subgraph(mods: &Vec<EntityModification>, id: &DeploymentHash) -> bool {