refactor: use subgraph client as state

Signed-off-by: Gustavo Inacio <[email protected]>

Author: gusinacio

crates/service/src/routes/static_subgraph.rs

@@ -1,7 +1,7 @@
 // Copyright 2023-, Edge & Node, GraphOps, and Semiotic Labs.
 // SPDX-License-Identifier: Apache-2.0
 
-use axum::{body::Bytes, response::IntoResponse, Extension, Json};
+use axum::{body::Bytes, extract::State, response::IntoResponse, Json};
 use reqwest::StatusCode;
 use serde_json::json;
 use tracing::warn;
@@ -10,7 +10,7 @@ use indexer_common::SubgraphClient;
 
 #[autometrics::autometrics]
 pub async fn static_subgraph_request_handler(
-    Extension(subgraph_client): Extension<&'static SubgraphClient>,
+    State(subgraph_client): State<&'static SubgraphClient>,
     body: Bytes,
 ) -> Result<impl IntoResponse, StaticSubgraphError> {
     let response = subgraph_client.query_raw(body).await?;

crates/service/src/service/indexer_service.rs

@@ -5,6 +5,7 @@ use alloy::dyn_abi::Eip712Domain;
 use anyhow;
 use axum::extract::MatchedPath;
 use axum::extract::Request as ExtractRequest;
+use axum::handler::Handler;
 use axum::http::{Method, Request};
 use axum::{
     async_trait,
@@ -414,7 +415,7 @@ impl IndexerService {
                     "/network",
                     post(static_subgraph_request_handler)
                         .route_layer(auth_layer)
-                        .route_layer(Extension(network_subgraph))
+                        .with_state(network_subgraph)
                         .route_layer(static_subgraph_rate_limiter.clone()),
                 );
             } else {
@@ -428,14 +429,13 @@ impl IndexerService {
 
                 let auth_layer = ValidateRequestHeaderLayer::bearer(free_auth_token);
 
-                misc_routes = misc_routes
-                    .route(
-                        "/escrow",
-                        post(static_subgraph_request_handler)
-                            .route_layer(auth_layer)
-                            .route_layer(Extension(escrow_subgraph)),
-                    )
-                    .route_layer(static_subgraph_rate_limiter);
+                misc_routes = misc_routes.route(
+                    "/escrow",
+                    post(static_subgraph_request_handler)
+                        .route_layer(auth_layer)
+                        .with_state(escrow_subgraph)
+                        .route_layer(static_subgraph_rate_limiter),
+                )
             } else {
                 warn!("`serve_escrow_subgraph` is enabled but no `serve_auth_token` provided. Disabling it.");
             }