#apollo-federation #graphql #federation #gateway #http-request #websocket #back-end

apollo-gateway-rs

Apollo-gateway-rs is Apollo Federation implemented in Rust

11 releases

0.9.1 Feb 23, 2024
0.9.0 Nov 14, 2023
0.8.3 Jul 1, 2023
0.8.2 Jun 30, 2023
0.7.8 Apr 24, 2022

#241 in HTTP server

Download history 3/week @ 2024-02-15 127/week @ 2024-02-22 10/week @ 2024-02-29 52/week @ 2024-03-07 15/week @ 2024-03-14

205 downloads per month

MIT/Apache

365KB
11K SLoC

Apollo-gateway-rs is Apollo Federation implemented in Rust.

Please create issue if you have a question or find a bug or need a feature.

Quick start

Define your remote source and implement RemoteGraphQLDataSource for it

pub struct CommonSource {
    pub name: String,
    pub addr: String,
    pub tls: bool,
}
impl RemoteGraphQLDataSource for CommonSource {
    fn name(&self) -> &str {
        &self.name
    }
    fn address(&self) -> &str {
        &self.addr
    }
    fn tls(&self) -> bool {
        self.tls
    }
}

Build a gateway-server with your sources

let gateway_server = GatewayServer::builder()
        .with_source(CommonSource::new("countries", "countries.trevorblades.com", true))
        .build();

Configure reqwest handlers

use async_graphql::http::{GraphQLPlaygroundConfig, playground_source};
use apollo_gateway_rs::{actix::{graphql_request, graphql_subscription}};

pub async fn playground() -> HttpResponse {
    let html = playground_source(GraphQLPlaygroundConfig::new("/").subscription_endpoint("/"));
    HttpResponse::Ok()
        .content_type("text/html; charset=utf-8")
        .body(html)
}
fn configure_api(config: &mut actix_web::web::ServiceConfig) {
    config.service(
        actix_web::web::resource("/")
            .route(actix_web::web::post().to(graphql_request))
            .route(
                actix_web::web::get()
                    .guard(actix_web::guard::Header("upgrade", "websocket"))
                    .to(graphql_subscription),
            )
            .route(actix_web::web::get().to(playground)),
    );
}

And spawn actix-web server!

async fn main() -> std::io::Result<()> {
    let gateway_server = GatewayServer::builder()
        .with_source(CommonSource::new("countries", "countries.trevorblades.com", true))
        .build();
    let gateway_server = Data::new(gateway_server);
    HttpServer::new(move || App::new()
        .app_data(gateway_server.clone())
        .configure(configure_api)
    )
        .bind("0.0.0.0:3000")?
        .run()
        .await
}

You can see full example in examples/actix/common_usage

Features

The gateway can modify the details of an incoming request before executing it across your subgraphs. For example, your subgraphs might all use the same authorization token to associate an incoming request with a particular user. The gateway can add that token to each operation it sends to your subgraphs.

#[async_trait::async_trait]
impl GraphqlSourceMiddleware for AuthSource {
    async fn did_receive_response(&self, response: &mut Response, ctx: &Context) -> anyhow::Result<()> {
        let session = ctx.get_session();
        if let Some(jwt) = response.headers.get("user-id")
            .and_then(|header| header.parse().ok())
            .and_then(|user_id| create_jwt(user_id).ok()) {
            session.insert("auth", jwt);
        }
        Ok(())
    }
}
#[async_trait::async_trait]
impl GraphqlSourceMiddleware for UserSource {
    async fn will_send_request(&self, request: &mut Request, ctx: &Context) -> anyhow::Result<()> {
        let session = ctx.get_session();
        if let Some(user_id) = session.get("auth")
                .ok()
                .flatten()
                .and_then(|identity| decode_identity(identity).ok())
                .map(|claims| claims.claims.id) {
            request.headers.insert("user-id".to_string(), user_id.to_string());
        }
        Ok(())
    }
}

You can find full example in examples/actix/authentication

Loading sources from config

You can define your source or use a DefaultSource and load it from json file.

let gateway_server = GatewayServer::builder()
    .with_sources_from_json::<DefaultSource>("sources.json")?
    .build();

You can see full example in examples/actix/from_config

Subscription support

Apollo-gateway-rs support subscription, use apollo_gateway_rs::actix::graphql_subscription if you want it.

Backend implementations

  • Actix-web
  • Rocket
  • Warp
  • Axum

Contribute

Welcome to contribute !

Dependencies

~74MB
~1.5M SLoC