Quote name attribute if necessary

This commit updates MetadataNamingStrategy to quote an ObjectName
attribute value if necessary. For now, only the name attribute is
handled as it is usually a bean name, and we have no control over
its structure.

Closes gh-31708
This commit is contained in:
Stéphane Nicoll
2023-11-28 17:02:09 +01:00
parent 87730f76b1
commit 3783d31c09
2 changed files with 114 additions and 2 deletions

View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2022 the original author or authors.
* Copyright 2002-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.
@@ -50,6 +50,9 @@ import org.springframework.util.StringUtils;
*/
public class MetadataNamingStrategy implements ObjectNamingStrategy, InitializingBean {
private static final char[] QUOTABLE_CHARS = new char[] {',', '=', ':', '"'};
/**
* The {@code JmxAttributeSource} implementation to use for reading metadata.
*/
@@ -132,10 +135,23 @@ public class MetadataNamingStrategy implements ObjectNamingStrategy, Initializin
}
Hashtable<String, String> properties = new Hashtable<>();
properties.put("type", ClassUtils.getShortName(managedClass));
properties.put("name", beanKey);
properties.put("name", quoteIfNecessary(beanKey));
return ObjectNameManager.getInstance(domain, properties);
}
}
}
private static String quoteIfNecessary(String value) {
return shouldQuote(value) ? ObjectName.quote(value) : value;
}
private static boolean shouldQuote(String value) {
for (char quotableChar : QUOTABLE_CHARS) {
if (value.indexOf(quotableChar) != -1) {
return true;
}
}
return false;
}
}

View File

@@ -0,0 +1,96 @@
/*
* Copyright 2002-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.jmx.export.naming;
import java.util.function.Consumer;
import javax.management.MalformedObjectNameException;
import javax.management.ObjectName;
import org.junit.jupiter.api.Test;
import org.springframework.jmx.export.annotation.AnnotationJmxAttributeSource;
import org.springframework.util.ClassUtils;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.entry;
/**
* Tests for {@link MetadataNamingStrategy}.
*
* @author Stephane Nicoll
*/
class MetadataNamingStrategyTests {
private static final TestBean TEST_BEAN = new TestBean();
private final MetadataNamingStrategy strategy;
MetadataNamingStrategyTests() {
this.strategy = new MetadataNamingStrategy();
this.strategy.setDefaultDomain("com.example");
this.strategy.setAttributeSource(new AnnotationJmxAttributeSource());
}
@Test
void getObjectNameWhenBeanNameIsSimple() throws MalformedObjectNameException {
ObjectName name = this.strategy.getObjectName(TEST_BEAN, "myBean");
assertThat(name.getDomain()).isEqualTo("com.example");
assertThat(name).satisfies(hasDefaultProperties(TEST_BEAN, "myBean"));
}
@Test
void getObjectNameWhenBeanNameIsValidObjectName() throws MalformedObjectNameException {
ObjectName name = this.strategy.getObjectName(TEST_BEAN, "com.another:name=myBean");
assertThat(name.getDomain()).isEqualTo("com.another");
assertThat(name.getKeyPropertyList()).containsOnly(entry("name", "myBean"));
}
@Test
void getObjectNameWhenBeanNamContainsComma() throws MalformedObjectNameException {
ObjectName name = this.strategy.getObjectName(TEST_BEAN, "myBean,");
assertThat(name).satisfies(hasDefaultProperties(TEST_BEAN, "\"myBean,\""));
}
@Test
void getObjectNameWhenBeanNamContainsEquals() throws MalformedObjectNameException {
ObjectName name = this.strategy.getObjectName(TEST_BEAN, "my=Bean");
assertThat(name).satisfies(hasDefaultProperties(TEST_BEAN, "\"my=Bean\""));
}
@Test
void getObjectNameWhenBeanNamContainsColon() throws MalformedObjectNameException {
ObjectName name = this.strategy.getObjectName(TEST_BEAN, "my:Bean");
assertThat(name).satisfies(hasDefaultProperties(TEST_BEAN, "\"my:Bean\""));
}
@Test
void getObjectNameWhenBeanNamContainsQuote() throws MalformedObjectNameException {
ObjectName name = this.strategy.getObjectName(TEST_BEAN, "\"myBean\"");
assertThat(name).satisfies(hasDefaultProperties(TEST_BEAN, "\"\\\"myBean\\\"\""));
}
private Consumer<ObjectName> hasDefaultProperties(Object instance, String expectedName) {
return objectName -> assertThat(objectName.getKeyPropertyList()).containsOnly(
entry("type", ClassUtils.getShortName(instance.getClass())),
entry("name", expectedName));
}
static class TestBean {}
}