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"
|
"encoding/json"
|
||||||
|
|
||||||
|
"github.com/abcum/surreal/util/dupe"
|
||||||
"github.com/abcum/surreal/util/pack"
|
"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.
|
// Copy returns a duplicated copy of the internal data object.
|
||||||
func (d *Doc) Copy() *Doc {
|
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.
|
// 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