Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/**
* Copyright 2015 Groupon.com
*
* 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 com.arpnetworking.metrics.impl;
import ch.qos.logback.classic.encoder.PatternLayoutEncoder;
import ch.qos.logback.classic.spi.ILoggingEvent;
import ch.qos.logback.core.encoder.Encoder;
import com.arpnetworking.metrics.CompoundUnit;
import com.arpnetworking.metrics.Event;
import com.arpnetworking.metrics.Quantity;
import com.arpnetworking.metrics.Unit;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.module.SimpleModule;
import org.slf4j.LoggerFactory;
import java.io.IOException;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import javax.annotation.Nullable;
/**
* Implementation of Sink for the query log file.
*
* @author Ville Koskela (vkoskela at groupon dot com)
*/
public class TsdLogSink extends BaseFileSink {
/**
* {@inheritDoc}
*/
@Override
public void record(final Event event) {
try {
getMetricsLogger().info(_objectMapper.writeValueAsString(event));
} catch (final IOException e) {
_logger.warn("Exception recording event", e);
}
}
private static Encoder createEncoder(final boolean immediateFlush) {
final PatternLayoutEncoder encoder = new PatternLayoutEncoder();
encoder.setPattern("%msg%n");
encoder.setImmediateFlush(immediateFlush);
return encoder;
}
/**
* Protected constructor.
*
* @param builder Instance of Builder.
*/
protected TsdLogSink(final Builder builder) {
this(builder, OBJECT_MAPPER, LOGGER);
}
// NOTE: Package private for testing
/* package private */ TsdLogSink(final Builder builder, final ObjectMapper objectMapper, final org.slf4j.Logger logger) {
super(builder, createEncoder(builder._immediateFlush.booleanValue()));
_objectMapper = objectMapper;
_logger = logger;
}
private final ObjectMapper _objectMapper;
private final org.slf4j.Logger _logger;
private static final ObjectMapper OBJECT_MAPPER = new ObjectMapper();
private static final org.slf4j.Logger LOGGER = LoggerFactory.getLogger(StenoLogSink.class);
static {
final SimpleModule simpleModule = new SimpleModule("TsdLogSink");
simpleModule.addSerializer(Event.class, EventSerializer.newInstance());
simpleModule.addSerializer(Quantity.class, QuantitySerializer.newInstance());
OBJECT_MAPPER.setSerializationInclusion(JsonInclude.Include.ALWAYS);
OBJECT_MAPPER.registerModule(simpleModule);
}
private static final class EventSerializer extends JsonSerializer {
public static JsonSerializer newInstance() {
return new EventSerializer();
}
@Override
public void serialize(
final Event event,
final JsonGenerator jsonGenerator,
final SerializerProvider provider)
throws IOException {
jsonGenerator.writeStartObject();
jsonGenerator.writeStringField("version", VERSION);
jsonGenerator.writeObjectField("annotations", event.getAnnotations());
if (!event.getCounterSamples().isEmpty()) {
jsonGenerator.writeObjectFieldStart("counters");
serializeSamples(event.getCounterSamples(), jsonGenerator);
jsonGenerator.writeEndObject();
}
if (!event.getGaugeSamples().isEmpty()) {
jsonGenerator.writeObjectFieldStart("gauges");
serializeSamples(event.getGaugeSamples(), jsonGenerator);
jsonGenerator.writeEndObject();
}
if (!event.getTimerSamples().isEmpty()) {
jsonGenerator.writeObjectFieldStart("timers");
serializeSamples(event.getTimerSamples(), jsonGenerator);
jsonGenerator.writeEndObject();
}
jsonGenerator.writeEndObject();
}
private void serializeSamples(
final Map> samples,
final JsonGenerator jsonGenerator)
throws IOException {
for (final Map.Entry> entry : samples.entrySet()) {
jsonGenerator.writeObjectFieldStart(entry.getKey());
jsonGenerator.writeObjectField("values", entry.getValue());
jsonGenerator.writeEndObject();
}
}
private EventSerializer() {}
private static final String VERSION = "2f";
}
private static final class QuantitySerializer extends JsonSerializer {
public static JsonSerializer newInstance() {
return new QuantitySerializer();
}
@Override
public void serialize(
final Quantity valueWithUnit,
final JsonGenerator jsonGenerator,
final SerializerProvider provider)
throws IOException {
jsonGenerator.writeStartObject();
jsonGenerator.writeObjectField("value", valueWithUnit.getValue());
@Nullable final Unit unit = valueWithUnit.getUnit();
if (unit != null) {
final Unit simplifiedUnit = new TsdCompoundUnit.Builder()
.addNumeratorUnit(unit)
.build();
if (simplifiedUnit instanceof CompoundUnit) {
final CompoundUnit compoundUnit = (CompoundUnit) simplifiedUnit;
if (!compoundUnit.getNumeratorUnits().isEmpty()) {
writeUnits(jsonGenerator, compoundUnit.getNumeratorUnits(), "unitNumerators");
}
if (!compoundUnit.getDenominatorUnits().isEmpty()) {
writeUnits(jsonGenerator, compoundUnit.getDenominatorUnits(), "unitDenominators");
}
} else if (simplifiedUnit != null) {
writeUnits(jsonGenerator, Collections.singletonList(simplifiedUnit), "unitNumerators");
}
}
jsonGenerator.writeEndObject();
}
private void writeUnits(
final JsonGenerator jsonGenerator,
final List units,
final String name) throws IOException {
jsonGenerator.writeArrayFieldStart(name);
for (final Unit unit : units) {
jsonGenerator.writeString(unit.getName());
}
jsonGenerator.writeEndArray();
}
}
/**
* Builder for TsdLogSink.
*
* This class is thread safe.
*
* @author Ville Koskela (vkoskela at groupon dot com)
*/
public static class Builder extends BaseFileSink.Builder {
/**
* {@inheritDoc}
*/
@Override
protected TsdLogSink createSink() {
return new TsdLogSink(this);
}
/**
* {@inheritDoc}
*/
@Override
protected Builder self() {
return this;
}
}
}
