转换结构为 CSV 字符串

本指南介绍了一种将数据库响应中扫描的结构转换为 CSV 字符串的方法。为了避免使用可能降低性能的反射，建议将数据转换为二维切片，其中每一行包含一个数据点，而每一列包含该数据点的属性。然后，可以使用 encoding/csv 包轻松地将二维切片转换为 CSV 字符串。此外，该指南还提供了将 CSV 写入字符串变量的方法。

我有一个在数据库响应后被扫描的结构，如下所示。每个字段都是相同的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 字符串，由 a 分隔逗号。

`
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
`

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

正确答案

如果可能，请避免使用 reflect 迭代结构体，因为它可能导致性能下降和代码可读性降低。不要陷入 xy 问题 - 这里的要求是将 data 结构转换为 csv 字符串（y 问题），但这里的 x 问题是避免使用 data 等结构类型作为起点。

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

• [][]string: encoding/csv 和 struct2csv

• []struct{}：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

以上就是《转换结构为 CSV 字符串》的详细内容，更多关于的资料请关注golang学习网公众号！