Python基于TCP实现聊天功能
Server端
import socket
import queue
import threading
import time
#socket.AF_INET指定使用IPv4地址族,socket.SOCK_STREAM指定使用TCP协议
serversocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
host = socket.gethostname()
print("服务器IP: "+socket.gethostbyname(host))
serversocket.bind((host, 9090))
serversocket.listen(5)
#存放已连接的对象
clients = []
#存放公共消息的容器
public_message = dict()
#接收新的对象
def init():
while True:
# 当心客户端连接时
client,addr = serversocket.accept() #阻塞线程
if client in clients:
print('老用户')
else:
print('新的用户加入:',end='')
print(client.getpeername()[0])
client.send(bytes('欢迎来到聊天室(匿名)!'.encode('utf-8')))
clients.append(client)
#线程启动时会调用receive_msg函数,args为传递给函数的参数元组
r = threading.Thread(target=receive_msg,args=(client,))
r.start()
#接收消息
def receive_msg(client):
while True:
time.sleep(1)
try:
if client in clients:
data = client.recv(1024).decode('utf-8')
if data!='':
print(data)
public_message[client] = queue.Queue()
public_message[client].put(data)
else:
if client in clients:
print("用户优雅的退出了")
clients.remove(client)
except BaseException as error:
print('用户强制中断了一个连接')
# print('错误:',error)
if client in clients:
clients.remove(client)
#转发消息(非/阻塞)
def broadcast():
while True:
if len(clients)>1:
public_message_clone = [i for i in public_message] #解决字典迭代中操作报错的问题
for client in clients:
for i in public_message_clone:
if i!=client and public_message[i].empty()==False:
data = public_message[i].get_nowait() #注意
if data !='':
# 转发消息给客户端
client.send(bytes(data.encode('utf-8')))
print('服务器转发了消息')
t1 = threading.Thread(target=init)
t2 = threading.Thread(target=broadcast)
t1.start()
t2.start()
#主线程监听在线人数
while (True):
print("当前在线人数为:%d"%(len(clients)))
time.sleep(5)
Client
import socket,threading
#客户端想要发消息和收消息同时进行,需要使用多线程达到并发效果
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 获取计算机name
host = socket.gethostname()
# 获取根据计算机name获取计算机IP
host = socket.gethostbyname(host);
print(host)
s.connect((host, 9090))
# 接收消息
def receive():
while True:
data = s.recv(1024).decode('utf-8')
if data!='':
print(data)
# 发送消息
def send_msg():
while (True):
msg = input(':')
if msg=='exit':
s.close()
break
s.send(bytes(msg.encode('utf-8')))
t1 = threading.Thread(target=receive,daemon=True)
t1.start()
send_msg()
如果这篇文章对你有用,可以关注本人微信公众号获取更多ヽ(^ω^)ノ ~
![微信公众号二维码](https://images.cnblogs.com/cnblogs_com/aeolian/1679458/o_wechat_gzh_qrcode.jpg)