千家信息网

golang如何实现websocket聊天室功能

发表于:2024-11-23 作者:千家信息网编辑
千家信息网最后更新 2024年11月23日,这篇文章将为大家详细讲解有关golang如何实现websocket聊天室功能,小编觉得挺实用的,因此分享给大家做个参考,希望大家阅读完这篇文章后可以有所收获。基本原理:1.引入了 golang.org
千家信息网最后更新 2024年11月23日golang如何实现websocket聊天室功能

这篇文章将为大家详细讲解有关golang如何实现websocket聊天室功能,小编觉得挺实用的,因此分享给大家做个参考,希望大家阅读完这篇文章后可以有所收获。

基本原理:
1.引入了 golang.org/x/net/websocket 包。
2.监听端口。
3.客户端连接时,发送结构体: {"type":"login","uid":"我是用户名","msg":"登陆成功"}' .服务端根据login信息,维护一个map,用来存放不同用户的连接体。
4.有用户发言时,将msg内容轮询发给给一个用户。
5.客户端使用js websocket功能,当作客户端,开始聊天。
不足之处:只是根据客户端传的uid来区分用户,没有进行身份校验,不过对一个demo来说,也没必要。

server代码:

package main import (        "encoding/json"        "fmt"        "golang.org/x/net/websocket"        "log"        "net/http") //ProxyConfig 配置type MsgConfig struct {        Type string `json:"type,omitempty"`        Uid  string `json:"uid,omitempty"`        Msg  string `json:"msg,omitempty"`} var connMap = make(map[string]*websocket.Conn) func Echo(ws *websocket.Conn) {        var err error        for {                var reply string                 if err = websocket.Message.Receive(ws, &reply); err != nil {                        fmt.Println("Can't receive")                        break                }                replyMsg := MsgConfig{}                json.Unmarshal([]byte(reply), &replyMsg)                 if replyMsg.Type == "login" && replyMsg.Uid != "" {                        connMap[replyMsg.Uid] = ws                        fmt.Println(connMap)                }                for k, v := range connMap {                        go sendMessage(replyMsg, v, k)                }         }} func sendMessage(replyMsg MsgConfig, conn *websocket.Conn, connUid string) {        msg := replyMsg.Uid + "说:" + replyMsg.Msg        if connUid == replyMsg.Uid {                fmt.Println(msg)                if replyMsg.Type == "login" {                        msg = "登陆成功"                } else {                        msg = "你说:" + replyMsg.Msg                }         }        if err := websocket.Message.Send(conn, msg); err != nil {                fmt.Println("Can't send")        } }func main() {         http.Handle("/", websocket.Handler(Echo))         if err := http.ListenAndServe(":1234", nil); err != nil {                log.Fatal("ListenAndServe:", err)        }}

客户端聊天:
可以在浏览器的控制台里面输入以下代码,进行聊天:
//ip:port设置成自己的

ws = new WebSocket("ws://127.0.0.1:1234");ws.onopen = function (ev) {        ws.send('{"type":"login","uid":"我是用户名","msg":"我登陆了"}');    };ws.onmessage = function (ev) {        if (ev.data){            console.log(ev.data)        }};//下面这句可以多次发送,当作是聊天。ws.send('{"type":"msg","uid":"我是用户名","msg":"你好啊"}');

客户端效果图:

服务器效果图:

关于"golang如何实现websocket聊天室功能"这篇文章就分享到这里了,希望以上内容可以对大家有一定的帮助,使各位可以学到更多知识,如果觉得文章不错,请把它分享出去让更多的人看到。

0