在MongoDB中存储Golang的big.Int的方法

从现在开始，努力学习吧！本文《在MongoDB中存储Golang的big.Int的方法》主要讲解了等等相关知识点，我会在golang学习网中持续更新相关的系列文章，欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧，希望能帮到你！

我有一个引用 *big.int 的结构。当将此结构天真地存储到 mongodb 中（使用官方驱动程序）时，在取回该结构时该字段将变为 nil。将 big.int 存储到 mongodb 中的正确/最佳方式是什么？

    type MyStruct struct {
        Number *big.Int
    }

    nb := MyStruct{Number: big.NewInt(42)}
    r, _ := db.Collection("test").InsertOne(context.TODO(), nb)

    result := &MyStruct{}
    db.Collection("test").FindOne(context.TODO(), bson.D{{"_id", r.InsertedID}}).Decode(result)
    fmt.Println(result) // <== Number will be 0 here

到目前为止，我最好的想法是围绕 big.int 创建一个包装器，它实现 marshalbson 和 unmarshalbson （老实说，我什至不确定如何正确执行）。但这样就很不方便了。

解决方案

这是我想出的一个可能的实现，它将 big.int 作为纯文本存储到 mongodb 中。还可以使用 big.int 的 bytes 和 setbytes 方法（而不是 marshaltext/unmarshaltext）轻松存储为字节数组。

package common

import (
    "fmt"
    "math/big"

    "go.mongodb.org/mongo-driver/bson"
)

type BigInt struct {
    i *big.Int
}

func NewBigInt(bigint *big.Int) *BigInt {
    return &BigInt{i: bigint}
}

func (bi *BigInt) Int() *big.Int {
    return bi.i
}
func (bi *BigInt) MarshalBSON() ([]byte, error) {
    txt, err := bi.i.MarshalText()
    if err != nil {
        return nil, err
    }
    a, err := bson.Marshal(map[string]string{"i": string(txt)})
    return a, err
}

func (bi *BigInt) UnmarshalBSON(data []byte) error {
    var d bson.D
    err := bson.Unmarshal(data, &d)
    if err != nil {
        return err
    }
    if v, ok := d.Map()["i"]; ok {
        bi.i = big.NewInt(0)
        return bi.i.UnmarshalText([]byte(v.(string)))
    }
    return fmt.Errorf("key 'i' missing")
}

本篇关于《在MongoDB中存储Golang的big.Int的方法》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！