1 files changed, 34 insertions, 5 deletions
diff --git a/src/main/java/design/model/StrokeLeague.java b/src/main/java/design/model/StrokeLeague.java
index f09cabf..487a2b5 100644
--- a/src/main/java/design/model/StrokeLeague.java
+++ b/src/main/java/design/model/StrokeLeague.java
@@ -3,32 +3,61 @@ package design.model;
 import com.fasterxml.jackson.annotation.JsonCreator;
 
 import java.util.ArrayList;
+import java.util.Collections;
+import java.util.Comparator;
 import java.util.Date;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 public class StrokeLeague extends League {
-    private final List<Golfer> participants;
+    private List<Golfer> participants;
+    private Map<Golfer, Integer> totalStrokes;
 
     @JsonCreator
     private StrokeLeague(int id, String name, Date registrationDate, Date startDate, Date endDate, Golfer owner, List<Golfer> participants, List<Match> schedule) {
         super(id, name, registrationDate, startDate, endDate, owner, schedule);
         this.participants = participants;
+        this.totalStrokes = new HashMap<>();
+        participants.forEach(p -> totalStrokes.putIfAbsent(p, 0));
     }
 
     public StrokeLeague(String name, Date registrationDate, Date startDate, Date endDate, Golfer owner) {
         super(name, registrationDate, startDate, endDate, owner);
         this.participants = new ArrayList<>();
+        this.totalStrokes = new HashMap<>();
     }
 
-    public boolean addParticipants(Golfer e) {
-        return participants.add(e);
+    public boolean addParticipants(Golfer g) {
+        boolean added = participants.add(g);
+        if(added) totalStrokes.putIfAbsent(g, 0);
+        return added;
     }
 
-    public boolean removeParticipants(Golfer o) {
-        return participants.remove(o);
+    public boolean removeParticipants(Golfer g) {
+        totalStrokes.remove(g);
+        return participants.remove(g);
     }
 
     public Golfer[] getParticipants() {
         return participants.toArray(Golfer[]::new);
     }
+
+    @Override
+    public void recordPlay(Golfer player, Match match, Round round){
+        if(!isPlayable() || !participants.contains(player)) return;
+        int strokes = round.getTotalSwings();
+        totalStrokes.merge(player, strokes, Integer::sum);
+        match.addRound(round);
+    }
+
+    @Override
+    public void finalizeLeague(){
+        markCompleted();
+        participants.sort(Comparator.comparingInt(totalStrokes::get));
+    }
+    
+    public Map<Golfer, Integer> getTotalStrokes(){
+        return Collections.unmodifiableMap(totalStrokes);
+    }
 }