add fake commands

bevyengine · Jun 14, 2021 · bbcdc6f · bbcdc6f
1 parent 5565020
commit bbcdc6f
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/benches/benches/bevy_ecs/commands.rs b/benches/benches/bevy_ecs/commands.rs
@@ -72,9 +72,17 @@ fn spawn_commands(criterion: &mut Criterion) {
     group.finish();
 }
 
-struct FakeCommand;
+struct FakeCommandA;
+struct FakeCommandB(u64);
 
-impl Command for FakeCommand {
+impl Command for FakeCommandA {
+    fn write(self: Box<Self>, world: &mut World) {
+        black_box(self);
+        black_box(world);
+    }
+}
+
+impl Command for FakeCommandB {
     fn write(self: Box<Self>, world: &mut World) {
         black_box(self);
         black_box(world);
@@ -91,10 +99,17 @@ fn fake_commands(criterion: &mut Criterion) {
             let mut world = World::default();
             let mut command_queue = CommandQueue::default();
 
+            let mut rng = SmallRng::seed_from_u64(42);
+            let mut rng_bool = || rng.next_u32() % 2 == 0;
+
             bencher.iter(|| {
                 let mut commands = Commands::new(&mut command_queue, &world);
                 for _ in 0..command_count {
-                    commands.add(FakeCommand);
+                    if rng_bool() {
+                        commands.add(FakeCommandA);
+                    } else {
+                        commands.add(FakeCommandB(0));
+                    }
                 }
                 drop(commands);
                 command_queue.apply(&mut world);