Fixing setter for recovery codes, adding creation and update time set…

…ting to constructors
indigo-iam · Mar 4, 2022 · e7ce000 · e7ce000
1 parent f648ec4
commit e7ce000
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/iam-persistence/src/main/java/it/infn/mw/iam/persistence/model/IamTotpMfa.java b/iam-persistence/src/main/java/it/infn/mw/iam/persistence/model/IamTotpMfa.java
@@ -64,10 +64,17 @@ public class IamTotpMfa implements Serializable {
       orphanRemoval = true)
   private Set<IamTotpRecoveryCode> recoveryCodes = new HashSet<>();
 
-  public IamTotpMfa() {}
+  public IamTotpMfa() {
+    Date now = new Date();
+    setCreationTime(now);
+    setLastUpdateTime(now);
+  }
 
   public IamTotpMfa(IamAccount account) {
     this.account = account;
+    Date now = new Date();
+    setCreationTime(now);
+    setLastUpdateTime(now);
   }
 
   public Long getId() {
@@ -134,12 +141,8 @@ public Set<IamTotpRecoveryCode> getRecoveryCodes() {
   }
 
   public void setRecoveryCodes(final Set<IamTotpRecoveryCode> recoveryCodes) {
-    if (this.recoveryCodes.isEmpty()) {
-      this.recoveryCodes = recoveryCodes;
-    } else {
-      this.recoveryCodes.clear();
-      this.recoveryCodes.addAll(recoveryCodes);
-    }
+    this.recoveryCodes.clear();
+    this.recoveryCodes.addAll(recoveryCodes);
   }
 
   @Override