遍历元素并将其插入MongoDB

对于一个Golang开发者来说，牢固扎实的基础是十分重要的，golang学习网就来带大家一点点的掌握基础知识点。今天本篇文章带大家了解《遍历元素并将其插入MongoDB》，主要介绍了，希望对大家的知识积累有所帮助，快点收藏起来吧，否则需要时就找不到了！

我遇到一个问题，不知道如何解决。我有一个 go 端点来读取 txt 文件的内容并返回其内容。该文件的结构如下：

usertest1,usertest1
usertest2,usertest2

第一步是读取行并在逗号处分割。当我得到结果时，这工作得很好：

[
{
   name: userttest1
   surname: usertest1
},
{
   name: userttest1
   surname: usertest1
}
]

这个逻辑与下面的代码块配合得很好

var dataslice = make([]element, 0)
    lines := strings.split(string(filebytes), "\n")
    var txt *models.textparser
    if err := ctx.shouldbindjson(&txt); err != nil {
        ctx.json(http.statusbadrequest, err.error())
        return
    }
    for _, line := range lines {

        keyval := strings.split(line, ",")
        dataslice = append(dataslice, element{name: keyval[0], surname: keyval[1]})
ctx.json(http.statusok, dataslice)

}

我的元素如下所示：

type element struct {
    name string `json:"name" bson:"name"`
    surname string `json:"surname" bson:"surname"`
}

但现在我想测试这个循环并将每个元素插入 mongo 数据库中。所以我从我的模型开始

型号：

type textparser struct {
    name  string    `json:"name" bson:"name"`
    surname  string    `json:"surname" bson:"surname"`
    createat  time.time `json:"created_at" bson:"created_at"`
    updatedat time.time `json:"updated_at" bson:"updated_at"`
}

type dbtxt struct {
    id        primitive.objectid `json:"id" bson:"_id"`
    name  string             `json:"name" bson:"name"`
    surname  string             `json:"surname" bson:"surname"`
    createat  time.time          `json:"created_at" bson:"created_at"`
    updatedat time.time          `json:"updated_at" bson:"updated_at"`
}

这是我的短信服务：

type txtservice interface {
    createtxt(*models.textparser) (*models.dbtxt, error)
}

这是文本服务实现：

type txtserviceimpl struct {
    txtcollection *mongo.collection
    ctx           context.context
}

func newtextservice(txtcollection *mongo.collection, ctx context.context) txtservice {
    return &txtserviceimpl{txtcollection, ctx}
}

func (t txtserviceimpl) createtxt(text *models.textparser) (*models.dbtxt, error) {
    text.createat = time.now()
    text.updatedat = text.createat
    res, err := t.txtcollection.insertone(t.ctx, text)

    var newtxt *models.dbtxt
    query := bson.m{"_id": res.insertedid}
    if err = t.txtcollection.findone(t.ctx, query).decode(&newtxt); err != nil {
        return nil, err
    }
    return newtxt, nil
}

现在我已经拥有了向 mongo db 实际添加值所需的所有代码。我更新了控制器以执行数据库插入。我的逻辑（实际上完全错误）是，当循环运行并更新我的元素数组时，我可以使用该循环将值插入 mongo 数据库中。所以我做了以下事情：

var dataslice = make([]element, 0)
    lines := strings.split(string(filebytes), "\n")
    var txt *models.textparser
    if err := ctx.shouldbindjson(&txt); err != nil {
        ctx.json(http.statusbadrequest, err.error())
        return
    }
    for _, line := range lines {

        keyval := strings.split(line, ",")
        dataslice = append(dataslice, element{name: keyval[0], surname: keyval[1]})
        newtxt, err := tc.txtservice.createtxt(dataslice)
        if err != nil {
            if strings.contains(err.error(), "title already exists") {
                ctx.json(http.statusconflict, gin.h{"status": "fail", "message": err.error()})
                return
            }

            ctx.json(http.statusbadgateway, gin.h{"status": "fail", "message": err.error()})
            return
            ctx.json(http.statuscreated, gin.h{"status": "success", "data": newtxt})

        }
    }

