1 files changed, 39 insertions, 31 deletions

diff --git a/sdc-distribution-client/src/main/java/org/openecomp/sdc/impl/DistributionStatusMessageImpl.java b/sdc-distribution-client/src/main/java/org/openecomp/sdc/impl/DistributionStatusMessageImpl.java
index d570142..d9ed654 100644
--- a/sdc-distribution-client/src/main/java/org/openecomp/sdc/impl/DistributionStatusMessageImpl.java
+++ b/sdc-distribution-client/src/main/java/org/openecomp/sdc/impl/DistributionStatusMessageImpl.java
@@ -20,78 +20,80 @@
 
 package org.openecomp.sdc.impl;
 
+import org.openecomp.sdc.api.consumer.IComponentDoneStatusMessage;
 import org.openecomp.sdc.api.consumer.IDistributionStatusMessage;
+import org.openecomp.sdc.api.consumer.IFinalDistrStatusMessage;
 import org.openecomp.sdc.utils.DistributionStatusEnum;
 
 class DistributionStatusMessageImpl implements IDistributionStatusMessage {
-	
-	
+
 	String distributionID;
 	String consumerID;
 	long timestamp;
 	String artifactURL;
 	DistributionStatusEnum status;
 	String errorReason;
-	
-	
-	public DistributionStatusMessageImpl(IDistributionStatusMessage message){
+	private String componentName;
+
+	public DistributionStatusMessageImpl(IDistributionStatusMessage message) {
 		super();
 		distributionID = message.getDistributionID();
 		consumerID = message.getConsumerID();
 		timestamp = message.getTimestamp();
 		artifactURL = message.getArtifactURL();
 		status = message.getStatus();
-		
+
+	}
+
+	public DistributionStatusMessageImpl(IComponentDoneStatusMessage message) {
+		super();
+		distributionID = message.getDistributionID();
+		consumerID = message.getConsumerID();
+		timestamp = message.getTimestamp();
+		artifactURL = message.getArtifactURL();
+		status = message.getStatus();
+		componentName = message.getComponentName();
+	}
+
+	public DistributionStatusMessageImpl(IFinalDistrStatusMessage message) {
+		super();
+		distributionID = message.getDistributionID();
+		consumerID = message.getConsumerID();
+		timestamp = message.getTimestamp();
+		 
+		artifactURL = "";
+		status = message.getStatus();
+		componentName = message.getComponentName();
 	}
 
 	@Override
 	public String getDistributionID() {
-		
+
 		return distributionID;
 	}
 
 	@Override
 	public String getConsumerID() {
-		
+
 		return consumerID;
 	}
 
 	@Override
 	public long getTimestamp() {
-		
+
 		return timestamp;
 	}
 
 	@Override
 	public String getArtifactURL() {
-		
+
 		return artifactURL;
 	}
 
 	@Override
 	public DistributionStatusEnum getStatus() {
-		
-		return status;
-	}
-
-	public void setDistributionID(String distributionID) {
-		this.distributionID = distributionID;
-	}
-
-	public void setConsumerID(String consumerID) {
-		this.consumerID = consumerID;
-	}
-
-	public void setTimestamp(long timestamp) {
-		this.timestamp = timestamp;
-	}
-
-	public void setArtifactURL(String artifactURL) {
-		this.artifactURL = artifactURL;
-	}
 
-	public void setStatus(DistributionStatusEnum status) {
-		this.status = status;
+		return status;
 	}
 
 	public String getErrorReason() {
@@ -102,4 +104,10 @@ class DistributionStatusMessageImpl implements IDistributionStatusMessage {
 		this.errorReason = errorReason;
 	}
 
+	public String getComponentName() {
+		return componentName;
+	}
+
+	
+
 }