Improve memory allocations when data.Doc cloning
This commit is contained in:
parent
21e05b2e27
commit
2e1b699a87
2 changed files with 64 additions and 1 deletions
|
@ -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
62
util/dupe/dupe.go
Normal 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
|
||||
}
|
||||
|
||||
}
|
Loading…
Reference in a new issue