Merge branch 'is-set-fixes'

Author: aarondl

bool.go

@@ -13,13 +13,15 @@ import (
 type Bool struct {
 	Bool  bool
 	Valid bool
+	Set   bool
 }
 
 // NewBool creates a new Bool
-func NewBool(b bool, valid bool) Bool {
+func NewBool(b, valid bool) Bool {
 	return Bool{
 		Bool:  b,
 		Valid: valid,
+		Set:   true,
 	}
 }
 
@@ -36,8 +38,21 @@ func BoolFromPtr(b *bool) Bool {
 	return NewBool(*b, true)
 }
 
+// IsValid returns true if this carries and explicit value and
+// is not null.
+func (b Bool) IsValid() bool {
+	return b.Set && b.Valid
+}
+
+// IsSet returns true if this carries an explicit value (null inclusive)
+func (b Bool) IsSet() bool {
+	return b.Set
+}
+
 // UnmarshalJSON implements json.Unmarshaler.
 func (b *Bool) UnmarshalJSON(data []byte) error {
+	b.Set = true
+
 	if bytes.Equal(data, NullBytes) {
 		b.Bool = false
 		b.Valid = false
@@ -54,7 +69,8 @@ func (b *Bool) UnmarshalJSON(data []byte) error {
 
 // UnmarshalText implements encoding.TextUnmarshaler.
 func (b *Bool) UnmarshalText(text []byte) error {
-	if text == nil || len(text) == 0 {
+	b.Set = true
+	if len(text) == 0 {
 		b.Valid = false
 		return nil
 	}
@@ -99,6 +115,7 @@ func (b Bool) MarshalText() ([]byte, error) {
 func (b *Bool) SetValid(v bool) {
 	b.Bool = v
 	b.Valid = true
+	b.Set = true
 }
 
 // Ptr returns a pointer to this Bool's value, or a nil pointer if this Bool is null.
@@ -117,10 +134,10 @@ func (b Bool) IsZero() bool {
 // Scan implements the Scanner interface.
 func (b *Bool) Scan(value interface{}) error {
 	if value == nil {
-		b.Bool, b.Valid = false, false
+		b.Bool, b.Valid, b.Set = false, false, false
 		return nil
 	}
-	b.Valid = true
+	b.Valid, b.Set = true, true
 	return convert.ConvertAssign(&b.Bool, value)
 }
 

bool_test.go

@@ -6,8 +6,7 @@ import (
 )
 
 var (
-	boolJSON  = []byte(`true`)
-	falseJSON = []byte(`false`)
+	boolJSON = []byte(`true`)
 )
 
 func TestBoolFrom(t *testing.T) {
@@ -31,15 +30,13 @@ func TestBoolFromPtr(t *testing.T) {
 }
 
 func TestUnmarshalBool(t *testing.T) {
-	var b Bool
-	err := json.Unmarshal(boolJSON, &b)
-	maybePanic(err)
-	assertBool(t, b, "bool json")
-
 	var null Bool
-	err = json.Unmarshal(nullJSON, &null)
+	err := json.Unmarshal(nullJSON, &null)
 	maybePanic(err)
 	assertNullBool(t, null, "null json")
+	if !null.Set {
+		t.Error("should be Set", err)
+	}
 
 	var badType Bool
 	err = json.Unmarshal(intJSON, &badType)

byte.go

@@ -11,13 +11,15 @@ import (
 type Byte struct {
 	Byte  byte
 	Valid bool
+	Set   bool
 }
 
 // NewByte creates a new Byte
 func NewByte(b byte, valid bool) Byte {
 	return Byte{
 		Byte:  b,
 		Valid: valid,
+		Set:   true,
 	}
 }
 
@@ -34,8 +36,21 @@ func ByteFromPtr(b *byte) Byte {
 	return NewByte(*b, true)
 }
 
+// IsValid returns true if this carries and explicit value and
+// is not null.
+func (b Byte) IsValid() bool {
+	return b.Set && b.Valid
+}
+
+// IsSet returns true if this carries an explicit value (null inclusive)
+func (b Byte) IsSet() bool {
+	return b.Set
+}
+
 // UnmarshalJSON implements json.Unmarshaler.
 func (b *Byte) UnmarshalJSON(data []byte) error {
+	b.Set = true
+
 	if len(data) == 0 || bytes.Equal(data, NullBytes) {
 		b.Valid = false
 		b.Byte = 0
@@ -58,7 +73,8 @@ func (b *Byte) UnmarshalJSON(data []byte) error {
 
 // UnmarshalText implements encoding.TextUnmarshaler.
 func (b *Byte) UnmarshalText(text []byte) error {
-	if text == nil || len(text) == 0 {
+	b.Set = true
+	if len(text) == 0 {
 		b.Valid = false
 		return nil
 	}
@@ -92,6 +108,7 @@ func (b Byte) MarshalText() ([]byte, error) {
 func (b *Byte) SetValid(n byte) {
 	b.Byte = n
 	b.Valid = true
+	b.Set = true
 }
 
 // Ptr returns a pointer to this Byte's value, or a nil pointer if this Byte is null.
@@ -110,19 +127,17 @@ func (b Byte) IsZero() bool {
 // Scan implements the Scanner interface.
 func (b *Byte) Scan(value interface{}) error {
 	if value == nil {
-		b.Byte, b.Valid = 0, false
+		b.Byte, b.Valid, b.Set = 0, false, false
 		return nil
 	}
 
 	val := value.(string)
 	if len(val) == 0 {
-		b.Valid = false
-		b.Byte = 0
+		b.Byte, b.Valid, b.Set = 0, false, false
 		return nil
 	}
 
-	b.Valid = true
-	b.Byte = byte(val[0])
+	b.Byte, b.Valid, b.Set = val[0], true, true
 	return nil
 }
 

byte_test.go

@@ -5,10 +5,6 @@ import (
 	"testing"
 )
 
-var (
-	byteJSON = []byte(`"b"`)
-)
-
 func TestByteFrom(t *testing.T) {
 	i := ByteFrom('b')
 	assertByte(t, i, "ByteFrom()")
@@ -34,6 +30,9 @@ func TestUnmarshalByte(t *testing.T) {
 	err := json.Unmarshal(nullJSON, &null)
 	maybePanic(err)
 	assertNullByte(t, null, "null json")
+	if !null.Set {
+		t.Error("expected Set to be true")
+	}
 
 	var badType Byte
 	err = json.Unmarshal(boolJSON, &badType)

bytes.go

@@ -15,13 +15,15 @@ var NullBytes = []byte("null")
 type Bytes struct {
 	Bytes []byte
 	Valid bool
+	Set   bool
 }
 
 // NewBytes creates a new Bytes
 func NewBytes(b []byte, valid bool) Bytes {
 	return Bytes{
 		Bytes: b,
 		Valid: valid,
+		Set:   true,
 	}
 }
 
@@ -39,27 +41,41 @@ func BytesFromPtr(b *[]byte) Bytes {
 	return n
 }
 
+// IsValid returns true if this carries and explicit value and
+// is not null.
+func (b Bytes) IsValid() bool {
+	return b.Set && b.Valid
+}
+
+// IsSet returns true if this carries an explicit value (null inclusive)
+func (b Bytes) IsSet() bool {
+	return b.Set
+}
+
 // UnmarshalJSON implements json.Unmarshaler.
 func (b *Bytes) UnmarshalJSON(data []byte) error {
+	b.Set = true
+
 	if bytes.Equal(data, NullBytes) {
 		b.Valid = false
 		b.Bytes = nil
 		return nil
 	}
 
-	var s string
-	if err := json.Unmarshal(data, &s); err != nil {
+	var bv []byte
+	if err := json.Unmarshal(data, &bv); err != nil {
 		return err
 	}
 
-	b.Bytes = []byte(s)
+	b.Bytes = bv
 	b.Valid = true
 	return nil
 }
 
 // UnmarshalText implements encoding.TextUnmarshaler.
 func (b *Bytes) UnmarshalText(text []byte) error {
-	if text == nil || len(text) == 0 {
+	b.Set = true
+	if len(text) == 0 {
 		b.Bytes = nil
 		b.Valid = false
 	} else {
@@ -72,10 +88,10 @@ func (b *Bytes) UnmarshalText(text []byte) error {
 
 // MarshalJSON implements json.Marshaler.
 func (b Bytes) MarshalJSON() ([]byte, error) {
-	if len(b.Bytes) == 0 || b.Bytes == nil {
+	if len(b.Bytes) == 0 {
 		return NullBytes, nil
 	}
-	return b.Bytes, nil
+	return json.Marshal(b.Bytes)
 }
 
 // MarshalText implements encoding.TextMarshaler.
@@ -90,6 +106,7 @@ func (b Bytes) MarshalText() ([]byte, error) {
 func (b *Bytes) SetValid(n []byte) {
 	b.Bytes = n
 	b.Valid = true
+	b.Set = true
 }
 
 // Ptr returns a pointer to this Bytes's value, or a nil pointer if this Bytes is null.
@@ -108,10 +125,10 @@ func (b Bytes) IsZero() bool {
 // Scan implements the Scanner interface.
 func (b *Bytes) Scan(value interface{}) error {
 	if value == nil {
-		b.Bytes, b.Valid = []byte{}, false
+		b.Bytes, b.Valid, b.Set = nil, false, false
 		return nil
 	}
-	b.Valid = true
+	b.Valid, b.Set = true, true
 	return convert.ConvertAssign(&b.Bytes, value)
 }