脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → Golang → go WebSocket服务器

如何使用go实现创建WebSocket服务器

 更新时间：2024年11月09日 10:44:10   作者：半桶水专家  
文章介绍了如何使用Go语言和gorilla/websocket库创建一个简单的WebSocket服务器,并实现商品信息的实时广播,感兴趣的朋友一起看看吧

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

安装依赖

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

go get github.com/gorilla/websocket

创建WebSocket服务器

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

1. 导入必要的包

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

2. 设置WebSocket升级器

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

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    // 允许跨域访问
    CheckOrigin: func(r *http.Request) bool {
        return true
    },
}

3. 创建WebSocket处理器

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

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服务器。

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服务器代码：

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 run main.go

测试连接

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

扩展功能

  • 多客户端支持：可以在服务器端维护一个连接列表，以便向所有连接的客户端广播消息。
  • 消息类型：可以定义不同的消息类型，以便客户端根据消息类型做出相应的处理。
  • 错误处理：添加更详细的错误处理和日志记录，以便更好地调试和监控。

到此这篇关于用go实现创建WebSocket服务器的文章就介绍到这了,更多相关go WebSocket服务器内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

相关文章

  • Go处理JSON数据的实现

    Go处理JSON数据的实现

    本文主要介绍了Go处理JSON数据的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2023-02-02
  • go语言make初始化的实现

    go语言make初始化的实现

    Go语言中的make函数用于初始化切片、映射和通道,本文就来介绍一下go语言make初始化的实现,具有一定的参考价值,感兴趣的可以了解一下
    2024-12-12
  • Golang字符串变位词示例详解

    Golang字符串变位词示例详解

    这篇文章主要给大家介绍了关于GoLang字符串变位词的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧。
    2017-10-10
  • 使用Go实现webrtc播放音频的流程步骤

    使用Go实现webrtc播放音频的流程步骤

    WebRTC是一项实时通信技术,允许网络应用或站点,在不需要中间媒介的情况下,建立浏览器之间点对点(Peer-to-Peer)的连接,实现视频流、音频流或普通数据的传输,本文给大家介绍了使用Go实现webrtc播放音频的流程步骤,需要的朋友可以参考下
    2025-07-07
  • GoAdminGroup/go-admin的安装和运行的教程详解

    GoAdminGroup/go-admin的安装和运行的教程详解

    这篇文章主要介绍了GoAdminGroup/go-admin的安装和运行的教程详解,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
    2020-09-09
  • 浅谈go中defer的一个隐藏功能

    浅谈go中defer的一个隐藏功能

    这篇文章主要介绍了浅谈go中defer的一个隐藏功能，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2019-12-12
  • Go条件控制语句详解(if-else、switch和select)

    Go条件控制语句详解(if-else、switch和select)

    条件语句用于检查一个条件是否为真,并根据条件的真假来决定是否执行相应的代码,下面这篇文章主要给大家介绍了关于Go条件控制语句(if-else、switch和select)的相关资料,需要的朋友可以参考下
    2024-03-03
  • Makefile构建Golang项目示例详解

    Makefile构建Golang项目示例详解

    这篇文章主要为大家介绍了Makefile构建Golang项目的过程示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2022-07-07
  • Go gRPC服务端流式RPC教程示例

    Go gRPC服务端流式RPC教程示例

    这篇文章主要为大家介绍了Go gRPC服务端流式RPC教程示例，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2022-06-06
  • Go语言数据结构之二叉树可视化详解

    Go语言数据结构之二叉树可视化详解

    这篇文章主要为大家详细介绍了Go语言数据结构中二叉树可视化的方法详解，文中的示例代码讲解详细，感兴趣的小伙伴可以跟随小编一起学习一下
    2022-09-09

最新评论