Rename indexes feat

Author: vgvoleg

ydb/_session_impl.py

@@ -268,6 +268,7 @@ def alter_table_request_factory(
     alter_partitioning_settings,
     set_key_bloom_filter,
     set_read_replicas_settings,
+    rename_indexes,
 ):
     request = session_state.attach_request(_apis.ydb_table.AlterTableRequest(path=path))
     if add_columns is not None:
@@ -316,6 +317,10 @@ def alter_table_request_factory(
     if set_read_replicas_settings is not None:
         request.set_read_replicas_settings.MergeFrom(set_read_replicas_settings.to_pb())
 
+    if rename_indexes is not None:
+        for rename_index in rename_indexes:
+            request.rename_indexes.add().MergeFrom(rename_index.to_pb)
+
     return request
 
 

ydb/aio/table.py

@@ -105,6 +105,7 @@ async def alter_table(
         alter_partitioning_settings=None,
         set_key_bloom_filter=None,
         set_read_replicas_settings=None,
+        rename_indexes=None,
     ):  # pylint: disable=W0236,R0913,R0914
         return await super().alter_table(
             path,
@@ -123,6 +124,7 @@ async def alter_table(
             alter_partitioning_settings,
             set_key_bloom_filter,
             set_read_replicas_settings,
+            rename_indexes,
         )
 
     def transaction(self, tx_mode=None, *, allow_split_transactions=None):
@@ -250,6 +252,7 @@ async def alter_table(
         alter_partitioning_settings: Optional["ydb.PartitioningSettings"] = None,
         set_key_bloom_filter: Optional["ydb.FeatureFlag"] = None,
         set_read_replicas_settings: Optional["ydb.ReadReplicasSettings"] = None,
+        rename_indexes: Optional[List["ydb.RenameIndexItem"]] = None,
     ) -> "ydb.Operation":
         """
         Alter a YDB table.
@@ -269,6 +272,7 @@ async def alter_table(
         :param set_compaction_policy: Compaction policy
         :param alter_partitioning_settings: ydb.PartitioningSettings to alter
         :param set_key_bloom_filter: ydb.FeatureFlag to set key bloom filter
+        :param rename_indexes: List of ydb.RenameIndexItem to rename
 
         :return: Operation or YDB error otherwise.
         """
@@ -293,6 +297,7 @@ async def callee(session: Session):
                 alter_partitioning_settings=alter_partitioning_settings,
                 set_key_bloom_filter=set_key_bloom_filter,
                 set_read_replicas_settings=set_read_replicas_settings,
+                rename_indexes=rename_indexes,
             )
 
         return await self._pool.retry_operation(callee)

ydb/table.py

@@ -327,6 +327,20 @@ def to_pb(self):
         return self._pb
 
 
+class RenameIndexItem:
+    def __init__(self, source_name: str, destination_name: str, replace_destination: bool = False):
+        self._source_name = source_name
+        self._destination_name = destination_name
+        self._replace_destination = replace_destination
+
+    def to_pb(self):
+        return _apis.ydb_table.RenameIndexItem(
+            source_name=self._source_name,
+            destination_name=self._destination_name,
+            replace_destination=self._replace_destination,
+        )
+
+
 class ReplicationPolicy(object):
     def __init__(self):
         self._pb = _apis.ydb_table.ReplicationPolicy()
@@ -1124,6 +1138,7 @@ def alter_table(
         alter_partitioning_settings=None,
         set_key_bloom_filter=None,
         set_read_replicas_settings=None,
+        rename_indexes=None,
     ):
         pass
 
@@ -1321,6 +1336,7 @@ def alter_table(
         alter_partitioning_settings: Optional["ydb.PartitioningSettings"] = None,
         set_key_bloom_filter: Optional["ydb.FeatureFlag"] = None,
         set_read_replicas_settings: Optional["ydb.ReadReplicasSettings"] = None,
+        rename_indexes: Optional[List["ydb.RenameIndexItem"]] = None,
     ) -> "ydb.Operation":
         """
         Alter a YDB table.
@@ -1340,6 +1356,7 @@ def alter_table(
         :param set_compaction_policy: Compaction policy
         :param alter_partitioning_settings: ydb.PartitioningSettings to alter
         :param set_key_bloom_filter: ydb.FeatureFlag to set key bloom filter
+        :param rename_indexes: List of ydb.RenameIndexItem to rename
 
         :return: Operation or YDB error otherwise.
         """
@@ -1364,6 +1381,7 @@ def callee(session: Session):
                 alter_partitioning_settings=alter_partitioning_settings,
                 set_key_bloom_filter=set_key_bloom_filter,
                 set_read_replicas_settings=set_read_replicas_settings,
+                rename_indexes=rename_indexes,
             )
 
         return self._pool.retry_operation_sync(callee)
@@ -1857,6 +1875,7 @@ def alter_table(
         alter_partitioning_settings=None,
         set_key_bloom_filter=None,
         set_read_replicas_settings=None,
+        rename_indexes=None,
     ):
         return self._driver(
             _session_impl.alter_table_request_factory(
@@ -1876,6 +1895,7 @@ def alter_table(
                 alter_partitioning_settings,
                 set_key_bloom_filter,
                 set_read_replicas_settings,
+                rename_indexes,
             ),
             _apis.TableService.Stub,
             _apis.TableService.AlterTable,
@@ -2088,6 +2108,7 @@ def async_alter_table(
         alter_partitioning_settings=None,
         set_key_bloom_filter=None,
         set_read_replicas_settings=None,
+        rename_indexes=None,
     ):
         return self._driver.future(
             _session_impl.alter_table_request_factory(
@@ -2107,6 +2128,7 @@ def async_alter_table(
                 alter_partitioning_settings,
                 set_key_bloom_filter,
                 set_read_replicas_settings,
+                rename_indexes,
             ),
             _apis.TableService.Stub,
             _apis.TableService.AlterTable,