Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.atlan.serde.AtlanRequestSerializer Maven / Gradle / Ivy
/* SPDX-License-Identifier: Apache-2.0
Copyright 2022 Atlan Pte. Ltd. */
package com.atlan.serde;
import com.atlan.AtlanClient;
import com.atlan.exception.AtlanException;
import com.atlan.exception.NotFoundException;
import com.atlan.model.admin.*;
import com.atlan.model.assets.Asset;
import com.atlan.util.JacksonUtils;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.jsontype.TypeSerializer;
import com.fasterxml.jackson.databind.ser.std.StdSerializer;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
/**
* Serialization of all {@link AtlanRequest} objects.
* This custom serialization is necessary to create some specific aspects of complexity in Atlan's payloads:
*
* The nested payload structures, which varies depending on Atlan tag and custom metadata details.
*
*/
public class AtlanRequestSerializer extends StdSerializer {
private static final long serialVersionUID = 2L;
private final AtlanClient client;
public AtlanRequestSerializer(AtlanClient client) {
this(AtlanRequest.class, client);
}
public AtlanRequestSerializer(Class t, AtlanClient client) {
super(t);
this.client = client;
}
/**
* {@inheritDoc}
*/
@Override
public void serializeWithType(
AtlanRequest value, JsonGenerator gen, SerializerProvider serializers, TypeSerializer typeSer)
throws IOException {
serialize(value, gen, serializers);
}
/**
* {@inheritDoc}
*/
@Override
public void serialize(AtlanRequest request, JsonGenerator gen, SerializerProvider sp)
throws IOException, JsonProcessingException {
String destinationAttribute = request.getDestinationAttribute();
Object translatedPayload = null;
if (request instanceof AtlanTagRequest) {
AtlanTagPayload cls = ((AtlanTagRequest) request).getPayload();
String clsName = cls.getTypeName();
String clsId;
try {
clsId = client.getAtlanTagCache().getIdForName(clsName);
} catch (NotFoundException e) {
clsId = AtlanClient.DELETED_AUDIT_OBJECT;
} catch (AtlanException e) {
throw new IOException("Unable to translate Atlan tag with name: " + clsName, e);
}
translatedPayload = cls.toBuilder().typeName(clsId).build();
} else if (request instanceof CustomMetadataRequest) {
CustomMetadataPayload cm = ((CustomMetadataRequest) request).getPayload();
String cmName = destinationAttribute;
Map attrValues = new HashMap<>();
try {
destinationAttribute = client.getCustomMetadataCache().getIdForName(cmName);
client.getCustomMetadataCache()
.getAttributesFromCustomMetadata(destinationAttribute, cmName, cm.getAttributes(), attrValues);
} catch (NotFoundException e) {
destinationAttribute = AtlanClient.DELETED_AUDIT_OBJECT;
} catch (AtlanException e) {
throw new IOException("Unable to find custom metadata with name: " + cmName, e);
}
translatedPayload = attrValues;
}
gen.writeStartObject();
JacksonUtils.serializeString(gen, "id", request.getId());
JacksonUtils.serializeString(gen, "version", request.getVersion());
JacksonUtils.serializeBoolean(gen, "isActive", request.getIsActive());
JacksonUtils.serializeLong(gen, "createdAt", request.getCreatedAt());
JacksonUtils.serializeLong(gen, "updatedAt", request.getUpdatedAt());
JacksonUtils.serializeString(gen, "createdBy", request.getCreatedBy());
JacksonUtils.serializeString(gen, "tenantId", request.getTenantId());
JacksonUtils.serializeString(gen, "sourceType", request.getSourceType());
JacksonUtils.serializeString(gen, "sourceGuid", request.getSourceGuid());
JacksonUtils.serializeString(gen, "sourceQualifiedName", request.getSourceQualifiedName());
JacksonUtils.serializeString(gen, "sourceAttribute", request.getSourceAttribute());
JacksonUtils.serializeString(gen, "destinationGuid", request.getDestinationGuid());
JacksonUtils.serializeString(gen, "destinationQualifiedName", request.getDestinationQualifiedName());
JacksonUtils.serializeString(gen, "destinationAttribute", destinationAttribute);
JacksonUtils.serializeString(gen, "destinationValue", request.getDestinationValue());
JacksonUtils.serializeString(gen, "destinationValueType", request.getDestinationValueType());
JacksonUtils.serializeString(gen, "entityType", request.getEntityType());
JacksonUtils.serializeObject(gen, "requestType", request.getRequestType());
JacksonUtils.serializeString(gen, "approvedBy", request.getApprovedBy());
JacksonUtils.serializeString(gen, "rejectedBy", request.getRejectedBy());
JacksonUtils.serializeObject(gen, "status", request.getStatus());
JacksonUtils.serializeString(gen, "message", request.getMessage());
JacksonUtils.serializeString(gen, "approvalType", request.getApprovalType());
JacksonUtils.serializeLong(gen, "hash", request.getHash());
JacksonUtils.serializeBoolean(gen, "isDuplicate", request.getIsDuplicate());
JacksonUtils.serializeString(gen, "destinationValueAction", request.getDestinationValueAction());
JacksonUtils.serializeObject(gen, "requestApproverUsers", request.getRequestApproverUsers());
JacksonUtils.serializeObject(gen, "requestApproverGroups", request.getRequestApproverGroups());
JacksonUtils.serializeObject(gen, "requestApproverRoles", request.getRequestApproverRoles());
JacksonUtils.serializeObject(gen, "requestDenyUsers", request.getRequestDenyUsers());
JacksonUtils.serializeObject(gen, "requestDenyGroups", request.getRequestDenyGroups());
JacksonUtils.serializeObject(gen, "requestDenyRoles", request.getRequestDenyRoles());
JacksonUtils.serializeObject(gen, "payload", translatedPayload);
Asset destinationAsset = request.getDestinationEntity();
if (destinationAsset != null) {
// Remove typeName when serializing any asset back
JacksonUtils.serializeObject(
gen,
"destinationEntity",
destinationAsset.toBuilder().typeName(null).build());
}
/* Unused.
Object destinationValueArray;
Object destinationValueObject;
Object confidenceScore;
Object botRunId;
Object requestsBatch;
Object assignedApprovers;
Object accessStartDate;
Object accessEndDate;
Object sourceEntity;
*/
gen.writeEndObject();
}
}
