Mirror/mochi
1
0
Fork 0
mirror of https://github.com/sot-tech/mochi.git synced 2026-07-16 21:28:11 -07:00
Code Issues Packages Projects Releases Wiki Activity

add helper methods to get IPv4 and IPv6 peer

Browse Source
This commit is contained in:
Leo Balduf
2016-04-18 15:49:36 -04:00
parent 468eefee57
commit 9f229c4ab6
2 changed files with 24 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/store/middleware/response/response.go
+2 -4
View File
@@ -27,16 +27,14 @@ func (f FailedToRetrievePeers) Error() string { return string(f) }
func responseAnnounceClient(next tracker.AnnounceHandler) tracker.AnnounceHandler {
return func(cfg *chihaya.TrackerConfig, req *chihaya.AnnounceRequest, resp *chihaya.AnnounceResponse) (err error) {
storage := store.MustGetStore()
peer4 := chihaya.Peer{ID: req.PeerID, IP: req.IPv4, Port: req.Port}
peer6 := chihaya.Peer{ID: req.PeerID, IP: req.IPv6, Port: req.Port}
resp.MinInterval = cfg.MinAnnounceInterval
resp.Compact = req.Compact
resp.Complete = int32(storage.NumSeeders(req.InfoHash))
resp.Incomplete = int32(storage.NumLeechers(req.InfoHash))
resp.IPv4Peers, resp.IPv6Peers, err = storage.AnnouncePeers(req.InfoHash, req.Left == 0, int(req.NumWant), peer4, peer6)
resp.IPv4Peers, resp.IPv6Peers, err = storage.AnnouncePeers(req.InfoHash, req.Left == 0, int(req.NumWant), req.Peer4(), req.Peer6())
if err != nil {
return err.(FailedToRetrievePeers)
return FailedToRetrievePeers(err.Error())
}
return next(cfg, req, resp)