@@ -26,19 +26,21 @@ import (
26
26
type ProxyBuilder struct {
27
27
reporter reporter.Reporter
28
28
manager httpserver.ClientConfigManager
29
+ tchanRep * Reporter
29
30
}
30
31
31
32
// NewCollectorProxy creates ProxyBuilder
32
33
func NewCollectorProxy (builder * Builder , mFactory metrics.Factory , logger * zap.Logger ) (* ProxyBuilder , error ) {
33
- b := & ProxyBuilder {}
34
34
r , err := builder .CreateReporter (logger )
35
35
if err != nil {
36
36
return nil , err
37
37
}
38
38
tchannelMetrics := mFactory .Namespace ("" , map [string ]string {"protocol" : "tchannel" })
39
- b .reporter = reporter .WrapWithMetrics (r , tchannelMetrics )
40
- b .manager = httpserver .WrapWithMetrics (httpserver .NewCollectorProxy (r .CollectorServiceName (), r .Channel ()), tchannelMetrics )
41
- return b , nil
39
+ return & ProxyBuilder {
40
+ tchanRep : r ,
41
+ reporter : reporter .WrapWithMetrics (r , tchannelMetrics ),
42
+ manager : httpserver .WrapWithMetrics (httpserver .NewCollectorProxy (r .CollectorServiceName (), r .Channel ()), tchannelMetrics ),
43
+ }, nil
42
44
}
43
45
44
46
// GetReporter returns Reporter
@@ -53,6 +55,6 @@ func (b ProxyBuilder) GetManager() httpserver.ClientConfigManager {
53
55
54
56
// Close closes connections used by proxy.
55
57
func (b ProxyBuilder ) Close () error {
56
- b .reporter .Channel ().Close ()
58
+ b .tchanRep .Channel ().Close ()
57
59
return nil
58
60
}
0 commit comments