如何使用socket的select模型

本篇内容主要讲解“如何使用socket的select模型”，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让小编来带大家学习“如何使用socket的select模型”吧!

十多年的崇州网站建设经验，针对设计、前端、开发、售后、文案、推广等六对一服务，响应快，48小时及时工作处理。网络营销推广的优势是能够根据用户设备显示端的尺寸不同，自动调整崇州建站的显示方式，使网站能够适用不同显示终端，在浏览器中调整网站的宽度，无论在任何一种浏览器上浏览网站，都能展现优雅布局与设计，从而大程度地提升浏览体验。创新互联从事“崇州网站设计”,“崇州网站推广”以来，每个客户项目都认真落实执行。

Select模型在windows下和linux下都可以使用，更高级的epoll模型只能在linux下使用。

在开始select之前，先纠正一个错误，MsgContainer中的__check_head函数应当改为下面的样子：

def__check_head(self):
    ifself.msg_len == 0 andlen(self.msgpond) > 5:
        self.__get_msg_len()
    self.__get_msg()

这个错误竟然没有人反馈，可见，大家只是看一看，都没有实际验证或使用。。。。。

select模型其实很好理解，我们给它三个数组，数组里存放的是socket，每一次的select，模型会从这三个数组中分别挑出来可读的，可写的，发生异常的socket，并分别放入到三个数组中，这样，应用层遍历这三个数组，做相应的操作。看代码：

#coding=utf-8
import select
import socket
import sys
from MsgContainer import MsgContainer

def start_server(port):
   HOST='0.0.0.0'
   PORT=port

   server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
   #server.setblocking(False)
   server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR  , 1)
   server.bind((HOST,PORT))   #套接字绑定的IP与端口
   server.listen(10)           #开始TCP监听

   inputs = [server]           #存放需要被检测可读的socket
   outputs = []                #存放需要被检测可写的socket
   message_queues = {}         #存储可发送的数据

   while inputs:
       readable , writable , exceptional = select.select(inputs, outputs, inputs)
       # 可读
       for s in readable:
           if s is server:
               connection, client_address = s.accept()
               inputs.append(connection)
               message_queues[connection] = MsgContainer()
           else:
               data = s.recv(3)        #故意设置的这么小
               if data :
                   message_queues[s].add_data(data)
                   #已经从这个socket当中收到数据,如果你想写,那么就将其加入到outputs中,等到select模型检查它是否可写
                   if s not in outputs:
                       outputs.append(s)
               else:
                   #收到为空的数据,意味着对方已经断开连接,需要做清理工作
                   if s in outputs :
                       outputs.remove(s)
                   inputs.remove(s)
                   s.close()
                   del message_queues[s]

       #可写
       for w in writable:
           #此处一定要判断w是否真的可写,有可能w既在readable中,也在writable中,而读到的数据是空,这样其实是关闭了连接
           if not w in message_queues:
               continue
           mc = message_queues[w]
           msgs = mc.get_all_msg()
           print msgs
           for msg in msgs:
               msg = mc.pack_msg(msg)
               w.send(msg)
           mc.clear_msg()
           outputs.remove(w)
       # 异常
       for s in exceptional:
           inputs.remove(s)
           if s in outputs:
               outputs.remove(s)
           s.close()
           del message_queues[s]

if __name__ == '__main__':
   if len(sys.argv) == 2:
       port = int(sys.argv[1])
       start_server(port)
   else:
       print u'请输入端口号'

这里就几个你可能疑惑的问题做重点讲解。

发现一个可读的socket时，如何去接收数据呢？此前我提供的例子，都是在wihle循环里不停的读取，在select模型中，仍然可以这样使用，但是要注意，需要把socket设置为非阻塞的，这样才能从while循环中退出来。本篇的例子没有使用while循环，而是每次读取3个字节的数据，如果对方发过来的数据长度是9，那么第一次读取后，接收缓冲区里还有6个字节的数据可读，在下一次select操作过程中，这个socket仍然会被放入到readtable中，这样，就可以继续读剩下的数据了。

       在遍历writable数组时，一定要判断该socket是不是真的可写。我在实际测试中发现了这样的情况，一个socket既在readtable中也在writable中，但在读的时候，数据是空，对方关闭了连接，此时在去写数据，就会发生错误，所以我通过对message_queues做了一个简单的检查来判断socket是不是真的可写。

在每一次写操作执行后，都从socket从writable中删除，这样做的原因很简单，该写的数据已经写完了，如果不删除，下一次select操作时，又会把他放入到writable中，可是现在已经没有数据需要写了啊，这样做没有意义，只会浪费select操作的时间，因为它要遍历outputs中的每一个socket，判断他们是否可写以决定是否将其放入到writtable中。

       在exceptional中，是发生错误和异常的socket，有了这个数组，就在也不用操心错误和异常了，不然程序写起来非常的复杂，有了统一的管理，发生错误后的清理工作将变得非常简单。

到此，相信大家对“如何使用socket的select模型”有了更深的了解，不妨来实际操作一番吧！这里是创新互联网站，更多相关内容可以进入相关频道进行查询，关注我们，继续学习！