go web socket

最新推荐文章于 2024-07-29 09:53:41 发布

harryhare

最新推荐文章于 2024-07-29 09:53:41 发布

阅读量292

点赞数

分类专栏： golang

本文链接：https://blog.csdn.net/harryhare/article/details/91361282

版权

golang 专栏收录该内容

32 篇文章 0 订阅

订阅专栏

这篇博客探讨了如何在Go中使用WebSocket进行前端通信，提到了两个可用的库，并提供了相关参考资料，包括GitHub上的示例代码和教程链接。

摘要由CSDN通过智能技术生成

有两个可用库，下面是两个简单的例子
web socket 似乎只是相当于在普通的http 协议中加入了一个升级协议的部分（协议升级后依然使用第一次通信的端口和 tcp 连接似乎）

gorilla/websocket 中显式的体现了这一步
x/net/websocket 中则把这一步封装成 websocket.Handler(handler)

`github.com/gorilla/websocket`

package main

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

func IndexHandler(w http.ResponseWriter, r *http.Request) {

	fmt.Fprintln(w, "hello world")
}
var upgrader = websocket.Upgrader{} // use default options

func webSocketHandler(w http.ResponseWriter, r *http.Request) {
	c, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		log.Print("upgrade:", err)
		return
	}
	defer c.Close()
	for {
		mt, message, err := c.ReadMessage()
		if err != nil {
			log.Println("read:", err)
			break
		}
		log.Printf("recv: %s", message)
		err = c.WriteMessage(mt, message)
		if err != nil {
			log.Println("write:", err)
			break
		}
	}
}

func main() {
	http.HandleFunc("/", IndexHandler)
	http.HandleFunc("/ws", webSocketHandler)
	http.ListenAndServe(":8000", nil)
}

`golang.org/x/net/websocket`

package main

import (
	"fmt"
	"net/http"
	"log"
	"golang.org/x/net/websocket"
)



func main() {
http.Handle("/ws",websocket.Handler(Echo))
	if err:=http.ListenAndServe(":8000",nil);err!=nil{
		log.Fatal(err)
	}


}

func Echo(w *websocket.Conn)  {
	for   {
		var msg string
		if  err:= websocket.Message.Receive(w,&msg);err!=nil{
			fmt.Println("read ",err)
			break
		}
		fmt.Println("reciv： "+msg)

		if err := websocket.Message.Send(w, msg); err != nil {
			fmt.Println("write:",err)
			break
		}
	}
}

前端

<html>
<head>
    <title>好好学习</title>
</head>
<body>
<script type="text/javascript">
    var sock = null;
    // var wsuri = "wss://127.0.0.1:8080"; //本地的地址 是可以改变的哦
     var wsuri = "ws://localhost:8080/shiming"; //本地的地址 是可以改变的哦


    window.onload = function() {
        //可以看到客户端JS，很容易的就通过WebSocket函数建立了一个与服务器的连接sock，当握手成功后，会触发WebScoket对象的onopen事件，告诉客户端连接已经成功建立。客户端一共绑定了四个事件。
        console.log("开始了 onload");

        sock = new WebSocket(wsuri);
        //建立连接后触发
        sock.onopen = function() {
            console.log(" 建立连接后触发 connected to " + wsuri);
        }
        // 关闭连接时候触发
        sock.onclose = function(e) {
            console.log("关闭连接时候触发 connection closed (" + e.code + ")");
        }
        // 收到消息后触发
        sock.onmessage = function(e) {
            console.log("收到消息后触发 message received: " + e.data);
        }
        //发生错误的时候触发
        sock.onerror=function (e) {
            console.log("发生错误时候触发"+wsuri)
        }
    };
     //如果sock被关闭掉了 这里 也会报错的啊
    function send() {
        var msg = document.getElementById('message').value;
        sock.send(msg);
    };
</script>
<h1>GoWebSocketDemo</h1>
<form>
    <p>
        Message: <input id="message" type="text" value="你好啊  shiming 小哥哥  嘿嘿   ">
    </p>
</form>
<button onclick="send();">给服务器发送消息</button>
</body>
</html>