feat: update marshal (#20)

xiaojiaoyu100 · Nov 6, 2020 · 7176808 · 7176808
1 parent 40e4e20
commit 7176808
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/sonyflake/id.go b/sonyflake/id.go
@@ -1,7 +1,6 @@
 package sonyflake
 
 import (
-	"fmt"
 	"strconv"
 	"time"
 )
@@ -16,12 +15,12 @@ func IDFromString(s string) (ID, error) {
 	return ID(id), nil
 }
 
-func (i *ID) MarshalJSON() ([]byte, error) {
-	return []byte(fmt.Sprintf("\"%v\"", i)), nil
+func (i *ID) MarshalText() ([]byte, error) {
+	return []byte(strconv.FormatInt(int64(*i), 10)), nil
 }
 
-func (i *ID) UnmarshalJSON(b []byte) error {
-	id, err := strconv.ParseInt(string(b[1:len(b)-1]), 10, 64)
+func (i *ID) UnmarshalText(b []byte) error {
+	id, err := strconv.ParseInt(string(b), 10, 64)
 	if err != nil {
 		return err
 	}