Struct TablePartitioner 

pub(crate) struct TablePartitioner {
    table_id: TableId,
    rule: PartitionRuleRef,
    versions: HashMap<RegionNumber, Option<PartitionExprVersion>>,
}

A single snapshot of a table’s partition rule and expression versions.

Fields

table_id: TableId

rule: PartitionRuleRef

versions: HashMap<RegionNumber, Option<PartitionExprVersion>>

Implementations

impl TablePartitioner

pub async fn try_new( partition_manager: &PartitionRuleManager, table_info: &TableInfo, ) -> Result<Self>

Reads a table’s partition rule and expression versions into one routing snapshot.

pub(crate) fn rebuild_write_requests( &self, request: RegionRequest, table_info: &TableInfo, ) -> Result<Vec<RegionRequest>>

Rebuilds write requests using this partition snapshot after a route refresh.

Takes ownership so row payloads can move into their new regions. Requests whose region and partition version still match retain their original payloads. The table metadata comes from the same refresh that created this partitioner.

pub fn partition_insert_requests( &self, rows: Rows, skip_wal: bool, ) -> Result<Vec<InsertRequest>>

Moves insert rows into requests for this snapshot’s destination regions.

pub fn partition_delete_requests( &self, rows: Rows, ) -> Result<Vec<DeleteRequest>>

Moves delete rows into requests for this snapshot’s destination regions.

pub fn partition_bulk_insert_requests( &self, batch: RecordBatch, original_ipc: Option<ArrowIpc>, skip_wal: bool, ) -> Result<Vec<BulkInsertRequest>>

Splits an Arrow batch, reusing the original IPC when one region selects all rows.

fn build_bulk_insert_requests( &self, batch: RecordBatch, masks: HashMap<RegionNumber, RegionMask>, original_ipc: Option<ArrowIpc>, skip_wal: bool, ) -> Result<Vec<BulkInsertRequest>>

fn rebuild_insert_requests(&self, inserts: InsertRequests) -> Result<Vec<Body>>

fn rebuild_delete_requests(&self, deletes: DeleteRequests) -> Result<Vec<Body>>

fn rebuild_bulk_insert_request( &self, bulk: BulkInsertRequest, table_info: &TableInfo, ) -> Result<Vec<Body>>

fn restore_partition_tags( &self, batch: &RecordBatch, table_info: &TableInfo, ) -> Result<Option<RecordBatch>>

Restores omitted physical-table tags for routing without changing the write payload.

fn is_partition_expr_unchanged( &self, region_id: RegionId, version: Option<PartitionExprVersion>, ) -> bool

Whether the request’s region still exists with the same partition expression version.

A leader change leaves the partition unchanged, so the original payload can be sent to the new peer. A removed region or changed partition expression version requires rebuilding the request using this snapshot’s partition rule.

fn version(&self, region: RegionNumber) -> Option<PartitionExprVersion>