[go] Go 언어에서 웹소켓 연결 유지하기

22 Dec 2023

웹소켓은 실시간 양방향 통신을 제공하기 위한 프로토콜로, 클라이언트와 서버 간의 지속적인 연결을 유지할 수 있습니다. Go 언어에서 웹소켓 연결을 유지하는 방법을 알아보겠습니다.

웹소켓 패키지 설치하기

먼저, Go 언어의 내장 웹소켓 패키지를 설치해야 합니다. 아래의 명령어를 사용하여 패키지를 설치할 수 있습니다:

go get github.com/gorilla/websocket

웹소켓 연결 설정하기

웹소켓 연결을 설정하고 유지하기 위해서는 다음과 같은 단계를 따릅니다:

클라이언트의 연결 요청을 받는 웹소켓 핸들러를 작성합니다.
클라이언트의 연결이 수립되면, 연결을 유지하기 위한 루프를 실행합니다.
연결된 클라이언트로부터 메시지를 수신하거나 전송합니다.
클라이언트와의 연결이 끊어졌을 때, 연결을 종료하고 정리합니다.

예제 코드

다음은 간단한 예제 코드로, 클라이언트와의 웹소켓 연결을 유지하는 방법을 보여줍니다:

package main

import (
	"log"
	"net/http"

	"github.com/gorilla/websocket"
)

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

func websocketHandler(w http.ResponseWriter, r *http.Request) {
	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		log.Println("웹소켓 연결 설정 중 에러:", err)
		return
	}
	defer conn.Close()

	for {
		_, msg, err := conn.ReadMessage()
		if err != nil {
			log.Println("메시지 수신 중 에러:", err)
			break
		}
		log.Printf("수신한 메시지: %s", msg)

		err = conn.WriteMessage(websocket.TextMessage, msg)
		if err != nil {
			log.Println("메시지 전송 중 에러:", err)
			break
		}
	}
}

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

이 예제는 /ws 엔드포인트를 통해 웹소켓 연결을 수립하고 유지합니다.

마무리

Go 언어를 이용하여 웹소켓 연결을 유지하는 방법에 대해 알아보았습니다. 웹소켓은 실시간 통신이 필요한 애플리케이션 개발에 유용한 기술입니다. Go 언어의 강력한 동시성 모델과 함께 웹소켓을 사용하여 실시간 어플리케이션을 구축할 수 있습니다.

참고 자료: