如何执行查询并以 CSV

一分耕耘，一分收获！既然打开了这篇文章《如何执行查询并以 CSV》，就坚持看下去吧！文中内容包含等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

我正在尝试使用 bigquery 命令行工具转换查询，而不是使用 go 来完成，但我没有找到如何正确配置查询。我的命令将格式设置为 csv、输出的最大行数、计费的最大字节数以及项目 id；然后运行标准 sql 查询并将输出写入 csv 文件，同时删除顶部的标题和空行。

下面是我使用命令行工具正确运行的命令：

bq query --format csv \
  --max_rows <max_row_int> --maximum_bytes_billed <max_bytes_billed_int> \
  --project_id <project_id> "#standardsql
    <standard_sql_statment>
  " \
  | tail -n +3 >results.csv

我可以从文档中看到，要在 go 中运行查询，我需要创建客户端/连接，然后我应该能够运行查询，如下所示：

ctx := context.Background()

client, error := bigquery.NewClient(ctx, "<project_id_string>")
if err != nil {
    return nil, err
}

q := client.Query(<standard_sql_query>)

如何在 go 代码中的命令行工具代码中设置上面的配置标志？

解决方案

您可以使用包 bigquery [1] [2] 中的 QueryConfig，它保存查询作业的配置。例如，您会在那里找到 MaxBytesBilled，它是 --maximum_bytes_billed 标志的类比。

保存输出结果不是 BigQuery 包的特定功能。为此，您可以使用 encoding/csv 包 [3]。

[1] - https://godoc.org/cloud.google.com/go/bigquery

[2] - https://github.com/googleapis/google-cloud-go/blob/master/bigquery/query.go#L26

[3] - https://golang.org/pkg/encoding/csv/

这里是库 https://cloud.google.com/bigquery/docs/quickstarts/quickstart-client-libraries 的参考实现

然后你可以执行你的脚本：

GOOGLE_CLOUD_PROJECT=项目 ID ./script.go

本篇关于《如何执行查询并以 CSV》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！