>  기사  >  백엔드 개발  >  golang을 사용하여 WebSocket 서버를 만드는 방법

golang을 사용하여 WebSocket 서버를 만드는 방법

WBOY
WBOY원래의
2023-12-17 22:32:22960검색

golang을 사용하여 WebSocket 서버를 만드는 방법

golang을 사용하여 WebSocket 서버를 만드는 방법

WebSocket은 클라이언트와 서버 간의 양방향 통신을 가능하게 하는 프로토콜입니다. 이를 통해 지속적인 연결을 통해 클라이언트와 서버 간에 데이터를 전송할 수 있습니다. 이번 글에서는 golang을 사용하여 간단한 WebSocket 서버를 생성하는 방법을 소개하고 구체적인 코드 예시를 제공하겠습니다.

  1. 준비
    시작하기 전에 golang이 올바르게 설치되었는지 확인해야 합니다. 공식 웹사이트에서 설치 패키지를 다운로드하고 지침에 따라 설치할 수 있습니다.
  2. 종속성 가져오기
    net/httpgithub.com/gorilla/websocket 두 가지 종속성 패키지를 가져와야 합니다. net/http는 HTTP 서버를 생성하는 데 사용되고 gorilla/websocket은 WebSocket 연결을 처리하는 데 사용됩니다. net/httpgithub.com/gorilla/websocket这两个依赖包。net/http用于创建HTTP服务器,gorilla/websocket用于处理WebSocket连接。
import (
    "net/http"
    "github.com/gorilla/websocket"
)
  1. 创建WebSocket服务器
    我们使用http.HandleFunc()函数来处理WebSocket连接请求,并使用websocket.Upgrade()函数将HTTP协议升级为WebSocket协议。
func wsHandler(w http.ResponseWriter, r *http.Request) {
    upgrader := websocket.Upgrader{
        CheckOrigin: func(r *http.Request) bool { return true },
    }

    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Upgrade error:", err)
        return
    }

    // 在这里处理WebSocket连接
}

wsHandler()函数中,我们创建了一个Upgrader对象,并设置了CheckOrigin属性为一个匿名函数,返回true表示接受所有的跨域请求。

接下来,我们使用Upgrade()函数将HTTP协议升级为WebSocket协议,如果升级失败则打印错误信息并返回。如果升级成功,我们可以使用conn对象来处理WebSocket连接。

  1. 处理WebSocket连接
    在WebSocket连接建立后,我们可以使用conn对象进行读写操作。下面是一个简单的处理示例:
for {
    messageType, message, err := conn.ReadMessage()
    if err != nil {
        log.Println("Read error:", err)
        break
    }

    log.Println("received message:", string(message))

    err = conn.WriteMessage(messageType, message)
    if err != nil {
        log.Println("Write error:", err)
        break
    }
}

在上面的示例中,我们使用ReadMessage()函数读取客户端发送的消息,并使用WriteMessage()函数将消息回传给客户端。如果读取或写入过程中出现错误,我们打印错误信息并跳出循环。

  1. 启动WebSocket服务器
    最后,我们使用http.HandleFunc()函数将wsHandler()函数注册为处理WebSocket连接的处理器,并使用http.ListenAndServe()函数在指定端口上启动WebSocket服务器。
func main() {
    http.HandleFunc("/ws", wsHandler)
    log.Println("WebSocket server started.")

    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal("ListenAndServe error:", err)
    }
}

在上面的示例中,我们将wsHandler()

rrreee
      WebSocket 서버 만들기
    1. http.HandleFunc() 함수를 사용하여 WebSocket 연결 요청을 처리하고 websocket.Upgrade()함수는 HTTP 프로토콜을 WebSocket 프로토콜로 업그레이드합니다. <br>
    2. rrreee
    wsHandler() 함수에서 Upgrader 객체를 생성하고 CheckOrigin 속성을 ​​익명 함수로 설정하여 true를 반환한다는 것은 의미합니다. 모든 도메인 간 요청을 수락합니다.


    다음으로 Upgrade() 함수를 사용하여 HTTP 프로토콜을 WebSocket 프로토콜로 업그레이드합니다. 업그레이드에 실패하면 오류 메시지가 인쇄되어 반환됩니다. 업그레이드가 성공하면 conn 개체를 사용하여 WebSocket 연결을 처리할 수 있습니다.

      🎜WebSocket 연결 처리🎜WebSocket 연결이 설정된 후 읽기 및 쓰기 작업에 conn 개체를 사용할 수 있습니다. 다음은 간단한 처리 예시입니다. 🎜🎜rrreee🎜위 예시에서는 ReadMessage() 함수를 사용하여 클라이언트가 보낸 메시지를 읽고, WriteMessage()를 사용합니다. >함수는 클라이언트에게 메시지를 다시 보냅니다. 읽거나 쓰는 동안 오류가 발생하면 오류 메시지를 인쇄하고 루프에서 벗어납니다. 🎜
        🎜WebSocket 서버 시작🎜마지막으로 http.HandleFunc() 함수를 사용하여 wsHandler() 함수를 WebSocket 처리로 등록합니다. 연결 서버에 연결하고 http.ListenAndServe() 함수를 사용하여 지정된 포트에서 WebSocket 서버를 시작합니다. 🎜🎜rrreee🎜위의 예에서는 wsHandler() 함수를 "/ws" 경로에 대한 핸들러로 등록했습니다. 필요에 따라 경로를 사용자 정의할 수 있습니다. 🎜🎜🎜클라이언트 테스트🎜 이제 WebSocket 서버 생성이 완료되었으므로 WebSocket 프로토콜을 지원하는 모든 클라이언트를 사용하여 서버와 통신할 수 있습니다. 브라우저의 개발자 도구를 사용하거나 일부 WebSocket 테스트 도구를 사용하여 이를 테스트할 수 있습니다. 🎜🎜🎜요약🎜이 글에서는 golang을 사용하여 간단한 WebSocket 서버를 만드는 방법을 소개하고 구체적인 코드 예제를 제공합니다. golang을 사용하여 WebSocket 서버를 만드는 것은 매우 간단하며 양방향 통신을 쉽게 구현할 수 있습니다. 이 글이 WebSocket을 이해하고 사용하는 데 도움이 되기를 바랍니다. 🎜

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

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.