Python socket实现简单聊天室

服务端使用了select模块，实现了对多个socket的监控。客户端由于select在Windows下只能对socket使用，所以使用了多线程来实现对客户端输入和socket连接的同时监控。注意这里的socket设置为了非阻塞。这样就实现了在一个线程中同时进行socket的接收和发送。

服务器代码

# -*- coding: utf-8 -*-
import socket,select

connection_list = []
host = ''
port = 10001

def board_cast(sock,message):
    for socket in connection_list:
        if socket != server_sock and socket != sock:
            try:
                socket.send(message)
            except:
                socket.close()
                print str(socket.getpeername())+' is offline'
                connection_list.remove(socket)

server_sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
server_sock.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
server_sock.setblocking(0)
server_sock.bind((host,port))
server_sock.listen(10)
connection_list.append(server_sock)

while 1:
    readable,writable,error = select.select(connection_list,[],[])
    for sock in readable:
        if sock == server_sock:
            connection,connection_add = sock.accept()
            message = str(connection_add)+'enter room'
            board_cast(connection,message)
            print connection_add,'%s connect'
            connection_list.append(connection)
        else:
            try:
                date = sock.recv(1024)
                print date
                board_cast(sock,'('+str(sock.getpeername())+') :'+date)
            except:
                message2 = str(sock.getpeername())+ 'is offline'
                board_cast(sock,message2)
                print str(sock.getpeername())+ ' is offline'
                sock.close()
                connection_list.remove(sock)
                continue

客户端代码

# -*- coding: utf-8 -*-
import socket,threading,time
flag = 0
date = ''
lock = threading.Lock()

host = 'localhost'
port = 10001
client_sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
client_sock.setblocking(0)

class Mythread1(threading.Thread):
    def __init__(self):
        threading.Thread.__init__(self)
    def run(self):
        global flag, date
        while 1:
            date = raw_input()
            if len(date):
                lock.acquire()
                flag = 1
                lock.release()

class Mythread2(threading.Thread):
    def __init__(self):
        threading.Thread.__init__(self)
    def run(self):
        global flag
        global date
        while 1:
            try:
                buf = client_sock.recv(1024)
                if len(buf):
                    print buf
            except:
                pass
            if flag:
                try:
                    client_sock.send(date)
                except socket.error, e:
                    print e
                lock.acquire()
                flag = 0
                lock.release()



try:
    client_sock.connect((host,port))
    print"连接成功"
except socket.error,e:
    print e

t1 = Mythread1()
t2 = Mythread2()
t1.start()
t2.start()