Simplify workload topic init

Author: nshestakov

ydb/public/lib/ydb_cli/commands/topic_operations_scenario.cpp

@@ -54,6 +54,11 @@ TString TTopicOperationsScenario::GetWriteOnlyTableName() const
     return TableName;
 }
 
+ui32 TTopicOperationsScenario::GetTopicMaxPartitionCount() const
+{
+    return TopicMaxPartitionCount >= TopicPartitionCount ? TopicMaxPartitionCount : (TopicPartitionCount << 3);
+}
+
 THolder<TLogBackend> TTopicOperationsScenario::MakeLogBackend(TConfig::EVerbosityLevel level)
 {
     return CreateLogBackend("cerr",

ydb/public/lib/ydb_cli/commands/topic_operations_scenario.h

@@ -45,6 +45,8 @@ class TTopicOperationsScenario {
     TString GetReadOnlyTableName() const;
     TString GetWriteOnlyTableName() const;
 
+    ui32 GetTopicMaxPartitionCount() const;
+
     TDuration TotalSec;
     TDuration WindowSec;
     TDuration WarmupSec;
@@ -54,7 +56,7 @@ class TTopicOperationsScenario {
     TString TopicName;
     ui32 TopicPartitionCount = 1;
     bool TopicAutoscaling = false;
-    ui32 TopicMaxPartitionCount = 100;
+    ui32 TopicMaxPartitionCount = 1024;
     ui32 StabilizationWindowSeconds = 15;
     ui32 UpUtilizationPercent = 90;
     ui32 DownUtilizationPercent = 30;

ydb/public/lib/ydb_cli/commands/topic_workload/topic_workload_init.cpp

@@ -10,7 +10,7 @@ using namespace NYdb::NConsoleClient;
 
 int TCommandWorkloadTopicInit::TScenario::DoRun(const TConfig& config)
 {
-    CreateTopic(config.Database, TopicName, TopicPartitionCount, ConsumerCount, TopicAutoscaling, TopicMaxPartitionCount, StabilizationWindowSeconds, UpUtilizationPercent, DownUtilizationPercent);
+    CreateTopic(config.Database, TopicName, TopicPartitionCount, ConsumerCount, TopicAutoscaling, GetTopicMaxPartitionCount(), StabilizationWindowSeconds, UpUtilizationPercent, DownUtilizationPercent);
 
     return EXIT_SUCCESS;
 }
@@ -45,8 +45,10 @@ void TCommandWorkloadTopicInit::Config(TConfig& config)
         .DefaultValue(false)
         .StoreTrue(&Scenario.TopicAutoscaling);
     config.Opts->AddLongOption('m', "auto-partitioning-max-partitions-count", "Max number of partitions in the topic.")
+        .DefaultValue(1024)
         .StoreResult(&Scenario.TopicMaxPartitionCount);
     config.Opts->AddLongOption("auto-partitioning-stabilization-window-seconds", "Duration in seconds of high or low load before automatically scale the number of partitions")
+        .DefaultValue(15)
         .Optional()
         .StoreResult(&Scenario.StabilizationWindowSeconds);
     config.Opts->AddLongOption("auto-partitioning-up-utilization-percent", "The load percentage at which the number of partitions will increase")