feat: add WirteCsvFile

fileutil/file.go

@@ -454,6 +454,28 @@ func ReadCsvFile(filepath string) ([][]string, error) {
 	return records, nil
 }
 
+// WriteCsvFile write content to target csv file.
+// Play: todo
+func WriteCsvFile(filepath string, records [][]string, append bool) error {
+	flag := os.O_RDWR | os.O_CREATE
+
+	if append {
+		flag = flag | os.O_APPEND
+	}
+
+	f, err := os.OpenFile(filepath, flag, 0644)
+	if err != nil {
+		return err
+	}
+
+	defer f.Close()
+
+	writer := csv.NewWriter(f)
+	writer.Comma = ','
+
+	return writer.WriteAll(records)
+}
+
 // WriteStringToFile write string to target file.
 // Play: https://go.dev/play/p/GhLS6d8lH_g
 func WriteStringToFile(filepath string, content string, append bool) error {

fileutil/file_example_test.go

@@ -276,7 +276,7 @@ func ExampleSha() {
 }
 
 func ExampleReadCsvFile() {
-	content, err := ReadCsvFile("./testdata/test.csv")
+	content, err := ReadCsvFile("./testdata/demo.csv")
 
 	fmt.Println(content)
 	fmt.Println(err)
@@ -286,6 +286,22 @@ func ExampleReadCsvFile() {
 	// <nil>
 }
 
+func ExampleWriteCsvFile() {
+	data := [][]string{
+		{"Lili", "22", "female"},
+		{"Jim", "21", "male"},
+	}
+	err := WriteCsvFile("./testdata/test2.csv", data, false)
+	fmt.Println(err)
+
+	content, _ := ReadCsvFile("./testdata/test2.csv")
+	fmt.Println(content)
+
+	// Output:
+	// <nil>
+	// [[Lili 22 female] [Jim 21 male]]
+}
+
 func ExampleWriteStringToFile() {
 	filepath := "./test.txt"
 

fileutil/file_test.go

@@ -288,7 +288,7 @@ func TestSha(t *testing.T) {
 func TestReadCsvFile(t *testing.T) {
 	assert := internal.NewAssert(t, "TestReadCsvFile")
 
-	content, err := ReadCsvFile("./testdata/test.csv")
+	content, err := ReadCsvFile("./testdata/demo.csv")
 
 	assert.IsNil(err)
 
@@ -297,6 +297,29 @@ func TestReadCsvFile(t *testing.T) {
 	assert.Equal("Bob", content[0][0])
 }
 
+func TestWriteCsvFile(t *testing.T) {
+	assert := internal.NewAssert(t, "TestWriteCsvFile")
+
+	csvFilePath := "./testdata/test1.csv"
+	content := [][]string{
+		{"Lili", "22", "female"},
+		{"Jim", "21", "male"},
+	}
+
+	err := WriteCsvFile(csvFilePath, content, false)
+	assert.IsNil(err)
+
+	readContent, err := ReadCsvFile(csvFilePath)
+
+	assert.IsNil(err)
+
+	assert.Equal(2, len(readContent))
+	assert.Equal(3, len(readContent[0]))
+	assert.Equal("Lili", content[0][0])
+
+	// RemoveFile(csvFilePath)
+}
+
 func TestWriteStringToFile(t *testing.T) {
 	assert := internal.NewAssert(t, "TestWriteStringToFile")
 

fileutil/testdata/test.csv

@@ -1,3 +0,0 @@
-Bob, 12, male
-Duke, 14, male
-Lucy, 16, female