storage: add ScrapeSwarm method

This commit is contained in:
Leo Balduf 2016-09-08 09:33:58 -04:00
parent 0cdbf9b416
commit b1576da355

View file

@ -12,11 +12,12 @@ var ErrResourceDoesNotExist = bittorrent.ClientError("resource does not exist")
// PeerStore is an interface that abstracts the interactions of storing and // PeerStore is an interface that abstracts the interactions of storing and
// manipulating Peers such that it can be implemented for various data stores. // manipulating Peers such that it can be implemented for various data stores.
type PeerStore interface { type PeerStore interface {
// PutSeeder adds a Seeder to the Swarm identified by the provided infoHash. // PutSeeder adds a Seeder to the Swarm identified by the provided
// infoHash.
PutSeeder(infoHash bittorrent.InfoHash, p bittorrent.Peer) error PutSeeder(infoHash bittorrent.InfoHash, p bittorrent.Peer) error
// DeleteSeeder removes a Seeder from the Swarm identified by the provided // DeleteSeeder removes a Seeder from the Swarm identified by the
// infoHash. // provided infoHash.
// //
// If the Swarm or Peer does not exist, this function should return // If the Swarm or Peer does not exist, this function should return
// ErrResourceDoesNotExist. // ErrResourceDoesNotExist.
@ -26,32 +27,44 @@ type PeerStore interface {
// infoHash. // infoHash.
PutLeecher(infoHash bittorrent.InfoHash, p bittorrent.Peer) error PutLeecher(infoHash bittorrent.InfoHash, p bittorrent.Peer) error
// DeleteLeecher removes a Leecher from the Swarm identified by the provided // DeleteLeecher removes a Leecher from the Swarm identified by the
// infoHash. // provided infoHash.
// //
// If the Swarm or Peer does not exist, this function should return // If the Swarm or Peer does not exist, this function should return
// ErrResourceDoesNotExist. // ErrResourceDoesNotExist.
DeleteLeecher(infoHash bittorrent.InfoHash, p bittorrent.Peer) error DeleteLeecher(infoHash bittorrent.InfoHash, p bittorrent.Peer) error
// GraduateLeecher promotes a Leecher to a Seeder in the Swarm identified by // GraduateLeecher promotes a Leecher to a Seeder in the Swarm
// the provided infoHash. // identified by the provided infoHash.
// //
// If the given Peer is not present as a Leecher, add the Peer as a Seeder // If the given Peer is not present as a Leecher, add the Peer as a
// and return no error. // Seeder and return no error.
GraduateLeecher(infoHash bittorrent.InfoHash, p bittorrent.Peer) error GraduateLeecher(infoHash bittorrent.InfoHash, p bittorrent.Peer) error
// AnnouncePeers is a best effort attempt to return Peers from the Swarm // AnnouncePeers is a best effort attempt to return Peers from the Swarm
// identified by the provided infoHash. The returned Peers are required to be // identified by the provided infoHash. The returned Peers are required
// either all IPv4 or all IPv6. // to be either all IPv4 or all IPv6.
// //
// The returned Peers should strive be: // The returned Peers should strive be:
// - as close to length equal to numWant as possible without going over // - as close to length equal to numWant as possible without going over
// - all IPv4 or all IPv6 depending on the provided peer // - all IPv4 or all IPv6 depending on the provided peer
// - if seeder is true, should ideally return more leechers than seeders // - if seeder is true, should ideally return more leechers than seeders
// - if seeder is false, should ideally return more seeders than leechers // - if seeder is false, should ideally return more seeders than
// leechers
AnnouncePeers(infoHash bittorrent.InfoHash, seeder bool, numWant int, p bittorrent.Peer) (peers []bittorrent.Peer, err error) AnnouncePeers(infoHash bittorrent.InfoHash, seeder bool, numWant int, p bittorrent.Peer) (peers []bittorrent.Peer, err error)
// Stopper is an interface that expects a Stop method to stops the PeerStore. // ScrapeSwarm returns information required to answer a scrape request
// about a swarm identified by the given infohash.
// The v6 flag indicates whether or not the IPv6 swarm should be
// scraped.
// The Complete and Incomplete fields of the Scrape must be filled,
// filling the Snatches field is optional.
// If the infohash is unknown to the PeerStore, an empty Scrape is
// returned.
ScrapeSwarm(infoHash bittorrent.InfoHash, v6 bool) bittorrent.Scrape
// Stopper is an interface that expects a Stop method to stop the
// PeerStore.
// For more details see the documentation in the stopper package. // For more details see the documentation in the stopper package.
stopper.Stopper stopper.Stopper
} }