Python的Tornado框架異步編程入門實例

更新時間：2015年04月24日 09:51:23 作者：C Wong

這篇文章主要介紹了Python的Tornado框架異步編程入門實例,異步編程的思維與普通編程比起來有些不同,需要的朋友可以參考下

Tornado

Tornado 是一款非阻塞可擴展的使用Python編寫的web服務(wù)器和Python Web框架, 可以使用Tornado編寫Web程序并不依賴任何web服務(wù)器直接提供高效的web服務(wù).所以Tornado不僅僅是一個web框架而且還是一款可以用于生產(chǎn)環(huán)境的高效的web服務(wù)器

Torando 在Linux和FreeBSD上使用高效的異步I/O模型 epoll 和kqueue來實現(xiàn)高效的web服務(wù)器, 所以 tornado在Linux上和FreeBSD系列性能可以達到最高
接口

當然我們可以不僅僅把Tornado看作是一個web框架和web服務(wù)器, 我們可以利用Tornado提供的接口進行高效的網(wǎng)絡(luò)異步編程,

tornado.ioloop.IOLoop 提供了三個接口可以用于網(wǎng)絡(luò)編程:

add_handler

def add_handler(self, fd, handler, events):
  self._handlers[fd] = stack_context.wrap(handler)
  self._impl.register(fd, events | self.ERROR)

add_handler用于添加socket到主循環(huán)中, 接受三個參數(shù): fd 是socket的文件描述符 handler 是處理此socket的 callback函數(shù) * events 是此socket注冊的事件

update_handler

def update_handler(self, fd, events):
  self._impl.modify(fd, events | self.ERROR)

update_handler用于更新住循環(huán)中已存在的socket響應(yīng)事件, 接受兩個參數(shù): fd 是socket對應(yīng)的文件描述符 events 是注冊的新事件

remove_handler

def remove_handler(self, fd):
  self._handlers.pop(fd, None)
  self._events.pop(fd, None)
  try:
    self._impl.unregister(fd)
  except Exception:
    gen_log.debug("Error deleting fd from IOLoop", exc_info=True)

remove_handler用于移除主循環(huán)中已存在的socket
事件

tornado.ioloop.IOLoop同時提供了4種響應(yīng)事件:
201542494640809.jpg (391×210)

實例

根據(jù)上面的接口和事件我們就可以寫出一個簡單的 echo server

#!/usr/bin/env python
# -*- coding:utf-8 -*-
#
#  Author :  cold
#  E-mail :  wh_linux@126.com
#  Date  :  13/04/15 15:08:51
#  Desc  :  Tornado Echo Server
#  HOME  :  http://www.linuxzen.com
#
import Queue
import socket

from functools import partial

from tornado.ioloop import IOLoop

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setblocking(0)       # 將socket設(shè)置為非阻塞

server_address = ("localhost", 10000)

sock.bind(server_address)
sock.listen(5)

fd_map = {}       # 文件描述符到socket的映射
message_queue_map = {}  # socket到消息隊列的映射

fd = sock.fileno()
fd_map[fd] = sock

ioloop = IOLoop.instance()

def handle_client(cli_addr, fd, event):
  s = fd_map[fd]
  if event & IOLoop.READ:
    data = s.recv(1024)
    if data:
      print "   received '%s' from %s" % (data, cli_addr)
      # 接收到消息更改事件為寫, 用于發(fā)送數(shù)據(jù)到對端
      ioloop.update_handler(fd, IOLoop.WRITE)
      message_queue_map[s].put(data)
    else:
      print "   closing %s" % cli_addr
      ioloop.remove_handler(fd)
      s.close()
      del message_queue_map[s]

  if event & IOLoop.WRITE:
    try:
      next_msg = message_queue_map[s].get_nowait()
    except Queue.Empty:
      print "%s queue empty" % cli_addr
      ioloop.update_handler(fd, IOLoop.READ)
    else:
      print 'sending "%s" to %s' % (next_msg, cli_addr)
      s.send(next_msg)

  if event & IOLoop.ERROR:
    print " exception on %s" % cli_addr
    ioloop.remove_handler(fd)
    s.close()
    del message_queue_map[s]


def handle_server(fd, event):
  s = fd_map[fd]
  if event & IOLoop.READ:
    conn, cli_addr = s.accept()
    print "   connection %s" % cli_addr[0]
    conn.setblocking(0)
    conn_fd = conn.fileno()
    fd_map[conn_fd] = conn
    handle = partial(handle_client, cli_addr[0])  # 將cli_addr作為第一個參數(shù)
    # 將連接和handle注冊為讀事件加入到 tornado ioloop
    ioloop.add_handler(conn_fd, handle, IOLoop.READ)
    message_queue_map[conn] = Queue.Queue()  # 創(chuàng)建對應(yīng)的消息隊列


ioloop.add_handler(fd, handle_server, IOLoop.READ)

ioloop.start()

上面代碼就建立了一個非阻塞的高效的異步的echo server

您可能感興趣的文章: