将指针作为参数传递到切片,而不是创建切片。
func processFiles(db *gorm.DB, contains string, records interface{}) {
for _, fileName := range getCSVFiles(contains) {
fmt.Println("Processing file", fileName)
file, err := os.Open(filepath.Join(csvPath, fileName))
if err != nil {
panic(err)
}
defer file.Close()
reader := csv.NewReader(file)
reader.Comma = ';'
// note that argument is records, not &records
if err := gocsv.UnmarshalCSVWithoutHeaders(reader, records); err != nil {
panic(err)
}
fmt.Printf("Creating records in database... ")
db.CreateInBatches(records, 50)
fmt.Println("Done")
}
}
来电者如下所示:
var records []models.Empresa
processFiles(db, "EMPRE", &records)