Round data is stored to each golfer after a match is concluded

1 files changed, 12 insertions, 0 deletions

diff --git a/src/main/java/design/model/StrokeLeague.java b/src/main/java/design/model/StrokeLeague.java
index 8a61659..374a079 100644
--- a/src/main/java/design/model/StrokeLeague.java
+++ b/src/main/java/design/model/StrokeLeague.java
@@ -70,4 +70,16 @@ public class StrokeLeague extends League {
     public Map<Golfer, Integer> getResults() {
         return getTotalStrokes();
     }
+
+    @Override
+    public void finalizeMatch(Match match) {
+        getCompletedMatches().add(match);
+        for(int i = 0; i < participants.size(); i++){
+            Golfer player = participants.get(i);
+            Round round = match.getRoundFor(i);
+            if(round != null){
+                player.addRound(round);
+            }
+        }
+    }
 }