forked from vmware-archive/atc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
garden_connection_factory.go
72 lines (62 loc) · 2.2 KB
/
garden_connection_factory.go
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
package worker
import (
"net/http"
gconn "code.cloudfoundry.org/garden/client/connection"
"code.cloudfoundry.org/garden/routes"
"code.cloudfoundry.org/lager"
"github.com/concourse/atc/worker/transport"
"github.com/concourse/retryhttp"
"github.com/tedsuo/rata"
)
//go:generate counterfeiter . GardenConnectionFactory
type GardenConnectionFactory interface {
BuildConnection() gconn.Connection
}
type gardenConnectionFactory struct {
db transport.TransportDB
logger lager.Logger
workerName string
workerHost *string
retryBackOffFactory retryhttp.BackOffFactory
}
func NewGardenConnectionFactory(
db transport.TransportDB,
logger lager.Logger,
workerName string,
workerHost *string,
retryBackOffFactory retryhttp.BackOffFactory,
) GardenConnectionFactory {
return &gardenConnectionFactory{
db: db,
logger: logger,
workerName: workerName,
workerHost: workerHost,
retryBackOffFactory: retryBackOffFactory,
}
}
func (gcf *gardenConnectionFactory) BuildConnection() gconn.Connection {
retryer := &transport.UnreachableWorkerRetryer{
DelegateRetryer: &retryhttp.DefaultRetryer{},
}
httpClient := &http.Client{
Transport: &retryhttp.RetryRoundTripper{
Logger: gcf.logger.Session("retryable-http-client"),
BackOffFactory: gcf.retryBackOffFactory,
RoundTripper: transport.NewGardenRoundTripper(gcf.workerName, gcf.workerHost, gcf.db, &http.Transport{DisableKeepAlives: true}),
Retryer: retryer,
},
}
hijackableClient := &retryhttp.RetryHijackableClient{
Logger: gcf.logger.Session("retry-hijackable-client"),
BackOffFactory: gcf.retryBackOffFactory,
HijackableClient: transport.NewHijackableClient(gcf.workerName, gcf.db, retryhttp.DefaultHijackableClient),
Retryer: retryer,
}
// the request generator's address doesn't matter because it's overwritten by the worker lookup clients
hijackStreamer := &transport.WorkerHijackStreamer{
HttpClient: httpClient,
HijackableClient: hijackableClient,
Req: rata.NewRequestGenerator("http://127.0.0.1:8080", routes.Routes),
}
return gconn.NewWithHijacker(hijackStreamer, gcf.logger)
}