Add code to handle id serialization

1 files changed, 8 insertions, 1 deletions

diff --git a/src/main/java/design/model/Course.java b/src/main/java/design/model/Course.java
index 8dfecd8..f90c8d9 100644
--- a/src/main/java/design/model/Course.java
+++ b/src/main/java/design/model/Course.java
@@ -3,6 +3,8 @@ package design.model;
 import java.util.List;
 
 public class Course implements ICourse {
+
+    private final int id;
     private final String name;
     private final int difficultyRating;
     private final String location;
@@ -10,7 +12,8 @@ public class Course implements ICourse {
     private final int totalPar;
     private final List<Hole> holes;
 
-    public Course(String name, int difficultyRating, String location, int holeCount, int totalPar, List<Hole> holes) {
+    public Course(int id, String name, int difficultyRating, String location, int holeCount, int totalPar, List<Hole> holes) {
+        this.id = id;
         this.name = name;
         this.difficultyRating = difficultyRating;
         this.location = location;
@@ -19,6 +22,10 @@ public class Course implements ICourse {
         this.holes = holes;
     }
 
+    public int getId() {
+        return id;
+    }
+
     public String getName() {
         return name;
     }