Fix votes remaining lag.

bilus · Sep 15, 2020 · 5bea62a · 5bea62a
1 parent 930e05a
commit 5bea62a
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/actions/votes.go b/actions/votes.go
@@ -31,6 +31,11 @@ func (c PostsController) VotesCreate() error {
 		return c.Render(http.StatusUnprocessableEntity, r.JavaScript("error.js"))
 	}
 
+	votesRemaining, err := c.VotingService.VotesRemaining(&c.CurrentUser, c.Board)
+	if err == nil {
+		c.Set("votesRemaining", votesRemaining)
+	}
+
 	c.Set("post", c.Post)
 	c.Set("board", &c.Post.Board)
 	return c.Render(http.StatusOK, r.JavaScript("votes/create.js"))
@@ -56,6 +61,12 @@ func (c PostsController) VotesDestroy() error {
 	} else {
 		return c.Render(http.StatusUnprocessableEntity, r.JavaScript("votes/fail.js"))
 	}
+
+	votesRemaining, err := c.VotingService.VotesRemaining(&c.CurrentUser, c.Board)
+	if err == nil {
+		c.Set("votesRemaining", votesRemaining)
+	}
+
 	c.Set("post", c.Post)
 	c.Set("board", &c.Post.Board)
 	return c.Render(http.StatusOK, r.JavaScript("votes/destroy.js"))