如何在Go中正确将*big.Rat转换为decimal256.Num？

核心思路 #

big.Rat是分数类型（分子/分母），而decimal256.Num是定点小数类型（整数值+缩放因子，即小数点后位数）。转换的关键是将分数转换为符合decimal256精度要求的定点小数，需分三种场景处理：

场景1：分数为整数（分母=1） #

直接使用decimal256.FromBigInt转换分子即可：

import (
    "math/big"
    "github.com/apache/arrow/go/v16/arrow/decimal256"
)

func convertRatToDecimal(rat *big.Rat) (decimal256.Num, error) {
    den := rat.Denom()
    if den.Cmp(big.NewInt(1)) == 0 {
        return decimal256.FromBigInt(rat.Num()), nil
    }
    // 其他场景处理逻辑
}

场景2：分数为有限小数（分母仅含2和5的质因数） #

这类分数可以通过缩放分子，将其转换为整数+缩放因子的形式：

1. 计算分母中2和5的因子个数，确定需要将分母放大为10的幂次的倍数
2. 将分子乘以该倍数，得到整数部分
3. 使用decimal256.New创建带对应缩放因子的decimal值

代码实现：

import (
    "math/big"
    "github.com/apache/arrow/go/v16/arrow/decimal256"
)

// countFactor 计算n中包含factor因子的次数
func countFactor(n *big.Int, factor int64) int {
    cnt := 0
    temp := new(big.Int).Set(n)
    f := big.NewInt(factor)
    zero := big.NewInt(0)
    for temp.Cmp(big.NewInt(1)) > 0 && temp.Mod(temp, f).Cmp(zero) == 0 {
        cnt++
        temp.Div(temp, f)
    }
    return cnt
}

func convertRatToDecimal(rat *big.Rat) (decimal256.Num, error) {
    den := rat.Denom()
    if den.Cmp(big.NewInt(1)) == 0 {
        return decimal256.FromBigInt(rat.Num()), nil
    }

    // 检查分母是否仅含2和5的质因数
    temp := new(big.Int).Set(den)
    cnt2 := countFactor(den, 2)
    cnt5 := countFactor(den, 5)
    temp.Div(temp, new(big.Int).Exp(big.NewInt(2), big.NewInt(int64(cnt2)), nil))
    temp.Div(temp, new(big.Int).Exp(big.NewInt(5), big.NewInt(int64(cnt5)), nil))
    
    if temp.Cmp(big.NewInt(1)) == 0 {
        // 可转换为有限小数
        scale := max(cnt2, cnt5)
        // 计算需要乘的倍数：10^scale / 分母
        multiplier := new(big.Int).Exp(big.NewInt(10), big.NewInt(int64(scale)), nil)
        multiplier.Div(multiplier, den)
        // 分子放大后得到整数
        scaledNum := new(big.Int).Mul(rat.Num(), multiplier)
        return decimal256.New(scaledNum, int32(scale))
    }

    // 场景3：无限循环小数，转字符串后解析
    // 使用big.Rat的FloatString生成足够精度的小数字符串（decimal256最大支持256位精度）
    decimalStr := rat.FloatString(256)
    return decimal256.FromString(decimalStr)
}

func max(a, b int) int {
    if a > b {
        return a
    }
    return b
}

场景3：分数为无限循环小数（分母含2、5以外的质因数） #

这类分数无法精确转换为有限小数，需将其转换为高精度小数字符串后，用decimal256.FromString解析。代码中已包含该逻辑，会自动处理这种情况。

注意事项 #

• 转换过程中需处理decimal256.New和decimal256.FromString可能返回的错误（比如数值超出decimal256的范围）
• 无限循环小数转换会截断到256位精度，若需要四舍五入可调整FloatString的参数或后续处理

内容的提问来源于stack exchange，提问作者Oleksandr Korzhenevskyi