fix unmarshal null character string to return 0 (#25)

* feat: add sonyflake * fix: unmarshal 0
xiaojiaoyu100 · Apr 9, 2021 · 4678acb · 4678acb
1 parent b557172
commit 4678acb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sonyflake/id.go b/sonyflake/id.go
@@ -24,6 +24,10 @@ func (i ID) MarshalText() ([]byte, error) {
 
 // UnmarshalText ...
 func (i *ID) UnmarshalText(b []byte) error {
+	if len(string(b)) == 0 {
+		*i = ID(0)
+		return nil
+	}
 	id, err := strconv.ParseInt(string(b), 10, 64)
 	if err != nil {
 		return err