YT-22250: Fill tableMountInfo.UpperCapBound for chaos queues like it's done for replicated tables

commit_hash:5c2672609977be17b7392405cae5ab086d728dee

Author: olegsidorkin

yt/yt/client/api/rpc_proxy/table_mount_cache.cpp

@@ -117,7 +117,11 @@ class TTableMountCache
                     tableInfo->UpperCapBound = MaxKey();
                 } else {
                     tableInfo->LowerCapBound = MakeUnversionedOwningRow(static_cast<int>(0));
-                    tableInfo->UpperCapBound = MakeUnversionedOwningRow(static_cast<int>(tableInfo->Tablets.size()));
+
+                    auto tabletCount = tableInfo->IsChaosReplicated()
+                        ? rsp->tablet_count()
+                        : static_cast<int>(tableInfo->Tablets.size());
+                    tableInfo->UpperCapBound = MakeUnversionedOwningRow(tabletCount);
                 }
 
                 YT_LOG_DEBUG("Table mount info received (Path: %v, TableId: %v, TabletCount: %v, Dynamic: %v)",

yt/yt/client/driver/driver.cpp

@@ -230,6 +230,7 @@ class TDriver
         REGISTER    (TGetTablePivotKeysCommand,            "get_table_pivot_keys",            Null,       Structured, false, false, ApiVersion4);
         REGISTER    (TGetTabletInfosCommand,               "get_tablet_infos",                Null,       Structured, true,  false, ApiVersion4);
         REGISTER    (TGetTabletErrorsCommand,              "get_tablet_errors",               Null,       Structured, true,  false, ApiVersion4);
+        REGISTER    (TGetTableMountInfoCommand,             "get_table_mount_info",           Null,       Structured, false, false, ApiVersion4);
 
         REGISTER    (TCreateTableBackupCommand,            "create_table_backup",             Null,       Null,       true,  false, ApiVersion3);
         REGISTER    (TRestoreTableBackupCommand,           "restore_table_backup",            Null,       Null,       true,  false, ApiVersion3);

yt/yt/client/driver/internal_commands.cpp

@@ -4,6 +4,8 @@
 
 #include <yt/yt/client/chunk_client/config.h>
 
+#include <yt/yt/client/tablet_client/table_mount_cache.h>
+
 #include <yt/yt/core/ytree/fluent.h>
 
 namespace NYT::NDriver {

yt/yt/client/driver/table_commands.cpp

@@ -1814,4 +1814,35 @@ void TGetTabletErrorsCommand::DoExecute(ICommandContextPtr context)
 
 ////////////////////////////////////////////////////////////////////////////////
 
+void TGetTableMountInfoCommand::Register(TRegistrar registrar)
+{
+    registrar.Parameter("path", &TThis::Path_);
+}
+
+void TGetTableMountInfoCommand::DoExecute(ICommandContextPtr context)
+{
+    auto tableMountCache = context->GetClient()->GetTableMountCache();
+    auto tableInfo = WaitFor(tableMountCache->GetTableInfo(Path_))
+        .ValueOrThrow();
+
+    // Rudimentary, for tests only
+    context->ProduceOutputValue(BuildYsonStringFluently()
+        .BeginMap()
+            .Item("lower_cap_bound").Value(tableInfo->LowerCapBound)
+            .Item("upper_cap_bound").Value(tableInfo->UpperCapBound)
+            .Item("primary_revision").Value(tableInfo->PrimaryRevision)
+            .Item("secondary_revision").Value(tableInfo->SecondaryRevision)
+            .Item("schemas")
+                .DoMap([&tableInfo] (auto fluent) {
+                    for (auto kind : TEnumTraits<ETableSchemaKind>::GetDomainValues()) {
+                        if (auto schemaPtr = tableInfo->Schemas[kind]) {
+                            fluent.Item(ToString(kind)).Value(schemaPtr);
+                        }
+                    }
+                })
+        .EndMap());
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
 } // namespace NYT::NDriver

yt/yt/client/driver/table_commands.h

@@ -606,4 +606,23 @@ class TRestoreTableBackupCommand
 
 ////////////////////////////////////////////////////////////////////////////////
 
+struct TGetTableMountInfoCommandOptions
+{ };
+
+class TGetTableMountInfoCommand
+    : public TTypedCommand<TGetTableMountInfoCommandOptions>
+{
+public:
+    REGISTER_YSON_STRUCT_LITE(TGetTableMountInfoCommand);
+
+    static void Register(TRegistrar registrar);
+
+private:
+    NYTree::TYPath Path_;
+
+    void DoExecute(ICommandContextPtr context) override;
+};
+
+////////////////////////////////////////////////////////////////////////////////
+
 } // namespace NYT::NDriver

yt/yt_proto/yt/client/api/rpc_proxy/proto/api_service.proto

@@ -1465,6 +1465,7 @@ message TRspGetTableMountInfo
     repeated TReplicaInfo replicas = 6;
     optional string physical_path = 7;
     repeated TIndexInfo indices = 8;
+    optional int32 tablet_count = 9;
 }
 
 ////////////////////////////////////////////////////////////////////////////////