Wednesday 20 July 2022

Atlas client: Delete entity by unique attribute

AtlasClientV2#deleteEntityByAttribute method is used to delete an entity by unique attribute search.

 

Example

Map<String, String> uniqAttributes = new HashMap<>();
uniqAttributes.put("qualifiedName", "DeleteEntityByUniqueAttribute_DEMO1");
EntityMutationResponse entityMutationResponse = atlasClient.deleteEntityByAttribute(TYPE_NAME, uniqAttributes);

 

Find the below working application.

 

Create atlas-application.properties file under src/main/resources folder.

 

atlas-application.properties

 

atlas.client.readTimeoutMSecs=30000
atlas.client.connectTimeoutMSecs=30000

 

Define DeleteEntityByUniqueAttribute class.

 

DeleteEntityByUniqueAttribute.java

 

package com.sample.app.entity;

import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.apache.atlas.AtlasClientV2;
import org.apache.atlas.AtlasServiceException;
import org.apache.atlas.model.instance.AtlasEntity;
import org.apache.atlas.model.instance.AtlasEntityHeader;
import org.apache.atlas.model.instance.EntityMutationResponse;
import org.apache.atlas.model.typedef.AtlasEntityDef;
import org.apache.atlas.model.typedef.AtlasStructDef.AtlasAttributeDef;
import org.apache.atlas.model.typedef.AtlasTypesDef;

public class DeleteEntityByUniqueAttribute {

	private static final String TYPE_NAME = "DemoType20";

	private static void createType(AtlasClientV2 atlasClient) throws AtlasServiceException {
		AtlasAttributeDef attributeDef1 = new AtlasAttributeDef();
		attributeDef1.setName("a");
		attributeDef1.setTypeName("int");
		attributeDef1.setCardinality(AtlasAttributeDef.Cardinality.SINGLE);
		attributeDef1.setIsIndexable(true);
		attributeDef1.setIsUnique(false);

		AtlasAttributeDef attributeDef2 = new AtlasAttributeDef();
		attributeDef2.setName("b");
		attributeDef2.setTypeName("int");
		attributeDef2.setCardinality(AtlasAttributeDef.Cardinality.SINGLE);
		attributeDef2.setIsIndexable(true);
		attributeDef2.setIsUnique(false);

		AtlasEntityDef atlasEntityDef = new AtlasEntityDef();
		atlasEntityDef.setName(TYPE_NAME);
		atlasEntityDef.setCreatedBy("Krishna");
		atlasEntityDef.setUpdatedBy("krishna");
		atlasEntityDef.setDescription("Represent a laptoo specification");
		atlasEntityDef.setAttributeDefs(Arrays.asList(attributeDef1, attributeDef2));

		Set<String> superTypes = new HashSet<>();
		superTypes.add("DataSet");
		atlasEntityDef.setSuperTypes(superTypes);

		AtlasTypesDef atlasTypesDef = new AtlasTypesDef();

		atlasTypesDef.getEntityDefs().add(atlasEntityDef);

		atlasClient.createAtlasTypeDefs(atlasTypesDef);
	}

	private static String createEntity(AtlasClientV2 atlasClient) throws AtlasServiceException {

		AtlasEntity atlasEntity = new AtlasEntity();

		atlasEntity.setTypeName(TYPE_NAME);

		atlasEntity.setAttribute("a", 10);
		atlasEntity.setAttribute("b", 20);

		// Following are the mandatory attributes
		atlasEntity.setAttribute("qualifiedName", "DeleteEntityByUniqueAttribute_DEMO1");
		atlasEntity.setAttribute("name", "DeleteEntityByUniqueAttribute_DEMO1");

		AtlasEntity.AtlasEntityWithExtInfo atlasEntityWithExtInfo = new AtlasEntity.AtlasEntityWithExtInfo();
		atlasEntityWithExtInfo.setEntity(atlasEntity);

		EntityMutationResponse entityMutationResponse = atlasClient.createEntity(atlasEntityWithExtInfo);
		return entityMutationResponse.getCreatedEntities().get(0).getGuid();
	}

	public static void main(String[] args) throws AtlasServiceException {

		AtlasClientV2 atlasClient = new AtlasClientV2(new String[] { "http://localhost:21000" },
				new String[] { "admin", "admin" });

		createType(atlasClient);
		createEntity(atlasClient);

		Map<String, String> uniqAttributes = new HashMap<>();
		uniqAttributes.put("qualifiedName", "DeleteEntityByUniqueAttribute_DEMO1");

		System.out.println("Deleting the entity by unique attirbute... ");
		EntityMutationResponse entityMutationResponse = atlasClient.deleteEntityByAttribute(TYPE_NAME, uniqAttributes);
		
		System.out.println("Deleted Item is ");
		List<AtlasEntityHeader> deletedEntities = entityMutationResponse.getDeletedEntities();
		for(AtlasEntityHeader header: deletedEntities) {
			System.out.println(header.getAttributes().get("name"));
		}

	}
}

 

Run above application.

 

Go to Atlas UI and query for the type ‘DemoType20’ and select the check box ‘Show historical entities’ to see soft deleted entities.

 


 

 

Previous                                                    Next                                                    Home

No comments:

Post a Comment