wulingxiao
3 years ago
parent
462f528e77
commit
9fb368347f
@ -1,129 +0,0 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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 com.tencent.cloud.rpc.enhancement.stat.config.plugin;
|
||||
|
||||
import java.net.UnknownHostException;
|
||||
import java.time.Duration;
|
||||
import java.util.Map;
|
||||
import java.util.Objects;
|
||||
import java.util.concurrent.Executors;
|
||||
import java.util.concurrent.ScheduledExecutorService;
|
||||
import java.util.concurrent.ScheduledFuture;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
import com.tencent.cloud.rpc.enhancement.stat.config.PolarisStatProperties;
|
||||
import com.tencent.polaris.client.util.NamedThreadFactory;
|
||||
import io.prometheus.client.CollectorRegistry;
|
||||
import io.prometheus.client.exporter.PushGateway;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
|
||||
/**
|
||||
* PrometheusPushGatewayContainer will push the metrics aggregated by
|
||||
* {@link com.tencent.polaris.plugins.stat.prometheus.handler.PrometheusHandler} to Prometheus pushGateway.
|
||||
*
|
||||
* @author lingxiao.wlx
|
||||
*/
|
||||
public class PrometheusPushGatewayContainer {
|
||||
|
||||
private final Logger logger = LoggerFactory.getLogger(PrometheusPushGatewayContainer.class);
|
||||
/**
|
||||
* {@link com.tencent.polaris.plugins.stat.prometheus.handler.PrometheusHandler} register the Collector with
|
||||
* CollectorRegistry.defaultRegistry.
|
||||
*/
|
||||
private final CollectorRegistry collectorRegistry = CollectorRegistry.defaultRegistry;
|
||||
private final String address;
|
||||
private final String job;
|
||||
private final PolarisStatProperties.ShutDownStrategy shutDownStrategy;
|
||||
private final PushGateway pushGateway;
|
||||
private final Map<String, String> groupingKey;
|
||||
|
||||
private final ScheduledExecutorService executorService;
|
||||
private final ScheduledFuture<?> scheduledFuture;
|
||||
|
||||
public PrometheusPushGatewayContainer(String address, Duration pushRate, String job,
|
||||
PolarisStatProperties.ShutDownStrategy shutDownStrategy, Map<String, String> groupingKey) {
|
||||
this.address = address;
|
||||
this.job = job;
|
||||
this.shutDownStrategy = shutDownStrategy;
|
||||
this.pushGateway = new PushGateway(address);
|
||||
this.groupingKey = groupingKey;
|
||||
this.executorService = Executors.newSingleThreadScheduledExecutor(new NamedThreadFactory("PrometheusPushGateway") {
|
||||
@Override
|
||||
public Thread newThread(Runnable r) {
|
||||
Thread thread = super.newThread(r);
|
||||
thread.setDaemon(true);
|
||||
return thread;
|
||||
}
|
||||
});
|
||||
this.scheduledFuture = executorService.scheduleAtFixedRate(() -> {
|
||||
this.push();
|
||||
if (logger.isDebugEnabled()) {
|
||||
logger.debug("push metrics to Prometheus pushGateway success!");
|
||||
}
|
||||
}, 0, pushRate.toMillis(), TimeUnit.MILLISECONDS);
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* Call by Spring to destroy PrometheusPushGatewayContainer instance.
|
||||
*/
|
||||
public void shutdown() {
|
||||
shutdown(this.shutDownStrategy);
|
||||
}
|
||||
|
||||
private void shutdown(PolarisStatProperties.ShutDownStrategy shutDownStrategy) {
|
||||
executorService.shutdown();
|
||||
scheduledFuture.cancel(false);
|
||||
if (Objects.isNull(shutDownStrategy)) {
|
||||
return;
|
||||
}
|
||||
switch (shutDownStrategy) {
|
||||
case PUSH:
|
||||
push();
|
||||
break;
|
||||
case DELETE:
|
||||
delete();
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
private void push() {
|
||||
try {
|
||||
pushGateway.pushAdd(collectorRegistry, this.job, this.groupingKey);
|
||||
}
|
||||
catch (UnknownHostException e) {
|
||||
logger.error("Unable to locate host {}. No longer attempting metrics publication to this host", this.address);
|
||||
// if cache UnknownHostException,shutdown task
|
||||
this.shutdown(null);
|
||||
}
|
||||
catch (Throwable t) {
|
||||
logger.error("Unable to push metrics to Prometheus pushGateway", t);
|
||||
}
|
||||
}
|
||||
|
||||
private void delete() {
|
||||
try {
|
||||
pushGateway.delete(this.job, this.groupingKey);
|
||||
}
|
||||
catch (Throwable t) {
|
||||
logger.error("Unable to delete metrics from Prometheus pushGateway", t);
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -1,102 +0,0 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* 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 com.tencent.cloud.rpc.enhancement.stat.plugin;
|
||||
|
||||
import java.util.Objects;
|
||||
|
||||
import com.tencent.cloud.rpc.enhancement.stat.config.PolarisStatProperties;
|
||||
import com.tencent.cloud.rpc.enhancement.stat.config.PolarisStatPropertiesAutoConfiguration;
|
||||
import com.tencent.cloud.rpc.enhancement.stat.config.plugin.PrometheusPushGatewayContainer;
|
||||
import org.junit.Test;
|
||||
import org.junit.jupiter.api.Assertions;
|
||||
|
||||
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
||||
import org.springframework.boot.test.context.runner.ApplicationContextRunner;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
|
||||
/**
|
||||
* Test for {@link PrometheusPushGatewayContainer}.
|
||||
*
|
||||
* @author lingxiao.wlx
|
||||
*/
|
||||
public class PrometheusPushGatewayContainerTest {
|
||||
|
||||
@Test
|
||||
public void testWithPushGatewayAndPushGatewayEnabled() {
|
||||
ApplicationContextRunner contextRunner = new ApplicationContextRunner()
|
||||
.withPropertyValues("spring.cloud.polaris.stat.pushgateway.enabled=true")
|
||||
.withPropertyValues("spring.cloud.polaris.stat.enabled=true")
|
||||
.withConfiguration(AutoConfigurations.of(PolarisStatPropertiesAutoConfiguration.class));
|
||||
contextRunner.run(context -> {
|
||||
assertThat(context).hasSingleBean(PrometheusPushGatewayContainer.class);
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testWithoutPushGatewayEnabled() {
|
||||
ApplicationContextRunner contextRunner = new ApplicationContextRunner()
|
||||
.withPropertyValues("spring.cloud.polaris.stat.pushgateway.enabled=false")
|
||||
.withPropertyValues("spring.cloud.polaris.stat.enabled=true")
|
||||
.withConfiguration(AutoConfigurations.of(PolarisStatPropertiesAutoConfiguration.class));
|
||||
contextRunner.run(context -> {
|
||||
assertThat(context).doesNotHaveBean(PrometheusPushGatewayContainer.class);
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testWithoutStatEnabled() {
|
||||
ApplicationContextRunner contextRunner = new ApplicationContextRunner()
|
||||
.withPropertyValues("spring.cloud.polaris.stat.pushgateway.enabled=true")
|
||||
.withPropertyValues("spring.cloud.polaris.stat.enabled=false")
|
||||
.withConfiguration(AutoConfigurations.of(PolarisStatPropertiesAutoConfiguration.class));
|
||||
contextRunner.run(context -> {
|
||||
assertThat(context).doesNotHaveBean(PrometheusPushGatewayContainer.class);
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testWithoutPushGatewayAndStatEnabled() {
|
||||
ApplicationContextRunner contextRunner = new ApplicationContextRunner()
|
||||
.withPropertyValues("spring.cloud.polaris.stat.pushgateway.enabled=false")
|
||||
.withPropertyValues("spring.cloud.polaris.stat.enabled=false")
|
||||
.withConfiguration(AutoConfigurations.of(PolarisStatPropertiesAutoConfiguration.class));
|
||||
contextRunner.run(context -> {
|
||||
assertThat(context).doesNotHaveBean(PrometheusPushGatewayContainer.class);
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testPushGatewayProperties() {
|
||||
ApplicationContextRunner contextRunner = new ApplicationContextRunner()
|
||||
.withPropertyValues("spring.cloud.polaris.stat.pushgateway.enabled=true")
|
||||
.withPropertyValues("spring.cloud.polaris.stat.enabled=true")
|
||||
.withPropertyValues("spring.cloud.polaris.stat.pushgateway.job=test")
|
||||
.withPropertyValues("spring.cloud.polaris.stat.pushgateway.grouping-keys.instance=test")
|
||||
.withPropertyValues("spring.cloud.polaris.stat.pushgateway.push-rate=1m")
|
||||
.withConfiguration(AutoConfigurations.of(PolarisStatPropertiesAutoConfiguration.class));
|
||||
contextRunner.run(context -> {
|
||||
assertThat(context).hasSingleBean(PrometheusPushGatewayContainer.class);
|
||||
PolarisStatProperties properties = context.getBean(PolarisStatProperties.class);
|
||||
PolarisStatProperties.PushGatewayProperties pushgateway = properties.getPushgateway();
|
||||
Assertions.assertFalse(Objects.isNull(pushgateway));
|
||||
Assertions.assertEquals(pushgateway.getJob(), "test");
|
||||
Assertions.assertEquals(pushgateway.getPushRate().toMillis(), 60000);
|
||||
});
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue