feat: 투표 블럭 서비스 로직 DB와 연동

Author: se030

server/apis/mom/block/model.ts

@@ -2,7 +2,7 @@ import mongoose from '@db';
 import { Schema } from 'mongoose';
 import { BlockType } from '@wabinar/api-types/block';
 
-import { Vote } from '../vote/service';
+import { Vote } from './vote/service';
 import { Question } from './question/service';
 
 interface Block {

server/apis/mom/block/service.ts

@@ -2,6 +2,7 @@ import LinkedList from '@wabinar/crdt/linked-list';
 import { BlockType } from '@wabinar/api-types/block';
 import blockModel from './model';
 import { Question } from './question/service';
+import { Vote } from './vote/service';
 
 export const getBlock = async (id: string) => {
   const block = await blockModel.findOne({ id });
@@ -27,7 +28,7 @@ export const putBlockType = async (id: string, type: BlockType) => {
 export const putBlock = async (
   id: string,
   type: BlockType,
-  data: LinkedList | Question[],
+  data: LinkedList | Question[] | Vote,
 ) => {
   switch (type) {
     case BlockType.H1:
@@ -37,6 +38,7 @@ export const putBlock = async (
       await putTextBlock(id, data as LinkedList);
       break;
     case BlockType.VOTE:
+      await putVoteBlock(id, data as Vote);
       break;
     case BlockType.QUESTION:
       await putQuestionBlock(id, data as Question[]);
@@ -53,6 +55,10 @@ const putTextBlock = async (id: string, data: LinkedList) => {
   );
 };
 
+const putVoteBlock = async (id: string, vote: Vote) => {
+  await blockModel.updateOne({ id }, { voteProperties: vote });
+};
+
 const putQuestionBlock = async (id: string, questions: Question[]) => {
   await blockModel.updateOne({ id }, { questionProperties: questions });
 };

server/apis/mom/block/vote/service.ts

@@ -0,0 +1,86 @@
+import { BlockType } from '@wabinar/api-types/block';
+import { getBlock, putBlock } from '../service';
+
+export interface Option {
+  id: number;
+  text: string;
+  count: number;
+}
+
+interface Options {
+  [index: number]: Omit<Option, 'id'>;
+}
+
+export interface Vote {
+  title: string;
+  options: Options;
+  isDoing: boolean;
+  participants: { [index: number]: number };
+}
+
+export const createVote = async (blockId: string, options: Option[]) => {
+  const processedOptions = options.reduce((obj, { id, text }) => {
+    obj[id] = { text, count: 0 };
+    return obj;
+  }, {} as Options);
+
+  await putBlock(blockId, BlockType.VOTE, {
+    title: '투표',
+    options: processedOptions,
+    isDoing: true,
+    participants: {},
+  });
+};
+
+export const updateVote = async (
+  blockId: string,
+  optionIdString: string,
+  userId: number,
+) => {
+  const voteBlock = await getBlock(blockId);
+
+  const { voteProperties: vote } = voteBlock;
+
+  if (!vote) return;
+
+  const optionId = Number(optionIdString);
+  const option = vote.options[optionId];
+
+  if (!option) return;
+
+  const count = option.count;
+
+  if (!vote.participants[userId]) {
+    vote.options[optionId] = { ...option, count: count + 1 };
+  }
+
+  vote.participants[userId] = optionId;
+
+  await putBlock(blockId, BlockType.VOTE, vote);
+
+  const participantCount = Object.keys(vote.participants).length;
+
+  return participantCount;
+};
+
+export const endVote = async (blockId: string) => {
+  const voteBlock = await getBlock(blockId);
+
+  const { voteProperties: vote } = voteBlock;
+
+  vote.isDoing = false;
+
+  const participantCount = Object.keys(vote.participants).length;
+
+  await putBlock(blockId, BlockType.VOTE, vote);
+
+  const result = {
+    options: Object.entries(vote.options).map(([id, rest]) => ({
+      id,
+      ...rest,
+    })),
+    participantCount,
+  };
+
+  return result;
+};

server/apis/mom/vote/service.ts

@@ -1,4 +1,9 @@
-import * as Vote from '@apis/mom/vote/service';
+import {
+  createVote,
+  endVote,
+  Option,
+  updateVote,
+} from '@apis/mom/block/vote/service';
 import { BLOCK_EVENT } from '@wabinar/constants/socket-message';
 import { Server, Socket } from 'socket.io';
 
@@ -7,23 +12,29 @@ export default function handleVoteBlock(
   namespace: string,
   socket: Socket,
 ) {
-  socket.on(BLOCK_EVENT.CREATE_VOTE, (options: Vote.Option[]) => {
+  socket.on(BLOCK_EVENT.CREATE_VOTE, async (blockId, options: Option[]) => {
     const momId = socket.data.momId;
-    const newVote = Vote.createVote(momId, options);
 
-    socket.to(momId).emit(BLOCK_EVENT.CREATE_VOTE, newVote);
+    await createVote(blockId, options);
+
+    socket.to(momId).emit(`${BLOCK_EVENT.CREATE_VOTE}-${blockId}`, options);
   });
 
-  socket.on(BLOCK_EVENT.UPDATE_VOTE, (optionId, userId) => {
+  socket.on(BLOCK_EVENT.UPDATE_VOTE, async (blockId, optionId, userId) => {
     const momId = socket.data.momId;
-    const participantCount = Vote.updateVote(momId, Number(optionId), userId);
 
-    io.of(namespace).to(momId).emit(BLOCK_EVENT.UPDATE_VOTE, participantCount);
+    const participantCount = await updateVote(blockId, optionId, userId);
+
+    io.of(namespace)
+      .to(momId)
+      .emit(`${BLOCK_EVENT.UPDATE_VOTE}-${blockId}`, participantCount);
   });
 
-  socket.on(BLOCK_EVENT.END_VOTE, () => {
+  socket.on(BLOCK_EVENT.END_VOTE, async (blockId) => {
     const momId = socket.data.momId;
-    const res = Vote.endVote(momId);
-    io.of(namespace).to(momId).emit(BLOCK_EVENT.END_VOTE, res);
+
+    const res = await endVote(blockId);
+
+    io.of(namespace).to(momId).emit(`${BLOCK_EVENT.END_VOTE}-${blockId}`, res);
   });
 }