path: root/openecomp-be/lib/openecomp-sdc-vendor-license-lib/openecomp-sdc-vendor-license-api/src/main/java/org/openecomp/sdc/vendorlicense/dao/types/EntitlementPoolEntity.java

/*-
 * ============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;
  }
}