>  기사  >  웹 프론트엔드  >  HTML5는 go+websocket을 사용하여 websocket 서비스 예제를 구축합니다.

HTML5는 go+websocket을 사용하여 websocket 서비스 예제를 구축합니다.

黄舟
黄舟원래의
2017-10-31 10:39:572752검색

이번에는 코드 스크린샷만 게시하겠습니다. go 언어를 사용하여 백그라운드에서 간단한 채팅을 해야 합니다. 여기서는 효과를 얻는 가장 간단한 방법입니다. h5와 함께 websocket을 사용하고 websocket 서비스 구축의 주요 부분으로 이동하십시오.

go 코드 부분:

// WebChat project main.go
package main

import (
    "fmt"
    "net/http"
    "time"

    "encoding/json"

    "strings"

    "golang.org/x/net/websocket"
)

//全局信息
var datas Datas
var users map[*websocket.Conn]string

func main() {
    fmt.Println("启动时间")
    fmt.Println(time.Now())

    //初始化
    datas = Datas{}
    users = make(map[*websocket.Conn]string)

    //绑定效果页面
    http.HandleFunc("/", h_index)
    //绑定socket方法
    http.Handle("/webSocket", websocket.Handler(h_webSocket))
    //开始监听
    http.ListenAndServe(":8", nil)
}

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

    http.ServeFile(w, r, "index.html")
}

func h_webSocket(ws *websocket.Conn) {

    var userMsg UserMsg
    var data string
    for {

        //判断是否重复连接
        if _, ok := users[ws]; !ok {
            users[ws] = "匿名"
        }
        userMsgsLen := len(datas.UserMsgs)
        fmt.Println("UserMsgs", userMsgsLen, "users长度:", len(users))

        //有消息时,全部分发送数据
        if userMsgsLen > 0 {
            b, errMarshl := json.Marshal(datas)
            if errMarshl != nil {
                fmt.Println("全局消息内容异常...")
                break
            }
            for key, _ := range users {
                errMarshl = websocket.Message.Send(key, string(b))
                if errMarshl != nil {
                    //移除出错的链接
                    delete(users, key)
                    fmt.Println("发送出错...")
                    break
                }
            }
            datas.UserMsgs = make([]UserMsg, 0)
        }

        fmt.Println("开始解析数据...")
        err := websocket.Message.Receive(ws, &data)
        fmt.Println("data:", data)
        if err != nil {
            //移除出错的链接
            delete(users, ws)
            fmt.Println("接收出错...")
            break
        }

        data = strings.Replace(data, "\n", "", 0)
        err = json.Unmarshal([]byte(data), &userMsg)
        if err != nil {
            fmt.Println("解析数据异常...")
            break
        }
        fmt.Println("请求数据类型:", userMsg.DataType)

        switch userMsg.DataType {
        case "send":
            //赋值对应的昵称到ws
            if _, ok := users[ws]; ok {
                users[ws] = userMsg.UserName

                //清除连接人昵称信息
                datas.UserDatas = make([]UserData, 0)
                //重新加载当前在线连接人
                for _, item := range users {

                    userData := UserData{UserName: item}
                    datas.UserDatas = append(datas.UserDatas, userData)
                }
            }
            datas.UserMsgs = append(datas.UserMsgs, userMsg)
        }
    }

}

type UserMsg struct {
    UserName string
    Msg      string
    DataType string
}

type UserData struct {
    UserName string
}

type Datas struct {
    UserMsgs  []UserMsg
    UserDatas []UserData
}

 html 코드 부분:

<!DOCTYPE html>
<html>
<head>
    <title></title>
    <meta name="viewport" content="width=device-width, initial-scale=1, maximum-scale=1, user-scalable=no">
    <!-- 新 Bootstrap 核心 CSS 文件 -->
    <link rel="stylesheet" href="//cdn.bootcss.com/bootstrap/3.3.5/css/bootstrap.min.css">
    <script src="//cdn.bootcss.com/jquery/1.11.3/jquery.min.js"></script>
    <!--        <script src="//cdn.bootcss.com/bootstrap/3.3.5/js/bootstrap.min.js"></script>-->
