diff --git a/v2/netmap/json.go b/v2/netmap/json.go index 7210501..3445f05 100644 --- a/v2/netmap/json.go +++ b/v2/netmap/json.go @@ -124,3 +124,23 @@ func (ni *NodeInfo) UnmarshalJSON(data []byte) error { return nil } + +func (i *NetworkInfo) MarshalJSON() ([]byte, error) { + return protojson.MarshalOptions{ + EmitUnpopulated: true, + }.Marshal( + NetworkInfoToGRPCMessage(i), + ) +} + +func (i *NetworkInfo) UnmarshalJSON(data []byte) error { + msg := new(netmap.NetworkInfo) + + if err := protojson.Unmarshal(data, msg); err != nil { + return err + } + + *i = *NetworkInfoFromGRPCMessage(msg) + + return nil +} diff --git a/v2/netmap/json_test.go b/v2/netmap/json_test.go index 561dbf1..3f01b03 100644 --- a/v2/netmap/json_test.go +++ b/v2/netmap/json_test.go @@ -66,3 +66,15 @@ func TestNodeInfoJSON(t *testing.T) { require.Equal(t, i, i2) } + +func TestNetworkInfoJSON(t *testing.T) { + i := generateNetworkInfo() + + data, err := i.MarshalJSON() + require.NoError(t, err) + + i2 := new(netmap.NetworkInfo) + require.NoError(t, i2.UnmarshalJSON(data)) + + require.Equal(t, i, i2) +}