Add various prometheus metrics

Add routes for tracking command usage via influxdb
Add routes for getting analytics like cmd usage
Update year in LICENSE
Response code cleanup
Make get post route work with slugs and only public/unlisted visibility
Add new features to readme

Cargo.toml

@@ -35,6 +35,11 @@ sha1 = "0.10.6"
 hex = "0.4.3"
 gql_client = "1.0.7"
 actix-cors = "0.7.0"
+prometheus = { version = "0.13.4", features = ["process"] }
+futures-util = "0.3.30"
+prometheus-http-query = "0.8.3"
+influxdb2 = "0.5.1"
+influxdb2-structmap = "0.2.0"
 
 [profile.release]
 lto = true

LICENSE

@@ -1,4 +1,4 @@
-MIT License Copyright (c) 2022 Dustin Rouillard
+MIT License Copyright (c) 2024 Dustin Rouillard
 
 Permission is hereby granted,
 free of charge, to any person obtaining a copy of this software and associated

README.md

@@ -1,10 +1,15 @@
-# Dustin API
+# dstn.to API
 
 There is probably a lot of better ways to do many of the things I've done here, but this is one of the first things I've done in rust, make suggestions if you see anything done weirdly!
 
+Built with actix-web, makes use of prisma to handle queries to a postgresql database and uses valkey/redis for storage of tokens and caching data.
+
 ## Used for
 
-- Spotify History and Now Playing API / Queue Messages
-- File and Screenshot Uploads
+- Spotify History and Now Playing API / Provides realtime queue for [gateway](https://github.com/dustinrouillard/dustin-gateway)
+- File and Screenshot Uploads (Multipart uploads to an s3 bucket)
 - Github pinned repositories
-- Blog System for Personal Site
+- Blog System for [Personal Site](https://github.com/dustinrouillard/personal-site)
+- Local weather (This just proxies my [weather worker](https://github.com/dustinrouillard/weather-worker))
+- Analytics tracking (commands per day, etc)
+- Prometheus metrics (API route and process metrics)

src/config.rs

@@ -11,6 +11,9 @@ pub struct Config {
   #[envconfig(from = "LISTEN_PORT", default = "8080")]
   pub listen_port: u16,
 
+  #[envconfig(from = "METRICS_LISTEN_PORT", default = "8081")]
+  pub metrics_listen_port: u16,
+
   #[envconfig(
     from = "SPOTIFY_CLIENT_ID",
     default = "01ba26764aca4594a26f4cc59cd3f01f"
@@ -64,4 +67,28 @@ pub struct Config {
 
   #[envconfig(from = "GITHUB_PAT", default = "")]
   pub github_pat: String,
+
+  #[envconfig(from = "WEATHER_COORDS", default = "37.8283/-96.5795")]
+  pub weather_coords: String,
+
+  #[envconfig(
+    from = "PROMETHEUS_HOST",
+    default = "https://prometheus.monit.kush/"
+  )]
+  pub prometheus_host: String,
+
+  #[envconfig(from = "INFLUXDB_TOKEN", default = "")]
+  pub influxdb_token: String,
+
+  #[envconfig(
+    from = "INFLUXDB_HOST",
+    default = "http://influxdb.kube-system"
+  )]
+  pub influxdb_host: String,
+
+  #[envconfig(from = "INFLUXDB_ORG", default = "lab")]
+  pub influxdb_org: String,
+
+  #[envconfig(from = "INFLUXDB_BUCKET", default = "api")]
+  pub influxdb_bucket: String,
 }

src/connectivity/influxdb.rs

@@ -0,0 +1,23 @@
+use envconfig::Envconfig;
+use influxdb2::Client;
+
+use crate::config::Config;
+
+#[derive(Clone)]
+pub struct InfluxManager {
+  pub client: Client,
+}
+
+impl InfluxManager {
+  pub async fn new() -> Self {
+    let config = Config::init_from_env().unwrap();
+    let client = influxdb2::Client::new(
+      config.influxdb_host,
+      config.influxdb_org,
+      config.influxdb_token,
+    );
+
+    tracing::info!("Connected to influxdb");
+    Self { client }
+  }
+}