This repository has been archived by the owner on Jul 12, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 3
/
amount.go
147 lines (121 loc) · 4.29 KB
/
amount.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
package crosschain
import (
"fmt"
"math"
"math/big"
"github.com/shopspring/decimal"
)
const FLOAT_PRECISION = 6
// AmountBlockchain is a big integer amount as blockchain expects it for tx.
type AmountBlockchain big.Int
// AmountHumanReadable is a decimal amount as a human expects it for readability.
type AmountHumanReadable decimal.Decimal
func (amount AmountBlockchain) String() string {
bigInt := big.Int(amount)
return bigInt.String()
}
// Int converts an AmountBlockchain into *bit.Int
func (amount AmountBlockchain) Int() *big.Int {
bigInt := big.Int(amount)
return &bigInt
}
func (amount AmountBlockchain) Sign() int {
bigInt := big.Int(amount)
return bigInt.Sign()
}
// Uint64 converts an AmountBlockchain into uint64
func (amount AmountBlockchain) Uint64() uint64 {
bigInt := big.Int(amount)
return bigInt.Uint64()
}
// UnmaskFloat64 converts an AmountBlockchain into float64 given the number of decimals
func (amount AmountBlockchain) UnmaskFloat64() float64 {
bigInt := big.Int(amount)
bigFloat := new(big.Float).SetInt(&bigInt)
exponent := new(big.Float).SetFloat64(math.Pow10(FLOAT_PRECISION))
bigFloat = bigFloat.Quo(bigFloat, exponent)
f64, _ := bigFloat.Float64()
return f64
}
// Use the underlying big.Int.Cmp()
func (amount *AmountBlockchain) Cmp(other *AmountBlockchain) int {
return amount.Int().Cmp(other.Int())
}
// Use the underlying big.Int.Add()
func (amount *AmountBlockchain) Add(x *AmountBlockchain) AmountBlockchain {
sum := *amount
return AmountBlockchain(*sum.Int().Add(sum.Int(), x.Int()))
}
// Use the underlying big.Int.Sub()
func (amount *AmountBlockchain) Sub(x *AmountBlockchain) AmountBlockchain {
diff := *amount
return AmountBlockchain(*diff.Int().Sub(diff.Int(), x.Int()))
}
// Use the underlying big.Int.Mul()
func (amount *AmountBlockchain) Mul(x *AmountBlockchain) AmountBlockchain {
prod := *amount
return AmountBlockchain(*prod.Int().Mul(prod.Int(), x.Int()))
}
// Use the underlying big.Int.Div()
func (amount *AmountBlockchain) Div(x *AmountBlockchain) AmountBlockchain {
quot := *amount
return AmountBlockchain(*quot.Int().Div(quot.Int(), x.Int()))
}
func (amount *AmountBlockchain) Abs() AmountBlockchain {
abs := *amount
return AmountBlockchain(*abs.Int().Abs(abs.Int()))
}
func (amount *AmountBlockchain) ToHuman(decimals int32) AmountHumanReadable {
dec := decimal.NewFromBigInt(amount.Int(), -decimals)
return AmountHumanReadable(dec)
}
// NewAmountBlockchainFromUint64 creates a new AmountBlockchain from a uint64
func NewAmountBlockchainFromUint64(u64 uint64) AmountBlockchain {
bigInt := new(big.Int).SetUint64(u64)
return AmountBlockchain(*bigInt)
}
// NewAmountBlockchainToMaskFloat64 creates a new AmountBlockchain as a float64 times 10^FLOAT_PRECISION
func NewAmountBlockchainToMaskFloat64(f64 float64) AmountBlockchain {
bigFloat := new(big.Float).SetFloat64(f64)
exponent := new(big.Float).SetFloat64(math.Pow10(FLOAT_PRECISION))
bigFloat = bigFloat.Mul(bigFloat, exponent)
var bigInt big.Int
bigFloat.Int(&bigInt)
return AmountBlockchain(bigInt)
}
// NewAmountBlockchainFromStr creates a new AmountBlockchain from a string
func NewAmountBlockchainFromStr(str string) AmountBlockchain {
bigInt, _ := new(big.Int).SetString(str, 10)
return AmountBlockchain(*bigInt)
}
// NewAmountHumanReadableFromStr creates a new AmountHumanReadable from a string
func NewAmountHumanReadableFromStr(str string) AmountHumanReadable {
decimal, _ := decimal.NewFromString(str)
return AmountHumanReadable(decimal)
}
func (amount AmountHumanReadable) ToBlockchain(decimals int32) AmountBlockchain {
factor := decimal.NewFromInt32(10).Pow(decimal.NewFromInt32(decimals))
raised := ((decimal.Decimal)(amount)).Mul(factor)
return AmountBlockchain(*raised.BigInt())
}
func (amount AmountHumanReadable) String() string {
return decimal.Decimal(amount).String()
}
func (amount AmountHumanReadable) Div(x AmountHumanReadable) AmountHumanReadable {
return AmountHumanReadable(decimal.Decimal(amount).Div(decimal.Decimal(x)))
}
func (b *AmountBlockchain) MarshalJSON() ([]byte, error) {
return []byte(b.String()), nil
}
func (b *AmountBlockchain) UnmarshalJSON(p []byte) error {
if string(p) == "null" {
return nil
}
var z big.Int
_, ok := z.SetString(string(p), 10)
if !ok {
return fmt.Errorf("not a valid big integer: %s", p)
}
*b = AmountBlockchain(z)
return nil
}