hpz937/zurg
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Do not verify if fresh link

Browse Source
This commit is contained in:
Ben Adrian Sarmiento
2024-08-26 10:22:37 +02:00
parent 78be877efe
commit 83d928a540
2 changed files with 4 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pkg/http/client.go
View File

@@ -337,10 +337,6 @@ func (r *HTTPClient) VerifyLink(link string) error {
if err != nil { if err != nil {
return err return err
} }
timeout := time.Duration(r.timeoutSecs) * time.Second
ctx, cancel := context.WithTimeout(context.Background(), timeout)
defer cancel()
req = req.WithContext(ctx)
resp, err := r.Do(req) resp, err := r.Do(req)
if err != nil { if err != nil {
return err return err

21
pkg/realdebrid/api.go
View File

@@ -107,18 +107,7 @@ func (rd *RealDebrid) UnrestrictAndVerify(link string) (*Download, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
unrestrictCache.Set(link, download) unrestrictCache.Set(link, download)
rd.verifiedLinks.Remove(download.ID)
err = rd.downloadClient.VerifyLink(download.Download)
if utils.IsBytesLimitReached(err) {
rd.TokenManager.SetTokenAsExpired(token, "bandwidth limit exceeded")
continue
} else if err != nil {
return nil, err
}
rd.verifiedLinks.Set(download.ID, now+DOWNLOAD_LINK_EXPIRY) rd.verifiedLinks.Set(download.ID, now+DOWNLOAD_LINK_EXPIRY)
return download, err return download, err
@@ -433,7 +422,6 @@ func (rd *RealDebrid) MonitorExpiredTokens() {
expiredTokens := rd.TokenManager.GetExpiredTokens() expiredTokens := rd.TokenManager.GetExpiredTokens()
for _, token := range expiredTokens { for _, token := range expiredTokens {
unrestrictCache, _ := rd.UnrestrictCache.Get(token) unrestrictCache, _ := rd.UnrestrictCache.Get(token)
stillExpired := true
skipAll := false skipAll := false
unrestrictCache.IterCb(func(key string, download *Download) { unrestrictCache.IterCb(func(key string, download *Download) {
if skipAll { if skipAll {
@@ -442,20 +430,19 @@ func (rd *RealDebrid) MonitorExpiredTokens() {
err := rd.downloadClient.VerifyLink(download.Download) err := rd.downloadClient.VerifyLink(download.Download)
if err != nil { if err != nil {
if utils.IsBytesLimitReached(err) { if utils.IsBytesLimitReached(err) {
if i%15 == 0 { if i%10 == 0 {
rd.log.Debugf("Token %s is still expired", utils.MaskToken(token)) rd.log.Debugf("Token %s is still expired", utils.MaskToken(token))
} }
// we already know that it is still expired
skipAll = true skipAll = true
} }
return return
} }
stillExpired = false
skipAll = true skipAll = true
rd.verifiedLinks.Set(download.ID, time.Now().Unix()+DOWNLOAD_LINK_EXPIRY) rd.verifiedLinks.Set(download.ID, time.Now().Unix()+DOWNLOAD_LINK_EXPIRY)
})
if !stillExpired {
rd.TokenManager.SetTokenAsUnexpired(token) rd.TokenManager.SetTokenAsUnexpired(token)
} })
} }
time.Sleep(sleepPeriod) time.Sleep(sleepPeriod)
} }