tornado websocket编程 (2)： 简单购物车改进

tornado websocket编程 ： 简单购物车改进

上面例子显然有个问题：无法知道是哪些用户下了订单/取消订单。现在我们对其相关的修改
——记录用户,使其更加像一个真实的购物车应用。

可选的解决方案:
1. 客户端连接服务端时发送一个session，WebSocket类中添加一个成员变量记录客户端
session
2. 服务端WebSocket中on_open方法中调用request相关属性

在这个demo中使用方案2，至于原因呢，只要是想展示下客户端向服务端send消息的一些细
节。

修改cart.py

(如无说明，则只贴出相关修改，没贴出来部分表示没有修改, 下同)

# 引用两个新模块
from uuid import uuid4 as uuid
import json

修改IndexHandler的get方法

class Index(tornado.web.RequestHandler):
    def get(self):
        context = {
            'total': self.application.cart.total,
            # 在template渲染session数据
            session': uuid()
        }
        self.render('cart.html', **context)

实现CartStatusHandler中on_message方法，用以处理客户端发送的数据

class CartStatusHandler(tornado.websocket.WebSocketHandler):
    # 添加成员变量， 用以记录客户端发送的session
    session = ''

    # 不再在连接时注册
    def open(self):
        pass

    def on_close(self):
        self.application.cart.unregister(self)

    # 接收客户端数据，当客户端发送的数据中的status为register时注册到Cart实例中
    def on_message(self, message):
        data = json.loads(message)

        self.session = data.get('session')

        if data.get('status') == 'connected':
            if not data.get('session') == '':
                # 注意此处修改！
                self.application.cart.register(self)

    def callback(self, customer, count):
        context = {
            'customer': customer,
            'count': count
        }
        self.write_message(json.dumps(context))

修改Cart

(由于修改过多，此处贴出完整修改)

class Cart(object):
    # 记录client实例
    # 注册/取消注册/发送数据方法时调用client实例方法和属性
    clients = []

    def __init__(self):
        self.__total = 100

    def register(self, client):
        self.clients.append(client)

    def unregister(self, current_clients):
        for client in self.clients:
            if client.session == current_clients.session:
                self.clients.remove(client)

    @property
    def total(self):
        return self.__total

    def add_order(self, customer_session):
        if not self.total == 0:
            self.__total -= 1

        self.notify(customer_session)

    def cancel_order(self, customer_session):
        if not self.total == 100:
            self.__total += 1

        self.notify(customer_session)

    def notify(self, customer_session):
        for client in self.clients:
            client.callback(customer_session, self.total)

前端js修改

var goodsClient = new WebSocket('ws://127.0.0.1:8000/websocket_goods');

goodsClient.onopen = function(e) {
    var session = $('input[name="session"]'),
        data = {
            'status': 'register',
            'session': session
        };
    this.send(JSON.stringify(data));

    console.log('connected');
}

goodsClient.onmessage = function(e) {
    // 接受websocket消息
    var data = null,
        message = { data: data };

    try {
        message.data = JSON.parse(evt.data);
    } catch (e) {
        message.data = evt;
    }

    $("#total").text(message.data.count);
}