讲师中心 微信公众号
AI工具推荐 视频效率加速

Polars 读取 S3 Parquet 文件的连接复用与性能优化实践

大伟同学_5379

大伟同学_5379

发布时间:2026-02-11 16:46:16

|

392人浏览过

|

来源于php中文网

原创

Polars 读取 S3 Parquet 文件的连接复用与性能优化实践

本文详解在 aws lambda 等无服务器环境中,如何避免 polars 每次调用重复建立 s3 连接,并对比原生 boto3 下载 + polars 本地解析的性能优势,提供可落地的连接池替代方案。

Polars 自 0.20 版本起通过 object_store 库统一管理云存储 I/O,支持直接读取 s3:// 路径的 Parquet 文件(如 pl.read_parquet("s3://bucket/data.parquet"))。但需明确:当前 Polars(截至 v0.21)不支持传入已初始化的 boto3.client 实例进行连接复用。其 storage_options 参数仅接受配置字典(如 {"aws_region": "us-east-1", "aws_access_key_id": "...", ...}),用于每次 I/O 时新建 object_store::aws::Client——这意味着在 Lambda 中,若将 read_parquet 直接写在 handler 内,每次调用(包括温启动)都会触发新的 HTTP 客户端初始化与连接建立,无法复用底层 TCP 连接或凭证缓存。

虽然 Polars 团队已在 PR #14598 中优化了 object_store 的内部客户端缓存逻辑(例如复用 reqwest::Client 实例),但该缓存作用于库内部、不可由用户控制,且实测仍无法匹敌原生 boto3 的连接复用效率。根据真实 Lambda 场景压测(见 issue #14572 讨论),直接使用全局 boto3.client("s3") 下载对象至内存/临时磁盘,再交由 Polars 解析,平均快 2–5 倍,尤其在小文件高频读取场景下优势显著。

因此,推荐采用以下经过验证的生产级模式:

import boto3
import polars as pl
from io import BytesIO

# ✅ 全局初始化:Lambda 冷启动时创建一次,后续所有 invocations 复用
s3_client = boto3.client("s3", region_name="us-east-1")

def handler(event, context):
    bucket = "my-bucket"
    key = "data/part-00000.parquet"

    # 步骤1:复用 s3_client 下载(自动复用连接池、凭证缓存)
    response = s3_client.get_object(Bucket=bucket, Key=key)
    parquet_bytes = response["Body"].read()

    # 步骤2:Polars 从内存解析(零磁盘 IO)
    df = pl.read_parquet(BytesIO(parquet_bytes))

    return df.shape[0]
⚠️ 注意事项: 若 Parquet 文件较大(>100MB),建议改用 tempfile.NamedTemporaryFile 写入本地 /tmp 后再 pl.read_parquet(),避免内存压力; 确保 Lambda 执行角色拥有对应 S3 s3:GetObject 权限; boto3.client 默认启用连接池(max_pool_connections=10),无需额外配置; 避免在 handler 内重复创建 boto3.client 或 pl.Config,否则失去复用意义。

总结而言,Polars 当前的 S3 集成更适合开发调试或低频批量场景;在 Lambda、Fargate 等对冷启动和吞吐敏感的环境,“boto3 下载 + Polars 本地解析”仍是更可控、更高效的选择。未来若 Polars 开放 client 参数或支持 reqwest::Client 注入,可平滑迁移。在此之前,拥抱成熟稳定的 boto3 生态,是兼顾性能与可靠性的务实之选。

相关文章

数码产品性能查询
数码产品性能查询

该软件包括了市面上所有手机CPU,手机跑分情况,电脑CPU,电脑产品信息等等,方便需要大家查阅数码产品最新情况,了解产品特性,能够进行对比选择最具性价比的商品。

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

热门AI工具

更多
小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

相关专题

更多
lambda表达式
lambda表达式

Lambda表达式是一种匿名函数的简洁表示方式,它可以在需要函数作为参数的地方使用,并提供了一种更简洁、更灵活的编码方式,其语法为“lambda 参数列表: 表达式”,参数列表是函数的参数,可以包含一个或多个参数,用逗号分隔,表达式是函数的执行体,用于定义函数的具体操作。本专题为大家提供lambda表达式相关的文章、下载、课程内容,供大家免费下载体验。

1051

2023.09.15

python lambda函数
python lambda函数

本专题整合了python lambda函数用法详解,阅读专题下面的文章了解更多详细内容。

345

2025.11.08

Python lambda详解
Python lambda详解

本专题整合了Python lambda函数相关教程,阅读下面的文章了解更多详细内容。

574

2026.01.05

C++ Lambda 表达式与函数式编程
C++ Lambda 表达式与函数式编程

深入讲解 C++ Lambda 表达式与函数式编程范式,涵盖 Lambda 基础语法与返回值推导、值捕获与引用捕获的区别与陷阱、初始化捕获(C++14 广义捕获)、泛型 Lambda(auto 参数)、mutable 关键字修改捕获变量、std::function 类型擦除与函数包装、std::bind 参数绑定(及其被 Lambda 替代的趋势)、Lambda 在 STL 算法(sort/transform/for_each/remo

729

2026.05.06

Java 函数式接口与 Lambda 进阶技巧
Java 函数式接口与 Lambda 进阶技巧

在 Java Stream 基础之上进一步深入函数式编程,涵盖 @FunctionalInterface 注解与自定义函数式接口设计、四种方法引用(静态/实例/对象/构造器)的使用场景、Lambda 中的 effectively final 变量捕获规则、函数组合(andThen/compose)与柯里化技巧、Comparator 链式比较器构建、CompletableFuture 异步函数式编排、Reactor / RxJava 响应

428

2026.05.08

http500解决方法
http500解决方法

http500解决方法有检查服务器日志、检查代码错误、检查服务器配置、检查文件和目录权限、检查资源不足、更新软件版本、重启服务器或寻求专业帮助等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

3643

2023.11.09

http请求415错误怎么解决
http请求415错误怎么解决

解决方法:1、检查请求头中的Content-Type;2、检查请求体中的数据格式;3、使用适当的编码格式;4、使用适当的请求方法;5、检查服务器端的支持情况。更多http请求415错误怎么解决的相关内容,可以阅读下面的文章。

894

2023.11.14

HTTP 503错误解决方法
HTTP 503错误解决方法

HTTP 503错误表示服务器暂时无法处理请求。想了解更多http错误代码的相关内容,可以阅读本专题下面的文章。

6586

2024.03.12

Gradle插件开发入门教程
Gradle插件开发入门教程

本专题介绍Gradle插件开发的基本流程,涵盖插件类编写、自定义任务注册、扩展参数设计和插件应用方法,帮助你理解插件如何组织构建逻辑,减少多个项目之间的重复配置。

60

2026.10.22

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Webpack4.x---十天技能课堂
Webpack4.x---十天技能课堂

共20课时 | 2.1万人学习

简单聊聊mysql8与网络通信
简单聊聊mysql8与网络通信

共1课时 | 0.1万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn