From 4aad66197348b6f19f1156eb2408480d0707f946 Mon Sep 17 00:00:00 2001
From: Soumen Mallick <smnk107@gmail.com>
Date: Sun, 22 Dec 2024 23:44:14 +0530
Subject: [PATCH] added updatePet in Owner

---
 .../samples/petclinic/owner/Owner.java              | 13 +++++++++++++
 .../samples/petclinic/owner/PetController.java      |  4 +++-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/main/java/org/springframework/samples/petclinic/owner/Owner.java b/src/main/java/org/springframework/samples/petclinic/owner/Owner.java
index 675b2140e..789d463ee 100644
--- a/src/main/java/org/springframework/samples/petclinic/owner/Owner.java
+++ b/src/main/java/org/springframework/samples/petclinic/owner/Owner.java
@@ -172,4 +172,17 @@ public class Owner extends Person {
 		pet.addVisit(visit);
 	}
 
+	public void updatePet(Pet newPet)
+	{
+		for(int i=0 ; i<pets.size() ; i++)
+		{
+			Pet existingPet = pets.get(i);
+			if(existingPet.getId().equals(newPet.getId()))
+			{
+				pets.set(i,newPet);
+				break;
+			}
+		}
+	}
+
 }
diff --git a/src/main/java/org/springframework/samples/petclinic/owner/PetController.java b/src/main/java/org/springframework/samples/petclinic/owner/PetController.java
index fcf431bff..e52c8b146 100644
--- a/src/main/java/org/springframework/samples/petclinic/owner/PetController.java
+++ b/src/main/java/org/springframework/samples/petclinic/owner/PetController.java
@@ -146,7 +146,9 @@ class PetController {
 			return VIEWS_PETS_CREATE_OR_UPDATE_FORM;
 		}
 
-		owner.addPet(pet);
+		//owner.addPet(pet);
+		owner.updatePet(pet);
+
 		this.owners.save(owner);
 		redirectAttributes.addFlashAttribute("message", "Pet details has been edited");
 		return "redirect:/owners/{ownerId}";