1 files changed, 370 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-sdc-vendor-license-lib/openecomp-sdc-vendor-license-api/src/main/java/org/openecomp/sdc/vendorlicense/dao/types/EntitlementPoolEntity.java b/openecomp-be/lib/openecomp-sdc-vendor-license-lib/openecomp-sdc-vendor-license-api/src/main/java/org/openecomp/sdc/vendorlicense/dao/types/EntitlementPoolEntity.java
new file mode 100644
index 0000000000..8964458181
--- /dev/null
+++ b/openecomp-be/lib/openecomp-sdc-vendor-license-lib/openecomp-sdc-vendor-license-api/src/main/java/org/openecomp/sdc/vendorlicense/dao/types/EntitlementPoolEntity.java
@@ -0,0 +1,370 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * SDC
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.openecomp.sdc.vendorlicense.dao.types;
+
+import com.datastax.driver.mapping.annotations.ClusteringColumn;
+import com.datastax.driver.mapping.annotations.Column;
+import com.datastax.driver.mapping.annotations.Enumerated;
+import com.datastax.driver.mapping.annotations.Frozen;
+import com.datastax.driver.mapping.annotations.PartitionKey;
+import com.datastax.driver.mapping.annotations.Table;
+import org.openecomp.sdc.vendorlicense.dao.types.xml.AggregationFunctionForXml;
+import org.openecomp.sdc.vendorlicense.dao.types.xml.EntitlementMetricForXml;
+import org.openecomp.sdc.vendorlicense.dao.types.xml.EntitlementTimeForXml;
+import org.openecomp.sdc.vendorlicense.dao.types.xml.ThresholdForXml;
+import org.openecomp.sdc.versioning.dao.types.Version;
+import org.openecomp.sdc.versioning.dao.types.VersionableEntity;
+
+import java.util.HashSet;
+import java.util.Objects;
+import java.util.Set;
+
+@Table(keyspace = "dox", name = "entitlement_pool")
+public class EntitlementPoolEntity implements VersionableEntity {
+  private static final String ENTITY_TYPE = "Entitlement Pool";
+
+  @PartitionKey
+  @Column(name = "vlm_id")
+  private String vendorLicenseModelId;
+
+  @PartitionKey(value = 1)
+  @Frozen
+  private Version version;
+
+  @ClusteringColumn
+  @Column(name = "ep_id")
+  private String id;
+  private String name;
+  private String description;
+
+
+  @Column(name = "threshold")
+  private int thresholdValue;
+
+  @Column(name = "threshold_unit")
+  @Enumerated
+  private ThresholdUnit thresholdUnit;
+
+  @Column(name = "entitlement_metric")
+  @Frozen
+  private ChoiceOrOther<EntitlementMetric> entitlementMetric;
+  private String increments;
+
+  @Column(name = "aggregation_func")
+  @Frozen
+  private ChoiceOrOther<AggregationFunction> aggregationFunction;
+
+  @Column(name = "operational_scope")
+  @Frozen
+  private MultiChoiceOrOther<OperationalScope> operationalScope;
+
+  @Frozen
+  private ChoiceOrOther<EntitlementTime> time;
+
+  @Column(name = "manufacturer_ref_num")
+  private String manufacturerReferenceNumber;
+
+  @Column(name = "ref_fg_ids")
+  private Set<String> referencingFeatureGroups = new HashSet<>();
+
+  @Column(name = "version_uuid")
+  private String versionUuId;
+
+  public EntitlementPoolEntity() {
+  }
+
+  /**
+   * Instantiates a new Entitlement pool entity.
+   *
+   * @param vlmId   the vlm id
+   * @param version the version
+   * @param id      the id
+   */
+  public EntitlementPoolEntity(String vlmId, Version version, String id) {
+    this.vendorLicenseModelId = vlmId;
+    this.version = version;
+    this.id = id;
+  }
+
+  @Override
+  public String getEntityType() {
+    return ENTITY_TYPE;
+  }
+
+  @Override
+  public String getFirstClassCitizenId() {
+    return getVendorLicenseModelId();
+  }
+
+  public String getVendorLicenseModelId() {
+    return vendorLicenseModelId;
+  }
+
+  public void setVendorLicenseModelId(String vendorLicenseModelId) {
+    this.vendorLicenseModelId = vendorLicenseModelId;
+  }
+
+  @Override
+  public String getId() {
+    return id;
+  }
+
+  @Override
+  public void setId(String id) {
+    this.id = id;
+  }
+
+  @Override
+  public Version getVersion() {
+    return version;
+  }
+
+  @Override
+  public void setVersion(Version version) {
+    this.version = version;
+  }
+
+  public Set<String> getReferencingFeatureGroups() {
+    return referencingFeatureGroups;
+  }
+
+  public void setReferencingFeatureGroups(Set<String> referencingFeatureGroups) {
+    this.referencingFeatureGroups = referencingFeatureGroups;
+  }
+
+  public String getName() {
+    return name;
+  }
+
+  public void setName(String name) {
+    this.name = name;
+  }
+
+  public String getDescription() {
+    return description;
+  }
+
+  public void setDescription(String description) {
+    this.description = description;
+  }
+
+  public int getThresholdValue() {
+    return thresholdValue;
+  }
+
+  public void setThresholdValue(int thresholdValue) {
+    this.thresholdValue = thresholdValue;
+  }
+
+  public ThresholdUnit getThresholdUnit() {
+    return thresholdUnit;
+  }
+
+  public void setThresholdUnit(ThresholdUnit thresholdUnits) {
+    this.thresholdUnit = thresholdUnits;
+  }
+
+  public ChoiceOrOther<EntitlementMetric> getEntitlementMetric() {
+    return entitlementMetric;
+  }
+
+  public void setEntitlementMetric(ChoiceOrOther<EntitlementMetric> entitlementMetric) {
+    entitlementMetric.resolveEnum(EntitlementMetric.class);
+    this.entitlementMetric = entitlementMetric;
+  }
+
+  public String getIncrements() {
+    return increments;
+  }
+
+  public void setIncrements(String increments) {
+    this.increments = increments;
+  }
+
+  public ChoiceOrOther<AggregationFunction> getAggregationFunction() {
+    return aggregationFunction;
+  }
+
+  public void setAggregationFunction(ChoiceOrOther<AggregationFunction> aggregationFunction) {
+    aggregationFunction.resolveEnum(AggregationFunction.class);
+    this.aggregationFunction = aggregationFunction;
+  }
+
+  public MultiChoiceOrOther<OperationalScope> getOperationalScope() {
+    return operationalScope;
+  }
+
+  public void setOperationalScope(MultiChoiceOrOther<OperationalScope> operationalScope) {
+    operationalScope.resolveEnum(OperationalScope.class);
+    this.operationalScope = operationalScope;
+  }
+
+  public ChoiceOrOther<EntitlementTime> getTime() {
+    return time;
+  }
+
+  public void setTime(ChoiceOrOther<EntitlementTime> time) {
+    time.resolveEnum(EntitlementTime.class);
+    this.time = time;
+  }
+
+  public String getManufacturerReferenceNumber() {
+    return manufacturerReferenceNumber;
+  }
+
+  public void setManufacturerReferenceNumber(String manufacturerReferenceNumber) {
+    this.manufacturerReferenceNumber = manufacturerReferenceNumber;
+  }
+
+
+  /**
+   * Gets threshold for artifact.
+   *
+   * @return the threshold for artifact
+   */
+  //util methods for XML
+  public ThresholdForXml getThresholdForArtifact() {
+    ThresholdForXml threshold = new ThresholdForXml();
+    threshold.setUnit(getThresholdUnit() == null ? null : getThresholdUnit().name());
+    threshold.setValue(getThresholdValue());
+    return threshold;
+  }
+
+  /**
+   * Gets entitlement metric for artifact.
+   *
+   * @return the entitlement metric for artifact
+   */
+  public EntitlementMetricForXml getEntitlementMetricForArtifact() {
+    EntitlementMetricForXml metric = new EntitlementMetricForXml();
+    if (entitlementMetric != null) {
+      metric.setValue(entitlementMetric.getResult());
+    } else {
+      metric.setValue(null);
+    }
+    return metric;
+  }
+
+  /**
+   * Gets time for artifact.
+   *
+   * @return the time for artifact
+   */
+  public EntitlementTimeForXml getTimeForArtifact() {
+    EntitlementTimeForXml timeForXml = new EntitlementTimeForXml();
+    if (time != null) {
+      timeForXml.setValue(time.getResult());
+    }
+
+    return timeForXml;
+  }
+
+
+  @Override
+  public String toString() {
+    return "EntitlementPoolEntity{"
+        + "vendorLicenseModelId='" + vendorLicenseModelId + '\''
+        + ", version=" + version
+        + ", id='" + id + '\''
+        + ", name='" + name + '\''
+        + ", description='" + description + '\''
+        + ", thresholdValue=" + thresholdValue
+        + ", thresholdUnit='" + thresholdUnit + '\''
+        + ", entitlementMetric=" + entitlementMetric
+        + ", increments='" + increments + '\''
+        + ", aggregationFunction=" + aggregationFunction
+        + ", operationalScope=" + operationalScope
+        + ", time=" + time
+        + ", manufacturerReferenceNumber='" + manufacturerReferenceNumber + '\''
+        + ", referencingFeatureGroups=" + referencingFeatureGroups
+        + ", version_uuid=" + versionUuId
+        + '}';
+  }
+
+  @Override
+  public boolean equals(Object obj) {
+    if (this == obj) {
+      return true;
+    }
+    if (obj == null || getClass() != obj.getClass()) {
+      return false;
+    }
+    EntitlementPoolEntity that = (EntitlementPoolEntity) obj;
+    return Float.compare(that.thresholdValue, thresholdValue) == 0
+        && Objects.equals(vendorLicenseModelId, that.vendorLicenseModelId)
+        && Objects.equals(version, that.version)
+        && Objects.equals(id, that.id)
+        && Objects.equals(name, that.name)
+        && Objects.equals(description, that.description)
+        && Objects.equals(thresholdUnit, that.thresholdUnit)
+        && Objects.equals(entitlementMetric, that.entitlementMetric)
+        && Objects.equals(increments, that.increments)
+        && Objects.equals(aggregationFunction, that.aggregationFunction)
+        && Objects.equals(operationalScope, that.operationalScope)
+        && Objects.equals(time, that.time)
+        && Objects.equals(manufacturerReferenceNumber, that.manufacturerReferenceNumber)
+        && Objects.equals(referencingFeatureGroups, that.referencingFeatureGroups);
+  }
+
+  @Override
+  public int hashCode() {
+    return Objects
+        .hash(vendorLicenseModelId, version, id, name, description, thresholdValue, thresholdUnit,
+            entitlementMetric, increments, aggregationFunction, operationalScope, time,
+            manufacturerReferenceNumber, referencingFeatureGroups);
+  }
+
+  /**
+   * Gets aggregation function for artifact.
+   *
+   * @return the aggregation function for artifact
+   */
+  public AggregationFunctionForXml getAggregationFunctionForArtifact() {
+    AggregationFunctionForXml aggregationFunctionForXml = new AggregationFunctionForXml();
+    if (entitlementMetric != null) {
+      aggregationFunctionForXml.setValue(aggregationFunction.getResult());
+    } else {
+      aggregationFunctionForXml.setValue(null);
+    }
+    return aggregationFunctionForXml;
+  }
+
+  /**
+   * Gets operational scope for artifact.
+   *
+   * @return the operational scope for artifact
+   */
+  public Set<String> getOperationalScopeForArtifact() {
+    if (operationalScope != null) {
+      return operationalScope.getResults();
+    } else {
+      return null;
+    }
+  }
+
+  public String getVersionUuId() {
+    return versionUuId;
+  }
+
+  public void setVersionUuId(String uuId) {
+    versionUuId = uuId;
+  }
+}