Merge pull request #1196 from vishh/disable-metrics

Fix --disable_metrics parsing to accept empty string as input.
This commit is contained in:
Vish Kannan 2016-04-06 13:51:06 -07:00
commit 81f06c22ea
2 changed files with 10 additions and 0 deletions

View File

@ -76,6 +76,10 @@ func (ml *metricSetValue) String() string {
}
func (ml *metricSetValue) Set(value string) error {
ignoreMetrics = metricSetValue{}
if value == "" {
return nil
}
for _, metric := range strings.Split(value, ",") {
if ignoreWhitelist.Has(container.MetricKind(metric)) {
(*ml).Add(container.MetricKind(metric))

View File

@ -27,3 +27,9 @@ func TestTcpMetricsAreDisabledByDefault(t *testing.T) {
flag.Parse()
assert.True(t, ignoreMetrics.Has(container.NetworkTcpUsageMetrics))
}
func TestTcpMetricsAreEnabledOnDemand(t *testing.T) {
assert.True(t, ignoreMetrics.Has(container.NetworkTcpUsageMetrics))
ignoreMetrics.Set("")
assert.False(t, ignoreMetrics.Has(container.NetworkTcpUsageMetrics))
}