notifier: Add proxy parameter to webhook notifier

This commit is contained in:
Quentin Machu 2016-04-12 13:51:05 -04:00
parent ed1a2b6ffd
commit 2fb815dc37
2 changed files with 21 additions and 7 deletions

View File

@ -72,3 +72,6 @@ clair:
cafile:
keyfile:
certfile:
# Optional HTTP Proxy: must be a valid URL (including the scheme).
proxy:

View File

@ -49,6 +49,7 @@ type WebhookNotifierConfiguration struct {
CertFile string
KeyFile string
CAFile string
Proxy string
}
func init() {
@ -78,22 +79,32 @@ func (h *WebhookNotifier) Configure(config *config.NotifierConfig) (bool, error)
return false, nil
}
if _, err := url.Parse(httpConfig.Endpoint); err != nil {
return false, errors.New("invalid endpoint URL")
return false, fmt.Errorf("could not parse endpoint URL: %s\n", err)
}
h.endpoint = httpConfig.Endpoint
// Setup HTTP client.
transport := &http.Transport{}
h.client = &http.Client{
Transport: transport,
Timeout: timeout,
}
// Initialize TLS.
tlsConfig, err := loadTLSClientConfig(&httpConfig)
transport.TLSClientConfig, err = loadTLSClientConfig(&httpConfig)
if err != nil {
return false, fmt.Errorf("could not initialize client cert auth: %s\n", err)
}
h.client = &http.Client{
Transport: &http.Transport{
TLSClientConfig: tlsConfig,
},
Timeout: timeout,
// Set proxy.
if httpConfig.Proxy != "" {
proxyURL, err := url.ParseRequestURI(httpConfig.Proxy)
if err != nil {
return false, fmt.Errorf("could not parse proxy URL: %s\n", err)
}
transport.Proxy = http.ProxyURL(proxyURL)
}
return true, nil
}