Python编写基于socket的非阻塞多人聊天室程序（单线程&多线程）

前置知识：socket非阻塞函数（socket.setblocking(False)），简单多线程编程

代码示例：

1. 单线程非阻塞版本：

服务端：

#!/usr/bin/env python
# encoding: utf-8
import socket
import os
import sys
import time
import traceback

reload(sys)
sys.setdefaultencoding("utf-8")

def socketInit(HOST,PORT):
	s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
	s.setblocking(False)
	s.bind((HOST,PORT))
	s.listen(5)
	return s

def socketAccept(s,socketAddrList):
	try:
		sT,sTAddr = s.accept()
	except Exception as e:
		pass
	else:
		socketAddrList[sT] = sTAddr[0] + ":" + str(sTAddr[1])
		print "[*]User [%s:%s] is Online" % (sTAddr[0],str(sTAddr[1]))
		sT.send("[*]Hello,This is Server >_< ...")

def socketWorking(socketAddrList):
	for sTarg in socketAddrList.keys():
		try:
			sBuffer = sTarg.recv(2048)
		except Exception as e:
			continue
		else:
			if len(sBuffer) > 0:
				t = time.ctime()
				sBuffer = "[%s %s]%s" % (t,socketAddrList[sTarg],sBuffer)
				print sBuffer
				sTarg.send(sBuffer)
			else:
				pass


def main():
	HOST = "127.0.0.1"
	PORT = 7676
	socketAddrList = {}
	s = socketInit(HOST,PORT)
	try:
		while True:
			socketAccept(s,socketAddrList)
			socketWorking(socketAddrList)
	except KeyboardInterrupt:
		sys.exit(0)


if __name__ == '__main__':
	main()

客户端：

#!/usr/bin/env python
# encoding: utf-8
import socket
import os
import sys
import time
import traceback

reload(sys)
sys.setdefaultencoding("utf-8")

def SocketInit():
	try:
		s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
	except Exception as e:
		print "[-]Socket Create Error"
		sys.exit(-1)
	else:
		print "[*]Socket Create Success"
	return s

def Conn(s,host,port):
	while 1:
		try:
			s.connect((host,port))
		except Exception as e:
			print "[-]Socket Connect Error"
			#traceback.print_exc()
			time.sleep(2)
			#sys.exit(-1)
		else:
			print "[*]Socket Connect Success"
			break

def main():
	host = "127.0.0.1"
	port = 7676
	sC = SocketInit()
	Conn(sC, host, port)
	print sC.recv(1024)
	try:
		while 1:
			Str_buffer = raw_input(host+":"+str(port)+">")
			try:
				sC.send(Str_buffer)
			except Exception as e:
				print "[-]Can not send Data"
			else:
				recv_buffer = sC.recv(1024)
				if not recv_buffer:
					print "[-]Server is Off"
				else:
					print recv_buffer
	except KeyboardInterrupt:
		sC.close()
		sys.exit(0)

if __name__ == '__main__':
	main()

2. 多线程非阻塞版本

服务端：

#!/usr/bin/env python
# encoding: utf-8
import socket
import os
import sys
import time
import traceback
from threading import Thread

reload(sys)
sys.setdefaultencoding("utf-8")

def socketInit(HOST,PORT):
	s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
	s.setblocking(False)
	s.bind((HOST,PORT))
	s.listen(5)
	return s


def socketWorking(sTarg,addr):
	while True:
		try:
			sBuffer = sTarg.recv(2048)
		except Exception as e:
			pass
		else:
			if len(sBuffer) > 0:
				t = time.ctime()
				sBuffer = "[%s %s]%s" % (t,addr[0]+":"+str(addr[1]),sBuffer)
				print sBuffer
				sTarg.send(sBuffer)
			else:
				pass
		time.sleep(0.2)

def socketAccept(s,threadsList):
	try:
		sT,addr = s.accept()
	except Exception as e:
		pass
	else:
		print "[*]User [%s:%s] is Online" % (addr[0],str(addr[1]))
		sT.send("[*]Hello,This is Server >_< ...")
		t = Thread(target=socketWorking,args=(sT,addr,))
		threadsList.append(t)
		t.start()


def main():
	HOST = "127.0.0.1"
	PORT = 7676
	threadsList = []
	s = socketInit(HOST,PORT)
	try:
		while True:
			socketAccept(s,threadsList)
	except KeyboardInterrupt:
		sys.exit(0)


if __name__ == '__main__':
	main()

客户端：

#!/usr/bin/env python
# encoding: utf-8
import socket
import os
import sys
import time
import traceback

reload(sys)
sys.setdefaultencoding("utf-8")

def SocketInit():
	try:
		s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
	except Exception as e:
		print "[-]Socket Create Error"
		sys.exit(-1)
	else:
		print "[*]Socket Create Success"
	return s

def Conn(s,host,port):
	while 1:
		try:
			s.connect((host,port))
		except Exception as e:
			print "[-]Socket Connect Error"
			#traceback.print_exc()
			time.sleep(2)
			#sys.exit(-1)
		else:
			print "[*]Socket Connect Success"
			break

def main():
	host = "127.0.0.1"
	port = 7676
	sC = SocketInit()
	Conn(sC, host, port)
	print sC.recv(1024)
	try:
		while 1:
			Str_buffer = raw_input(host+":"+str(port)+">")
			try:
				sC.send(Str_buffer)
			except Exception as e:
				print "[-]Can not send Data"
			else:
				recv_buffer = sC.recv(1024)
				if not recv_buffer:
					print "[-]Server is Off"
				else:
					print recv_buffer
	except KeyboardInterrupt:
		sC.close()
		sys.exit(0)

if __name__ == '__main__':
	main()