如何读取由特定分隔符分隔的TCP套接字数据

import socket

def read_tcp_data(sock, delimiter):
    data = b''  # 存储接收到的数据
    while True:
        chunk = sock.recv(1024)  # 接收数据
        if not chunk:
            break
        data += chunk
        while delimiter in data:
            message, data = data.split(delimiter, 1)  # 根据分隔符分割数据
            process_message(message)  # 处理数据

def process_message(message):
    # 处理接收到的消息
    print("Received message:", message)

# 创建TCP套接字并连接到目标主机
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect(('127.0.0.1', 8000))

# 读取由换行符分隔的TCP套接字数据
read_tcp_data(sock, b'\n')

# 关闭套接字连接
sock.close()