extern/rclone
1
0
Fork 0
mirror of https://github.com/rclone/rclone.git synced 2024-12-23 07:29:35 +01:00
Code Issues Packages Projects Releases Wiki Activity

sync: replace container/heap with github.com/aalpar/deheap

Browse Source
This commit is contained in:
Nick Craig-Wood 2020-03-13 17:10:06 +00:00
parent 84369286df
commit 1e3d899db8
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
fs/sync/pipe.go
View File

@ -1,11 +1,11 @@
package sync
import (
"container/heap"
"context"
"strings"
"sync"
"github.com/aalpar/deheap"
"github.com/pkg/errors"
"github.com/rclone/rclone/fs"
"github.com/rclone/rclone/fs/fserrors"
@ -38,7 +38,7 @@ func newPipe(orderBy string, stats func(items int, totalSize int64), maxBacklog
less: less,
}
if p.less != nil {
heap.Init(p)
deheap.Init(p)
}
return p, nil
}
@ -73,9 +73,6 @@ func (p *pipe) Pop() interface{} {
return item
}
// Check interface satisfied
var _ heap.Interface = (*pipe)(nil)
// Put an pair into the pipe
//
// It returns ok = false if the context was cancelled
@ -90,7 +87,7 @@ func (p *pipe) Put(ctx context.Context, pair fs.ObjectPair) (ok bool) {
// no order-by
p.queue = append(p.queue, pair)
} else {
heap.Push(p, pair)
deheap.Push(p, pair)
}
size := pair.Src.Size()
if size > 0 {
@ -129,7 +126,7 @@ func (p *pipe) Get(ctx context.Context) (pair fs.ObjectPair, ok bool) {
p.queue[0] = fs.ObjectPair{} // avoid memory leak
p.queue = p.queue[1:]
} else {
pair = heap.Pop(p).(fs.ObjectPair)
pair = deheap.Pop(p).(fs.ObjectPair)
}
size := pair.Src.Size()
if size > 0 {

4
fs/sync/pipe_test.go
View File

@ -1,6 +1,7 @@
package sync
import (
"container/heap"
"context"
"sync"
"sync/atomic"
@ -12,6 +13,9 @@ import (
"github.com/stretchr/testify/require"
)
// Check interface satisfied
var _ heap.Interface = (*pipe)(nil)
func TestPipe(t *testing.T) {
var queueLength int
var queueSize int64