mirror of
https://github.com/rclone/rclone.git
synced 2024-12-23 15:38:57 +01:00
f78cd1e043
- Change rclone/fs interfaces to accept context.Context - Update interface implementations to use context.Context - Change top level usage to propagate context to lover level functions Context propagation is needed for stopping transfers and passing other request-scoped values.
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
package memtest
|
|
|
|
import (
|
|
"context"
|
|
"runtime"
|
|
"sync"
|
|
|
|
"github.com/ncw/rclone/cmd"
|
|
"github.com/ncw/rclone/fs"
|
|
"github.com/ncw/rclone/fs/operations"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func init() {
|
|
cmd.Root.AddCommand(commandDefintion)
|
|
}
|
|
|
|
var commandDefintion = &cobra.Command{
|
|
Use: "memtest remote:path",
|
|
Short: `Load all the objects at remote:path and report memory stats.`,
|
|
Hidden: true,
|
|
Run: func(command *cobra.Command, args []string) {
|
|
cmd.CheckArgs(1, 1, command, args)
|
|
fsrc := cmd.NewFsSrc(args)
|
|
cmd.Run(false, false, command, func() error {
|
|
ctx := context.Background()
|
|
objects, _, err := operations.Count(ctx, fsrc)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
objs := make([]fs.Object, 0, objects)
|
|
var before, after runtime.MemStats
|
|
runtime.GC()
|
|
runtime.ReadMemStats(&before)
|
|
var mu sync.Mutex
|
|
err = operations.ListFn(ctx, fsrc, func(o fs.Object) {
|
|
mu.Lock()
|
|
objs = append(objs, o)
|
|
mu.Unlock()
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
runtime.GC()
|
|
runtime.ReadMemStats(&after)
|
|
usedMemory := after.Alloc - before.Alloc
|
|
fs.Logf(nil, "%d objects took %d bytes, %.1f bytes/object", len(objs), usedMemory, float64(usedMemory)/float64(len(objs)))
|
|
fs.Logf(nil, "System memory changed from %d to %d bytes a change of %d bytes", before.Sys, after.Sys, after.Sys-before.Sys)
|
|
return nil
|
|
})
|
|
},
|
|
}
|