core: types: implements UnmarshalJSON from DKGMasterPublicKey (#223)

1 files changed, 8 insertions, 0 deletions

diff --git a/core/types/dkg.go b/core/types/dkg.go
index c761504..c13740b 100644
--- a/core/types/dkg.go
+++ b/core/types/dkg.go
@@ -18,6 +18,7 @@
 package types
 
 import (
+	"encoding/json"
 	"fmt"
 
 	"github.com/dexon-foundation/dexon-consensus-core/common"
@@ -59,6 +60,13 @@ func NewDKGMasterPublicKey() *DKGMasterPublicKey {
 	}
 }
 
+// UnmarshalJSON implements json.Unmarshaller.
+func (d *DKGMasterPublicKey) UnmarshalJSON(data []byte) error {
+	type innertDKGMasterPublicKey DKGMasterPublicKey
+	d.PublicKeyShares = *dkg.NewEmptyPublicKeyShares()
+	return json.Unmarshal(data, (*innertDKGMasterPublicKey)(d))
+}
+
 // DKGComplaint describe a complaint in DKG protocol.
 type DKGComplaint struct {
 	ProposerID   NodeID           `json:"proposer_id"`