python非阻塞多线程socket_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()