doutuanxiao4619 2019-04-07 23:29
浏览 946
已采纳

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

I am trying to get some familiarity with the official mongo-go-driver and ran into an hour-long delay trying to figure out the right syntax for UpdateOne.

My simplest full example follows (NOTE: in order to use this code you will need to substitute in your own user and server names as well as export the login password to the environment as 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
}

The function signature for UpdateOne from the package docs is:

func (coll *Collection) UpdateOne(ctx context.Context, filter interface{}, 
    update interface{}, opts ...*options.UpdateOptions) (*UpdateResult, error)

So I am clearly making some sort of mistake in creating the update interface{} argument to the function because I am presented with this error

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

The most popular answer here shows that I need to use a document sort of like this

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

but taken verbatim this will not compile in the mongo-go-driver.

I think I need some form of a bson document to comply with the Go syntax. What is the best and/or most efficient syntax to create this bson document for the update?

  • 写回答

1条回答 默认 最新

  • dozoqn3347 2019-04-08 02:28
    关注

    After playing around with this for a little while longer I was able to solve the problem after A LOT OF TRIAL AND ERROR using the mongodb bson package by changing the UpdateByName function in my code above as follows:

    // 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
    }
    

    Note the use of bson.D{{$"set", .... It is unfortunate the way MongoDB has implemented the bson package this syntax still does not pass the go-vet. If anyone has a comment to fix the lint conflict below it would be appreciated.

    go.mongodb.org/mongo-driver/bson/primitive.E composite literal uses unkeyed fields
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

悬赏问题

  • ¥15 nslt的可用模型,或者其他可以进行推理的现有模型
  • ¥15 arduino上连sim900a实现连接mqtt服务器
  • ¥15 vncviewer7.0安装后如何正确注册License许可证,激活使用
  • ¥15 phython如何实现以下功能?查找同一用户名的消费金额合并2
  • ¥66 关于人体营养与饮食规划的线性规划模型
  • ¥15 基于深度学习的快递面单识别系统
  • ¥15 Multisim仿真设计地铁到站提醒电路
  • ¥15 怎么用一个500W电源给5台60W的电脑供电
  • ¥15 请推荐一个轻量级规则引擎,配合流程引擎使用,规则引擎负责判断出符合规则的流程引擎模板id
  • ¥15 Excel表只有年月怎么计算年龄