13 _ "github.com/anacrolix/envpprof"
14 "github.com/anacrolix/squirrel"
15 "github.com/anacrolix/torrent/storage"
16 test_storage "github.com/anacrolix/torrent/storage/test"
17 "github.com/anacrolix/torrent/test"
18 "github.com/dustin/go-humanize"
19 qt "github.com/frankban/quicktest"
22 func TestLeecherStorage(t *testing.T) {
23 test.TestLeecherStorage(t, test.LeecherStorageTestCase{
25 func(s string) storage.ClientImplCloser {
26 path := filepath.Join(s, "sqlite3.db")
27 var opts NewDirectStorageOpts
29 cl, err := NewDirectStorage(opts)
39 func BenchmarkMarkComplete(b *testing.B) {
40 const pieceSize = test_storage.DefaultPieceSize
41 const noTriggers = false
42 const noCacheBlobs = false
43 var capacity int64 = test_storage.DefaultNumPieces * pieceSize / 2
45 // Since we won't push out old pieces, we have to mark them incomplete manually.
48 runBench := func(b *testing.B, ci storage.ClientImpl) {
49 test_storage.BenchmarkPieceMarkComplete(b, ci, pieceSize, test_storage.DefaultNumPieces, capacity)
52 b.Run("CustomDirect", func(b *testing.B) {
53 var opts squirrel.NewCacheOpts
54 opts.Capacity = capacity
55 opts.NoTriggers = noTriggers
56 opts.NoCacheBlobs = noCacheBlobs
57 benchOpts := func(b *testing.B) {
58 opts.Path = filepath.Join(b.TempDir(), "storage.db")
59 ci, err := NewDirectStorage(opts)
60 c.Assert(err, qt.IsNil)
64 b.Run("Default", benchOpts)
66 for _, memory := range []bool{false, true} {
67 b.Run(fmt.Sprintf("Memory=%v", memory), func(b *testing.B) {
68 b.Run("Direct", func(b *testing.B) {
69 var opts NewDirectStorageOpts
71 opts.Capacity = capacity
72 // opts.GcBlobs = true
73 opts.BlobFlushInterval = time.Second
74 opts.NoTriggers = noTriggers
75 opts.NoCacheBlobs = noCacheBlobs
76 directBench := func(b *testing.B) {
77 opts.Path = filepath.Join(b.TempDir(), "storage.db")
78 ci, err := NewDirectStorage(opts)
79 var ujm squirrel.ErrUnexpectedJournalMode
80 if errors.As(err, &ujm) {
81 b.Skipf("setting journal mode %q: %v", opts.SetJournalMode, err)
83 c.Assert(err, qt.IsNil)
87 for _, journalMode := range []string{"", "wal", "off", "truncate", "delete", "persist", "memory"} {
88 opts.SetJournalMode = journalMode
89 b.Run("JournalMode="+journalMode, func(b *testing.B) {
90 for _, mmapSize := range []int64{-1} {
91 if memory && mmapSize >= 0 {
94 b.Run(fmt.Sprintf("MmapSize=%s", func() string {
98 return humanize.IBytes(uint64(mmapSize))
100 }()), func(b *testing.B) {
101 opts.MmapSize = mmapSize
102 opts.MmapSizeOk = true