com.rollbar.notifier.util.BodyFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rollbar-java Show documentation
For connecting your applications built on the JVM to Rollbar for Error Reporting
The newest version!
package com.rollbar.notifier.util;

import com.rollbar.api.payload.data.TelemetryEvent;
import com.rollbar.api.payload.data.body.Body;
import com.rollbar.api.payload.data.body.ExceptionInfo;
import com.rollbar.api.payload.data.body.Frame;
import com.rollbar.api.payload.data.body.Message;
import com.rollbar.api.payload.data.body.Trace;
import com.rollbar.api.payload.data.body.TraceChain;
import com.rollbar.jvmti.CacheFrame;
import com.rollbar.jvmti.ThrowableCache;
import com.rollbar.notifier.wrapper.RollbarThrowableWrapper;
import com.rollbar.notifier.wrapper.ThrowableWrapper;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * Body factory helper to build the proper body depending on the throwable or the description.
 */
public class BodyFactory {

  /**
   * Builds the body for the throwable and description supplied.
   *
   * @param throwable   the throwable.
   * @param description the description.
   * @return the body.
   * @deprecated Replaced by {@link #from(ThrowableWrapper, String)}.
   */
  @Deprecated
  public Body from(Throwable throwable, String description) {
    if (throwable == null) {
      return new Body.Builder().bodyContent(message(description)).build();
    }
    return from(new RollbarThrowableWrapper(throwable), description);
  }

  /**
   * Builds the body from the {@link ThrowableWrapper throwableWrapper} and the description
   * supplied.
   *
   * @param throwableWrapper the throwable proxy.
   * @param description      the description.
   * @return the body.
   */
  public Body from(ThrowableWrapper throwableWrapper, String description) {
    Body.Builder builder = new Body.Builder();
    return from(throwableWrapper, description, builder);
  }

  /**
   * Builds the body from the {@link ThrowableWrapper throwableWrapper}, the description
   * supplied and telemetry events.
   *
   * @param throwableWrapper the throwable proxy.
   * @param description      the description.
   * @param telemetryEvents  the telemetry events.
   * @return the body.
   */
  public Body from(
      ThrowableWrapper throwableWrapper,
      String description,
      List telemetryEvents
  ) {
    Body.Builder builder = new Body.Builder().telemetryEvents(telemetryEvents);
    return from(throwableWrapper, description, builder);
  }

  private Body from(ThrowableWrapper throwableWrapper, String description, Body.Builder builder) {
    if (throwableWrapper == null) {
      return builder.bodyContent(message(description)).build();
    }
    if (throwableWrapper.getCause() == null) {
      return builder.bodyContent(trace(throwableWrapper, description)).build();
    }
    return builder.bodyContent(traceChain(throwableWrapper, description)).build();
  }

  private static Message message(String description) {
    return new Message.Builder()
      .body(description)
      .build();
  }

  private static Trace trace(ThrowableWrapper throwableWrapper, String description) {
    return new Trace.Builder()
      .frames(frames(throwableWrapper))
      .exception(info(throwableWrapper, description))
      .build();
  }

  private static TraceChain traceChain(ThrowableWrapper throwableWrapper, String description) {
    ArrayList chain = new ArrayList<>();
    do {
      chain.add(trace(throwableWrapper, description));
      description = null;
      throwableWrapper = throwableWrapper.getCause();
    } while (throwableWrapper != null);
    return new TraceChain.Builder()
        .traces(chain)
        .build();
  }

  private static List frames(ThrowableWrapper throwableWrapper) {
    StackTraceElement[] elements = throwableWrapper.getStackTrace();
    CacheFrame[] cachedFrames = ThrowableCache.get(throwableWrapper.getThrowable());
    int j = 0;
    if (cachedFrames != null) {
      j = cachedFrames.length - 1;
    }

    ArrayList result = new ArrayList<>();
    for (int i = elements.length - 1; i >= 0; i--, j--) {
      StackTraceElement element = elements[i];
      Map locals = null;
      if (cachedFrames != null) {
        while (j >= 0 && !cachedFrames[j].getMethod().getName().equals(element.getMethodName())) {
          j--;
        }
        if (j >= 0) {
          locals = cachedFrames[j].getLocals();
        }
      }
      result.add(frame(element, locals));
    }

    return result;
  }

  private static ExceptionInfo info(ThrowableWrapper throwableWrapper, String description) {
    String className = throwableWrapper.getClassName();
    String message = throwableWrapper.getMessage();
    return new ExceptionInfo.Builder()
        .className(className)
        .message(message)
        .description(description)
        .build();
  }

  private static Frame frame(StackTraceElement element, Map locals) {
    String filename = element.getFileName();
    Integer lineNumber = element.getLineNumber();
    String method = element.getMethodName();
    String className = element.getClassName();

    return new Frame.Builder()
        .filename(filename)
        .lineNumber(lineNumber)
        .method(method)
        .className(className)
        .locals(locals)
        .build();
  }
}