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()