读取位于密码保护 zip 文件中的 CSV 文件

大家好，今天本人给大家带来文章《读取位于密码保护 zip 文件中的 CSV 文件》，文中内容主要涉及到，如果你对Golang方面的知识点感兴趣，那就请各位朋友继续看下去吧~希望能真正帮到你们，谢谢！

我有这样的链接：https://storage.googleapis.com/data/test_file.csv.zip 内容是带有密码的受保护 zip 中的一个 csv 文件。如何从 csv 读取数据？

我已经尝试过这个但它的错误

func ReadCSVZIPFromURL(fileURL string) (data [][]string, err error) {
    resp, err := http.Get(fileURL)
    if err != nil {
        return data, errors.AddTrace(err)
    }
    defer resp.Body.Close()

    r, err := zip.OpenReader(resp.Body)
    if err != nil {
        log.Fatal(err)
    }
    defer r.Close()

    for _, file := range r.File {
        if file.IsEncrypted() {
            file.SetPassword("password")
        }
        reader := csv.NewReader(file)

        data, err := reader.ReadAll()
    }

    return data, nil
}

正确答案

我已经解决了这个问题，这就是解决方案。我们的想法是，我们将字节复制到 zip 阅读器，然后我们将得到 ioreader，我们使用 csv 库读取 ioreader。 为了使用密码加密和解密数据，我们使用这个库“github.com/alexmullins/zip”

func GetCSVFromZipURL(ctx context.Context, fileURL, filePassword string) (ioReader io.Reader, err error) {
    span, ctx := tracer.StartSpanFromContext(ctx)
    defer span.Finish()

    resp, err := http.Get(fileURL)
    if err != nil {
        return ioReader, errors.AddTrace(err)
    }
    defer resp.Body.Close()

    buf := &bytes.Buffer{}

    _, err = io.Copy(buf, resp.Body)
    if err != nil {
        return ioReader, errors.AddTrace(err)
    }

    b := bytes.NewReader(buf.Bytes())
    r, err := zip.NewReader(b, int64(b.Len()))
    if err != nil {
        return ioReader, errors.AddTrace(err)
    }

    for _, f := range r.File {
        if f.IsEncrypted() {
            f.SetPassword(filePassword)
        }
        ioReader, err = f.Open()
        if err != nil {
            return ioReader, errors.AddTrace(err)
        }

        return ioReader, nil
    }

    return ioReader, nil
}

func getUserBenefitListFromCSV(ioReader io.Reader) (userBenefitList []UserBenefit, err error) {
    reader := csv.NewReader(ioReader)

    row := 1
    for {
        csvRowsStr, err := reader.Read()
        if err == io.EOF {
            break
        }
        if err != nil {
            return userBenefitList, errors.AddTrace(err)
        }

        // check if 1st row (header), skip
        if row == 1 {
            row++
            continue
        }

        if len(csvRowsStr) > 0 {
            userID, err := strconv.ParseInt(csvRowsStr[0], 10, 64)
            if err != nil {
                return userBenefitList, errors.AddTrace(err)
            }

            catalogID := 0
            if len(csvRowsStr) > 1 {
                catalogID, err = strconv.ParseInt(csvRowsStr[1], 10, 64)
                if err != nil {
                    return userBenefitList, errors.AddTrace(err)
                }
            }

            userBenefitTemp := UserBenefit{
                UserID:    userID,
                CatalogID: catalogID,
            }

            userBenefitList = append(userBenefitList, userBenefitTemp)
        }
    }

    return userBenefitList, nil
}

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~