From 44fc3bfa488c8ce6fa0f9a128e84441b4f9d2e12 Mon Sep 17 00:00:00 2001
From: anujraghuwanshi9900 <raghuanuj810@gmail.com>
Date: Mon, 18 Nov 2024 19:17:24 +0530
Subject: [PATCH] Update Advisor.java

---
 .../wellsfargo/counselor/entity/Advisor.java  | 35 +++++++++++++++----
 1 file changed, 28 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/wellsfargo/counselor/entity/Advisor.java b/src/main/java/com/wellsfargo/counselor/entity/Advisor.java
index 1a6861ec..02ea17b9 100644
--- a/src/main/java/com/wellsfargo/counselor/entity/Advisor.java
+++ b/src/main/java/com/wellsfargo/counselor/entity/Advisor.java
@@ -1,16 +1,14 @@
 package com.wellsfargo.counselor.entity;
 
-
-import jakarta.persistence.Column;
-import jakarta.persistence.Entity;
-import jakarta.persistence.GeneratedValue;
-import jakarta.persistence.Id;
+import jakarta.persistence.*;
+import java.util.ArrayList;
+import java.util.List;
 
 @Entity
 public class Advisor {
 
     @Id
-    @GeneratedValue()
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
     private long advisorId;
 
     @Column(nullable = false)
@@ -28,10 +26,14 @@ public class Advisor {
     @Column(nullable = false)
     private String email;
 
-    protected Advisor() {
+    @OneToMany(mappedBy = "advisor", cascade = CascadeType.ALL, orphanRemoval = true)
+    private List<Client> clients = new ArrayList<>();
 
+    // Default constructor required by JPA
+    protected Advisor() {
     }
 
+    // Constructor for initialization
     public Advisor(String firstName, String lastName, String address, String phone, String email) {
         this.firstName = firstName;
         this.lastName = lastName;
@@ -40,6 +42,7 @@ public Advisor(String firstName, String lastName, String address, String phone,
         this.email = email;
     }
 
+    // Getters and Setters
     public Long getAdvisorId() {
         return advisorId;
     }
@@ -83,4 +86,22 @@ public String getEmail() {
     public void setEmail(String email) {
         this.email = email;
     }
+
+    public List<Client> getClients() {
+        return clients;
+    }
+
+    public void setClients(List<Client> clients) {
+        this.clients = clients;
+    }
+
+    public void addClient(Client client) {
+        clients.add(client);
+        client.setAdvisor(this);
+    }
+
+    public void removeClient(Client client) {
+        clients.remove(client);
+        client.setAdvisor(null);
+    }
 }