Python中使用select模块实现非阻塞的IO


Posted in Python onFebruary 03, 2015

Socket的英文原义是“孔”或“插座”。作为BSD UNIX的进程通信机制,取后一种意思。通常也称作”套接字”,用于描述IP地址和端口,是一个通信链的句柄。在Internet上的主机一般运行了多个服务软件,同时提供几种服务。每种服务都打开一个Socket,并绑定到一个端口上,不同的端口对应于不同的服务。Socket正如其英文原意那样,像一个多孔插座。一台主机犹如布满各种插座的房间,每个插座有一个编号,有的插座提供220伏交流电, 有的提供110伏交流电,有的则提供有线电视节目。 客户软件将插头插到不同编号的插座,就可以得到不同的服务。?百度百科

socket如此重要,现在的网络编程几乎都是用的它,它起源于Unix,而Unix/Linux基本哲学之一就是“一切皆文件”,都可以用打开,读写,关闭的模式来操作。然而,对于网络服务来说,往往针对大量的客户群体,例如web,对于这类服务,必须要保证既能并行处理请求,又能保证服务的稳定。但传统的socket在处理并发方面有所欠缺,借助与select模块,能够较好的是要非阻塞的IO。

Python中的select模块以列表形式接受四个参数,分别是需要监控的可读文件对象,可写文件对象,产生异常的文件对象和超时设置,当监控的对象发生变化时,select会返回发生变化的对象列表。下面是用select实现一个简单的聊天室:

#!/usr/bin/env python
#*-* coding:utf-8 *-*
import socket
import select
import sys
import signal
class ChatServer():
  def __init__(self,host,port,timeout=10,backlog=5):
    #记录连接的客户端数量
    self.clients =0
    #存储连接的客户端socket和地址对应的字典
    self.clientmap={}
    #存储连接的客户端socket
    self.outputs = []
    #建立socket
    self.server=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    self.server.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
    self.server.bind((host,port))
    self.server.listen(backlog)
    #增加信号处理
    signal.signal(signal.SIGINT,self.sighandler) 
  def sighandler(self):
    sys.stdout.write("Shutdown Server......\n")
    #向已经连接客户端发送关系信息,并主动关闭socket
    for output in self.outputs:
      output.send("Shutdown Server")
      output.close()
    #关闭listen
    self.server.close()
    sys.stdout.flush()
  #主函数,用来启动服务器
  def run(self):
    #需要监听的可读对象
    inputs=[self.server]
    
    runing=True
    #添加监听主循环
    while runing:
      try:
        readable,writeable,exceptional = select.select(inputs,self.outputs,[])
        #此处会被select模块阻塞,只有当监听的三个参数发生变化时,select才会返回
      except select.error,e:
        break
      #当返回的readable中含有本地socket的信息时,表示有客户端正在请求连接
      if self.server in readable:
        #接受客户端连接请求
        client,addr=self.server.accept()
        sys.stdout.write("New Connection from %s\n"%str(addr))
        sys.stdout.flush()
        #更新服务器上客户端连接情况
        #1,数量加1
        #2,self.outputs增加一列
        #3,self.clientmap增加一对
        #4, 给input添加可读监控
        self.clients += 1
        self.outputs.append(client)
        self.clientmap[client]=addr
        inputs.append(client)
      
      #readable中含有已经添加的客户端socket,并且可读
      #说明 1,客户端有数据发送过来或者 2,客户端请求关闭
      elif len(readable) != 0:
        #1, 取出这个列表中的socket
        csock=readable[0]
        #2, 根据这个socket,在事先存放的clientmap中,去除客户端的地址,端口的详细信息
        host,port = self.clientmap[csock]
        #3,取数据, 或接受关闭请求,并处理
        #注意,这个操作是阻塞的,但是由于数据是在本地缓存之后,所以速度会非常快
        try:
          data = csock.recv(1024).strip()
          for cs in self.outputs:
            if cs != csock:
              cs.send("%s\n"%data)
        except socket.error,e:
          self.clients -= 1
          inputs.remove(csock)
          self.outputs.remove(csock)
          del self.clientmap[csock]
      #print self.outputs
    self.server.close()
        
if __name__ == "__main__":
  chat=ChatServer("",8008)
  chat.run()

