Skip to content

Commit 068f848

Browse files
committed
Partition autosplit feature
1 parent 228bb52 commit 068f848

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

ydb/_grpc/grpcwrapper/ydb_topic.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -696,6 +696,20 @@ def to_proto(self) -> ydb_topic_pb2.StreamReadMessage.StopPartitionSessionRespon
696696
partition_session_id=self.partition_session_id,
697697
)
698698

699+
@dataclass
700+
class EndPartitionSession(IFromProto):
701+
partition_session_id: int
702+
adjacent_partition_ids: List[int]
703+
child_partition_ids: List[int]
704+
705+
@staticmethod
706+
def from_proto(msg: ydb_topic_pb2.StreamReadMessage.EndPartitionSession):
707+
return StreamReadMessage.EndPartitionSession(
708+
partition_session_id=msg.partition_session_id,
709+
adjacent_partition_ids=list(msg.adjacent_partition_ids),
710+
child_partition_ids=list(msg.child_partition_ids),
711+
)
712+
699713
@dataclass
700714
class FromClient(IToProto):
701715
client_message: "ReaderMessagesFromClientToServer"

0 commit comments

Comments
 (0)