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

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

前端代码

<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

package mainimport ("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 errorfor {var reply stringerr = 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}}}