9 _ "github.com/anacrolix/envpprof"
10 "github.com/anacrolix/squirrel"
11 "github.com/dustin/go-humanize"
12 qt "github.com/frankban/quicktest"
16 "github.com/anacrolix/torrent/storage"
17 test_storage "github.com/anacrolix/torrent/storage/test"
18 "github.com/anacrolix/torrent/test"
21 func TestLeecherStorage(t *testing.T) {
22 test.TestLeecherStorage(t, test.LeecherStorageTestCase{
24 func(s string) storage.ClientImplCloser {
25 path := filepath.Join(s, "sqlite3.db")
26 var opts NewDirectStorageOpts
28 cl, err := NewDirectStorage(opts)
38 func BenchmarkMarkComplete(b *testing.B) {
39 const pieceSize = test_storage.DefaultPieceSize
40 const noTriggers = false
41 var capacity int64 = test_storage.DefaultNumPieces * pieceSize / 2
43 // Since we won't push out old pieces, we have to mark them incomplete manually.
46 runBench := func(b *testing.B, ci storage.ClientImpl) {
47 test_storage.BenchmarkPieceMarkComplete(b, ci, pieceSize, test_storage.DefaultNumPieces, capacity)
50 b.Run("CustomDirect", func(b *testing.B) {
51 var opts squirrel.NewCacheOpts
52 opts.Capacity = capacity
53 opts.NoTriggers = noTriggers
54 benchOpts := func(b *testing.B) {
55 opts.Path = filepath.Join(b.TempDir(), "storage.db")
56 ci, err := NewDirectStorage(opts)
57 c.Assert(err, qt.IsNil)
61 b.Run("Default", benchOpts)
63 for _, memory := range []bool{false, true} {
64 b.Run(fmt.Sprintf("Memory=%v", memory), func(b *testing.B) {
65 b.Run("Direct", func(b *testing.B) {
66 var opts NewDirectStorageOpts
68 opts.Capacity = capacity
69 opts.NoTriggers = noTriggers
70 directBench := func(b *testing.B) {
71 opts.Path = filepath.Join(b.TempDir(), "storage.db")
72 ci, err := NewDirectStorage(opts)
73 var ujm squirrel.ErrUnexpectedJournalMode
74 if errors.As(err, &ujm) {
75 b.Skipf("setting journal mode %q: %v", opts.SetJournalMode, err)
77 c.Assert(err, qt.IsNil)
81 for _, journalMode := range []string{"", "wal", "off", "truncate", "delete", "persist", "memory"} {
82 opts.SetJournalMode = journalMode
83 b.Run("JournalMode="+journalMode, func(b *testing.B) {
84 for _, mmapSize := range []int64{-1} {
85 if memory && mmapSize >= 0 {
88 b.Run(fmt.Sprintf("MmapSize=%s", func() string {
92 return humanize.IBytes(uint64(mmapSize))
94 }()), func(b *testing.B) {
95 opts.MmapSize = mmapSize
96 opts.MmapSizeOk = true