com.gruelbox.transactionoutbox.jackson.JacksonInvocationSerializer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of transactionoutbox-jackson Show documentation

A safe implementation of the transactional outbox pattern for Java (Jackson extension library)

There is a newer version: 5.5.447

Show newest version

package com.gruelbox.transactionoutbox.jackson;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.gruelbox.transactionoutbox.Beta;
import com.gruelbox.transactionoutbox.DefaultInvocationSerializer;
import com.gruelbox.transactionoutbox.Invocation;
import com.gruelbox.transactionoutbox.InvocationSerializer;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.Reader;
import java.io.Writer;
import lombok.Builder;

/**
 * A general-purpose {@link InvocationSerializer} which can handle pretty much anything that you
 * throw at it.
 *
 * Note that if there is any risk that you might not trust the source of the serialized {@link
 * Invocation}, do not use this. This serializer is vulnerable to a
 * deserialization of untrusted data vulnerability (more information here)
 * which is why it is not included in the core library.
 */
@Beta
public final class JacksonInvocationSerializer implements InvocationSerializer {
  private final ObjectMapper mapper;
  private final InvocationSerializer defaultInvocationSerializer;

  @Builder
  private JacksonInvocationSerializer(
      ObjectMapper mapper, DefaultInvocationSerializer defaultInvocationSerializer) {
    this.mapper = mapper.copy();
    this.defaultInvocationSerializer = defaultInvocationSerializer;
    this.mapper.setDefaultTyping(TransactionOutboxJacksonModule.typeResolver());
    this.mapper.registerModule(new TransactionOutboxJacksonModule());
  }

  @Override
  public void serializeInvocation(Invocation invocation, Writer writer) {
    try {
      mapper.writeValue(writer, invocation);
    } catch (IOException e) {
      throw new RuntimeException(e);
    }
  }

  @Override
  public Invocation deserializeInvocation(Reader reader) {
    try {
      // read ahead to check if old style
      BufferedReader br = new BufferedReader(reader);
      if (checkForOldSerialization(br)) {
        if (defaultInvocationSerializer == null) {
          throw new UnsupportedOperationException(
              "Can't deserialize GSON-format tasks without a "
                  + DefaultInvocationSerializer.class.getSimpleName()
                  + ". Supply one when building "
                  + getClass().getSimpleName());
        }
        return defaultInvocationSerializer.deserializeInvocation(br);
      }
      return mapper.readValue(br, Invocation.class);
    } catch (IOException e) {
      throw new RuntimeException(e);
    }
  }

  private boolean checkForOldSerialization(BufferedReader reader) throws IOException {
    reader.mark(1);
    char[] chars = new char[6];
    int charsRead = reader.read(chars, 0, 6);

    String result = "";
    if (charsRead != -1) {
      result = new String(chars, 0, charsRead);
    }
    reader.reset();
    return result.startsWith("{\"c\":");
  }
}