用go实现创建WebSocket服务器

使用Go语言创建WebSocket服务器可以利用现有的库来简化开发过程。gorilla/websocket 是一个非常流行且功能强大的库,适用于Go语言的WebSocket应用。下面是一个详细的步骤指南,介绍如何使用 gorilla/websocket 创建一个WebSocket服务器,并实现商品信息的实时广播。

安装依赖

首先,你需要安装 gorilla/websocket 库。可以通过以下命令安装:

Go 复制代码
go get github.com/gorilla/websocket

创建WebSocket服务器

接下来,我们将创建一个简单的WebSocket服务器,该服务器可以接收客户端连接,并向所有连接的客户端广播商品信息。

1. 导入必要的包
Go 复制代码
package main

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

    "github.com/gorilla/websocket"
)
2. 设置WebSocket升级器

WebSocket连接需要通过HTTP握手来建立。gorilla/websocket 提供了一个 Upgrader 结构体来处理这个过程。

Go 复制代码
var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    // 允许跨域访问
    CheckOrigin: func(r *http.Request) bool {
        return true
    },
}
3. 创建WebSocket处理器

定义一个函数来处理WebSocket连接。在这个函数中,我们将读取消息并广播商品信息。

Go 复制代码
func serveWs(w http.ResponseWriter, r *http.Request) {
    // 升级HTTP连接到WebSocket连接
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Failed to set websocket upgrade:", err)
        return
    }
    defer conn.Close()

    // 模拟商品添加事件
    for {
        item := map[string]interface{}{
            "id":    1,
            "name":  "Apple",
            "price": 2.5,
        }

        // 将商品信息发送给客户端
        err := conn.WriteJSON(item)
        if err != nil {
            log.Println("Error writing message:", err)
            break
        }

        // 每隔5秒发送一次
        time.Sleep(5 * time.Second)
    }
}
4. 注册路由并启动HTTP服务器

最后,我们需要注册路由并启动HTTP服务器。

Go 复制代码
func main() {
    http.HandleFunc("/ws", serveWs)

    fmt.Println("Starting server on :8080")
    if err := http.ListenAndServe(":8080", nil); err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

完整代码

将上述代码片段组合在一起,得到完整的WebSocket服务器代码:

Go 复制代码
package main

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

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    CheckOrigin: func(r *http.Request) bool {
        return true
    },
}

func serveWs(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Failed to set websocket upgrade:", err)
        return
    }
    defer conn.Close()

    for {
        item := map[string]interface{}{
            "id":    1,
            "name":  "Apple",
            "price": 2.5,
        }

        err := conn.WriteJSON(item)
        if err != nil {
            log.Println("Error writing message:", err)
            break
        }

        time.Sleep(5 * time.Second)
    }
}

func main() {
    http.HandleFunc("/ws", serveWs)

    fmt.Println("Starting server on :8080")
    if err := http.ListenAndServe(":8080", nil); err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

运行服务器

保存文件并运行:

Go 复制代码
go run main.go

测试连接

你可以使用任何WebSocket客户端工具(如浏览器的开发者工具)来测试连接。连接到 ws://localhost:8080/ws,你应该每5秒接收到一次商品信息。

扩展功能

  • 多客户端支持:可以在服务器端维护一个连接列表,以便向所有连接的客户端广播消息。
  • 消息类型:可以定义不同的消息类型,以便客户端根据消息类型做出相应的处理。
  • 错误处理:添加更详细的错误处理和日志记录,以便更好地调试和监控。
相关推荐
九河云9 分钟前
华为云国内版与国际版的差异
大数据·服务器·华为云
yaoxin52112337 分钟前
第三十二章 UDP 客户端 服务器通信
服务器·网络协议·udp
LightOfNight2 小时前
【设计模式】创建型模式之单例模式(饿汉式 懒汉式 Golang实现)
单例模式·设计模式·golang
筱小虾米3 小时前
解决SSL VPN客户端一直提示无法连接服务器的问题
运维·服务器
MelonTe7 小时前
Golang网络模型netpoll源码解析
golang
wanhengwangluo7 小时前
裸金属服务器能够帮助企业解决哪些问题?
运维·服务器
titxixYY8 小时前
SElinux
linux·运维·服务器
聚名网9 小时前
手机无法连接服务器1302什么意思?
运维·服务器·智能手机
代码欢乐豆9 小时前
软件工程第13章小测
服务器·前端·数据库·软件工程
Clown9510 小时前
go-zero(十) 数据缓存和Redis使用
redis·缓存·golang