initial commit
This commit is contained in:
+11
@@ -0,0 +1,11 @@
|
||||
# MIT License
|
||||
|
||||
Copyright © 2026 Aneurin Barker Snook a@aneur.in
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||
|
||||
https://mit-license.org/
|
||||
@@ -0,0 +1,7 @@
|
||||
# Go Record
|
||||
|
||||
Record stores mixed scalar data (Boolean, numeric, and string) and provides type-safe methods to retrieve it. It automatically casts values where possible.
|
||||
|
||||
## License
|
||||
|
||||
See [LICENSE.md](./LICENSE.md)
|
||||
@@ -0,0 +1,13 @@
|
||||
module code.aneur.in/go/record
|
||||
|
||||
go 1.25.0
|
||||
|
||||
require github.com/stretchr/testify v1.11.1
|
||||
|
||||
require (
|
||||
github.com/davecgh/go-spew v1.1.1 // indirect
|
||||
github.com/kr/pretty v0.3.1 // indirect
|
||||
github.com/pmezard/go-difflib v1.0.0 // indirect
|
||||
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
)
|
||||
@@ -0,0 +1,19 @@
|
||||
github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
|
||||
github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
|
||||
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
|
||||
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
|
||||
github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
|
||||
github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
|
||||
github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U=
|
||||
github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo=
|
||||
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
@@ -0,0 +1,558 @@
|
||||
package record
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"reflect"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
// Record stores mixed scalar data (Boolean, numeric, and string) and provides type-safe methods to retrieve it.
|
||||
// It automatically casts values where possible.
|
||||
//
|
||||
// https://go.dev/ref/spec#Types
|
||||
type Record map[string]any
|
||||
|
||||
func (r Record) Bool(column string) bool {
|
||||
value, _ := r.BoolSafe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) BoolSafe(column string) (bool, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "bool":
|
||||
return r[column].(bool), true
|
||||
case "int":
|
||||
return r[column].(int) > 0, true
|
||||
case "int16":
|
||||
return r[column].(int16) > 0, true
|
||||
case "int32":
|
||||
return r[column].(int32) > 0, true
|
||||
case "int64":
|
||||
return r[column].(int64) > 0, true
|
||||
case "string":
|
||||
if value, err := strconv.ParseBool(r[column].(string)); err == nil {
|
||||
return value, true
|
||||
}
|
||||
case "uint":
|
||||
return r[column].(uint) > 0, true
|
||||
case "uint8":
|
||||
return r[column].(uint8) > 0, true
|
||||
case "uint16":
|
||||
return r[column].(uint16) > 0, true
|
||||
case "uint32":
|
||||
return r[column].(uint32) > 0, true
|
||||
case "uint64":
|
||||
return r[column].(uint64) > 0, true
|
||||
}
|
||||
}
|
||||
|
||||
return false, false
|
||||
}
|
||||
|
||||
func (r Record) Byte(column string) byte {
|
||||
value, _ := r.ByteSafe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) ByteSafe(column string) (byte, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "byte":
|
||||
return r[column].(byte), true
|
||||
case "uint8":
|
||||
return byte(r[column].(uint8)), true
|
||||
}
|
||||
}
|
||||
|
||||
return byte(0), false
|
||||
}
|
||||
|
||||
func (r Record) Float32(column string) float32 {
|
||||
value, _ := r.Float32Safe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) Float32Safe(column string) (float32, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "float32":
|
||||
return r[column].(float32), true
|
||||
case "float64":
|
||||
return float32(r[column].(float64)), true
|
||||
case "int":
|
||||
return float32(r[column].(int)), true
|
||||
case "int16":
|
||||
return float32(r[column].(int16)), true
|
||||
case "int32":
|
||||
return float32(r[column].(int32)), true
|
||||
case "int64":
|
||||
return float32(r[column].(int64)), true
|
||||
case "string":
|
||||
if value, err := strconv.ParseFloat(r[column].(string), 32); err == nil {
|
||||
return float32(value), true
|
||||
}
|
||||
case "uint":
|
||||
return float32(r[column].(uint)), true
|
||||
case "uint8":
|
||||
return float32(r[column].(uint8)), true
|
||||
case "uint16":
|
||||
return float32(r[column].(uint16)), true
|
||||
case "uint32":
|
||||
return float32(r[column].(uint32)), true
|
||||
case "uint64":
|
||||
return float32(r[column].(uint64)), true
|
||||
}
|
||||
}
|
||||
|
||||
return 0, false
|
||||
}
|
||||
|
||||
func (r Record) Float64(column string) float64 {
|
||||
value, _ := r.Float64Safe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) Float64Safe(column string) (float64, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "float32":
|
||||
return float64(r[column].(float32)), true
|
||||
case "float64":
|
||||
return r[column].(float64), true
|
||||
case "int":
|
||||
return float64(r[column].(int)), true
|
||||
case "int16":
|
||||
return float64(r[column].(int16)), true
|
||||
case "int32":
|
||||
return float64(r[column].(int32)), true
|
||||
case "int64":
|
||||
return float64(r[column].(int64)), true
|
||||
case "string":
|
||||
if value, err := strconv.ParseFloat(r[column].(string), 64); err == nil {
|
||||
return float64(value), true
|
||||
}
|
||||
case "uint":
|
||||
return float64(r[column].(uint)), true
|
||||
case "uint8":
|
||||
return float64(r[column].(uint8)), true
|
||||
case "uint16":
|
||||
return float64(r[column].(uint16)), true
|
||||
case "uint32":
|
||||
return float64(r[column].(uint32)), true
|
||||
case "uint64":
|
||||
return float64(r[column].(uint64)), true
|
||||
}
|
||||
}
|
||||
|
||||
return 0, false
|
||||
}
|
||||
|
||||
func (r Record) Has(column string) bool {
|
||||
_, ok := r.TypeOf(column)
|
||||
return ok
|
||||
}
|
||||
|
||||
func (r Record) Int(column string) int {
|
||||
value, _ := r.IntSafe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) IntSafe(column string) (int, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "bool":
|
||||
if r[column].(bool) {
|
||||
return 1, true
|
||||
} else {
|
||||
return 0, true
|
||||
}
|
||||
case "int":
|
||||
return r[column].(int), true
|
||||
case "int16":
|
||||
return int(r[column].(int16)), true
|
||||
case "int32":
|
||||
return int(r[column].(int32)), true
|
||||
case "int64":
|
||||
return int(r[column].(int64)), true
|
||||
case "string":
|
||||
if value, err := strconv.ParseInt(r[column].(string), 10, 0); err == nil {
|
||||
return int(value), true
|
||||
}
|
||||
case "uint":
|
||||
return int(r[column].(uint)), true
|
||||
case "uint8":
|
||||
return int(r[column].(uint8)), true
|
||||
case "uint16":
|
||||
return int(r[column].(uint16)), true
|
||||
case "uint32":
|
||||
return int(r[column].(uint32)), true
|
||||
case "uint64":
|
||||
return int(r[column].(uint64)), true
|
||||
}
|
||||
}
|
||||
|
||||
return 0, false
|
||||
}
|
||||
|
||||
func (r Record) Int16(column string) int16 {
|
||||
value, _ := r.Int16Safe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) Int16Safe(column string) (int16, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "bool":
|
||||
if r[column].(bool) {
|
||||
return 1, true
|
||||
} else {
|
||||
return 0, true
|
||||
}
|
||||
case "int":
|
||||
return int16(r[column].(int)), true
|
||||
case "int16":
|
||||
return r[column].(int16), true
|
||||
case "int32":
|
||||
return int16(r[column].(int32)), true
|
||||
case "int64":
|
||||
return int16(r[column].(int64)), true
|
||||
case "string":
|
||||
if value, err := strconv.ParseInt(r[column].(string), 10, 16); err == nil {
|
||||
return int16(value), true
|
||||
}
|
||||
case "uint":
|
||||
return int16(r[column].(uint)), true
|
||||
case "uint8":
|
||||
return int16(r[column].(uint8)), true
|
||||
case "uint16":
|
||||
return int16(r[column].(uint16)), true
|
||||
case "uint32":
|
||||
return int16(r[column].(uint32)), true
|
||||
case "uint64":
|
||||
return int16(r[column].(uint64)), true
|
||||
}
|
||||
}
|
||||
|
||||
return 0, false
|
||||
}
|
||||
|
||||
func (r Record) Int32(column string) int32 {
|
||||
value, _ := r.Int32Safe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) Int32Safe(column string) (int32, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "bool":
|
||||
if r[column].(bool) {
|
||||
return 1, true
|
||||
} else {
|
||||
return 0, true
|
||||
}
|
||||
case "int":
|
||||
return int32(r[column].(int)), true
|
||||
case "int16":
|
||||
return int32(r[column].(int16)), true
|
||||
case "int32":
|
||||
return r[column].(int32), true
|
||||
case "int64":
|
||||
return int32(r[column].(int64)), true
|
||||
case "string":
|
||||
if value, err := strconv.ParseInt(r[column].(string), 10, 32); err == nil {
|
||||
return int32(value), true
|
||||
}
|
||||
case "uint":
|
||||
return int32(r[column].(uint)), true
|
||||
case "uint8":
|
||||
return int32(r[column].(uint8)), true
|
||||
case "uint16":
|
||||
return int32(r[column].(uint16)), true
|
||||
case "uint32":
|
||||
return int32(r[column].(uint32)), true
|
||||
case "uint64":
|
||||
return int32(r[column].(uint64)), true
|
||||
}
|
||||
}
|
||||
|
||||
return 0, false
|
||||
}
|
||||
|
||||
func (r Record) Int64(column string) int64 {
|
||||
value, _ := r.Int64Safe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) Int64Safe(column string) (int64, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "bool":
|
||||
if r[column].(bool) {
|
||||
return 1, true
|
||||
} else {
|
||||
return 0, true
|
||||
}
|
||||
case "int":
|
||||
return int64(r[column].(int)), true
|
||||
case "int16":
|
||||
return int64(r[column].(int16)), true
|
||||
case "int32":
|
||||
return int64(r[column].(int32)), true
|
||||
case "int64":
|
||||
return r[column].(int64), true
|
||||
case "string":
|
||||
if value, err := strconv.ParseInt(r[column].(string), 10, 64); err == nil {
|
||||
return value, true
|
||||
}
|
||||
case "uint":
|
||||
return int64(r[column].(uint)), true
|
||||
case "uint8":
|
||||
return int64(r[column].(uint8)), true
|
||||
case "uint16":
|
||||
return int64(r[column].(uint16)), true
|
||||
case "uint32":
|
||||
return int64(r[column].(uint32)), true
|
||||
case "uint64":
|
||||
return int64(r[column].(uint64)), true
|
||||
}
|
||||
}
|
||||
|
||||
return 0, false
|
||||
}
|
||||
|
||||
func (r Record) String(column string) string {
|
||||
value, _ := r.StringSafe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) StringSafe(column string) (string, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "string":
|
||||
return r[column].(string), true
|
||||
default:
|
||||
return fmt.Sprint(r[column]), true
|
||||
}
|
||||
}
|
||||
|
||||
return "", false
|
||||
}
|
||||
|
||||
func (r Record) TypeOf(column string) (reflect.Type, bool) {
|
||||
if r[column] != nil {
|
||||
return reflect.TypeOf(r[column]), true
|
||||
}
|
||||
|
||||
return nil, false
|
||||
}
|
||||
|
||||
func (r Record) Uint(column string) uint {
|
||||
value, _ := r.UintSafe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) UintSafe(column string) (uint, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "bool":
|
||||
if r[column].(bool) {
|
||||
return 1, true
|
||||
} else {
|
||||
return 0, true
|
||||
}
|
||||
case "int":
|
||||
return uint(r[column].(int)), true
|
||||
case "int16":
|
||||
return uint(r[column].(int16)), true
|
||||
case "int32":
|
||||
return uint(r[column].(int32)), true
|
||||
case "int64":
|
||||
return uint(r[column].(int64)), true
|
||||
case "string":
|
||||
if value, err := strconv.ParseUint(r[column].(string), 10, 0); err == nil {
|
||||
return uint(value), true
|
||||
}
|
||||
case "uint":
|
||||
return r[column].(uint), true
|
||||
case "uint8":
|
||||
return uint(r[column].(uint8)), true
|
||||
case "uint16":
|
||||
return uint(r[column].(uint16)), true
|
||||
case "uint32":
|
||||
return uint(r[column].(uint32)), true
|
||||
case "uint64":
|
||||
return uint(r[column].(uint64)), true
|
||||
}
|
||||
}
|
||||
|
||||
return 0, false
|
||||
}
|
||||
|
||||
func (r Record) Uint8(column string) uint8 {
|
||||
value, _ := r.Uint8Safe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) Uint8Safe(column string) (uint8, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "bool":
|
||||
if r[column].(bool) {
|
||||
return 1, true
|
||||
} else {
|
||||
return 0, true
|
||||
}
|
||||
case "int":
|
||||
return uint8(r[column].(int)), true
|
||||
case "int16":
|
||||
return uint8(r[column].(int16)), true
|
||||
case "int32":
|
||||
return uint8(r[column].(int32)), true
|
||||
case "int64":
|
||||
return uint8(r[column].(int64)), true
|
||||
case "string":
|
||||
if value, err := strconv.ParseUint(r[column].(string), 10, 16); err == nil {
|
||||
return uint8(value), true
|
||||
}
|
||||
case "uint":
|
||||
return uint8(r[column].(uint)), true
|
||||
case "uint8":
|
||||
return r[column].(uint8), true
|
||||
case "uint16":
|
||||
return uint8(r[column].(uint16)), true
|
||||
case "uint32":
|
||||
return uint8(r[column].(uint32)), true
|
||||
case "uint64":
|
||||
return uint8(r[column].(uint64)), true
|
||||
}
|
||||
}
|
||||
|
||||
return 0, false
|
||||
}
|
||||
|
||||
func (r Record) Uint16(column string) uint16 {
|
||||
value, _ := r.Uint16Safe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) Uint16Safe(column string) (uint16, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "bool":
|
||||
if r[column].(bool) {
|
||||
return 1, true
|
||||
} else {
|
||||
return 0, true
|
||||
}
|
||||
case "int":
|
||||
return uint16(r[column].(int)), true
|
||||
case "int16":
|
||||
return uint16(r[column].(int16)), true
|
||||
case "int32":
|
||||
return uint16(r[column].(int32)), true
|
||||
case "int64":
|
||||
return uint16(r[column].(int64)), true
|
||||
case "string":
|
||||
if value, err := strconv.ParseUint(r[column].(string), 10, 16); err == nil {
|
||||
return uint16(value), true
|
||||
}
|
||||
case "uint":
|
||||
return uint16(r[column].(uint)), true
|
||||
case "uint8":
|
||||
return uint16(r[column].(uint8)), true
|
||||
case "uint16":
|
||||
return r[column].(uint16), true
|
||||
case "uint32":
|
||||
return uint16(r[column].(uint32)), true
|
||||
case "uint64":
|
||||
return uint16(r[column].(uint64)), true
|
||||
}
|
||||
}
|
||||
|
||||
return 0, false
|
||||
}
|
||||
|
||||
func (r Record) Uint32(column string) uint32 {
|
||||
value, _ := r.Uint32Safe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) Uint32Safe(column string) (uint32, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "bool":
|
||||
if r[column].(bool) {
|
||||
return 1, true
|
||||
} else {
|
||||
return 0, true
|
||||
}
|
||||
case "int":
|
||||
return uint32(r[column].(int)), true
|
||||
case "int16":
|
||||
return uint32(r[column].(int16)), true
|
||||
case "int32":
|
||||
return uint32(r[column].(int32)), true
|
||||
case "int64":
|
||||
return uint32(r[column].(int64)), true
|
||||
case "string":
|
||||
if value, err := strconv.ParseUint(r[column].(string), 10, 32); err == nil {
|
||||
return uint32(value), true
|
||||
}
|
||||
case "uint":
|
||||
return uint32(r[column].(uint)), true
|
||||
case "uint8":
|
||||
return uint32(r[column].(uint8)), true
|
||||
case "uint16":
|
||||
return uint32(r[column].(uint16)), true
|
||||
case "uint32":
|
||||
return r[column].(uint32), true
|
||||
case "uint64":
|
||||
return uint32(r[column].(uint64)), true
|
||||
}
|
||||
}
|
||||
|
||||
return 0, false
|
||||
}
|
||||
|
||||
func (r Record) Uint64(column string) uint64 {
|
||||
value, _ := r.Uint64Safe(column)
|
||||
return value
|
||||
}
|
||||
|
||||
func (r Record) Uint64Safe(column string) (uint64, bool) {
|
||||
if t, ok := r.TypeOf(column); ok {
|
||||
switch t.Name() {
|
||||
case "bool":
|
||||
if r[column].(bool) {
|
||||
return 1, true
|
||||
} else {
|
||||
return 0, true
|
||||
}
|
||||
case "int":
|
||||
return uint64(r[column].(int)), true
|
||||
case "int16":
|
||||
return uint64(r[column].(int16)), true
|
||||
case "int32":
|
||||
return uint64(r[column].(int32)), true
|
||||
case "int64":
|
||||
return uint64(r[column].(int64)), true
|
||||
case "string":
|
||||
if value, err := strconv.ParseUint(r[column].(string), 10, 64); err == nil {
|
||||
return value, true
|
||||
}
|
||||
case "uint":
|
||||
return uint64(r[column].(uint)), true
|
||||
case "uint8":
|
||||
return uint64(r[column].(uint8)), true
|
||||
case "uint16":
|
||||
return uint64(r[column].(uint16)), true
|
||||
case "uint32":
|
||||
return uint64(r[column].(uint32)), true
|
||||
case "uint64":
|
||||
return r[column].(uint64), true
|
||||
}
|
||||
}
|
||||
|
||||
return 0, false
|
||||
}
|
||||
+238
@@ -0,0 +1,238 @@
|
||||
package record
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
// TODO
|
||||
func TestRecord(t *testing.T) {
|
||||
t.Run("bool value", func(t *testing.T) {
|
||||
r := Record{"a": true, "b": false}
|
||||
var value any
|
||||
|
||||
// Case A: value is true
|
||||
value, ok := r.BoolSafe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, true, value)
|
||||
|
||||
_, ok = r.ByteSafe("a")
|
||||
assert.False(t, ok)
|
||||
|
||||
_, ok = r.Float32Safe("a")
|
||||
assert.False(t, ok)
|
||||
|
||||
_, ok = r.Float64Safe("a")
|
||||
assert.False(t, ok)
|
||||
|
||||
value, ok = r.IntSafe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, 1, value)
|
||||
|
||||
value, ok = r.Int16Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, int16(1), value)
|
||||
|
||||
value, ok = r.Int32Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, int32(1), value)
|
||||
|
||||
value, ok = r.Int64Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, int64(1), value)
|
||||
|
||||
value, ok = r.StringSafe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, "true", value)
|
||||
|
||||
value, ok = r.UintSafe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint(1), value)
|
||||
|
||||
value, ok = r.Uint8Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint8(1), value)
|
||||
|
||||
value, ok = r.Uint16Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint16(1), value)
|
||||
|
||||
value, ok = r.Uint32Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint32(1), value)
|
||||
|
||||
value, ok = r.Uint64Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint64(1), value)
|
||||
|
||||
// Case B: value is false
|
||||
value, ok = r.BoolSafe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, false, value)
|
||||
|
||||
_, ok = r.ByteSafe("b")
|
||||
assert.False(t, ok)
|
||||
|
||||
_, ok = r.Float32Safe("b")
|
||||
assert.False(t, ok)
|
||||
|
||||
_, ok = r.Float64Safe("b")
|
||||
assert.False(t, ok)
|
||||
|
||||
value, ok = r.IntSafe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, 0, value)
|
||||
|
||||
value, ok = r.Int16Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, int16(0), value)
|
||||
|
||||
value, ok = r.Int32Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, int32(0), value)
|
||||
|
||||
value, ok = r.Int64Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, int64(0), value)
|
||||
|
||||
value, ok = r.StringSafe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, "false", value)
|
||||
|
||||
value, ok = r.UintSafe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint(0), value)
|
||||
|
||||
value, ok = r.Uint8Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint8(0), value)
|
||||
|
||||
value, ok = r.Uint16Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint16(0), value)
|
||||
|
||||
value, ok = r.Uint32Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint32(0), value)
|
||||
|
||||
value, ok = r.Uint64Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint64(0), value)
|
||||
})
|
||||
|
||||
t.Run("byte value", func(t *testing.T) {
|
||||
r := Record{"a": byte(1), "b": byte(0)}
|
||||
var value any
|
||||
|
||||
// Case A: value is true
|
||||
value, ok := r.BoolSafe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, true, value)
|
||||
|
||||
value, ok = r.ByteSafe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, byte(1), value)
|
||||
|
||||
_, ok = r.Float32Safe("a")
|
||||
assert.False(t, ok)
|
||||
|
||||
_, ok = r.Float64Safe("a")
|
||||
assert.False(t, ok)
|
||||
|
||||
value, ok = r.IntSafe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, 1, value)
|
||||
|
||||
value, ok = r.Int16Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, int16(1), value)
|
||||
|
||||
value, ok = r.Int32Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, int32(1), value)
|
||||
|
||||
value, ok = r.Int64Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, int64(1), value)
|
||||
|
||||
value, ok = r.StringSafe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, "1", value)
|
||||
|
||||
value, ok = r.UintSafe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint(1), value)
|
||||
|
||||
value, ok = r.Uint8Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint8(1), value)
|
||||
|
||||
value, ok = r.Uint16Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint16(1), value)
|
||||
|
||||
value, ok = r.Uint32Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint32(1), value)
|
||||
|
||||
value, ok = r.Uint64Safe("a")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint64(1), value)
|
||||
|
||||
// Case B: value is false
|
||||
value, ok = r.BoolSafe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, false, value)
|
||||
|
||||
value, ok = r.ByteSafe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, byte(0), value)
|
||||
|
||||
_, ok = r.Float32Safe("b")
|
||||
assert.False(t, ok)
|
||||
|
||||
_, ok = r.Float64Safe("b")
|
||||
assert.False(t, ok)
|
||||
|
||||
value, ok = r.IntSafe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, 0, value)
|
||||
|
||||
value, ok = r.Int16Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, int16(0), value)
|
||||
|
||||
value, ok = r.Int32Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, int32(0), value)
|
||||
|
||||
value, ok = r.Int64Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, int64(0), value)
|
||||
|
||||
value, ok = r.StringSafe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, "0", value)
|
||||
|
||||
value, ok = r.UintSafe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint(0), value)
|
||||
|
||||
value, ok = r.Uint8Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint8(0), value)
|
||||
|
||||
value, ok = r.Uint16Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint16(0), value)
|
||||
|
||||
value, ok = r.Uint32Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint32(0), value)
|
||||
|
||||
value, ok = r.Uint64Safe("b")
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint64(0), value)
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user