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 深入理解yield
Sep 06 Python
python算法学习之基数排序实例
Dec 18 Python
分享几道你可能遇到的python面试题
Jul 24 Python
python基础练习之几个简单的游戏
Nov 10 Python
浅谈Django学习migrate和makemigrations的差别
Jan 18 Python
Python实现的用户登录系统功能示例
Feb 05 Python
tensorflow实现KNN识别MNIST
Mar 12 Python
解决Python3 抓取微信账单信息问题
Jul 19 Python
pycharm中显示CSS提示的知识点总结
Jul 29 Python
Pytorch 数据加载与数据预处理方式
Dec 31 Python
通过Python pyecharts输出保存图片代码实例
Nov 25 Python
Python+Appium自动化测试的实战
Jun 30 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中关于socket的系列函数总结
2015/05/18 PHP
详解如何在云服务器上部署Laravel
2017/06/30 PHP
PHP实现的AES双向加密解密功能示例【128位】
2018/09/03 PHP
PHP接入微信H5支付的方法示例
2019/10/28 PHP
js随机颜色代码的多种实现方式
2013/04/23 Javascript
javascript中Object使用详解
2015/01/26 Javascript
JavaScript中的对象与JSON
2015/07/03 Javascript
jQuery Timelinr实现垂直水平时间轴插件(附源码下载)
2016/02/16 Javascript
郁闷!ionic中获取ng-model绑定的值为undefined如何解决
2016/08/27 Javascript
VUE元素的隐藏和显示(v-show指令)
2017/06/23 Javascript
分享vue.js devtools遇到一系列问题
2017/10/24 Javascript
es6数据变更同步到视图层的方法
2019/03/04 Javascript
Promise扫盲贴
2019/06/24 Javascript
ES6如何用一句代码实现函数的柯里化
2020/01/18 Javascript
[00:57]林俊杰助阵DOTA2亚洲邀请赛
2015/01/28 DOTA
[55:03]完美世界DOTA2联赛PWL S2 LBZS vs FTD.C 第二场 11.20
2020/11/20 DOTA
纯Python开发的nosql数据库CodernityDB介绍和使用实例
2014/10/23 Python
python获得文件创建时间和修改时间的方法
2015/06/30 Python
Python实现Smtplib发送带有各种附件的邮件实例
2017/06/05 Python
浅谈机器学习需要的了解的十大算法
2017/12/15 Python
Django框架教程之正则表达式URL误区详解
2018/01/28 Python
numpy中实现二维数组按照某列、某行排序的方法
2018/04/04 Python
python爬虫刷访问量 2019 7月
2019/08/01 Python
Python集成开发工具Pycharm的安装和使用详解
2020/03/18 Python
CSS3实现闪烁动画效果的方法
2015/02/09 HTML / CSS
波兰办公用品和学校用品在线商店:Dlabiura24.pl
2020/11/18 全球购物
如何查找和删除数据库中的重复数据
2014/11/05 面试题
就业自荐信
2013/12/04 职场文书
小学学校门卫岗位职责
2014/08/03 职场文书
应用心理学专业求职信
2014/08/04 职场文书
省委召开党的群众路线教育实践活动总结大会报告
2014/10/21 职场文书
人事局接收函
2015/01/31 职场文书
人事文员岗位职责
2015/02/04 职场文书
怒海潜将观后感
2015/06/11 职场文书
《我要的是葫芦》教学反思
2016/02/18 职场文书
MySQL中TIMESTAMP类型返回日期时间数据中带有T的解决
2022/12/24 MySQL