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

1 files changed, 11 insertions, 0 deletions

diff --git a/src/main/java/design/model/Match.java b/src/main/java/design/model/Match.java
index 8126e7c..525f281 100644
--- a/src/main/java/design/model/Match.java
+++ b/src/main/java/design/model/Match.java
@@ -5,7 +5,9 @@ import com.fasterxml.jackson.annotation.JsonCreator;
 import java.time.LocalDateTime;
 import java.util.ArrayList;
 import java.util.Date;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 public class Match {
     private final Course course;
@@ -14,6 +16,7 @@ public class Match {
     private final LocalDateTime end;
     private final int holeCount;
     private final List<Round> rounds;
+    private final Map<Integer, Round> roundMap = new HashMap<>();
 
     @JsonCreator
     private Match(Course course, Date dateScheduled, LocalDateTime start, LocalDateTime end, int holeCount, List<Round> rounds) {
@@ -61,4 +64,12 @@ public class Match {
     public boolean checkCompletion() {
         return LocalDateTime.now().isAfter(end);
     }
+
+    public void addRoundFor(int participantIndex, Round r) {
+        roundMap.put(participantIndex, r);
+    }
+
+    public Round getRoundFor(int participantIndex) {
+        return roundMap.get(participantIndex);
+    }
 }