欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

[Go] 实现websocket服务端

程序员文章站 2023-08-30 15:44:07
直接使用官方子包可以实现websocket协议, golang.org/x/net/websocket 如果在这个目录没有这个包就去github下载net库,放进这个目录$GOPATH/src/golang.org/x/net就行了 服务端代码: package main import ( "tim ......

直接使用官方子包可以实现websocket协议,

golang.org/x/net/websocket
如果在这个目录没有这个包就去github下载net库,放进这个目录$gopath/src/golang.org/x/net就行了
服务端代码:
package main

import (
    "time"
    "log"
    "net/http"
    "golang.org/x/net/websocket"
)
func main(){
    http.handlefunc("/",index)
    http.handle("/websocket",websocket.handler(instantmessage))
    log.println("start server")
    err:=http.listenandserve(":1234",nil)
    log.println(err)
}
func index(res http.responsewriter,req *http.request){
    res.write([]byte("hello world"))
}
func instantmessage(w *websocket.conn)  {
    var error error
    for {
        var reply string
        if  error= websocket.message.receive(w,&reply);error!=nil{
            log.println("接受消息失败",error)
            break
        }
        log.println("客户端:",reply)
        
        msg:="服务端:"+reply+time.now().string()
        log.println(msg)
        if error = websocket.message.send(w, msg); error != nil {
            log.println("发送消息失败")
            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:1234/websocket"; //本地的地址 是可以改变的哦


    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);
            document.getelementbyid('receive').innerhtml=e.data;
        }
        //发生错误的时候触发
        sock.onerror=function (e) {
            console.log("发生错误时候触发"+wsuri)
        }
    };
     //如果sock被关闭掉了 这里 也会报错的啊
    function send() {
        var msg = document.getelementbyid('message').value;
        sock.send(msg);
    };
</script>
<h1>客户端</h1>
<form>
    <p>发送: </p>
    <textarea id="message" >我是陶士涵</textarea>
    <p>接收: </p>
    <textarea id="receive" ></textarea>
</form>
<button onclick="send();">给服务器发送消息</button>
</body>
</html>

[Go] 实现websocket服务端