toki/StartRNR
1
0
Fork 0
Code Issues Pull requests Activity

Lightly refactor query code

Browse source
This commit is contained in:
Kiana Sheibani 2023-08-27 16:44:18 -04:00
parent 07284bef06
commit 53cde72ede
Signed by: toki
GPG key ID: 6CB106C25E86A9F7
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
cli/src/queries.rs
View file

@ -25,17 +25,16 @@ pub struct Timestamp(pub u64);
// Query machinery // Query machinery
pub trait QueryUnwrap<Vars>: QueryBuilder<Vars> { pub trait QueryUnwrap<Vars>: 'static + QueryBuilder<Vars> {
type VarsUnwrapped; type VarsUnwrapped;
type Unwrapped; type Unwrapped;
fn wrap_vars(vars: Self::VarsUnwrapped) -> Vars; fn wrap_vars(vars: Self::VarsUnwrapped) -> Vars;
fn unwrap_response(response: GraphQlResponse<Self>) -> Option<Self::Unwrapped>; fn unwrap_response(response: GraphQlResponse<Self>) -> Option<Self::Unwrapped>;
} }
// Generic function for running start.gg queries // Generic function for running start.gg queries
pub async fn run_query<Builder: 'static, Vars>( pub async fn run_query<Builder, Vars>(
vars: Builder::VarsUnwrapped, vars: Builder::VarsUnwrapped,
auth: &str, auth: &str,
) -> Option<Builder::Unwrapped> ) -> Option<Builder::Unwrapped>
@ -46,12 +45,12 @@ where
{ {
use cynic::http::SurfExt; use cynic::http::SurfExt;
let query = Builder::build(<Builder as QueryUnwrap<Vars>>::wrap_vars(vars)); let query = Builder::build(Builder::wrap_vars(vars));
let response = surf::post("https://api.start.gg/gql/alpha") let response = surf::post("https://api.start.gg/gql/alpha")
.header("Authorization", String::from("Bearer ") + auth) .header("Authorization", String::from("Bearer ") + auth)
.run_graphql(query) .run_graphql(query)
.await; .await;
<Builder as QueryUnwrap<Vars>>::unwrap_response(response.unwrap()) Builder::unwrap_response(response.unwrap())
} }