当我尝试运行该项目时，出现以下错误：

#13 15.09 controllers/parser.controller.go:77:42: cannot use dataSlice (variable of type []Element) as *models.TextParser value in argument to tc.txtService.CreateTxt
------
failed to solve: rpc error: code = Unknown desc = executor failed running [/bin/sh -c go install -v ./...]: exit code: 1

如果有人可以帮助理解我的错误，我将不胜感激。

正确答案

您似乎想通过调用 createtxt 来单独插入每个文档。

要插入的文档是 textparser 类型（也许名称 parsedtext 更合适，因为它只是一个数据对象而不是解析器）。

获得 element 切片后，您希望将其转换为 parsedtext，并可能在 created_at 和 updated_at 数据库字段中添加当前时间。

可以对 element 类型的每个 el 执行此操作，如下所示：

doc := parsedtext{
        el.name,
        el.surname,
        time.now(),
        time.now(),
    }

也许将解析功能提取到单独的函数中是个好主意。例如，这可以称为 parse 并返回元素的切片。

func parse() ([]element, error)

一个有效的、简化的、独立的示例可能看起来像这样：

package main

import (
    "context"
    "fmt"
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/bson/primitive"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
    "log"
    "time"
)

type element struct {
    name    string `json:"name" bson:"name"`
    surname string `json:"surname" bson:"surname"`
}

type parsedtext struct {
    name      string    `json:"name" bson:"name"`
    surname   string    `json:"surname" bson:"surname"`
    createat  time.time `json:"created_at" bson:"created_at"`
    updatedat time.time `json:"updated_at" bson:"updated_at"`
}

type dbtxt struct {
    id        primitive.objectid `json:"id" bson:"_id"`
    name      string             `json:"name" bson:"name"`
    surname   string             `json:"surname" bson:"surname"`
    createat  time.time          `json:"created_at" bson:"created_at"`
    updatedat time.time          `json:"updated_at" bson:"updated_at"`
}

func createtxt(ctx context.context, collection *mongo.collection, text *parsedtext) (*dbtxt, error) {
    res, err := collection.insertone(context.todo(), text)
    if err != nil {
        return nil, err
    }

    var newtxt *dbtxt
    query := bson.m{"_id": res.insertedid}
    if err = collection.findone(ctx, query).decode(&newtxt); err != nil {
        return nil, err
    }
    return newtxt, nil
}

func parse() ([]element, error) {
    return []element{
        {name: "c.j.", surname: "cherryh"},
        {name: "george", surname: "orwell"},
    }, nil
}

func main() {
    ctx := context.todo()
    client, err := mongo.connect(ctx, options.client().applyuri("mongodb://localhost:27017"))
    if err != nil {
        log.fatal(err)
    }
    defer client.disconnect(ctx)

    collection := client.database("db").collection("elements")

    dataslice, err := parse()
    if err != nil {
        log.fatal(err)
    }

    for _, el := range dataslice {
        doc := parsedtext{
            el.name,
            el.surname,
            time.now(),
            time.now(),
        }
        dbtxt, err := createtxt(ctx, collection, &doc)
        if err != nil {
            log.fatal(err)
        }
        fmt.printf("result: %v\n", dbtxt)
    }
}

parse 显然只提供了一些模拟数据，但这只是为了演示转换部分。

补充说明

值得注意的是，mongodb 还提供了 insertmany()。使用它，您可以一次插入整个切片。

如果您想采用此路线，则 createtxt 的参数可以是 dataslice []interface{}。 dataslice 将使用 []interface{} 类型创建。您将添加几个 parsedtext 对象，类似于上面显示的代码。

插入代码看起来像这样：

res, err := t.txtCollection.InsertMany(t.ctx, dataSlice)

由于您似乎对 id 感兴趣，因此您可以使用 res.insertedids 检索它们。

好了，本文到此结束，带大家了解了《遍历元素并将其插入MongoDB》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！