1 files changed, 222 insertions, 0 deletions
diff --git a/models-base/src/main/java/org/onap/policy/models/base/PfReferenceTimestampKey.java b/models-base/src/main/java/org/onap/policy/models/base/PfReferenceTimestampKey.java
new file mode 100644
index 000000000..5cb639ed5
--- /dev/null
+++ b/models-base/src/main/java/org/onap/policy/models/base/PfReferenceTimestampKey.java
@@ -0,0 +1,222 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * Copyright (C) 2021 Nordix Foundation.
+ * ================================================================================
+ * 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.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.models.base;
+
+import java.time.Instant;
+import javax.persistence.Column;
+import javax.persistence.Embeddable;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.NonNull;
+import lombok.ToString;
+import org.onap.policy.common.parameters.annotations.NotNull;
+import org.onap.policy.common.utils.validation.Assertions;
+
+/**
+ * This class is an extension of PfReferenceKey. It has similar behaviour as of PfReferenceKey with an
+ * additional option to have timestamp as a parameter.
+ *
+ */
+@Embeddable
+@Data
+@EqualsAndHashCode
+@ToString(callSuper = true)
+public class PfReferenceTimestampKey extends PfReferenceKey {
+    private static final String TIMESTAMP_TOKEN = "timeStamp";
+    private static final Instant DEFAULT_TIMESTAMP = Instant.EPOCH;
+
+    @Column(name = TIMESTAMP_TOKEN)
+    @NotNull
+    private Instant timeStamp;
+
+
+    /**
+     * The default constructor creates a null reference timestamp key.
+     */
+    public PfReferenceTimestampKey() {
+        super();
+        this.timeStamp = DEFAULT_TIMESTAMP;
+    }
+
+    /**
+     * The Copy Constructor creates a key by copying another key.
+     *
+     * @param referenceKey
+     *        the reference key to copy from
+     */
+    public PfReferenceTimestampKey(final PfReferenceTimestampKey referenceKey) {
+        super(referenceKey);
+        this.timeStamp = referenceKey.getTimeStamp();
+    }
+
+    /**
+     * Constructor to create a null reference key for the specified parent concept key.
+     *
+     * @param pfConceptKey
+     *        the parent concept key of this reference key
+     */
+    public PfReferenceTimestampKey(final PfConceptKey pfConceptKey) {
+        super(pfConceptKey);
+        this.timeStamp = DEFAULT_TIMESTAMP;
+    }
+
+    /**
+     * Constructor to create a reference timestamp key for the given parent concept key with the given local name.
+     *
+     * @param pfConceptKey
+     *        the parent concept key of this reference key
+     * @param localName
+     *        the local name of this reference key
+     * @param timeStamp
+     *        the timestamp for this reference key
+     */
+    public PfReferenceTimestampKey(final PfConceptKey pfConceptKey, final String localName, final Instant timeStamp) {
+        super(pfConceptKey, localName);
+        this.timeStamp = timeStamp;
+    }
+
+    /**
+     * Constructor to create a reference timestamp key for the given parent reference key with the given local name.
+     *
+     * @param parentReferenceKey
+     *        the parent reference key of this reference key
+     * @param localName
+     *        the local name of this reference key
+     * @param timeStamp
+     *        the timestamp for this reference key
+     */
+    public PfReferenceTimestampKey(final PfReferenceKey parentReferenceKey, final String localName,
+                                   final Instant timeStamp) {
+        super(parentReferenceKey, localName);
+        this.timeStamp = timeStamp;
+    }
+
+    /**
+     * Constructor to create a reference timestamp key for the given parent reference key (specified by the parent
+     * reference key's concept key and local name) with the given local name.
+     *
+     * @param pfConceptKey
+     *        the concept key of the parent reference key of this reference key
+     * @param parentLocalName
+     *        the local name of the parent reference key of this reference key
+     * @param localName
+     *        the local name of this reference key
+     * @param timeStamp
+     *        the timestamp for this reference key
+     */
+    public PfReferenceTimestampKey(final PfConceptKey pfConceptKey, final String parentLocalName,
+                                   final String localName, final Instant timeStamp) {
+        super(pfConceptKey, parentLocalName, localName);
+        this.timeStamp = timeStamp;
+    }
+
+    /**
+     * Constructor to create a reference timestamp key for the given parent concept key (specified by the
+     * parent concept key's name and version) with the given local name.
+     *
+     * @param parentKeyName
+     *        the name of the parent concept key of this reference key
+     * @param parentKeyVersion
+     *        the version of the parent concept key of this reference key
+     * @param localName
+     *        the local name of this reference key
+     * @param timeStamp
+     *        the timestamp for this reference key
+     */
+    public PfReferenceTimestampKey(final String parentKeyName, final String parentKeyVersion, final String localName,
+                                   final Instant timeStamp) {
+        super(parentKeyName, parentKeyVersion, NULL_KEY_NAME, localName);
+        this.timeStamp = timeStamp;
+    }
+
+    /**
+     * Constructor to create a reference timestamp key for the given parent key (specified by the parent key's name,
+     * version and local name) with the given local name.
+     *
+     * @param parentKeyName
+     *        the parent key name of this reference key
+     * @param parentKeyVersion
+     *        the parent key version of this reference key
+     * @param parentLocalName
+     *        the parent local name of this reference key
+     * @param localName
+     *        the local name of this reference key
+     * @param timeStamp
+     *        the timestamp for this reference key
+     */
+    public PfReferenceTimestampKey(final String parentKeyName, final String parentKeyVersion,
+                                   final String parentLocalName, final String localName, final Instant timeStamp) {
+        super(parentKeyName, parentKeyVersion, parentLocalName, localName);
+        this.timeStamp = timeStamp;
+    }
+
+
+    /**
+     * Constructor to create a key using the key and version from the specified key ID.
+     *
+     * @param id the key ID in a format that respects the KEY_ID_REGEXP
+     */
+    public PfReferenceTimestampKey(final String id) {
+        super(id.substring(0, id.lastIndexOf(':')));
+        this.timeStamp = Instant.ofEpochSecond(Long.parseLong(id.substring(id.lastIndexOf(':') + 1)));
+    }
+
+
+    /**
+     * Get a null reference timestamp key.
+     *
+     * @return a null reference key
+     */
+    public static PfReferenceTimestampKey getNullKey() {
+        return new PfReferenceTimestampKey(PfKey.NULL_KEY_NAME, PfKey.NULL_KEY_VERSION, PfKey.NULL_KEY_NAME,
+            PfKey.NULL_KEY_NAME, DEFAULT_TIMESTAMP);
+    }
+
+
+    @Override
+    public PfReferenceTimestampKey getKey() {
+        return this;
+    }
+
+    @Override
+    public String getId() {
+        return super.getId() + ':' + getTimeStamp().getEpochSecond();
+    }
+
+    @Override
+    public boolean isNullKey() {
+        return super.isNullKey() && getTimeStamp().getEpochSecond() == 0;
+    }
+
+    @Override
+    public boolean isNewerThan(@NonNull PfKey otherKey) {
+        Assertions.instanceOf(otherKey, PfReferenceTimestampKey.class);
+        final PfReferenceTimestampKey otherConceptKey = (PfReferenceTimestampKey) otherKey;
+
+        if (this.equals(otherConceptKey)) {
+            return false;
+        }
+        if (!getTimeStamp().equals(otherConceptKey.timeStamp)) {
+            return timeStamp.isAfter(otherConceptKey.timeStamp);
+        }
+        return super.isNewerThan(otherKey);
+    }
+}