Go 언어를 사용하여 WebSocket 서버를 구축하는 방법

WebSockets는 단일 TCP 연결을 통해 클라이언트와 서버 간의 양방향 통신을 허용하는 고급 네트워크 프로토콜입니다. WebSocket은 실시간 데이터 전송, 온라인 게임, 채팅방 및 기타 애플리케이션을 지원하므로 개발자들 사이에서 매우 인기가 있습니다.

Go 언어는 WebSocket 서버 구축에 적합한 고성능의 간결한 프로그래밍 언어입니다. 이번 글에서는 Go 언어를 사용하여 WebSocket 서버를 구축하는 방법을 자세히 소개하겠습니다.

1. WebSocket 프로토콜 이해

WebSocket 프로토콜은 HTTP 프로토콜의 핸드셰이크 및 업그레이드 헤더를 사용하여 WebSocket 연결로 업그레이드합니다. WebSocket 연결은 항상 캐리지 리턴 및 줄 바꿈 문자를 메시지 종결자로 사용합니다. WebSocket 메시지는 텍스트 또는 바이너리 데이터일 수 있습니다.

2. Gorilla WebSocket 라이브러리 설치

Gorilla WebSocket 라이브러리는 Go 언어에서 가장 널리 사용되는 WebSocket 라이브러리 중 하나이며 WebSocket 클라이언트 및 서버 구현을 제공합니다. 다음 명령을 사용하여 Gorilla WebSocket 라이브러리를 설치할 수 있습니다.

go get github.com/gorilla/websocket

3. WebSocket 서버 만들기

다음은 간단한 WebSocket 서버 예입니다.

package main

import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
)

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

func main() {
    http.HandleFunc("/echo", echoHandler)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

func echoHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()

    for {
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            break
        }
        log.Printf("Received message: %s
", message)
        err = conn.WriteMessage(messageType, message)
        if err != nil {
            log.Println(err)
            break
        }
    }
}

위 코드에서 echoHandler 함수는 모든 WebSocket 연결을 처리합니다. Gorilla WebSocket 라이브러리에서 제공하는 업그레이더를 사용하여 HTTP 요청을 WebSocket 연결로 업그레이드합니다. 연결이 업그레이드된 후 서버는 conn.ReadMessage()를 사용하여 클라이언트에서 메시지를 읽고 conn.WriteMessage()를 사용하여 클라이언트에 메시지를 다시 보냅니다.

4. WebSocket 서버 실행

이제 WebSocket 서버를 실행해 보세요. 터미널에서 Go 파일이 있는 디렉터리로 이동하여 다음 명령을 실행합니다.

go run main.go

이 시점에서 다음과 유사한 출력이 표시됩니다.

2021/08/11 15:32:31 Listening on port 8080...

이는 서버가 포트 8080에서 수신 대기 중임을 의미합니다. 들어오는 연결.

5. WebSocket 서버 테스트

이제 WebSocket 서버를 성공적으로 생성했으므로 WebSocket 클라이언트를 사용하여 테스트할 수 있습니다. Chrome 브라우저에는 WebSocket 클라이언트가 내장되어 있으므로 테스트에는 Chrome 브라우저를 사용하는 것이 좋습니다. Chrome에서 개발자 도구를 열고 콘솔 탭으로 전환한 후 다음 코드를 입력하세요.

var ws = new WebSocket("ws://localhost:8080/echo");
ws.onopen = function() {
    console.log("Connected");
    ws.send("Hello, World!");
};
ws.onmessage = function(event) {
    console.log("Received message: " + event.data);
};
ws.onclose = function() {
    console.log("Disconnected");
};

Chrome이 서버에 WebSocket 연결을 설정하고 "Hello, World!" 메시지를 보냅니다. 서버는 메시지를 에코하여 클라이언트에 다시 보냅니다. 콘솔에 다음과 유사한 출력이 표시되어야 합니다.

Connected
Received message: Hello, World!
Disconnected

6. 요약

이 문서에서는 Gorilla WebSocket 라이브러리와 Go 언어를 사용하여 WebSocket 서버를 구축하는 방법을 설명합니다. 이제 WebSocket 프로토콜을 사용하여 실시간 통신 기능을 구현하는 방법을 알았습니다. WebSocket을 사용하면 각 요청에 대해 새 연결을 설정하지 않고도 단일 TCP 연결을 사용하여 데이터를 전송할 수 있습니다. 이는 성능과 확장성을 향상시켜 애플리케이션에서 사용하기에 가치가 있습니다.

위 내용은 Go 언어를 사용하여 WebSocket 서버를 구축하는 방법의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!