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

如何高效提取 BigQuery 中单条超大记录(如 1.8MB)

小静酱_5157

小静酱_5157

发布时间:2026-08-01 21:47:06

|

775人浏览过

|

来源于php中文网

原创

如何高效提取 BigQuery 中单条超大记录(如 1.8MB)

当从 BigQuery 查询仅返回一条但体积庞大的记录(如 1.8MB)时,传统 bigQuery.query() 同步拉取方式易导致内存压力与超时风险;推荐改用 BigQuery Storage Read API,通过二进制流式读取提升吞吐、降低延迟并支持精准控制。

当从 bigquery 查询仅返回一条但体积庞大的记录(如 1.8mb)时,传统 `bigquery.query()` 同步拉取方式易导致内存压力与超时风险;推荐改用 bigquery storage read api,通过二进制流式读取提升吞吐、降低延迟并支持精准控制。

对于单条重型记录(heavy record)的场景——例如某一行包含 Base64 编码的大型二进制对象、JSON 嵌套文档或长文本字段——使用标准 QueryJob 拉取整个结果集(即使只有一行)会将全部数据一次性加载至 JVM 堆内存,不仅可能触发 OutOfMemoryError,还因序列化/反序列化开销显著拖慢响应。此时,BigQuery Storage Read API 是更优选择:它绕过传统查询执行引擎,直接从底层存储层以 Avro 或 Arrow 格式流式读取数据,支持分片、并行、按需解码,并天然适配大 payload 场景。

以下为 Java 中使用 Storage Read API 提取单条大记录的核心示例(需添加依赖 com.google.cloud:google-cloud-bigquerystorage:2.40.0+):

import com.google.cloud.bigquery.storage.v1.*;
import com.google.cloud.bigquery.storage.v1.ReadOptions.TableReadOptions;
import com.google.protobuf.ByteString;

// 构建 ReadSession(自动选择最优分区)
ReadSession.Builder sessionBuilder = ReadSession.newBuilder()
    .setTableReadOptions(TableReadOptions.newBuilder()
        .addSelectedFields("id")      // 显式指定所需字段,减少传输量
        .addSelectedFields("payload") // 尤其重要:避免读取无关大字段
        .build())
    .setDataFormat(DataFormat.ARROW) // 推荐 Arrow:零拷贝、高效列式解析
    .setReadOptions(ReadOptions.newBuilder()
        .setUseAvroLogicalTypes(true)
        .build());

// 创建 ReadSession(需指定项目 ID 和表路径)
String tableName = "projects/your-project/datasets/your_dataset/tables/your_table";
ReadSession session = client.createReadSession(
    CreateReadSessionRequest.newBuilder()
        .setParent("projects/your-project")
        .setReadSession(sessionBuilder.build())
        .setMaxStreamCount(1) // 单条记录 → 1 stream 足够
        .build()
);

if (session.getStreamsList().isEmpty()) {
    throw new IllegalStateException("No streams created — check permissions & table existence");
}

// 流式读取第一条消息(即目标大记录)
String streamName = session.getStreamsList().get(0).getName();
ReadRowsRequest request = ReadRowsRequest.newBuilder()
    .setReadStream(streamName)
    .build();

ServerStreamingCallable<ReadRowsRequest, ReadRowsResponse> callable =
    client.getStub().readRowsCallable();

// 使用阻塞流(也可用异步方式)
Iterator<ReadRowsResponse> responseIterator = callable
    .call(request)
    .iterateAll();

if (responseIterator.hasNext()) {
    ReadRowsResponse response = responseIterator.next();
    // Arrow 格式:使用 ArrowReader 解析(需引入 arrow-memory-core)
    ArrowStreamReader reader = new ArrowStreamReader(
        response.getArrowRecordBatch().getData(), 
        new RootAllocator()
    );
    VectorSchemaRoot root = reader.getVectorSchemaRoot();
    // 逐行访问(此处仅处理第 0 行)
    if (root.getRowCount() > 0) {
        Object id = root.getVector("id").getObject(0);
        ByteString payloadBytes = (ByteString) root.getVector("payload").getObject(0);
        // ✅ payloadBytes 可直接转 byte[] 或流式处理,避免全量驻留内存
        byte[] rawPayload = payloadBytes.toByteArray();
        System.out.println("Loaded heavy record, size: " + rawPayload.length + " bytes");
    }
}

