fix: use geo enum for sql queries

ultrasoundmoney · Sep 4, 2024 · 95b317c · 95b317c
1 parent 379fd80
commit 95b317c
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/src/phoenix/demotion_monitor.rs b/src/phoenix/demotion_monitor.rs
@@ -16,12 +16,12 @@ use crate::{
 
 use super::{
     checkpoint::{self, CheckpointId},
-    env::APP_CONFIG,
+    env::{Geo, APP_CONFIG},
 };
 
 #[derive(Debug, Clone)]
 pub struct BuilderDemotion {
-    pub geo: String,
+    pub geo: Geo,
     pub builder_pubkey: String,
     pub builder_id: Option<String>,
     pub slot: i64,

diff --git a/src/phoenix/env.rs b/src/phoenix/env.rs
@@ -87,8 +87,12 @@ fn default_missed_slots_alert_threshold() -> i64 {
 
 /// Auction geography
 #[allow(clippy::upper_case_acronyms)]
+#[derive(Debug, Clone, sqlx::Type)]
+#[sqlx(type_name = "geo")]
 pub enum Geo {
+    #[sqlx(rename = "rbx")]
     RBX,
+    #[sqlx(rename = "vin")]
     VIN,
 }
 

diff --git a/src/phoenix/inclusion_monitor/mod.rs b/src/phoenix/inclusion_monitor/mod.rs
@@ -29,7 +29,7 @@ use super::{
         SendAlert,
     },
     checkpoint::{self, CheckpointId},
-    env::APP_CONFIG,
+    env::{Geo, APP_CONFIG},
 };
 
 #[derive(Debug)]
@@ -40,7 +40,7 @@ struct DeliveredPayload {
     inserted_at: DateTime<Utc>,
     proposer_pubkey: String,
     slot: i64,
-    geo: String,
+    geo: Geo,
 }
 
 async fn get_delivered_payloads(