文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

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

2024-04-04 23:11

关注

大家好,今天本人给大家带来文章《读取位于密码保护 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的技术知识我们会一点点深入介绍,欢迎大家关注编程网公众号,一起学习编程~

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