Golang-CSV 落日映苍穹つ 2023-10-15 19:02 25阅读 0赞 ### 读写csv文件 ### csv文件是一种以,为字段分隔符的文件，常常第一行表示字段名，之后的行为对应的数据集合，每一行为每一独立的记录，最后的一行记录可以有换行符也可以没有 一个典型的csv文件如下: id, name, age 1, Tom, 20 2, Jack, 30 在标准库encoding/csv中提供了对csv文件的基本操作，该package中声明了两个结构体对象Reader, Writer，分别支持csv文件的读取以及写入。 package 中提供了NewReader的方法，可以返回一个Reader对象，从源码中我们可以看出构造出来的Reader的分隔符为`,`当然我们可以修改，但是在大多数情况下并没有必要 func NewReader(r io.Reader) *Reader { return &Reader{ Comma: ',', r: bufio.NewReader(r), } } NewReader需要一个io.Reader类型的参数，常见的实现了该接口的类型有os.File、strings.Reader、bufio.Reader、bytes.Buffer、bytes.Reader，当然我们在该方法中使用最多的就是os.File了 创建好Reader对象之后，我们便可以调用其中的方法读取数据了，其中Read方法每次读取一条记录，ReadAll方法一次性读完所有记录，它们都有一个error类型的返回值，如果为io.EOF表示文件读取完成，并不是发生错误. package main import ( "encoding/csv" "fmt" "io" "os" ) func main() { file, err := os.Open("data.csv") // 延迟关闭文件 defer file.Close() if err != nil { fmt.Println("文件打开失败: ", err) return } reader := csv.NewReader(file) for { line, err := reader.Read() // 读取一行数据, 返回 []string if err == io.EOF { fmt.Println("文件读取完毕") break } if err != nil { fmt.Println("读取文件时发生错误: ", err) return } fmt.Println("该行内容为: ", line) } } package中提供了NewWriter方法可以构造一个Writer对象，从源码中可以看出，默认以,为分隔符 func NewWriter(w io.Writer) *Writer { return &Writer{ Comma: ',', w: bufio.NewWriter(w), } } NewWriter需要接收一个io.Writer类型的参数，常见实现了该接口的类型有os.File、strings.Writer、bufio.Writer、bytes.Buffer、bytes.Writer,在这里我们使用os.File 然后调用Wirter对象的Write和WriteAll方法便可以写入数据，调用Flush方法可以将缓冲区的数据进入到文件中，调用Error方法可以返回Write和Flush执行时候遇到的问题 package main import ( "encoding/csv" "fmt" "os" ) func main() { file, err := os.Create("data2.csv") defer file.Close() if err != nil { fmt.Println(err) return } writer := csv.NewWriter(file) writer.Write([]string{ "id", "name", "age"}) //写入标题 writer.WriteAll([][]string{ { "1", "Tom", "18"}, { "2", "Jack", "19"}, { "3", "Letty", "20"}}) //写入数据 if err := writer.Error(); err != nil { fmt.Println("执行中发生错误: ", err) } // 可以调用Flush函数立刻将缓冲区中的数据写入文件中 writer.Flush() } ### csv转结构体 ### package main import ( "fmt" "os" "github.com/gocarina/gocsv" ) type NotUsed struct { Name string } type Client struct { // Our example struct, you can use "-" to ignore a field Id string `csv:"client_id"` Name string `csv:"client_name"` Age string `csv:"client_age"` NotUsedString string `csv:"-"` NotUsedStruct NotUsed `csv:"-"` } func main() { clientsFile, err := os.OpenFile("clients.csv", os.O_RDWR|os.O_CREATE, os.ModePerm) if err != nil { panic(err) } defer clientsFile.Close() clients := []*Client{ } if err := gocsv.UnmarshalFile(clientsFile, &clients); err != nil { // Load clients from file panic(err) } for _, client := range clients { fmt.Println("Hello", client.Name) } } 注意: 结构体必须是数组或分片 ### 结构体转csv ### package main import ( "fmt" "os" "github.com/gocarina/gocsv" ) type NotUsed struct { Name string } type Client struct { // Our example struct, you can use "-" to ignore a field Id string `csv:"client_id"` Name string `csv:"client_name"` Age string `csv:"client_age"` NotUsedString string `csv:"-"` NotUsedStruct NotUsed `csv:"-"` } func main() { clientsFile, err := os.OpenFile("clients.csv", os.O_RDWR|os.O_CREATE, os.ModePerm) if err != nil { panic(err) } defer clientsFile.Close() clients := []*Client{ } clients = append(clients, &Client{ Id: "12", Name: "John", Age: "21"}) // Add clients clients = append(clients, &Client{ Id: "13", Name: "Fred"}) clients = append(clients, &Client{ Id: "14", Name: "James", Age: "32"}) clients = append(clients, &Client{ Id: "15", Name: "Danny"}) err = gocsv.MarshalFile(&clients, clientsFile) // Use this to save the CSV back to the file if err != nil { panic(err) } } 注意: 结构体必须是数组或分片 ### 自定义类型转换器 ### csv和结构体的互转函数 type TypeMarshaller interface { MarshalCSV() (string, error) } // TypeUnmarshaller is implemented by any value that has an UnmarshalCSV method // This converter is used to convert a string to your value representation of that string type TypeUnmarshaller interface { UnmarshalCSV(string) error } 结构体转换成csv字符串时，需要实现如下接口 // MarshalText encodes the receiver into UTF-8-encoded text and returns the result. type TextMarshaler interface { MarshalText() (text []byte, err error) } type TextUnmarshaler interface { UnmarshalText(text []byte) error } 例如，我们定义了一个结构体DateTime，里面有一个time.Time类型的属性。并且DateTime类型实现了TypeMarshaller接口的MarshalCSV函数和TypeUnmarshaller接口的UnmarshalCSV函数。如下： type DateTime struct { time.Time } // Convert the internal date as CSV string func (date *DateTime) MarshalCSV() (string, error) { return date.Time.Format("20060201"), nil } // You could also use the standard Stringer interface func (date *DateTime) String() (string) { return date.String() // Redundant, just for example } // Convert the CSV string as internal date func (date *DateTime) UnmarshalCSV(csv string) (err error) { date.Time, err = time.Parse("20060201", csv) return err } type Client struct { // Our example struct with a custom type (DateTime) Id string `csv:"id"` Name string `csv:"name"` Employed DateTime `csv:"employed"` } func main() { client := []Client{ { Id: "001", Name: "Go学堂", Employed: DateTime{ time.Now()}, }, } csvContent, _ := gocsv.MarshalString(client) fmt.Println("csv:", csvContent) //输出内容是 001,Go学堂,20231003 } ### 自定义CSV的Reader/Writer ### csv文件中的分隔符默认是逗号。但也可以是其他字符。这就要求我们在读取或写入之前指定好内容的分隔号。那么就可以通过自定义的Reader/Writer来覆盖默认的Reader/Writer的选项。如下： gocsv.SetCSVReader(func(in io.Reader) gocsv.CSVReader { r := csv.NewReader(in) r.Comma = '|' return r // Allows use pipe as delimiter }) 指定写入的内容是用 分割符 “|” 进行分割的 gocsv.SetCSVWriter(func(out io.Writer) *gocsv.SafeCSVWriter { writer := csv.NewWriter(out) writer.Comma = '|' return gocsv.NewSafeCSVWriter(writer) }) ![在这里插入图片描述][241d60d901ab468c8532f2a2d4413093.gif] 点赞 -收藏-关注-便于以后复习和收到最新内容有其他问题在评论区讨论-或者私信我-收到会在第一时间回复在本博客学习的技术不得以任何方式直接或者间接的从事违反中华人民共和国法律,内容仅供学习、交流与参考 免责声明：本文部分素材来源于网络，版权归原创者所有，如存在文章/图片/音视频等使用不当的情况，请随时私信联系我、以迅速采取适当措施，避免给双方造成不必要的经济损失。感谢,配合,希望我的努力对你有帮助^\_^ [241d60d901ab468c8532f2a2d4413093.gif]: https://img-blog.csdnimg.cn/241d60d901ab468c8532f2a2d4413093.gif
还没有评论，来说两句吧...