运行这个脚本,然后用任意客户端如telnet或netcat连接8008端口,多个客户端之间就可以进行对话。

其实select模块本身是阻塞的,当需要监控的socket发生变化时,select作出返回,下面的程序会继续执行,程序根据select的返回值,对各种情况作出处理。

Python 相关文章推荐
使用Python编写一个简单的tic-tac-toe游戏的教程
Apr 16 Python
python shell根据ip获取主机名代码示例
Nov 25 Python
TensorFlow变量管理详解
Mar 10 Python
django 多数据库配置教程
May 30 Python
Python二进制串转换为通用字符串的方法
Jul 23 Python
Python实现字典排序、按照list中字典的某个key排序的方法示例
Dec 18 Python
Python字符串逆序输出的实例讲解
Feb 16 Python
Python操作Mongodb数据库的方法小结
Sep 10 Python
使用pickle存储数据dump 和 load实例讲解
Dec 30 Python
TensorFlow设置日志级别的几种方式小结
Feb 04 Python
Python爬虫获取页面所有URL链接过程详解
Jun 04 Python
python中pandas库中DataFrame对行和列的操作使用方法示例
Jun 14 Python
Python异常学习笔记
Feb 03 #Python
Python中的迭代器漫谈
Feb 03 #Python
Python描述器descriptor详解
Feb 03 #Python
理解Python中的With语句
Feb 02 #Python
Linux环境下MySQL-python安装过程分享
Feb 02 #Python
Python中用pycurl监控http响应时间脚本分享
Feb 02 #Python
Python列表(list)常用操作方法小结
Feb 02 #Python
You might like
xml+php动态载入与分页
2006/10/09 PHP
百度ping方法使用示例 自动ping百度
2014/01/26 PHP
浅谈php扩展imagick
2014/06/02 PHP
PHP输出一个等腰三角形的方法
2015/05/12 PHP
thinkphp中字符截取函数msubstr()用法分析
2016/01/09 PHP
javascript检测浏览器flash版本的实现代码
2011/12/06 Javascript
jquery mobile页面跳转后样式丢失js失效的解决方法
2014/09/06 Javascript
JS实现仿QQ效果的三级竖向菜单
2015/09/25 Javascript
在Ubuntu系统上安装Node.JS的教程
2015/10/15 Javascript
详解Javascript模板引擎mustache.js
2016/01/20 Javascript
三种带箭头提示框总结实例
2016/06/14 Javascript
JavaScript错误处理和堆栈追踪详解
2017/04/18 Javascript
mui back 返回刷新页面的实例
2017/12/06 Javascript
详解react、redux、react-redux之间的关系
2018/04/11 Javascript
在vue中获取token,并将token写进header的方法
2018/09/26 Javascript
vue 表单验证按钮事件交由父组件触发的方法
2018/12/17 Javascript
JavaScript链式调用实例浅析
2018/12/19 Javascript
[01:29:46]DOTA2上海特级锦标赛C组资格赛#1 OG VS LGD第二局
2016/02/27 DOTA
跟老齐学Python之让人欢喜让人忧的迭代
2014/10/02 Python
在Python中使用dict和set方法的教程
2015/04/27 Python
python检测空间储存剩余大小和指定文件夹内存占用的实例
2018/06/11 Python
利用PyCharm Profile分析异步爬虫效率详解
2019/05/08 Python
如何通过python的fabric包完成代码上传部署
2019/07/29 Python
解决tensorflow模型压缩的问题_踩坑无数,总算搞定
2021/03/02 Python
使用html5+css3来实现slider切换效果告别javascript+css
2013/01/08 HTML / CSS
职高毕业生自我鉴定
2013/10/21 职场文书
煤矿班组长岗位职责
2013/12/29 职场文书
美术教师自我鉴定
2014/02/12 职场文书
关于环保的建议书
2014/05/12 职场文书
2014年党的群众路线学习心得体会
2014/11/05 职场文书
护士求职自荐信范文
2015/03/04 职场文书
教师旷工检讨书
2015/08/15 职场文书
gojs实现蚂蚁线动画效果
2022/02/18 Javascript
Redis 限流器
2022/05/15 Redis
Golang gRPC HTTP协议转换示例
2022/06/16 Golang
详解CSS3浏览器兼容
2022/12/24 HTML / CSS