jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-06-10 03:57:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: cache errored request

Browse Source
This commit is contained in:
Simon Ding
2024-08-13 11:05:46 +08:00
parent 32914344d1
commit b487c81865
2 changed files with 22 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/torznab/cache.go
View File

@@ -5,4 +5,4 @@ import (
"time" "time"
) )
var cc = cache.NewCache[string, Response](time.Minute * 30) var cc = cache.NewCache[string, *Response](time.Minute * 30)

17
pkg/torznab/torznab.go
View File

@@ -135,6 +135,18 @@ func Search(indexer *db.TorznabInfo, keyWord string) ([]Result, error) {
cacheRes, ok := cc.Get(key) cacheRes, ok := cc.Get(key)
if !ok { if !ok {
res, err := doRequest(req)
if err != nil {
cc.Set(key, &Response{})
return nil, errors.Wrap(err, "do http request")
}
cacheRes = res
cc.Set(key, cacheRes)
}
return cacheRes.ToResults(indexer), nil
}
func doRequest(req *http.Request) (*Response, error) {
resp, err := http.DefaultClient.Do(req) resp, err := http.DefaultClient.Do(req)
if err != nil { if err != nil {
return nil, errors.Wrap(err, "do http") return nil, errors.Wrap(err, "do http")
@@ -149,10 +161,7 @@ func Search(indexer *db.TorznabInfo, keyWord string) ([]Result, error) {
if err != nil { if err != nil {
return nil, errors.Wrapf(err, "xml unmarshal data: %v", string(data)) return nil, errors.Wrapf(err, "xml unmarshal data: %v", string(data))
} }
cacheRes = res return &res, nil
cc.Set(key, cacheRes)
}
return cacheRes.ToResults(indexer), nil
} }
type Result struct { type Result struct {