refactor: move graphql queries to own crate (#487)

Author: gusinacio

Cargo.lock

@@ -3,6 +3,7 @@ members = [
   "crates/common",
   "crates/config",
   "crates/dips",
+  "crates/query",
   "crates/service",
   "crates/tap-agent",
 ]

Cargo.toml

@@ -5,6 +5,7 @@ edition = "2021"
 
 [dependencies]
 indexer-config = { path = "../config" }
+indexer-query = { path = "../query" }
 thiserror.workspace = true
 async-trait.workspace = true
 alloy.workspace = true
@@ -29,14 +30,14 @@ tokio = { workspace = true, features = ["fs", "tokio-macros"] }
 cost-model = { git = "https://github.com/graphprotocol/agora", rev = "3ed34ca" }
 regex = "1.7.1"
 axum-extra = { version = "0.9.3", features = [
-    "typed-header",
+  "typed-header",
 ], default-features = false }
 autometrics = { version = "1.0.1", features = ["prometheus-exporter"] }
 tower_governor = "0.4.0"
 tower-http = { version = "0.5.2", features = [
-    "cors",
-    "normalize-path",
-    "trace",
+  "cors",
+  "normalize-path",
+  "trace",
 ] }
 tokio-util = "0.7.10"
 bip39 = "2.0.0"

crates/common/Cargo.toml

@@ -9,23 +9,11 @@ use std::{
 
 use super::Allocation;
 use crate::{prelude::SubgraphClient, watcher::new_watcher};
-use alloy::primitives::{TxHash, B256, U256};
-use graphql_client::GraphQLQuery;
+use alloy::primitives::TxHash;
+use indexer_query::allocations_query::{self, AllocationsQuery};
 use thegraph_core::{Address, DeploymentId};
 use tokio::sync::watch::Receiver;
 
-type BigInt = U256;
-type Bytes = B256;
-
-#[derive(GraphQLQuery)]
-#[graphql(
-    schema_path = "../graphql/network.schema.graphql",
-    query_path = "../graphql/allocations.query.graphql",
-    response_derives = "Debug",
-    variables_derives = "Clone"
-)]
-pub struct AllocationsQuery;
-
 impl TryFrom<allocations_query::AllocationFragment> for Allocation {
     type Error = anyhow::Error;
 

crates/common/src/allocations/monitor.rs

@@ -5,21 +5,10 @@ use crate::subgraph_client::SubgraphClient;
 use crate::watcher::new_watcher;
 use alloy::primitives::Address;
 use anyhow::Error;
-use graphql_client::GraphQLQuery;
+use indexer_query::dispute_manager::{self, DisputeManager};
 use std::time::Duration;
 use tokio::sync::watch::Receiver;
 
-type Bytes = Address;
-
-#[derive(GraphQLQuery)]
-#[graphql(
-    schema_path = "../graphql/network.schema.graphql",
-    query_path = "../graphql/dispute.query.graphql",
-    response_derives = "Debug",
-    variables_derives = "Clone"
-)]
-struct DisputeManager;
-
 pub async fn dispute_manager(
     network_subgraph: &'static SubgraphClient,
     interval: Duration,

crates/common/src/attestations/dispute_manager.rs

@@ -9,7 +9,7 @@ use std::{
 
 use alloy::primitives::{Address, U256};
 use anyhow::{anyhow, Result};
-use graphql_client::GraphQLQuery;
+use indexer_query::escrow_account::{self, EscrowAccountQuery};
 use thiserror::Error;
 use tokio::sync::watch::Receiver;
 use tracing::{error, warn};
@@ -87,17 +87,6 @@ impl EscrowAccounts {
     }
 }
 
-type BigInt = String;
-
-#[derive(GraphQLQuery)]
-#[graphql(
-    schema_path = "../graphql/tap.schema.graphql",
-    query_path = "../graphql/escrow_account.query.graphql",
-    response_derives = "Debug",
-    variables_derives = "Clone"
-)]
-pub struct EscrowAccountQuery;
-
 pub async fn escrow_accounts(
     escrow_subgraph: &'static SubgraphClient,
     indexer_address: Address,
@@ -121,7 +110,7 @@ async fn get_escrow_accounts(
     // isAuthorized == true means that the signer is still authorized to sign
     // payments in the name of the sender.
     let response = escrow_subgraph
-        .query::<EscrowAccountQuery, _>(escrow_account_query::Variables {
+        .query::<EscrowAccountQuery, _>(escrow_account::Variables {
             indexer: format!("{:x?}", indexer_address),
             thaw_end_timestamp: if reject_thawing_signers {
                 U256::ZERO.to_string()

crates/common/src/escrow_accounts.rs

@@ -8,19 +8,11 @@ use axum::{
 };
 use graphql_client::GraphQLQuery;
 use indexer_config::GraphNodeConfig;
+use indexer_query::{health_query, HealthQuery};
 use reqwest::StatusCode;
 use serde_json::json;
 use thiserror::Error;
 
-#[derive(GraphQLQuery)]
-#[graphql(
-    schema_path = "../graphql/indexing_status.schema.graphql",
-    query_path = "../graphql/subgraph_health.query.graphql",
-    response_derives = "Debug",
-    variables_derives = "Clone"
-)]
-pub struct HealthQuery;
-
 #[derive(Debug, Error)]
 pub enum CheckHealthError {
     #[error("Failed to send request")]

crates/common/src/indexer_service/http/health.rs

@@ -312,6 +312,7 @@ impl SubgraphClient {
 mod test {
     use std::str::FromStr;
 
+    use indexer_query::{current_epoch, user_query, CurrentEpoch, UserQuery};
     use serde_json::json;
     use wiremock::matchers::{method, path};
     use wiremock::{Mock, MockServer, ResponseTemplate};
@@ -356,15 +357,6 @@ mod test {
         .await
     }
 
-    #[derive(GraphQLQuery)]
-    #[graphql(
-        schema_path = "../graphql/network.schema.graphql",
-        query_path = "../graphql/epoch.query.graphql",
-        response_derives = "Debug",
-        variables_derives = "Clone"
-    )]
-    struct CurrentEpoch;
-
     #[tokio::test]
     #[ignore = "depends on the defunct hosted-service"]
     async fn test_network_query() {
@@ -380,15 +372,6 @@ mod test {
         assert!(result.is_ok());
     }
 
-    #[derive(GraphQLQuery)]
-    #[graphql(
-        schema_path = "../graphql/test.schema.graphql",
-        query_path = "../graphql/user.query.graphql",
-        response_derives = "Debug",
-        variables_derives = "Clone"
-    )]
-    struct UserQuery;
-
     #[tokio::test]
     async fn test_uses_local_deployment_if_healthy_and_synced() {
         let deployment =

crates/common/src/subgraph_client/client.rs

@@ -3,24 +3,18 @@
 
 use std::time::Duration;
 
-use deployment_status_query::Health;
 use graphql_client::GraphQLQuery;
+use indexer_query::{
+    deployment_status_query::{self, Health},
+    DeploymentStatusQuery,
+};
 use reqwest::Url;
 use serde::Deserialize;
 use thegraph_core::DeploymentId;
 use tokio::sync::watch::Receiver;
 
 use crate::watcher::new_watcher;
 
-#[derive(GraphQLQuery)]
-#[graphql(
-    schema_path = "../graphql/indexing_status.schema.graphql",
-    query_path = "../graphql/subgraph_deployment_status.graphql",
-    response_derives = "Debug",
-    variables_derives = "Clone"
-)]
-pub struct DeploymentStatusQuery;
-
 #[derive(Clone, Debug, Deserialize, Eq, PartialEq)]
 pub struct DeploymentStatus {
     pub synced: bool,

crates/common/src/subgraph_client/monitor.rs

@@ -0,0 +1,11 @@
+[package]
+name = "indexer-query"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+graphql_client.workspace = true
+thegraph-core.workspace = true
+serde.workspace = true
+alloy.workspace = true
+anyhow.workspace = true