Generating load using repeat_fn.

hydro-project · Dec 16, 2024 · 77f3558 · 77f3558
1 parent 87ee9da
commit 77f3558
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 11 deletions.
diff --git a/datastores/gossip_kv/kv/server.rs b/datastores/gossip_kv/kv/server.rs
@@ -148,6 +148,14 @@ where
             })
             -> writes;
 
+        simulated_puts = repeat_fn(2000, || {
+            upsert_row(Clock::new(100), rand::random(), "value".to_string())
+        })
+            -> inspect (|_| {
+                SETS_COUNTER.inc();
+            })
+            -> writes;
+
         client_in[Delete]
             -> null();
             // -> inspect(|req| trace!("{:?}: Received Delete request: {:?}.", context.current_tick(), req))

diff --git a/datastores/gossip_kv/load_test_server/server.rs b/datastores/gossip_kv/load_test_server/server.rs
@@ -84,17 +84,17 @@ fn run_server(
             // });
 
             let local = task::LocalSet::new();
-            local.spawn_local(async move {
-                // let key_master: Key = "/usr/table/key".parse().unwrap();
-                loop {
-                    let request = ClientRequest::Set {
-                        key: 100,
-                        value: "FOOBAR".to_string(),
-                    };
-                    client_input_tx.send((request, UNKNOWN_ADDRESS)).await.unwrap();
-                    SETS_SENT.inc();
-                }
-            });
+            // local.spawn_local(async move {
+            //     // let key_master: Key = "/usr/table/key".parse().unwrap();
+            //     loop {
+            //         let request = ClientRequest::Set {
+            //             key: 100,
+            //             value: "FOOBAR".to_string(),
+            //         };
+            //         client_input_tx.send((request, UNKNOWN_ADDRESS)).await.unwrap();
+            //         SETS_SENT.inc();
+            //     }
+            // });
 
 
             // Networking