python 的 websocket的使用

一：websocket使用場景

https://django-websocket-redis.readthedocs.io/en/latest/running.html

最典型的使用場景：聊天。

還有就是：當我們做異步處理的任務的時候，之前采用的長輪詢或者計時器的方法，但是這種方法的開銷和性能比較差。對於網頁的性能的不是很好。

而html5的時候 支持websocket和python 本身也支持websocket，由於websocket的特性和開銷更小，性能更好。所以采用websocket方式。接下來的django中，會開發一個叫做channels。這種方式更為不錯。

今天介紹websocket。至於websocket的好處，網上的介紹一大堆，可以自己百度看，不做累述。

二  使用:

簡單介紹下代碼：前端通過jquery的$() 來采集頁面的IP信息，然后通過html5的websocket跟后端python通信，后端通過傳過的IP信息，建立ansible任務，執行對應的采集腳本，隔幾秒返回給前端，

前端將結果替換到相應的頁面，來更新前端的監控信息。

1、python 安裝websocket：

pip2.7  install dwebsocket

 

前端代碼：

if (window.s) {
                window.s.close()
            }
            /*創建socket連接*/
            var socket = new WebSocket("ws://192.168.31.206:1601/get_monitor/");
            socket.onopen = function () {
                console.log('WebSocket open');//成功連接上Websocket
                socket.send(msg)
            };
            socket.onmessage = function (e) {
                data=JSON.parse(e.data);  
                console.log('message: ' + e.data);//打印出服務端返回過來的數據
                $.each(data['success_ip'],
                                  function(key,val){
                                     $( 'td:contains('+key+')').siblings().find('.cpu_val').text(val[0]);
                                     $( 'td:contains('+key+')').siblings().find('.io_val').text(val[2]+'m');
                                     $( 'td:contains('+key+')').siblings().find('.disk_val').text(val[1]+'%');
                                  }

                )
            };
            // Call onopen directly if socket is already open
            if (socket.readyState == WebSocket.OPEN) socket.onopen();
            window.s = socket;


    }

 

后端代碼：

@accept_websocket
def get_monitor(request):#get monior infomation func.
    import time 
    if  request.is_websocket():
        while True:
            if request.websocket.has_messages():
                msg=request.websocket.read()
                ip,task_id=msg.split(':')
                ip_list=[]
                ip_list.append(ip)
                print ip_list,task_id
                host_obj=models.Host.objects.filter(task_id=task_id).first()
                user=host_obj.sys_user
                pwd=host_obj.sys_pwd
                while True:
                    time.sleep(2)
                    task_obj=task_excu.Task_Runner(ip_list, user, pwd)
                    ret=task_obj.get_monitor()
                    for i,v in ret['success_ip'].items():
                        v=v.split('\n')
                        ret['success_ip'][i]=v
                        request.websocket.send(json.dumps(ret))
            else:
                time.sleep(1)
                continue

 注釋：

      通過裝飾器：@accept_websocket來讓view函數的即可以接受http協議也可以接收wss 協議及websocket。

      通過while死循環來，不斷等待接收前端發送的消息和發送消息。

      通過time模塊來控制循環速度。

      前端通過websocket 給后端發送消息和接收消息。