golang搭建websocket

Golang是一种快速编译、高效、安全的编程语言，在Web开发方面也有着广泛的应用。其中，使用Golang搭建Websocket是一个非常流行的应用场景。本文将介绍如何使用Golang搭建一个Websocket应用程序。

一、什么是Websocket

Websocket是一种协议，它允许在浏览器和服务器之间建立双向通信的连接。传统的HTTP协议只能由浏览器向服务器发出请求，并接收服务器响应。但是，使用Websocket可以让服务器主动向浏览器发送消息，实现了服务器与浏览器的全双工通信。

二、Golang如何实现Websocket

Golang自带了一个标准库net/http，这个包提供了一个http.Server结构体，它可以用来构建Web服务器。除此之外，net/http也提供了一个websocket子包，它为我们提供了构建WebSocket服务器所需的全部功能。

建立WebSocket连接

要建立WebSocket连接，客户端需要向服务器发送一个HTTP请求，并携带Upgrade头部，指示请求升级为WebSocket协议。服务器端在收到此请求后，需要检查Upgrade头部，并根据是否为WebSocket请求来进行相应的处理。

服务器端可以使用websocket.IsWebSocketUpgrade函数来检查请求是否为WebSocket请求，并使用websocket.Upgrade函数升级请求。升级成功后，服务器就可以通过*websocket.Conn类型的连接对象与客户端进行通信了。

func handler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()
    // 使用conn进行通信
}

登录后复制

其中，upgrader是一个websocket.Upgrader类型的变量，通过设置它的属性可以控制WebSocket连接的处理方式。Upgrade函数在升级请求时会阻塞，因此要将其放在goroutine中运行。

发送和接收WebSocket消息

一旦WebSocket连接建立成功，客户端和服务器都可以通过*websocket.Conn对象进行通信。使用这个对象可以发送和接收WebSocket消息。

发送消息很简单，只需要调用WriteMessage方法即可：

err := conn.WriteMessage(websocket.TextMessage, []byte("hello"))

登录后复制

接收消息的方法有三种：

ReadMessage：阻塞读取，直到有消息到来
NextReader：阻塞等待下一个消息到来
NextWriter：返回一个io.WriteCloser对象，可以使用这个对象发送消息

一般使用ReadMessage方法接收WebSocket消息：

_, message, err := conn.ReadMessage()
if err != nil {
    log.Println(err)
    return
}
log.Printf("recv: %s", message)

登录后复制

三、一个简单的WebSocket应用

在这个例子中，我们将创建一个简单的聊天室应用，客户端发送消息之后，服务器将消息广播给所有连接的客户端。

客户端代码：（使用JavaScript）

<!DOCTYPE html>
<html>
<head>
    <title>Chatroom</title>
</head>
<body>
    <input type="text" id="input-message"/>
    <button onclick="sendMessage()">发送</button>
    <ul id="message-list"></ul>

    <script>
        var ws = new WebSocket("ws://localhost:8080/ws");
        ws.onmessage = function(event) {
            var message = event.data;
            var li = document.createElement("li");
            li.appendChild(document.createTextNode(message));
            document.getElementById("message-list").appendChild(li);
        };

        function sendMessage() {
            var message = document.getElementById("input-message").value;
            ws.send(message);
        }
    </script>
</body>
</html>

登录后复制

服务器端代码：

package main

import (
    "fmt"
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{}

func main() {
    http.HandleFunc("/ws", wsHandler)
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

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

    for {
        _, message, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            break
        }
        fmt.Printf("recv: %s
", message)

        // 广播给所有客户端
        for _, conn := range conns {
            err := conn.WriteMessage(websocket.TextMessage, message)
            if err != nil {
                log.Println(err)
                break
            }
        }
    }
}

登录后复制

在上面的代码中，我们将所有连接的websocket.Conn对象存储在conns变量中。每当有一个新消息到来时，服务器会将消息广播给所有客户端。

总结

使用Golang搭建WebSocket应用程序非常简单，只需要导入net/http和websocket包即可，直接使用标准库提供的API即可完成WebSocket应用的开发。希望此文章能够对Golang Websocket开发有所帮助。

以上就是golang搭建websocket的详细内容，更多请关注Work网其它相关文章！