added workload log run parse param validation for PK mode

Author: Emgariko

ydb/library/workload/abstract/workload_query_generator.h

@@ -6,6 +6,7 @@
 #include <ydb/public/sdk/cpp/include/ydb-cpp-sdk/client/value/value.h>
 #include <ydb/library/accessor/accessor.h>
 #include <library/cpp/getopt/last_getopt.h>
+#include <ydb/public/lib/ydb_cli/common/command.h>
 
 #include <list>
 #include <string>
@@ -172,6 +173,7 @@ class TWorkloadParams {
     virtual ~TWorkloadParams() = default;
     virtual void ConfigureOpts(NLastGetopt::TOpts& /*opts*/, const ECommandType /*commandType*/, int /*workloadType*/) {
     };
+    virtual void Parse(NYdb::NConsoleClient::TClientCommand::TConfig& /*config*/) {};
     virtual THolder<IWorkloadQueryGenerator> CreateGenerator() const = 0;
     virtual TWorkloadDataInitializer::TList CreateDataInitializers() const {
         return {};

ydb/library/workload/log/log.cpp

@@ -429,6 +429,7 @@ void TLogWorkloadParams::ConfigureOpts(NLastGetopt::TOpts& opts, const ECommandT
                 // .DefaultValue(TimestampStandardDeviationMinutes)
                 .Optional()
                 .StoreResult(&TimestampStandardDeviationMinutes);
+            // TODO: maybe it shoudn't be optional
 
             opts.AddLongOption("date-from", "Left boundary of the interval to generate "
                 "timestamp uniformly from specified interval. Presents as seconds since epoch. Once this option passed, 'date-to' "
@@ -472,6 +473,21 @@ void TLogWorkloadParams::ConfigureOpts(NLastGetopt::TOpts& opts, const ECommandT
     }
 }
 
+void TLogWorkloadParams::Parse(NYdb::NConsoleClient::TClientCommand::TConfig& config) {
+    // TODO: should I check if it's a specific command type: run ?
+    auto timestamp_dev_passed = config.ParseResult->Has("timestamp_deviation");
+    auto date_from_passed = config.ParseResult->Has("date-from");
+    auto date_to_passed = config.ParseResult->Has("date-to");
+
+    if (timestamp_dev_passed && (date_from_passed || date_to_passed)) {
+        throw yexception() << "timestamp_deviation and date_from, date_to are mutually exclusive and shouldn't be passed at once";
+    }
+
+    if ((date_from_passed && !date_to_passed) || (!date_from_passed && date_to_passed)) {
+        throw yexception() << "The `date_from` and `date_to` parameters must be provided together to specify the interval for uniform PK generation";
+    }
+}
+
 THolder<IWorkloadQueryGenerator> TLogWorkloadParams::CreateGenerator() const {
     return MakeHolder<TLogGenerator>(this);
 }

ydb/library/workload/log/log.h

@@ -14,6 +14,7 @@ class TLogWorkloadParams : public TWorkloadParams {
     };
 
     void ConfigureOpts(NLastGetopt::TOpts& opts, const ECommandType commandType, int workloadType) override;
+    void Parse(NYdb::NConsoleClient::TClientCommand::TConfig& config) override;
     THolder<IWorkloadQueryGenerator> CreateGenerator() const override;
     TString GetWorkloadName() const override;
     ui64 MinPartitions = 40;

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

@@ -374,6 +374,10 @@ void TWorkloadCommandBase::Config(TConfig& config) {
     Params.ConfigureOpts(config.Opts->GetOpts(), CommandType, Type);
 }
 
+void TWorkloadCommandBase::Parse(TConfig& config) {
+    Params.Parse(config);
+}
+
 int TWorkloadCommandBase::Run(TConfig& config) {
     Driver = MakeHolder<NYdb::TDriver>(CreateDriver(config));
     if (!DryRun) {

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

@@ -89,6 +89,7 @@ class TWorkloadCommandBase: public TYdbCommand {
         const TString& description = TString(),
         int type = 0);
     virtual void Config(TConfig& config) override;
+    virtual void Parse(TConfig& config) override;
     virtual int Run(TConfig& config) override final;
 
 protected: