Go에서 WebSocket을 어떻게 사용하나요?

최근 몇 년 동안 WebSocket 기술은 웹 개발에 없어서는 안될 부분이 되었습니다. WebSocket은 단일 TCP 연결을 통한 전이중 통신을 위한 프로토콜로, 클라이언트와 서버 간의 통신을 보다 원활하고 효율적으로 만듭니다. 오늘날 많은 최신 웹 애플리케이션은 실시간 채팅, 온라인 게임, 실시간 데이터 시각화 등 WebSocket 기술을 사용합니다.

현대 프로그래밍 언어인 Go 언어는 자연스럽게 WebSocket 기술에 대한 우수한 지원을 제공합니다. 이 글에서는 Go에서 WebSocket 기술을 사용하는 방법에 대해 자세히 살펴보겠습니다.

Go 언어로 WebSocket 구현

WebSocket 프로토콜은 HTTP에서 업그레이드된 프로토콜이므로 Go 언어로 WebSocket을 사용하려면 이를 HTTP 프로토콜과 결합해야 합니다. Go 언어에는 다양한 HTTP 관련 기능을 제공하는 net/http 패키지가 내장되어 있습니다. Go에서 WebSocket을 사용하려면 net/http 및 github.com/gorilla/websocket 라이브러리만 가져오면 됩니다. net/http包，该包提供了各种HTTP相关的功能。如果我们想要在Go中使用WebSocket，只需导入net/http和github.com/gorilla/websocket库即可。

下面，我们就来看一下如何使用github.com/gorilla/websocket库来实现WebSocket。

创建WebSocket服务端

在Go语言中，创建WebSocket服务端的方式也比较简单。我们直接看下列代码，它展示了如何创建一个HTTP服务器，等待客户端的WebSocket连接：

package main

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

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

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

func handleWebSocket(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)
            return
        }
        log.Printf("Received message: %s
", message)

        err = conn.WriteMessage(messageType, message)
        if err != nil {
            log.Println(err)
            return
        }
        log.Printf("Sent message: %s
", message)
    }
}

在上面的代码中，我们首先创建了一个websocket.Upgrader对象，它负责将传入的HTTP连接升级为WebSocket连接。然后，我们创建了一个handleWebSocket函数，它会在客户端进行连接时被调用。该函数会先将传入的HTTP连接升级为WebSocket连接，然后进入循环，不断地接收来自客户端的信息，并将接收到的信息回传给客户端。其中，conn.ReadMessage()和conn.WriteMessage()分别用于读取和写入信息。

最后，我们将该handleWebSocket函数绑定到/ws路由上，并启动了HTTP服务器，以便客户端可以连接到该WebSocket服务器。如果一切正常，当我们启动上述代码后，您会看到类似如下输出：

2021/10/13 15:20:06 http: Server started at http://127.0.0.1:8080

这意味着您已经成功地创建了一个WebSocket服务器，客户端可以通过ws://localhost:8080/ws地址来连接该服务器。

创建WebSocket客户端

与创建WebSocket服务端类似，我们也可以很容易地在Go语言中创建一个WebSocket客户端。我们来看下面的代码，它描述了如何创建一个与WebSocket服务器建立连接的客户端：

package main

import (
    "fmt"
    "log"
    "net/url"
    "os"
    "os/signal"
    "syscall"
    "github.com/gorilla/websocket"
)

func main() {
    interrupt := make(chan os.Signal, 1)
    signal.Notify(interrupt, os.Interrupt, syscall.SIGTERM)

    u := url.URL{Scheme: "ws", Host: "localhost:8080", Path: "/ws"}
    log.Printf("Connecting to %s", u.String())

    conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
    if err != nil {
        log.Fatal("dial:", err)
    }
    defer conn.Close()

    done := make(chan struct{})
    go func() {
        defer close(done)
        for {
            messageType, message, err := conn.ReadMessage()
            if err != nil {
                log.Println("read:", err)
                return
            }
            log.Printf("Received message: %s
", message)
            if err := conn.WriteMessage(messageType, message); err != nil {
                log.Println("write:", err)
                return
            }
            log.Printf("Sent message: %s
", message)
        }
    }()

    for {
        select {
        case <-done:
            return
        case <-interrupt:
            log.Println("interrupt")

            if err := conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "")); err != nil {
                log.Println("write close:", err)
                return
            }
            select {
            case <-done:
            case <-time.After(time.Second):
            }
            return
        }
    }
}

