Python Socket中的recv方法用法介绍|极客教程

Python Socket中的recv方法用法介绍

1. 简介

Socket是一种在计算机网络中实现通信的方式，它提供了一种可靠的、面向连接的通信方式，可以用于在不同的计算机之间传输数据。Python的socket库为我们提供了方便的接口，可以轻松实现网络编程。

recv方法是Python Socket库中常用的方法之一，用于接收数据。本文将详细介绍recv方法的使用，包括参数的含义、使用方法以及示例代码。

2. `recv`方法的基本使用

recv方法用于从Socket对象接收数据，其基本语法如下：

recv(bufsize)

参数说明：

bufsize：表示一次接收的最大字节数。

recv方法返回接收到的数据，以字节串的形式进行返回。如果没有数据可接收，recv方法将会阻塞。

下面是一个简单的示例，演示了如何使用recv方法：

import socket

# 创建Socket对象
s = socket.socket()

# 连接远程服务器
s.connect(('www.example.com', 80))

# 接收数据
data = s.recv(1024)
print(data)

# 关闭连接
s.close()

在这个示例中，我们首先创建了一个Socket对象，并使用connect方法连接到www.example.com的80端口。然后使用recv方法接收最多1024个字节的数据，并将结果打印出来。最后关闭连接。

3. `recv`方法的返回值

recv方法返回接收到的数据，以字节串的形式进行返回。需要注意的是，返回的数据可能少于请求的字节数。

为了处理这种情况，我们可以使用一个循环来接收所有数据。下面是一个示例代码，演示了如何使用recv方法接收所有数据：

import socket

# 创建Socket对象
s = socket.socket()

# 连接远程服务器
s.connect(('www.example.com', 80))

# 接收数据
data = b''  # 创建一个空字节串
while True:
    chunk = s.recv(1024)  # 一次接收1024个字节的数据
    if not chunk:  # 如果没有数据可接收，退出循环
        break
    data += chunk
print(data)

# 关闭连接
s.close()

在这个示例中，我们使用一个循环来接收所有数据，每次接收1024个字节的数据，并将其追加到data变量中。循环退出的条件是recv方法返回一个空字节串，表示没有数据可接收。

4. `recv`方法的示例

下面是更多示例代码，演示了recv方法在不同场景下的使用。

示例1：接收固定长度的数据

import socket

# 创建Socket对象
s = socket.socket()

# 绑定端口
s.bind(('localhost', 8888))

# 监听请求
s.listen(5)

# 等待客户端连接
client, address = s.accept()

# 接收固定长度的数据
data = client.recv(10)
print(data)

# 关闭连接
client.close()
s.close()

在这个示例中，我们创建了一个服务器端的Socket对象，并绑定到本地主机的8888端口。然后等待客户端的连接，并接受客户端发送的10个字节的数据。

示例2：接收指定分隔符的数据

import socket

# 创建Socket对象
s = socket.socket()

# 绑定端口
s.bind(('localhost', 8888))

# 监听请求
s.listen(5)

# 等待客户端连接
client, address = s.accept()

# 接收指定分隔符的数据
data = b''  # 创建一个空字节串
while True:
    chunk = client.recv(1024)
    if b'\r\n' in chunk:
        data += chunk[:chunk.index(b'\r\n')]
        break
    data += chunk
print(data)

# 关闭连接
client.close()
s.close()

在这个示例中，我们接收了包含指定分隔符\r\n的数据。循环接收数据，直到接收到包含分隔符的数据。同时，我们使用index方法找到分隔符的位置，并将分隔符之前的数据保存到data变量中。

示例3：接收指定长度的数据并超时处理

import socket

# 创建Socket对象
s = socket.socket()

# 连接远程服务器
s.connect(('www.example.com', 80))

# 设置超时时间为5秒
s.settimeout(5)

# 接收指定长度的数据
try:
    data = s.recv(1024)
    print(data)
except socket.timeout:
    print('接收超时')

# 关闭连接
s.close()

在这个示例中，我们将连接超时时间设置为5秒，如果在规定时间内没有接收到数据，将抛出socket.timeout异常。

示例4：接收数据直到连接关闭

import socket

# 创建Socket对象
s = socket.socket()

# 绑定端口
s.bind(('localhost', 8888))

# 监听请求
s.listen(5)

# 等待客户端连接
client, address = s.accept()

# 接收数据直到连接关闭
try:
    while True:
        data = client.recv(1024)
        if not data:
            break
        print(data)
except ConnectionResetError:
    print('连接被重置')

# 关闭连接
client.close()
s.close()

在这个示例中，我们接收数据直到连接关闭。我们使用一个循环来接收数据，当recv方法返回一个空字节串时，表示连接已关闭，退出循环。

示例5：接收大量数据的进度显示

import socket
import sys

# 创建Socket对象
s = socket.socket()

# 连接远程服务器
s.connect(('www.example.com', 80))

# 接收大量数据的进度显示
received_bytes = 0
total_bytes = 1000000
while received_bytes < total_bytes:
    chunk = s.recv(1024)
    received_bytes += len(chunk)
    progress = received_bytes / total_bytes * 100
    sys.stdout.write(f'\r已接收：{int(progress)}%')
    sys.stdout.flush()
print('\n数据接收完成')

# 关闭连接
s.close()

在这个示例中，我们用来模拟接收一大块数据，并显示接收进度。我们使用两个变量来记录已接收的字节数和总字节数。每次接收数据后，更新进度并打印到终端。