feat(Type): add Scan implementation for Optional and Result types for use with databases
This commit is contained in:
102
type/optional.go
102
type/optional.go
@@ -1,6 +1,11 @@
|
|||||||
package Type
|
package Type
|
||||||
|
|
||||||
import Assert "github.com/lbatuska/goutils/assert"
|
import (
|
||||||
|
"database/sql"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
Assert "github.com/lbatuska/goutils/assert"
|
||||||
|
)
|
||||||
|
|
||||||
// CTORS BEGIN
|
// CTORS BEGIN
|
||||||
func Some[T any](value T) Optional[T] {
|
func Some[T any](value T) Optional[T] {
|
||||||
@@ -106,3 +111,98 @@ func (opt *Optional[T]) OkOrElse(f func() error) Result[T] {
|
|||||||
}
|
}
|
||||||
return Err[T](f())
|
return Err[T](f())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (opt *Optional[T]) Scan(src interface{}) error {
|
||||||
|
// DB had a null value
|
||||||
|
if src == nil {
|
||||||
|
opt.present = false
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
// If T is a scanner
|
||||||
|
if scanner, ok := any(&opt.value).(sql.Scanner); ok {
|
||||||
|
if err := scanner.Scan(src); err != nil {
|
||||||
|
opt.present = false
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
opt.present = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
// We implement parsing for some builtin types
|
||||||
|
opt.present = false
|
||||||
|
switch v := any(&opt.value).(type) {
|
||||||
|
|
||||||
|
case *string:
|
||||||
|
if str, ok := src.(string); ok {
|
||||||
|
*v = str
|
||||||
|
opt.present = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if b, ok := src.([]byte); ok {
|
||||||
|
*v = string(b)
|
||||||
|
opt.present = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
case *int:
|
||||||
|
if s, ok := src.(int); ok {
|
||||||
|
*v = s
|
||||||
|
opt.present = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
case *int32:
|
||||||
|
if s, ok := src.(int32); ok {
|
||||||
|
*v = s
|
||||||
|
opt.present = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
case *int64:
|
||||||
|
if s, ok := src.(int64); ok {
|
||||||
|
*v = s
|
||||||
|
opt.present = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
case *bool:
|
||||||
|
switch s := src.(type) {
|
||||||
|
case bool:
|
||||||
|
*v = s
|
||||||
|
opt.present = true
|
||||||
|
return nil
|
||||||
|
// We could technically allow this, however we try to avoid implicit conversions to ensure type safety.
|
||||||
|
// case string:
|
||||||
|
// if s == "1" || s == "true" || s == "t" {
|
||||||
|
// *v = true
|
||||||
|
// opt.present = true
|
||||||
|
// return nil
|
||||||
|
// }
|
||||||
|
// if s == "0" || s == "false" || s == "f" {
|
||||||
|
// *v = false
|
||||||
|
// opt.present = true
|
||||||
|
// return nil
|
||||||
|
// }
|
||||||
|
}
|
||||||
|
|
||||||
|
case *float64:
|
||||||
|
switch s := src.(type) {
|
||||||
|
case float64:
|
||||||
|
*v = s
|
||||||
|
opt.present = true
|
||||||
|
return nil
|
||||||
|
case float32:
|
||||||
|
*v = float64(s)
|
||||||
|
opt.present = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
case *float32:
|
||||||
|
switch s := src.(type) {
|
||||||
|
case float32:
|
||||||
|
*v = s
|
||||||
|
opt.present = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// We couldnt parse the value
|
||||||
|
opt.present = false
|
||||||
|
return fmt.Errorf("unsupported type %T or differs from Optional[%T], and the type doesn't implement sql.Scanner", src, opt.value)
|
||||||
|
}
|
||||||
|
|||||||
102
type/result.go
102
type/result.go
@@ -1,7 +1,9 @@
|
|||||||
package Type
|
package Type
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"database/sql"
|
||||||
"errors"
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
Assert "github.com/lbatuska/goutils/assert"
|
Assert "github.com/lbatuska/goutils/assert"
|
||||||
)
|
)
|
||||||
@@ -134,3 +136,103 @@ func (res *Result[T]) Err() Optional[error] {
|
|||||||
}
|
}
|
||||||
return Optional[error]{present: false}
|
return Optional[error]{present: false}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (res *Result[T]) Scan(src interface{}) error {
|
||||||
|
// DB had a null value
|
||||||
|
if src == nil {
|
||||||
|
res.err = errors.New("src was nil!")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
// If T is a scanner
|
||||||
|
if scanner, ok := any(&res.value).(sql.Scanner); ok {
|
||||||
|
if err := scanner.Scan(src); err != nil {
|
||||||
|
res.err = err
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
res.err = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
// We implement parsing for some builtin types
|
||||||
|
mismatchErr := fmt.Errorf("Type of src (%T) doesn't match type of Result[%T]!", src, res.value)
|
||||||
|
|
||||||
|
switch v := any(&res.value).(type) {
|
||||||
|
|
||||||
|
case *string:
|
||||||
|
if str, ok := src.(string); ok {
|
||||||
|
*v = str
|
||||||
|
res.err = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if b, ok := src.([]byte); ok {
|
||||||
|
*v = string(b)
|
||||||
|
res.err = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
res.err = mismatchErr
|
||||||
|
return res.err
|
||||||
|
|
||||||
|
case *int:
|
||||||
|
if s, ok := src.(int); ok {
|
||||||
|
*v = s
|
||||||
|
res.err = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
res.err = mismatchErr
|
||||||
|
return res.err
|
||||||
|
case *int32:
|
||||||
|
if s, ok := src.(int32); ok {
|
||||||
|
*v = s
|
||||||
|
res.err = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
res.err = mismatchErr
|
||||||
|
return res.err
|
||||||
|
case *int64:
|
||||||
|
if s, ok := src.(int64); ok {
|
||||||
|
*v = s
|
||||||
|
res.err = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
res.err = mismatchErr
|
||||||
|
return res.err
|
||||||
|
|
||||||
|
case *bool:
|
||||||
|
switch s := src.(type) {
|
||||||
|
case bool:
|
||||||
|
*v = s
|
||||||
|
res.err = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
res.err = mismatchErr
|
||||||
|
return res.err
|
||||||
|
|
||||||
|
case *float64:
|
||||||
|
switch s := src.(type) {
|
||||||
|
case float64:
|
||||||
|
*v = s
|
||||||
|
res.err = nil
|
||||||
|
return nil
|
||||||
|
case float32:
|
||||||
|
*v = float64(s)
|
||||||
|
res.err = nil
|
||||||
|
return nil
|
||||||
|
|
||||||
|
}
|
||||||
|
res.err = mismatchErr
|
||||||
|
return res.err
|
||||||
|
|
||||||
|
case *float32:
|
||||||
|
switch s := src.(type) {
|
||||||
|
case float32:
|
||||||
|
*v = s
|
||||||
|
res.err = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
res.err = mismatchErr
|
||||||
|
return res.err
|
||||||
|
}
|
||||||
|
// We couldnt parse the value
|
||||||
|
err := fmt.Errorf("Unsupported type %T, and the type doesn't implement sql.Scanner!", src)
|
||||||
|
res.err = err
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user