Merge pull request #30 from dreamblack86/master

Offers the possibility to use an internal proxy.

config.sample.toml

@@ -18,6 +18,7 @@ max_size = 4000
 [app.http_client]
 max_idle_conns =  100
 request_timeout = 8000
+proxy_url = "http://internal-squid-proxy.com:3128"
 
 [app.chat.alertManagerTestRoom]
 notification_url = "https://chat.googleapis.com/v1/spaces/xxx/messages?key=abc-xyz&token=token-unique-key%3D"

helm/templates/configmap.yaml

@@ -30,6 +30,7 @@ data:
     [app.http_client]
     max_idle_conns = {{ .Values.configmap.app.http_client.max_idle_conns }}
     request_timeout = {{ .Values.configmap.app.http_client.request_timeout }}
+    proxy_url = {{ .Values.configmap.app.http_client.proxy_url | quote }}
 
     {{- with .Values.configmap.rooms }}
     {{ tpl . $ | indent 4 }}

helm/values.yaml

@@ -18,6 +18,7 @@ configmap:
     http_client:
       max_idle_conns: "100"
       request_timeout: "8000"
+      proxy_url: "http://internal-squid-proxy.com:3128"
   rooms: |
 
     [app.chat.alertManagerTestRoom]

kustomize/overlays/prod/raw/config.toml

@@ -17,6 +17,7 @@ template_file = "message.tmpl"
 [app.http_client]
 max_idle_conns =  100
 request_timeout = 8000
+proxy_url = "http://internal-squid-proxy.com:3128"
 
 [app.chat.alertManagerTestRoom]
 notification_url = "https://chat.googleapis.com/v1/spaces/xxx/messages?key=abc-xyz&token=token-unique-key%3D"

main.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"log"
 	"net/http"
+	"net/url"
 	"os"
 	"time"
 
@@ -72,6 +73,7 @@ func initConfig() {
 	viper.SetDefault("server.read_timeout", 1000)
 	viper.SetDefault("server.write_timeout", 5000)
 	viper.SetDefault("server.keepalive_timeout", 30000)
+	viper.SetDefault("app.http_client.proxy_url", "")
 	viper.SetDefault("app.max_size", 4000)
 	// Process flags.
 	flagSet.Parse(os.Args[1:])
@@ -94,13 +96,29 @@ func initConfig() {
 }
 
 func initClient() *http.Client {
+
+	transport := &http.Transport{
+		MaxIdleConnsPerHost:   viper.GetInt("app.http_client.max_idle_conns"),
+		ResponseHeaderTimeout: time.Duration(viper.GetDuration("app.http_client.request_timeout") * time.Millisecond),
+	}
+
+	proxyURLString := viper.GetString("app.http_client.proxy_url")
+
+	if proxyURLString != "" {
+
+		proxyURL, err := url.Parse(proxyURLString)
+		if err != nil {
+			errLog.Fatalf("Unable to parse `proxy_url`: %s", err)
+		}
+
+		transport.Proxy = http.ProxyURL(proxyURL)
+	}
+
 	// Generic HTTP handler for communicating with the Chat webhook endpoint.
 	return &http.Client{
-		Timeout: time.Duration(viper.GetDuration("app.http_client.request_timeout") * time.Millisecond),
-		Transport: &http.Transport{
-			MaxIdleConnsPerHost:   viper.GetInt("app.http_client.max_idle_conns"),
-			ResponseHeaderTimeout: time.Duration(viper.GetDuration("app.http_client.request_timeout") * time.Millisecond),
-		}}
+		Timeout:   time.Duration(viper.GetDuration("app.http_client.request_timeout") * time.Millisecond),
+		Transport: transport}
+
 }
 
 // prog initialisation.