Updated logic and completed holeplay subsystem

1 files changed, 10 insertions, 0 deletions

diff --git a/src/main/java/design/model/Golfer.java b/src/main/java/design/model/Golfer.java
index 04ad8bc..19d6ac6 100644
--- a/src/main/java/design/model/Golfer.java
+++ b/src/main/java/design/model/Golfer.java
@@ -11,6 +11,7 @@ public class Golfer {
     private String fullName;
     private final List<Course> courses;
     private final List<Round> rounds;
+    private final List<Club> clubs = new ArrayList<>(); // Keep track of golfer's clubs
 
     @JsonCreator
     private Golfer(String username, int passwordHash, String fullName, List<Course> courses, List<Round> rounds) {
@@ -72,4 +73,13 @@ public class Golfer {
     public void addRound(Round round) {
         rounds.add(round);
     }
+
+    // Helpers dealing with clubs
+    public void addClub(Club c) {
+        clubs.add(c);
+    }
+
+    public boolean hasClub(Club c) {
+        return clubs.contains(c);
+    }
 }