如何使用golang从查询直接导出到CSV

import (
    "database/sql"
    "encoding/csv"
    "log"
    "os"
)

db, err := sql.Open("驱动程序名称", "连接字符串")
if err != nil {
    log.Fatal(err)
}
defer db.Close()

rows, err := db.Query("SELECT 列1, 列2, ... FROM 表名")
if err != nil {
    log.Fatal(err)
}
defer rows.Close()

var data [][]string
for rows.Next() {
    var col1, col2 string
    err := rows.Scan(&col1, &col2)
    if err != nil {
        log.Fatal(err)
    }
    data = append(data, []string{col1, col2})
}

file, err := os.Create("output.csv")
if err != nil {
    log.Fatal(err)
}
defer file.Close()

writer := csv.NewWriter(file)
defer writer.Flush()

for _, row := range data {
    err := writer.Write(row)
    if err != nil {
        log.Fatal(err)
    }
}

writer.Flush()
if err := writer.Error(); err != nil {
    log.Fatal(err)
}