csv - 将结构转换为 CSV 字符串

Question

我有一个在数据库响应后被扫描的结构，如下所示。每个字段都是相同的len()。我想采用这个结构并生成一个 CSV 分隔字符串/

package main

import "fmt"

type Data struct {
    id   []string
    col1 []float64
    col2 []float64
}

func main() {
    d := &Data{
        id:   []string{"id_1", "id_1", "id_1", "id_1"},
        col1: []float64{340.384926, 321.385028, 520.341473, 500.385473},
        col2: []float64{123.285031, 4087.284675, -8958.284216, -7612.283668},
    }
    fmt.Printf("%+v", d)
}

我想循环遍历我认为可以使用 reflect 的结构，然后使用结构字段名称作为标题并将值作为单独的列构造一个如下所示的 CSV 字符串对于该 header ，以逗号分隔。

`
id,col1,col2
id_1,340.384926,123.285031
id_1,321.385028,4087.284675
id_1,520.341473,-8958.284216
id_1,500.385473,-7612.283668
`

实现这一目标的有效方法是什么？

Answer 1

如果可能，请避免使用 reflect 来迭代结构，因为它可能导致性能下降和代码可读性降低。不要上当 XY problem - 这里的要求是将 Data 结构转换为 csv 字符串(Y 问题)，但这里的 X 问题是避免使用 Data 等结构类型作为起点。

许多操作 csv 的 golang 包更喜欢:

• [][]字符串:encoding/csv和 struct2csv

• []结构{}:gocsv和 csv2struct

但是，如果Data类型不可避免，您可以先编写一个将Data转换为[][]string的函数，同时避免使用反射(reflect):

func TransformDataTo2DSlice(d Data) [][]string {

    numRows := len(d.id)
    result := make([][]string, numRows+1)

    // Add header row
    result[0] = []string{"id", "col1", "col2"}

    // Add data rows
    for i := 0; i < numRows; i++ {
        result[i+1] = []string{d.id[i],
            strconv.FormatFloat(d.col1[i], 'f', -1, 64),
            strconv.FormatFloat(d.col2[i], 'f', -1, 64),
        }
    }

    return result
}

接下来，使用 encoding/csv 中的 w.WriteAll() 方法轻松将 [][]string 转换为 csv

func main() {

    d := Data{
        id:   []string{"id_1", "id_1", "id_1", "id_1"},
        col1: []float64{340.384926, 321.385028, 520.341473, 500.385473},
        col2: []float64{123.285031, 4087.284675, -8958.284216, -7612.283668},
    }

    d2dslice := TransformDataTo2DSlice(d)

    // fmt.Printf("%+v", d2dslice)
    // [[id, col1, col2],
    // [id_1, 340.384926, 123.285031],
    // [id_1, 321.385028, 4087.284675],
    // [id_1, 520.341473, -8958.284216],
    // [id_1,500.385473,-7612.283668]]

    w := csv.NewWriter(os.Stdout)
    w.WriteAll(d2dslice)

    if err := w.Error(); err != nil {
        log.Fatalln("error writing csv:", err)
    }

    // stdout:
    // id,col1,col2
    // id_1,340.384926,123.285031
    // id_1,321.385028,4087.284675
    // id_1,520.341473,-8958.284216
    // id_1,500.385473,-7612.283668
}

在此处运行上面的程序:go-playground

要将 csv 写入字符串变量，请传入缓冲区:

    buf := new(bytes.Buffer)
    w := csv.NewWriter(buf)
    w.WriteAll(d2dslice)

    if err := w.Error(); err != nil {
        log.Fatalln("error writing csv:", err)
    }

    csvString := buf.String()

    fmt.Printf("%T\n", csvString)  // print the variable type
    // string

    fmt.Printf("%+v\n", csvString) // print the variable value
    // id,col1,col2
    // id_1,340.384926,123.285031
    // id_1,321.385028,4087.284675
    // id_1,520.341473,-8958.284216
    // id_1,500.385473,-7612.283668