在上面的代码中，我们使用websocket.Dial方法来与WebSocket服务器建立连接。建立连接后，我们再进入一个无限循环，不断地发送和接收来自WebSocket服务器的信息。其中，conn.ReadMessage()和conn.WriteMessage()用于读取和写入信息。

值得注意的是，在上述代码中，我们使用了channels来等待操作完成。如果接收到interrupt信号，我们将使用websocket.CloseMessage消息发送给服务器，以关闭WebSocket连接。

结论

在本文中，我们学习了如何在Go语言中使用github.com/gorilla/websocket

이제 github.com/gorilla/websocket 라이브러리를 사용하여 WebSocket을 구현하는 방법을 살펴보겠습니다.

WebSocket 서버 만들기🎜🎜Go 언어에서 WebSocket 서버를 만드는 방법은 비교적 간단합니다. HTTP 서버를 생성하고 클라이언트의 WebSocket 연결을 기다리는 방법을 보여주는 다음 코드를 직접 살펴보겠습니다. 🎜rrreee🎜위 코드에서는 먼저 websocket.Upgrader 객체를 생성합니다. 들어오는 HTTP 연결을 WebSocket 연결로 업그레이드하는 작업을 담당합니다. 그런 다음 클라이언트가 연결될 때 호출될 handleWebSocket 함수를 만듭니다. 이 함수는 먼저 들어오는 HTTP 연결을 WebSocket 연결로 업그레이드한 다음 루프에 들어가 클라이언트로부터 지속적으로 정보를 수신하고 수신된 정보를 클라이언트에 다시 전달합니다. 그 중 conn.ReadMessage()와 conn.WriteMessage()는 각각 정보를 읽고 쓰는 데 사용됩니다. 🎜🎜마지막으로 handleWebSocket 함수를 /ws 경로에 바인딩하고 클라이언트가 WebSocket 서버에 연결할 수 있도록 HTTP 서버를 시작했습니다. 모든 것이 순조롭게 진행되면 위 코드를 시작하면 다음과 유사한 출력이 표시됩니다. 🎜rrreee🎜 이는 WebSocket 서버가 성공적으로 생성되었으며 클라이언트가 ws://localhost를 통해 해당 서버에 액세스할 수 있음을 의미합니다. 8080/ws서버에 연결하기 위한 주소입니다. 🎜🎜Creating a WebSocket client🎜🎜WebSocket 서버를 만드는 것과 유사하게 Go 언어로 WebSocket 클라이언트를 쉽게 만들 수도 있습니다. WebSocket 서버와 연결을 설정하는 클라이언트를 생성하는 방법을 설명하는 다음 코드를 살펴보겠습니다. 🎜rrreee🎜 위 코드에서는 websocket.Dial 메서드를 사용하여 연결을 설정합니다. WebSocket 서버. 연결이 설정된 후에는 WebSocket 서버에서 지속적으로 정보를 주고받으며 무한 루프에 들어갑니다. 그 중 conn.ReadMessage()와 conn.WriteMessage()는 정보를 읽고 쓰는 데 사용됩니다. 🎜🎜위 코드에서는 채널을 사용하여 작업이 완료될 때까지 기다립니다. interrupt 신호가 수신되면 websocket.CloseMessage 메시지를 사용하여 서버에 전송하여 WebSocket 연결을 닫습니다. 🎜🎜결론🎜🎜이 기사에서는 github.com/gorilla/websocket 라이브러리를 사용하여 Go 언어로 WebSocket을 구현하는 방법을 배웠습니다. WebSocket 기술을 사용하면 웹 애플리케이션에 더 많은 역동성과 상호 작용을 추가하여 애플리케이션을 더욱 현대적이고 사용자 친화적으로 만들 수 있습니다. 🎜🎜WebSocket 기술과 Go 언어에서 이를 사용하는 방법에 대해 질문이나 의견이 있는 경우 아래 메시지를 남겨서 의견을 공유해 주세요. 🎜

위 내용은 Go에서 WebSocket을 어떻게 사용하나요?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!