#!/usr/bin/env python
#-*- coding:utf-8 -*-
import socket
import select
import Queue
#創(chuàng)建socket對象
serversocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
#設(shè)置IP地址復(fù)用
serversocket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
#ip地址和端口號
server_address = ("127.0.0.1", 8888)
#綁定IP地址
serversocket.bind(server_address)
#監(jiān)聽，并設(shè)置最大連接數(shù)
serversocket.listen(10)
print ?"服務(wù)器啟動(dòng)成功，監(jiān)聽IP：" , server_address
#服務(wù)端設(shè)置非阻塞
serversocket.setblocking(False) ?
#超時(shí)時(shí)間
timeout = 10
#創(chuàng)建epoll事件對象，后續(xù)要監(jiān)控的事件添加到其中
epoll = select.epoll()
#注冊服務(wù)器監(jiān)聽fd到等待讀事件集合
epoll.register(serversocket.fileno(), select.EPOLLIN)
#保存連接客戶端消息的字典，格式為{}
message_queues = {}
#文件句柄到所對應(yīng)對象的字典，格式為{句柄：對象}
fd_to_socket = {serversocket.fileno():serversocket,}
while True:
? print "等待活動(dòng)連接......"
? #輪詢注冊的事件集合，返回值為[(文件句柄，對應(yīng)的事件)，(...),....]
? events = epoll.poll(timeout)
? if not events:
? ? ?print "epoll超時(shí)無活動(dòng)連接，重新輪詢......"
? ? ?continue
? print "有" , len(events), "個(gè)新事件，開始處理......"
? ?
? for fd, event in events:
? ? ?socket = fd_to_socket[fd]
? ? ?#如果活動(dòng)socket為當(dāng)前服務(wù)器socket，表示有新連接
? ? ?if socket == serversocket:
? ? ? ? ? ? connection, address = serversocket.accept()
? ? ? ? ? ? print "新連接：" , address
? ? ? ? ? ? #新連接socket設(shè)置為非阻塞
? ? ? ? ? ? connection.setblocking(False)
? ? ? ? ? ? #注冊新連接fd到待讀事件集合
? ? ? ? ? ? epoll.register(connection.fileno(), select.EPOLLIN)
? ? ? ? ? ? #把新連接的文件句柄以及對象保存到字典
? ? ? ? ? ? fd_to_socket[connection.fileno()] = connection
? ? ? ? ? ? #以新連接的對象為鍵值，值存儲(chǔ)在隊(duì)列中，保存每個(gè)連接的信息
? ? ? ? ? ? message_queues[connection] ?= Queue.Queue()
? ? ?#關(guān)閉事件
? ? ?elif event & select.EPOLLHUP:
? ? ? ? print 'client close'
? ? ? ? #在epoll中注銷客戶端的文件句柄
? ? ? ? epoll.unregister(fd)
? ? ? ? #關(guān)閉客戶端的文件句柄
? ? ? ? fd_to_socket[fd].close()
? ? ? ? #在字典中刪除與已關(guān)閉客戶端相關(guān)的信息
? ? ? ? del fd_to_socket[fd]
? ? ?#可讀事件
? ? ?elif event & select.EPOLLIN:
? ? ? ? #接收數(shù)據(jù)
? ? ? ? data = socket.recv(1024)
? ? ? ? if data:
? ? ? ? ? ?print "收到數(shù)據(jù)：" , data , "客戶端：" , socket.getpeername()
? ? ? ? ? ?#將數(shù)據(jù)放入對應(yīng)客戶端的字典
? ? ? ? ? ?message_queues[socket].put(data)
? ? ? ? ? ?#修改讀取到消息的連接到等待寫事件集合(即對應(yīng)客戶端收到消息后，再將其fd修改并加入寫事件集合)
? ? ? ? ? ?epoll.modify(fd, select.EPOLLOUT)
? ? ?#可寫事件
? ? ?elif event & select.EPOLLOUT:
? ? ? ? try:
? ? ? ? ? ?#從字典中獲取對應(yīng)客戶端的信息
? ? ? ? ? ?msg = message_queues[socket].get_nowait()
? ? ? ? except Queue.Empty:
? ? ? ? ? ?print socket.getpeername() , " queue empty"
? ? ? ? ? ?#修改文件句柄為讀事件
? ? ? ? ? ?epoll.modify(fd, select.EPOLLIN)
? ? ? ? else :
? ? ? ? ? ?print "發(fā)送數(shù)據(jù)：" , data , "客戶端：" , socket.getpeername()
? ? ? ? ? ?#發(fā)送數(shù)據(jù)
? ? ? ? ? ?socket.send(msg)
#在epoll中注銷服務(wù)端文件句柄
epoll.unregister(serversocket.fileno())
#關(guān)閉epoll
epoll.close()
#關(guān)閉服務(wù)器socket
serversocket.close()

#!/usr/bin/env python
#-*- coding:utf-8 -*-
import socket
#創(chuàng)建客戶端socket對象
clientsocket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
#服務(wù)端IP地址和端口號元組
server_address = ('127.0.0.1',8888)
#客戶端連接指定的IP地址和端口號
clientsocket.connect(server_address)
while True:
? ? #輸入數(shù)據(jù)
? ? data = raw_input('please input:')
? ? #客戶端發(fā)送數(shù)據(jù)
? ? clientsocket.sendall(data)
? ? #客戶端接收數(shù)據(jù)
? ? server_data = clientsocket.recv(1024)
? ? print '客戶端收到的數(shù)據(jù)：'server_data
? ? #關(guān)閉客戶端socket
? ? clientsocket.close()