Python Socket 实现简陋的聊天室模拟

⭐ 简介：大家好，我是zy阿二，我是一名对知识充满渴望的自由职业者。
☘️ 最近我沉溺于Python的学习中。你所看到的是我的学习笔记。
❤️ 如果对你有帮助，请关注、点赞，让我们共同进步。有不足之处请留言指正！

1. 将服务器使用的方法详细说明.

套接字(Socket)，就是对网络中不同主机上的应用进程之间进行双向通信的端点的抽象。一个套接字就是网络上进程通信的一端，提供了应用层进程利用网络协议交换数据的机制。从所处的地位来讲，套接字上联应用进程，下联网络协议栈，是应用程序通过网络协议进行通信的接口，是应用程序与网络协议根进行交互的接口。

2. 描述tcp服务器开发流程.

• 实例化socket

• 绑定端口

• 监听客户端

• 接收客户端报文

• 返回报文给客户端

• socket通讯服务端开发完毕

3. 你的hacker小伙伴潜入了内网，不能连接外部网络，作为接头人，你需要尽快搭建一个服务器用于接收小伙伴的信息。接头暗号为：力拔山兮气盖世，时不利兮骓不逝。当接收到暗号后，请及时回复消息哦~

4.	import socket
5.	# 这是进行定义一个ip协议版本AF_INET（IPv4），定义一个传输TCP协议，SOCK_STREAM
6.	# socket.AF_INET 表示的是IPv4（这也是socket里默认的）
7.	# socket.AF_INET6 表示的是IPv6
8.	# socket.AF_UNIX 这个主要用于单一的UNIX进程间的通信
9.	sk = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
10.	
11.	# 定义ip地址与端口号，ip地址就是服务端的ip地址，端口随便定义，但要与客户端脚本一致
12.	# 本机IP在终端中输入 ipconfig  以太网中的 IPv4地址 就是本机ip
13.	sk.bind(('', 9000))
14.	
15.	# 监听一个端口,这里的数字是一个常量，表示阻塞连接的数量，也就是最大等待数量
16.	sk.listen(99)
17.	
18.	# 接受客户端的数据，并返回两个参数，info为连接信息，ip为客户端的ip地址与端口号
19.	info, ip = sk.accept()
20.	
21.	while True:
22.	    # 客户端发送的数据存储在recv里，1024指最大接受数据的量,单位字节，byte
23.	    data = info.recv(1024)
24.	    # 客户端传过来的数据是二进制的，所以需要解码，根据客户端那边的编码方式解码。
25.	    # 即客户端那边的字符串是用utf-8编码的，那么这边就也需要用utf-8进行解码。
26.	    data = data.decode('utf-8')
27.	    # 同时回传过去的内容也需要转换成二进制
28.	
29.	    if data == '力拔山兮气盖世，时不利兮骓不逝':
30.	        message = "暗号正确".encode('utf-8')
31.	        # 返回数据给 客户端
32.	        info.send(message)
33.	        # 关闭socket连接
34.	        sk.close()
35.	        break

4.你们成功的建立起了链接，其他的hacker也需要加入你的服务器，请搭建一个支持多人同步聊天的服务器，并显示所有加入你的服务器的客户端的ip地址和端口号。

"""整体思路
1. 需要群聊，那么服务端就需要把所有客户端发送过来的消息转发给所有客户端。
2. 服务端需要有3个核心功能，1. 监听客户端连接，2. 监听客户端信息，3. 转发客户端信息。
3. 因为要保证可以同时监听接受和转发，所以至少需要2个线程。
4. 但是由于多线程无法有效的保存数据，所以需要用队列存储客户端发来的消息。
5. 最后检查列队不为空,那么就说明有消息，那么就遍历所有已连接的对象，进行消息分发
"""

import socket
import queue
import threading
import time

sk = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sk.bind((socket.gethostname(), 9090))  # socket.gethostname() 方法 取到本机 IPv4 地址
# 限制最大连接客户端数量。5
sk.listen(5)

# 存放已连接的对象
client_list = []
# 存放消息
q = queue.Queue()


def init():
    """初始化，客户端建立连接
    """
    while True:
        client, addr = sk.accept()  # 阻塞线程
        if client in client_list:
            print('~~老用户~~')
        else:
            # 打印新用户的IP
            print(f'~~新用户:{client.getpeername()[0]}')
            client.send('欢迎来到聊天室!'.encode('utf-8'))
            client_list.append(client)
            r = threading.Thread(target=receive_msg, args=(client,))
            r.start()


def receive_msg(client):
    """建立链接后开始监听该客户端的消息

    Args:
        client (socket.socket()): _description_
    """
    while True:
        time.sleep(1)
        try:
            if client in client_list:
                message = client.recv(1024).decode('utf-8')
                # if not message   等于   if message !=""
                if message != '':
                    print(f'{client}发来消息：{message}')
                    q.put(message)

                elif client in client_list:
                    print("用户退出了")
                    client_list.remove(client)

        except BaseException as error:
            print('一个连接中断了')
            if client in client_list:
                client_list.remove(client)


def broad_cast():
    """群发消息，判断已连接的客户端的list不为空和队列不为空
    """
    while True:
        if client_list and not q.empty():
            # get_nowait方法，避免堵塞。
            data = q.get_nowait()
            if data != '':  # 判断消息不为空
                for client in client_list:
                    client.send((f'广播：{data}'.encode('utf-8')))
                    print('服务器转发了消息')


if __name__ == "__main__":
    t1 = threading.Thread(target=init)
    t2 = threading.Thread(target=broad_cast)
    t1.start()
    t2.start()
    # 主线程监听在线人数
    while (True):
        print("当前在线人数为:%d" % (len(client_list)))
        time.sleep(5)

客户端代码：

import socket
import threading


sk = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
host = socket.gethostname()
print(host)
sk.connect((host, 9090))


# 接受消息
def receive():
    while True:
        data = sk.recv(1024).decode('utf-8')
        if data != '':
            print(data)


# 发送消息
def send_msg():
    while (True):
        message = input('请输入消息:')
        if message == 'bye':
            sk.close()
            break
        sk.send(message.encode('utf-8'))


if __name__ == "__main__":
    # 开一个独立的子线程用来接收消息
    t1 = threading.Thread(target=receive, daemon=True)
    t1.start()
    # 主线程用来发消息
    send_msg()