apunkt/imageproxy
1
0
Fork 0
mirror of https://github.com/willnorris/imageproxy.git synced 2026-05-09 03:52:40 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Add option to override expire headers

Browse source
This commit is contained in:
Medya Gh 2019-12-27 23:14:02 -08:00 • committed by Will Norris
parent 95bcf70004
commit 3c7ff75344
4 changed files with 28 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
imageproxy.go
View file

@ -85,12 +85,16 @@ type Proxy struct {
// The User-Agent used by imageproxy when requesting origin image
UserAgent string
// Request header override key values. provided by user. (for example Expires)
HeaderOverrides map[string]string
}
// NewProxy constructs a new proxy. The provided http RoundTripper will be
// used to fetch remote URLs. If nil is provided, http.DefaultTransport will
// be used.
func NewProxy(transport http.RoundTripper, cache Cache) *Proxy {
// overrideHeader are key/values to override the request headers, such as expires
func NewProxy(transport http.RoundTripper, cache Cache, overrideHeader map[string]string) *Proxy {
if transport == nil {
transport, _ = aia.NewTransport()
}
@ -217,6 +221,10 @@ func (p *Proxy) serveImage(w http.ResponseWriter, r *http.Request) {
copyHeader(w.Header(), resp.Header, "Cache-Control", "Last-Modified", "Expires", "Etag", "Link")
for k, v := range p.HeaderOverrides {
p.logf("overriding request header %s:%s", k, v)
resp.Header.Set(k, v)
}
if should304(r, resp) {
w.WriteHeader(http.StatusNotModified)
return