Function main

sdk/python/feast/infra/utils/hbase_utils.py:179–217

()

def main():
    from feast.infra.key_encoding_utils import serialize_entity_key
    from feast.protos.feast.types.EntityKey_pb2 import EntityKey
    from feast.protos.feast.types.Value_pb2 import Value

    pool = ConnectionPool(
        host="localhost",
        port=9090,
        size=2,
    )
    with pool.connection() as connection:
        table = connection.table("test_hbase_driver_hourly_stats")
        row_keys = [
            serialize_entity_key(
                EntityKey(
                    join_keys=["driver_id"], entity_values=[Value(int64_val=1004)]
                ),
                entity_key_serialization_version=3,
            ).hex(),
            serialize_entity_key(
                EntityKey(
                    join_keys=["driver_id"], entity_values=[Value(int64_val=1005)]
                ),
                entity_key_serialization_version=3,
            ).hex(),
            serialize_entity_key(
                EntityKey(
                    join_keys=["driver_id"], entity_values=[Value(int64_val=1024)]
                ),
                entity_key_serialization_version=3,
            ).hex(),
        ]
        rows = table.rows(row_keys)

        for _, row in rows:
            for key, value in row.items():
                col_name = bytes.decode(key, "utf-8").split(":")[1]
                print(col_name, value)
            print()


if __name__ == "__main__":