1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
// Copyright 2016 Michal Witkowski. All Rights Reserved.
// See LICENSE for licensing terms.
package main
import (
"flag"
"log"
"net"
"net/http"
"crypto/tls"
"fmt"
"time"
"github.com/mwitkow/go-conntrack"
"github.com/mwitkow/go-conntrack/connhelpers"
"github.com/prometheus/client_golang/prometheus"
"golang.org/x/net/context/ctxhttp"
_ "golang.org/x/net/trace"
)
var (
port = flag.Int("port", 9090, "whether to use tls or not")
useTls = flag.Bool("tls", true, "Whether to use TLS and HTTP2.")
tlsCertFilePath = flag.String("tls_cert_file", "certs/localhost.crt", "Path to the CRT/PEM file.")
tlsKeyFilePath = flag.String("tls_key_file", "certs/localhost.key", "Path to the private key file.")
)
func main() {
flag.Parse()
// Make sure all outbound connections use the wrapped dialer.
http.DefaultTransport.(*http.Transport).DialContext = conntrack.NewDialContextFunc(
conntrack.DialWithTracing(),
conntrack.DialWithDialer(&net.Dialer{
Timeout: 30 * time.Second,
KeepAlive: 30 * time.Second,
}),
)
// Since we're using a dynamic name, let's preregister it with prometheus.
conntrack.PreRegisterDialerMetrics("google")
handler := func(resp http.ResponseWriter, req *http.Request) {
resp.WriteHeader(http.StatusOK)
resp.Header().Add("Content-Type", "application/json")
resp.Write([]byte(`{"msg": "hello"}`))
callCtx := conntrack.DialNameToContext(req.Context(), "google")
_, err := ctxhttp.Get(callCtx, http.DefaultClient, "https://www.google.comx")
log.Printf("Google reached with err: %v", err)
log.Printf("Got request: %v", req)
}
http.DefaultServeMux.Handle("/", http.HandlerFunc(handler))
http.DefaultServeMux.Handle("/metrics", prometheus.Handler())
httpServer := http.Server{
Handler: http.DefaultServeMux,
}
var httpListener net.Listener
listener, err := net.Listen("tcp", fmt.Sprintf(":%d", *port))
if err != nil {
log.Fatalf("Failed to listen: %v", err)
}
listener = conntrack.NewListener(listener, conntrack.TrackWithTracing())
if !*useTls {
httpListener = listener
} else {
tlsConfig, err := connhelpers.TlsConfigForServerCerts(*tlsCertFilePath, *tlsKeyFilePath)
if err != nil {
log.Fatalf("Failed configuring TLS: %v", err)
}
tlsConfig, err = connhelpers.TlsConfigWithHttp2Enabled(tlsConfig)
if err != nil {
log.Fatalf("Failed configuring TLS: %v", err)
}
log.Printf("Listening with TLS")
tlsListener := tls.NewListener(listener, tlsConfig)
httpListener = tlsListener
}
//httpListener.Addr()
log.Printf("Listening on: %s", listener.Addr().String())
if err := httpServer.Serve(httpListener); err != nil {
log.Fatalf("Failed listning: %v", err)
}
}
|
