h2cp/main.go at main · moonrhythm/h2cp · GitHub

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
package main

import (
	"crypto/tls"
	"flag"
	"log"
	"net/http"
	"strings"
	"time"

	"github.com/moonrhythm/parapet"
	"github.com/moonrhythm/parapet/pkg/prom"
	"github.com/moonrhythm/parapet/pkg/upstream"
)

func main() {
	addr := flag.String("addr", ":8080", "listen address")
	target := flag.String("target", "", "target address")
	enableProm := flag.Bool("prom", false, "enable prometheus metrics")
	flag.Parse()

	if *addr == "" {
		log.Fatal("-addr required")
	}
	if *target == "" {
		log.Fatal("-upstream required")
	}

	var tr http.RoundTripper
	switch {
	case strings.HasPrefix(*target, "unix://"):
		*target = strings.TrimPrefix(*target, "unix://")
		tr = &upstream.UnixTransport{
			MaxIdleConns:          5000,
			ResponseHeaderTimeout: -1,
		}
	case strings.HasPrefix(*target, "https://"):
		*target = strings.TrimPrefix(*target, "https://")
		*target = injectDefaultPort(*target, "443")
		tr = &upstream.HTTPSTransport{
			DialTimeout:           time.Second,
			MaxIdleConns:          1000,
			ResponseHeaderTimeout: -1,
			TLSClientConfig: &tls.Config{
				InsecureSkipVerify: true,
			},
		}
	case strings.HasPrefix(*target, "h2c://"):
		*target = strings.TrimPrefix(*target, "h2c://")
		*target = injectDefaultPort(*target, "80")
		tr = &upstream.H2CTransport{}
	default:
		*target = strings.TrimPrefix(*target, "http://")
		*target = injectDefaultPort(*target, "80")
		tr = &upstream.HTTPTransport{
			DialTimeout:           time.Second,
			MaxIdleConns:          1000,
			ResponseHeaderTimeout: -1,
		}
	}

	svc := parapet.NewBackend()

	if *enableProm {
		prom.Connections(svc)
		prom.Networks(svc)
		svc.Use(prom.Requests())

		log.Printf("enabled Prometheus metrics on :9187")
		go prom.Start(":9187")
	}

	svc.Addr = *addr
	svc.H2C = true
	svc.Use(upstream.SingleHost(*target, tr))
	err := svc.ListenAndServe()
	if err != nil {
		log.Fatal(err)
	}
}

func injectDefaultPort(addr string, port string) string {
	if strings.Contains(addr, ":") {
		return addr
	}
	return addr + ":" + port
}