chore: add prometheus metrics

revoltchat · Jul 4, 2023 · 76150db · 76150db
1 parent 815c3fe
commit 76150db
Show file tree

Hide file tree

Showing 10 changed files with 52 additions and 21 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/bonfire/Cargo.toml b/crates/bonfire/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "revolt-bonfire"
-version = "0.6.4"
+version = "0.6.5"
 license = "AGPL-3.0-or-later"
 edition = "2021"
 

diff --git a/crates/core/database/Cargo.toml b/crates/core/database/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "revolt-database"
-version = "0.6.4"
+version = "0.6.5"
 edition = "2021"
 license = "AGPL-3.0-or-later"
 authors = [ "Paul Makles <me@insrt.uk>" ]
@@ -22,10 +22,10 @@ default = [ "mongodb", "async-std-runtime" ]
 
 [dependencies]
 # Core
-revolt-result = { version = "0.6.4", path = "../result" }
-revolt-models = { version = "0.6.4", path = "../models" }
-revolt-presence = { version = "0.6.4", path = "../presence" }
-revolt-permissions = { version = "0.6.4", path = "../permissions", features = [ "serde", "bson" ] }
+revolt-result = { version = "0.6.5", path = "../result" }
+revolt-models = { version = "0.6.5", path = "../models" }
+revolt-presence = { version = "0.6.5", path = "../presence" }
+revolt-permissions = { version = "0.6.5", path = "../permissions", features = [ "serde", "bson" ] }
 
 # Utility
 log = "0.4"

diff --git a/crates/core/models/Cargo.toml b/crates/core/models/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "revolt-models"
-version = "0.6.4"
+version = "0.6.5"
 edition = "2021"
 license = "AGPL-3.0-or-later"
 authors = [ "Paul Makles <me@insrt.uk>" ]
@@ -18,7 +18,7 @@ default = [ "serde", "partials" ]
 
 [dependencies]
 # Core
-revolt-permissions = { version = "0.6.4", path = "../permissions" }
+revolt-permissions = { version = "0.6.5", path = "../permissions" }
 
 # Serialisation
 revolt_optional_struct = { version = "0.2.0", optional = true }

diff --git a/crates/core/permissions/Cargo.toml b/crates/core/permissions/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "revolt-permissions"
-version = "0.6.4"
+version = "0.6.5"
 edition = "2021"
 license = "AGPL-3.0-or-later"
 authors = [ "Paul Makles <me@insrt.uk>" ]

diff --git a/crates/core/presence/Cargo.toml b/crates/core/presence/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "revolt-presence"
-version = "0.6.4"
+version = "0.6.5"
 edition = "2021"
 license = "AGPL-3.0-or-later"
 authors = [ "Paul Makles <me@insrt.uk>" ]

diff --git a/crates/core/result/Cargo.toml b/crates/core/result/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "revolt-result"
-version = "0.6.4"
+version = "0.6.5"
 edition = "2021"
 license = "AGPL-3.0-or-later"
 authors = [ "Paul Makles <me@insrt.uk>" ]

diff --git a/crates/delta/Cargo.toml b/crates/delta/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "revolt-delta"
-version = "0.6.4"
+version = "0.6.5"
 license = "AGPL-3.0-or-later"
 authors = ["Paul Makles <paulmakles@gmail.com>"]
 edition = "2018"
@@ -47,6 +47,7 @@ lettre = "0.10.0-alpha.4"
 rocket = { version = "0.5.0-rc.2", default-features = false, features = ["json"] }
 rocket_empty = { version = "0.1.1", features = ["schema"] }
 rocket_authifier = { version = "1.0.7" }
+rocket_prometheus = "0.10.0-rc.3"
 
 # spec generation
 schemars = "0.8.8"

diff --git a/crates/delta/src/main.rs b/crates/delta/src/main.rs
@@ -8,6 +8,7 @@ extern crate serde_json;
 pub mod routes;
 pub mod util;
 
+use rocket_prometheus::PrometheusMetrics;
 use std::net::Ipv4Addr;
 
 use async_std::channel::unbounded;
@@ -65,7 +66,11 @@ async fn rocket() -> _ {
 
     // Configure Rocket
     let rocket = rocket::build();
+    let prometheus = PrometheusMetrics::new();
+
     routes::mount(rocket)
+        .attach(prometheus.clone())
+        .mount("/metrics", prometheus)
         .mount("/", revolt_quark::web::cors::catch_all_options_routes())
         .mount("/", revolt_quark::web::ratelimiter::routes())
         .mount("/swagger/", revolt_quark::web::swagger::routes())

diff --git a/crates/quark/Cargo.toml b/crates/quark/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "revolt-quark"
-version = "0.6.4"
+version = "0.6.5"
 edition = "2021"
 license = "AGPL-3.0-or-later"