Browse Source
For example, setting management.observations.enable.denied.prefix=false will prevent all observations starting with 'denied.prefix' Closes gh-34802pull/35890/head
10 changed files with 230 additions and 142 deletions
@ -1,51 +0,0 @@
@@ -1,51 +0,0 @@
|
||||
/* |
||||
* Copyright 2012-2023 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.springframework.boot.actuate.autoconfigure.observation; |
||||
|
||||
import java.util.Map.Entry; |
||||
|
||||
import io.micrometer.common.KeyValues; |
||||
import io.micrometer.observation.Observation.Context; |
||||
import io.micrometer.observation.ObservationFilter; |
||||
|
||||
/** |
||||
* {@link ObservationFilter} to apply settings from {@link ObservationProperties}. |
||||
* |
||||
* @author Moritz Halbritter |
||||
*/ |
||||
class PropertiesObservationFilter implements ObservationFilter { |
||||
|
||||
private final ObservationFilter delegate; |
||||
|
||||
PropertiesObservationFilter(ObservationProperties properties) { |
||||
this.delegate = createDelegate(properties); |
||||
} |
||||
|
||||
@Override |
||||
public Context map(Context context) { |
||||
return this.delegate.map(context); |
||||
} |
||||
|
||||
private static ObservationFilter createDelegate(ObservationProperties properties) { |
||||
if (properties.getKeyValues().isEmpty()) { |
||||
return (context) -> context; |
||||
} |
||||
KeyValues keyValues = KeyValues.of(properties.getKeyValues().entrySet(), Entry::getKey, Entry::getValue); |
||||
return (context) -> context.addLowCardinalityKeyValues(keyValues); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,83 @@
@@ -0,0 +1,83 @@
|
||||
/* |
||||
* Copyright 2012-2023 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.springframework.boot.actuate.autoconfigure.observation; |
||||
|
||||
import java.util.Map; |
||||
import java.util.Map.Entry; |
||||
import java.util.function.Supplier; |
||||
|
||||
import io.micrometer.common.KeyValues; |
||||
import io.micrometer.observation.Observation.Context; |
||||
import io.micrometer.observation.ObservationFilter; |
||||
import io.micrometer.observation.ObservationPredicate; |
||||
|
||||
import org.springframework.util.StringUtils; |
||||
|
||||
/** |
||||
* {@link ObservationFilter} to apply settings from {@link ObservationProperties}. |
||||
* |
||||
* @author Moritz Halbritter |
||||
*/ |
||||
class PropertiesObservationFilterPredicate implements ObservationFilter, ObservationPredicate { |
||||
|
||||
private final ObservationFilter commonKeyValuesFilter; |
||||
|
||||
private final ObservationProperties properties; |
||||
|
||||
PropertiesObservationFilterPredicate(ObservationProperties properties) { |
||||
this.properties = properties; |
||||
this.commonKeyValuesFilter = createCommonKeyValuesFilter(properties); |
||||
} |
||||
|
||||
@Override |
||||
public Context map(Context context) { |
||||
return this.commonKeyValuesFilter.map(context); |
||||
} |
||||
|
||||
@Override |
||||
public boolean test(String name, Context context) { |
||||
return lookupWithFallbackToAll(this.properties.getEnable(), name, true); |
||||
} |
||||
|
||||
private static <T> T lookupWithFallbackToAll(Map<String, T> values, String name, T defaultValue) { |
||||
if (values.isEmpty()) { |
||||
return defaultValue; |
||||
} |
||||
return doLookup(values, name, () -> values.getOrDefault("all", defaultValue)); |
||||
} |
||||
|
||||
private static <T> T doLookup(Map<String, T> values, String name, Supplier<T> defaultValue) { |
||||
while (StringUtils.hasLength(name)) { |
||||
T result = values.get(name); |
||||
if (result != null) { |
||||
return result; |
||||
} |
||||
int lastDot = name.lastIndexOf('.'); |
||||
name = (lastDot != -1) ? name.substring(0, lastDot) : ""; |
||||
} |
||||
return defaultValue.get(); |
||||
} |
||||
|
||||
private static ObservationFilter createCommonKeyValuesFilter(ObservationProperties properties) { |
||||
if (properties.getKeyValues().isEmpty()) { |
||||
return (context) -> context; |
||||
} |
||||
KeyValues keyValues = KeyValues.of(properties.getKeyValues().entrySet(), Entry::getKey, Entry::getValue); |
||||
return (context) -> context.addLowCardinalityKeyValues(keyValues); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,100 @@
@@ -0,0 +1,100 @@
|
||||
/* |
||||
* Copyright 2012-2023 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.springframework.boot.actuate.autoconfigure.observation; |
||||
|
||||
import io.micrometer.common.KeyValue; |
||||
import io.micrometer.common.KeyValues; |
||||
import io.micrometer.observation.Observation.Context; |
||||
import org.junit.jupiter.api.Test; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
||||
/** |
||||
* Tests for {@link PropertiesObservationFilterPredicate}. |
||||
* |
||||
* @author Moritz Halbritter |
||||
*/ |
||||
class PropertiesObservationFilterPredicateTests { |
||||
|
||||
@Test |
||||
void shouldDoNothingIfKeyValuesAreEmpty() { |
||||
PropertiesObservationFilterPredicate filter = createFilter(); |
||||
Context mapped = mapContext(filter, "a", "alpha"); |
||||
assertThat(mapped.getLowCardinalityKeyValues()).containsExactly(KeyValue.of("a", "alpha")); |
||||
} |
||||
|
||||
@Test |
||||
void shouldAddKeyValues() { |
||||
PropertiesObservationFilterPredicate filter = createFilter("b", "beta"); |
||||
Context mapped = mapContext(filter, "a", "alpha"); |
||||
assertThat(mapped.getLowCardinalityKeyValues()).containsExactly(KeyValue.of("a", "alpha"), |
||||
KeyValue.of("b", "beta")); |
||||
} |
||||
|
||||
@Test |
||||
void shouldFilter() { |
||||
PropertiesObservationFilterPredicate predicate = createPredicate("spring.security"); |
||||
Context context = new Context(); |
||||
assertThat(predicate.test("spring.security.filterchains", context)).isFalse(); |
||||
assertThat(predicate.test("spring.security", context)).isFalse(); |
||||
assertThat(predicate.test("spring.data", context)).isTrue(); |
||||
assertThat(predicate.test("spring", context)).isTrue(); |
||||
} |
||||
|
||||
@Test |
||||
void filterShouldFallbackToAll() { |
||||
PropertiesObservationFilterPredicate predicate = createPredicate("all"); |
||||
Context context = new Context(); |
||||
assertThat(predicate.test("spring.security.filterchains", context)).isFalse(); |
||||
assertThat(predicate.test("spring.security", context)).isFalse(); |
||||
assertThat(predicate.test("spring.data", context)).isFalse(); |
||||
assertThat(predicate.test("spring", context)).isFalse(); |
||||
} |
||||
|
||||
@Test |
||||
void shouldNotFilterIfDisabledNamesIsEmpty() { |
||||
PropertiesObservationFilterPredicate predicate = createPredicate(); |
||||
Context context = new Context(); |
||||
assertThat(predicate.test("spring.security.filterchains", context)).isTrue(); |
||||
assertThat(predicate.test("spring.security", context)).isTrue(); |
||||
assertThat(predicate.test("spring.data", context)).isTrue(); |
||||
assertThat(predicate.test("spring", context)).isTrue(); |
||||
} |
||||
|
||||
private static Context mapContext(PropertiesObservationFilterPredicate filter, String... initialKeyValues) { |
||||
Context context = new Context(); |
||||
context.addLowCardinalityKeyValues(KeyValues.of(initialKeyValues)); |
||||
return filter.map(context); |
||||
} |
||||
|
||||
private static PropertiesObservationFilterPredicate createFilter(String... keyValues) { |
||||
ObservationProperties properties = new ObservationProperties(); |
||||
for (int i = 0; i < keyValues.length; i += 2) { |
||||
properties.getKeyValues().put(keyValues[i], keyValues[i + 1]); |
||||
} |
||||
return new PropertiesObservationFilterPredicate(properties); |
||||
} |
||||
|
||||
private static PropertiesObservationFilterPredicate createPredicate(String... disabledNames) { |
||||
ObservationProperties properties = new ObservationProperties(); |
||||
for (String name : disabledNames) { |
||||
properties.getEnable().put(name, false); |
||||
} |
||||
return new PropertiesObservationFilterPredicate(properties); |
||||
} |
||||
|
||||
} |
||||
@ -1,62 +0,0 @@
@@ -1,62 +0,0 @@
|
||||
/* |
||||
* Copyright 2012-2023 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.springframework.boot.actuate.autoconfigure.observation; |
||||
|
||||
import io.micrometer.common.KeyValue; |
||||
import io.micrometer.common.KeyValues; |
||||
import io.micrometer.observation.Observation.Context; |
||||
import org.junit.jupiter.api.Test; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
||||
/** |
||||
* Tests for {@link PropertiesObservationFilter}. |
||||
* |
||||
* @author Moritz Halbritter |
||||
*/ |
||||
class PropertiesObservationFilterTests { |
||||
|
||||
@Test |
||||
void shouldDoNothingIfKeyValuesAreEmpty() { |
||||
PropertiesObservationFilter filter = createFilter(); |
||||
Context mapped = mapContext(filter, "a", "alpha"); |
||||
assertThat(mapped.getLowCardinalityKeyValues()).containsExactly(KeyValue.of("a", "alpha")); |
||||
} |
||||
|
||||
@Test |
||||
void shouldAddKeyValues() { |
||||
PropertiesObservationFilter filter = createFilter("b", "beta"); |
||||
Context mapped = mapContext(filter, "a", "alpha"); |
||||
assertThat(mapped.getLowCardinalityKeyValues()).containsExactly(KeyValue.of("a", "alpha"), |
||||
KeyValue.of("b", "beta")); |
||||
} |
||||
|
||||
private static Context mapContext(PropertiesObservationFilter filter, String... initialKeyValues) { |
||||
Context context = new Context(); |
||||
context.addLowCardinalityKeyValues(KeyValues.of(initialKeyValues)); |
||||
return filter.map(context); |
||||
} |
||||
|
||||
private static PropertiesObservationFilter createFilter(String... keyValues) { |
||||
ObservationProperties properties = new ObservationProperties(); |
||||
for (int i = 0; i < keyValues.length; i += 2) { |
||||
properties.getKeyValues().put(keyValues[i], keyValues[i + 1]); |
||||
} |
||||
return new PropertiesObservationFilter(properties); |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue