40 lines
1.0 KiB
Go
40 lines
1.0 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 redblacktree
|
|
|
|
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 list's elements.
|
|
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 list's elements 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
|
|
}
|