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

Merge pull request #153 from mrd0ll4r/middleware-smallfix

Browse Source 
hotfix: initialized scrape map
This commit is contained in:
Jimmy Zelinskie
2016-04-02 20:31:40 -04:00
parent 52f6fb664e 9c1168746a
commit d5c00c24b1
5 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server/http/writer.go
View File

@@ -71,7 +71,7 @@ func writeAnnounceResponse(w http.ResponseWriter, resp *chihaya.AnnounceResponse
func writeScrapeResponse(w http.ResponseWriter, resp *chihaya.ScrapeResponse) error {
filesDict := bencode.NewDict()
for infohash, scrape := range resp.Files {
filesDict[infohash] = bencode.Dict{
filesDict[string(infohash)] = bencode.Dict{
"complete": scrape.Complete,
"incomplete": scrape.Incomplete,
}

2
server/store/middleware/response/response.go
View File

@@ -47,7 +47,7 @@ func responseScrapeClient(next tracker.ScrapeHandler) tracker.ScrapeHandler {
return func(cfg *chihaya.TrackerConfig, req *chihaya.ScrapeRequest, resp *chihaya.ScrapeResponse) (err error) {
storage := store.MustGetStore()
for _, infoHash := range req.InfoHashes {
resp.Files[string(infoHash)] = chihaya.Scrape{
resp.Files[infoHash] = chihaya.Scrape{
Complete: int32(storage.NumSeeders(infoHash)),
Incomplete: int32(storage.NumLeechers(infoHash)),
}