[#48] feat: add/delete request returns count

handong-life · Jan 26, 2022 · b6234b4 · b6234b4
1 parent 6b9459d
commit b6234b4
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 10 deletions.
diff --git a/server/controllers/timetable.js b/server/controllers/timetable.js
@@ -1,6 +1,7 @@
 const Lecture = require('../models/lecture');
 const Timetable = require('../models/timetable');
 const TimetableLectureRelation = require('../models/timetable_lecture_relation');
+const { addCount } = require('../utils/counter_helper');
 
 exports.getTimetable = async (req, res) => {
   const timetable = await Timetable.findOne({
@@ -39,7 +40,12 @@ exports.addLecture = async (req, res) => {
     timetableId: +timetableId,
     lectureId: +lectureId,
   });
-  res.send('complete');
+  return res.json({
+    msg: 'complete',
+    count: {
+      add: await addCount(+lectureId),
+    },
+  });
 };
 
 exports.deleteLecture = async (req, res) => {
@@ -50,5 +56,10 @@ exports.deleteLecture = async (req, res) => {
       lectureId: +lectureId,
     },
   });
-  res.send('complete');
+  return res.json({
+    msg: 'complete',
+    count: {
+      add: await addCount(lectureId),
+    },
+  });
 };
diff --git a/server/controllers/user.js b/server/controllers/user.js
@@ -4,6 +4,7 @@ const Timetable = require('../models/timetable');
 const User = require('../models/user');
 const UserLectureRelation = require('../models/user_lecture_relation');
 const UserLectureGleaningRelation = require('../models/user_lecture_gleaning_relation');
+const { bookmarkCount, spikeCount } = require('../utils/counter_helper');
 
 exports.getUser = async (req, res) => {
   await User.update(
@@ -26,39 +27,63 @@ exports.getUser = async (req, res) => {
 };
 
 exports.bookmarkLecture = async (req, res) => {
+  const lectureId = +req.params.lectureId;
   await UserLectureRelation.create({
     userId: req.user.id,
-    lectureId: +req.params.lectureId,
+    lectureId,
+  });
+  return res.json({
+    msg: 'complete',
+    count: {
+      bookmark: await bookmarkCount(lectureId),
+    },
   });
-  res.send('complete');
 };
 
 exports.unbookmarkLecture = async (req, res) => {
+  const lectureId = +req.params.lectureId;
   await UserLectureRelation.destroy({
     where: {
       userId: req.user.id,
-      lectureId: +req.params.lectureId,
+      lectureId,
+    },
+  });
+  return res.json({
+    msg: 'complete',
+    count: {
+      bookmark: await bookmarkCount(lectureId),
     },
   });
-  res.send('complete');
 };
 
 exports.addSpikeLecture = async (req, res) => {
+  const lectureId = +req.params.lectureId;
   await UserLectureGleaningRelation.create({
     userId: req.user.id,
-    lectureId: +req.params.lectureId,
+    lectureId,
+  });
+  return res.json({
+    msg: 'complete',
+    count: {
+      spike: await spikeCount(lectureId),
+    },
   });
-  res.send('complete');
 };
 
 exports.deleteSpikeLecture = async (req, res) => {
+  const lectureId = +req.params.lectureId;
   await UserLectureGleaningRelation.destroy({
     where: {
       userId: req.user.id,
-      lectureId: +req.params.lectureId,
+      lectureId,
+    },
+  });
+  return res.json({
+    msg: 'complete',
+    count: {
+      spike: await bookmarkCount(lectureId),
     },
   });
-  res.send('complete');
 };
 
 exports.getBookmarks = async (req, res) => {