使用Socket与多线程实现聊天室

版本：Python3
知识点：Socket，threading

一、服务器端

1，服务器绑定地址与端口号，保持监听状态。
2，无限循环接收客户端的连接，验证身份判断是否是客户端，如果不是立即关闭连接，如果是则创建一个新线程管理对客户端的所有操作，并将其加入到连接池。
3，当客户端主动终止连接时，通知其他用户其已离开聊天室，关闭连接，从连接池里将其剔除。

运行演示.png

代码：

# encoding: utf-8
# Author: Timeashore
# Time: 2017-12-31
# Email: [email protected]

# 服务器端
import threading
import socket

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind(('127.0.0.1',5000))
s.listen(5)

print('Server',socket.gethostbyname('127.0.0.1'),'listening ...')
mydict = dict()     # fileno:nickname
mylist = list()     # 连接池列表

# 向在线人发送消息（除了发消息本人）
def send_everybody(number, message):
    for user in mylist:
        if user.fileno() != number:
            user.send(message.encode())


# 每个线程的调用方法，管理对客户端的所有操作
def fun(conn, number):
    mylist.append(conn) # 添加到连接池
    nickname = conn.recv(1024).decode()
    conn.send(('Hello,'+nickname).encode())
    mydict[number] = nickname # 添加到fileno:nickname
    print('connection',number,'has nickname : ',nickname)   # 打印链接信息
    send_everybody(number, '【系统提示：'+nickname+' 进入聊天室】')
    while True:
        try:
            conn.send(''.encode())
            getmessage = conn.recv(1024).decode()
            if getmessage:
                print(nickname,':',getmessage)
                send_everybody(number,nickname+': '+getmessage)
        except (OSError, ConnectionResetError):
            try:
                mylist.remove(conn)   # 从连接池列表剔除
            except:
                pass
            print(nickname, 'exit, ', len(mylist), ' person left')  # 服务器终端打印离开信息
            send_everybody(number, '【系统提示：' + nickname + ' 离开聊天室】')  # 告知所有人nickname已经离开聊天室
            conn.close()  # 关闭连接
            return

while True:
    conn, addr = s.accept()
    print('Accept new connection',conn.getsockname(),conn.fileno())
    try:
        verification = conn.recv(1024).decode()
        if verification == 'user':
            conn.send('welcome to server!\n'.encode())
            conn.send('nickname:\n'.encode())
            t = threading.Thread(target=fun, args=(conn, conn.fileno()))
            t.setDaemon(True)
            t.start()
        else:
            conn.send('Your verification NOT pass'.encode())
            conn.close()
    except:
        pass

二、客户端

与服务器端相比，客户端比较简单。
1，与服务器建立连接，发送身份验证信息。
2，启动两个线程，一个发送消息，一个接收消息。

用户1.png

用户2.png

代码：

# encoding: utf-8
# Author: Timeashore
# Time: 2017-12-31
# Email: [email protected]


import socket
import threading

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('127.0.0.1',5000))

s.send('user'.encode())  # 验证身份
print(s.recv(1024).decode()) # 接受欢迎消息
print(s.recv(1024).decode()) # 接受欢迎消息
nickname = input()
s.send(nickname.encode())

# 接收消息
def recv():
    while True:
        try:
            message = s.recv(1024).decode()
            if message:
                print(message)
            else:
                pass
        except ConnectionAbortedError:
            print('Server closed this connection!')
        except ConnectionResetError:
            print('Server is closed!')

# 发送消息
def send():
    while True:
        try:
            ready_send = input('')
            s.send(ready_send.encode())
        except ConnectionAbortedError:
            print('Server closed this connection!')
        except ConnectionResetError:
            print('Server is closed!')

# 启动两个线程，一个发送一个接收
t1 = threading.Thread(target=recv)
t2 = threading.Thread(target=send)

for x in [t1,t2]:
    x.setDaemon(True)
    x.start()
t1.join()
t2.join()

---

以上是在终端运行，可以使用Python GUI工具 Tkinter 给客户端增加一个界面，客户端收到的消息实时显示在界面上，需要处理线程和GUI的配合。

GUI客户端.png