多線程原理:TCP服務器會創建一個線程池,每當有客戶端請求連接的時候,它便會從線程池中分配一個線程同客戶端建立連接,當客戶端中斷連接后,線程便銷毀。

  SocketServer 是標準庫中一個高級別的模塊。用于簡化網絡客戶與服務器的實現。

  socketserver模塊中分兩大類:server類(解決連接問題)和request類(解決通信問題)

  我們將服務器做為中轉站來處理信息,一方面與客戶端互動,另一方面進行消息轉發。

  大體思路確定下來后,需要確定一些通信規則:

  1. 客戶端與服務器建立連接后,需要輸入用戶名登入,若用戶名已存在,將reuse反饋給用戶,用戶輸出錯誤信息,退出

  2. 用戶輸入正確的用戶名后,即可進行通信了。如果未選擇通信對象,則服務器會反饋信息,提示用戶選擇通信對象

  3. 選擇通信對象的方法為,輸入to:username,如果所選擇的對象不存在,反饋錯誤信息,重新輸入

  4.當正確選擇通信對象后,雙方建立連接,通過服務器中轉信息進行通信

  5.在通信中,若發送‘quit',則結束發送消息的線程,并指示服務器該用戶準備登出,服務器刪除該用戶后,反饋消息給用戶,用戶結束接收消息的線程并退出

  6.如果A正在與C通信,此時B向A發送信息,則A的通信窗口變為與B的通信窗口,即接收到B消息后,A發出的消息不再是給C,而是默認給B

  實現代碼:

  #!/usr/bin/python

  'test TCP server'

  from socket import *

  from time import ctime

  import threading #多線程模塊

  import re #正則表達式模塊

  HOST = ''

  PORT = 21567

  BUFSIZ = 1024

  ADDR = (HOST, PORT)

  def Deal(sock, user):

  while True:

  data = sock.recv(BUFSIZ) #接收用戶的數據

  if data == 'quit': #用戶退出

  del clients[user]

  sock.send(data)

  sock.close()

  print '%s logout' %user

  break

  elif re.match('to:.+', data) is not None: #選擇通信對象

  data = data[3:]

  if clients.has_key(data):

  chatwith[sock] = clients[data]

  chatwith[clients[data]] = sock

  else:

  sock.send('the user %s is not exist' %data)

  else:

  if chatwith.has_key(sock): #進行通信

  chatwith[sock].send("[%s] %s: %s" %(ctime(), user, data))

  else:

  sock.send('Please input the user who you want to chat with')

  tcpSerSock = socket(AF_INET, SOCK_STREAM)

  tcpSerSock.bind(ADDR)

  tcpSerSock.listen(5)

  clients = {} #提供 用戶名->socket 映射

  chatwith = {} #提供通信雙方映射

  while True:

  print 'waiting for connection...'

  tcpCliSock, addr = tcpSerSock.accept()

  print '...connected from:',addr

  username = tcpCliSock.recv(BUFSIZ) #接收用戶名

  print 'The username is:',username

  if clients.has_key(username): #查找用戶名

  tcpCliSock.send("Reuse") #用戶名已存在

  tcpCliSock.close()

  else: 無錫婦科醫院排行 http://www.0510bhyy.com/

  tcpCliSock.send("Welcome!") #登入成功

  clients[username] = tcpCliSock

  chat = threading.Thread(target = Deal, args = (tcpCliSock,username)) #創建新線程進行處理

  chat.start() #啟動線程

  tcpSerSock.close()

  #!/usr/bin/python

  'test tcp client'

  from socket import *

  import threading

  HOST = 'localhost'

  PORT = 21567

  BUFSIZ = 1024

  ADDR = (HOST, PORT)

  threads = []

  def Send(sock, test): #發送消息

  while True:

  data = raw_input('>')

  tcpCliSock.send(data)

  if data == 'quit':

  break

  def Recv(sock, test): #接收消息

  while True:

  data = tcpCliSock.recv(BUFSIZ)

  if data == 'quit':

  sock.close() #退出時關閉socket

  break

  print data

  tcpCliSock = socket(AF_INET, SOCK_STREAM)

  tcpCliSock.connect(ADDR)

  print 'Please input your username:',

  username = raw_input()

  tcpCliSock.send(username)

  data = tcpCliSock.recv(BUFSIZ)

  if data == 'Reuse':

  print 'The username has been used!'

  else:

  print 'Welcome!'

  chat = threading.Thread(target = Send, args = (tcpCliSock,None)) #創建發送信息線程

  threads.append(chat)

  chat = threading.Thread(target = Recv, args = (tcpCliSock,None)) #創建接收信息線程

  threads.append(chat)

  for i in range(len(threads)): #啟動線程

  threads[i].start()

  threads[0].join() #在我們的設計中,send線程必然先于recv線程結束,所以此處只需要調用send的join,等待recv線程的結束。