[csv] Add method 'WithComma' to CSV reader

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ### 12.69.0
 
+* `[csv]` Added method `WithComma` to CSV reader
 * `[spinner]` Added symbols customization
 * `[spinner]` Change default skip symbol to check mark
 * `[spinner]` Change default skip symbol color to dark grey

csv/csv.go

@@ -79,8 +79,20 @@ func (r *Reader) ReadTo(dst []string) error {
  return nil
 }
 
+// WithComma sets comma (fields delimiter) for CSV reader
+func (r *Reader) WithComma(comma rune) *Reader {
+ if r == nil {
+ return nil
+ }
+
+ r.Comma = comma
+
+ return r
+}
+
 // ////////////////////////////////////////////////////////////////////////////////// //
 
+// parseAndFill parses line
 func parseAndFill(src string, dst []string, sep string) {
  l := len(dst)
 
@@ -116,6 +128,7 @@ func parseAndFill(src string, dst []string, sep string) {
  }
 }
 
+// clean cleans destination slice
 func clean(dst []string, since, to int) {
  for i := since; i < to; i++ {
  dst[i] = ""

csv/csv_test.go

@@ -56,9 +56,7 @@ func (s *CSVSuite) TestRead(c *C) {
 
  defer fd.Close()
 
- reader := NewReader(fd)
- reader.Comma = ','
-
+ reader := NewReader(fd).WithComma(',')
  count := 0
 
  for {
@@ -146,6 +144,7 @@ func (s *CSVSuite) TestNil(c *C) {
 
  c.Assert(err, DeepEquals, ErrNilReader)
  c.Assert(r.ReadTo(b), DeepEquals, ErrNilReader)
+ c.Assert(r.WithComma('X'), IsNil)
 }
 
 // ////////////////////////////////////////////////////////////////////////////////// //

csv/examples_test.go

@@ -64,3 +64,26 @@ func ExampleReader_ReadTo() {
  fmt.Printf("%#v\n", data)
  }
 }
+
+func ExampleReader_WithComma() {
+ fd, err := os.Open("file.csv")
+
+ if err != nil {
+ fmt.Println(err.Error())
+ return
+ }
+
+ defer fd.Close()
+
+ reader := NewReader(fd).WithComma(',')
+
+ for {
+ data, err := reader.Read()
+
+ if err == io.EOF {
+ break
+ }
+
+ fmt.Printf("%#v\n", data)
+ }
+}