com.newrelic.telemetry.logs.json.LogJsonTelemetryBlockWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of telemetry-core Show documentation

Used to send telemetry data to New Relic

There is a newer version: 0.16.0

/*
 * Copyright 2020 New Relic Corporation. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 */
package com.newrelic.telemetry.logs.json;

import com.google.gson.stream.JsonWriter;
import com.newrelic.telemetry.json.AttributesJson;
import com.newrelic.telemetry.logs.Log;
import com.newrelic.telemetry.logs.LogBatch;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.PrintStream;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

public final class LogJsonTelemetryBlockWriter {

  private final AttributesJson attributesJson;

  public LogJsonTelemetryBlockWriter(AttributesJson attributesJson) {
    this.attributesJson = attributesJson;
  }

  public void appendTelemetryJson(LogBatch batch, JsonWriter jsonWriter) {
    try {
      jsonWriter.name("logs");
      jsonWriter.beginArray();
      Collection telemetry = batch.getTelemetry();
      for (Log log : telemetry) {
        jsonWriter.beginObject();
        jsonWriter.name("timestamp").value(log.getTimestamp());
        jsonWriter.name("attributes").jsonValue(attributesJson.toJson(enhanceAttributes(log)));
        if (log.getMessage() != null) {
          jsonWriter.name("message").value(log.getMessage());
        }
        jsonWriter.endObject();
      }
      jsonWriter.endArray();
    } catch (IOException e) {
      throw new RuntimeException("Failed to generate span telemetry json", e);
    }
  }

  private Map enhanceAttributes(Log log) {
    Map result = new HashMap<>(log.getAttributes().asMap());
    result.putIfAbsent("service.name", log.getServiceName());
    if (log.getLevel() != null) {
      result.put("log.level", log.getLevel());
    }
    Throwable throwable = log.getThrowable();
    if (throwable != null) {
      ByteArrayOutputStream bytes = new ByteArrayOutputStream();
      throwable.printStackTrace(new PrintStream(bytes));
      result.put("error.message", throwable.getMessage());
      result.put("error.class", throwable.getClass().getName());
      result.put("error.stack", bytes.toString());
    }
    return result;
  }

  @Override
  public String toString() {
    return "LogJsonTelemetryBlockWriter{" + "attributesJson=" + attributesJson + '}';
  }
}