如何将 ClickHouse 的 groupArray 函数的结果初始化为数组

Question

在我的 Go 应用程序中，我通过 clickhouse-go 包向 ClickHouse 数据库发出请求。我所做的这个查询总是只返回一条记录，它是一个整数数组。有什么方法可以将该结果初始化为 Go 中的数组？

var ids []int

ids, err := database.ClickHouse.Exec("SELECT groupArray(ID) FROM layers;")
if err != nil {
    fmt.Println(err)
}

我尝试了这样的代码，但它引发了错误：cannot assign Result to ids (type []int) in multiple assignment.

Answer 1

Exec 方法不返回行，它返回 driver.Result 和 error。

func (stmt *stmt) Exec(args []driver.Value) (driver.Result, error)
                                             ^^^^^^^^^^^^^

并且，driver.Result 类型具有以下定义（已删除 cmets）：

type Result interface {
    LastInsertId() (int64, error)
    RowsAffected() (int64, error)
}

您正在寻找的是返回 driver.Rows 的 Query 方法：

func (stmt *stmt) Query(args []driver.Value) (driver.Rows, error)
                                              ^^^^^^^^^^^

然后您可以遍历行以生成所需的数组。

README.md 中列出了一个示例（在此处复制）：

rows, err := connect.Query("SELECT country_code, os_id, browser_id, categories, action_day, action_time FROM example")
if err != nil {
    log.Fatal(err)
}
defer rows.Close()

for rows.Next() {
    var (
        country               string
        os, browser           uint8
        categories            []int16
        actionDay, actionTime time.Time
    )
    if err := rows.Scan(&country, &os, &browser, &categories, &actionDay, &actionTime); err != nil {
        log.Fatal(err)
    }
    log.Printf("country: %s, os: %d, browser: %d, categories: %v, action_day: %s, action_time: %s", country, os, browser, categories, actionDay, actionTime)
}

希望有帮助！