python利用select实现的Socket Server

# 利用python的select模块实现简单的Socket Sever

#实现多用户访问，再次基础上可以实现FTP Server应用程序

# 发布目的，在于解决了客户端强行终止时，服务器端也跟着程序终止

# 程序的关键在：读就是读，写就是写 ，不要读写混着来

# 代码如下：

Server

 __author__ = 'Stone'
 # -*- coding: UTF-8 -*-
 # !/usr/bin/env python3
 import socket
 import queue
 import select
 HOST = '0.0.0.0'
 PORT = 8000
 s = socket.socket()
 s.bind((HOST,PORT))
 s.listen(500)
 # 设定关闭程序后，能马上释放服务器的端口，供后续程序使用
 s.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
 s.setblocking(0) # 设置为非阻塞模式
 # 设置列表，新建、发送、异常
 inputs = []
 outputs = []
 #exceptions = []
 # 存放信息
 msg_dic = {}
 inputs.append(s)  # 把自身加进去，侦听是否有新连接
 while True:
     readable,writable,exceptional = select.select(inputs,outputs,inputs)  # select 会阻塞socket
22
     for r in readable:
         if r is s:
             # 说明有新连接过来
             conn,addr = s.accept()
             print("新连接:",addr)
             conn.setblocking(0) # 把单个连接也设置为非阻塞模式，比如：某个连接接收的文件比较大，将会一直占用着（别的程序没机会处理）；设置为非阻塞后，可以等着下次for循环继续接收
             inputs.append(conn)
             msg_dic[conn] = queue.Queue()  # 为每个连接创建消息队列
         else:
             # 说明有连接是活动的
                 try:
                     data = r.recv(1024)
                     if data:
                         print("接收到了数据:",data.decode('utf-8'))
                         # 放进消息队列
                         msg_dic[r].put(data)
                         if r not in outputs:
                             outputs.append(r)  # 并放入发送数据列表
                     else:
                         # 连接断开
                         print("客户端断开连接")
                         if r in outputs:
                             outputs.remove(r)
                         inputs.remove(r)
                         del msg_dic[r]
                 except socket.error:     # 解决问题（出现客户端异常断开，服务器也跟着断开），后面不做任何处理，留给exceptional做处理
                     pass

     for w in writable:
         try:
             send_msg = msg_dic[w].get_nowait()
         except queue.Empty:
             print("client [%s]"% w.getpeername()[0],"queue is empty...")
             outputs.remove(w)
         else:
             print("sending message to [%s]"% w.getpeername()[0],send_msg)
             w.send(send_msg)
             outputs.remove(w)  # 防止再次执行时，发生empty的异常

     for e in exceptional:
         if e in outputs:
             outputs.remove(e)
         inputs.remove(e)
         e.close()
         del msg_dic[e]
 s.close()

Client:

 __author__ = 'Stone'
 # -*- coding: UTF-8 -*-
 # !/usr/bin/env python3
 import socket
 HOST = 'localhost'
 PORT = 8000
 s_client = socket.socket()
 s_client.connect((HOST,PORT))
 while True:
     data = input('>>:').strip()
     if not data:
         continue
     s_client.send(data.encode('utf-8'))
     recv = s_client.recv(1024)
     print(recv)
     #break      # 测试，作为客户端自动离开
 s_client.close()