mirror of https://github.com/emirpasic/gods
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
// Copyright (c) 2015, Emir Pasic. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package btree
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/emirpasic/gods/containers"
|
|
"github.com/emirpasic/gods/utils"
|
|
)
|
|
|
|
func assertSerializationImplementation() {
|
|
var _ containers.JSONSerializer = (*Tree)(nil)
|
|
var _ containers.JSONDeserializer = (*Tree)(nil)
|
|
}
|
|
|
|
// ToJSON outputs the JSON representation of the tree.
|
|
func (tree *Tree) ToJSON() ([]byte, error) {
|
|
elements := make(map[string]interface{})
|
|
it := tree.Iterator()
|
|
for it.Next() {
|
|
elements[utils.ToString(it.Key())] = it.Value()
|
|
}
|
|
return json.Marshal(&elements)
|
|
}
|
|
|
|
// FromJSON populates the tree from the input JSON representation.
|
|
func (tree *Tree) FromJSON(data []byte) error {
|
|
elements := make(map[string]interface{})
|
|
err := json.Unmarshal(data, &elements)
|
|
if err == nil {
|
|
tree.Clear()
|
|
for key, value := range elements {
|
|
tree.Put(key, value)
|
|
}
|
|
}
|
|
return err
|
|
}
|
|
|
|
// UnmarshalJSON @implements json.Unmarshaler
|
|
func (tree *Tree) UnmarshalJSON(bytes []byte) error {
|
|
return tree.FromJSON(bytes)
|
|
}
|
|
|
|
// MarshalJSON @implements json.Marshaler
|
|
func (tree *Tree) MarshalJSON() ([]byte, error) {
|
|
return tree.ToJSON()
|
|
}
|