feat : Count open and closed issues(#10)

pbg0205 · Jun 17, 2021 · 14f889d · 14f889d
1 parent 014f54f
commit 14f889d
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/BE/src/main/java/com/codesquad/issuetracker/issue/domain/Issue.java b/BE/src/main/java/com/codesquad/issuetracker/issue/domain/Issue.java
@@ -74,9 +74,21 @@ public void removeAssignee(User user) {
 
     public void setMilestone(Milestone milestone) {
         this.milestone = milestone;
+
+        if(this.open) {
+            this.milestone.countOpenIssuesUp();
+        } else {
+            this.milestone.countClosedIssuesUp();
+        }
     }
 
     public void removeMilestone() {
+        if(this.open) {
+            this.milestone.countOpenIssuesDown();
+        } else {
+            this.milestone.countClosedIssuesDown();
+        }
+
         if (this.milestone == null) {
             return;
         }

diff --git a/BE/src/main/java/com/codesquad/issuetracker/milestone/domain/Milestone.java b/BE/src/main/java/com/codesquad/issuetracker/milestone/domain/Milestone.java
@@ -53,4 +53,20 @@ public void update(MilestoneRequestDto milestoneRequest) {
         this.description = milestoneRequest.getDescription();
         this.dueDate = milestoneRequest.getDueDate();
     }
+
+    public void countOpenIssuesUp() {
+        this.openIssues++;
+    }
+
+    public void countClosedIssuesUp() {
+        this.closedIssues++;
+    }
+
+    public void countOpenIssuesDown() {
+        this.openIssues--;
+    }
+
+    public void countClosedIssuesDown() {
+        this.closedIssues--;
+    }
 }