datahub.event.EventFormatter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of datahub-client Show documentation

DataHub Java client for metadata integration

There is a newer version: 0.14.1

package datahub.event;

import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.linkedin.common.urn.Urn;
import com.linkedin.data.ByteString;
import com.linkedin.data.template.JacksonDataTemplateCodec;
import com.linkedin.mxe.GenericAspect;
import com.linkedin.mxe.MetadataChangeProposal;

import java.io.IOException;
import java.net.URISyntaxException;
import java.nio.charset.StandardCharsets;
import lombok.SneakyThrows;


/**
 * A class that helps to format Metadata events for transport
 */
public class EventFormatter {

  private final ObjectMapper objectMapper = new ObjectMapper().setSerializationInclusion(JsonInclude.Include.NON_NULL);

  private final JacksonDataTemplateCodec dataTemplateCodec = new JacksonDataTemplateCodec(objectMapper.getFactory());
  private final Format serializationFormat;

  public EventFormatter(Format serializationFormat) {
    this.serializationFormat = serializationFormat;
  }

  public EventFormatter() {
    this(Format.PEGASUS_JSON);
  }

  @SneakyThrows(URISyntaxException.class)
  public MetadataChangeProposal convert(MetadataChangeProposalWrapper mcpw) throws IOException {
    
    String serializedAspect = StringEscapeUtils.escapeJava(dataTemplateCodec.dataTemplateToString(mcpw.getAspect()));
    MetadataChangeProposal mcp = new MetadataChangeProposal().setEntityType(mcpw.getEntityType())
        .setAspectName(mcpw.getAspectName())
        .setEntityUrn(Urn.createFromString(mcpw.getEntityUrn()))
        .setChangeType(mcpw.getChangeType());

    switch (this.serializationFormat) {
      case PEGASUS_JSON: {
        mcp.setAspect(new GenericAspect().setContentType("application/json")
            .setValue(ByteString.unsafeWrap(serializedAspect.getBytes(StandardCharsets.UTF_8))));
      }
      break;
      default:
        throw new EventValidationException("Cannot handle serialization format " + this.serializationFormat);
    }
    return mcp;
  }

  public enum Format {
    PEGASUS_JSON,
  }
}