Add boil.Byte type
This commit is contained in:
parent
d27823de53
commit
3bfdae6b5c
3 changed files with 137 additions and 2 deletions
61
types/byte.go
Normal file
61
types/byte.go
Normal file
|
@ -0,0 +1,61 @@
|
||||||
|
package types
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql/driver"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Byte is an alias for byte.
|
||||||
|
// Byte implements Marshal and Unmarshal.
|
||||||
|
type Byte byte
|
||||||
|
|
||||||
|
// String output your byte.
|
||||||
|
func (b Byte) String() string {
|
||||||
|
return string(b)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnmarshalJSON sets *b to a copy of data.
|
||||||
|
func (b *Byte) UnmarshalJSON(data []byte) error {
|
||||||
|
if b == nil {
|
||||||
|
return errors.New("json: unmarshal json on nil pointer to byte")
|
||||||
|
}
|
||||||
|
|
||||||
|
var x string
|
||||||
|
if err := json.Unmarshal(data, &x); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(x) > 1 {
|
||||||
|
return errors.New("json: cannot convert to byte, text len is greater than one")
|
||||||
|
}
|
||||||
|
|
||||||
|
*b = Byte(x[0])
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// MarshalJSON returns the JSON encoding of b.
|
||||||
|
func (b Byte) MarshalJSON() ([]byte, error) {
|
||||||
|
return []byte{'"', byte(b), '"'}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Value returns b as a driver.Value.
|
||||||
|
func (b Byte) Value() (driver.Value, error) {
|
||||||
|
return []byte{byte(b)}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Scan stores the src in *b.
|
||||||
|
func (b *Byte) Scan(src interface{}) error {
|
||||||
|
switch src.(type) {
|
||||||
|
case uint8:
|
||||||
|
*b = Byte(src.(uint8))
|
||||||
|
case string:
|
||||||
|
*b = Byte(src.(string)[0])
|
||||||
|
case []byte:
|
||||||
|
*b = Byte(src.([]byte)[0])
|
||||||
|
default:
|
||||||
|
return errors.New("incompatible type for byte")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
74
types/byte_test.go
Normal file
74
types/byte_test.go
Normal file
|
@ -0,0 +1,74 @@
|
||||||
|
package types
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestByteString(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
b := Byte('b')
|
||||||
|
if b.String() != "b" {
|
||||||
|
t.Errorf("Expected %q, got %s", "b", b.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestByteUnmarshal(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
var b Byte
|
||||||
|
err := json.Unmarshal([]byte(`"b"`), &b)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if b != 'b' {
|
||||||
|
t.Errorf("Expected %q, got %s", "b", b)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestByteMarshal(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
b := Byte('b')
|
||||||
|
res, err := json.Marshal(&b)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !bytes.Equal(res, []byte(`"b"`)) {
|
||||||
|
t.Errorf("expected %s, got %s", `"b"`, b.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestByteValue(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
b := Byte('b')
|
||||||
|
v, err := b.Value()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !bytes.Equal([]byte{byte(b)}, v.([]byte)) {
|
||||||
|
t.Errorf("byte mismatch, %v %v", b, v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestByteScan(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
var b Byte
|
||||||
|
|
||||||
|
s := "b"
|
||||||
|
err := b.Scan(s)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !bytes.Equal([]byte{byte(b)}, []byte{'b'}) {
|
||||||
|
t.Errorf("bad []byte: %#v ≠ %#v\n", b, "b")
|
||||||
|
}
|
||||||
|
}
|
|
@ -35,7 +35,7 @@ func (j *JSON) Marshal(obj interface{}) error {
|
||||||
// UnmarshalJSON sets *j to a copy of data.
|
// UnmarshalJSON sets *j to a copy of data.
|
||||||
func (j *JSON) UnmarshalJSON(data []byte) error {
|
func (j *JSON) UnmarshalJSON(data []byte) error {
|
||||||
if j == nil {
|
if j == nil {
|
||||||
return errors.New("JSON: UnmarshalJSON on nil pointer")
|
return errors.New("json: unmarshal json on nil pointer to json")
|
||||||
}
|
}
|
||||||
|
|
||||||
*j = append((*j)[0:0], data...)
|
*j = append((*j)[0:0], data...)
|
||||||
|
@ -68,7 +68,7 @@ func (j *JSON) Scan(src interface{}) error {
|
||||||
case []byte:
|
case []byte:
|
||||||
source = src.([]byte)
|
source = src.([]byte)
|
||||||
default:
|
default:
|
||||||
return errors.New("Incompatible type for JSON")
|
return errors.New("incompatible type for json")
|
||||||
}
|
}
|
||||||
|
|
||||||
*j = JSON(append((*j)[0:0], source...))
|
*j = JSON(append((*j)[0:0], source...))
|
||||||
|
|
Loading…
Reference in a new issue