all: Add AMP chain name aliases to config

AMP-powered subgraphs use different network names than graph-node's
internal chain names (e.g., AMP uses "ethereum-mainnet" while graph-node
uses "mainnet"). Add a config-level `amp` field on chains that maps AMP
names to internal names, so AMP manifests work without renaming chains
in the database.

Author: lutter

core/src/subgraph/registrar.rs

@@ -5,6 +5,7 @@ use graph::amp;
 use graph::blockchain::{Blockchain, BlockchainKind, BlockchainMap};
 use graph::components::{
     link_resolver::LinkResolverContext,
+    network_provider::AmpChainNames,
     store::{DeploymentId, DeploymentLocator, SubscriptionManager},
     subgraph::Settings,
 };
@@ -30,6 +31,7 @@ pub struct SubgraphRegistrar<P, S, SM, AC> {
     version_switching_mode: SubgraphVersionSwitchingMode,
     assignment_event_stream_cancel_guard: CancelGuard, // cancels on drop
     settings: Arc<Settings>,
+    amp_chain_names: Arc<AmpChainNames>,
 }
 
 impl<P, S, SM, AC> SubgraphRegistrar<P, S, SM, AC>
@@ -50,6 +52,7 @@ where
         node_id: NodeId,
         version_switching_mode: SubgraphVersionSwitchingMode,
         settings: Arc<Settings>,
+        amp_chain_names: Arc<AmpChainNames>,
     ) -> Self {
         let logger = logger_factory.component_logger("SubgraphRegistrar", None);
         let logger_factory = logger_factory.with_parent(logger.clone());
@@ -69,6 +72,7 @@ where
             version_switching_mode,
             assignment_event_stream_cancel_guard: CancelGuard::new(),
             settings,
+            amp_chain_names,
         }
     }
 
@@ -314,6 +318,7 @@ where
                     &resolver,
                     self.amp_client.cheap_clone(),
                     history_blocks,
+                    &self.amp_chain_names,
                 )
                 .await?
             }
@@ -333,6 +338,7 @@ where
                     &resolver,
                     self.amp_client.cheap_clone(),
                     history_blocks,
+                    &self.amp_chain_names,
                 )
                 .await?
             }
@@ -460,6 +466,7 @@ async fn create_subgraph_version<C: Blockchain, S: SubgraphStore, AC: amp::Clien
     resolver: &Arc<dyn LinkResolver>,
     amp_client: Option<Arc<AC>>,
     history_blocks_override: Option<i32>,
+    amp_chain_names: &AmpChainNames,
 ) -> Result<DeploymentLocator, SubgraphRegistrarError> {
     let raw_string = serde_yaml::to_string(&raw).unwrap();
 
@@ -488,9 +495,10 @@ async fn create_subgraph_version<C: Blockchain, S: SubgraphStore, AC: amp::Clien
         .map_err(SubgraphRegistrarError::ManifestValidationError)?;
 
     let network_name: Word = manifest.network_name().into();
+    let resolved_name = amp_chain_names.resolve(&network_name);
 
     let chain = chains
-        .get::<C>(network_name.clone())
+        .get::<C>(resolved_name.clone())
         .map_err(SubgraphRegistrarError::NetworkNotSupported)?
         .cheap_clone();
 
@@ -570,7 +578,7 @@ async fn create_subgraph_version<C: Blockchain, S: SubgraphStore, AC: amp::Clien
             &manifest.schema,
             deployment,
             node_id,
-            network_name.into(),
+            resolved_name.into(),
             version_switching_mode,
         )
         .await

docs/config.md

@@ -114,6 +114,9 @@ The configuration for a chain `name` is specified in the section
 - `protocol`: the protocol type being indexed, default `ethereum`
   (alternatively `near`, `cosmos`,`arweave`,`starknet`)
 - `polling_interval`: the polling interval for the block ingestor (default 500ms)
+- `amp`: the network name used by AMP for this chain; defaults to the chain name.
+  Set this when AMP uses a different name than graph-node (e.g., `amp = "ethereum-mainnet"`
+  on a chain named `mainnet`).
 - `provider`: a list of providers for that chain
 
 A `provider` is an object with the following characteristics:
@@ -164,6 +167,7 @@ optimisations.
 ingestor = "block_ingestor_node"
 [chains.mainnet]
 shard = "vip"
+amp = "ethereum-mainnet"
 provider = [
   { label = "mainnet1", url = "http://..", features = [], headers = { Authorization = "Bearer foo" } },
   { label = "mainnet2", url = "http://..", features = [ "archive", "traces" ] }

graph/src/components/network_provider/mod.rs

@@ -16,10 +16,60 @@ pub use self::provider_check::ProviderCheckStatus;
 pub use self::provider_manager::ProviderCheckStrategy;
 pub use self::provider_manager::ProviderManager;
 
+use std::collections::HashMap;
+
 // Used to increase memory efficiency.
 // Currently, there is no need to create a separate type for this.
 pub type ChainName = crate::data::value::Word;
 
 // Used to increase memory efficiency.
 // Currently, there is no need to create a separate type for this.
 pub type ProviderName = crate::data::value::Word;
+
+/// Maps AMP network names to internal graph-node chain names.
+///
+/// AMP-powered subgraphs may use different network names than graph-node
+/// (e.g., AMP uses `"ethereum-mainnet"` while graph-node uses `"mainnet"`).
+/// This type provides a config-driven translation layer.
+#[derive(Clone, Debug, Default)]
+pub struct AmpChainNames(HashMap<ChainName, ChainName>);
+
+impl AmpChainNames {
+    pub fn new(mapping: HashMap<ChainName, ChainName>) -> Self {
+        AmpChainNames(mapping)
+    }
+
+    /// Returns the internal chain name for an AMP alias, or the input
+    /// unchanged if no alias matches.
+    pub fn resolve(&self, name: &ChainName) -> ChainName {
+        self.0.get(name).cloned().unwrap_or_else(|| name.clone())
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn amp_chain_names_resolve_known_alias() {
+        let mut map = HashMap::new();
+        map.insert(
+            ChainName::from("ethereum-mainnet"),
+            ChainName::from("mainnet"),
+        );
+        let names = AmpChainNames::new(map);
+        assert_eq!(
+            names.resolve(&ChainName::from("ethereum-mainnet")),
+            ChainName::from("mainnet")
+        );
+    }
+
+    #[test]
+    fn amp_chain_names_resolve_unknown_passthrough() {
+        let names = AmpChainNames::default();
+        assert_eq!(
+            names.resolve(&ChainName::from("mainnet")),
+            ChainName::from("mainnet")
+        );
+    }
+}

node/resources/tests/full_config.toml

@@ -45,6 +45,7 @@ ingestor = "index_0"
 
 [chains.mainnet]
 shard = "primary"
+amp = "ethereum-mainnet"
 provider = [
   { label = "mainnet-0", url = "http://rpc.mainnet.io", features = ["archive", "traces"] },
   { label = "mainnet-1", details = { type = "web3call", url = "http://rpc.mainnet.io", features = ["archive", "traces"] }},