django入门进阶11_websocket

<script type="text/javascript">
    var socket = new WebSocket("ws:" + window.location.host + "/drug/drug_connect/");
    socket.onopen = function () {
        console.log('WebSocket open');//成功连接上Websocket
        socket.send('adasdasda。。。。');//发送数据到服务端
    };
    socket.onmessage = function (e) {
        console.log('message: ' + e.data);//打印服务端返回的数据
    };
    socket.onclose=function(e){
      console.log(e);
      socket.close(); //关闭TCP连接
    };
    if (socket.readyState == WebSocket.OPEN) socket.onopen();
</sctipt>

@accept_websocket
def start_server_script(request):
    if request.is_websocket():
        # "这里实现wbsocket连接逻辑"
        for info in request.websocket:# 这里需要注意的是，这个for后面的对象，request.websocket是阻塞的，也就是说如果对方发送消息,info=新消息，循环走一圈，如果对方不发消息，for这里会“卡住”，这里容易忽略
            request.websocket.send("你刚才对我说:%s"%info)
         print('这里其实不会被执行')
    else:
        # "这里实现http连接逻辑"
        pass

wobsocket_map=dict()
@accept_websocket
def start_server_script(request):
    if request.is_websocket():
        wobsocket_map[id(request.websocket)]=request.websocket
        # "这里实现wbsocket连接逻辑"
        for info in request.websocket:# 这里需要注意的是，这个for后面的对象，request.websocket是阻塞的，也就是说如果对方发送消息,info=新消息，循环走一圈，如果对方不发消息，for这里会“卡住”，这里容易忽略
            [websocket.send("你刚才对我说:%s"%info) for websocket in wobsocket_map.values()]
         print('这里其实不会被执行')
    else:
        # "这里实现http连接逻辑"
        pass

# 伪代码，不保证能跑，意思差不多
# 其他函数通过next(),or .send()调用此生成器，不断生成各个视频帧的base64编码（看不懂的话，百度"yield"）
url='rtmp://xxxx'
@accept_websocket
def show_video(request):
    video=video_img_base64(url)
    base64=next(video)
    while base64 is not None: # 这个就是类似死循环的东西（当然这个并非真正死循环，但很多情况退出只能放到循环里break，这里只能采用while True）
        request.websocket.send({'img_base64':base64})
        base64=video.send(None)

def video_img_base64(url):
    cap=cv2.VideoCapture(url)
    ret,frame=cap.read()
    while ret:
        yield base64(frame.tobytes()).decode('utf8')
        ret,frame=cap.read()
    else:
        yield None　#调用方一旦收到None，避免调用next()　or send()否则抛出异常，需要做异常处理

send_queue=Queue()
@accept_websocket
def websocket_ctrl(request):
    if request.is_websocket:
        while True:# 由于使用同步方式处理异步，所以这里必然死循环
            if request.websocket.count_messages()>0:
                message=request.websocket.read()
                while message:
                    onmessage(message)
                    message=request.websocket.read()
           　if not send_queue.empty:# 需向send_queue放东西，send函数中会放message
           　    messsage=send_queue.pop()
           　    while message:
                    message=request.websocket.send(message)
                    messsage=send_queue.pop()

def onmessage(message):
    pass # 这里可以放你想在收到消息时做的事情,类似异步方法的onmessage，如果需要反馈修改，比如改变上一级的执行逻辑，则可以通过返回值的方式，传递给上一级调用者，让上一级调用者通过返回值调整自身执行逻辑

# 他人调用这个方法，将消息加入发送队列，在websocket_ctrl的循环中会取得，并发出
def send(message):
   send_queue.put(message)

for message in request.websocket:
    print(message)