Python中检测Socket是否已打开及定时任务中复用Socket的方法

一、复用Socket的核心方案：全局变量持久化连接 #

在Python Azure定时器函数中，要避免每次触发都新建Socket，关键是把Socket对象放在函数外部的全局作用域——同一个函数实例的多次定时触发会共享这个全局变量。如果Azure回收了闲置的函数实例，下次触发会重新初始化全局变量，此时再新建连接即可。

调整你的代码结构，用全局变量存储Socket：

import socket
import logging
import time
import azure.functions as func

# 全局变量存储Socket连接，初始为None
global_sock = None
# 替换为你的目标服务器地址和端口
server_address = ("目标网站域名/IP", 目标端口)

def main(mytimer: func.TimerRequest) -> None:
    global global_sock
    try:
        # 检查全局Socket是否存在且可用
        if global_sock is not None and is_socket_alive(global_sock):
            logging.info("复用已建立的Socket连接")
        else:
            logging.info("新建Socket连接...")
            # 清理无效的旧连接（如果存在）
            if global_sock is not None:
                try:
                    global_sock.close()
                except Exception:
                    pass
            # 初始化并建立新连接
            global_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            global_sock.settimeout(20)
            global_sock.connect(server_address)
            global_sock.settimeout(None)  # 按需调整超时策略

        # 这里添加你的数据获取逻辑
        logging.info(f当前连接状态：{global_sock}")
        # 示例：发送请求并接收数据
        # global_sock.send(b"请求内容")
        # response = global_sock.recv(1024)

    except socket.gaierror as e:
        logging.exception(f解析服务器地址失败：{e}")
        reset_global_socket()
    except socket.error as e:
        logging.exception(f连接异常：{e}")
        reset_global_socket()
    except Exception as e:
        logging.exception(f未知错误：{e}")
        reset_global_socket()

def reset_global_socket():
    """重置全局Socket，清理无效连接"""
    global global_sock
    if global_sock is not None:
        try:
            global_sock.close()
        except Exception:
            pass
        global_sock = None

def is_socket_alive(sock: socket.socket) -> bool:
    """准确检测Socket连接是否存活"""
    if sock is None:
        return False
    try:
        # 发送空TCP心跳包（无实际数据，仅触发TCP保活机制）
        sock.send(b"", socket.MSG_DONTWAIT)
        return True
    except BlockingIOError:
        # Socket处于阻塞模式但无数据可发送，说明连接正常
        return True
    except BrokenPipeError:
        # 对方主动关闭连接
        return False
    except ConnectionResetError:
        # 连接被重置
        return False
    except socket.timeout:
        # 超时，连接已失效
        return False
    except Exception as e:
        logging.exception(f检测Socket状态出错：{e}")
        return False

二、Socket存活检测的修正说明 #

你原有的is_socket_open函数逻辑存在错误，这里做了关键调整：

• recv(16, socket.MSG_PEEK)返回空字节（len(data)==0），实际意味着对方已主动关闭连接，此时应返回False而非True。
• 用send(b"", socket.MSG_DONTWAIT)发送空心跳包是更可靠的检测方式：TCP协议会处理这个空包，连接正常则无异常；连接断开则会抛出对应错误。
• 区分阻塞/非阻塞Socket的异常场景，避免误判连接状态。

三、Azure函数的特殊注意点 #

• 实例回收：Azure函数长时间闲置会被回收，全局变量会丢失，因此每次触发必须先检查连接有效性，无效则重建。
• 多实例场景：如果定时器函数配置为多实例运行，每个实例会有独立的全局Socket，这是正常的分布式运行逻辑。
• 超时配置：根据业务需求调整Socket超时时间，避免长时间阻塞函数执行导致超时。

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