Python Azure Function App SQL Server连接复用问题咨询

首先直接回答你的核心疑问：你当前的代码是会复用SQL Server连接的，但存在一些需要优化的点，下面给你逐步拆解：

1. 当前代码的连接复用逻辑 #

在Azure Functions的Python运行时中，函数的模块（也就是你的__init__.py）是在进程启动时加载一次的。模块级别的代码（比如你写的connection = getConnection()）只会执行一次，之后同一个进程内的所有函数触发（也就是处理Service Bus消息的main函数调用）都会复用这个已经初始化好的connection变量。

那连接会不会在Function运行期间持续保持？

• 理论上，只要承载函数的进程没有被回收，连接就会保持。但要注意两个情况：
  • SQL Server本身有连接超时设置（默认一般是30分钟左右），如果函数长时间没有触发，连接可能会被SQL Server主动断开。
  • Azure Functions的宿主会在函数空闲一段时间后回收进程（默认是20分钟），这时候下次触发会重新加载模块，创建新的连接。

不过你的当前代码没有处理连接断开的情况，如果连接失效后，后续的insert操作会直接报错，这是需要优化的地方。

2. 如何正确实现Python Function的连接复用 #

要实现可靠的连接复用，除了模块级初始化连接，还要加入连接有效性检查和自动重连的逻辑，同时可以考虑使用连接池来更高效地管理连接。这里给你两种优化方案：

方案一：在使用连接前检查有效性，失效则重建 #

修改你的代码，在每次使用连接前先检查是否可用，如果不可用就重新创建：

import azure.functions as func
import pyodbc  # 假设你用pyodbc连接SQL Server
import logging

# 模块级变量存储连接
connection = None

def getConnection():
    # 这里替换成你的SQL Server连接字符串
    conn_str = "你的连接字符串"
    return pyodbc.connect(conn_str)

def check_and_reconnect():
    global connection
    if connection is None:
        connection = getConnection()
    else:
        try:
            # 执行一个简单的查询检查连接是否有效
            cursor = connection.cursor()
            cursor.execute("SELECT 1")
            cursor.fetchone()
        except pyodbc.Error:
            # 连接失效，重新创建
            logging.warning("Connection lost, reconnecting...")
            connection = getConnection()

def main(msg: func.ServiceBusMessage):
    # 先检查并确保连接有效
    check_and_reconnect()
    
    mess = msg.get_body().decode("utf-8")
    logging.info(mess)
    message = eval(str(mess))  # 注意：eval有安全风险，建议用json.loads替代
    data = handle_message_from_device(message)
    insert(connection, data)

方案二：使用连接池（推荐） #

如果你的SQL操作比较频繁，使用连接池会比单连接更高效，比如用pyodbc的连接池，或者SQLAlchemy的引擎（它会自动管理连接池）：

import azure.functions as func
from sqlalchemy import create_engine
import logging

# 模块级初始化SQLAlchemy引擎（自动管理连接池）
engine = create_engine("mssql+pyodbc:///?odbc_connect=你的连接字符串")

def main(msg: func.ServiceBusMessage):
    mess = msg.get_body().decode("utf-8")
    logging.info(mess)
    message = eval(str(mess))
    data = handle_message_from_device(message)
    
    # 使用引擎获取连接，用完自动归还到连接池
    with engine.connect() as conn:
        insert(conn, data)

3. 关于主文件（init.py）的调用疑问 #

你担心“新消息触发时会重新调用主文件”，其实这个理解是不对的：

• 当Function第一次被触发（冷启动），或者进程被回收后再次触发，宿主会加载__init__.py模块，执行所有模块级代码（包括初始化连接）。
• 之后同一个进程内的所有消息触发，只会调用main函数，不会重新执行模块级代码，所以连接会被复用。
• 只有当进程被销毁（比如空闲超时、自动缩放创建新进程），才会重新加载模块创建新连接。

最后提醒一下：eval(str(mess))存在安全风险，如果消息是不可信的，可能会执行恶意代码，建议用json.loads(mess)来解析JSON格式的消息。

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