client.go | 6 ++---- t.go | 5 ++++- diff --git a/client.go b/client.go index fe83190072738e63f9a63670befe47b97790fd30..0de33272c20fcb842da268c888ef0861debd3fc5 100644 --- a/client.go +++ b/client.go @@ -1258,15 +1258,13 @@ } return t.MergeSpec(TorrentSpecFromMetaInfo(&mi)) } -func (cl *Client) dropTorrent(infoHash metainfo.Hash) (err error) { +func (cl *Client) dropTorrent(infoHash metainfo.Hash, wg *sync.WaitGroup) (err error) { t, ok := cl.torrents[infoHash] if !ok { err = fmt.Errorf("no such torrent") return } - var wg sync.WaitGroup - defer wg.Wait() - err = t.close(&wg) + err = t.close(wg) if err != nil { panic(err) } diff --git a/t.go b/t.go index ebb328835deaa3cfa8f42ba7346b9d5ffcc4e4a4..153264e928d5c62823c0ff222bb65f39f82dc7be 100644 --- a/t.go +++ b/t.go @@ -5,6 +5,7 @@ "strconv" "strings" "github.com/anacrolix/missinggo/pubsub" + "github.com/anacrolix/sync" "github.com/anacrolix/torrent/metainfo" ) @@ -95,9 +96,11 @@ // Drop the torrent from the client, and close it. It's always safe to do // this. No data corruption can, or should occur to either the torrent's data, // or connected peers. func (t *Torrent) Drop() { + var wg sync.WaitGroup + defer wg.Wait() t.cl.lock() defer t.cl.unlock() - t.cl.dropTorrent(t.infoHash) + t.cl.dropTorrent(t.infoHash, &wg) } // Number of bytes of the entire torrent we have completed. This is the sum of