twilight-gateway
twilight-gateway
is an implementation of Discord's sharding gateway sessions.
This is responsible for receiving stateful events in real-time from Discord and
sending some stateful information.
The primary type is the Shard
, a stateful interface to maintain a Websocket
connection to Discord's gateway. Much of its functionality can be configured,
and it's used to receive gateway events or raw Websocket messages, useful for
load balancing and microservices.
Using the stream
module, shards can be easily managed in groups.
Features
simd-json
: use simd-json
instead of serde_json
for deserializing
events
- TLS (mutually exclusive)
native
: platform's native TLS implementation via native-tls
rustls-native-roots
(default): rustls
using native root certificates
rustls-webpki-roots
: rustls
using webpki-roots
for root
certificates, useful for scratch
containers
twilight-http
(default): enable the stream::create_recommended
function
- Zlib (mutually exclusive)
zlib-stock
(default): flate2
's stock zlib implementation
zlib-ng
: use zlib-ng
for zlib, may have better performance
Examples
Create a shard and loop over guild and voice state events:
use std::env;
use twilight_gateway::{Intents, Shard, ShardId};
#[tokio::main]
async fn main() -> anyhow::Result<()> {
tracing_subscriber::fmt::init();
let token = env::var("DISCORD_TOKEN")?;
let intents = Intents::GUILDS | Intents::GUILD_VOICE_STATES;
let mut shard = Shard::new(ShardId::ONE, token, intents);
tracing::info!("started shard");
loop {
let event = match shard.next_event().await {
Ok(event) => event,
Err(source) => {
tracing::warn!(?source, "error receiving event");
if source.is_fatal() {
break;
}
continue;
},
};
tracing::debug!(?event, "received event");
}
Ok(())
}
Create the recommended number of shards and run them concurrently through
ShardEventStream
:
use futures::StreamExt;
use std::env;
use twilight_gateway::{
stream::{self, ShardEventStream},
Config, Intents,
};
use twilight_http::Client;
#[tokio::main]
async fn main() -> anyhow::Result<()> {
tracing_subscriber::fmt::init();
let token = env::var("DISCORD_TOKEN")?;
let client = Client::new(token.clone());
let config = Config::new(token, Intents::GUILDS);
let mut shards = stream::create_recommended(&client, config, |_, builder| builder.build())
.await?
.collect::<Vec<_>>();
let mut stream = ShardEventStream::new(shards.iter_mut());
while let Some((shard, event)) = stream.next().await {
let event = match event {
Ok(event) => event,
Err(source) => {
tracing::warn!(?source, "error receiving event");
if source.is_fatal() {
break;
}
continue;
}
};
tracing::debug!(?event, shard = ?shard.id(), "received event");
}
Ok(())
}
There are a few additional examples located in the
repository.