109 lines
3.9 KiB
Java
109 lines
3.9 KiB
Java
/*
|
|
* Copyright (c) 2014 Hewlett-Packard Development Company, L.P.
|
|
*
|
|
* 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
|
|
*
|
|
* http://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 monasca.api.app;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Collections;
|
|
import java.util.Comparator;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.Map.Entry;
|
|
|
|
import javax.annotation.Nullable;
|
|
import javax.inject.Inject;
|
|
|
|
import kafka.javaapi.producer.Producer;
|
|
import kafka.producer.KeyedMessage;
|
|
|
|
import com.codahale.metrics.Meter;
|
|
import com.codahale.metrics.MetricRegistry;
|
|
import com.google.common.base.Strings;
|
|
import com.google.common.collect.ImmutableMap;
|
|
import com.google.common.collect.ImmutableMap.Builder;
|
|
|
|
import monasca.api.MonApiConfiguration;
|
|
import monasca.common.model.metric.Metric;
|
|
import monasca.common.model.metric.MetricEnvelope;
|
|
import monasca.common.model.metric.MetricEnvelopes;
|
|
|
|
/**
|
|
* Metric service implementation.
|
|
*/
|
|
public class MetricService {
|
|
private final MonApiConfiguration config;
|
|
private final Producer<String, String> producer;
|
|
private final Meter metricMeter;
|
|
private final Meter batchMeter;
|
|
|
|
@Inject
|
|
public MetricService(MonApiConfiguration config, Producer<String, String> producer,
|
|
MetricRegistry metricRegistry) {
|
|
this.config = config;
|
|
this.producer = producer;
|
|
metricMeter =
|
|
metricRegistry.meter(MetricRegistry.name(MetricService.class, "metrics.published"));
|
|
batchMeter =
|
|
metricRegistry.meter(MetricRegistry.name(MetricService.class, "batches.published"));
|
|
}
|
|
|
|
public void create(List<Metric> metrics, String tenantId, @Nullable String crossTenantId) {
|
|
String metricTenantId = Strings.isNullOrEmpty(crossTenantId) ? tenantId : crossTenantId;
|
|
Builder<String, Object> metaBuilder =
|
|
new ImmutableMap.Builder<String, Object>().put("tenantId", metricTenantId).put("region",
|
|
config.region);
|
|
ImmutableMap<String, Object> meta = metaBuilder.build();
|
|
|
|
List<KeyedMessage<String, String>> keyedMessages = new ArrayList<>(metrics.size());
|
|
for (Metric metric : metrics) {
|
|
MetricEnvelope envelope = new MetricEnvelope(metric, meta);
|
|
keyedMessages.add(new KeyedMessage<>(config.metricsTopic, buildKey(metricTenantId, metric),
|
|
MetricEnvelopes.toJson(envelope)));
|
|
metricMeter.mark();
|
|
}
|
|
|
|
producer.send(keyedMessages);
|
|
batchMeter.mark();
|
|
}
|
|
|
|
private String buildKey(String metricTenantId, Metric metric) {
|
|
final StringBuilder key = new StringBuilder(metricTenantId);
|
|
key.append(metric.name);
|
|
|
|
// Dimensions are optional.
|
|
if (metric.dimensions != null && !metric.dimensions.isEmpty()) {
|
|
|
|
// Key must be the same for the same metric so sort the dimensions so they will be
|
|
// in a known order
|
|
for (final Map.Entry<String, String> dim : buildSortedDimSet(metric.dimensions)) {
|
|
key.append(dim.getKey());
|
|
key.append(dim.getValue());
|
|
}
|
|
}
|
|
String keyValue = key.toString();
|
|
return keyValue;
|
|
}
|
|
|
|
private List<Map.Entry<String, String>> buildSortedDimSet(final Map<String, String> dimMap) {
|
|
final List<Map.Entry<String, String>> dims = new ArrayList<>(dimMap.entrySet());
|
|
Collections.sort(dims, new Comparator<Map.Entry<String, String>>() {
|
|
@Override
|
|
public int compare(Entry<String, String> o1, Entry<String, String> o2) {
|
|
int nameCmp = o1.getKey().compareTo(o2.getKey());
|
|
return (nameCmp != 0 ? nameCmp : o1.getValue().compareTo(o2.getValue()));
|
|
}
|
|
});
|
|
return dims;
|
|
}
|
|
}
|