如何在函数内部判断其返回值是否被调用方需要？

场景说明 #

假设有两种调用drawPoint函数的场景：
直接调用，不接收返回值：

drawPoint(1,2)

调用后将返回值赋值给变量：

newPointId = drawPoint(1,2)

该函数会发起HTTP请求，服务器响应状态不确定（立即返回、延迟、长时间等待或无响应）。需求明确：

• 当调用方不需要返回值时，采用**发后即弃（fire-and-forget）**模式：发送请求后立即返回无关值，超时设为0，无需等待响应。
• 当调用方需要返回值时，采用**阻塞（blocking）**模式：设置合理超时时间，搭配重试机制；若超时未收到响应，返回False，确保后续代码能正常使用返回值变量。

实现方案 #

在Python中，可以通过inspect模块分析调用栈的字节码，判断当前函数的返回值是否会被调用方使用。核心思路是查看调用帧的下一条字节码指令，判断是否存在存储返回值的操作（比如STORE_FAST、STORE_NAME等）。

以下是完整的代码实现：

import inspect
import opcode

def drawPoint(x, y):
    # 判断返回值是否被调用方需要
    frame = inspect.currentframe().f_back
    bytecode = frame.f_code.co_code
    offset = frame.f_lasti
    # 获取下一条指令的操作码
    next_opcode = bytecode[offset + 2] if offset + 2 < len(bytecode) else None
    # 检查是否是存储类指令（表示返回值会被保存）
    return_value_needed = next_opcode in (opcode.opmap['STORE_FAST'], 
                                         opcode.opmap['STORE_NAME'],
                                         opcode.opmap['STORE_GLOBAL'],
                                         opcode.opmap['STORE_ATTR'])
    
    if return_value_needed:
        # 阻塞模式：带超时和重试的HTTP请求
        print("执行阻塞请求，等待响应...")
        # 替换为实际HTTP请求逻辑，示例返回有效ID
        return "point_123"
    else:
        # 发后即弃模式：无超时，发送后立即返回
        print("执行发后即弃请求，无需等待响应...")
        # 可通过异步线程处理请求，避免阻塞当前流程
        return None

注意事项 #

• 该实现依赖Python字节码机制，不同版本的字节码指令可能存在细微差异，需按需测试适配。
• 针对复杂调用场景（如返回值作为表达式参数、传递给其他函数），需扩展判断逻辑，增加对CALL_FUNCTION、BINARY_OP等指令的检查。
• 发后即弃模式建议用异步线程/进程处理请求，避免占用当前线程资源。

内容的提问来源于stack exchange，提问作者Tom Grundy