You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@atlas.apache.org by am...@apache.org on 2018/10/31 16:58:55 UTC
atlas git commit: ATLAS-2943: Export options null check added.
Repository: atlas
Updated Branches:
refs/heads/master 3051df254 -> 2f8966286
ATLAS-2943: Export options null check added.
Project: http://git-wip-us.apache.org/repos/asf/atlas/repo
Commit: http://git-wip-us.apache.org/repos/asf/atlas/commit/2f896628
Tree: http://git-wip-us.apache.org/repos/asf/atlas/tree/2f896628
Diff: http://git-wip-us.apache.org/repos/asf/atlas/diff/2f896628
Branch: refs/heads/master
Commit: 2f896628645b584f6e70651facb54ee5dc985fb6
Parents: 3051df2
Author: Ashutosh Mestry <am...@hortonworks.com>
Authored: Tue Oct 30 10:43:21 2018 -0700
Committer: Ashutosh Mestry <am...@hortonworks.com>
Committed: Wed Oct 31 09:46:47 2018 -0700
----------------------------------------------------------------------
.../atlas/model/impexp/AtlasExportRequest.java | 12 +++++++++++
.../atlas/model/impexp/AtlasImportRequest.java | 14 ++++++++++++-
.../atlas/repository/impexp/AuditsWriter.java | 21 +++++---------------
3 files changed, 30 insertions(+), 17 deletions(-)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/atlas/blob/2f896628/intg/src/main/java/org/apache/atlas/model/impexp/AtlasExportRequest.java
----------------------------------------------------------------------
diff --git a/intg/src/main/java/org/apache/atlas/model/impexp/AtlasExportRequest.java b/intg/src/main/java/org/apache/atlas/model/impexp/AtlasExportRequest.java
index 8fb7c68..0e93744 100644
--- a/intg/src/main/java/org/apache/atlas/model/impexp/AtlasExportRequest.java
+++ b/intg/src/main/java/org/apache/atlas/model/impexp/AtlasExportRequest.java
@@ -19,11 +19,13 @@ package org.apache.atlas.model.impexp;
import com.fasterxml.jackson.annotation.JsonAutoDetect;
+import com.fasterxml.jackson.annotation.JsonIgnore;
import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.databind.annotation.JsonSerialize;
import org.apache.atlas.model.instance.AtlasObjectId;
import org.apache.atlas.model.typedef.AtlasBaseTypeDef;
import org.apache.commons.collections.MapUtils;
+import org.apache.commons.lang.StringUtils;
import javax.xml.bind.annotation.XmlAccessType;
import javax.xml.bind.annotation.XmlAccessorType;
@@ -133,6 +135,16 @@ public class AtlasExportRequest implements Serializable {
return Long.parseLong(getOptions().get(AtlasExportRequest.FETCH_TYPE_INCREMENTAL_CHANGE_MARKER).toString());
}
+ @JsonIgnore
+ public boolean isReplicationOptionSet() {
+ return MapUtils.isNotEmpty(options) && options.containsKey(OPTION_KEY_REPLICATED_TO);
+ }
+
+ @JsonIgnore
+ public String getOptionKeyReplicatedTo() {
+ return isReplicationOptionSet() ? (String) options.get(OPTION_KEY_REPLICATED_TO) : StringUtils.EMPTY;
+ }
+
public StringBuilder toString(StringBuilder sb) {
if (sb == null) {
sb = new StringBuilder();
http://git-wip-us.apache.org/repos/asf/atlas/blob/2f896628/intg/src/main/java/org/apache/atlas/model/impexp/AtlasImportRequest.java
----------------------------------------------------------------------
diff --git a/intg/src/main/java/org/apache/atlas/model/impexp/AtlasImportRequest.java b/intg/src/main/java/org/apache/atlas/model/impexp/AtlasImportRequest.java
index 06bc231..0b3ede9 100644
--- a/intg/src/main/java/org/apache/atlas/model/impexp/AtlasImportRequest.java
+++ b/intg/src/main/java/org/apache/atlas/model/impexp/AtlasImportRequest.java
@@ -25,6 +25,8 @@ import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.databind.annotation.JsonSerialize;
import org.apache.atlas.model.typedef.AtlasBaseTypeDef;
+import org.apache.commons.collections.MapUtils;
+import org.apache.commons.lang.StringUtils;
import java.io.Serializable;
import java.util.HashMap;
@@ -102,13 +104,23 @@ public class AtlasImportRequest implements Serializable {
}
private String getOptionForKey(String key) {
- if (this.options == null || !this.options.containsKey(key)) {
+ if (MapUtils.isEmpty(this.options) || !this.options.containsKey(key)) {
return null;
}
return (String) this.options.get(key);
}
+ @JsonIgnore
+ public boolean isReplicationOptionSet() {
+ return MapUtils.isNotEmpty(options) && options.containsKey(OPTION_KEY_REPLICATED_FROM);
+ }
+
+ @JsonIgnore
+ public String getOptionKeyReplicatedFrom() {
+ return isReplicationOptionSet() ? options.get(OPTION_KEY_REPLICATED_FROM) : StringUtils.EMPTY;
+ }
+
@JsonAnySetter
public void setOption(String key, String value) {
if (null == options) {
http://git-wip-us.apache.org/repos/asf/atlas/blob/2f896628/repository/src/main/java/org/apache/atlas/repository/impexp/AuditsWriter.java
----------------------------------------------------------------------
diff --git a/repository/src/main/java/org/apache/atlas/repository/impexp/AuditsWriter.java b/repository/src/main/java/org/apache/atlas/repository/impexp/AuditsWriter.java
index 3612c45..9bf30f1 100644
--- a/repository/src/main/java/org/apache/atlas/repository/impexp/AuditsWriter.java
+++ b/repository/src/main/java/org/apache/atlas/repository/impexp/AuditsWriter.java
@@ -30,6 +30,7 @@ import org.apache.atlas.model.impexp.AtlasImportResult;
import org.apache.atlas.model.impexp.ExportImportAuditEntry;
import org.apache.atlas.repository.Constants;
import org.apache.atlas.type.AtlasType;
+import org.apache.commons.collections.MapUtils;
import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@@ -70,10 +71,6 @@ public class AuditsWriter {
auditForImport.add(userName, result, startTime, endTime, entityCreationOrder);
}
- private boolean isReplicationOptionSet(Map<String, ? extends Object> options, String replicatedKey) {
- return options.containsKey(replicatedKey);
- }
-
private void updateReplicationAttribute(boolean isReplicationSet,
String serverName, String serverFullName,
List<String> exportedGuids,
@@ -87,12 +84,6 @@ public class AuditsWriter {
atlasServerService.updateEntitiesWithServer(server, exportedGuids, attrNameReplicated);
}
- private String getClusterNameFromOptions(Map options, String key) {
- return options.containsKey(key)
- ? (String) options.get(key)
- : StringUtils.EMPTY;
- }
-
private AtlasServer saveServer(String clusterName, String serverFullName,
String entityGuid,
long lastModifiedTimestamp) throws AtlasBaseException {
@@ -138,20 +129,18 @@ public class AuditsWriter {
private class ExportAudits {
private AtlasExportRequest request;
private String targetServerName;
- private String optionKeyReplicatedTo;
private boolean replicationOptionState;
private String targetServerFullName;
public void add(String userName, AtlasExportResult result,
long startTime, long endTime,
List<String> entityGuids) throws AtlasBaseException {
- optionKeyReplicatedTo = AtlasExportRequest.OPTION_KEY_REPLICATED_TO;
request = result.getRequest();
- replicationOptionState = isReplicationOptionSet(request.getOptions(), optionKeyReplicatedTo);
+ replicationOptionState = request.isReplicationOptionSet();
saveCurrentServer();
- targetServerFullName = getClusterNameFromOptions(request.getOptions(), optionKeyReplicatedTo);
+ targetServerFullName = request.getOptionKeyReplicatedTo();
targetServerName = getServerNameFromFullName(targetServerFullName);
auditService.add(userName, getCurrentClusterName(), targetServerName,
ExportImportAuditEntry.OPERATION_EXPORT,
@@ -178,11 +167,11 @@ public class AuditsWriter {
List<String> entityGuids) throws AtlasBaseException {
optionKeyReplicatedFrom = AtlasImportRequest.OPTION_KEY_REPLICATED_FROM;
request = result.getRequest();
- replicationOptionState = isReplicationOptionSet(request.getOptions(), optionKeyReplicatedFrom);
+ replicationOptionState = request.isReplicationOptionSet();
saveCurrentServer();
- sourceServerFullName = getClusterNameFromOptions(request.getOptions(), optionKeyReplicatedFrom);
+ sourceServerFullName = request.getOptionKeyReplicatedFrom();
sourceServerName = getServerNameFromFullName(sourceServerFullName);
auditService.add(userName,
sourceServerName, getCurrentClusterName(),