假设需要使用这些函数,我怎样才能使这个调用通用,这样我就不会重复几乎相同的代码。
使用“编码/csv”
func getDataFromCSVFiles(files []string) (error, Data) {
data := Data{}
for _, file := range files {
f, err := os.Open(file)
if err != nil {
panic(err)
return err, data
}
defer f.Close()
r := charmap.ISO8859_1.NewDecoder().Reader(f)
reader := csv.NewReader(r)
for i := 1;;i++ {
rec, err := reader.Read()
if i == 1 {
//Skipping header
continue
}
if err != nil {
if err == io.EOF {
break
}
//TODO log error line and csv filename
log.Fatal(err)
}
addWorkbook(rec, &data)
}
}
return nil, data
}
还有 import fw "github.com/hduplooy/gofixedwidth"除了调用 fw.NewReader 几乎是一样的
func getDataFromPRNFiles(files []string) (error, Data) {
data := Data{}
for _, file := range files {
f, err := os.Open(file)
if err != nil {
panic(err)
return err, data
}
defer f.Close()
r := charmap.ISO8859_1.NewDecoder().Reader(f)
reader := fw.NewReader(r)
for i := 1;;i++ {
rec, err := reader.Read()
if i == 1 {
//Skipping header
continue
}
if err != nil {
if err == io.EOF {
break
}
//TODO log error line and csv filename
log.Fatal(err)
}
addWorkbook(rec, &data)
}
}
return nil, data
}
最佳答案
唯一明显的区别是:
reader := csv.NewReader(r)
对比:
reader := fw.NewReader(r)
我不确定 fw
是什么,但大概两个读者都实现了一个通用接口(interface):
type StringSliceReader interface {
Read() ([]string, error)
}
因此您可以将开启器(csv.NewReader
和 fw.NewReader
)作为函数参数传递:
func getDataFromFiles(files []string, func(r io.Reader) StringArrayReader) (error, Data) {
//...
}
但是您需要将它们包装在小函数中以绕过返回类型:
func newCSVReader(r io.Reader) StringSliceReader {
return csv.NewReader(r)
}
func newFWReader(r io.Reader) StringSliceReader {
return fw.NewReader(r)
}
此外,defer
会在函数 退出时将要执行的事情排队,而不是在循环的下一次迭代时执行。所以如果你这样做:
for _, file := range files {
f, err := os.Open(file)
if err != nil {
panic(err)
return err, data
}
defer f.Close()
//...
}
和 files
有一百个条目,那么在关闭任何文件之前,您将有一百个打开的文件。您可能希望将该循环体移动到一个单独的函数中,这样您一次只能打开一个文件。
此外,error
通常是函数的最后一个返回值,因此您应该return data, err
更加地道。
结果可能是这样的:
type StringSliceReader interface {
Read() ([]string, error)
}
type NewReader func(r io.Reader) StringSliceReader
func newCSVReader(r io.Reader) StringSliceReader {
return csv.NewReader(r)
}
func newFWReader(r io.Reader) StringSliceReader {
return fw.NewReader(r)
}
func getDataFrom(file string, data *Data, newReader NewReader) error {
f, err := os.Open(file)
if err != nil {
return err
}
defer f.Close()
r := charmap.ISO8859_1.NewDecoder().Reader(f)
reader := newReader(r)
for i := 1; ; i++ {
rec, err := reader.Read()
if i == 1 {
continue
}
if err != nil {
if err == io.EOF {
break
}
log.Fatal(err)
}
addWorkbook(rec, data)
}
return nil
}
func getDataFromFiles(files []string, newReader NewReader) (Data, error) {
data := Data{}
for _, file := range files {
err := getDataFrom(file, newReader, &data)
if err != nil {
panic(err)
return data, err
}
}
return data, nil
}
您可以说 getDataFromFiles(files, newCSVReader)
读取 CSV 或 getDataFromFiles(files, newFWReader)
读取 FW 文件。如果你想从其他东西读取,你只需要一个 NewReader
函数和实现 StringSliceReader
接口(interface)的东西。
您可能希望将 charmap.ISO8859_1.NewDecoder().Reader(f)
内容隐藏/隐藏到 NewReader
函数中,以便更容易阅读非Latin-1 编码文件。您还可以在 getDataFromFiles
中将 newReader NewReader
替换为 map[string]NewReader
并选择要使用的 NewReader
在文件的扩展名或其他格式标识符上。
关于不同包的Golang函数泛化,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48837797/