登录
首页 >  Golang >  Go问答

将 interface{} 类型的 parquet-go 转换为结构体切片的方法有哪些?

来源:stackoverflow

时间:2024-02-14 16:18:25 398浏览 收藏

从现在开始,我们要努力学习啦!今天我给大家带来《将 interface{} 类型的 parquet-go 转换为结构体切片的方法有哪些?》,感兴趣的朋友请继续看下去吧!下文中的内容我们主要会涉及到等等知识点,如果在阅读本文过程中有遇到不清楚的地方,欢迎留言呀!我们一起讨论,一起学习!

问题内容

我正在读取镶木地板文件,如下所示。下面的代码读取 parquet 文件并将其转换为 parquetproduct 结构,稍后我将使用它来从中获取数据。

func (r *clientRepository) read(logg log.Prot, file string, bucket string) error {
    var err error
    fr, err := pars3.NewS3FileReader(context.Background(), bucket, file, r.s3Client.GetSession().Config)
    if err != nil {
        return errs.Wrap(err)
    }
    defer xio.CloseIgnoringErrors(fr)

    pr, err := reader.NewParquetReader(fr, nil, int64(r.cfg.Workers))
    if err != nil {
        return errs.Wrap(err)
    }

    if pr.GetNumRows() == 0 {
        logg.Infof("Skipping %s due to 0 rows", file)
        return nil
    }

    for {
        rows, err := pr.ReadByNumber(r.cfg.RowsToRead)
        if err != nil {
            return errs.Wrap(err)
        }
        if len(rows) <= 0 {
            break
        }

        // doing Marshal here first
        byteSlice, err := json.Marshal(rows)
        if err != nil {
            return errs.Wrap(err)
        }

        var productRows []ParquetProduct
        // and then Unmarshal here
        err = json.Unmarshal(byteSlice, &productRows)
        if err != nil {
            return errs.Wrap(err)
        }

        //.....
        // use productRows here
        //.....
    
    }
    return nil
}

问题陈述

我首先执行 marshal 然后执行 unmarshalling 来获取所需的对象。有什么办法可以避免这一切吗? readbynumber 函数(parquet-go 库)返回 []interface{} 那么是否有办法从 []interface{} 返回我的 []parquetproduct 结构?

我正在使用 go 1.19。这是我用来读取 parquet 文件的库 - https://github.com/xitongsys/parquet-go

是否有更好、更有效的方法来完成此任务?


正确答案


不要使用 readbynumer,而是制作具有所需长度的 []parquetproduct 切片并使用 read

products := make([]ParquetProduct, r.cfg.RowsToRead) 
// ^ slice with length and capacity equal to r.cfg.RowsToRead
err = pr.Read(&products)
if err != nil {
    // ...
}

今天关于《将 interface{} 类型的 parquet-go 转换为结构体切片的方法有哪些?》的内容就介绍到这里了,是不是学起来一目了然!想要了解更多关于的内容请关注golang学习网公众号!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>