Merge pull request #93 from gfa-cc-after/SCRUM-94

Scrum 94
gfa-cc-after · Aug 31, 2024 · b81d0fb · b81d0fb
2 parents 5703a56 + 4a39ae7
commit b81d0fb
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/backend/src/main/java/com/greenfoxacademy/backend/models/Pet.java b/backend/src/main/java/com/greenfoxacademy/backend/models/Pet.java
@@ -1,9 +1,12 @@
 package com.greenfoxacademy.backend.models;
 
+import jakarta.persistence.CascadeType;
 import jakarta.persistence.Column;
 import jakarta.persistence.Entity;
 import jakarta.persistence.GeneratedValue;
 import jakarta.persistence.Id;
+import jakarta.persistence.JoinColumn;
+import jakarta.persistence.ManyToOne;
 import jakarta.persistence.Table;
 import java.util.Date;
 import lombok.AllArgsConstructor;
@@ -34,4 +37,8 @@ public class Pet {
   private Date petBirthDate;
   private Date lastCheckUp;
   private Date nextCheckUp;
-}
+
+  @ManyToOne(cascade = CascadeType.ALL)
+  @JoinColumn(name = "petOwner_Id")
+  private User petOwner;
+}
diff --git a/backend/src/main/java/com/greenfoxacademy/backend/models/User.java b/backend/src/main/java/com/greenfoxacademy/backend/models/User.java
@@ -1,10 +1,14 @@
 package com.greenfoxacademy.backend.models;
 
+import jakarta.persistence.CascadeType;
 import jakarta.persistence.Column;
 import jakarta.persistence.Entity;
+import jakarta.persistence.FetchType;
 import jakarta.persistence.GeneratedValue;
 import jakarta.persistence.Id;
+import jakarta.persistence.OneToMany;
 import jakarta.persistence.Table;
+import java.util.ArrayList;
 import java.util.Collection;
 import java.util.List;
 import java.util.UUID;
@@ -38,6 +42,9 @@ public class User implements UserDetails {
   private String password;
   private UUID verificationId;
 
+  @OneToMany(mappedBy = "petOwner", cascade = CascadeType.ALL, fetch = FetchType.LAZY)
+  private List<Pet> pets = new ArrayList<>();
+
   @Override
   public Collection<? extends GrantedAuthority> getAuthorities() {
     return List.of(new SimpleGrantedAuthority("ROLE_USER"));