Query specific store

glean-core/examples/sample.rs

@@ -6,7 +6,7 @@ lazy_static! {
     pub static ref GLOBAL_METRIC: BooleanMetric = BooleanMetric::new(CommonMetricData {
         name: "global_metric".into(),
         category: "global".into(),
-        send_in_pings: vec!["core".into()],
+        send_in_pings: vec!["core".into(), "metrics".into()],
         lifetime: Lifetime::Ping,
         disabled: false,
     });
@@ -26,5 +26,6 @@ fn main() {
     GLOBAL_METRIC.set(true);
     local_metric.set("I can set this");
 
-    println!("{}", storage::StorageManager.dump());
+    println!("Core Data:\n{}", storage::StorageManager.dump("core"));
+    println!("Metrics Data:\n{}", storage::StorageManager.dump("metrics"));
 }

glean-core/src/storage/boolean.rs

@@ -26,7 +26,7 @@ impl BooleanStorageImpl {
 }
 
 impl StorageDump for BooleanStorageImpl {
-    fn dump(&self) -> Option<JsonValue> {
-        self.store.get("core").map(|store| json!(store))
+    fn dump(&self, store_name: &str) -> Option<JsonValue> {
+        self.store.get(store_name).map(|store| json!(store))
     }
 }

glean-core/src/storage/mod.rs

@@ -18,16 +18,16 @@ lazy_static! {
 }
 
 pub trait StorageDump {
-    fn dump(&self) -> Option<JsonValue>;
+    fn dump(&self, store_name: &str) -> Option<JsonValue>;
 }
 
 pub struct StorageManager;
 
 macro_rules! dump_storages {
-    ($(($name:expr, $storage:tt),)+) => {{
+    ($store_name:expr => $(( $name:expr, $storage:tt),)+) => {{
         let data = json!({
             $(
-                $name: $storage.read().unwrap().dump(),
+                $name: $storage.read().unwrap().dump($store_name),
             )+
         });
         data
@@ -35,8 +35,8 @@ macro_rules! dump_storages {
 }
 
 impl StorageManager {
-    pub fn dump(&self) -> String {
-        let data = dump_storages!(("bool", BooleanStorage), ("string", StringStorage),);
+    pub fn dump(&self, store_name: &str) -> String {
+        let data = dump_storages!(store_name => ("bool", BooleanStorage), ("string", StringStorage),);
 
         ::serde_json::to_string_pretty(&data).unwrap()
     }

glean-core/src/storage/string.rs

@@ -26,7 +26,7 @@ impl StringStorageImpl {
 }
 
 impl StorageDump for StringStorageImpl {
-    fn dump(&self) -> Option<JsonValue> {
-        self.store.get("core").map(|store| json!(store))
+    fn dump(&self, store_name: &str) -> Option<JsonValue> {
+        self.store.get(store_name).map(|store| json!(store))
     }
 }