Open the metric unit vocabulary

1 files changed, 6 insertions, 13 deletions

diff --git a/crates/fidget-spinner-store-sqlite/src/lib.rs b/crates/fidget-spinner-store-sqlite/src/lib.rs
index fbdbb32..253929e 100644
--- a/crates/fidget-spinner-store-sqlite/src/lib.rs
+++ b/crates/fidget-spinner-store-sqlite/src/lib.rs
@@ -3033,19 +3033,12 @@ fn parse_frontier_status(raw: &str) -> Result<FrontierStatus, rusqlite::Error> {
 }
 
 fn parse_metric_unit(raw: &str) -> Result<MetricUnit, rusqlite::Error> {
-    match raw {
-        "seconds" => Ok(MetricUnit::Seconds),
-        "bytes" => Ok(MetricUnit::Bytes),
-        "count" => Ok(MetricUnit::Count),
-        "ratio" => Ok(MetricUnit::Ratio),
-        "custom" => Ok(MetricUnit::Custom),
-        _ => Err(to_sql_conversion_error(StoreError::Json(
-            serde_json::Error::io(io::Error::new(
-                io::ErrorKind::InvalidData,
-                format!("invalid metric unit `{raw}`"),
-            )),
-        ))),
-    }
+    MetricUnit::new(raw).map_err(|error| {
+        to_sql_conversion_error(StoreError::Json(serde_json::Error::io(io::Error::new(
+            io::ErrorKind::InvalidData,
+            error.to_string(),
+        ))))
+    })
 }
 
 fn parse_optimization_objective(raw: &str) -> Result<OptimizationObjective, rusqlite::Error> {