从 MySQL 结果生成 .CSV 文件

Question

我正在尝试使用 Go 生成一个 CSV 文件，该文件将存储 MySQL 查询的转储。

我目前可以将结果导出到预先存在的 CSV 文件，但我尝试在 main.go 运行后自动生成 CSV 文件。我尝试使用WriteFile，我知道它会将 CSV 文件写入指定的文件名。我知道这是设计使然，但我希望生成该文件。

Answer 1

rows, _ := db.Query("SELECT * FROM orderTest limit 100;")    err := sqltocsv.WriteFile("orderTest.csv", rows)    if err != nil {        panic(err)    }    columns, _ := rows.Columns()    count := len(columns)    values := make([]interface{}, count)    valuePtrs := make([]interface{}, count)    for rows.Next() {        for i := range columns {            valuePtrs[i] = &values[i]        }        rows.Scan(valuePtrs...)        for i, col := range columns {            val := values[i]            b, ok := val.([]byte)            var v interface{}            if ok {                v = string(b)            } else {                v = val            }            fmt.Println(col, v)        }    }}我的目标是让 OrdeTest.csv 文件在运行 main.go 时自动创建

Answer 2

sqltocsv.WriteFile(...)如果文件不存在，应该为您创建该文件。在底层，它只使用os.Create(...)标准库中的内容。github.com/joho/sqltocsv/sqltocsv.go:// WriteFile writes the CSV to the filename specified, return an error if problemfunc (c Converter) WriteFile(csvFileName string) error {    f, err := os.Create(csvFileName)    if err != nil {        return err    }    err = c.Write(f)    if err != nil {        f.Close() // close, but only return/handle the write error        return err    }    return f.Close()}文档os.Create(...)：// Create creates the named file with mode 0666 (before umask), truncating// it if it already exists. If successful, methods on the returned// File can be used for I/O; the associated file descriptor has mode// O_RDWR.// If there is an error, it will be of type *PathError.func Create(name string) (*File, error) {    return OpenFile(name, O_RDWR|O_CREATE|O_TRUNC, 0666)}