feat: add subgraph query auth support

edgeandnode · May 30, 2024 · 1443cce · 1443cce
1 parent e8384a7
commit 1443cce
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/config.rs b/src/config.rs
@@ -15,6 +15,8 @@ pub struct Config {
     pub kafka: Kafka,
     #[serde_as(as = "DisplayFromStr")]
     pub network_subgraph: Url,
+    #[serde(default)]
+    pub query_auth: Option<String>,
     #[serde_as(as = "DisplayFromStr")]
     pub rpc_url: Hidden<Url>,
     pub secret_key: Hidden<B256>,

diff --git a/src/main.rs b/src/main.rs
@@ -62,8 +62,13 @@ async fn main() -> anyhow::Result<()> {
     let http_client = reqwest::Client::builder()
         .timeout(Duration::from_secs(10))
         .build()?;
-    let mut network_subgraph = SubgraphClient::new(http_client.clone(), config.network_subgraph);
-    let mut escrow_subgraph = SubgraphClient::new(http_client.clone(), config.escrow_subgraph);
+    let mut network_subgraph =
+        SubgraphClient::builder(http_client.clone(), config.network_subgraph)
+            .with_auth_token(config.query_auth.clone())
+            .build();
+    let mut escrow_subgraph = SubgraphClient::builder(http_client.clone(), config.escrow_subgraph)
+        .with_auth_token(config.query_auth)
+        .build();
 
     let authorized_signers = authorized_signers(&mut escrow_subgraph, &sender_address)
         .await