Django Channels로 실시간 알람 기능 만들기 (2)

2022.07.04 - [프로젝트/capstone] - Django Channels로 실시간 알람 기능 만들기 (1)

Django Channels로 실시간 알람 기능 만들기 (1)

 

위 글에서는 websocket을 이용해 Vue.js와 Django를 연결했습니다.

 

DB에 데이터가 저장되면 프론트로 알림 전송하기

오늘 만들 기능은

DB에 데이터가 저장되면 프론트로 알림을 전송하는 기능입니다.

 

폭력행위가 발생하면 DB에 데이터가 저장되고 해당 데이터가 프론트로 전송되어야합니다.

즉, Django 모델에 데이터가 추가되면 해당 데이터가 웹소켓이 연결된 페이지로 전송되어야합니다.

 

Django Signal

이 때 사용할 기능은 바로 Django의 Signal입니다.

Django Signal

Signals | Django documentation | Django

Django Signal은 DB model에 save가 작동하면, 저장이 완료된 이후에 지정한 동작을 수행할 수 있도록 할 수 있습니다.

 

 

from django.dispatch import receiver 와 from django.db.models.signals import post_save 를 사용합니다

• post_save()의 내용은 모델의 save() 함수의 끝에 전해지게 된다.

 

인자

• sender: 모델 클래스를 저장
• instance: 모델의 record 데이터를 가지고 있습니다.
• created: 새로운 record이 생성되었다면 True 아니라면 False를 반환

 

@receiver(post_save, sender=Notification)
def announce_likes(sender, instance, created, **kwargs):
    if created:
        channel_layer=get_channel_layer()
        async_to_sync(channel_layer.group_send)(
            "shares", {
                "type": "share_notification",
                "message": instance.content,
            }
    )

 

• decorater(@receiver)를 통해서 signal에 필요한 인자들(post_save, sender)을 넘겨줍니다.
• 만약 Django 모델에 새로운 record가 추가되면
  • channel layer의 group에 Django 모델의 content를 전송합니다

---

코드

# consumers.py

from channels.generic.websocket import WebsocketConsumer
from channels.db import database_sync_to_async
from channels.layers import get_channel_layer
from asgiref.sync import async_to_sync

from django.db.models.signals import post_save
from django.dispatch import receiver

from .models import Notification
from .serializers import NotificationSerializer

# DB model에 관련해서 save가 작동하면, 저장이 완료된 이후에 지정한 동작을 수행
@receiver(post_save, sender=Notification)
def send_update(sender, instance, created, **kwargs):
    print("New reading in DB")
    serializer = NotificationSerializer(instance)

    if created:
        print("New saving in DB")
        channel_layer=get_channel_layer()
        async_to_sync(channel_layer.group_send)(
            "center_name", {
                "type": "notify",
                "data": serializer.data
            }
        )

 

• @receiver(post_save, sender=Notification)
  • Notifiation 모델에 데이터가 저장된 후(post_save) 실행할 코드를 작성해주면 됩니다
• 전송할 Notification 데이터를 serializer에 담아주었습니다
• if created
  • Notification에 데이터가 저장되면 아래 코드를 실행합니다
• channel layer를 가져온 후 serializer의 data를 전송합니다

 

그 겨로가, Notification 모델에 데이터가 추가되면 해당 데이터를 웹소켓이 연결된 페이지로 전송합니다