处理上传的csv文件

Question

我有一个 gin 应用程序，它接收一个 post 请求，其中包含一个我想读取但不保存的 csv 文件。我被困在这里尝试读取 post 请求并出现以下错误消息：cannot use file (variable of type *multipart.FileHeader) as io.Reader value in argument to csv.NewReader: missing method Read

file, err := c.FormFile("file")
if err != nil {
    errList["Invalid_body"] = "Unable to get request"
    c.JSON(http.StatusUnprocessableEntity, gin.H{
        "status": http.StatusUnprocessableEntity,
        "error":  errList,
    })
}

r := csv.NewReader(file) // <= Error message

records, err := r.ReadAll()

for _, record := range records {
    fmt.Println(record)
}

有没有我可以使用的好例子？

Answer 1

首先读取文件和文件头

csvPartFile, csvHeader, openErr := r.FormFile("file")
if openErr != nil {
    // handle error 
}

然后从文件中读取行

csvLines, readErr := csv.NewReader(csvPartFile).ReadAll()
if readErr != nil {
  //handle error 
}

你可以通过循环遍历记录的行

for _, line := range csvLines {
        fmt.Println(line)
    }

Answer 2

正如其他答案所提到的，您应该先 Open() 它。
最新版gin.Context.FromFile(string)好像return只有两个值。
这对我有用：

func (c *gin.Context) {
    file_ptr, err := c.FormFile("file")
    if err != nil {
        log.Println(err.Error())
        c.Status(http.StatusUnprocessableEntity)
        return
    }
    log.Println(file_ptr.Filename)
    file, err := file_ptr.Open()
    if err != nil {
        log.Println(err.Error())
        c.Status(http.StatusUnprocessableEntity)
        return
    }
    defer file.Close()
    records, err := csv.NewReader(file).ReadAll()
    if err != nil {
        log.Println(err.Error())
        c.Status(http.StatusUnprocessableEntity)
        return
    }
    for _, line := range records {
        fmt.Println(line)
    }
}

处理上传的csv文件

Process csv file from upload

csv

go

go-gin