前言
最近在学习socket的时候,想要做一个简易的聊天室。在网上看了很多的代码、文章都不胜求解,再经过2天的努力之后终于基本实现了,以供与我有相同问题的朋友参考。
前置知识
在实现这个简易聊天室时需要的简易知识有:socket模块、threading模块和select模块。
开始
实现一个简易的聊天室需要两个文件,server.py和client.py,由简至繁,我们先从client.py模块开始。
client客户端
client功能较为简单:发送消息和接收消息,因为我们需要实现发送和接收同时进行,所以需要用到threading多线程模块。
#实现步骤:
#1 连接服务器
#2 使用两个线程实现接收消息和发送消息
import socket
import threading
def send(sock,nic): ##发送信息
while True:
msg = input("发送信息:")
msg = nic+':'+msg
sock.send(msg.encode('utf-8'))
def rev(sock): #接收信息
while True:
msg = sock.recv(1024)
print(msg.decode('utf-8'))
if __name__ == '__main__':
try: #异常处理,避免服务器端断开连接报错
sock = socket.socket();
sock.connect((socket.gethostname(),9991)) #服务器端ip和端口
nic = input('输入昵称:')
sock.send(nic.encode('utf-8'))
threading.Thread(target=send,args=(sock,nic,)).start() #两个线程实现并行处理
threading.Thread(target=rev, args=(sock,)).start()
except:
sock.close()
server服务器
服务器端需要实现的功能是信息转发
import socket
import sys
import threading
from select import select
from tkinter import *
server = socket.socket()
server.bind((socket.gethostname(),9991))
server.listen(5)
clients = [server,]
print('xxx的聊天室')
while True:
cls,x,y = select(clients,[],[]) #监听套接字模块,用法网上查
for c in cls:
if c is server: #如果是server服务器被监听说明有新客户端连接
client, addr = server.accept()
print('欢迎' + client.recv(1024).decode('utf-8') + '来到聊天室')
clients.append(client) #将新客户端加入监听列表
else: #表示客户端被监听
try: #异常处理,避免客户端退出导致服务器端退出
data = c.recv(1024)
print(data.decode('utf-8'))
for r in clients[1:]: #循环所有的客户端
if r != c:
r.send(data)
except:
c.close()
clients.remove(c)
关于为什么要使用select模块,假如不使用的话会出现,客户端只能被监听一次,即用户只能给服务器发送一条消息。这是由于server.accept() 处发生了阻塞,阻止程序向下运行。
此外,除了使用select模块实现多路IO监听,还可以使用多线程的方法来绕过server.accept() 阻塞。
运行结果不展示了,截图太麻烦。