beego + websocket 向页面推送数据(一)

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/u012210379/article/details/72901387

我已经用了3年的beego,虽然beego的版本在不断更新,但当别人谈论到它时,我越来越“冷静“,它对新手很友好,同时也发现了各种限制,尝试不同的框架还是好的。
Go语言的伟大之处在于:如果你也不是写网站出身,没有时间去学tomcat的配置,看到Java厚重的SSH框架书籍就脑袋疼,看到python群里各种2、3版本争论不休,Go可以在最短的时间内把自己的服务器搭起来,在最短的时间内把自己的工资提起来。


参考链接

如果你把上面3个链接都看完 下面就不用看了

我们开始试着把上面提到的文章内的代码拆开再自己组合起来

  • 目的,使用Gorilla WebSocket而不是官方的x/net下的websocket,毕竟要往beego项目里集成
  • 只要把数据通过websocket的方式显示到页面上就算成功
  • 先实现个简单版,不往项目里扔

效果

这里写图片描述

项目结构
这里写图片描述

main.go

package main

import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
    "time"
)

var clients = make(map[*websocket.Conn]bool)
var broadcast = make(chan Message)

var upgrader = websocket.Upgrader{}

type Message struct {
    Message string `json:"message"`
}

func main() {
    fs := http.FileServer(http.Dir("public"))
    http.Handle("/", fs)

    http.HandleFunc("/ws", handleConnections)

    go handleMessages()

    log.Println("http server started on :8000")
    err := http.ListenAndServe(":8000", nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

//注册成为 websocket
func handleConnections(w http.ResponseWriter, r *http.Request) {

    ws, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Fatal(err)
    }
    defer ws.Close()

    clients[ws] = true

    //不断的从页面上获取数据 然后广播发送出去
    for {
        //将从页面上接收数据改为不接收 直接发送
        //var msg Message
        //err := ws.ReadJSON(&msg)
        //if err != nil {
        //  log.Printf("error: %v", err)
        //  delete(clients, ws)
        //  break
        //}

        //目前存在问题 定时效果不好 需要在业务代码替换时改为beego toolbox中的定时器
        time.Sleep(time.Second * 3)
        msg := Message{Message: "这是向页面发送的数据 " + time.Now().Format("2006-01-02 15:04:05")}
        broadcast <- msg
    }
}

//广播发送至页面
func handleMessages() {
    for {
        msg := <-broadcast
        for client := range clients {
            err := client.WriteJSON(msg)
            if err != nil {
                log.Printf("client.WriteJSON error: %v", err)
                client.Close()
                delete(clients, client)
            }
        }
    }
}

index.html

<!DOCTYPE html>
<html>
<head>
    <meta charset="UTF-8"/>
    <title>Sample of websocket with golang</title>
    <script src="http://apps.bdimg.com/libs/jquery/2.1.4/jquery.min.js"></script>

    <script>
        $(function() {
            var ws = new WebSocket('ws://' + window.location.host + '/ws');
            ws.onmessage = function(e) {
                $('<li>').text(event.data).appendTo($ul);
            };
            var $ul = $('#msg-list');
        });
    </script>
</head>
<body>
<ul id="msg-list"></ul>
</body>
</html>

是不是代码简单到爆炸?从上面你可以看到,在go文件中核心代码就2句

ws, err := upgrader.Upgrade(w, r, nil)
err := client.WriteJSON(msg)

一句注册,一句发送,搞定,收工

展开阅读全文

没有更多推荐了,返回首页