Commit 8808d5e4 authored by Stephane Nicoll's avatar Stephane Nicoll

Polish "Support sending metrics to InfluxDB v2"

See gh-25721
parent 8eb73bcf
...@@ -21,7 +21,6 @@ import io.micrometer.influx.InfluxConsistency; ...@@ -21,7 +21,6 @@ import io.micrometer.influx.InfluxConsistency;
import org.springframework.boot.actuate.autoconfigure.metrics.export.properties.StepRegistryProperties; import org.springframework.boot.actuate.autoconfigure.metrics.export.properties.StepRegistryProperties;
import org.springframework.boot.context.properties.ConfigurationProperties; import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.util.StringUtils;
/** /**
* {@link ConfigurationProperties @ConfigurationProperties} for configuring Influx metrics * {@link ConfigurationProperties @ConfigurationProperties} for configuring Influx metrics
...@@ -35,7 +34,7 @@ import org.springframework.util.StringUtils; ...@@ -35,7 +34,7 @@ import org.springframework.util.StringUtils;
public class InfluxProperties extends StepRegistryProperties { public class InfluxProperties extends StepRegistryProperties {
/** /**
* Database to send metrics to when using an InfluxDB 1.x database. * Database to send metrics to. InfluxDB v1 only.
*/ */
private String db = "mydb"; private String db = "mydb";
...@@ -45,12 +44,12 @@ public class InfluxProperties extends StepRegistryProperties { ...@@ -45,12 +44,12 @@ public class InfluxProperties extends StepRegistryProperties {
private InfluxConsistency consistency = InfluxConsistency.ONE; private InfluxConsistency consistency = InfluxConsistency.ONE;
/** /**
* Login user of the Influx server. * Login user of the Influx server. InfluxDB v1 only.
*/ */
private String userName; private String userName;
/** /**
* Login password of the Influx server. * Login password of the Influx server. InfluxDB v1 only.
*/ */
private String password; private String password;
...@@ -91,7 +90,7 @@ public class InfluxProperties extends StepRegistryProperties { ...@@ -91,7 +90,7 @@ public class InfluxProperties extends StepRegistryProperties {
/** /**
* Whether to create the Influx database if it does not exist before attempting to * Whether to create the Influx database if it does not exist before attempting to
* publish metrics to it. * publish metrics to it. InfluxDB v1 only.
*/ */
private boolean autoCreateDb = true; private boolean autoCreateDb = true;
...@@ -102,13 +101,13 @@ public class InfluxProperties extends StepRegistryProperties { ...@@ -102,13 +101,13 @@ public class InfluxProperties extends StepRegistryProperties {
private InfluxApiVersion apiVersion; private InfluxApiVersion apiVersion;
/** /**
* InfluxDB v2 org to write metrics. * Org to write metrics to. InfluxDB v2 only.
*/ */
private String org; private String org;
/** /**
* InfluxDB v2 bucket for metrics. Use either the bucket name or ID. Defaults to the * Bucket for metrics. Use either the bucket name or ID. Defaults to the value of the
* value of the db property if not set. * db property if not set. InfluxDB v2 only.
*/ */
private String bucket; private String bucket;
...@@ -207,10 +206,7 @@ public class InfluxProperties extends StepRegistryProperties { ...@@ -207,10 +206,7 @@ public class InfluxProperties extends StepRegistryProperties {
} }
public InfluxApiVersion getApiVersion() { public InfluxApiVersion getApiVersion() {
if (this.apiVersion != null) { return this.apiVersion;
return this.apiVersion;
}
return StringUtils.hasText(this.org) ? InfluxApiVersion.V2 : InfluxApiVersion.V1;
} }
public void setApiVersion(InfluxApiVersion apiVersion) { public void setApiVersion(InfluxApiVersion apiVersion) {
...@@ -226,10 +222,7 @@ public class InfluxProperties extends StepRegistryProperties { ...@@ -226,10 +222,7 @@ public class InfluxProperties extends StepRegistryProperties {
} }
public String getBucket() { public String getBucket() {
if (this.bucket != null) { return this.bucket;
return this.bucket;
}
return this.db;
} }
public void setBucket(String bucket) { public void setBucket(String bucket) {
......
/*
* Copyright 2012-2021 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.metrics.export.influx;
import io.micrometer.influx.InfluxApiVersion;
import org.junit.jupiter.api.Test;
import static org.assertj.core.api.Assertions.assertThat;
/**
* Tests for {@link InfluxPropertiesConfigAdapter}.
*
* @author Stephane Nicoll
*/
class InfluxPropertiesConfigAdapterTests {
@Test
void adaptInfluxV1BasicConfig() {
InfluxProperties properties = new InfluxProperties();
properties.setDb("test-db");
properties.setUri("https://influx.example.com:8086");
properties.setUserName("user");
properties.setPassword("secret");
InfluxPropertiesConfigAdapter adapter = new InfluxPropertiesConfigAdapter(properties);
assertThat(adapter.apiVersion()).isEqualTo(InfluxApiVersion.V1);
assertThat(adapter.db()).isEqualTo("test-db");
assertThat(adapter.uri()).isEqualTo("https://influx.example.com:8086");
assertThat(adapter.userName()).isEqualTo("user");
assertThat(adapter.password()).isEqualTo("secret");
}
@Test
void adaptInfluxV2BasicConfig() {
InfluxProperties properties = new InfluxProperties();
properties.setOrg("test-org");
properties.setBucket("test-bucket");
properties.setUri("https://influx.example.com:8086");
properties.setToken("token");
InfluxPropertiesConfigAdapter adapter = new InfluxPropertiesConfigAdapter(properties);
assertThat(adapter.apiVersion()).isEqualTo(InfluxApiVersion.V2);
assertThat(adapter.org()).isEqualTo("test-org");
assertThat(adapter.bucket()).isEqualTo("test-bucket");
assertThat(adapter.uri()).isEqualTo("https://influx.example.com:8086");
assertThat(adapter.token()).isEqualTo("token");
}
}
...@@ -46,9 +46,7 @@ class InfluxPropertiesTests extends StepRegistryPropertiesTests { ...@@ -46,9 +46,7 @@ class InfluxPropertiesTests extends StepRegistryPropertiesTests {
assertThat(properties.getUri()).isEqualTo(config.uri()); assertThat(properties.getUri()).isEqualTo(config.uri());
assertThat(properties.isCompressed()).isEqualTo(config.compressed()); assertThat(properties.isCompressed()).isEqualTo(config.compressed());
assertThat(properties.isAutoCreateDb()).isEqualTo(config.autoCreateDb()); assertThat(properties.isAutoCreateDb()).isEqualTo(config.autoCreateDb());
assertThat(properties.getApiVersion()).isEqualTo(config.apiVersion());
assertThat(properties.getOrg()).isEqualTo(config.org()); assertThat(properties.getOrg()).isEqualTo(config.org());
assertThat(properties.getBucket()).isEqualTo(config.bucket());
assertThat(properties.getToken()).isEqualTo(config.token()); assertThat(properties.getToken()).isEqualTo(config.token());
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment