fix test/code validation issues

This commit is contained in:
Lukas Rist
2022-12-23 15:51:54 +01:00
parent 738852f932
commit 85d73be641
3 changed files with 17 additions and 13 deletions

View File

@ -2,13 +2,14 @@ package client
import (
"context"
"github.com/geziyor/geziyor/internal"
"net/http"
"net/url"
"sync/atomic"
"github.com/geziyor/geziyor/internal"
)
const ProxyURLKey = 0
type ProxyURLKey int
type roundRobinProxy struct {
proxyURLs []*url.URL
@ -20,7 +21,7 @@ func (r *roundRobinProxy) GetProxy(pr *http.Request) (*url.URL, error) {
u := r.proxyURLs[index%uint32(len(r.proxyURLs))]
// Set proxy url to context
ctx := context.WithValue(pr.Context(), ProxyURLKey, u.String())
ctx := context.WithValue(pr.Context(), ProxyURLKey(0), u.String())
*pr = *pr.WithContext(ctx)
return u, nil
}