translate sp name to entity key

helium · Jan 3, 2024 · 4626b99 · 4626b99
1 parent e9b7f8e
commit 4626b99
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/reward_index/src/indexer.rs b/reward_index/src/indexer.rs
@@ -162,7 +162,7 @@ impl Indexer {
                         if let Some(sp) = ServiceProvider::from_i32(r.service_provider_id) {
                             Ok((
                                 RewardKey {
-                                    key: sp.as_str_name().to_string(),
+                                    key: service_provider_to_entity_key(sp)?,
                                     reward_type: RewardType::MobileServiceProvider,
                                 },
                                 r.amount,
@@ -204,3 +204,9 @@ impl Indexer {
         }
     }
 }
+
+fn service_provider_to_entity_key(sp: ServiceProvider) -> anyhow::Result<String> {
+    match sp {
+        ServiceProvider::HeliumMobile => Ok("Helium Mobile".to_string()),
+    }
+}