single user mode

private-attribution · Oct 31, 2023 · cfd1f21 · cfd1f21
1 parent 545bce5
commit cfd1f21
Show file tree

Hide file tree

Showing 3 changed files with 154 additions and 19 deletions.
diff --git a/benches/oneshot/ipa.rs b/benches/oneshot/ipa.rs
@@ -115,16 +115,10 @@ async fn run(args: Args) -> Result<(), Error> {
     // Note that this is a temporary solution. If the max number of rows become too
     // large, we need a better way to generate the steps, likely by synthesizing the
     // steps in the `Step` derive macro.
-    let (max_number_of_users, query_size) = if cfg!(feature = "step-trace") && args.oprf {
-        (
-            NonZeroU32::new(1).unwrap(),
-            usize::try_from(args.records_per_user).unwrap(),
-        )
+    let (single_user_mode, query_size) = if cfg!(feature = "step-trace") && args.oprf {
+        (true, usize::try_from(args.records_per_user).unwrap())
     } else {
-        (
-            EventGeneratorConfig::default().max_number_of_users,
-            args.query_size,
-        )
+        (false, args.query_size)
     };
 
     let seed = args.random_seed.unwrap_or_else(|| random());
@@ -139,7 +133,7 @@ async fn run(args: Args) -> Result<(), Error> {
             max_trigger_value: NonZeroU32::try_from(args.max_trigger_value).unwrap(),
             max_breakdown_key: NonZeroU32::try_from(args.breakdown_keys).unwrap(),
             max_events_per_user: NonZeroU32::try_from(args.records_per_user).unwrap(),
-            max_number_of_users,
+            single_user_mode,
             ..Default::default()
         },
     )