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编写百度贴吧的简单爬虫
Apr 02 Python
Python3搜索及替换文件中文本的方法
May 22 Python
以一个投票程序的实例来讲解Python的Django框架使用
Feb 18 Python
Python Pandas中根据列的值选取多行数据
Jul 08 Python
Python数据可视化 pyecharts实现各种统计图表过程详解
Aug 15 Python
关于Python3 类方法、静态方法新解
Aug 30 Python
python类中super() 的使用解析
Dec 19 Python
python识别验证码图片实例详解
Feb 17 Python
python 实现分组求和与分组累加求和代码
May 18 Python
Python如何设置指定窗口为前台活动窗口
Aug 12 Python
Pytest如何使用skip跳过执行测试
Aug 13 Python
python 如何执行控制台命令与操作剪切板
May 20 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
PHP中获取时间的下一周下个月的方法
2014/03/18 PHP
PHPMailer发送HTML内容、带附件的邮件实例
2014/07/01 PHP
Yii框架函数简单用法分析
2019/09/09 PHP
ext combobox动态加载数据库数据(附前后台)
2014/06/17 Javascript
JavaScript中函数(Function)的apply与call理解
2015/07/08 Javascript
JavaScript实现时间倒计时跳转(推荐)
2016/06/28 Javascript
终于实现了!精彩的jquery弹幕效果
2016/07/18 Javascript
jquery实现垂直和水平菜单导航栏
2020/08/27 Javascript
JS表单数据验证的正则表达式(常用)
2017/02/18 Javascript
javascript使用btoa和atob来进行Base64转码和解码
2017/03/20 Javascript
Vue组件全局注册实现警告框的实例详解
2018/06/11 Javascript
JavaScript引用类型Function实例详解
2018/08/09 Javascript
Intellij IDEA搭建vue-cli项目的方法步骤
2018/10/20 Javascript
three.js实现炫酷的全景3D重力感应
2018/12/30 Javascript
20个必会的JavaScript面试题(小结)
2019/07/02 Javascript
基于vue实现简易打地鼠游戏
2020/08/21 Javascript
[14:50]2018DOTA2亚洲邀请赛开幕式
2018/04/03 DOTA
Python使用PIL库实现验证码图片的方法
2016/03/11 Python
好用的Python编辑器WingIDE的使用经验总结
2016/08/31 Python
OpenCV2从摄像头获取帧并写入视频文件的方法
2018/08/03 Python
Python实现DDos攻击实例详解
2019/02/02 Python
Python列表(list)所有元素的同一操作解析
2019/08/01 Python
Kears+Opencv实现简单人脸识别
2019/08/28 Python
pyinstaller打包单文件时--uac-admin选项不起作用怎么办
2020/04/15 Python
python将dict中的unicode打印成中文实例
2020/05/11 Python
python实现npy格式文件转换为txt文件操作
2020/07/01 Python
Python list和str互转的实现示例
2020/11/16 Python
在PyCharm中安装PaddlePaddle的方法
2021/02/05 Python
html5摇一摇代码优化包括DeviceMotionEvent等等
2014/09/01 HTML / CSS
html2canvas生成清晰的图片实现打印的示例代码
2019/09/30 HTML / CSS
施华洛世奇巴西官网:SWAROVSKI巴西
2019/12/03 全球购物
门卫人员岗位职责
2013/12/24 职场文书
函授生自我鉴定
2014/03/25 职场文书
养生餐厅创业计划书范文
2014/03/26 职场文书
法制教育演讲稿
2014/09/10 职场文书
报名委托书
2015/01/29 职场文书