登录
首页 >  Golang >  Go问答

将 CSV 解组为结构,然后导入

来源:stackoverflow

时间:2024-04-28 19:18:32 365浏览 收藏

Golang不知道大家是否熟悉?今天我将给大家介绍《将 CSV 解组为结构,然后导入》,这篇文章主要会讲到等等知识点,如果你在看完本篇文章后,有更好的建议或者发现哪里有问题,希望大家都能积极评论指出,谢谢!希望我们能一起加油进步!

问题内容

我正在尝试学习 go。 我正在努力将 csv 记录解组到结构中,以便我可以创建插入并将这些数据导入到我的数据库中。

我正在寻找一种自动化编组的方法,其方式与编码类似

当我运行 main 时,我收到错误

panic: strconv.parseint: parsing "id": invalid syntax

我已经编写了一些代码,应该将我的 .csv 记录解组为结构

func unmarshal(reader *csv.reader, v interface{}) error {
    record, err := reader.read()
    if err != nil {
        return err
    }
    s := reflect.valueof(v).elem()
    if s.numfield() != len(record) {
        return &fieldmismatch{s.numfield(), len(record)}
    }
    for i := 0; i < s.numfield(); i++ {
        f := s.field(i)
        switch f.type().string() {
        case "string":
            f.setstring(record[i])
        case "int":
            ival, err := strconv.parseint(record[i], 10, 0)
            if err != nil {
                return err
            }
            f.setint(ival)
        default:
            return &unsupportedtype{f.type().string()}
        }
    }
    return nil
}

我在另一个文件中创建了这个结构:

type inserttodatabase struct {
    id         int `csv:"id"`
    groupid    int `csv:"group_id"`
    name    string `csv:"name"`
    status int `csv:"status"`
}

当 main 运行时,它应该运行这段代码,然后解析我的 .csv 文件以创建插入,然后在我的本地数据库上运行。

var InsertTest InsertToDatabase
    reader := csv.NewReader(file)
    for {
        err := Unmarshal(reader, &InsertTest)
        if err == io.EOF {
            break
        }
        if err != nil {
            panic(err)
        }
        channel <- InsertTest
    }
}

有没有办法提高这段代码的可读性?另外,我应该循环并调用 .read 一次处理一行吗?


解决方案


当我运行 main 时,我收到错误

恐慌:strconv.ParseInt:解析“id”:语法无效

我猜您是无意中解析了 CSV 文件的标题行。

本篇关于《将 CSV 解组为结构,然后导入》的介绍就到此结束啦,但是学无止境,想要了解学习更多关于Golang的相关知识,请关注golang学习网公众号!

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