All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.rollbar.notifier.sender.json.JsonSerializerImpl Maven / Gradle / Ivy

Go to download

For connecting your applications built on the JVM to Rollbar for Error Reporting

There is a newer version: 2.0.0-alpha.1
Show newest version
package com.rollbar.notifier.sender.json;

import static java.util.regex.Pattern.compile;

import com.rollbar.api.json.JsonSerializable;
import com.rollbar.api.payload.Payload;
import com.rollbar.notifier.sender.result.Result;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * Implementation of the {@link JsonSerializer json serializer}.
 */
public class JsonSerializerImpl implements JsonSerializer {

  private static final Pattern CODE_PATTERN = compile("\"err\"\\s*:\\s*(\\d)");

  private static final Pattern MESSAGE_PATTERN = compile("\"message\"\\s*:\\s*\"([^\"]*)\"");

  private static final Pattern UUID_PATTERN = compile("\"uuid\"\\s*:\\s*\"([^\"]*)\"");

  private static final String[] REPLACEMENT_CHARS;

  static {
    REPLACEMENT_CHARS = new String[128];
    for (int i = 0; i <= 0x1f; i++) {
      REPLACEMENT_CHARS[i] = String.format("\\u%04x", (int) i);
    }
    REPLACEMENT_CHARS['"'] = "\\\"";
    REPLACEMENT_CHARS['\\'] = "\\\\";
    REPLACEMENT_CHARS['\t'] = "\\t";
    REPLACEMENT_CHARS['\b'] = "\\b";
    REPLACEMENT_CHARS['\n'] = "\\n";
    REPLACEMENT_CHARS['\r'] = "\\r";
    REPLACEMENT_CHARS['\f'] = "\\f";
  }

  private final boolean prettyPrint;

  /**
   * Construct a JsonSerializerImpl that does not pretty print the Payload.
   */
  public JsonSerializerImpl() {
    this(false);
  }

  /**
   * Construct a JsonSerializerImpl.
   * @param prettyPrint whether or not to pretty print the payload.
   */
  public JsonSerializerImpl(boolean prettyPrint) {
    this.prettyPrint = prettyPrint;
  }

  @Override
  public String toJson(Payload payload) {
    if (payload.json != null) {
      return payload.json;
    }

    StringBuilder builder = new StringBuilder();
    serializeValue(builder, payload, 0);
    return builder.toString();
  }

  @Override
  public Result resultFrom(String response) {
    Matcher codeMatcher = CODE_PATTERN.matcher(response);
    codeMatcher.find();
    String codeStr = codeMatcher.group(1);
    int code;
    try {
      code = Integer.parseInt(codeStr);
    } catch (Exception e) {
      code = -1;
    }
    Pattern p;
    if (code == 0) {
      p = UUID_PATTERN;
    } else {
      p = MESSAGE_PATTERN;
    }
    Matcher m = p.matcher(response);
    m.find();

    String body = m.group(1);

    return new Result.Builder()
        .code(code)
        .body(body)
        .build();
  }


  private void serializeObject(Map content, StringBuilder builder, int level) {
    builder.append('{');

    String comma = "";
    for (Map.Entry entry : content.entrySet()) {
      builder.append(comma);
      comma = ",";

      if (prettyPrint) {
        builder.append("\n");
        indent(builder, level);
      }
      serializeString(builder, entry.getKey());

      builder.append(':');
      if (prettyPrint) {
        builder.append(" ");
      }

      serializeValue(builder, entry.getValue(), level + 1);
    }
    if (prettyPrint) {
      builder.append("\n");
    }

    builder.append('}');
  }

  private void serializeValue(StringBuilder builder, Object value, int level) {
    if (value == null) {
      serializeNull(builder);
    } else if (value instanceof Boolean) {
      serializeBoolean(builder, (Boolean) value);
    } else if (value instanceof Number) {
      serializeNumber(builder, (Number) value);
    } else if (value instanceof String) {
      serializeString(builder, (String) value);
    } else if (value instanceof JsonSerializable) {
      serializeValue(builder, ((JsonSerializable) value).asJson(), level);
    } else if (value instanceof Map) {
      Map obj = asMap((Map) value);
      serializeObject(obj, builder, level);
    } else if (value instanceof Collection) {
      serializeArray(builder, ((Collection) value).toArray(), level);
    } else if (value instanceof Object[]) {
      serializeArray(builder, (Object[]) value, level);
    } else if (value instanceof Throwable) {
      serializeThrowable(builder, (Throwable) value);
    } else {
      serializeDefault(builder, value);
    }
  }

  private static void serializeThrowable(StringBuilder builder, Throwable value) {
    final StringWriter writer = new StringWriter();
    value.printStackTrace(new PrintWriter(writer));
    builder.append(String.format("\"%s\"", writer.toString()));
  }

  private static void serializeDefault(StringBuilder builder, Object value) {
    builder.append(String.format("\"%s\"", value));
  }

  private static void serializeNumber(StringBuilder builder, Number value) {
    builder.append(value);
  }

  private static void serializeBoolean(StringBuilder builder, Boolean value) {
    builder.append(value ? "true" : "false");
  }

  private static void serializeNull(StringBuilder builder) {
    builder.append("null");
  }

  private void serializeArray(StringBuilder builder, Object[] array, int level) {
    builder.append('[');
    String comma = "";
    for (Object obj : array) {
      builder.append(comma);
      comma = ",";

      if (prettyPrint) {
        builder.append("\n");
        indent(builder, level);
      }
      serializeValue(builder, obj, level + 1);
    }
    builder.append(']');
  }

  private static Map asMap(Map value) {
    try {
      @SuppressWarnings("unchecked")
      Map obj = (Map) value;
      return obj;
    } catch (ClassCastException e) {
      Map obj = new LinkedHashMap<>();
      for (Object o : value.entrySet()) {
        Map.Entry thisOne = (Map.Entry) o;
        Object key = thisOne.getKey();
        Object val = thisOne.getValue();
        obj.put(key.toString(), val);
      }
      return obj;
    }
  }

  // Borrowed from
  // https://github.com/google/gson/blob/59edfc1caf2bb30e30f523f8502f23e8f8edc38e/gson/src/main/java/com/google/gson/stream/JsonWriter.java
  private static void serializeString(StringBuilder builder, String str) {
    builder.append('"');
    int last = 0;
    int length = str.length();
    for (int i = 0; i < length; i++) {
      char c = str.charAt(i);
      String replacement;
      if (c < 128) {
        replacement = REPLACEMENT_CHARS[c];
        if (replacement == null) {
          continue;
        }
      } else if (c == '\u2028') {
        replacement = "\\u2028";
      } else if (c == '\u2029') {
        replacement = "\\u2029";
      } else {
        continue;
      }
      if (last < i) {
        builder.append(str, last, i);
      }
      builder.append(replacement);
      last = i + 1;
    }
    if (last < length) {
      builder.append(str, last, length);
    }
    builder.append('"');
  }

  private static void indent(StringBuilder builder, int i) {
    for (int x = 0; x <= i; x++) {
      builder.append("  ");
    }
  }

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy