socket实现简易的聊天室

前言

最近在学习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() 阻塞。

运行结果不展示了，截图太麻烦。