如果我有一个 csv 读入一个结构，我该如何操作输入来构建我想要的结构？在各种教程之后，我陷入了困境。这是我离得最近的一次。

我基本上想打开一个 csv，读取选定的列，确保在引用该列时从同一行记录值。然后以可以放入数据库的格式生成数据。

示例 CSV:

Ignore,Customer,Fruit,Number
123,A,Apple,1
123,A,Apple,3
123,B,Orange,4
123,C,Melon,5

示例代码:

package main
import (
    "bufio"
    "encoding/csv"
    "encoding/json"
    "fmt"
    "io"
    "log"
    "os"
)

type Account struct {
    Customer string `json:"Customer"`
    LineItem *LineItem  `json:"LineItem"`
}

type LineItem struct {
    ProductName string `json:"ProductName"`
    Count string `json:"Count"`
}


func main() {
    csvFile, _ := os.Open("/home/frank/gocode/src/local/billing/fruit.csv")

    reader := csv.NewReader(bufio.NewReader(csvFile))
    var billData []Account
    for {
        line, error := reader.Read()
        if error == io.EOF {
            break
        } else if error != nil {
            log.Fatal(error)
        }
        billData = append(billData, Account{
            Customer: line[1],
            LineItem: &LineItem{
                ProductName:   line[2],
                Count: line[3],
            },
        })
    }

    billingJson, _ := json.Marshal(billData)
    fmt.Println(string(billingJson))
}

当前输出为:

[{"Customer":"Customer","LineItem":{"ProductName":"Fruit","Count":"Number"}},{"Customer":"A","LineItem":{"ProductName":"Apple","Count":"1"}},{"Customer":"A","LineItem":{"ProductName":"Apple","Count":"3"}},{"Customer":"B","LineItem":{"ProductName":"Orange","Count":"4"}},{"Customer":"C","LineItem":{"ProductName":"Melon","Count":"5"}}]

我想去掉第一条记录，这样标题就不会保留了。例如

[{"Customer":"A","LineItem":{"ProductName":"Apple","Count":"1"}},{"Customer":"A","LineItem":{"ProductName":"Apple","Count":"3"}},{"Customer":"B","LineItem":{"ProductName":"Orange","Count":"4"}},{"Customer":"C","LineItem":{"ProductName":"Melon","Count":"5"}}]

合并，使客户 A 成为包含两个 LineItem 的一条记录，例如

[{"Customer":"A","LineItem":{"ProductName":"Apple","Count":"1"},"LineItem":{"ProductName":"Apple","Count":"3"}},{"Customer":"B","LineItem":{"ProductName":"Orange","Count":"4"}},{"Customer":"C","LineItem":{"ProductName":"Melon","Count":"5"}}]

任何最佳实践 - 欢迎使用替代方法(不确定此处的 map 是否更好)。希望有足够的信息来帮助我。

最佳答案

删除第一个条目就像 billData = billData[1:] 一样简单。那，或者进行初始读取以提取列名称。

在第二部分，您当前的数据结构不能容忍一对多关系(每个帐户只有一个 LineItem)。之后您需要对列表进行一些处理。 CSV 文件必须是 1:1，因为每一行都被视为一个独立的记录。最简单的方法是使用映射使其成为一对多，但您也可以简单地遍历 slice (保留更接近现有代码):

https://play.golang.org/p/3uevo0taKR5

package main

import (
    "bytes"
    "encoding/csv"
    "encoding/json"
    "fmt"
    "io"
    "log"
)

var data = `Ignore,Customer,Fruit,Number
123,A,Apple,1
123,A,Apple,3
123,B,Orange,4
123,C,Melon,5`

type Account struct {
    Customer  string     `json:"Customer"`
    LineItems []LineItem `json:"LineItems"`
}

type LineItem struct {
    ProductName string `json:"ProductName"`
    Count       string `json:"Count"`
}

func main() {
    reader := csv.NewReader(bytes.NewBufferString(data))

    // Read column label data and discard
    if _, err := reader.Read(); err != nil {
        log.Fatal(err)
    }

    var billData []Account
    for {
        line, err := reader.Read()
        if err == io.EOF {
            break
        }
        if err != nil {
            log.Fatal(err)
        }
        found := false
        for i := range billData {
            if billData[i].Customer == line[1] {
                found = true
                billData[i].LineItems = append(billData[i].LineItems, LineItem{
                    ProductName: line[2],
                    Count:       line[3],
                })
                break
            }
        }
        if !found {
            billData = append(billData, Account{
                Customer: line[1],
                LineItems: []LineItem{
                    {
                        ProductName: line[2],
                        Count:       line[3],
                    },
                },
            })
        }
    }

    billingJson, err := json.MarshalIndent(billData, "", "  ")
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(string(billingJson))
}

输出:

[
    {
        "Customer": "A",
        "LineItems": [
            {
                "ProductName": "Apple",
                "Count": "1"
            },
            {
                "ProductName": "Apple",
                "Count": "3"
            }
        ]
    },
    {
        "Customer": "B",
        "LineItems": [
            {
                "ProductName": "Orange",
                "Count": "4"
            }
        ]
    },
    {
        "Customer": "C",
        "LineItems": [
            {
                "ProductName": "Melon",
                "Count": "5"
            }
        ]
    }
]

最后，我建议使用 err 或类似的错误变量。 error 是内置错误类型的名称，因此通过命名您的变量，您将隐藏该类型并使其无法在同一范围内声明该类型的变量。虽然这不会影响您当前的代码，但它仍然是一种非常糟糕的做法，最终可能会给您带来麻烦。

关于json - CSV 到结构建议，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/48890994/