使用Bottle实现websocket和http同端口复用

arklearn

因为工作时候，有需求，同一个端口去接收websocket数据和http请求。
这个需求让我愁得掉了好些头发，没有办法硬着头皮去查资料。
一开始一发狠，用socket来做，反正http和websocket实际上也是一种socket，拼了。
后面发现脑袋大了，这是一个无底洞啊，时间有限，不符合python开发。pyhton里有那多框架，完全不用再造轮子。
突然，Bottle框架出现在我的眼前，忽如一夜春风来，千树万树梨花开，哈哈！！
bottle框架只是搭起了一个架子，实际websocket 还是用了gevent，这个用来解决websocket的。

[Python] 纯文本查看 复制代码

from gevent.pywsgi import WSGIServer
from geventwebsocket import WebSocketError
from geventwebsocket.handler import WebSocketHandler

部分代码，附件里还加了线程：

[Python] 纯文本查看 复制代码

'''
Bottle服务，绑定websocket和http服务
'''

class MyBottleServer(Bottle):
    def __init__(self, ip="127.0.0.1", port=8737):
        Bottle.__init__(self)

        self.ip = ip
        self.port = port

    def __del__(self):
        try:
            if (self.server != None):
                self.server.close_server()
        except:
            pass

    def start_server(self):
        self.server = WSGIServer((self.ip, self.port), self, handler_class=WebSocketHandler)
        self.server.serve_forever()

    def close_server(self):
        if (self.server != None):
            self.server.close()
            self.server = None

    def register_websocket_server(self):
        @self.route('/wspost')
        def handle_websocket():
            wsock = request.environ.get('wsgi.websocket')
            if not wsock:
                abort(400, '此功能只能处理websocket')
            message = ""
            while True:
                try:
                    if wsock.closed:
                        break
                    message = wsock.receive()

                    result = {
                        "flag": False,
                        "msg": ""
                    }
                    if message == None or message.strip() == "":
                        msg = u"获取websocket消息失败"
                        result["msg"] = msg
                        break
                    
                    ### 处理其他业务

                    if wsock.closed != True:
                        wsock.close()
                    break

                except WebSocketError, ex:
                    logging.error(u"WebSocket异常信息")
                    logging.exception(ex)
                    # 如果异常报错，则断开连接
                    break

    def register_http_server(self):
        @self.route('/')
        def index():
            return template('<b>Hello {{name}}</b>!', name="xx")

        @self.route('/datapost', method="POST")
        def do_printform():
            response.headers['Access-Control-Allow-Origin'] = '*'
            response.headers['Access-Control-Allow-Methods'] = 'GET,POST,PUT,OPTIONS'
            allow_headers = 'Referer, Accept, Origin, User-Agent'
            response.headers[
                'Access-Control-Allow-Headers'] = "Content-Type,Content-Length, Authorization, Accept,X-Requested-With"
            response.content_type = "application/javascript; charset=utf-8"

            message = request.params.get("message")
            if message is None:
                message = ""
            try:
                message = urllib.unquote(message);
            except Exception, ex:
                pass

            ##处理单据业务

            time.sleep(5)

    def register_error(self):
        @self.error(404)
        def error404(error):
            return u'404'

    def response_header(self):

        # set cross headers
        headers = {'Content-type': 'application/json'}
        headers['Access-Control-Allow-Origin'] = '*'
        headers['Access-Control-Allow-Methods'] = 'GET,POST,PUT,OPTIONS'
        allow_headers = 'Referer, Accept, Origin, User-Agent'
        headers['Access-Control-Allow-Headers'] = allow_headers

我把代码上传到附件里，去除工作所用的业务逻辑，只留下框架的实现，希望能给大家带来思路。

因为工作时间的关系，没有优化，感觉后面可以去掉bottle框架，只使用gevent框架。