Skip to content
Snippets Groups Projects
Commit ed6661f5 authored by tpgus2603's avatar tpgus2603
Browse files

refact: 친구 추가 로직,데이터모델 변경(#7)

parent dcb0f956
Branches
No related tags found
2 merge requests!31Develop,!10[#7] 친구서비스 로직 수정 및 모델추가
DB_HOST=localhost
DB_USER=root
DB_PASSWORD=0000
DB_NAME=meeting
MONGO_URI=mongodb://localhost:27017/your_mongo_db
PORT=8080
CALLBACK_URL=http://localhost:8080/auth/google/callback
\ No newline at end of file
node_modules/ node_modules/
.env/ .env
config.json/ config.json/
const FriendService = require('../services/friendService'); const FriendService = require('../services/friendService');
class friendController { class friendController {
/**
* 친구 요청 보내기
* POST /api/friend/request/:friendId
*/
async sendRequest(req, res) {
try {
const userId = req.user.id;
const { friendId } = req.params;
const request = await FriendService.sendFriendRequest(userId, friendId);
return res.status(201).json({
success: true, class FriendController {
data: request /**
}); * 친구 요청 보내기
} catch (error) { * 클라이언트는 userId와 요청을 보낼 사용자의 email을 전송
return res.status(400).json({ * @param {Object} req - Express 요청 객체
success: false, * @param {Object} res - Express 응답 객체
error: { * @param {Function} next - Express next 미들웨어 함수
message: error.message, */
code: 'FRIEND_REQUEST_ERROR' async sendFriendRequest(req, res, next) {
const { userId, email } = req.body;
try {
if (!userId || !email) {
return res.status(400).json({ message: 'userId와 email은 필수 입력 항목입니다.' });
} }
}); // 친구 요청을 받을 사용자의 정보 조회 (서비스로 분리할지 생각)
const receiver = await User.findOne({ where: { email: email } });
if (!receiver) {
return res.status(404).json({ message: '요청을 받을 사용자를 찾을 수 없습니다.' });
}
const friendId = receiver.id;
// 친구 요청 보내기 서비스 호출
const friendRequest = await friendService.sendFriendRequest(userId, friendId);
return res.status(201).json({
success:true,
data:friendRequest
});
} catch (error) {
// 유니크 제약조건 오류 처리
if (error.message === 'Friend request already exists') {
return res.status(409).json({ message: error.message });
}
// 일반 오류 처리
return res.status(500).json({ message: '서버 오류가 발생했습니다.', error: error.message });
}
} }
} }
......
...@@ -13,12 +13,22 @@ const Friend = sequelize.define('Friend', { ...@@ -13,12 +13,22 @@ const Friend = sequelize.define('Friend', {
}, { }, {
tableName: 'Friends', tableName: 'Friends',
timestamps: true, timestamps: true,
indexes: [
{
unique: true,
fields: ['requester_id', 'receiver_id']
},
{
fields: ['status']
}
]
}); });
Friend.belongsTo(User, { foreignKey: 'user_id', as: 'user' }); // 관계 설정
Friend.belongsTo(User, { foreignKey: 'friend_id', as: 'friend' }); Friend.belongsTo(User, { foreignKey: 'requester_id', as: 'requester' }); // 친구 요청을 보낸 사용자
Friend.belongsTo(User, { foreignKey: 'receiver_id', as: 'receiver' }); // 친구 요청을 받은 사용자
User.hasMany(Friend, { foreignKey: 'user_id', as: 'friends' }); User.hasMany(Friend, { foreignKey: 'requester_id', as: 'sentRequests' }); // 친구 요청을 보낸 목록
User.hasMany(Friend, { foreignKey: 'friend_id', as: 'friendRequests' }); User.hasMany(Friend, { foreignKey: 'receiver_id', as: 'receivedRequests' }); // 친구 요청을 받은 목록
module.exports = Friend; module.exports = Friend;
\ No newline at end of file
This diff is collapsed.
...@@ -29,6 +29,7 @@ ...@@ -29,6 +29,7 @@
"sequelize-cli": "^6.6.2" "sequelize-cli": "^6.6.2"
}, },
"devDependencies": { "devDependencies": {
"artillery": "^2.0.21",
"nodemon": "^3.1.7" "nodemon": "^3.1.7"
} }
} }
// services/friendService.js
const { Op } = require('sequelize'); const { Op } = require('sequelize');
const Friend = require('../models/Friend'); const Friend = require('../models/Friend');
const User = require('../models/User'); const User = require('../models/User');
const sequelize = require('../config/sequelize'); // 트랜잭션을 위해 추가
class friendService { class FriendService {
/** /**
* User 존재 여부 유효성 검사 * User 존재 여부 유효성 검사
* @param {number} userId - 검사할 사용자 ID
* @returns {Promise<User>} - 유효한 사용자 객체
* @throws {Error} - 사용자가 존재하지 않을 경우
*/ */
async validUser(userId) { async validUser(userId) {
const user = await User.findByPk(userId); const user = await User.findByPk(userId);
...@@ -14,10 +19,13 @@ class friendService { ...@@ -14,10 +19,13 @@ class friendService {
} }
return user; return user;
} }
/** /**
* 친구 요청 보내기 * 친구 요청 보내기
* 나 자신에게 보내기 or 이미 존재하는 친구 -> X * @param {number} userId - 친구 요청을 보내는 사용자 ID
* 이후, PENDING 상태로 변환 -> 수락/거절에 따라 변화 * @param {number} friendId - 친구 요청을 받는 사용자 ID
* @returns {Promise<Friend>} - 생성된 친구 요청 객체
* @throws {Error} - 유효하지 않은 요청일 경우
*/ */
async sendFriendRequest(userId, friendId) { async sendFriendRequest(userId, friendId) {
await this.validUser(userId); await this.validUser(userId);
...@@ -27,38 +35,34 @@ class friendService { ...@@ -27,38 +35,34 @@ class friendService {
throw new Error('Cannot send friend request to yourself'); throw new Error('Cannot send friend request to yourself');
} }
const existingFriend = await Friend.findOne({ try {
where: { return await Friend.create({
[Op.or]: [ requester_id: userId,
{ user_id: userId, friend_id: friendId }, receiver_id: friendId,
{ user_id: friendId, friend_id: userId } status: 'PENDING'
] });
} catch (error) {
if (error.name === 'SequelizeUniqueConstraintError') {
throw new Error('Friend request already exists');
} }
}); throw error;
if (existingFriend) {
throw new Error('Friend request already exists');
} }
return Friend.create({
user_id: userId,
friend_id: friendId,
status: 'PENDING'
});
} }
/** /**
* 받은 친구 요청 목록 조회 * 받은 친구 요청 목록 조회
* @param {number} userId - 요청을 받은 사용자 ID
* @returns {Promise<Array>} - 받은 친구 요청 목록
*/ */
async getReceivedRequests(userId) { async getReceivedRequests(userId) {
return Friend.findAll({ return Friend.findAll({
where: { where: {
friend_id: userId, receiver_id: userId,
status: 'PENDING' status: 'PENDING'
}, },
include: [{ include: [{
model: User, model: User,
as: 'user', as: 'requester',
attributes: ['id', 'name', 'email'] attributes: ['id', 'name', 'email']
}] }]
}); });
...@@ -66,16 +70,18 @@ class friendService { ...@@ -66,16 +70,18 @@ class friendService {
/** /**
* 보낸 친구 요청 목록 조회 * 보낸 친구 요청 목록 조회
* @param {number} userId - 요청을 보낸 사용자 ID
* @returns {Promise<Array>} - 보낸 친구 요청 목록
*/ */
async getSentRequests(userId) { async getSentRequests(userId) {
return Friend.findAll({ return Friend.findAll({
where: { where: {
user_id: userId, requester_id: userId,
status: 'PENDING' status: 'PENDING'
}, },
include: [{ include: [{
model: User, model: User,
as: 'user', as: 'receiver',
attributes: ['id', 'name', 'email'] attributes: ['id', 'name', 'email']
}] }]
}); });
...@@ -83,96 +89,116 @@ class friendService { ...@@ -83,96 +89,116 @@ class friendService {
/** /**
* 친구 요청 수락 * 친구 요청 수락
* @param {number} userId - 요청을 수락하는 사용자 ID
* @param {number} friendId - 친구 요청을 보낸 사용자 ID
* @returns {Promise<Friend>} - 업데이트된 친구 요청 객체
* @throws {Error} - 친구 요청이 존재하지 않을 경우
*/ */
async acceptFriendRequest(userId, friendId) { async acceptFriendRequest(userId, friendId) {
const request = await Friend.findOne({ const transaction = await sequelize.transaction();
where: { try {
user_id: friendId, const request = await Friend.findOne({
friend_id: userId, where: {
status: 'PENDING' requester_id: friendId,
receiver_id: userId,
status: 'PENDING'
},
transaction
});
if (!request) {
throw new Error('Friend request not found');
} }
});
await request.update({ status: 'ACCEPTED' }, { transaction });
if (!request) {
throw new Error('Friend request not found'); await transaction.commit();
return request;
} catch (error) {
await transaction.rollback();
throw error;
} }
return request.update({ status: 'ACCEPTED' });
} }
/** /**
* 친구 요청 거절 * 친구 요청 거절
* @param {number} userId - 요청을 거절하는 사용자 ID
* @param {number} friendId - 친구 요청을 보낸 사용자 ID
* @returns {Promise<number>} - 삭제된 친구 요청 수
* @throws {Error} - 친구 요청이 존재하지 않을 경우
*/ */
async rejectFriendRequest(userId, friendId) { async rejectFriendRequest(userId, friendId) {
const result = await Friend.destroy({ const result = await Friend.destroy({
where: { where: {
user_id: friendId, requester_id: friendId,
friend_id: userId, receiver_id: userId,
status: 'PENDING' status: 'PENDING'
} }
}); });
if (!result) { if (!result) {
throw new Error('Friend request not found'); throw new Error('Friend request not found');
} }
return result; return result;
} }
/** /**
* 친구 목록 조회 * 친구 목록 조회
* @param {number} userId - 친구 목록을 조회할 사용자 ID
* @returns {Promise<Array>} - 친구 목록
*/ */
async getFriendList(userId) { async getFriendList(userId) {
const friends = await Friend.findAll({ const friends = await Friend.findAll({
where: { where: {
[Op.or]: [ [Op.or]: [
{ user_id: userId }, { requester_id: userId },
{ friend_id: userId } { receiver_id: userId }
], ],
status: 'ACCEPTED' status: 'ACCEPTED'
}
});
const friendIds = friends.map(friend =>
friend.user_id === userId ? friend.friend_id : friend.user_id
);
const friendUsers = await User.findAll({
where: {
id: friendIds
}, },
attributes: ['id', 'name', 'email'] include: [
{
model: User,
as: 'requester',
attributes: ['id', 'name', 'email']
},
{
model: User,
as: 'receiver',
attributes: ['id', 'name', 'email']
}
]
}); });
const friendsMap = friendUsers.reduce((map, user) => {
map[user.id] = user;
return map;
}, {});
return friends.map(friend => { return friends.map(friend => {
const isSender = friend.user_id === userId; const isRequester = friend.requester_id === userId;
const friendId = isSender ? friend.friend_id : friend.user_id; const friendInfo = isRequester ? friend.receiver : friend.requester;
return { return {
id: friend.id, id: friend.id,
status: friend.status, status: friend.status,
createdAt: friend.createdAt, createdAt: friend.createdAt,
updatedAt: friend.updatedAt, updatedAt: friend.updatedAt,
friendInfo: friendsMap[friendId], friendInfo: friendInfo,
relationshipType: isSender ? 'sent' : 'received' relationshipType: isRequester ? 'sent' : 'received'
}; };
}); });
} }
/** /**
* 친구 삭제 * 친구 삭제
* @param {number} userId - 친구를 삭제하는 사용자 ID
* @param {number} friendId - 삭제할 친구의 사용자 ID
* @returns {Promise<number>} - 삭제된 친구 관계 수
* @throws {Error} - 친구 관계가 존재하지 않을 경우
*/ */
async deleteFriend(userId, friendId) { async deleteFriend(userId, friendId) {
const result = await Friend.destroy({ const result = await Friend.destroy({
where: { where: {
[Op.or]: [ [Op.or]: [
{user_id: userId, friend_id: friendId}, { requester_id: userId, receiver_id: friendId },
{user_id: friendId, friend_id: userId} { requester_id: friendId, receiver_id: userId }
], ],
status: 'ACCEPTED' status: 'ACCEPTED'
} }
...@@ -183,7 +209,6 @@ class friendService { ...@@ -183,7 +209,6 @@ class friendService {
} }
return result; return result;
} }
} }
module.exports = new friendService(); module.exports = new FriendService();
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment