Custom JSON marshal/unmarshal

January 28, 2023

Name: Moch Lutfi
Twitter: @kaptenupi

Problem

Using Go we can have some freedom to use custom data type and customize the way of data will be encoded or decoded. Today we will learn about reading custom data with unique behavior.

I have json string like below, the requirement is reading price number as integer even though in json representative stored as string. How to solve this?

{"price":"6550","code":"ASII"}

Solution

In simple ways we have 2 solution:

Using intermediate struct to marshal/unmarshal that act as data transfer object then transform to struct with price type int64.

// marshal/unmarshal json using DtoPrice
type DtoTransaction struct {
	Price string `json:"price"`
	Code  string `json:"code"`     
}
// then convert to this struct
type Transaction struct {
	Price int64 
	Code  string
}

Custom JSON marshal/unmarshal

Create custom type StrInt64 with type alias int64
Implement MarshalJSON and UnmarshalJSON to manage read string as int64 and vise versa.

Let's digging more in the code.

Function MarshalJSON is quite simple we need to convert as string instead put value directly
Function UnmarshalJSON have 3 steps
- Unmarshall as string
- Convert string to int64
- Type casting from int64 to StrInt64

intstring.go

instring_test.go

package customtype
import (
	"encoding/json"
	"fmt"
	"strconv"
)
// StrInt64 is a custom type to easily parse integers in string format
type StrInt64 int64
// MarshalJSON encodes the value into string
func (i StrInt64) MarshalJSON() ([]byte, error) {
	return json.Marshal(fmt.Sprintf("%d", i))
}
// UnmarshalJSON parse encoded string data into int64
func (i *StrInt64) UnmarshalJSON(data []byte) (err error) {
	var str string
	if err := json.Unmarshal(data, &str); err != nil {
		return err
	}
	x, err := strconv.ParseInt(str, 10, 64)
	if err != nil {
		return err
	}
	*i = StrInt64(x)
	return nil
}

Prove it yourself if the implementation above is correct at https://go.dev/play/p/cnx_KdH3e2n (opens in a new tab)

main.go

package main
import (
	"encoding/json"
	"fmt"
	"strconv"
)
// StrInt64 is a custom type to easily parse integers in string format
type StrInt64 int64
// MarshalJSON encodes the value into string
func (i StrInt64) MarshalJSON() ([]byte, error) {
	return json.Marshal(fmt.Sprintf("%d", i))
}
// UnmarshalJSON parse encoded string data into int64
func (i *StrInt64) UnmarshalJSON(data []byte) (err error) {
	var str string
	if err := json.Unmarshal(data, &str); err != nil {
		return err
	}
	x, err := strconv.ParseInt(str, 10, 64)
	if err != nil {
		return err
	}
	*i = StrInt64(x)
	return nil
}
type Transaction struct {
	Price StrInt64 `json:"price"`
	Code  string   `json:"code"`
}
func main() {
	var t Transaction
	json.Unmarshal([]byte(`{"price":"6550","code":"ASII"}`), &t)
	t.Price = t.Price * 2
	fmt.Println(t)
	s, _ := json.Marshal(t)
	fmt.Println(string(s))
}
// Output
// {13100 ASII}
// {"price":"13100","code":"ASII"}

Check symmetric binary tree Rotate Matrix