本文介绍如何在EMR on ECS集群环境中安装 PyIceberg,并配置其通过 Iceberg REST 协议访问DLF。
环境准备与安装
安装 pyiceberg-dlf
pyiceberg-dlf 是适配 DLF 的 PyIceberg 发行版,已发布到 PyPI,包含 REST sigv4 修复、vended OSS 存储凭证以及凭证到期前自动刷新能力。
环境要求:Python 3.10 及以上版本。
python3 -m venv venv
source venv/bin/activate
pip install -U pip
# 卸载官方版(与 DLF 适配版不能共存)
pip uninstall -y pyiceberg
# rest-sigv4 为必选项(安装 boto3,用于 REST sigv4 签名)
pip install "pyiceberg-dlf[rest-sigv4,pyarrow,pandas]"包名与 import 名不同:包名为
pyiceberg-dlf,import 名仍为pyiceberg。DLF 适配版与官方 pyiceberg 不能共存:
pyiceberg-dlf与官方pyiceberg都提供import pyiceberg,后装的会覆盖前者。建议使用独立虚拟环境,或安装前执行pip uninstall -y pyiceberg。rest-sigv4 不可省略:该 extra 会安装 boto3 等签名依赖,缺少时
load_catalog()将报ModuleNotFoundError: No module named 'boto3'。pandas 按需添加:示例中使用了
scan.to_pandas(),因此需带上pandasextra。pyarrow 须低于 22:
pyiceberg-dlf[pyarrow]已将 pyarrow 锁定到 21.x。pyarrow 22 及以上版本上传 OSS 时会报aws-chunked encoding is not supported。
代码示例
以下 Python 脚本演示了完整的生命周期操作:连接 Catalog、创建表、写入数据、读取数据以及删除表。
请将脚本中的占位符替换为实际配置。
${regionId}:DLF域名,如cn-hangzhou。详情请参见Iceberg REST服务接入点。${catalogName}:DLF Catalog 名称。${accessKey}:AccessKey ID。${accessKeySecret}:AccessKey Secret。
import pyarrow as pa
from pyiceberg.catalog import load_catalog
from pyiceberg.exceptions import TableAlreadyExistsError, NoSuchTableError
from pyiceberg.io.pyarrow import schema_to_pyarrow
from pyiceberg.partitioning import PartitionField, PartitionSpec
from pyiceberg.schema import Schema
from pyiceberg.transforms import IdentityTransform
from pyiceberg.types import LongType, NestedField
# 1. 配置与连接 Catalog
catalog = load_catalog(
"default",
**{
"type": "rest",
"uri": "http://${regionId}-vpc.dlf.aliyuncs.com/iceberg",
"warehouse": "${catalogName}",
"rest.signing-name": "DlfNext",
"rest.signing-region": "${regionId}",
"rest.sigv4-enabled": "true",
"client.access-key-id": "${accessKeyId}",
"client.secret-access-key": "${accessKeySecret}",
"client.region": "${regionId}",
},
)
# ---------------------------------------------------
# 2. 定义表结构与元数据
# ---------------------------------------------------
TEST_TABLE_SCHEMA = Schema(
NestedField(1, "x", LongType(), required=True),
NestedField(2, "y", LongType(), doc="comment", required=True),
NestedField(3, "z", LongType(), required=True),
)
TEST_TABLE_IDENTIFIER = ("default", "my_table")
TEST_TABLE_PARTITION_SPEC = PartitionSpec(
PartitionField(name="x", transform=IdentityTransform(), source_id=1, field_id=1000)
)
TEST_TABLE_PROPERTIES = {"read.split.target.size": "134217728"} # 128MB
# ---------------------------------------------------
# 3. 执行测试流程
# ---------------------------------------------------
# 清理旧表(如果存在)
try:
catalog.drop_table(identifier=TEST_TABLE_IDENTIFIER)
print("Existing table dropped.")
except NoSuchTableError:
print("No existing table to drop.")
# 创建新表
try:
catalog.create_table(
identifier=TEST_TABLE_IDENTIFIER,
schema=TEST_TABLE_SCHEMA,
partition_spec=TEST_TABLE_PARTITION_SPEC,
properties=TEST_TABLE_PROPERTIES,
)
print("Table created.")
except TableAlreadyExistsError:
print("Table already exists, will append data.")
# 加载表
table = catalog.load_table(identifier=TEST_TABLE_IDENTIFIER)
print(f"Loaded table: {table}")
# 构造 PyArrow 数据
arrow_schema = schema_to_pyarrow(table.schema())
data = pa.Table.from_pydict(
{
"x": [1, 2, 3],
"y": [10, 20, 30],
"z": [100, 200, 300],
},
schema=arrow_schema,
)
# 写入数据
print(f"Inserting {data.num_rows} rows...")
table.append(data)
print("Insert finished.")
# 读取并展示数据
scan = table.scan()
df = scan.to_pandas()
print("First 10 rows via to_pandas():")
print(df.head(10))
# 清理测试表
try:
catalog.drop_table(identifier=TEST_TABLE_IDENTIFIER)
print("Table dropped after test.")
except NoSuchTableError:
print("Table already dropped.")运行结果展示:
(myenv) root@iZbp1h4zr65vjcvz9w3080Z:~/workspace# python test.py
Existing table dropped.
Table created.
Loaded table: my_table(
1: x: required long,
2: y: required long (comment),
3: z: required long
),
partition by: [x],
sort order: [],
snapshot: null
Inserting 3 rows...
Insert finished.
First 10 rows via to_pandas():
x y z
0 1 10 100
1 2 20 200
2 3 30 300
Table dropped after test.
(myenv) root@iZbp1h4zr65vjcvz9w3080Z:~/workspace# pip list|grep -E "pyarrow|pyiceberg|boto3|pandas"
boto3 1.42.15
pandas 2.3.3
pyarrow 19.0.0
pyiceberg 0.10.0.dev0
(myenv) root@iZbp1h4zr65vjcvz9w3080Z:~/workspace#