benchmark_config.py 817 B

12345678910111213141516171819202122232425
  1. # Copyright Materialize, Inc. and contributors. All rights reserved.
  2. #
  3. # Use of this software is governed by the Business Source License
  4. # included in the LICENSE file at the root of this repository.
  5. #
  6. # As of the Change Date specified in that file, in accordance with
  7. # the Business Source License, use of this software will be governed
  8. # by the Apache License, Version 2.0.
  9. from dataclasses import dataclass
  10. from math import floor, sqrt
  11. from materialize.scalability.workload.workload import Workload
  12. @dataclass
  13. class BenchmarkConfiguration:
  14. workload_classes: list[type[Workload]]
  15. exponent_base: float
  16. min_concurrency: int
  17. max_concurrency: int
  18. count: int
  19. verbose: bool
  20. def get_count_for_concurrency(self, concurrency: int) -> int:
  21. return floor(self.count * sqrt(concurrency))