如何使用socket的select模型

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

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

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

def __check_head(self):
    if self.msg_len == 0 and len(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模型”有了更深的了解,不妨来实际操作一番吧!这里是亿速云网站,更多相关内容可以进入相关频道进行查询,关注我们,继续学习!

原创文章,作者:kepupublish,如若转载,请注明出处:https://blog.ytso.com/220164.html

(0)
上一篇 2022年1月2日
下一篇 2022年1月2日

相关推荐

发表回复

登录后才能评论