Golang-websocket编程

最新推荐文章于 2024-09-22 01:18:58 发布

Alphathur

最新推荐文章于 2024-09-22 01:18:58 发布

阅读量453

点赞数

分类专栏： Golang 文章标签： go websocket

本文链接：https://blog.csdn.net/mryang125/article/details/114649349

版权

Golang 专栏收录该内容

12 篇文章 1 订阅

订阅专栏

Go实现websocket编程，需要引入外部包gorilla/websocket。

我们启动一个http服务器，指定根路径路由到一个html页面，该页面用来模拟websocket通信的客户端，页面会提供一个按钮触发一段执行websocket通信的js。服务端接收到websocket请求，然后将请求的内容完整地响应给浏览器。

接收websocket请求的服务端：

package main

import (
	"fmt"
	"net/http"

	"github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
	ReadBufferSize:  1024,
	WriteBufferSize: 1024,
}

func main() {
	http.HandleFunc("/echo", func(w http.ResponseWriter, r *http.Request) {
		conn, _ := upgrader.Upgrade(w, r, nil) // error ignored for sake of simplicity

		for {
			// Read message from browser
			msgType, msg, err := conn.ReadMessage()
			if err != nil {
				return
			}

			// Print the message to the console
			fmt.Printf("%s sent: %s\n", conn.RemoteAddr(), string(msg))

			// Write message back to browser
			if err = conn.WriteMessage(msgType, msg); err != nil {
				return
			}
		}
	})

	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		http.ServeFile(w, r, "H:\\go\\main\\websockets.html")
	})

	http.ListenAndServe(":8080", nil)
}

发送websocket请求的客户端：

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>WebSockets</title>
</head>
<body>
<input id="input" type="text" />
<button onclick="send()">Send</button>
<pre id="output"></pre>
<script>
    var input = document.getElementById("input");
    var output = document.getElementById("output");
    var socket = new WebSocket("ws://localhost:8080/echo");

    socket.onopen = function () {
        output.innerHTML += "Status: Connected\n";
    };

    socket.onmessage = function (e) {
        output.innerHTML += "Server: " + e.data + "\n";
    };

    function send() {
        socket.send(input.value);
        input.value = "";
    }
</script>
</body>
</html>