[TIL_Carrotww] 66 - 22/12/06

📝Carrotww의 코딩 기록장

🧲 Daphne Channels

🔍 Channels를 이용하여 사용자 일대일 채팅 기능을 구현하였는데 말도 안되는 오류에서 시간을 너무 많이 써 혹시 몰라 끄적여 본다. 일대일 채팅의 기능을 모두 구현한 후 서로 방을 만들고 채널 레이어에서 채팅 전송까지 테스트를 확인하였으나 DB 저장 로직을 추가 하였을때 자꾸 리로드 되어 연결된 websocket이 해제되는 현상이 있었다.

daphne로 서버를 돌려보기도 하고 여러 방법으로 수행하였는데 하루종일 고생한 것 치고 조금 어이없는 곳이 문제였다.

vscode 한 폴더에서 같이 열고 front 테스트는 live server를 django server는 runserver로 테스트 하였는데 다른 vscode에서 liveserver를 열고 테스트를 진행하였을때는 잘 되는 것이였다.

위 오류의 이유인 즉슨 한 vscode에서 db 저장 로직을 추가 후 테스트 하게 되면 db가 저장, 갱신되며 파일의 저장이 이루어지고 vscode가 새로고침되어 liveserver도 새로고침과 동시에 소켓연결이 끊기게 되는 것이였다..
이걸 하루동안 찾았는데 역시 단순한곳에 문제가 있었다 ㅠㅠㅠ 어이없는 오류..? 였는데 무튼 이제는 잘 된다.

class CreateRoom(AsyncWebsocketConsumer):
    # django channels authentication 로그인 관련한 인증 기능 추가해야 함
    async def connect(self):
        # get room_id value
        self.room_name = self.scope["url_route"]["kwargs"]["room_id"]
        self.room_group_name = "chat_%s" % self.room_name

        await self.channel_layer.group_add(
            self.room_group_name,
            self.channel_name
        )

        await self.accept()

    async def disconnect(self, close_code):
        await self.channel_layer.group_discard(
            self.room_group_name,
            self.channel_name
        )

    async def receive(self, text_data):
        text_data_json = json.loads(text_data)
        room_id = text_data_json['room_id']
        message = text_data_json['message']
        # 프론트에서 sender_id, receiver_id 보내주는 로직 필요
        sender_id = text_data_json['sender_id']
        receiver_id = text_data_json['receiver_id']

        sender = await self.get_user_db(sender_id)
        receiver = await self.get_user_db(receiver_id)
        room_object = await self.get_chatroom_db(room_id)

        if not sender:
            print('Sender user가 조회되지 않습니다.')
        if not receiver:
            print('Receiver user가 조회되지 않습니다.')
        if not message:
            print('message가 없습니다.')
            return

        await self.create_chat_log(room_object, sender, message)

        cur_datetime = datetime.now()
        ampm = cur_datetime.strftime('%p')

        cur_time = datetime.now().strftime('%I:%M')            
        date = datetime.now().strftime('%Y년 %m월 %d일')
        cur_time = f"AM {cur_time}" if ampm == 'AM' else f"PM {cur_time}"

        response_json = {
            'message': message,
            'sender': sender.id,
            'room_id': room_id,
            'cur_time': cur_time,
            'date': date
            }

        await self.channel_layer.group_send(
            self.room_group_name,
            {
                'type': 'chat_message',
                'message': json.dumps(response_json)
            }
        )

    # Receive message from room group
    async def chat_message(self, event):
        message_data = json.loads(event['message'])
        message = message_data['message']
        sender = message_data['sender']
        cur_time = message_data['cur_time']
        date = message_data['date']
        room_id = message_data['room_id']

        # Send message to WebSocket
        await self.send(text_data=json.dumps({
            "message": message,
            "sender": sender,
            "cur_time": cur_time,
            "date": date,
            "room_id": room_id
            }))

    @database_sync_to_async
    def get_user_db(self, user_id):
        user = User.objects.filter(id=user_id)
        if user:
            return user[0]
        return None

    @database_sync_to_async
    def get_chatroom_db(self, room_id):
        room = ChatRoom.objects.filter(id=room_id)
        if room:
            return room[0]
        return None

    @database_sync_to_async
    def create_chat_log(self, room_object, sender, content):
        RoomMessage.objects.create(room=room_object, user=sender, content=content)

코드는 직관적이고 해당 함수들은 channels 내부 함수로 room id를 받아 group을 생성하고 해당 그룹에 채널 레이어를 만든다.
해당 채널 레이어로 메세지를 전송하며 전송함과 동시에 db에 메세지가 저장이 되는 것이다.

🧲 django orm chain

🔍 chatroom 모델에 송신자 수신자 순서 상관없이 먼저 보내는 사람이 송신자가 되어 채팅 방이 만들어 진다.
즉 어떤 사용자가 포함된 모든 방을 가져오려면 chatroom 모델에서 가져오고 싶은 사용자가 송신자일때, 수신자일때 모두 가져와서 Queryset을 합쳐야 한다.
검색해보니 chain이라는 것이 있었고 쉽게 구현할 수 있었따.

model 코드

class ChatRoom(models.Model):
    sender = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, related_name='Room_sender')
    receiver = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, related_name='Room_receiver')

    def __str__(self):
        return str(f'id : {self.id}, sender : {self.sender}, receiver : {self.receiver}')

view 작성 코드

opponent_receiver = ChatRoom.objects.filter(sender=cur_user_id)
opponent_sender = ChatRoom.objects.filter(receiver=cur_user_id)

temp_opp_receiver = [User.objects.get(email=x.receiver) for x in opponent_receiver]
temp_opp_sender = [User.objects.get(email=x.sender) for x in opponent_sender]

result_opp_user = list(chain(temp_opp_receiver, temp_opp_sender))

slz = UserListSerializer(result_opp_user, many=True)

코드를 다 짜고 생각해보니 채팅이 온 순서대로 정렬을 해주어야 한다...
그 코드는 바로 떠오르지는 않고 조금 생각을 하고 다시 짜보아야 될 것 같다.