Improve memory allocations when data.Doc cloning

This commit is contained in:
Tobie Morgan Hitchcock 2019-11-20 13:10:34 +00:00
parent 21e05b2e27
commit 2e1b699a87
2 changed files with 64 additions and 1 deletions

View file

@ -22,6 +22,7 @@ import (
"encoding/json"
"github.com/abcum/surreal/util/dupe"
"github.com/abcum/surreal/util/pack"
)
@ -66,7 +67,7 @@ func (d *Doc) Data() interface{} {
// Copy returns a duplicated copy of the internal data object.
func (d *Doc) Copy() *Doc {
return new(Doc).Decode(d.Encode())
return Consume(dupe.Duplicate(d.data))
}
// Encode encodes the data object to a byte slice.

62
util/dupe/dupe.go Normal file
View file

@ -0,0 +1,62 @@
// Copyright © 2016 Abcum Ltd
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package dupe
type Copyable interface {
Copy() interface{}
}
type Cloneable interface {
Clone() interface{}
}
func Duplicate(v interface{}) interface{} {
switch x := v.(type) {
case map[string]interface{}:
out := make(map[string]interface{}, len(x))
for k, v := range x {
switch y := v.(type) {
case map[string]interface{}:
out[k] = Duplicate(y)
case []interface{}:
out[k] = Duplicate(y)
default:
out[k] = y
}
}
return out
case []interface{}:
out := make([]interface{}, len(x))
for k, v := range x {
switch y := v.(type) {
case map[string]interface{}:
out[k] = Duplicate(y)
case []interface{}:
out[k] = Duplicate(y)
default:
out[k] = y
}
}
return out
case Cloneable:
return x.Clone()
case Copyable:
return x.Copy()
default:
return x
}
}