官方 mongo-go-driver UpdateOne 中 $set 的 bson 语法是什么

怎么入门Golang编程？需要学习哪些知识点？这是新手们刚接触编程时常见的问题；下面golang学习网就来给大家整理分享一些知识点，希望能够给初学者一些帮助。本篇文章就来介绍《官方 mongo-go-driver UpdateOne 中 $set 的 bson 语法是什么》，涉及到，有需要的可以收藏一下

我正在尝试熟悉官方 mongo-go-driver 和 updateone 的正确语法。

我最简单的完整示例如下：

（注意：为了使用此代码，您需要替换为您自己的用户名和服务器名，并将登录密码作为 mongo_pw 导出到环境中）：

package main

import (
    "context"
    "fmt"
    "os"

    "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"
)

type db struct {
    user       string
    server     string
    database   string
    collection string
    client     *mongo.client
    ctx        context.context
}

var db = db{
    user:       <username>,
    server:     <server_ip>,
    database:   "test",
    collection: "movies",
    ctx:        context.todo(),
}

type movie struct {
    id          primitive.objectid `bson:"_id" json:"id"`
    name        string             `bson:"name" json:"name"`
    description string             `bson:"description" json:"description"`
}

func main() {
    if err := db.connect(); err != nil {
        fmt.println("error: unable to connect")
        os.exit(1)
    }
    fmt.println("connected")

    // the code assumes the original entry for dunkirk is the following
    // {"name":"dunkirk", "description":"a world war 2 movie"}
    updatedmovie := movie{
        name:        "dunkirk",
        description: "movie about the british evacuation in wwii",
    }

    res, err := db.updatebyname(updatedmovie)
    if err != nil {
        fmt.println("error updating movie:", err)
        os.exit(1)
    }
    if res.matchedcount < 1 {
        fmt.println("error: update did not match any documents")
        os.exit(1)
    }
}

// updatebyname changes the description for a movie identified by its name
func (db *db) updatebyname(movie movie) (*mongo.updateresult, error) {
    filter := bson.d{{"name", movie.name}}

    res, err := db.client.database(db.database).collection(db.collection).updateone(
        db.ctx,
        filter,
        movie,
    )
    if err != nil {
        return nil, err
    }
    return res, nil
}

// connect assumes that the database password is stored in the
// environment variable mongo_pw
func (db *db) connect() error {
    pw, ok := os.lookupenv("mongo_pw")
    if !ok {
        fmt.println("error: unable to find mongo_pw in the environment")
        os.exit(1)
    }
    mongouri := fmt.sprintf("mongodb+srv://%s:%s@%s", db.user, pw, db.server)

    // set client options and verify connection
    clientoptions := options.client().applyuri(mongouri)
    client, err := mongo.connect(db.ctx, clientoptions)
    if err != nil {
        return err
    }
    err = client.ping(db.ctx, nil)
    if err != nil {
        return err
    }

    db.client = client
    return nil
}

包文档中 updateone 的函数签名是：

func (coll *collection) updateone(ctx context.context, filter interface{}, 
    update interface{}, opts ...*options.updateoptions) (*updateresult, error)

因此，我在为函数创建 update 接口{} 参数时显然犯了某种错误，因为我遇到了此错误

error updating movie: update document must contain key beginning with '$'

这里最受欢迎的答案表明我需要使用类似这样的文档

{ $set: {"Name" : "The Matrix", "Decription" "Neo and Trinity kick butt" } }

但逐字记录，这不会在 mongo-go-driver 中编译。

我想我需要某种形式的 bson 文档来符合 go 语法。为 update 创建此 bson 文档的最佳和/或最有效的语法是什么？

解决方案

经过一段时间的尝试后，我通过更改上面代码中的 updatebyname 函数，使用 mongodb bson package 解决了大量尝试和错误后的问题，如下所示：

// updatebyname changes the description for a movie identified by its name
func (db *db) updatebyname(movie movie) (*mongo.updateresult, error) {
    filter := bson.d{{"name", movie.name}}
    update := bson.d{{"$set",
        bson.d{
            {"description", movie.description},
        },
    }}

    res, err := db.client.database(db.database).collection(db.collection).updateone(
        db.ctx,
        filter,
        update,
    )
    if err != nil {
        return nil, err
    }
    return res, nil
}

注意 bson.d{{$"set", ... 的使用。不幸的是，mongodb 实现 bson 包的方式仍然没有通过 go-vet。如果有人有评论来解决下面的 lint 冲突，我们将不胜感激。

go.mongodb.org/mongo-driver/bson/primitive.e composite literal uses unkeyed fields

在许多情况下，您可以替换构造

filter := bson.d{{"name", movie.name}}

与

filter := bson.M{"name": movie.Name}

如果参数顺序不重要

到这里，我们也就讲完了《官方 mongo-go-driver UpdateOne 中 $set 的 bson 语法是什么》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！