add graphql support
This commit is contained in:
parent
d1ab9ae173
commit
ce66c16185
9 changed files with 919 additions and 77 deletions
65
services/api/src/graphql/mod.rs
Normal file
65
services/api/src/graphql/mod.rs
Normal file
|
|
@ -0,0 +1,65 @@
|
|||
use crate::redispool::RedisPool;
|
||||
|
||||
use self::types::World;
|
||||
use juniper::{graphql_object, FieldResult, ID};
|
||||
use rocket::response::content::RawHtml;
|
||||
|
||||
pub mod types;
|
||||
|
||||
pub struct Context {
|
||||
con: RedisPool,
|
||||
}
|
||||
|
||||
impl juniper::Context for Context {}
|
||||
|
||||
#[get("/graphiql")]
|
||||
pub fn graphiql() -> RawHtml<String> {
|
||||
juniper_rocket::graphiql_source("/graphql", None)
|
||||
}
|
||||
|
||||
#[get("/")]
|
||||
pub fn playground() -> RawHtml<String> {
|
||||
juniper_rocket::playground_source("/graphql", None)
|
||||
}
|
||||
|
||||
#[post("/", data = "<query>")]
|
||||
pub async fn post_graphql(
|
||||
query: juniper_rocket::GraphQLRequest,
|
||||
schema: &rocket::State<Schema>,
|
||||
con: &RedisPool,
|
||||
) -> juniper_rocket::GraphQLResponse {
|
||||
query.execute(&*schema, &Context { con: con.clone() }).await
|
||||
}
|
||||
|
||||
#[get("/?<query..>")]
|
||||
pub async fn get_graphql(
|
||||
query: juniper_rocket::GraphQLRequest,
|
||||
schema: &rocket::State<Schema>,
|
||||
con: &RedisPool,
|
||||
) -> juniper_rocket::GraphQLResponse {
|
||||
query.execute(&*schema, &Context { con: con.clone() }).await
|
||||
}
|
||||
|
||||
pub struct Query;
|
||||
|
||||
#[graphql_object(context = Context)]
|
||||
impl Query {
|
||||
fn world(id: ID) -> FieldResult<Option<World>> {
|
||||
Ok(Some(World { id }))
|
||||
}
|
||||
}
|
||||
|
||||
pub type Schema = juniper::RootNode<
|
||||
'static,
|
||||
Query,
|
||||
juniper::EmptyMutation<Context>,
|
||||
juniper::EmptySubscription<Context>,
|
||||
>;
|
||||
|
||||
pub fn schema() -> Schema {
|
||||
Schema::new(
|
||||
Query,
|
||||
juniper::EmptyMutation::<Context>::new(),
|
||||
juniper::EmptySubscription::<Context>::new(),
|
||||
)
|
||||
}
|
||||
271
services/api/src/graphql/types.rs
Normal file
271
services/api/src/graphql/types.rs
Normal file
|
|
@ -0,0 +1,271 @@
|
|||
use juniper::graphql_object;
|
||||
use once_cell::sync::Lazy;
|
||||
use rocket_db_pools::deadpool_redis::redis::{cmd, pipe};
|
||||
use std::{
|
||||
collections::HashMap,
|
||||
ops::Sub,
|
||||
time::{Duration, SystemTime},
|
||||
};
|
||||
|
||||
static WORLD_ID_TO_NAME: Lazy<HashMap<&str, &str>> = Lazy::new(|| {
|
||||
HashMap::from([
|
||||
("1", "Connery"),
|
||||
("10", "Miller"),
|
||||
("13", "Cobalt"),
|
||||
("17", "Emerald"),
|
||||
("19", "Jaeger"),
|
||||
("40", "SolTech"),
|
||||
("1000", "Genudine"),
|
||||
("2000", "Ceres"),
|
||||
])
|
||||
});
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct World {
|
||||
pub id: juniper::ID,
|
||||
}
|
||||
|
||||
#[graphql_object(context = super::Context)]
|
||||
impl World {
|
||||
pub fn name(&self) -> String {
|
||||
WORLD_ID_TO_NAME
|
||||
.get(&self.id.to_string().as_str())
|
||||
.unwrap_or(&"Unknown")
|
||||
.to_string()
|
||||
}
|
||||
|
||||
pub async fn population(&self, context: &mut super::Context) -> i32 {
|
||||
let mut con = (*context).con.get().await.unwrap();
|
||||
let id = self.id.to_string();
|
||||
|
||||
let filter_timestamp = SystemTime::now()
|
||||
.sub(Duration::from_secs(60 * 15))
|
||||
.duration_since(SystemTime::UNIX_EPOCH)
|
||||
.unwrap()
|
||||
.as_secs();
|
||||
|
||||
let (vs, nc, tr, ns): (i32, i32, i32, i32) = pipe()
|
||||
.zcount(format!("wp:{}/{}", id, 1), filter_timestamp, "+inf")
|
||||
.zcount(format!("wp:{}/{}", id, 2), filter_timestamp, "+inf")
|
||||
.zcount(format!("wp:{}/{}", id, 3), filter_timestamp, "+inf")
|
||||
.zcount(format!("wp:{}/{}", id, 4), filter_timestamp, "+inf")
|
||||
.query_async(&mut con)
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
tr + vs + nc + ns
|
||||
}
|
||||
|
||||
pub async fn faction_population(&self) -> FactionPopulation {
|
||||
FactionPopulation {
|
||||
world_id: self.id.clone(),
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn vehicles(&self) -> Vehicles {
|
||||
Vehicles {
|
||||
world_id: self.id.clone(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub struct FactionPopulation {
|
||||
world_id: juniper::ID,
|
||||
}
|
||||
|
||||
impl FactionPopulation {
|
||||
async fn by_faction(&self, context: &super::Context, world_id: String, faction: i32) -> i32 {
|
||||
let mut con = (*context).con.get().await.unwrap();
|
||||
|
||||
let filter_timestamp = SystemTime::now()
|
||||
.sub(Duration::from_secs(60 * 15))
|
||||
.duration_since(SystemTime::UNIX_EPOCH)
|
||||
.unwrap()
|
||||
.as_secs();
|
||||
|
||||
cmd("ZCOUNT")
|
||||
.arg(format!("wp:{}/{}", world_id, faction))
|
||||
.arg(filter_timestamp)
|
||||
.arg("+inf")
|
||||
.query_async(&mut con)
|
||||
.await
|
||||
.unwrap()
|
||||
}
|
||||
}
|
||||
|
||||
#[graphql_object(context = super::Context)]
|
||||
#[graphql(description = "The population of each faction on a world")]
|
||||
impl FactionPopulation {
|
||||
async fn vs(&self, context: &super::Context) -> i32 {
|
||||
self.by_faction(context, self.world_id.to_string(), 1).await
|
||||
}
|
||||
async fn nc(&self, context: &super::Context) -> i32 {
|
||||
self.by_faction(context, self.world_id.to_string(), 2).await
|
||||
}
|
||||
async fn tr(&self, context: &super::Context) -> i32 {
|
||||
self.by_faction(context, self.world_id.to_string(), 3).await
|
||||
}
|
||||
async fn ns(&self, context: &super::Context) -> i32 {
|
||||
self.by_faction(context, self.world_id.to_string(), 4).await
|
||||
}
|
||||
}
|
||||
|
||||
pub struct Vehicles {
|
||||
world_id: juniper::ID,
|
||||
}
|
||||
|
||||
impl Vehicles {
|
||||
async fn get_vehicle(
|
||||
&self,
|
||||
context: &super::Context,
|
||||
world_id: String,
|
||||
vehicle_name: &str,
|
||||
) -> i32 {
|
||||
let mut con = (*context).con.get().await.unwrap();
|
||||
|
||||
let filter_timestamp = SystemTime::now()
|
||||
.sub(Duration::from_secs(60 * 15))
|
||||
.duration_since(SystemTime::UNIX_EPOCH)
|
||||
.unwrap()
|
||||
.as_secs();
|
||||
|
||||
cmd("ZCOUNT")
|
||||
.arg(format!("v:{}/{}", world_id, vehicle_name))
|
||||
.arg(filter_timestamp)
|
||||
.arg("+inf")
|
||||
.query_async(&mut con)
|
||||
.await
|
||||
.unwrap()
|
||||
}
|
||||
}
|
||||
|
||||
#[graphql_object(context = super::Context)]
|
||||
#[graphql(description = "The count of active vehicles on a world")]
|
||||
impl Vehicles {
|
||||
// Transporters
|
||||
async fn flash(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "flash")
|
||||
.await
|
||||
}
|
||||
async fn sunderer(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "sunderer")
|
||||
.await
|
||||
}
|
||||
async fn ant(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "ant")
|
||||
.await
|
||||
}
|
||||
async fn harasser(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "harasser")
|
||||
.await
|
||||
}
|
||||
async fn javelin(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "javelin")
|
||||
.await
|
||||
}
|
||||
|
||||
// Tanks
|
||||
async fn lightning(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "lightning")
|
||||
.await
|
||||
}
|
||||
async fn prowler(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "prowler")
|
||||
.await
|
||||
}
|
||||
async fn vanguard(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "vanguard")
|
||||
.await
|
||||
}
|
||||
async fn magrider(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "magrider")
|
||||
.await
|
||||
}
|
||||
async fn chimera(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "chimera")
|
||||
.await
|
||||
}
|
||||
|
||||
// Air
|
||||
async fn mosquito(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "mosquito")
|
||||
.await
|
||||
}
|
||||
async fn liberator(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "liberator")
|
||||
.await
|
||||
}
|
||||
async fn galaxy(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "galaxy")
|
||||
.await
|
||||
}
|
||||
async fn valkyrie(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "valkyrie")
|
||||
.await
|
||||
}
|
||||
async fn reaver(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "reaver")
|
||||
.await
|
||||
}
|
||||
async fn scythe(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "scythe")
|
||||
.await
|
||||
}
|
||||
async fn dervish(&self, context: &super::Context) -> i32 {
|
||||
self.get_vehicle(context, self.world_id.to_string(), "dervish")
|
||||
.await
|
||||
}
|
||||
}
|
||||
|
||||
pub struct Classes {
|
||||
pub world_id: juniper::ID,
|
||||
}
|
||||
|
||||
impl Classes {
|
||||
async fn get_class(&self, context: &super::Context, world_id: String, class_name: &str) -> i32 {
|
||||
let mut con = (*context).con.get().await.unwrap();
|
||||
|
||||
let filter_timestamp = SystemTime::now()
|
||||
.sub(Duration::from_secs(60 * 15))
|
||||
.duration_since(SystemTime::UNIX_EPOCH)
|
||||
.unwrap()
|
||||
.as_secs();
|
||||
|
||||
cmd("ZCOUNT")
|
||||
.arg(format!("c:{}/{}", world_id, class_name))
|
||||
.arg(filter_timestamp)
|
||||
.arg("+inf")
|
||||
.query_async(&mut con)
|
||||
.await
|
||||
.unwrap()
|
||||
}
|
||||
}
|
||||
|
||||
#[graphql_object(context = super::Context)]
|
||||
#[graphql(description = "The count of active classes on a world")]
|
||||
impl Classes {
|
||||
async fn infiltrator(&self, context: &super::Context) -> i32 {
|
||||
self.get_class(context, self.world_id.to_string(), "infiltrator")
|
||||
.await
|
||||
}
|
||||
async fn light_assault(&self, context: &super::Context) -> i32 {
|
||||
self.get_class(context, self.world_id.to_string(), "light_assault")
|
||||
.await
|
||||
}
|
||||
async fn combat_medic(&self, context: &super::Context) -> i32 {
|
||||
self.get_class(context, self.world_id.to_string(), "combat_medic")
|
||||
.await
|
||||
}
|
||||
async fn engineer(&self, context: &super::Context) -> i32 {
|
||||
self.get_class(context, self.world_id.to_string(), "engineer")
|
||||
.await
|
||||
}
|
||||
async fn heavy_assault(&self, context: &super::Context) -> i32 {
|
||||
self.get_class(context, self.world_id.to_string(), "heavy_assault")
|
||||
.await
|
||||
}
|
||||
async fn max(&self, context: &super::Context) -> i32 {
|
||||
self.get_class(context, self.world_id.to_string(), "max")
|
||||
.await
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue