Python網(wǎng)絡(luò)編程使用select實(shí)現(xiàn)socket全雙工異步通信功能示例

本文實(shí)例講述了Python網(wǎng)絡(luò)編程使用select實(shí)現(xiàn)socket全雙工異步通信功能。分享給大家供大家參考，具體如下：
在前面一篇《Python網(wǎng)絡(luò)編程之TCP套接字簡(jiǎn)單用法》中，我們實(shí)現(xiàn)了tcp客戶端與服務(wù)器的通信，但是功能十分局限，發(fā)送消息與接收消息不能同時(shí)進(jìn)行。
接下來我將通過select這個(gè)模塊，來實(shí)現(xiàn)全雙工通信（隨時(shí)可以接收信息以及發(fā)送信息），當(dāng)然，用多線程也可以完成，這是后話。
那么，select為何物？
select  -在單線程網(wǎng)絡(luò)服務(wù)中器程序中，管理多個(gè)套接字連接
select的原型為(rlist,wlist,xlist[,timeout])，其中rlist是等待讀取的對(duì)象，wlist是等待寫入的對(duì)象，xlist是等待異常的對(duì)象，最后一個(gè)是可選對(duì)象，指定等待的時(shí)間，單位是s.
select()方法的返回值是準(zhǔn)備好的對(duì)象的三元組，若在timeout的時(shí)間內(nèi)，沒有對(duì)象準(zhǔn)備好，那么返回值將是空的列表。
它采用的是輪詢的方式來實(shí)現(xiàn)異步通信的。
在下面程序中，目前主要支持1對(duì)1通信，當(dāng)雙方任何一方發(fā)送字符串‘88'時(shí)，表示結(jié)束通信。
下面我們來看看具體實(shí)現(xiàn)：
首先是服務(wù)器。    
#!/usr/bin/python
'test TCP server'
from socket import *
from time import ctime
import select
import sys
HOST = ''
PORT = 21567
BUFSIZ = 1024
ADDR = (HOST, PORT)
tcpSerSock = socket(AF_INET, SOCK_STREAM)
tcpSerSock.bind(ADDR)
tcpSerSock.listen(5)
input = [tcpSerSock, sys.stdin]   #input是一個(gè)列表，初始有歡迎套接字以及標(biāo)準(zhǔn)輸入
while True:
  print 'waiting for connection...'
  tcpCliSock, addr = tcpSerSock.accept()
  print '...connected from:',addr
  input.append(tcpCliSock)  #將服務(wù)套接字加入到input列表中
  while True:
    readyInput,readyOutput,readyException = select.select(input,[],[]) #從input中選擇，輪流處理client的請(qǐng)求連接（tcpSerSock），client發(fā)送來的消息(tcpCliSock)，及服務(wù)器端的發(fā)送消息(stdin)
    for indata in readyInput:
      if indata==tcpCliSock:  #處理client發(fā)送來的消息
        data = tcpCliSock.recv(BUFSIZ)
        print data
        if data=='88':
          input.remove(tcpCliSock)
          break
      else:       #處理服務(wù)器端的發(fā)送消息
        data = raw_input('>')
        if data=='88':
          tcpCliSock.send('%s' %(data))
          input.remove(tcpCliSock)
          break
        tcpCliSock.send('[%s] %s' %(ctime(), data))
    if data=='88':
      break
  tcpCliSock.close()
tcpSerSock.close()
以下是客戶端的代碼，非常類似，只是與服務(wù)器相比不需要處理請(qǐng)求信息而已。
#!/usr/bin/python
'test tcp client'
from socket import *
from time import ctime
import select
import sys
HOST = 'localhost'
PORT = 21567
BUFSIZ = 1024
ADDR = (HOST, PORT)
tcpCliSock = socket(AF_INET, SOCK_STREAM)
tcpCliSock.connect(ADDR)
input = [tcpCliSock,sys.stdin]
while True:
  readyInput,readyOutput,readyException = select.select(input,[],[])
  for indata in readyInput:
    if indata==tcpCliSock:
      data = tcpCliSock.recv(BUFSIZ)
      print data
      if data=='88':
        break
    else:
      data = raw_input('>')
      if data=='88':
        tcpCliSock.send('%s' %(data))
        break
      tcpCliSock.send('[%s] %s' %(ctime(), data))
  if data=='88':
    break
tcpCliSock.close()
目前為止，已經(jīng)實(shí)現(xiàn)了一對(duì)一全雙工聊天通信啦。
下面當(dāng)然要思考下，如何實(shí)現(xiàn)多對(duì)多通信呢？
我們知道，一個(gè)服務(wù)器可以為多個(gè)客戶端服務(wù)，即，服務(wù)器與客戶端本身有著一對(duì)多的關(guān)系，那么，我們是否可以利用服務(wù)器作為中間人，用來傳達(dá)信息，從而實(shí)現(xiàn)多對(duì)多通信呢？
舉個(gè)例子，A想要和B通信，則A將信息發(fā)送給服務(wù)器，然后再通過服務(wù)器轉(zhuǎn)發(fā)給B。按照這個(gè)思路去寫，完成多對(duì)多通信應(yīng)該是可以的，有興趣的可以嘗試一下。