Add support for macaddr and macaddr8

Author: kyleconroy

macaddr.go

@@ -0,0 +1,66 @@
+package pqtype
+
+import (
+	"database/sql/driver"
+	"fmt"
+	"net"
+)
+
+type Macaddr struct {
+	Addr  net.HardwareAddr
+	Valid bool
+}
+
+func (dst *Macaddr) decodeText(src []byte) error {
+	if src == nil {
+		*dst = Macaddr{}
+		return nil
+	}
+
+	addr, err := net.ParseMAC(string(src))
+	if err != nil {
+		return err
+	}
+
+	*dst = Macaddr{Addr: addr, Valid: true}
+	return nil
+}
+
+func (src Macaddr) encodeText(buf []byte) ([]byte, error) {
+	if src.Valid {
+		return append(buf, src.Addr.String()...), nil
+	} else {
+		return nil, nil
+	}
+}
+
+// Scan implements the database/sql Scanner interface.
+func (dst *Macaddr) Scan(src interface{}) error {
+	if src == nil {
+		*dst = Macaddr{}
+		return nil
+	}
+
+	switch src := src.(type) {
+	case string:
+		return dst.decodeText([]byte(src))
+	case []byte:
+		srcCopy := make([]byte, len(src))
+		copy(srcCopy, src)
+		return dst.decodeText(srcCopy)
+	}
+
+	return fmt.Errorf("cannot scan %T", src)
+}
+
+// Value implements the database/sql/driver Valuer interface.
+func (src Macaddr) Value() (driver.Value, error) {
+	buf, err := src.encodeText(make([]byte, 0, 32))
+	if err != nil {
+		return nil, err
+	}
+	if buf == nil {
+		return nil, nil
+	}
+	return string(buf), err
+}

tests/macaddr_test.go

@@ -0,0 +1,133 @@
+package tests
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+	"github.com/tabbed/pqtype"
+)
+
+// https://www.postgresql.org/docs/current/datatype-net-types.html#DATATYPE-MACADDR
+func TestMacaddr(t *testing.T) {
+	for _, addr := range []struct {
+		input  string
+		output string
+	}{
+		{
+			"08:00:2b:01:02:03",
+			"08:00:2b:01:02:03",
+		},
+		{
+			"08-00-2b-01-02-03",
+			"08:00:2b:01:02:03",
+		},
+		{
+			"08002b:010203",
+			"08:00:2b:01:02:03",
+		},
+		{
+			"08002b-010203",
+			"08:00:2b:01:02:03",
+		},
+		{
+			"0800.2b01.0203",
+			"08:00:2b:01:02:03",
+		},
+		{
+			"0800-2b01-0203",
+			"08:00:2b:01:02:03",
+		},
+		{
+			"08002b010203",
+			"08:00:2b:01:02:03",
+		},
+	} {
+		addr = addr
+		t.Run(addr.input, func(t *testing.T) {
+			var cidr pqtype.Macaddr
+			if err := db.QueryRow(fmt.Sprintf(`SELECT '%s'::macaddr`, addr.input)).Scan(&cidr); err != nil {
+				t.Fatal(err)
+			}
+			if diff := cmp.Diff(true, cidr.Valid); diff != "" {
+				t.Errorf("valid mismatch (-want +got):\n%s", diff)
+			}
+			if diff := cmp.Diff(addr.output, cidr.Addr.String()); diff != "" {
+				t.Errorf("mismatch (-want +got):\n%s", diff)
+			}
+		})
+	}
+	t.Run("NULL", func(t *testing.T) {
+		var cidr pqtype.Macaddr
+		if err := db.QueryRow(fmt.Sprintf(`SELECT NULL::macaddr`)).Scan(&cidr); err != nil {
+			t.Fatal(err)
+		}
+		if diff := cmp.Diff(false, cidr.Valid); diff != "" {
+			t.Errorf("valid mismatch (-want +got):\n%s", diff)
+		}
+	})
+}
+
+// https://www.postgresql.org/docs/current/datatype-net-types.html#DATATYPE-MACADDR8
+func TestMacaddr8(t *testing.T) {
+	for _, addr := range []struct {
+		input  string
+		output string
+	}{
+		{
+			"08:00:2b:01:02:03:04:05",
+			"08:00:2b:01:02:03:04:05",
+		},
+		{
+			"08-00-2b-01-02-03-04-05",
+			"08:00:2b:01:02:03:04:05",
+		},
+		{
+			"08002b:0102030405",
+			"08:00:2b:01:02:03:04:05",
+		},
+		{
+			"08002b-0102030405",
+			"08:00:2b:01:02:03:04:05",
+		},
+		{
+			"0800.2b01.0203.0405",
+			"08:00:2b:01:02:03:04:05",
+		},
+		{
+			"0800-2b01-0203-0405",
+			"08:00:2b:01:02:03:04:05",
+		},
+		{
+			"08002b01:02030405",
+			"08:00:2b:01:02:03:04:05",
+		},
+		{
+			"08002b0102030405",
+			"08:00:2b:01:02:03:04:05",
+		},
+	} {
+		addr = addr
+		t.Run(addr.input, func(t *testing.T) {
+			var cidr pqtype.Macaddr
+			if err := db.QueryRow(fmt.Sprintf(`SELECT '%s'::macaddr8`, addr.input)).Scan(&cidr); err != nil {
+				t.Fatal(err)
+			}
+			if diff := cmp.Diff(true, cidr.Valid); diff != "" {
+				t.Errorf("valid mismatch (-want +got):\n%s", diff)
+			}
+			if diff := cmp.Diff(addr.output, cidr.Addr.String()); diff != "" {
+				t.Errorf("mismatch (-want +got):\n%s", diff)
+			}
+		})
+	}
+	t.Run("NULL", func(t *testing.T) {
+		var cidr pqtype.Macaddr
+		if err := db.QueryRow(fmt.Sprintf(`SELECT NULL::macaddr8`)).Scan(&cidr); err != nil {
+			t.Fatal(err)
+		}
+		if diff := cmp.Diff(false, cidr.Valid); diff != "" {
+			t.Errorf("valid mismatch (-want +got):\n%s", diff)
+		}
+	})
+}