init: basic tools
This commit is contained in:
36
input_scv.go
Normal file
36
input_scv.go
Normal file
@@ -0,0 +1,36 @@
|
||||
package script
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/csv"
|
||||
"os"
|
||||
)
|
||||
|
||||
// CSV creates csv.Reader reading from filename.
|
||||
func NewCSVReader(filename string) (*CSVReader, error) {
|
||||
return newCSVreader(filename)
|
||||
}
|
||||
|
||||
type CSVReader struct {
|
||||
f *os.File
|
||||
rdr *csv.Reader
|
||||
}
|
||||
|
||||
func (c *CSVReader) Read(context.Context) ([]string, error) {
|
||||
return c.rdr.Read()
|
||||
}
|
||||
|
||||
func (c *CSVReader) Close() error {
|
||||
return c.f.Close()
|
||||
}
|
||||
|
||||
func newCSVreader(name string) (*CSVReader, error) {
|
||||
f, err := os.Open(name)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
csvr := csv.NewReader(f)
|
||||
|
||||
return &CSVReader{f: f, rdr: csvr}, nil
|
||||
}
|
Reference in New Issue
Block a user