</head>
<body>
    <div>
        <div>内容:</div>
        <div id="divShow">
            <!--<div class="list-group-item list-group-item-success">1111</div>
            <div class="list-group-item list-group-item-info">1111</div>
            <div class="list-group-item list-group-item-warning">1111</div>
            <div class="list-group-item list-group-item-danger">1111</div>-->
        </div>
        <div id="divUsers">
            在线:<br />
            <!--<div class="btn btn-default">111</div>-->

        </div>
        <div>
            昵称:<input id="txtUserName" value="红领巾" type="text" maxlength="20" style="width: 30%; margin-bottom: 15px" />
            聊聊:<textarea id="txtContent" autofocus rows="6" placeholder="想聊的内容" maxlength="200" required style="width: 60%; "></textarea>
            <button class="btn btn-default" id="btnSend" style="margin-top:15px">发 送</button>
        </div>
    </div>
</body>
</html>

<script>

    var tool = function () {

        var paperLoopNum = 0;
        var paperTempleArr = [
            &#39;<div class="list-group-item list-group-item-success">{0}</div>&#39;,
            &#39;<div class="list-group-item list-group-item-info">{0}</div>&#39;,
            &#39;<div class="list-group-item list-group-item-warning">{0}</div>&#39;,
            &#39;<div class="list-group-item list-group-item-danger">{0}</div>&#39;
        ];

        return {

            paperDiv: function (val) {

                var hl = paperTempleArr[paperLoopNum];
                paperLoopNum++;
                if (paperLoopNum >= paperTempleArr.length) { paperLoopNum = 0; }

                return this.formart(hl, [val])
            },
            formart: function (str, arrVal) {

                for (var i = 0; i < arrVal.length; i++) {
                    str = str.replace("{" + i + "}", arrVal[i]);
                }
                return str;
            }
        }
    }

    function showMsg(id, hl, isAppend) {

        if (!isAppend) { $("#" + id).html(hl); } else {
            $("#" + id).append(hl);
        }
    }

    $(function () {

        //初始化工具方法
        var tl = new tool();

        var wsUrl = "ws://172.16.9.6:8/webSocket";
        ws = new WebSocket(wsUrl);

        try {

            ws.onopen = function () {

                //showMsg("divShow", tl.paperDiv("连接服务器-成功"));
            }

            ws.onclose = function () {
                if (ws) {
                    ws.close();
                    ws = null;
                }
                showMsg("divShow", tl.paperDiv("连接服务器-关闭"), true);
            }

            ws.onmessage = function (result) {

                //console.log(result.data);
                var data = JSON.parse(result.data);
                $(data.UserMsgs).each(function (i, item) {
                    showMsg("divShow", tl.paperDiv("【" + item.UserName + "】:" + item.Msg), true);
                });

                var userDataShow = [];
                $(data.UserDatas).each(function (i, item) {

                    userDataShow.push(&#39;<div class="btn btn-default">&#39; + item.UserName + &#39;</div>&#39;);

                });
                showMsg("divUsers", userDataShow.join(&#39;&#39;), false);
            }

            ws.onerror = function () {
                if (ws) {
                    ws.close();
                    ws = null;
                }
                showMsg("divShow", tl.paperDiv("连接服务器-关闭"), true);
            }

        } catch (e) {

            alert(e.message);
        }
        $("#btnSend").on("click", function () {

            var tContentObj = $("#txtContent");
            var tContent = $.trim( tContentObj.val()).replace("/[\n]/g", "");
            var tUserName = $.trim( $("#txtUserName").val()); tUserName = tUserName.length <= 0 ? "匿名" : tUserName;
            if (tContent.length <= 0 || $.trim(tContent).length <= 0) { alert("请输入发送内容!"); return; }
            if (ws == null) { alert("连接失败,请F5刷新页面!"); return; }

            var request = tl.formart(&#39;{"UserName": "{0}", "DataType": "{1}", "Msg": "{2}" }&#39;,
                                     [tUserName, "send", tContent]);
            ws.send(request);
            tContentObj.val("");
            tContentObj.val($.trim(tContentObj.val()).replace("/[\n]/g", ""));
        });
        $("#txtContent").on("keydown", function (event) {

            if (event.keyCode == 13) {

                $("#btnSend").trigger("click");
            }
        });
    })

</script>

 렌더링:

 

위 내용은 HTML5는 go+websocket을 사용하여 websocket 서비스 예제를 구축합니다.의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.