Skip to main content

Class RecordRetriever

class gpudb_multihead_io.RecordRetriever(gpudb, table_name, record_type: GPUdbRecordType, workers=None, is_table_replicated=False)

Retrieves records from all worker ranks directly. If multi-head retrieval is not set up, then automatically retrieves records from the head node.

Initializes the RecordRetriever instance.

Parameters

set_logger_level(log_level)

Set the log level for the GPUdb multi-head I/O module.

Parameters

get_records_by_key(key_values, expression=, options=None)

Fetches the record(s) from the appropriate worker rank directly (or, if multi-head record retrieval is not set up, then from the head node) that map to the given shard key.

Parameters

Returns