如何使用csv.Writer输出MySQL数据

package main

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

    _ "github.com/go-sql-driver/mysql"
)

func main() {
    // 连接到MySQL数据库
    db, err := sql.Open("mysql", "username:password@tcp(hostname:port)/database")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    // 执行SQL查询
    rows, err := db.Query("SELECT * FROM table")
    if err != nil {
        log.Fatal(err)
    }
    defer rows.Close()

    // 创建CSV文件
    file, err := os.Create("output.csv")
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close()

    // 创建csv.Writer
    writer := csv.NewWriter(file)

    // 写入CSV文件头部
    columns, err := rows.Columns()
    if err != nil {
        log.Fatal(err)
    }
    writer.Write(columns)

    // 写入查询结果
    values := make([]sql.RawBytes, len(columns))
    scanArgs := make([]interface{}, len(values))
    for i := range values {
        scanArgs[i] = &values[i]
    }
    for rows.Next() {
        err := rows.Scan(scanArgs...)
        if err != nil {
            log.Fatal(err)
        }
        var record []string
        for _, value := range values {
            record = append(record, string(value))
        }
        writer.Write(record)
    }

    // 关闭CSV文件
    writer.Flush()
    if err := writer.Error(); err != nil {
        log.Fatal(err)
    }

    fmt.Println("CSV文件已生成")
}