feat: add WirteCsvFile

docs/fileutil.md

@@ -43,6 +43,7 @@ import (
 -   [MTime](#MTime)
 -   [Sha](#Sha)
 -   [ReadCsvFile](#ReadCsvFile)
+-   [WriteCsvFile](#WriteCsvFile)
 -   [WriteStringToFile](#WriteStringToFile)
 -   [WriteBytesToFile](#WriteBytesToFile)
 
@@ -503,7 +504,6 @@ func main() {
 }
 ```
 
-
 ### <span id="IsZipFile">IsZipFile</span>
 
 <p>Checks if file is zip file or not.</p>
@@ -664,6 +664,42 @@ func main() {
 }
 ```
 
+### <span id="WriteCsvFile">WriteCsvFile</span>
+
+<p>Write content to target csv file.</p>
+
+<b>Signature:</b>
+
+```go
+func WriteCsvFile(filepath string, records [][]string, append bool) error
+```
+
+<b>Example:</b>
+
+```go
+package main
+
+import (
+    "fmt"
+    "github.com/duke-git/lancet/fileutil"
+)
+
+func main() {
+    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]]
+}
+```
 
 ### <span id="WriteBytesToFile">WriteBytesToFile</span>
 
@@ -688,33 +724,32 @@ import (
 func main() {
     filepath := "./bytes.txt"
 
-	file, err := os.Create(filepath)
-	if err != nil {
-		return
-	}
+    file, err := os.Create(filepath)
+    if err != nil {
+        return
+    }
 
-	defer file.Close()
+    defer file.Close()
 
-	err = fileutil.WriteBytesToFile(filepath, []byte("hello"))
-	if err != nil {
-		return
-	}
+    err = fileutil.WriteBytesToFile(filepath, []byte("hello"))
+    if err != nil {
+        return
+    }
 
-	content, err := fileutil.ReadFileToString(filepath)
-	if err != nil {
-		return
-	}
+    content, err := fileutil.ReadFileToString(filepath)
+    if err != nil {
+        return
+    }
 
-	os.Remove(filepath)
+    os.Remove(filepath)
 
-	fmt.Println(content)
+    fmt.Println(content)
 
-	// Output:
-	// hello
+    // Output:
+    // hello
 }
 ```
 
-
 ### <span id="WriteStringToFile">WriteStringToFile</span>
 
 <p>Writes string to target file.</p>
@@ -738,28 +773,28 @@ import (
 func main() {
     filepath := "./test.txt"
 
-	file, err := os.Create(filepath)
-	if err != nil {
-		return
-	}
+    file, err := os.Create(filepath)
+    if err != nil {
+        return
+    }
 
-	defer file.Close()
+    defer file.Close()
 
-	err = fileutil.WriteStringToFile(filepath, "hello", true)
-	if err != nil {
-		return
-	}
+    err = fileutil.WriteStringToFile(filepath, "hello", true)
+    if err != nil {
+        return
+    }
 
-	content, err := fileutil.ReadFileToString(filepath)
-	if err != nil {
-		return
-	}
+    content, err := fileutil.ReadFileToString(filepath)
+    if err != nil {
+        return
+    }
 
-	os.Remove(filepath)
+    os.Remove(filepath)
 
-	fmt.Println(content)
+    fmt.Println(content)
 
-	// Output:
-	// hello
+    // Output:
+    // hello
 }
-```
+```

docs/fileutil_zh-CN.md

@@ -43,6 +43,7 @@ import (
 -   [MTime](#MTime)
 -   [Sha](#Sha)
 -   [ReadCsvFile](#ReadCsvFile)
+-   [WriteCsvFile](#WriteCsvFile)
 -   [WriteStringToFile](#WriteStringToFile)
 -   [WriteBytesToFile](#WriteBytesToFile)
 
@@ -503,7 +504,6 @@ func main() {
 }
 ```
 
-
 ### <span id="IsZipFile">IsZipFile</span>
 
 <p>判断文件是否是zip压缩文件。</p>
@@ -664,6 +664,43 @@ func main() {
 }
 ```
 
+### <span id="WriteCsvFile">WriteCsvFile</span>
+
+<p>向csv文件写入内容。</p>
+
+<b>函数签名:</b>
+
+```go
+func WriteCsvFile(filepath string, records [][]string, append bool) error
+```
+
+<b>示例:</b>
+
+```go
+package main
+
+import (
+    "fmt"
+    "github.com/duke-git/lancet/v2/fileutil"
+)
+
+func main() {
+    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]]
+}
+```
+
 ### <span id="WriteBytesToFile">WriteBytesToFile</span>
 
 <p>将bytes写入文件。</p>
@@ -687,33 +724,32 @@ import (
 func main() {
     filepath := "./bytes.txt"
 
-	file, err := os.Create(filepath)
-	if err != nil {
-		return
-	}
+    file, err := os.Create(filepath)
+    if err != nil {
+        return
+    }
 
-	defer file.Close()
+    defer file.Close()
 
-	err = fileutil.WriteBytesToFile(filepath, []byte("hello"))
-	if err != nil {
-		return
-	}
+    err = fileutil.WriteBytesToFile(filepath, []byte("hello"))
+    if err != nil {
+        return
+    }
 
-	content, err := fileutil.ReadFileToString(filepath)
-	if err != nil {
-		return
-	}
+    content, err := fileutil.ReadFileToString(filepath)
+    if err != nil {
+        return
+    }
 
-	os.Remove(filepath)
+    os.Remove(filepath)
 
-	fmt.Println(content)
+    fmt.Println(content)
 
-	// Output:
-	// hello
+    // Output:
+    // hello
 }
 ```
 
-
 ### <span id="WriteStringToFile">WriteStringToFile</span>
 
 <p>将字符串写入文件。</p>
@@ -737,28 +773,28 @@ import (
 func main() {
     filepath := "./test.txt"
 
-	file, err := os.Create(filepath)
-	if err != nil {
-		return
-	}
+    file, err := os.Create(filepath)
+    if err != nil {
+        return
+    }
 
-	defer file.Close()
+    defer file.Close()
 
-	err = fileutil.WriteStringToFile(filepath, "hello", true)
-	if err != nil {
-		return
-	}
+    err = fileutil.WriteStringToFile(filepath, "hello", true)
+    if err != nil {
+        return
+    }
 
-	content, err := fileutil.ReadFileToString(filepath)
-	if err != nil {
-		return
-	}
+    content, err := fileutil.ReadFileToString(filepath)
+    if err != nil {
+        return
+    }
 
-	os.Remove(filepath)
+    os.Remove(filepath)
 
-	fmt.Println(content)
+    fmt.Println(content)
 
-	// Output:
-	// hello
+    // Output:
+    // hello
 }
 ```

fileutil/file.go

@@ -370,7 +370,7 @@ func MTime(filepath string) (int64, error) {
 	return f.ModTime().Unix(), nil
 }
 
-// MTime returns file sha value, param `shaType` should be 1, 256 or 512.
+// Sha returns file sha value, param `shaType` should be 1, 256 or 512.
 func Sha(filepath string, shaType ...int) (string, error) {
 	file, err := os.Open(filepath)
 	if err != nil {
@@ -420,6 +420,27 @@ func ReadCsvFile(filepath string) ([][]string, error) {
 	return records, nil
 }
 
+// WriteCsvFile write content to target csv file.
+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.
 func WriteStringToFile(filepath string, content string, append bool) error {
 	flag := os.O_RDWR | os.O_CREATE

fileutil/file_test.go

@@ -268,7 +268,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")
 	t.Log(content)
 
 	assert.IsNil(err)
@@ -278,6 +278,27 @@ 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])
+}
+
 func TestWriteStringToFile(t *testing.T) {
 	assert := internal.NewAssert(t, "TestWriteStringToFile")
 

fileutil/testdata/test1.csv

@@ -0,0 +1,2 @@
+Lili,22,female
+Jim,21,male