return PQRB tablet in ui (#14056)

ydb-platform · Feb 4, 2025 · 3defd52 · 3defd52
1 parent 0d1aa34
commit 3defd52
Show file tree

Hide file tree

Showing 3 changed files with 71 additions and 0 deletions.
diff --git a/ydb/core/viewer/tests/canondata/result.json b/ydb/core/viewer/tests/canondata/result.json
@@ -3,6 +3,48 @@
         "status_code": 200,
         "text": "1\n"
     },
+    "test.test_pqrb_tablet": {
+        "response_create_topic": {
+            "version": "not-zero-number"
+        },
+        "response_tablet_info": {
+            "ResponseTime": "not-zero-number-text",
+            "TabletStateInfo": [
+                {
+                    "ChangeTime": "not-zero-number-text",
+                    "FollowerId": 0,
+                    "Generation": 1,
+                    "HiveId": "not-zero-number-text",
+                    "Leader": true,
+                    "NodeId": "not-zero-number",
+                    "Overall": "Green",
+                    "State": "Active",
+                    "TabletId": "not-zero-number-text",
+                    "TenantId": {
+                        "PathId": "not-zero-number-text",
+                        "SchemeShard": "not-zero-number-text"
+                    },
+                    "Type": "PersQueue"
+                },
+                {
+                    "ChangeTime": "not-zero-number-text",
+                    "FollowerId": 0,
+                    "Generation": 1,
+                    "HiveId": "not-zero-number-text",
+                    "Leader": true,
+                    "NodeId": "not-zero-number",
+                    "Overall": "Green",
+                    "State": "Active",
+                    "TabletId": "not-zero-number-text",
+                    "TenantId": {
+                        "PathId": "not-zero-number-text",
+                        "SchemeShard": "not-zero-number-text"
+                    },
+                    "Type": "PersQueueReadBalancer"
+                }
+            ]
+        }
+    },
     "test.test_storage_groups": {
         "FieldsAvailable": "111111111111111111111",
         "FieldsRequired": "111111111111111111111",

diff --git a/ydb/core/viewer/tests/test.py b/ydb/core/viewer/tests/test.py
@@ -517,3 +517,29 @@ def test_viewer_query_issue_13945():
         'query': 'SELECT AsList();',
         'schema': 'multi'
     })
+
+
+def test_pqrb_tablet():
+    response_create_topic = call_viewer("/viewer/query", {
+        'database': dedicated_db,
+        'query': 'CREATE TOPIC topic1(CONSUMER consumer1)',
+        'schema': 'multi'
+    })
+    response_tablet_info = call_viewer("/viewer/tabletinfo", {
+        'database': dedicated_db,
+        'path': dedicated_db + '/topic1',
+        'enums': 'true'
+    })
+    result = {
+        'response_create_topic': response_create_topic,
+        'response_tablet_info': response_tablet_info,
+    }
+    return replace_values_by_key(result, ['version',
+                                          'ResponseTime',
+                                          'ChangeTime',
+                                          'HiveId',
+                                          'NodeId',
+                                          'TabletId',
+                                          'PathId',
+                                          'SchemeShard'
+                                          ])
diff --git a/ydb/core/viewer/viewer_tabletinfo.h b/ydb/core/viewer/viewer_tabletinfo.h
@@ -261,6 +261,9 @@ class TJsonTabletInfo : public TJsonWhiteboardRequest<TEvWhiteboard::TEvTabletSt
             for (const auto& partition : pathDescription.GetPersQueueGroup().GetPartitions()) {
                 Tablets[partition.GetTabletId()] = NKikimrTabletBase::TTabletTypes::PersQueue;
             }
+            if (pathDescription.HasPersQueueGroup()) {
+                Tablets[pathDescription.GetPersQueueGroup().GetBalancerTabletID()] = NKikimrTabletBase::TTabletTypes::PersQueueReadBalancer;
+            }
             if (pathDescription.HasRtmrVolumeDescription()) {
                 for (const auto& partition : pathDescription.GetRtmrVolumeDescription().GetPartitions()) {
                     Tablets[partition.GetTabletId()] = NKikimrTabletBase::TTabletTypes::RTMRPartition;