proxypool.go

package pylon

import (
	"net/http/httputil"
	"net/http"
)

type ProxyPool struct {
	pool chan *httputil.ReverseProxy
}

func NewProxyPool(capacity int) *ProxyPool {
	return &ProxyPool{
		make(chan *httputil.ReverseProxy, capacity),
	}
}

func (p *ProxyPool) Get() *httputil.ReverseProxy {
	var proxy *httputil.ReverseProxy
	select {
	case proxy = <-p.pool:
	default:
		proxy = NewProxy()
	}
	return proxy
}

func (p *ProxyPool) Put(rp *httputil.ReverseProxy) {
	select {
	case p.pool <- rp:
	default:
	}
}

func NewProxy() *httputil.ReverseProxy {
	return &httputil.ReverseProxy{
		Transport: NewPylonTransport(),
		FlushInterval: flushInterval,
	}
}

func setUpProxy(proxy *httputil.ReverseProxy, m *MicroService, host string) {
	proxy.Director = func(req *http.Request) {
		req.URL.Scheme = "http"
		req.URL.Host = host
	}
	proxy.Transport.(*PylonTransport).setMicroService(m)
}