MirrorMirror
/
Signal-Server
mirror of https://github.com/signalapp/Signal-Server.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Report Dropwizard metrics via the Wavefront proxy.

This commit is contained in:
Jon Chambers 2020-11-13 14:42:23 -05:00 committed by Jon Chambers
parent 251e1b51c5
commit 0a23b57ff8
2 changed files with 12 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
service/src/main/java/org/whispersystems/textsecuregcm/metrics/JsonMetricsReporter.java
View File

@ -21,6 +21,7 @@ import com.fasterxml.jackson.core.JsonGenerator;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.ws.rs.core.UriBuilder;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.net.InetAddress;
@ -44,13 +45,13 @@ public class JsonMetricsReporter extends ScheduledReporter {
private final URI uri;
private final HttpClient httpClient;
public JsonMetricsReporter(MetricRegistry registry, String token, String hostname,
public JsonMetricsReporter(MetricRegistry registry, URI uri,
MetricFilter filter, TimeUnit rateUnit, TimeUnit durationUnit)
throws UnknownHostException
{
super(registry, "json-reporter", filter, rateUnit, durationUnit);
this.httpClient = HttpClient.newBuilder().version(HttpClient.Version.HTTP_2).build();
this.uri = URI.create(String.format("https://%s/report/metrics?t=%s&h=%s", hostname, token, InetAddress.getLocalHost().getHostName()));
this.uri = UriBuilder.fromUri(uri).queryParam("h", InetAddress.getLocalHost().getHostName()).build();
}
@Override
@ -210,8 +211,7 @@ public class JsonMetricsReporter extends ScheduledReporter {
private MetricFilter filter = MetricFilter.ALL;
private TimeUnit rateUnit = TimeUnit.SECONDS;
private TimeUnit durationUnit = TimeUnit.MILLISECONDS;
private String token;
private String hostname;
private URI uri;
private Builder(MetricRegistry registry) {
this.registry = registry;
@ -235,26 +235,17 @@ public class JsonMetricsReporter extends ScheduledReporter {
return this;
}
public Builder withToken(String token) {
this.token = token;
return this;
}
public Builder withHostname(String hostname) {
this.hostname = hostname;
public Builder withUri(URI uri) {
this.uri = uri;
return this;
}
public JsonMetricsReporter build() throws UnknownHostException {
if (hostname == null) {
throw new IllegalArgumentException("No hostname specified!");
if (uri == null) {
throw new IllegalArgumentException("No URI specified!");
}
if (token == null) {
throw new IllegalArgumentException("No token specified!");
}
return new JsonMetricsReporter(registry, token, hostname, filter, rateUnit, durationUnit);
return new JsonMetricsReporter(registry, uri, filter, rateUnit, durationUnit);
}
}
}

10
service/src/main/java/org/whispersystems/textsecuregcm/metrics/JsonMetricsReporterFactory.java
View File

@ -11,6 +11,7 @@ import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.annotation.JsonTypeName;
import javax.validation.constraints.NotNull;
import java.net.URI;
import java.net.UnknownHostException;
import io.dropwizard.metrics.BaseReporterFactory;
@ -20,18 +21,13 @@ public class JsonMetricsReporterFactory extends BaseReporterFactory {
@JsonProperty
@NotNull
private String hostname;
@JsonProperty
@NotNull
private String token;
private URI uri;
@Override
public ScheduledReporter build(MetricRegistry metricRegistry) {
try {
return JsonMetricsReporter.forRegistry(metricRegistry)
.withHostname(hostname)
.withToken(token)
.withUri(uri)
.convertRatesTo(getRateUnit())
.convertDurationsTo(getDurationUnit())
.filter(getFilter())