You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@atlas.apache.org by sa...@apache.org on 2018/07/10 22:37:54 UTC
atlas git commit: ATLAS-1950: Import API: Improvement: Specify
Supertypes in Import Transforms
Repository: atlas
Updated Branches:
refs/heads/branch-0.8 b6d0e4013 -> d16324cb8
ATLAS-1950: Import API: Improvement: Specify Supertypes in Import Transforms
Signed-off-by: Sarath Subramanian <ss...@hortonworks.com>
Project: http://git-wip-us.apache.org/repos/asf/atlas/repo
Commit: http://git-wip-us.apache.org/repos/asf/atlas/commit/d16324cb
Tree: http://git-wip-us.apache.org/repos/asf/atlas/tree/d16324cb
Diff: http://git-wip-us.apache.org/repos/asf/atlas/diff/d16324cb
Branch: refs/heads/branch-0.8
Commit: d16324cb8c9cf0170f14b2187929ed0edd311bf0
Parents: b6d0e40
Author: Ashutosh Mestry <am...@hortonworks.com>
Authored: Tue Jul 10 15:37:16 2018 -0700
Committer: Sarath Subramanian <ss...@hortonworks.com>
Committed: Tue Jul 10 15:37:16 2018 -0700
----------------------------------------------------------------------
.../atlas/repository/impexp/ImportService.java | 28 +++++++++++++++++-
.../repository/impexp/ImportTransforms.java | 18 ++++++++++++
.../repository/impexp/ImportServiceTest.java | 30 ++++++++++++++++++++
.../repository/impexp/ImportTransformsTest.java | 12 ++++++++
4 files changed, 87 insertions(+), 1 deletion(-)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/atlas/blob/d16324cb/repository/src/main/java/org/apache/atlas/repository/impexp/ImportService.java
----------------------------------------------------------------------
diff --git a/repository/src/main/java/org/apache/atlas/repository/impexp/ImportService.java b/repository/src/main/java/org/apache/atlas/repository/impexp/ImportService.java
index c976c59..972e1e3 100644
--- a/repository/src/main/java/org/apache/atlas/repository/impexp/ImportService.java
+++ b/repository/src/main/java/org/apache/atlas/repository/impexp/ImportService.java
@@ -17,6 +17,7 @@
*/
package org.apache.atlas.repository.impexp;
+import com.google.common.annotations.VisibleForTesting;
import org.apache.atlas.AtlasErrorCode;
import org.apache.atlas.exception.AtlasBaseException;
import org.apache.atlas.model.impexp.AtlasImportRequest;
@@ -24,6 +25,7 @@ import org.apache.atlas.model.impexp.AtlasImportResult;
import org.apache.atlas.model.typedef.AtlasTypesDef;
import org.apache.atlas.repository.store.graph.BulkImporter;
import org.apache.atlas.store.AtlasTypeDefStore;
+import org.apache.atlas.type.AtlasEntityType;
import org.apache.atlas.type.AtlasTypeRegistry;
import org.apache.commons.collections.MapUtils;
import org.apache.commons.io.FileUtils;
@@ -75,7 +77,7 @@ public class ImportService {
String transforms = MapUtils.isNotEmpty(request.getOptions()) ? request.getOptions().get(AtlasImportRequest.TRANSFORMS_KEY) : null;
- source.setImportTransform(ImportTransforms.fromJson(transforms));
+ setImportTransform(source, transforms);
startTimestamp = System.currentTimeMillis();
processTypes(source.getTypesDef(), result);
setStartPosition(request, source);
@@ -99,6 +101,30 @@ public class ImportService {
return result;
}
+ @VisibleForTesting
+ void setImportTransform(ZipSource source, String transforms) throws AtlasBaseException {
+ ImportTransforms importTransform = ImportTransforms.fromJson(transforms);
+ if (importTransform == null) {
+ return;
+ }
+
+ updateTransformsWithSubTypes(importTransform);
+ source.setImportTransform(importTransform);
+ }
+
+ private void updateTransformsWithSubTypes(ImportTransforms importTransforms) throws AtlasBaseException {
+ String[] transformTypes = importTransforms.getTypes().toArray(new String[importTransforms.getTypes().size()]);
+ for (int i = 0; i < transformTypes.length; i++) {
+ String typeName = transformTypes[i];
+ AtlasEntityType entityType = typeRegistry.getEntityTypeByName(typeName);
+ if (entityType == null) {
+ continue;
+ }
+
+ importTransforms.addParentTransformsToSubTypes(typeName, entityType.getAllSubTypes());
+ }
+ }
+
private void setStartPosition(AtlasImportRequest request, ZipSource source) throws AtlasBaseException {
if (request.getStartGuid() != null) {
source.setPositionUsingEntityGuid(request.getStartGuid());
http://git-wip-us.apache.org/repos/asf/atlas/blob/d16324cb/repository/src/main/java/org/apache/atlas/repository/impexp/ImportTransforms.java
----------------------------------------------------------------------
diff --git a/repository/src/main/java/org/apache/atlas/repository/impexp/ImportTransforms.java b/repository/src/main/java/org/apache/atlas/repository/impexp/ImportTransforms.java
index 63e53c5..2f27448 100644
--- a/repository/src/main/java/org/apache/atlas/repository/impexp/ImportTransforms.java
+++ b/repository/src/main/java/org/apache/atlas/repository/impexp/ImportTransforms.java
@@ -30,6 +30,7 @@ import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
+import java.util.Set;
public class ImportTransforms {
private static final Logger LOG = LoggerFactory.getLogger(ImportTransforms.class);
@@ -53,6 +54,23 @@ public class ImportTransforms {
public Map<String, List<ImportTransformer>> getTransforms(String typeName) { return transforms.get(typeName); }
+ public Set<String> getTypes() {
+ return getTransforms().keySet();
+ }
+
+ public void addParentTransformsToSubTypes(String parentType, Set<String> subTypes) {
+ Map<String, List<ImportTransformer>> attribtueTransformMap = getTransforms().get(parentType);
+ for (String subType : subTypes) {
+ if(!getTransforms().containsKey(subType)) {
+ getTransforms().put(subType, attribtueTransformMap);
+ } else {
+ for (Map.Entry<String, List<ImportTransformer>> entry : attribtueTransformMap.entrySet()) {
+ getTransforms().get(subType).get(entry.getKey()).addAll(entry.getValue());
+ }
+ }
+ }
+ }
+
public AtlasEntity.AtlasEntityWithExtInfo apply(AtlasEntity.AtlasEntityWithExtInfo entityWithExtInfo) throws AtlasBaseException {
if (entityWithExtInfo != null) {
apply(entityWithExtInfo.getEntity());
http://git-wip-us.apache.org/repos/asf/atlas/blob/d16324cb/repository/src/test/java/org/apache/atlas/repository/impexp/ImportServiceTest.java
----------------------------------------------------------------------
diff --git a/repository/src/test/java/org/apache/atlas/repository/impexp/ImportServiceTest.java b/repository/src/test/java/org/apache/atlas/repository/impexp/ImportServiceTest.java
index 035ddac..c03fbc6 100644
--- a/repository/src/test/java/org/apache/atlas/repository/impexp/ImportServiceTest.java
+++ b/repository/src/test/java/org/apache/atlas/repository/impexp/ImportServiceTest.java
@@ -46,6 +46,7 @@ import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertNotNull;
+import static org.testng.Assert.assertTrue;
@Guice(modules = TestModules.TestOnlyModule.class)
public class ImportServiceTest {
@@ -221,4 +222,33 @@ public class ImportServiceTest {
private void loadHiveModel() throws IOException, AtlasBaseException {
loadModelFromJson("0030-hive_model.json", typeDefStore, typeRegistry);
}
+
+ @Test(dataProvider = "salesNewTypeAttrs-next")
+ public void transformUpdatesForSubTypes(ZipSource zipSource) throws IOException, AtlasBaseException {
+ loadModelFromJson("0010-base_model.json", typeDefStore, typeRegistry);
+ loadModelFromJson("0030-hive_model.json", typeDefStore, typeRegistry);
+
+ String transformJSON = "{ \"Asset\": { \"qualifiedName\":[ \"lowercase\", \"replace:@cl1:@cl2\" ] } }";
+ importService.setImportTransform(zipSource, transformJSON);
+ ImportTransforms importTransforms = zipSource.getImportTransform();
+
+ assertTrue(importTransforms.getTransforms().containsKey("Asset"));
+ assertTrue(importTransforms.getTransforms().containsKey("hive_table"));
+ assertTrue(importTransforms.getTransforms().containsKey("hive_column"));
+ }
+
+ @Test(dataProvider = "salesNewTypeAttrs-next")
+ public void transformUpdatesForSubTypesAddsToExistingTransforms(ZipSource zipSource) throws IOException, AtlasBaseException {
+ loadModelFromJson("0010-base_model.json", typeDefStore, typeRegistry);
+ loadModelFromJson("0030-hive_model.json", typeDefStore, typeRegistry);
+
+ String transformJSON = "{ \"Asset\": { \"qualifiedName\":[ \"replace:@cl1:@cl2\" ] }, \"hive_table\": { \"qualifiedName\":[ \"lowercase\" ] } }";
+ importService.setImportTransform(zipSource, transformJSON);
+ ImportTransforms importTransforms = zipSource.getImportTransform();
+
+ assertTrue(importTransforms.getTransforms().containsKey("Asset"));
+ assertTrue(importTransforms.getTransforms().containsKey("hive_table"));
+ assertTrue(importTransforms.getTransforms().containsKey("hive_column"));
+ assertEquals(importTransforms.getTransforms().get("hive_table").get("qualifiedName").size(), 2);
+ }
}
http://git-wip-us.apache.org/repos/asf/atlas/blob/d16324cb/repository/src/test/java/org/apache/atlas/repository/impexp/ImportTransformsTest.java
----------------------------------------------------------------------
diff --git a/repository/src/test/java/org/apache/atlas/repository/impexp/ImportTransformsTest.java b/repository/src/test/java/org/apache/atlas/repository/impexp/ImportTransformsTest.java
index ccedeb1..a73abcd 100644
--- a/repository/src/test/java/org/apache/atlas/repository/impexp/ImportTransformsTest.java
+++ b/repository/src/test/java/org/apache/atlas/repository/impexp/ImportTransformsTest.java
@@ -23,19 +23,23 @@ import org.apache.atlas.model.instance.AtlasEntity.AtlasEntityWithExtInfo;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.Test;
+import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
+import static org.apache.atlas.repository.impexp.ZipFileResourceTestUtils.loadModelFromJson;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertNotNull;
+import static org.testng.Assert.assertTrue;
public class ImportTransformsTest {
private final String qualifiedName = "qualifiedName";
private final String lowerCaseCL1 = "@cl1";
private final String lowerCaseCL2 = "@cl2";
private final String jsonTransforms = "{ \"hive_table\": { \"qualifiedName\":[ \"lowercase\", \"replace:@cl1:@cl2\" ] } }";
+ private final String jsonTransforms2 = "{ \"Asset\": { \"qualifiedName\":[ \"replace:@cl1:@cl2\" ] }, \"hive_table\": { \"qualifiedName\":[ \"lowercase\", \"replace:@cl1:@cl2\" ] } }";
private ImportTransforms transform;
@@ -86,6 +90,14 @@ public class ImportTransformsTest {
assertEquals(entityWithExtInfo.getEntity().getGuid(), transformedEntityWithExtInfo.getEntity().getGuid());
}
+ @Test
+ public void transformFromJsonWithMultipleEntries() {
+ ImportTransforms t = ImportTransforms.fromJson(jsonTransforms2);
+
+ assertNotNull(t);
+ assertEquals(t.getTransforms().size(), 2);
+ }
+
private String[] getExtEntityExpectedValues(AtlasEntityWithExtInfo entityWithExtInfo) {
String[] ret = new String[entityWithExtInfo.getReferredEntities().size()];