【go语言实现一个webSocket的一个demo】

岁月的眸2023-11-26 23:40

go语言实现一个webSocket的一个demo

前端代码

html 复制代码

<html lang="zh-CN">

<head></head>

<body>
<script type="text/javascript">
    // header('Access-Control-Allow-Origin:*');
    var sock = null;
    var wsuri = "ws://127.0.0.1:9999";

    window.onload = function () {

        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);
        }
    };

    function send() {
        var msg = document.getElementById('message').value;
        sock.send(msg);
    };
</script>
<h1>WebSocket Echo Test</h1>
<form>
    <p>
        Message: <input id="message" type="text" value="Hello, world!">
    </p>
</form>
<button onclick="send();">Send Message</button>
</body>

</html>

服务端

WebSocketServer.go

go 复制代码

package main

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

func main() {
	http.Handle("/", websocket.Handler(Echo)) //这里校验请求头的Origin字段
	if err := http.ListenAndServe("127.0.0.1:9999", nil); err != nil {
		log.Fatal("ListenAndServer: ", err)
	}
}

func Echo(ws *websocket.Conn) {
	var err error
	for {
		var reply string
		err = websocket.Message.Receive(ws, &reply)
		if err != nil {
			fmt.Println("Can't receive data...")
			break
		}
		fmt.Println("Receive back from client:" + reply)
		msg := "Receive: " + "接受到客户端的消息，ok"
		fmt.Println("Sending to client: " + msg)

		err = websocket.Message.Send(ws, msg)
		if err != nil {
			fmt.Println("发送消息失败")
			break
		}

	}

}