forked from concourse/concourse
/
factory.go
87 lines (75 loc) · 2.85 KB
/
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package gclient
import (
"net/http"
"time"
"code.cloudfoundry.org/garden/routes"
"code.cloudfoundry.org/lager"
"github.com/pf-qiu/concourse/v6/atc/worker/gclient/connection"
"github.com/pf-qiu/concourse/v6/atc/worker/transport"
"github.com/concourse/retryhttp"
"github.com/tedsuo/rata"
)
type gardenClientFactory struct {
db transport.TransportDB
logger lager.Logger
workerName string
workerHost *string
retryBackOffFactory retryhttp.BackOffFactory
streamClientRequestTimeout time.Duration
}
func NewGardenClientFactory(
db transport.TransportDB,
logger lager.Logger,
workerName string,
workerHost *string,
retryBackOffFactory retryhttp.BackOffFactory,
streamClientRequestTimeout time.Duration,
) *gardenClientFactory {
return &gardenClientFactory{
db: db,
logger: logger,
workerName: workerName,
workerHost: workerHost,
retryBackOffFactory: retryBackOffFactory,
streamClientRequestTimeout: streamClientRequestTimeout,
}
}
func (gcf *gardenClientFactory) NewClient() Client {
retryer := &transport.UnreachableWorkerRetryer{
DelegateRetryer: &retryhttp.DefaultRetryer{},
}
streamClient := &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,
},
Timeout: gcf.streamClientRequestTimeout,
}
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: streamClient,
HijackableClient: hijackableClient,
Req: rata.NewRequestGenerator("http://127.0.0.1:8080", routes.Routes),
}
return NewClient(NewRetryableConnection(connection.NewWithHijacker(hijackStreamer, gcf.logger)))
}
// Do not try any client method that requires hijack functionality (streaming logs)!
func BasicGardenClientWithRequestTimeout(logger lager.Logger, requestTimeout time.Duration, address string) Client {
streamClient := &http.Client{
Timeout: requestTimeout,
}
streamer := &transport.WorkerHijackStreamer{
HttpClient: streamClient,
HijackableClient: nil,
Req: rata.NewRequestGenerator(address, routes.Routes),
}
return NewClient(connection.NewWithHijacker(streamer, logger))
}