当前位置:首页 > 文章列表 > Golang > Go问答 > 使用 Go 将内容以 UUID 类型保存到 Mongo 数据库的方法

使用 Go 将内容以 UUID 类型保存到 Mongo 数据库的方法

来源:stackoverflow 2024-02-23 17:33:24 0浏览 收藏

Golang不知道大家是否熟悉?今天我将给大家介绍《使用 Go 将内容以 UUID 类型保存到 Mongo 数据库的方法》,这篇文章主要会讲到等等知识点,如果你在看完本篇文章后,有更好的建议或者发现哪里有问题,希望大家都能积极评论指出,谢谢!希望我们能一起加油进步!

问题内容

通过 mongodb 控制台我可以添加文档:

{
        "_id": objectid(),
        "name": "john",
        "sign": uuid("32e79135-76c4-4682-80b2-8c813be9b792")
}

接下来我可以阅读它:

{
    _id: objectid('64d4ce25ac7c4327e76a53e3'),
    name: "john",
    sign: uuid("32e7913576c4468280b28c813be9b792")
}

我能做到: db.getcollection('people').find({"sign": uuid("32e79135-76c4-4682-80b2-8c813be9b792")})

我可以找到人。

但是当我在 go 中尝试时:

package main

import (
    "context"
    "fmt"
    "log"

    "github.com/google/uuid"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

type Person struct {
    Name string    `bson:"name"`
    Sign uuid.UUID `bson:"sign"`
}

func main() {
    client, err := mongo.Connect(context.Background(), options.Client().ApplyURI("mongodb://localhost:27017"))
    if err != nil {
        log.Fatal(err)
    }
    defer client.Disconnect(context.Background())

    db := client.Database("mydatabase")
    collection := db.Collection("people")

    signUUID := uuid.New()

    person := Person{
        Name: "John",
        Sign: signUUID,
    }

    _, err = collection.InsertOne(context.Background(), person)
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println("Document inserted successfully")
}

然后在 mongo 我有:

{ _id: objectid('64d4cf0a308251580cffee4e'), 姓名: 'john', 符号: bindata(0, 'wqax6oylt1eureehicik9q==') }

我想要一个 uuid 类型的“sign”字段,并且能够在 mongo 中轻松搜索它。


正确答案


github.com/google/uuid 没有实现 bson.ValueMarshaler 接口来将自身编组为 mongodb uuid,这就是它不起作用的原因。

我们可以注册自己的编解码器,将其编码为 mongodb uuid 或从 mongodb uuid 解码。请参阅下面的演示:

package main

import (
    "context"
    "fmt"
    "log"
    "reflect"

    "github.com/google/uuid"
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/bson/bsoncodec"
    "go.mongodb.org/mongo-driver/bson/bsonrw"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

type person struct {
    name string    `bson:"name"`
    sign uuid.uuid `bson:"sign"`
}

func main() {
    tuuid := reflect.typeof(uuid.nil)
    bson.defaultregistry.registertypeencoder(tuuid, bsoncodec.valueencoderfunc(func(ec bsoncodec.encodecontext, vw bsonrw.valuewriter, val reflect.value) error {
        // borrowed from https://github.com/mongodb/mongo-go-driver/blob/89c7e1d1d9a2954472fe852baf27ecca9dbf9bbf/bson/bsoncodec/default_value_encoders.go#l684-l691
        if !val.isvalid() || val.type() != tuuid {
            return bsoncodec.valueencodererror{name: "uuidencodevalue", types: []reflect.type{tuuid}, received: val}
        }
        uuid := val.interface().(uuid.uuid)

        return vw.writebinarywithsubtype(uuid[:], bson.typebinaryuuid)
    }))
    bson.defaultregistry.registertypedecoder(tuuid, bsoncodec.valuedecoderfunc(func(dc bsoncodec.decodecontext, vr bsonrw.valuereader, val reflect.value) error {
        // borrowed from https://github.com/mongodb/mongo-go-driver/blob/89c7e1d1d9a2954472fe852baf27ecca9dbf9bbf/bson/bsoncodec/default_value_decoders.go#l694-l706
        if !val.canset() || val.type() != tuuid {
            return bsoncodec.valuedecodererror{name: "uuiddecodevalue", types: []reflect.type{tuuid}, received: val}
        }

        var data []byte
        var subtype byte
        var err error
        switch vrtype := vr.type(); vrtype {
        case bson.typebinary:
            data, subtype, err = vr.readbinary()
        case bson.typenull:
            err = vr.readnull()
        case bson.typeundefined:
            err = vr.readundefined()
        default:
            err = fmt.errorf("cannot decode %v into a binary", vrtype)
        }

        if err != nil {
            return err
        }

        if subtype != bson.typebinaryuuid {
            return fmt.errorf("cannot decode subtype %v into a uuid", subtype)
        }

        val.set(reflect.valueof(uuid.uuid(data)))
        return nil
    }))

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

    db := client.database("mydatabase")
    collection := db.collection("people")

    signuuid := uuid.new()

    person := person{
        name: "john",
        sign: signuuid,
    }

    if _, err = collection.insertone(context.background(), person); err != nil {
        log.fatal(err)
    }

    var p person

    if err := collection.findone(context.background(),
        bson.m{"sign": signuuid},
    ).decode(&p); err != nil {
        log.fatal(err)
    }
    log.printf("%+v\n", p)
}

请参阅此处的讨论:Add support for mongodb UUID type

为了完整起见,下面显示了实现 bson.valuemarshalerbson.valueunmarshaler 接口的方法:

package main

import (
    "context"
    "errors"
    "fmt"
    "log"

    "github.com/google/uuid"
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/bson/bsontype"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
    "go.mongodb.org/mongo-driver/x/bsonx/bsoncore"
)

type myUUID struct {
    uuid.UUID
}

var (
    _ bson.ValueMarshaler   = (*myUUID)(nil)
    _ bson.ValueUnmarshaler = (*myUUID)(nil)
)

func (u myUUID) MarshalBSONValue() (bsontype.Type, []byte, error) {
    return bson.TypeBinary, bsoncore.AppendBinary(nil, bson.TypeBinaryUUID, []byte(u.UUID[:])), nil
}

func (u *myUUID) UnmarshalBSONValue(typ bsontype.Type, value []byte) error {
    if typ != bson.TypeBinary {
        return fmt.Errorf("cannot unmarshal %v into a Binary", typ)
    }
    subtype, bin, rem, ok := bsoncore.ReadBinary(value)
    if subtype != bson.TypeBinaryUUID {
        return fmt.Errorf("cannot unmarshal binary subtype %v into a UUID", subtype)
    }
    if len(rem) > 0 {
        return fmt.Errorf("value has extra data: %v", rem)
    }
    if !ok {
        return errors.New("value does not have enough bytes")
    }
    *u = myUUID{UUID: uuid.UUID(bin)}
    return nil
}

type Person struct {
    Name string `bson:"name"`
    Sign myUUID `bson:"sign"`
}

func main() {
    client, err := mongo.Connect(context.Background(), options.Client().ApplyURI("mongodb://localhost:27017"))
    if err != nil {
        log.Fatal(err)
    }
    defer client.Disconnect(context.Background())

    db := client.Database("mydatabase")
    collection := db.Collection("people")

    signUUID := myUUID{UUID: uuid.New()}

    person := Person{
        Name: "John",
        Sign: signUUID,
    }

    if _, err = collection.InsertOne(context.Background(), person); err != nil {
        log.Fatal(err)
    }

    var p Person

    if err := collection.FindOne(context.Background(),
        bson.M{"sign": signUUID},
    ).Decode(&p); err != nil {
        log.Fatal(err)
    }
    log.Printf("%+v\n", p)
}

今天关于《使用 Go 将内容以 UUID 类型保存到 Mongo 数据库的方法》的内容就介绍到这里了,是不是学起来一目了然!想要了解更多关于的内容请关注golang学习网公众号!

版本声明
本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
Golang 中的方差计算方法Golang 中的方差计算方法
上一篇
Golang 中的方差计算方法
使用包含文件扩展名的httprouter路由路径
下一篇
使用包含文件扩展名的httprouter路由路径
查看更多
最新文章
查看更多
课程推荐
  • 前端进阶之JavaScript设计模式
    前端进阶之JavaScript设计模式
    设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
    542次学习
  • GO语言核心编程课程
    GO语言核心编程课程
    本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
    509次学习
  • 简单聊聊mysql8与网络通信
    简单聊聊mysql8与网络通信
    如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
    497次学习
  • JavaScript正则表达式基础与实战
    JavaScript正则表达式基础与实战
    在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
    487次学习
  • 从零制作响应式网站—Grid布局
    从零制作响应式网站—Grid布局
    本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
    484次学习
查看更多
AI推荐
  • AI边界平台:智能对话、写作、画图,一站式解决方案
    边界AI平台
    探索AI边界平台,领先的智能AI对话、写作与画图生成工具。高效便捷,满足多样化需求。立即体验!
    372次使用
  • 讯飞AI大学堂免费AI认证证书:大模型工程师认证,提升您的职场竞争力
    免费AI认证证书
    科大讯飞AI大学堂推出免费大模型工程师认证,助力您掌握AI技能,提升职场竞争力。体系化学习,实战项目,权威认证,助您成为企业级大模型应用人才。
    384次使用
  • 茅茅虫AIGC检测:精准识别AI生成内容,保障学术诚信
    茅茅虫AIGC检测
    茅茅虫AIGC检测,湖南茅茅虫科技有限公司倾力打造,运用NLP技术精准识别AI生成文本,提供论文、专著等学术文本的AIGC检测服务。支持多种格式,生成可视化报告,保障您的学术诚信和内容质量。
    528次使用
  • 赛林匹克平台:科技赛事聚合,赋能AI、算力、量子计算创新
    赛林匹克平台(Challympics)
    探索赛林匹克平台Challympics,一个聚焦人工智能、算力算法、量子计算等前沿技术的赛事聚合平台。连接产学研用,助力科技创新与产业升级。
    629次使用
  • SEO  笔格AIPPT:AI智能PPT制作,免费生成,高效演示
    笔格AIPPT
    SEO 笔格AIPPT是135编辑器推出的AI智能PPT制作平台,依托DeepSeek大模型,实现智能大纲生成、一键PPT生成、AI文字优化、图像生成等功能。免费试用,提升PPT制作效率,适用于商务演示、教育培训等多种场景。
    536次使用
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议隐私政策
返回登录
  • 重置密码