-
Notifications
You must be signed in to change notification settings - Fork 1
/
csv.go
48 lines (39 loc) · 1.04 KB
/
csv.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
package csv
import (
"bytes"
"encoding/csv"
"errors"
LogTool "github.com/adimax2953/log-tool"
)
func CreateCSVWithHeader(header []string, body [][]string) ([]byte, error) {
// 创建一个内存中的缓冲区来存储CSV数据
var buf bytes.Buffer
// 创建一个 CSV writer,将数据写入到缓冲区中
writer := csv.NewWriter(&buf)
if header == nil || len(header) == 0 {
return nil, errors.New("Header不得為空")
}
if body == nil || len(body) == 0 {
return nil, errors.New("body不得為空")
}
// 写入 CSV 头部
if err := writer.Write(header); err != nil {
LogTool.LogError("写入 CSV 头部时出错:", err)
return nil, err
}
// 逐行写入数据
for _, record := range body {
if err := writer.Write(record); err != nil {
LogTool.LogError("写入 CSV 数据时出错:", err)
return nil, err
}
}
// 检查是否有错误
if err := writer.Error(); err != nil {
LogTool.LogError("CSV 写入错误:", err)
return nil, err
}
// 刷新并关闭 CSV writer
writer.Flush()
return buf.Bytes(), nil
}