feat: bool, byte, time scanners and marshallers

Author: abdivasiyev

bool.go

@@ -0,0 +1,76 @@
+package null
+
+import (
+	"database/sql"
+	"database/sql/driver"
+	"encoding/json"
+	"fmt"
+)
+
+type Bool struct {
+	nullValue *sql.NullBool
+}
+
+func (s Bool) String() string {
+	if s.nullValue != nil {
+		return fmt.Sprint(s.nullValue.Bool)
+	}
+
+	return ""
+}
+
+func NewBool(value bool) Bool {
+	return Bool{
+		nullValue: &sql.NullBool{
+			Valid: true,
+			Bool:  value,
+		},
+	}
+}
+
+func (s Bool) Get() bool {
+	if s.nullValue != nil {
+		return s.nullValue.Bool
+	}
+
+	return false
+}
+
+func (s *Bool) Value() (driver.Value, error) {
+	if s.nullValue != nil {
+		return s.nullValue.Value()
+	}
+
+	return nil, nil
+}
+
+func (s *Bool) Scan(value any) error {
+	if err := s.nullValue.Scan(value); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (s *Bool) UnmarshalJSON(bytes []byte) error {
+	s.nullValue = &sql.NullBool{
+		Bool:  false,
+		Valid: false,
+	}
+
+	if len(bytes) > 0 {
+		if err := json.Unmarshal(bytes, &s.nullValue.Bool); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (s Bool) MarshalJSON() ([]byte, error) {
+	if s.nullValue.Valid {
+		return json.Marshal(s.nullValue.Bool)
+	}
+
+	return json.Marshal(nil)
+}

byte.go

@@ -0,0 +1,76 @@
+package null
+
+import (
+	"database/sql"
+	"database/sql/driver"
+	"encoding/json"
+	"fmt"
+)
+
+type Byte struct {
+	nullValue *sql.NullByte
+}
+
+func (s Byte) String() string {
+	if s.nullValue != nil {
+		return fmt.Sprint(s.nullValue.Byte)
+	}
+
+	return ""
+}
+
+func NewByte(value byte) Byte {
+	return Byte{
+		nullValue: &sql.NullByte{
+			Valid: true,
+			Byte:  value,
+		},
+	}
+}
+
+func (s Byte) Get() byte {
+	if s.nullValue != nil {
+		return s.nullValue.Byte
+	}
+
+	return 0
+}
+
+func (s *Byte) Value() (driver.Value, error) {
+	if s.nullValue != nil {
+		return s.nullValue.Value()
+	}
+
+	return nil, nil
+}
+
+func (s *Byte) Scan(value any) error {
+	if err := s.nullValue.Scan(value); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (s *Byte) UnmarshalJSON(bytes []byte) error {
+	s.nullValue = &sql.NullByte{
+		Byte:  0,
+		Valid: false,
+	}
+
+	if len(bytes) > 0 {
+		if err := json.Unmarshal(bytes, &s.nullValue.Byte); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (s Byte) MarshalJSON() ([]byte, error) {
+	if s.nullValue.Valid {
+		return json.Marshal(s.nullValue.Byte)
+	}
+
+	return json.Marshal(nil)
+}

go.mod

@@ -0,0 +1,3 @@
+module github.com/abdivasiyev/go-null
+
+go 1.18

time.go

@@ -0,0 +1,77 @@
+package null
+
+import (
+	"database/sql"
+	"database/sql/driver"
+	"encoding/json"
+	"fmt"
+	"time"
+)
+
+type Time struct {
+	nullValue *sql.NullTime
+}
+
+func (s Time) String() string {
+	if s.nullValue != nil {
+		return fmt.Sprint(s.nullValue.Time)
+	}
+
+	return ""
+}
+
+func NewTime(value time.Time) Time {
+	return Time{
+		nullValue: &sql.NullTime{
+			Valid: true,
+			Time:  value,
+		},
+	}
+}
+
+func (s Time) Get() time.Time {
+	if s.nullValue != nil {
+		return s.nullValue.Time
+	}
+
+	return time.Time{}
+}
+
+func (s *Time) Value() (driver.Value, error) {
+	if s.nullValue != nil {
+		return s.nullValue.Value()
+	}
+
+	return nil, nil
+}
+
+func (s *Time) Scan(value any) error {
+	if err := s.nullValue.Scan(value); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (s *Time) UnmarshalJSON(bytes []byte) error {
+	s.nullValue = &sql.NullTime{
+		Time:  time.Time{},
+		Valid: false,
+	}
+
+	if len(bytes) > 0 {
+		if err := json.Unmarshal(bytes, &s.nullValue.Time); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (s Time) MarshalJSON() ([]byte, error) {
+	if s.nullValue.Valid {
+		return json.Marshal(s.nullValue.Time)
+	}
+
+	return json.Marshal(nil)
+}