mirror/groupcache
1
0
Fork 0
mirror of https://github.com/mailgun/groupcache.git synced 2024-10-02 21:02:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #63 from pierrre/master

Browse Source 
use passed options
This commit is contained in:
Andrew Gerrand 2016-02-08 20:49:24 +11:00
commit 42b67f4f14
1 changed files with 19 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
http.go
View File

@ -46,12 +46,12 @@ type HTTPPool struct {
// If nil, the client uses http.DefaultTransport. // If nil, the client uses http.DefaultTransport.
Transport func(Context) http.RoundTripper Transport func(Context) http.RoundTripper
// base path including leading and trailing slash, e.g. "/_groupcache/"
basePath string
// this peer's base URL, e.g. "https://example.net:8000" // this peer's base URL, e.g. "https://example.net:8000"
self string self string
// opts specifies the options.
opts HTTPPoolOptions
mu sync.Mutex // guards peers and httpGetters mu sync.Mutex // guards peers and httpGetters
peers *consistenthash.Map peers *consistenthash.Map
httpGetters map[string]*httpGetter // keyed by e.g. "http://10.0.0.2:8008" httpGetters map[string]*httpGetter // keyed by e.g. "http://10.0.0.2:8008"
@ -78,7 +78,7 @@ type HTTPPoolOptions struct {
// for example "http://example.net:8000". // for example "http://example.net:8000".
func NewHTTPPool(self string) *HTTPPool { func NewHTTPPool(self string) *HTTPPool {
p := NewHTTPPoolOpts(self, nil) p := NewHTTPPoolOpts(self, nil)
http.Handle(p.basePath, p) http.Handle(p.opts.BasePath, p)
return p return p
} }
@ -93,23 +93,21 @@ func NewHTTPPoolOpts(self string, o *HTTPPoolOptions) *HTTPPool {
} }
httpPoolMade = true httpPoolMade = true
opts := HTTPPoolOptions{}
if o != nil {
opts = *o
}
if opts.BasePath == "" {
opts.BasePath = defaultBasePath
}
if opts.Replicas == 0 {
opts.Replicas = defaultReplicas
}
p := &HTTPPool{ p := &HTTPPool{
basePath: opts.BasePath,
self: self, self: self,
peers: consistenthash.New(opts.Replicas, opts.HashFn),
httpGetters: make(map[string]*httpGetter), httpGetters: make(map[string]*httpGetter),
} }
if o != nil {
p.opts = *o
}
if p.opts.BasePath == "" {
p.opts.BasePath = defaultBasePath
}
if p.opts.Replicas == 0 {
p.opts.Replicas = defaultReplicas
}
p.peers = consistenthash.New(p.opts.Replicas, p.opts.HashFn)
RegisterPeerPicker(func() PeerPicker { return p }) RegisterPeerPicker(func() PeerPicker { return p })
return p return p
} }
@ -120,11 +118,11 @@ func NewHTTPPoolOpts(self string, o *HTTPPoolOptions) *HTTPPool {
func (p *HTTPPool) Set(peers ...string) { func (p *HTTPPool) Set(peers ...string) {
p.mu.Lock() p.mu.Lock()
defer p.mu.Unlock() defer p.mu.Unlock()
p.peers = consistenthash.New(defaultReplicas, nil) p.peers = consistenthash.New(p.opts.Replicas, p.opts.HashFn)
p.peers.Add(peers...) p.peers.Add(peers...)
p.httpGetters = make(map[string]*httpGetter, len(peers)) p.httpGetters = make(map[string]*httpGetter, len(peers))
for _, peer := range peers { for _, peer := range peers {
p.httpGetters[peer] = &httpGetter{transport: p.Transport, baseURL: peer + p.basePath} p.httpGetters[peer] = &httpGetter{transport: p.Transport, baseURL: peer + p.opts.BasePath}
} }
} }
@ -142,10 +140,10 @@ func (p *HTTPPool) PickPeer(key string) (ProtoGetter, bool) {
func (p *HTTPPool) ServeHTTP(w http.ResponseWriter, r *http.Request) { func (p *HTTPPool) ServeHTTP(w http.ResponseWriter, r *http.Request) {
// Parse request. // Parse request.
if !strings.HasPrefix(r.URL.Path, p.basePath) { if !strings.HasPrefix(r.URL.Path, p.opts.BasePath) {
panic("HTTPPool serving unexpected path: " + r.URL.Path) panic("HTTPPool serving unexpected path: " + r.URL.Path)
} }
parts := strings.SplitN(r.URL.Path[len(p.basePath):], "/", 2) parts := strings.SplitN(r.URL.Path[len(p.opts.BasePath):], "/", 2)
if len(parts) != 2 { if len(parts) != 2 {
http.Error(w, "bad request", http.StatusBadRequest) http.Error(w, "bad request", http.StatusBadRequest)
return return