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

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了在官方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 interface{}参数时显然犯了一些错误,因为我遇到了这个错误

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

最流行的答案here表明我需要使用类似这样的文档

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

但是逐字逐句,这将无法在mongo-go-driver中编译。

我想我需要某种形式的bson文档来遵守Go语法。为update创建这个bson文档的最佳和/或最有效的语法是什么?

答案

在玩了一段时间之后,我能够通过更改上面代码中的mongodb bson package函数来使用UpdateByName进行了很多尝试和错误之后解决问题,如下所示:

// 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。如果有人有评论来修复下面的棉绒冲突,将不胜感激。

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

以上是关于在官方mongo-go-driver的UpdateOne中$ set的bson语法是什么的主要内容,如果未能解决你的问题,请参考以下文章

如何使用连接池将我的 mgo 会话转换为 mongo-go-driver 客户端?

go语言操作mongodb

labix.org/mgo 连接池泄漏问题

[杂项][转载]vs2015update3各版本官方下载地址

如何使用官方 c# 驱动程序在 MongoDB 中使用 Update.Set 更新多个字段?

16.Update Methods-官方文档摘录