go使用gorilla/websocket实现websocket

前言

Gorilla WebSocket 是一个用于 Go 语言的 WebSocket 协议实现,而且是一个优秀的组件库,能快速实现websocket协议

websocket实现

依赖安装

使用以下依赖安装对应的库

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

程序实现

go 复制代码
package main

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

var upgrader = websocket.Upgrader{
    ReadBufferSize:  4196,
    WriteBufferSize: 1124,
    CheckOrigin: func(r *http.Request) bool {

       //if r.Method != "GET" {
       // fmt.Println("method is not GET")
       // return false
       //}
       //if r.URL.Path != "/ws" {
       // fmt.Println("path error")
       // return false
       //}
       return true
    },
}

var m map[string]*websocket.Conn

// ServerHTTP 用于升级协议
func ServerHTTP(w http.ResponseWriter, r *http.Request) {
    // 收到http请求之后升级协议
    conn1, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
       log.Println("Error during connection upgrade:", err)
       return
    }
    userId := r.FormValue("userId")
    log.Println("userId:" + userId)
    defer conn1.Close()
    m[userId] = conn1

    conn := m[userId]

    for {
       // 服务端读取客户端请求
       messageType, message, err := conn.ReadMessage()
       if err != nil {
          log.Println("Error during message reading:", err)
          break
       }
       log.Printf("Received:%s", message)

       // 开启关闭连接监听
       conn.SetCloseHandler(func(code int, text string) error {
          fmt.Println(code, text) // 断开连接时将打印code和text
          return nil
       })

       //服务端给客户端返回请求
       err = conn.WriteMessage(messageType, message)
       conn.WriteMessage(websocket.TextMessage, []byte("aaa"))
       if err != nil {
          log.Println("Error during message writing:", err)
          return
       }

    }
}

func home(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Index Page")
}

func main() {
    //a, _ := strconv.Atoi("1")
    //fmt.Println(a)

    m = make(map[string]*websocket.Conn)

    // 创建HTTP服务器
    http.HandleFunc("/ws", ServerHTTP)
    log.Println("Server started on :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

启动之后

使用postman连接服务端

发送消息

总结

Gorilla WebSocket 提供了一个完整且经过测试的 WebSocket 协议实现,适用于构建高效的 WebSocket 应用程序

相关推荐
JohnYan17 分钟前
工作笔记 - 一种业务信息汇报机制的设计和实现
数据库·后端·postgresql
Goboy17 分钟前
弹窗大战15年祭,零信任iOA的枪口,正对准哪吒的方向盘
安全·程序员·产品
EdenX22 分钟前
MySQL详解:从基础到应用,附电商订单系统实战
后端
error_cn24 分钟前
匿名ftp服务器搭建指南
后端
就是帅我不改27 分钟前
深入实战责任链模式:在企业级审批流程中的优雅应用
后端·面试
武子康27 分钟前
大数据-54 Kafka 安装配置 从 ZooKeeper 到 KRaft 的演进之路
大数据·后端·kafka
星辰大海的精灵29 分钟前
深入解析 CopyOnWriteArrayList
java·后端·算法
云边散步31 分钟前
《校园生活平台从 0 到 1 的搭建》第三篇:后端的微信授权登录
前端·后端
hrrrrb42 分钟前
【Spring Boot 快速入门】二、请求与响应
spring boot·后端
程序员海军44 分钟前
使用 Kiro AI IDE 3小时实现全栈应用Admin系统
前端·后端·aigc