Skip to content

Commit b3127ba

Browse files
authored
Merge pull request #910 from togettoyou/master
fix: prevent panic when client initialization fails
2 parents 9f44105 + fdfa053 commit b3127ba

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

pkg/trimaran/collector.go

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,25 +56,32 @@ func NewCollector(logger klog.Logger, trimaranSpec *pluginConfig.TrimaranSpec) (
5656
logger.V(4).Info("Using TrimaranSpec", "type", trimaranSpec.MetricProvider.Type,
5757
"address", trimaranSpec.MetricProvider.Address, "watcher", trimaranSpec.WatcherAddress)
5858

59-
var client loadwatcherapi.Client
59+
var (
60+
client loadwatcherapi.Client
61+
err error
62+
)
6063
if trimaranSpec.WatcherAddress != "" {
61-
client, _ = loadwatcherapi.NewServiceClient(trimaranSpec.WatcherAddress)
64+
client, err = loadwatcherapi.NewServiceClient(trimaranSpec.WatcherAddress)
6265
} else {
6366
opts := watcher.MetricsProviderOpts{
6467
Name: string(trimaranSpec.MetricProvider.Type),
6568
Address: trimaranSpec.MetricProvider.Address,
6669
AuthToken: trimaranSpec.MetricProvider.Token,
6770
InsecureSkipVerify: trimaranSpec.MetricProvider.InsecureSkipVerify,
6871
}
69-
client, _ = loadwatcherapi.NewLibraryClient(opts)
72+
client, err = loadwatcherapi.NewLibraryClient(opts)
73+
}
74+
75+
if err != nil {
76+
return nil, err
7077
}
7178

7279
collector := &Collector{
7380
client: client,
7481
}
7582

7683
// populate metrics before returning
77-
err := collector.updateMetrics(logger)
84+
err = collector.updateMetrics(logger)
7885
if err != nil {
7986
logger.Error(err, "Unable to populate metrics initially")
8087
}

0 commit comments

Comments
 (0)