Merge pull request #4 from gaze-network/feat/add-marshalers

Add text and json marshalers and unmarshalers

Author: gazenw

uint128.go

@@ -525,3 +525,38 @@ func FromString(s string) (u Uint128, err error) {
 	_, err = fmt.Sscan(s, &u)
 	return
 }
+
+// MarshalText implements encoding.TextMarshaler.
+func (u Uint128) MarshalText() ([]byte, error) {
+	return []byte(u.String()), nil
+}
+
+// UnmarshalText implements encoding.TextUnmarshaler.
+func (u *Uint128) UnmarshalText(text []byte) error {
+	v, err := FromString(string(text))
+	if err != nil {
+		return err
+	}
+	*u = v
+	return nil
+}
+
+// MarshalJSON implements json.Marshaler.
+func (u Uint128) MarshalJSON() ([]byte, error) {
+	return []byte(`"` + u.String() + `"`), nil
+}
+
+// UnmarshalJSON implements json.Unmarshaler.
+func (u *Uint128) UnmarshalJSON(data []byte) error {
+	// data may be a JSON string or a JSON number
+	// check if data is a JSON string first
+	if len(data) > 0 && data[0] == '"' && data[len(data)-1] == '"' {
+		data = data[1 : len(data)-1]
+	}
+	v, err := FromString(string(data))
+	if err != nil {
+		return err
+	}
+	*u = v
+	return nil
+}

uint128_test.go

@@ -524,3 +524,67 @@ func TestFromBytesBE(t *testing.T) {
 		t.Fatalf("mismatch:\n%v !=\n%v", u1, u2)
 	}
 }
+
+func TestMarshal(t *testing.T) {
+	number := "42540766411282592856903984951653826562"
+
+	u, err := FromString(number)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	btext, err := u.MarshalText()
+	if err != nil {
+		t.Fatal(err)
+	}
+	bjson, err := u.MarshalJSON()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	expectedText := number
+	if string(btext) != expectedText {
+		t.Fatalf("text mismatch:\n%v !=\n%v", string(btext), expectedText)
+	}
+	expectedJSON := `"` + number + `"`
+	if string(bjson) != expectedJSON {
+		t.Fatalf("json mismatch:\n%v !=\n%v", string(bjson), expectedJSON)
+	}
+}
+
+func TestUnmarshal(t *testing.T) {
+	number := "42540766411282592856903984951653826562"
+
+	var u Uint128
+	err := u.UnmarshalText([]byte(number))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	expected, err := FromString(number)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if u != expected {
+		t.Fatalf("text mismatch:\n%v !=\n%v", u, expected)
+	}
+
+	err = u.UnmarshalText([]byte("invalid"))
+	if err == nil {
+		t.Fatal("expected error")
+	}
+
+	err = u.UnmarshalJSON([]byte(`"` + number + `"`))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if u != expected {
+		t.Fatalf("json mismatch:\n%v !=\n%v", u, expected)
+	}
+
+	err = u.UnmarshalJSON([]byte(`"invalid"`))
+	if err == nil {
+		t.Fatal("expected error")
+	}
+}