⚠️ 关键注意事项:

  • 权限要求:服务账号需具备 roles/bigquery.reader 和 roles/storage.objectViewer(Storage API 所需);
  • 字段裁剪:务必通过 TableReadOptions.selectedFields 限制读取字段,避免传输冗余大数据列;
  • 格式选型:优先选用 DataFormat.ARROW(较 Avro 更省内存、支持零拷贝),若需兼容旧系统再选 Avro;
  • 连接管理:BigQueryWriteClient 和 BigQueryReadClient 均为线程安全且建议复用,避免频繁创建;
  • 错误重试:Storage API 默认不自动重试流中断,建议在 responseIterator 外层封装幂等重连逻辑。

综上,面对“少而重”的查询模式,放弃 bigQuery.query() 的便利性,转向 Storage Read API 并配合字段精简、流式解析与二进制格式,是保障稳定性与性能的工程最佳实践。

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

热门AI工具

更多
二狗PPT
二狗PPT Hot

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

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

WorkBuddy

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

豆包大模型

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

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

LibLibAI
LibLibAI Hot

一款AI视频创作工具,主要用于国内领先的AI创意平台,以海量模型、低门槛操作与“创作-分享-商业化”生态,让小白与专业创作者都能高效实现图文乃至视频创意表达,适合需要提升相关任务效率的用户。

DeepSeek

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

咔片AIPPT

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

相关专题

更多
Kratos框架HTTP与gRPC服务开发教程
Kratos框架HTTP与gRPC服务开发教程

本专题围绕Kratos框架双协议服务开发,涵盖HTTP路由与处理器编写、参数获取、gRPC服务实现与客户端调用、metadata上下文传递、encoding编解码注册、统一响应封装、超时控制与流式响应实现方法。

0

2026.10.10

Kratos框架Protobuf接口定义与代码生成合集
Kratos框架Protobuf接口定义与代码生成合集

本专题讲解Kratos框架接口定义体系,涵盖proto编写规范、proto add/client/server生成命令、http注解路由、validate校验、OpenAPI文档生成、跨服务proto复用与兼容性设计。

0

2026.10.10

C++虚函数怎么定义和调用
C++虚函数怎么定义和调用

C++虚函数是实现运行时多态的重要机制。本专题从virtual关键字的基本用法入手,介绍基类与派生类之间的函数重写、基类指针调用派生类方法,以及动态绑定的执行过程,帮助初学者掌握虚函数的核心语法。

0

2026.10.10

C++类与对象的封装方法教程
C++类与对象的封装方法教程

C++封装是面向对象编程的核心特性之一,通过类将数据与操作数据的函数组织在一起,并利用访问权限控制外部访问。本专题介绍类的定义、成员变量、成员函数以及public、private和protected的使用方法,帮助初学者掌握封装的基本原理。

0

2026.10.10

C++构造函数定义与调用方法
C++构造函数定义与调用方法

C++构造函数用于初始化类对象,是面向对象编程的重要基础。本专题从构造函数的定义、声明和调用入手,介绍默认构造函数、带参数构造函数、拷贝构造函数及成员初始化列表,帮助初学者掌握对象创建与初始化的基本方法。

0

2026.10.10

Kratos框架零基础入门教程
Kratos框架零基础入门教程

本专题整理Kratos框架入门内容,涵盖Go环境准备、kratos CLI安装升级、new命令创建项目、目录结构分层说明、服务启动与双协议端口、依赖下载报错排查,帮助开发者快速跑通第一个Kratos框架微服务应用。

0

2026.10.10

C++条件判断语句怎么写
C++条件判断语句怎么写

C++条件判断是控制程序执行流程的重要基础。本专题介绍if、if-else、else if和switch等常见分支语句,结合条件表达式、比较运算符与代码示例,帮助初学者掌握不同场景下的判断逻辑。

0

2026.10.10

C++变量怎么声明和赋值
C++变量怎么声明和赋值

C++变量是编写程序和存储数据的基础。本专题围绕变量声明、定义、初始化、赋值和类型选择等内容展开,帮助初学者理解不同变量的用法,并掌握在实际代码中定义和使用变量的方法。

0

2026.10.10

C++运算符基础入门
C++运算符基础入门

本专题详细讲解了C++运算符的类型、语法与使用方法,涵盖算术运算符、关系运算符、逻辑运算符、位运算符、赋值运算符、条件运算符及其他特殊运算符,并通过代码示例解析优先级与结合性。

0

2026.10.09

热门下载

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

精品课程

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

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