本教程旨在解决go语言中`math/big.rat`类型在mongodb中存储高精度小数的问题。由于`big.rat`无法直接序列化为bson，我们将介绍一种实用的方法：将其分解为分子和分母（使用`int64`类型）存储在自定义结构体中，并在需要时从mongodb中检索并重构为`big.rat`，确保金融或科学计算中数值的精确性。
引言：Go语言中的高精度小数处理
在Go语言中进行金融计算、科学计算或其他需要极高精度小数处理的场景时，标准浮点数类型（float32, float64）可能因其固有的精度限制而导致误差累积。math/big包提供了big.Rat类型，它以有理数（分数）的形式表示数字，即分子除以分母，从而实现了任意精度的数值运算，有效避免了浮点数精度问题。
然而，将big.Rat类型直接存储到MongoDB这类文档型数据库时，会遇到挑战。mgo驱动无法直接识别并序列化big.Rat为MongoDB支持的BSON数字类型，因为big.Rat的内部结构并非简单的基本类型。
big.Rat的内部结构与MongoDB存储挑战
big.Rat类型在内部由两个*big.Int值构成：一个代表分子（Numerator），另一个代表分母（Denominator）。这两个big.Int值都是非导出字段，意味着我们无法直接通过结构体字段访问它们。但是，big.Rat提供了Num()和Denom()方法来获取这两个*big.Int值。
MongoDB的BSON规范支持整数（int32, int64）、浮点数（double）等基本数值类型。big.Int由于其任意精度特性，不能直接映射到BSON的固定大小整数类型。因此，我们需要一种策略来桥接big.Rat与MongoDB的存储能力。
立即学习“go语言免费学习笔记（深入）”；
解决方案：分解存储与重构
最直接且有效的方法是将big.Rat分解为其分子和分母，并以MongoDB支持的整数类型（如int64）存储。当从数据库中读取数据时，再将存储的分子和分母重构成big.Rat对象。
1. 定义自定义结构体
首先，我们需要定义一个自定义的Go结构体，用于在应用程序和MongoDB之间传递big.Rat的分解形式。这个结构体应该包含导出（大写开头）的字段，分别存储分子和分母。考虑到大多数实际应用场景，int64通常足以覆盖分母和分子的范围，如果需要更大的范围，可以考虑存储为字符串。
package mainimport ( "fmt" "math/big" "log" "go.mongodb.org/mongo-driver/bson" // 使用新的go.mongodb.org/mongo-driver "go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo/options" "context" "time")// CurrencyValue 用于在MongoDB中存储big.Rat的自定义结构体type CurrencyValue struct { Num int64 `bson:"num"` // 分子 Denom int64 `bson:"denom"` // 分母}// CustomTest 包含CurrencyValue的测试结构type CustomTest struct { ID string `bson:"_id,omitempty"` Budget CurrencyValue `bson:"budget"`}// ToBigRat 将CurrencyValue转换为*big.Ratfunc (cv *CurrencyValue) ToBigRat() *big.Rat { return big.NewRat(cv.Num, cv.Denom)}// FromBigRat 将*big.Rat转换为CurrencyValuefunc FromBigRat(r *big.Rat) CurrencyValue { // 获取分子和分母的big.Int值 numBigInt := r.Num() denomBigInt := r.Denom() // 转换为int64。如果超出int64范围，这里需要额外的错误处理或选择其他存储方式（如string） if !numBigInt.IsInt64() || !denomBigInt.IsInt64() { log.Fatalf("Error: big.Rat numerator or denominator exceeds int64 range. Num: %s, Denom: %s", numBigInt.String(), denomBigInt.String()) } return CurrencyValue{ Num: numBigInt.Int64(), Denom: denomBigInt.Int64(), }}
注意： 上述代码中，我将mgo驱动替换为官方推荐的go.mongodb.org/mongo-driver，因为labix.org/v2/mgo已经不再维护。核心逻辑（big.Rat的分解与重构）保持不变。
2. 存储流程示例
接下来，我们将在应用程序中实现存储big.Rat到MongoDB的逻辑。这包括将big.Rat实例转换为CurrencyValue结构，然后通过MongoDB驱动将其插入数据库。
func main() { ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) defer cancel() client, err := mongo.Connect(ctx, options.Client().ApplyURI("mongodb://localhost:27017")) if err != nil { log.Fatal(err) } defer func() { if err = client.Disconnect(ctx); err != nil { log.Fatal(err) } }() collection := client.Database("db_log").Collection("test_precision") // 1. 创建一个big.Rat实例 initialRat := big.NewRat(5, 10) // 0.5 fmt.Printf("Initial big.Rat: %s (FloatString: %s)n", initialRat.String(), initialRat.FloatString(10)) // 2. 将big.Rat转换为CurrencyValue cvToStore := FromBigRat(initialRat) // 3. 创建CustomTest对象并存储 testDoc := CustomTest{ ID: "initial_budget", Budget: cvToStore, } _, err = collection.InsertOne(ctx, testDoc) if err != nil { log.Fatal(err) } fmt.Println("Initial budget stored successfully.") // 演示更新操作，模拟多次减法 cmp := big.NewRat(1, 100000) // 0.00001 currentRat := initialRat // 从初始值开始 for i := 0; i < 20; i++ { currentRat.Sub(currentRat, cmp) fmt.Printf("Iteration %d: %s (FloatString: %s)n", i+1, currentRat.String(), currentRat.FloatString(10)) } // 4. 存储更新后的big.Rat cvUpdatedToStore := FromBigRat(currentRat) updatedDoc := CustomTest{ ID: "updated_budget", Budget: cvUpdatedToStore, } _, err = collection.InsertOne(ctx, updatedDoc) if err != nil { log.Fatal(err) } fmt.Println("Updated budget stored successfully.") // ... (读取流程将在下一节展示)}
3. 读取与重构流程示例
从MongoDB中读取数据时，我们将获取到CurrencyValue结构体。然后，利用CurrencyValue的ToBigRat()方法，可以轻松地将其转换回big.Rat对象，以便继续进行高精度计算。
func main() { // ... (连接MongoDB，存储数据部分同上) ... collection := client.Database("db_log").Collection("test_precision") // 5. 从MongoDB读取数据 fmt.Println("n--- Reading data from MongoDB ---") // 读取初始预算 var retrievedInitial CustomTest err = collection.FindOne(ctx, bson.M{"_id": "initial_budget"}).Decode(&retrievedInitial) if err != nil { log.Fatal(err) } initialBudgetRat := retrievedInitial.Budget.ToBigRat() fmt.Printf("Retrieved initial budget: %s (FloatString: %s)n", initialBudgetRat.String(), initialBudgetRat.FloatString(10)) // 读取更新后的预算 var retrievedUpdated CustomTest err = collection.FindOne(ctx, bson.M{"_id": "updated_budget"}).Decode(&retrievedUpdated) if err != nil { log.Fatal(err) } updatedBudgetRat := retrievedUpdated.Budget.ToBigRat() fmt.Printf("Retrieved updated budget: %s (FloatString: %s)n", updatedBudgetRat.String(), updatedBudgetRat.FloatString(10)) // 验证精度 expectedFinalRat := big.NewRat(5, 10) // 0.5 cmp := big.NewRat(1, 100000) // 0.00001 for i := 0; i < 20; i++ { expectedFinalRat.Sub(expectedFinalRat, cmp) } fmt.Printf("Expected final budget (recalculated): %s (FloatString: %s)n", expectedFinalRat.String(), expectedFinalRat.FloatString(10)) if expectedFinalRat.Cmp(updatedBudgetRat) == 0 { fmt.Println("Precision check passed: Retrieved updated budget matches recalculated expected value.") } else { fmt.Println("Precision check FAILED: Retrieved updated budget does NOT match recalculated expected value.") }}
注意事项
int64的适用性与big.Int的权衡： 尽管big.Rat内部使用big.Int，它支持任意大小的整数，但将其转换为int64存储意味着存在溢出的风险。如果你的分子或分母可能超出int64的范围（约±9 quintillion），那么你需要采取更复杂的策略，例如将big.Int转换为字符串存储。然而，对于大多数金融应用，int64通常足够。在FromBigRat函数中加入了溢出检查，如果发生溢出将导致程序退出，实际应用中应改为返回错误。错误处理： 在实际生产代码中，应仔细处理MongoDB操作可能返回的错误，例如连接错误、插入/查询错误等，而不是简单地使用log.Fatal。驱动选择： 本教程已更新为使用go.mongodb.org/mongo-driver，这是MongoDB官方推荐的Go语言驱动。如果你仍在使用labix.org/v2/mgo，原理是相同的，只需调整API调用方式。原子性操作： 如果在并发环境中对同一个big.Rat值进行频繁更新，需要考虑MongoDB的事务功能或乐观锁机制，以确保数据一致性。索引： 如果你需要根据CurrencyValue的某个字段（如num或denom）进行查询，请在MongoDB中为这些字段创建索引以优化查询性能。
总结
通过将big.Rat分解为int64类型的分子和分母进行存储，并在需要时重构，我们成功解决了Go语言中高精度小数在MongoDB中的持久化问题。这种方法简单、高效，且能确保数值的精确性，非常适用于金融、科学计算等对数据精度要求严格的场景。在实际应用中，请根据具体需求权衡int64的范围限制，并完善错误处理机制。
以上就是Go语言中mgo存储math/big.Rat高精度小数的策略的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1424765.html
微信扫一扫 
支付宝扫一扫 
