[go] Go 언어를 사용하여 웹소켓으로 실시간 뉴스 알림 구현하기

22 Dec 2023

최근에는 실시간 뉴스 알림이 매우 중요한 기능으로 여겨지고 있습니다. 실시간으로 뉴스를 받아오기 위해서는 웹소켓을 활용하는 것이 가장 효율적인 방법 중 하나입니다. 이번에는 Go 언어를 사용하여 실시간 뉴스 알림을 위한 웹소켓 기능을 구현하는 방법에 대해 알아보겠습니다.

웹소켓이란?

웹소켓은 서버와 클라이언트 간의 양방향 통신을 가능하게 하는 프로토콜입니다. HTTP와 마찬가지로 전이중(full-duplex) 통신을 지원하지만, 더 작은 오버헤드와 빠른 연결 설정 시간을 갖고 있습니다. 이러한 특징으로 인해 실시간 데이터 전송이 필요한 애플리케이션에 적합하며, 실시간 뉴스 알림 기능을 구현하기에 이상적입니다.

Go를 사용한 웹소켓 구현

필요한 패키지 설치

먼저, Go에서 웹소켓 기능을 구현하기 위해 gorilla/websocket 패키지를 설치해야 합니다. 다음 명령을 사용하여 패키지를 설치합니다.

go get github.com/gorilla/websocket

서버 구현

아래는 간단한 웹소켓 서버의 구현 예시입니다.

package main

import (
    "net/http"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize: 1024,
    WriteBufferSize: 1024,
}

func wsHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        // error handling
        return
    }
    defer conn.Close()

    for {
        // 클라이언트로부터 메시지 수신
        messageType, p, err := conn.ReadMessage()
        if err != nil {
            // error handling
            return
        }

        // 메시지 처리
        // ...

        // 클라이언트로 메시지 전송
        if err := conn.WriteMessage(messageType, p); err != nil {
            // error handling
            return
        }
    }
}

func main() {
    http.HandleFunc("/ws", wsHandler)
    http.ListenAndServe(":8080", nil)
}

클라이언트 구현

웹소켓 클라이언트를 구현할 때는 JavaScript를 사용하는 것이 일반적입니다. 웹 페이지에서 다음과 같이 웹소켓을 이용하여 서버와 연결하고 데이터를 주고받을 수 있습니다.

var socket = new WebSocket("ws://localhost:8080/ws");

socket.onopen = function(event) {
    console.log("웹소켓 연결됨");
};

socket.onmessage = function(event) {
    var receivedData = event.data;
    // 받은 데이터 처리
};

socket.onclose = function(event) {
    console.log("웹소켓 연결 종료됨");
};

마치며

위 예시를 참고하여 Go 언어를 사용하여 웹소켓을 통한 실시간 뉴스 알림을 구현할 수 있습니다. 실제로 더 복잡한 뉴스 알림 시스템을 구현할 때에는 보다 다양한 기능과 안정성을 고려해야 합니다. 하지만 기본적인 개념을 숙지하고 있는 것은 향후 실무에서 뉴스 알림 시스템을 구현할 때 큰 도움이 될 것입니다.