diff options
author | Meng-Ying Yang <garfield@dexon.org> | 2018-12-25 16:51:08 +0800 |
---|---|---|
committer | Wei-Ning Huang <w@dexon.org> | 2019-04-09 21:32:55 +0800 |
commit | 0f51629cd2dd0beabd809787453d041ea0d1a25c (patch) | |
tree | 7a90b5a04b41448b6276e47139744b761f578591 /common/big.go | |
parent | 6aee32cd8c3d7f0c7d99bf1fe4951fdf6c534be9 (diff) | |
download | dexon-0f51629cd2dd0beabd809787453d041ea0d1a25c.tar dexon-0f51629cd2dd0beabd809787453d041ea0d1a25c.tar.gz dexon-0f51629cd2dd0beabd809787453d041ea0d1a25c.tar.bz2 dexon-0f51629cd2dd0beabd809787453d041ea0d1a25c.tar.lz dexon-0f51629cd2dd0beabd809787453d041ea0d1a25c.tar.xz dexon-0f51629cd2dd0beabd809787453d041ea0d1a25c.tar.zst dexon-0f51629cd2dd0beabd809787453d041ea0d1a25c.zip |
core: add database/sql support for more types (#102)
* core: types: add database/sql support for BlockNonce
* common: add database/sql support with Big
New Big type is declared to let big.Int support database/sql by
implementing Scan() and Value() on new type.
Diffstat (limited to 'common/big.go')
-rw-r--r-- | common/big.go | 37 |
1 files changed, 36 insertions, 1 deletions
diff --git a/common/big.go b/common/big.go index 65d4377bf..dcc5269a7 100644 --- a/common/big.go +++ b/common/big.go @@ -16,7 +16,11 @@ package common -import "math/big" +import ( + "database/sql/driver" + "fmt" + "math/big" +) // Common big integers often used var ( @@ -28,3 +32,34 @@ var ( Big256 = big.NewInt(256) Big257 = big.NewInt(257) ) + +// Big support database/sql Scan and Value. +type Big big.Int + +// Scan implements Scanner for database/sql. +func (b *Big) Scan(src interface{}) error { + newB := new(big.Int) + switch t := src.(type) { + case int64: + *b = Big(*newB.SetInt64(t)) + case uint64: + *b = Big(*newB.SetUint64(t)) + case []byte: + *b = Big(*newB.SetBytes(t)) + case string: + v, ok := newB.SetString(t, 10) + if !ok { + return fmt.Errorf("invalid string format %v", src) + } + *b = Big(*v) + default: + return fmt.Errorf("can't scan %T into Big", src) + } + return nil +} + +// Value implements valuer for database/sql. +func (b Big) Value() (driver.Value, error) { + b2 := big.Int(b) + return (&b2).String(), nil +} |